dify/web/app/components/workflow/nodes/code/use-single-run-form-params.ts
Joel 1a1bfd4048
feat: last run frontend (#21369)
The frontend of feat: Persist Variables for Enhanced Debugging Workflow (#20699).

Co-authored-by: jZonG <jzongcode@gmail.com>
2025-06-24 09:10:30 +08:00

66 lines
1.7 KiB
TypeScript

import type { MutableRefObject } from 'react'
import type { InputVar, Variable } from '@/app/components/workflow/types'
import { useCallback, useMemo } from 'react'
import useNodeCrud from '../_base/hooks/use-node-crud'
import type { CodeNodeType } from './types'
type Params = {
id: string,
payload: CodeNodeType,
runInputData: Record<string, any>
runInputDataRef: MutableRefObject<Record<string, any>>
getInputVars: (textList: string[]) => InputVar[]
setRunInputData: (data: Record<string, any>) => void
toVarInputs: (variables: Variable[]) => InputVar[]
}
const useSingleRunFormParams = ({
id,
payload,
runInputData,
toVarInputs,
setRunInputData,
}: Params) => {
const { inputs } = useNodeCrud<CodeNodeType>(id, payload)
const varInputs = toVarInputs(inputs.variables)
const setInputVarValues = useCallback((newPayload: Record<string, any>) => {
setRunInputData(newPayload)
}, [setRunInputData])
const inputVarValues = (() => {
const vars: Record<string, any> = {}
Object.keys(runInputData)
.forEach((key) => {
vars[key] = runInputData[key]
})
return vars
})()
const forms = useMemo(() => {
return [
{
inputs: varInputs,
values: inputVarValues,
onChange: setInputVarValues,
},
]
}, [inputVarValues, setInputVarValues, varInputs])
const getDependentVars = () => {
return payload.variables.map(v => v.value_selector)
}
const getDependentVar = (variable: string) => {
const varItem = payload.variables.find(v => v.variable === variable)
if (varItem)
return varItem.value_selector
}
return {
forms,
getDependentVars,
getDependentVar,
}
}
export default useSingleRunFormParams