import { RiQuestionLine } from '@remixicon/react'
import {
  memo,
  useCallback,
} from 'react'
import { useTranslation } from 'react-i18next'
import {
  Economic,
  HighQuality,
} from '@/app/components/base/icons/src/vender/knowledge'
import Input from '@/app/components/base/input'
import Slider from '@/app/components/base/slider'
import Tooltip from '@/app/components/base/tooltip'
import { Field } from '@/app/components/workflow/nodes/_base/components/layout'
import { cn } from '@/utils/classnames'
import {
  ChunkStructureEnum,
  IndexMethodEnum,
} from '../types'
import OptionCard from './option-card'

type IndexMethodProps = {
  chunkStructure: ChunkStructureEnum
  indexMethod?: IndexMethodEnum
  onIndexMethodChange: (value: IndexMethodEnum) => void
  keywordNumber: number
  onKeywordNumberChange: (value: number) => void
  readonly?: boolean
}
const IndexMethod = ({
  chunkStructure,
  indexMethod,
  onIndexMethodChange,
  keywordNumber,
  onKeywordNumberChange,
  readonly = false,
}: IndexMethodProps) => {
  const { t } = useTranslation()
  const isHighQuality = indexMethod === IndexMethodEnum.QUALIFIED
  const isEconomy = indexMethod === IndexMethodEnum.ECONOMICAL

  const handleIndexMethodChange = useCallback((newIndexMethod: IndexMethodEnum) => {
    onIndexMethodChange(newIndexMethod)
  }, [onIndexMethodChange])

  const handleInputChange = useCallback((e: React.ChangeEvent<HTMLInputElement>) => {
    const value = Number(e.target.value)
    if (!Number.isNaN(value))
      onKeywordNumberChange(value)
  }, [onKeywordNumberChange])

  return (
    <Field
      fieldTitleProps={{
        title: t('stepTwo.indexMode', { ns: 'datasetCreation' }),
      }}
    >
      <div className="space-y-1">
        <OptionCard<IndexMethodEnum>
          id={IndexMethodEnum.QUALIFIED}
          selectedId={indexMethod}
          icon={(
            <HighQuality
              className={cn(
                'h-[15px] w-[15px] text-text-tertiary group-hover:text-util-colors-orange-orange-500',
                isHighQuality && 'text-util-colors-orange-orange-500',
              )}
            />
          )}
          title={t('stepTwo.qualified', { ns: 'datasetCreation' })}
          description={t('form.indexMethodHighQualityTip', { ns: 'datasetSettings' })}
          onClick={handleIndexMethodChange}
          isRecommended
          effectColor="orange"
        >
        </OptionCard>
        {
          chunkStructure === ChunkStructureEnum.general && (
            <OptionCard
              id={IndexMethodEnum.ECONOMICAL}
              selectedId={indexMethod}
              icon={(
                <Economic
                  className={cn(
                    'h-[15px] w-[15px] text-text-tertiary group-hover:text-util-colors-indigo-indigo-500',
                    isEconomy && 'text-util-colors-indigo-indigo-500',
                  )}
                />
              )}
              title={t('form.indexMethodEconomy', { ns: 'datasetSettings' })}
              description={t('form.indexMethodEconomyTip', { ns: 'datasetSettings', count: keywordNumber })}
              onClick={handleIndexMethodChange}
              effectColor="blue"
            >
              <div className="flex items-center">
                <div className="flex grow items-center">
                  <div className="system-xs-medium truncate text-text-secondary">
                    {t('form.numberOfKeywords', { ns: 'datasetSettings' })}
                  </div>
                  <Tooltip
                    popupContent="number of keywords"
                  >
                    <RiQuestionLine className="ml-0.5 h-3.5 w-3.5 text-text-quaternary" />
                  </Tooltip>
                </div>
                <Slider
                  disabled={readonly}
                  className="mr-3 w-24 shrink-0"
                  value={keywordNumber}
                  onChange={onKeywordNumberChange}
                />
                <Input
                  disabled={readonly}
                  className="shrink-0"
                  wrapperClassName="shrink-0 w-[72px]"
                  type="number"
                  value={keywordNumber}
                  onChange={handleInputChange}
                />
              </div>
            </OptionCard>
          )
        }
      </div>
    </Field>
  )
}

export default memo(IndexMethod)