'use client'
import { Button } from '@langgenius/dify-ui/button'
import { Checkbox } from '@langgenius/dify-ui/checkbox'
import { CheckboxGroup } from '@langgenius/dify-ui/checkbox-group'
import { cn } from '@langgenius/dify-ui/cn'
import { Field, FieldDescription, FieldItem, FieldLabel } from '@langgenius/dify-ui/field'
import { Fieldset, FieldsetLegend } from '@langgenius/dify-ui/fieldset'
import { useMemo, useState } from 'react'
import { useTranslation } from 'react-i18next'
import Badge from '@/app/components/base/badge'
import { SearchInput } from '@/app/components/base/search-input'
import SearchMenu from '@/assets/search-menu.svg'

type CheckboxListOption = {
  label: string
  value: string
  disabled?: boolean
}

type CheckboxListProps = {
  name?: string
  title?: string
  label?: string
  description?: string
  options: CheckboxListOption[]
  value?: string[]
  onChange?: (value: string[]) => void
  disabled?: boolean
  containerClassName?: string
  showSelectAll?: boolean
  showCount?: boolean
  showSearch?: boolean
  maxHeight?: string | number
}

export const CheckboxList = ({
  name,
  title = '',
  label,
  description,
  options,
  value = [],
  onChange,
  disabled = false,
  containerClassName,
  showSelectAll = true,
  showCount = true,
  showSearch = true,
  maxHeight,
}: CheckboxListProps) => {
  const { t } = useTranslation()
  const [searchQuery, setSearchQuery] = useState('')

  const filteredOptions = useMemo(() => {
    if (!searchQuery?.trim()) return options

    const query = searchQuery.toLowerCase()
    return options.filter(
      (option) =>
        option.label.toLowerCase().includes(query) || option.value.toLowerCase().includes(query),
    )
  }, [options, searchQuery])

  const selectedCount = value.length

  const selectableOptionValues = useMemo(
    () => options.filter((option) => !option.disabled).map((option) => option.value),
    [options],
  )

  return (
    <Field name={name} className={cn('flex w-full flex-col gap-1', containerClassName)}>
      <Fieldset
        render={
          <CheckboxGroup
            aria-label={!label && title ? title : undefined}
            value={value}
            onValueChange={(nextValue) => onChange?.(nextValue)}
            allValues={selectableOptionValues}
            disabled={disabled}
            className="flex flex-col gap-1"
          />
        }
      >
        {label && <FieldsetLegend className="mb-0">{label}</FieldsetLegend>}
        {description && (
          <FieldDescription className="body-xs-regular text-text-tertiary">
            {description}
          </FieldDescription>
        )}

        <div className="rounded-lg border border-components-panel-border bg-components-panel-bg">
          {(showSelectAll || title || showSearch) && (
            <div className="relative flex items-center gap-2 border-b border-divider-subtle px-3 py-2">
              {!searchQuery && showSelectAll && (
                <FieldItem disabled={disabled} className="shrink-0 gap-0">
                  <FieldLabel
                    className={cn('flex items-center p-0', !disabled && 'cursor-pointer')}
                  >
                    <Checkbox parent disabled={disabled} />
                    <span className="sr-only">
                      {t(($) => $['operation.selectAll'], { ns: 'common' })}
                    </span>
                  </FieldLabel>
                </FieldItem>
              )}
              {!searchQuery ? (
                <div className="flex min-w-0 flex-1 items-center gap-1">
                  {title && (
                    <span className="truncate system-xs-semibold-uppercase leading-5 text-text-secondary">
                      {title}
                    </span>
                  )}
                  {showCount && selectedCount > 0 && (
                    <Badge uppercase>
                      {t(($) => $['operation.selectCount'], { ns: 'common', count: selectedCount })}
                    </Badge>
                  )}
                </div>
              ) : (
                <div className="flex-1 system-sm-medium-uppercase leading-6 text-text-secondary">
                  {filteredOptions.length > 0
                    ? t(($) => $['operation.searchCount'], {
                        ns: 'common',
                        count: filteredOptions.length,
                        content: title,
                      })
                    : t(($) => $['operation.noSearchCount'], { ns: 'common', content: title })}
                </div>
              )}
              {showSearch && (
                <SearchInput
                  value={searchQuery}
                  onValueChange={setSearchQuery}
                  placeholder={t(($) => $['placeholder.search'], { ns: 'common' })}
                  className="w-40"
                />
              )}
            </div>
          )}

          <div
            className="p-1"
            style={maxHeight ? { maxHeight, overflowY: 'auto' } : {}}
            data-testid="options-container"
          >
            {!filteredOptions.length ? (
              <div className="px-3 py-6 text-center text-sm text-text-tertiary">
                {searchQuery ? (
                  <div className="flex flex-col items-center justify-center gap-2">
                    <img alt="search menu" src={SearchMenu.src} width={32} />
                    <span className="system-sm-regular text-text-secondary">
                      {t(($) => $['operation.noSearchResults'], { ns: 'common', content: title })}
                    </span>
                    <Button
                      variant="secondary-accent"
                      size="small"
                      onClick={() => setSearchQuery('')}
                    >
                      {t(($) => $['operation.resetKeywords'], { ns: 'common' })}
                    </Button>
                  </div>
                ) : (
                  t(($) => $.noData, { ns: 'common' })
                )}
              </div>
            ) : (
              filteredOptions.map((option) => (
                <FieldItem
                  key={option.value}
                  disabled={option.disabled || disabled}
                  className="gap-0"
                >
                  <FieldLabel
                    className={cn(
                      'flex w-full cursor-pointer items-center gap-2 rounded-md px-2 py-1.5 transition-colors hover:bg-state-base-hover',
                      (option.disabled || disabled) && 'cursor-not-allowed opacity-50',
                    )}
                  >
                    <Checkbox value={option.value} disabled={option.disabled || disabled} />
                    <span
                      className="flex-1 truncate system-sm-medium text-text-secondary"
                      title={option.label}
                    >
                      {option.label}
                    </span>
                  </FieldLabel>
                </FieldItem>
              ))
            )}
          </div>
        </div>
      </Fieldset>
    </Field>
  )
}