import type { ReactNode } from 'react'
import { cn } from '@langgenius/dify-ui/cn'
import * as React from 'react'
import { useTranslation } from 'react-i18next'
import Badge from '@/app/components/base/badge'
import { ArrowShape } from '../../base/icons/src/vender/knowledge'
import { EffectColor } from './chunk-structure/types'

const HEADER_EFFECT_MAP: Record<EffectColor, string> = {
  [EffectColor.indigo]: 'bg-util-colors-indigo-indigo-600 opacity-50',
  [EffectColor.blueLight]: 'bg-util-colors-blue-light-blue-light-600 opacity-80',
  [EffectColor.orange]: 'bg-util-colors-orange-orange-500 opacity-50',
  [EffectColor.purple]: 'bg-util-colors-purple-purple-600 opacity-80',
}
type OptionCardProps<T> = {
  id: T
  className?: string
  isActive?: boolean
  icon?: ReactNode
  iconActiveColor?: string
  title: string
  description?: string
  isRecommended?: boolean
  effectColor?: EffectColor
  showEffectColor?: boolean
  disabled?: boolean
  onClick?: (id: T) => void
  children?: ReactNode
  showChildren?: boolean
  ref?: React.Ref<HTMLDivElement>
}
const OptionCard = <T,>({
  id,
  className,
  isActive,
  icon,
  iconActiveColor,
  title,
  description,
  isRecommended,
  effectColor,
  showEffectColor,
  disabled,
  onClick,
  children,
  showChildren,
  ref,
}: OptionCardProps<T>) => {
  const { t } = useTranslation()

  return (
    <div
      ref={ref}
      className={cn(
        'cursor-pointer overflow-hidden rounded-xl border border-components-option-card-option-border bg-components-option-card-option-bg',
        isActive && 'border border-components-option-card-option-selected-border ring-[1px] ring-components-option-card-option-selected-border',
        disabled && 'cursor-not-allowed opacity-50',
      )}
      onClick={() => {
        if (isActive || disabled)
          return
        onClick?.(id)
      }}
    >
      <div className={cn(
        'relative flex rounded-t-xl p-2',
        className,
      )}
      >
        {
          effectColor && showEffectColor && (
            <div className={cn(
              'absolute top-[-2px] left-[-2px] h-14 w-14 rounded-full blur-[80px]',
              `${HEADER_EFFECT_MAP[effectColor]}`,
            )}
            />
          )
        }
        {
          !!icon && (
            <div className={cn(
              'flex size-6 shrink-0 items-center justify-center text-text-tertiary',
              isActive && iconActiveColor,
            )}
            >
              {icon}
            </div>
          )
        }
        <div className="flex grow flex-col gap-y-0.5 py-px">
          <div className="flex items-center gap-x-1">
            <span className="system-sm-medium text-text-secondary">
              {title}
            </span>
            {
              isRecommended && (
                <Badge className="h-[18px] border-text-accent-secondary text-text-accent-secondary">
                  {t('stepTwo.recommend', { ns: 'datasetCreation' })}
                </Badge>
              )
            }
          </div>
          {
            description && (
              <div className="system-xs-regular text-text-tertiary">
                {description}
              </div>
            )
          }
        </div>
      </div>
      {
        !!(children && showChildren) && (
          <div className="relative rounded-b-xl bg-components-panel-bg p-4">
            <ArrowShape className="absolute top-[-11px] left-[14px] size-4 text-components-panel-bg" />
            {children}
          </div>
        )
      }
    </div>
  )
}

export default React.memo(OptionCard) as typeof OptionCard