import * as React from "react"
import { useAtom } from "jotai"
import { Button } from "@/components/ui/button"
import {
  Popover,
  PopoverTrigger,
  PopoverContent,
} from "@/components/ui/popover"
import { cn } from "@/lib/utils"
import { LaIcon } from "@/components/custom/la-icon"
import { LEARNING_STATES, LearningStateValue } from "@/lib/constants"
import { linkLearningStateSelectorAtom } from "@/store/link"
import {
  Command,
  CommandInput,
  CommandList,
  CommandItem,
  CommandGroup,
} from "@/components/ui/command"
import { ScrollArea } from "@/components/ui/scroll-area"
import { icons } from "lucide-react"

interface LearningStateSelectorProps {
  showSearch?: boolean
  defaultLabel?: string
  searchPlaceholder?: string
  value?: string
  onChange: (value: LearningStateValue) => void
  className?: string
  defaultIcon?: keyof typeof icons
}

export const LearningStateSelector: React.FC<LearningStateSelectorProps> = ({
  showSearch = true,
  defaultLabel = "State",
  searchPlaceholder = "Search state...",
  value,
  onChange,
  className,
  defaultIcon,
}) => {
  const [isLearningStateSelectorOpen, setIsLearningStateSelectorOpen] = useAtom(
    linkLearningStateSelectorAtom,
  )
  const selectedLearningState = React.useMemo(
    () => LEARNING_STATES.find((ls) => ls.value === value),
    [value],
  )

  const handleSelect = (selectedValue: string) => {
    onChange(selectedValue as LearningStateValue)
    setIsLearningStateSelectorOpen(false)
  }

  const iconName = selectedLearningState?.icon || defaultIcon
  const labelText = selectedLearningState?.label || defaultLabel

  return (
    <Popover
      open={isLearningStateSelectorOpen}
      onOpenChange={setIsLearningStateSelectorOpen}
    >
      <PopoverTrigger asChild>
        <Button
          size="sm"
          type="button"
          role="combobox"
          variant="secondary"
          className={cn("h-7 gap-x-2 text-sm", className)}
        >
          {iconName && (
            <LaIcon
              name={iconName}
              className={cn(selectedLearningState?.className)}
            />
          )}
          {labelText && (
            <span
              className={cn("truncate", selectedLearningState?.className || "")}
            >
              {labelText}
            </span>
          )}
          <LaIcon name="ChevronDown" />
        </Button>
      </PopoverTrigger>
      <PopoverContent className="w-52 rounded-lg p-0" side="bottom" align="end">
        <LearningStateSelectorContent
          showSearch={showSearch}
          searchPlaceholder={searchPlaceholder}
          value={value}
          onSelect={handleSelect}
        />
      </PopoverContent>
    </Popover>
  )
}

interface LearningStateSelectorContentProps {
  showSearch: boolean
  searchPlaceholder: string
  value?: string
  onSelect: (value: string) => void
}

export const LearningStateSelectorContent: React.FC<
  LearningStateSelectorContentProps
> = ({ showSearch, searchPlaceholder, value, onSelect }) => {
  return (
    <Command>
      {showSearch && (
        <CommandInput placeholder={searchPlaceholder} className="h-9" />
      )}
      <CommandList>
        <ScrollArea>
          <CommandGroup>
            {LEARNING_STATES.map((ls) => (
              <CommandItem key={ls.value} value={ls.value} onSelect={onSelect}>
                {ls.icon && (
                  <LaIcon name={ls.icon} className={cn("mr-2", ls.className)} />
                )}
                <span className={ls.className}>{ls.label}</span>
                <LaIcon
                  name="Check"
                  className={cn(
                    "absolute right-3",
                    ls.value === value ? "text-primary" : "text-transparent",
                  )}
                />
              </CommandItem>
            ))}
          </CommandGroup>
        </ScrollArea>
      </CommandList>
    </Command>
  )
}