File size: 849 Bytes
4304c6d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
import { memo } from 'react'
import type { NodeProps } from 'reactflow'
import type { Node } from '../types'
import {
NodeComponentMap,
PanelComponentMap,
} from './constants'
import BaseNode from './_base/node'
import BasePanel from './_base/panel'
const CustomNode = memo((props: NodeProps) => {
const nodeData = props.data
const NodeComponent = NodeComponentMap[nodeData.type]
return (
<BaseNode { ...props }>
<NodeComponent />
</BaseNode>
)
})
CustomNode.displayName = 'CustomNode'
export const Panel = memo((props: Node) => {
const nodeData = props.data
const PanelComponent = PanelComponentMap[nodeData.type]
return (
<BasePanel key={props.id} {...props}>
<PanelComponent />
</BasePanel>
)
})
Panel.displayName = 'Panel'
export default memo(CustomNode)
|