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/end/default.ts

@@ -0,0 +1,33 @@
+import { BlockEnum } from '../../types'
+import type { NodeDefault } from '../../types'
+import { type EndNodeType } from './types'
+import { ALL_CHAT_AVAILABLE_BLOCKS, ALL_COMPLETION_AVAILABLE_BLOCKS } from '@/app/components/workflow/constants'
+
+const nodeDefault: NodeDefault<EndNodeType> = {
+  defaultValue: {
+    outputs: [],
+  },
+  getAvailablePrevNodes(isChatMode: boolean) {
+    const nodes = isChatMode
+      ? ALL_CHAT_AVAILABLE_BLOCKS
+      : ALL_COMPLETION_AVAILABLE_BLOCKS.filter(type => type !== BlockEnum.End)
+    return nodes
+  },
+  getAvailableNextNodes() {
+    return []
+  },
+  checkValid(payload: EndNodeType) {
+    let isValid = true
+    let errorMessages = ''
+    if (payload.type) {
+      isValid = true
+      errorMessages = ''
+    }
+    return {
+      isValid,
+      errorMessage: errorMessages,
+    }
+  },
+}
+
+export default nodeDefault

web/app/components/workflow/nodes/end/node.tsx

@@ -0,0 +1,93 @@
+import type { FC } from 'react'
+import React from 'react'
+import type { EndNodeType } from './types'
+import type { NodeProps, ValueSelector, Variable } from '@/app/components/workflow/types'
+import { isSystemVar, toNodeOutputVars } from '@/app/components/workflow/nodes/_base/components/variable/utils'
+import {
+  useIsChatMode,
+  useWorkflow,
+} from '@/app/components/workflow/hooks'
+import { VarBlockIcon } from '@/app/components/workflow/block-icon'
+import { Line3 } from '@/app/components/base/icons/src/public/common'
+import { Variable02 } from '@/app/components/base/icons/src/vender/solid/development'
+import { BlockEnum, VarType } from '@/app/components/workflow/types'
+
+const Node: FC<NodeProps<EndNodeType>> = ({
+  id,
+  data,
+}) => {
+  const { getBeforeNodesInSameBranch } = useWorkflow()
+  const availableNodes = getBeforeNodesInSameBranch(id)
+  const isChatMode = useIsChatMode()
+  const outputVars = toNodeOutputVars(availableNodes, isChatMode)
+
+  const startNode = availableNodes.find((node: any) => {
+    return node.data.type === BlockEnum.Start
+  })
+
+  const getNode = (id: string) => {
+    return availableNodes.find(node => node.id === id) || startNode
+  }
+
+  const getVarType = (nodeId: string, value: ValueSelector) => {
+    const targetVar = outputVars.find(v => v.nodeId === nodeId)
+    if (!targetVar)
+      return 'undefined'
+
+    let type: VarType = VarType.string
+    let curr: any = targetVar.vars
+    const isSystem = isSystemVar(value);
+    (value).slice(1).forEach((key, i) => {
+      const isLast = i === value.length - 2
+      curr = curr.find((v: any) => v.variable === isSystem ? `sys.${key}` : key)
+      if (isLast) {
+        type = curr.type
+      }
+      else {
+        if (curr.type === VarType.object)
+          curr = curr.children
+      }
+    })
+    return type
+  }
+  const { outputs } = data
+  const filteredOutputs = (outputs as Variable[]).filter(({ value_selector }) => value_selector.length > 0)
+
+  if (!filteredOutputs.length)
+    return null
+
+  return (
+    <div className='mb-1 px-3 py-1 space-y-0.5'>
+      {filteredOutputs.map(({ value_selector }, index) => {
+        const node = getNode(value_selector[0])
+        const isSystem = isSystemVar(value_selector)
+        const varName = isSystem ? `sys.${value_selector[value_selector.length - 1]}` : value_selector[value_selector.length - 1]
+        return (
+          <div key={index} className='flex items-center h-6 justify-between bg-gray-100 rounded-md  px-1 space-x-1 text-xs font-normal text-gray-700'>
+            <div className='flex items-center text-xs font-medium text-gray-500'>
+              <div className='p-[1px]'>
+                <VarBlockIcon
+                  className='!text-gray-900'
+                  type={node?.data.type || BlockEnum.Start}
+                />
+              </div>
+              <div className='max-w-[75px] truncate'>{node?.data.title}</div>
+              <Line3 className='mr-0.5'></Line3>
+              <div className='flex items-center text-primary-600'>
+                <Variable02 className='w-3.5 h-3.5' />
+                <div className='max-w-[50px] ml-0.5 text-xs font-medium truncate'>{varName}</div>
+              </div>
+            </div>
+            <div className='text-xs font-normal text-gray-700'>
+              <div className='ml-0.5 text-xs font-normal text-gray-500 capitalize'>{getVarType(node?.id || '', value_selector)}</div>
+
+            </div>
+          </div>
+        )
+      })}
+
+    </div>
+  )
+}
+
+export default React.memo(Node)

web/app/components/workflow/nodes/end/panel.tsx

@@ -0,0 +1,49 @@
+import { type FC } from 'react'
+import React from 'react'
+import { useTranslation } from 'react-i18next'
+import useConfig from './use-config'
+import type { EndNodeType } from './types'
+import VarList from '@/app/components/workflow/nodes/_base/components/variable/var-list'
+import Field from '@/app/components/workflow/nodes/_base/components/field'
+import AddButton from '@/app/components/base/button/add-button'
+import { type NodePanelProps } from '@/app/components/workflow/types'
+
+const i18nPrefix = 'workflow.nodes.end'
+
+const Panel: FC<NodePanelProps<EndNodeType>> = ({
+  id,
+  data,
+}) => {
+  const { t } = useTranslation()
+
+  const {
+    readOnly,
+    inputs,
+    handleVarListChange,
+    handleAddVariable,
+  } = useConfig(id, data)
+
+  const outputs = inputs.outputs
+  return (
+    <div className='mt-2'>
+      <div className='px-4 pb-4 space-y-4'>
+
+        <Field
+          title={t(`${i18nPrefix}.output.variable`)}
+          operations={
+            !readOnly ? <AddButton onClick={handleAddVariable} /> : undefined
+          }
+        >
+          <VarList
+            nodeId={id}
+            readonly={readOnly}
+            list={outputs}
+            onChange={handleVarListChange}
+          />
+        </Field>
+      </div>
+    </div>
+  )
+}
+
+export default React.memo(Panel)

web/app/components/workflow/nodes/end/types.ts

@@ -0,0 +1,5 @@
+import type { CommonNodeType, Variable } from '@/app/components/workflow/types'
+
+export type EndNodeType = CommonNodeType & {
+  outputs: Variable[]
+}

web/app/components/workflow/nodes/end/use-config.ts

@@ -0,0 +1,27 @@
+import useVarList from '../_base/hooks/use-var-list'
+import type { EndNodeType } 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: EndNodeType) => {
+  const { nodesReadOnly: readOnly } = useNodesReadOnly()
+  const { inputs, setInputs } = useNodeCrud<EndNodeType>(id, payload)
+
+  const { handleVarListChange, handleAddVariable } = useVarList<EndNodeType>({
+    inputs,
+    setInputs: (newInputs) => {
+      setInputs(newInputs)
+    },
+    varKey: 'outputs',
+  })
+
+  return {
+    readOnly,
+    inputs,
+    handleVarListChange,
+    handleAddVariable,
+  }
+}
+
+export default useConfig

web/app/components/workflow/nodes/end/utils.ts

@@ -0,0 +1,5 @@
+import type { EndNodeType } from './types'
+
+export const checkNodeValid = (payload: EndNodeType) => {
+  return true
+}