FEAT: NEW WORKFLOW ENGINE (#3160)

Co-authored-by: Joel <iamjoel007@gmail.com>
Co-authored-by: Yeuoly <admin@srmxy.cn>
Co-authored-by: JzoNg <jzongcode@gmail.com>
Co-authored-by: StyleZhang <jasonapring2015@outlook.com>
Co-authored-by: jyong <jyong@dify.ai>
Co-authored-by: nite-knite <nkCoding@gmail.com>
Co-authored-by: jyong <718720800@qq.com>

web/app/components/workflow/nodes/variable-assigner/use-config.ts

@@ -0,0 +1,81 @@
+import { useCallback, useEffect, useState } from 'react'
+import produce from 'immer'
+import useVarList from './components/var-list/use-var-list'
+import type { VariableAssignerNodeType } from './types'
+import useNodeCrud from '@/app/components/workflow/nodes/_base/hooks/use-node-crud'
+import type { ValueSelector, Var } from '@/app/components/workflow/types'
+import { BlockEnum, VarType } from '@/app/components/workflow/types'
+import {
+  useNodesReadOnly,
+  useWorkflow,
+} from '@/app/components/workflow/hooks'
+
+const useConfig = (id: string, payload: VariableAssignerNodeType) => {
+  const { nodesReadOnly: readOnly } = useNodesReadOnly()
+  const { inputs, setInputs } = useNodeCrud<VariableAssignerNodeType>(id, payload)
+  const { getBeforeNodeById } = useWorkflow()
+  const beforeNodes = getBeforeNodeById(id)
+
+  useEffect(() => {
+    if (beforeNodes.length !== 1 || inputs.variables.length > 0)
+      return
+    const beforeNode = beforeNodes[0]
+    if (beforeNode.data.type === BlockEnum.KnowledgeRetrieval) {
+      const newInputs = produce(inputs, (draft: VariableAssignerNodeType) => {
+        draft.output_type = VarType.array
+        draft.variables[0] = [beforeNode.id, 'result']
+      })
+      setInputs(newInputs)
+    }
+  // eslint-disable-next-line react-hooks/exhaustive-deps
+  }, [beforeNodes, inputs.variables])
+
+  const handleOutputTypeChange = useCallback((outputType: string) => {
+    const newInputs = produce(inputs, (draft: VariableAssignerNodeType) => {
+      draft.output_type = outputType as VarType
+    })
+    setInputs(newInputs)
+  }, [inputs, setInputs])
+
+  const { handleVarListChange, handleAddVariable } = useVarList({
+    id,
+    inputs,
+    setInputs,
+  })
+
+  const { variables } = inputs
+  const [currVarIndex, setCurrVarIndex] = useState(-1)
+  const currVar = variables[currVarIndex]
+  const handleOnVarOpen = useCallback((index: number) => {
+    setCurrVarIndex(index)
+  }, [])
+  const filterVar = useCallback((varPayload: Var, valueSelector: ValueSelector) => {
+    const type = varPayload.type
+    if ((inputs.output_type !== VarType.array && type !== inputs.output_type) || (
+      inputs.output_type === VarType.array && ![VarType.array, VarType.arrayString, VarType.arrayNumber, VarType.arrayObject].includes(type)
+    ))
+      return false
+
+    // can not choose the same node
+    if (!currVar)
+      return true
+
+    const selectNodeId = valueSelector[0]
+
+    if (selectNodeId !== currVar[0] && variables.find(v => v[0] === selectNodeId))
+      return false
+
+    return true
+  }, [currVar, inputs.output_type, variables])
+  return {
+    readOnly,
+    inputs,
+    handleOutputTypeChange,
+    handleVarListChange,
+    handleAddVariable,
+    handleOnVarOpen,
+    filterVar,
+  }
+}
+
+export default useConfig