import { cn } from '@langgenius/dify-ui/cn'
import {
  RiHashtag,
  RiTextSnippet,
  RiTimeLine,
} from '@remixicon/react'
import { memo } from 'react'
import { MetadataFilteringVariableType } from '@/app/components/workflow/nodes/knowledge-retrieval/types'

type MetadataIconProps = {
  type?: MetadataFilteringVariableType
  className?: string
}
const MetadataIcon = ({
  type,
  className,
}: MetadataIconProps) => {
  return (
    <>
      {
        (type === MetadataFilteringVariableType.string || type === MetadataFilteringVariableType.select) && (
          <RiTextSnippet className={cn('size-3.5', className)} />
        )
      }
      {
        type === MetadataFilteringVariableType.number && (
          <RiHashtag className={cn('size-3.5', className)} />
        )
      }
      {
        type === MetadataFilteringVariableType.time && (
          <RiTimeLine className={cn('size-3.5', className)} />
        )
      }
    </>
  )
}

export default memo(MetadataIcon)
