1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
| import { useCallback } from 'react'
| import produce from 'immer'
| import type { Variable } from '@/app/components/workflow/types'
|
| type Params<T> = {
| inputs: T
| setInputs: (newInputs: T) => void
| varKey?: string
| }
| function useVarList<T>({
| inputs,
| setInputs,
| varKey = 'variables',
| }: Params<T>) {
| const handleVarListChange = useCallback((newList: Variable[] | string) => {
| const newInputs = produce(inputs, (draft: any) => {
| draft[varKey] = newList as Variable[]
| })
| setInputs(newInputs)
| }, [inputs, setInputs, varKey])
|
| const handleAddVariable = useCallback(() => {
| const newInputs = produce(inputs, (draft: any) => {
| draft[varKey].push({
| variable: '',
| value_selector: [],
| })
| })
| setInputs(newInputs)
| }, [inputs, setInputs, varKey])
| return {
| handleVarListChange,
| handleAddVariable,
| }
| }
|
| export default useVarList
|
|