mirror of
https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git
synced 2025-05-14 03:58:16 +08:00
50 lines
1.2 KiB
TypeScript
50 lines
1.2 KiB
TypeScript
import type { FC } from 'react'
|
|
import {
|
|
memo,
|
|
useEffect,
|
|
} from 'react'
|
|
import {
|
|
Background,
|
|
useNodesInitialized,
|
|
useViewport,
|
|
} from 'reactflow'
|
|
import { useNodeIterationInteractions } from './use-interactions'
|
|
import type { IterationNodeType } from './types'
|
|
import AddBlock from './add-block'
|
|
import cn from '@/utils/classnames'
|
|
import type { NodeProps } from '@/app/components/workflow/types'
|
|
|
|
const Node: FC<NodeProps<IterationNodeType>> = ({
|
|
id,
|
|
data,
|
|
}) => {
|
|
const { zoom } = useViewport()
|
|
const nodesInitialized = useNodesInitialized()
|
|
const { handleNodeIterationRerender } = useNodeIterationInteractions()
|
|
|
|
useEffect(() => {
|
|
if (nodesInitialized)
|
|
handleNodeIterationRerender(id)
|
|
}, [nodesInitialized, id, handleNodeIterationRerender])
|
|
|
|
return (
|
|
<div className={cn(
|
|
'relative min-w-[258px] min-h-[118px] w-full h-full rounded-2xl bg-[#F0F2F7]/90',
|
|
)}>
|
|
<Background
|
|
id={`iteration-background-${id}`}
|
|
className='rounded-2xl !z-0'
|
|
gap={[14 / zoom, 14 / zoom]}
|
|
size={2 / zoom}
|
|
color='#E4E5E7'
|
|
/>
|
|
<AddBlock
|
|
iterationNodeId={id}
|
|
iterationNodeData={data}
|
|
/>
|
|
</div>
|
|
)
|
|
}
|
|
|
|
export default memo(Node)
|