mirror of
https://github.com/langgenius/dify.git
synced 2026-02-03 17:41:35 +08:00
56 lines
1.4 KiB
TypeScript
56 lines
1.4 KiB
TypeScript
import produce from 'immer'
|
|
import type { HumanInputNodeType, Timeout, UserAction } from './types'
|
|
import useNodeCrud from '@/app/components/workflow/nodes/_base/hooks/use-node-crud'
|
|
import {
|
|
useNodesReadOnly,
|
|
} from '@/app/components/workflow/hooks'
|
|
const useConfig = (id: string, payload: HumanInputNodeType) => {
|
|
const { nodesReadOnly: readOnly } = useNodesReadOnly()
|
|
const { inputs, setInputs } = useNodeCrud<HumanInputNodeType>(id, payload)
|
|
|
|
const handleUserActionAdd = (newAction: UserAction) => {
|
|
setInputs({
|
|
...inputs,
|
|
userActions: [...inputs.userActions, newAction],
|
|
})
|
|
}
|
|
|
|
const handleUserActionChange = (updatedAction: UserAction) => {
|
|
const newActions = produce(inputs.userActions, (draft) => {
|
|
const index = draft.findIndex(a => a.id === updatedAction.id)
|
|
if (index !== -1)
|
|
draft[index] = updatedAction
|
|
})
|
|
setInputs({
|
|
...inputs,
|
|
userActions: newActions,
|
|
})
|
|
}
|
|
|
|
const handleUserActionDelete = (actionId: string) => {
|
|
const newActions = inputs.userActions.filter(action => action.id !== actionId)
|
|
setInputs({
|
|
...inputs,
|
|
userActions: newActions,
|
|
})
|
|
}
|
|
|
|
const handleTimeoutChange = (timeout: Timeout) => {
|
|
setInputs({
|
|
...inputs,
|
|
timeout,
|
|
})
|
|
}
|
|
|
|
return {
|
|
readOnly,
|
|
inputs,
|
|
handleUserActionAdd,
|
|
handleUserActionChange,
|
|
handleUserActionDelete,
|
|
handleTimeoutChange,
|
|
}
|
|
}
|
|
|
|
export default useConfig
|