import React, { useState } from "react";

// react-popper
import { usePopper } from "react-popper";
// headless ui
import { Combobox } from "@headlessui/react";
// icons
import { ChevronDownIcon } from "@heroicons/react/20/solid";
import { CheckIcon, MagnifyingGlassIcon } from "@heroicons/react/24/outline";
// types
import { DropdownProps } from "./types";

export type CustomSearchSelectProps = DropdownProps & {
  footerOption?: JSX.Element;
  onChange: any;
  options:
    | {
        value: any;
        query: string;
        content: React.ReactNode;
      }[]
    | undefined;
} & (
    | { multiple?: false; value: any } // if multiple is false, value can be anything
    | {
        multiple?: true;
        value: any[] | null; // if multiple is true, value should be an array
      }
  );

export const CustomSearchSelect = ({
  customButtonClassName = "",
  buttonClassName = "",
  className = "",
  customButton,
  placement,
  disabled = false,
  footerOption,
  input = false,
  label,
  maxHeight = "md",
  multiple = false,
  noChevron = false,
  onChange,
  options,
  onOpen,
  optionsClassName = "",
  value,
  width = "auto",
}: CustomSearchSelectProps) => {
  const [query, setQuery] = useState("");

  const [referenceElement, setReferenceElement] = useState<HTMLButtonElement | null>(null);
  const [popperElement, setPopperElement] = useState<HTMLDivElement | null>(null);

  const { styles, attributes } = usePopper(referenceElement, popperElement, {
    placement: placement ?? "bottom-start",
  });

  const filteredOptions =
    query === "" ? options : options?.filter((option) => option.query.toLowerCase().includes(query.toLowerCase()));

  const props: any = {
    value,
    onChange,
    disabled,
  };

  if (multiple) props.multiple = true;

  return (
    <Combobox as="div" className={`relative flex-shrink-0 text-left ${className}`} {...props}>
      {({ open }: { open: boolean }) => {
        if (open && onOpen) onOpen();

        return (
          <>
            {customButton ? (
              <Combobox.Button as={React.Fragment}>
                <button
                  ref={setReferenceElement}
                  type="button"
                  className={`flex items-center justify-between gap-1 w-full text-xs ${
                    disabled
                      ? "cursor-not-allowed text-custom-text-200"
                      : "cursor-pointer hover:bg-custom-background-80"
                  }  ${customButtonClassName}`}
                >
                  {customButton}
                </button>
              </Combobox.Button>
            ) : (
              <Combobox.Button as={React.Fragment}>
                <button
                  ref={setReferenceElement}
                  type="button"
                  className={`flex items-center justify-between gap-1 w-full rounded-md border border-custom-border-300 shadow-sm duration-300 focus:outline-none ${
                    input ? "px-3 py-2 text-sm" : "px-2.5 py-1 text-xs"
                  } ${
                    disabled
                      ? "cursor-not-allowed text-custom-text-200"
                      : "cursor-pointer hover:bg-custom-background-80"
                  } ${buttonClassName}`}
                >
                  {label}
                  {!noChevron && !disabled && <ChevronDownIcon className="h-3 w-3" aria-hidden="true" />}
                </button>
              </Combobox.Button>
            )}
            <Combobox.Options as={React.Fragment}>
              <div
                className={`z-10 min-w-[10rem] border border-custom-border-300 p-2 rounded-md bg-custom-background-90 text-xs shadow-custom-shadow-rg focus:outline-none my-1 ${
                  width === "auto" ? "min-w-[8rem] whitespace-nowrap" : width
                } ${optionsClassName}`}
                ref={setPopperElement}
                style={styles.popper}
                {...attributes.popper}
              >
                <div className="flex w-full items-center justify-start rounded-sm border-[0.6px] border-custom-border-200 bg-custom-background-90 px-2">
                  <MagnifyingGlassIcon className="h-3 w-3 text-custom-text-200" />
                  <Combobox.Input
                    className="w-full bg-transparent py-1 px-2 text-xs text-custom-text-200 placeholder:text-custom-text-400 focus:outline-none"
                    value={query}
                    onChange={(e) => setQuery(e.target.value)}
                    placeholder="Type to search..."
                    displayValue={(assigned: any) => assigned?.name}
                  />
                </div>
                <div
                  className={`mt-2 space-y-1 ${
                    maxHeight === "lg"
                      ? "max-h-60"
                      : maxHeight === "md"
                      ? "max-h-48"
                      : maxHeight === "rg"
                      ? "max-h-36"
                      : maxHeight === "sm"
                      ? "max-h-28"
                      : ""
                  } overflow-y-scroll`}
                >
                  {filteredOptions ? (
                    filteredOptions.length > 0 ? (
                      filteredOptions.map((option) => (
                        <Combobox.Option
                          key={option.value}
                          value={option.value}
                          className={({ active, selected }) =>
                            `flex items-center justify-between gap-2 cursor-pointer select-none truncate rounded px-1 py-1.5 ${
                              active || selected ? "bg-custom-background-80" : ""
                            } ${selected ? "text-custom-text-100" : "text-custom-text-200"}`
                          }
                        >
                          {({ active, selected }) => (
                            <>
                              {option.content}
                              {multiple ? (
                                <div
                                  className={`flex items-center justify-center rounded border border-custom-border-400 p-0.5 ${
                                    active || selected ? "opacity-100" : "opacity-0"
                                  }`}
                                >
                                  <CheckIcon className={`h-3 w-3 ${selected ? "opacity-100" : "opacity-0"}`} />
                                </div>
                              ) : (
                                <CheckIcon className={`h-3 w-3 ${selected ? "opacity-100" : "opacity-0"}`} />
                              )}
                            </>
                          )}
                        </Combobox.Option>
                      ))
                    ) : (
                      <span className="flex items-center gap-2 p-1">
                        <p className="text-left text-custom-text-200 ">No matching results</p>
                      </span>
                    )
                  ) : (
                    <p className="text-center text-custom-text-200">Loading...</p>
                  )}
                </div>
                {footerOption}
              </div>
            </Combobox.Options>
          </>
        );
      }}
    </Combobox>
  );
};
