import type { NodeProps } from 'reactflow'
import { Tooltip, TooltipContent, TooltipTrigger } from '@langgenius/dify-ui/tooltip'
import { RiHome5Fill } from '@remixicon/react'
import { memo } from 'react'
import { useTranslation } from 'react-i18next'
import { NodeSourceHandle } from '@/app/components/workflow/nodes/_base/components/node-handle'

const IterationStartNode = ({ id, data }: NodeProps) => {
  const { t } = useTranslation()

  return (
    <div className="nodrag group mt-1 flex size-11 items-center justify-center rounded-2xl border border-workflow-block-border bg-workflow-block-bg shadow-xs">
      <Tooltip>
        <TooltipTrigger
          aria-label={t('blocks.iteration-start', { ns: 'workflow' })}
          className="flex h-6 w-6 items-center justify-center rounded-full border-[0.5px] border-components-panel-border-subtle bg-util-colors-blue-brand-blue-brand-500 p-0"
        >
          <RiHome5Fill className="size-3 text-text-primary-on-surface" />
        </TooltipTrigger>
        <TooltipContent>{t('blocks.iteration-start', { ns: 'workflow' })}</TooltipContent>
      </Tooltip>
      <NodeSourceHandle
        id={id}
        data={data}
        handleClassName="top-1/2! -right-[9px]! -translate-y-1/2!"
        handleId="source"
      />
    </div>
  )
}

export const IterationStartNodeDumb = () => {
  const { t } = useTranslation()

  return (
    <div className="nodrag relative top-[21px] left-[17px] z-11 flex h-11 w-11 items-center justify-center rounded-2xl border border-workflow-block-border bg-workflow-block-bg">
      <Tooltip>
        <TooltipTrigger
          aria-label={t('blocks.iteration-start', { ns: 'workflow' })}
          className="flex h-6 w-6 items-center justify-center rounded-full border-[0.5px] border-components-panel-border-subtle bg-util-colors-blue-brand-blue-brand-500 p-0"
        >
          <RiHome5Fill className="size-3 text-text-primary-on-surface" />
        </TooltipTrigger>
        <TooltipContent>{t('blocks.iteration-start', { ns: 'workflow' })}</TooltipContent>
      </Tooltip>
    </div>
  )
}

export default memo(IterationStartNode)