import * as React from "react";
import { CaretSortIcon, CheckIcon } from "@radix-ui/react-icons";

import { cn } from "@/lib/utils";
import { Button } from "@/components/ui/button";
import {
  Command,
  CommandEmpty,
  CommandGroup,
  CommandInput,
  CommandItem,
  CommandList,
} from "@/components/ui/command";
import {
  Popover,
  PopoverContent,
  PopoverTrigger,
} from "@/components/ui/popover";

interface GenericType {
  [key: string]: any;
}

export function GenericCombobox<Type extends GenericType>({
  list,
  name,
  defaultValue,
  valueField,
  searchPlaceholder,
  selectPlaceholder,
  emptyText,
  keywordFields,
  template,
}: {
  list: Type[];
  name: string;
  defaultValue?: string | null;
  valueField: string;
  searchPlaceholder: string;
  selectPlaceholder: string;
  emptyText: string;
  keywordFields: string[];
  template: (item: Type) => JSX.Element;
}) {
  const [open, setOpen] = React.useState(false);
  const [value, setValue] = React.useState(defaultValue);

  let label;
  if (value) {
    const item = list.find((item) => item[valueField] === value);
    if (item) {
      label = template(item);
    }
  }

  return (
    <Popover open={open} onOpenChange={setOpen}>
      <input type="hidden" name={name} value={value ?? ""} />
      <PopoverTrigger asChild>
        <Button
          variant="outline"
          role="combobox"
          aria-expanded={open}
          className="w-full justify-between"
        >
          {label ?? selectPlaceholder}
          <CaretSortIcon className="ml-2 h-4 w-4 shrink-0 opacity-50" />
        </Button>
      </PopoverTrigger>
      <PopoverContent className="w-[--radix-popover-trigger-width] max-h-[--radix-popover-content-available-height] p-0">
        <Command
          filter={(value, search, keywords) => {
            const extendValue = value + " " + keywords?.join(" ");
            if (extendValue.toLowerCase().includes(search.toLowerCase())) {
              return 1;
            }
            return 0;
          }}
        >
          <CommandInput placeholder={searchPlaceholder} className="h-9" />
          <CommandList>
            <CommandEmpty>{emptyText}</CommandEmpty>
            <CommandGroup>
              {list.map((item) => (
                <CommandItem
                  key={item[valueField]}
                  value={item[valueField]}
                  onSelect={(currentValue) => {
                    setValue(currentValue === value ? "" : currentValue);
                    setOpen(false);
                  }}
                  keywords={keywordFields.map((field) => item[field])}
                >
                  {template(item)}
                  <CheckIcon
                    className={cn(
                      "ml-auto h-4 w-4",
                      value === item[valueField] ? "opacity-100" : "opacity-0"
                    )}
                  />
                </CommandItem>
              ))}
            </CommandGroup>
          </CommandList>
        </Command>
      </PopoverContent>
    </Popover>
  );
}