import { Slider } from '@langgenius/dify-ui/slider'
import { noop } from 'es-toolkit/function'
import { memo } from 'react'
import { useTranslation } from 'react-i18next'

const weightedScoreSliderSlotClassNames = {
  track: 'bg-util-colors-teal-teal-500',
  indicator: 'bg-util-colors-blue-light-blue-light-500',
}

const formatNumber = (value: number) => {
  if (value > 0 && value < 1) return `0.${value * 10}`
  else if (value === 1) return '1.0'

  return value
}

type Value = {
  value: number[]
}

type WeightedScoreProps = {
  value: Value
  onChange: (value: Value) => void
  readonly?: boolean
}
const WeightedScore = ({ value, onChange = noop, readonly = false }: WeightedScoreProps) => {
  const { t } = useTranslation()

  return (
    <div>
      <div className="space-x-3 rounded-lg border border-components-panel-border px-3 pt-5 pb-2">
        <div className="grow">
          <Slider
            className="grow"
            max={1.0}
            min={0}
            step={0.1}
            value={value.value[0]}
            onValueChange={(v) => !readonly && onChange({ value: [v, (10 - v * 10) / 10] })}
            disabled={readonly}
            aria-label={t(($) => $['weightedScore.semantic'], { ns: 'dataset' })}
            slotClassNames={weightedScoreSliderSlotClassNames}
          />
        </div>
        <div className="mt-3 flex justify-between">
          <div className="flex w-[90px] shrink-0 items-center system-xs-semibold-uppercase text-util-colors-blue-light-blue-light-500">
            <div
              className="mr-1 truncate uppercase"
              title={t(($) => $['weightedScore.semantic'], { ns: 'dataset' }) || ''}
            >
              {t(($) => $['weightedScore.semantic'], { ns: 'dataset' })}
            </div>
            {formatNumber(value.value[0]!)}
          </div>
          <div className="flex w-[90px] shrink-0 items-center justify-end system-xs-semibold-uppercase text-util-colors-teal-teal-500">
            {formatNumber(value.value[1]!)}
            <div
              className="ml-1 truncate uppercase"
              title={t(($) => $['weightedScore.keyword'], { ns: 'dataset' }) || ''}
            >
              {t(($) => $['weightedScore.keyword'], { ns: 'dataset' })}
            </div>
          </div>
        </div>
      </div>
    </div>
  )
}

export default memo(WeightedScore)
