node.tsx 1.9 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647
  1. import type { FC } from 'react'
  2. import React from 'react'
  3. import { useNodes } from 'reactflow'
  4. import { useTranslation } from 'react-i18next'
  5. import NodeVariableItem from '../variable-assigner/components/node-variable-item'
  6. import { type AssignerNodeType } from './types'
  7. import { isConversationVar, isENV, isSystemVar } from '@/app/components/workflow/nodes/_base/components/variable/utils'
  8. import { BlockEnum, type Node, type NodeProps } from '@/app/components/workflow/types'
  9. const i18nPrefix = 'workflow.nodes.assigner'
  10. const NodeComponent: FC<NodeProps<AssignerNodeType>> = ({
  11. data,
  12. }) => {
  13. const { t } = useTranslation()
  14. const nodes: Node[] = useNodes()
  15. const { assigned_variable_selector: variable, write_mode: writeMode } = data
  16. if (!variable || variable.length === 0)
  17. return null
  18. const isSystem = isSystemVar(variable)
  19. const isEnv = isENV(variable)
  20. const isChatVar = isConversationVar(variable)
  21. const node = isSystem ? nodes.find(node => node.data.type === BlockEnum.Start) : nodes.find(node => node.id === variable[0])
  22. const varName = isSystem ? `sys.${variable[variable.length - 1]}` : variable.slice(1).join('.')
  23. return (
  24. <div className='relative px-3'>
  25. <div className='mb-1 system-2xs-medium-uppercase text-text-tertiary'>{t(`${i18nPrefix}.assignedVariable`)}</div>
  26. <NodeVariableItem
  27. node={node as Node}
  28. isEnv={isEnv}
  29. isChatVar={isChatVar}
  30. varName={varName}
  31. className='bg-workflow-block-parma-bg'
  32. />
  33. <div className='my-2 flex justify-between items-center h-[22px] px-[5px] bg-workflow-block-parma-bg radius-sm'>
  34. <div className='system-xs-medium-uppercase text-text-tertiary'>{t(`${i18nPrefix}.writeMode`)}</div>
  35. <div className='system-xs-medium text-text-secondary'>{t(`${i18nPrefix}.${writeMode}`)}</div>
  36. </div>
  37. </div>
  38. )
  39. }
  40. export default React.memo(NodeComponent)