import type { NodeProps } from 'reactflow'
import BaseNode from './base'
import { NodeComponentMap } from './constants'

const CustomNode = (props: NodeProps) => {
  const nodeData = props.data
  const NodeComponent = NodeComponentMap[nodeData.type]

  return (
    <>
      <BaseNode { ...props }>
        { NodeComponent && <NodeComponent /> }
      </BaseNode>
    </>
  )
}

export default CustomNode
