import type { FC } from 'react'
import React from 'react'
import { useTranslation } from 'react-i18next'
import type { AssignerNodeType } from './types'
import type { NodeProps } from '@/app/components/workflow/types'
import {
  VariableLabelInNode,
} from '@/app/components/workflow/nodes/_base/components/variable/variable-label'
import Badge from '@/app/components/base/badge'
import OperationItem from './components/operation-item'
import { useFindNode } from '@/app/components/workflow/hooks/use-find-node'

const i18nPrefix = 'workflow.nodes.assigner'

const NodeComponent: FC<NodeProps<AssignerNodeType>> = ({
  data,
}) => {
  const { t } = useTranslation()
  const node = useFindNode((data as any).assigned_variable_selector)
  if (data.version === '2') {
    const { items: operationItems } = data
    const validOperationItems = operationItems?.filter(item =>
      item.variable_selector && item.variable_selector.length > 0,
    ) || []

    if (validOperationItems.length === 0) {
      return (
        <div className='relative flex flex-col items-start gap-0.5 self-stretch px-3 py-1'>
          <div className='flex flex-col items-start gap-1 self-stretch'>
            <div className='flex items-center gap-1 self-stretch rounded-md bg-workflow-block-parma-bg px-[5px] py-1'>
              <div className='system-xs-medium flex-1 text-text-tertiary'>{t(`${i18nPrefix}.varNotSet`)}</div>
            </div>
          </div>
        </div>
      )
    }
    return (
      <div className='relative flex flex-col items-start gap-0.5 self-stretch px-3 py-1'>
        {operationItems.map((value, index) => {
          return <OperationItem key={index} value={value} />
        })}
      </div>
    )
  }
  // Legacy version
  const { assigned_variable_selector: variable, write_mode: writeMode } = data as any

  if (!variable || variable.length === 0)
    return null

  return (
    <div className='relative flex flex-col items-start gap-0.5 self-stretch px-3 py-1'>
      <VariableLabelInNode
        variables={variable}
        nodeType={node?.data.type}
        nodeTitle={node?.data.title}
        rightSlot={
          writeMode && <Badge className='!ml-auto shrink-0' text={t(`${i18nPrefix}.operations.${writeMode}`)} />
        }
      />
    </div>
  )
}

export default React.memo(NodeComponent)