| import type { FC } from 'react' |
| import React from 'react' |
| import cn from 'classnames' |
| import type { EndNodeType } from './types' |
| import type { NodeProps, Variable } from '@/app/components/workflow/types' |
| import { isConversationVar, isENV, isSystemVar } from '@/app/components/workflow/nodes/_base/components/variable/utils' |
| import { |
| useIsChatMode, |
| useWorkflow, |
| useWorkflowVariables, |
| } 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 { BubbleX, Env } from '@/app/components/base/icons/src/vender/line/others' |
| import { BlockEnum } from '@/app/components/workflow/types' |
|
|
| const Node: FC<NodeProps<EndNodeType>> = ({ |
| id, |
| data, |
| }) => { |
| const { getBeforeNodesInSameBranch } = useWorkflow() |
| const availableNodes = getBeforeNodesInSameBranch(id) |
| const { getCurrentVariableType } = useWorkflowVariables() |
| const isChatMode = useIsChatMode() |
|
|
| 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 { 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 isEnv = isENV(value_selector) |
| const isChatVar = isConversationVar(value_selector) |
| const varName = isSystem ? `sys.${value_selector[value_selector.length - 1]}` : value_selector[value_selector.length - 1] |
| const varType = getCurrentVariableType({ |
| valueSelector: value_selector, |
| availableNodes, |
| isChatMode, |
| }) |
| 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'> |
| {!isEnv && !isChatVar && ( |
| <> |
| <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'> |
| {!isEnv && !isChatVar && <Variable02 className='shrink-0 w-3.5 h-3.5 text-primary-500' />} |
| {isEnv && <Env className='shrink-0 w-3.5 h-3.5 text-util-colors-violet-violet-600' />} |
| {isChatVar && <BubbleX className='w-3.5 h-3.5 text-util-colors-teal-teal-700' />} |
| |
| <div className={cn('max-w-[50px] ml-0.5 text-xs font-medium truncate', (isEnv || isChatVar) && '!max-w-[70px] text-gray-900')}>{varName}</div> |
| </div> |
| </div> |
| <div className='text-xs font-normal text-gray-700'> |
| <div className='max-w-[42px] ml-0.5 text-xs font-normal text-gray-500 capitalize truncate' title={varType}>{varType}</div> |
| </div> |
| </div> |
| ) |
| })} |
|
|
| </div> |
| ) |
| } |
|
|
| export default React.memo(Node) |
|
|