dify

2025-12-01 17:21:38 +08:00
parent 32fee2b8ab
commit fab8c13cb3
7511 changed files with 996300 additions and 0 deletions
--- a/dify/web/app/components/workflow/nodes/assigner/node.tsx
+++ b/dify/web/app/components/workflow/nodes/assigner/node.tsx
@@ -0,0 +1,82 @@
+import type { FC } from 'react'
+import React from 'react'
+import useNodes from '@/app/components/workflow/store/workflow/use-nodes'
+import { useTranslation } from 'react-i18next'
+import type { AssignerNodeType } from './types'
+import { isSystemVar } from '@/app/components/workflow/nodes/_base/components/variable/utils'
+import { BlockEnum, type Node, type NodeProps } from '@/app/components/workflow/types'
+import {
+  VariableLabelInNode,
+} from '@/app/components/workflow/nodes/_base/components/variable/variable-label'
+import Badge from '@/app/components/base/badge'
+
+const i18nPrefix = 'workflow.nodes.assigner'
+
+const NodeComponent: FC<NodeProps<AssignerNodeType>> = ({
+  data,
+}) => {
+  const { t } = useTranslation()
+  const nodes: Node[] = useNodes()
+  if (data.version === '2') {
+    const { items: operationItems } = data
+    const validOperationItems = operationItems?.filter(item =>
+      item.variable_selector && item.variable_selector.length > 0,
+    ) || []
+
+    if (validOperationItems.length === 0) {
+      return (
+        <div className='relative flex flex-col items-start gap-0.5 self-stretch px-3 py-1'>
+          <div className='flex flex-col items-start gap-1 self-stretch'>
+            <div className='flex items-center gap-1 self-stretch rounded-md bg-workflow-block-parma-bg px-[5px] py-1'>
+              <div className='system-xs-medium flex-1 text-text-tertiary'>{t(`${i18nPrefix}.varNotSet`)}</div>
+            </div>
+          </div>
+        </div>
+      )
+    }
+    return (
+      <div className='relative flex flex-col items-start gap-0.5 self-stretch px-3 py-1'>
+        {operationItems.map((value, index) => {
+          const variable = value.variable_selector
+          if (!variable || variable.length === 0)
+            return null
+          const isSystem = isSystemVar(variable)
+          const node = isSystem ? nodes.find(node => node.data.type === BlockEnum.Start) : nodes.find(node => node.id === variable[0])
+          return (
+            <VariableLabelInNode
+              key={index}
+              variables={variable}
+              nodeType={node?.data.type}
+              nodeTitle={node?.data.title}
+              rightSlot={
+                value.operation && <Badge className='!ml-auto shrink-0' text={t(`${i18nPrefix}.operations.${value.operation}`)} />
+              }
+            />
+          )
+        })}
+      </div>
+    )
+  }
+  // Legacy version
+  const { assigned_variable_selector: variable, write_mode: writeMode } = data as any
+
+  if (!variable || variable.length === 0)
+    return null
+  const isSystem = isSystemVar(variable)
+  const node = isSystem ? nodes.find(node => node.data.type === BlockEnum.Start) : nodes.find(node => node.id === variable[0])
+
+  return (
+    <div className='relative flex flex-col items-start gap-0.5 self-stretch px-3 py-1'>
+      <VariableLabelInNode
+        variables={variable}
+        nodeType={node?.data.type}
+        nodeTitle={node?.data.title}
+        rightSlot={
+          writeMode && <Badge className='!ml-auto shrink-0' text={t(`${i18nPrefix}.operations.${writeMode}`)} />
+        }
+      />
+    </div>
+  )
+}
+
+export default React.memo(NodeComponent)