import { Combobox } from "@headlessui/react";
import { observer } from "mobx-react";
import type { ReactNode } from "react";
import { useRef, useState } from "react";
import { usePopper } from "react-popper";
// buzzwareTech imports
import { useTranslation } from "@buzzwaretech/i18n";
import { Logo } from "@buzzwaretech/propel/emoji-icon-picker";
import {
    CheckIcon,
    ChevronDownIcon,
    ProjectIcon,
    SearchIcon,
} from "@buzzwaretech/propel/icons";
import { ComboDropDown } from "@buzzwaretech/ui";
import { cn, sortBySelectedFirst } from "@buzzwaretech/utils";
// components
// hooks
import { useDropdown } from "@/hooks/use-dropdown";
// buzzwareTech web imports
import type { TProject } from "@buzzwaretech/types";
// local imports
import { DropdownButton } from "../buttons";
import { BUTTON_VARIANTS_WITH_TEXT } from "../constants";
import type { TDropdownProps } from "../types";

type Props = TDropdownProps & {
  button?: ReactNode;
  currentProjectId?: string;
  dropdownArrow?: boolean;
  dropdownArrowClassName?: string;
  getProjectById: (
    projectId: string | null | undefined,
  ) => Partial<TProject> | undefined;
  onClose?: () => void;
  projectIds: string[];
  renderByDefault?: boolean;
  renderCondition?: (projectId: string) => boolean;
} & (
    | {
        multiple: false;
        onChange: (val: string) => void;
        value: string | null;
      }
    | {
        multiple: true;
        onChange: (val: string[]) => void;
        value: string[];
      }
  );

export const ProjectDropdownBase = observer(function ProjectDropdownBase(
  props: Props,
) {
  const {
    button,
    buttonClassName,
    buttonContainerClassName,
    buttonVariant,
    className = "",
    currentProjectId,
    disabled = false,
    dropdownArrow = false,
    dropdownArrowClassName = "",
    getProjectById,
    hideIcon = false,
    multiple,
    onChange,
    onClose,
    placeholder = "Project",
    placement,
    projectIds,
    renderByDefault = true,
    renderCondition,
    showTooltip = false,
    tabIndex,
    value,
  } = props;
  // refs
  const dropdownRef = useRef<HTMLDivElement | null>(null);
  const inputRef = useRef<HTMLInputElement | null>(null);
  // popper-js refs
  const [referenceElement, setReferenceElement] =
    useState<HTMLButtonElement | null>(null);
  const [popperElement, setPopperElement] = useState<HTMLDivElement | null>(
    null,
  );
  // states
  const [query, setQuery] = useState("");
  const [isOpen, setIsOpen] = useState(false);
  // buzzwareTech hooks
  const { t } = useTranslation();
  // popper-js init
  const { styles, attributes } = usePopper(referenceElement, popperElement, {
    placement: placement ?? "bottom-start",
    modifiers: [
      {
        name: "preventOverflow",
        options: {
          padding: 12,
        },
      },
    ],
  });
  // store hooks
  const options = projectIds?.map((projectId) => {
    const projectDetails = getProjectById(projectId);
    if (renderCondition && !renderCondition(projectId)) return;
    return {
      value: projectId,
      query: `${projectDetails?.name}`,
      content: (
        <div className="flex items-center gap-2">
          {projectDetails?.logo_props && (
            <span className="grid h-4 w-4 flex-shrink-0 place-items-center">
              <Logo logo={projectDetails?.logo_props} size={12} />
            </span>
          )}
          <span className="flex-grow truncate">{projectDetails?.name}</span>
        </div>
      ),
    };
  });

  const filteredOptions = sortBySelectedFirst(
    (query === ""
      ? options?.filter((o) => o?.value !== currentProjectId)
      : options?.filter(
          (o) =>
            o?.value !== currentProjectId &&
            o?.query.toLowerCase().includes(query.toLowerCase()),
        )
    )?.filter((o): o is NonNullable<typeof o> => o !== undefined),
    value,
  );

  const { handleClose, handleKeyDown, handleOnClick, searchInputKeyDown } =
    useDropdown({
      dropdownRef,
      inputRef,
      isOpen,
      onClose,
      query,
      setIsOpen,
      setQuery,
    });

  const dropdownOnChange = (val: string & string[]) => {
    onChange(val);
    if (!multiple) handleClose();
  };

  const getDisplayName = (
    value: string | string[] | null,
    placeholder: string = "",
  ) => {
    if (Array.isArray(value)) {
      const firstProject = getProjectById(value[0]);
      return value.length
        ? value.length === 1
          ? firstProject?.name
          : `${value.length} projects`
        : placeholder;
    } else {
      return value ? (getProjectById(value)?.name ?? placeholder) : placeholder;
    }
  };

  const getProjectIcon = (value: string | string[] | null) => {
    const renderIcon = (logoProps: TProject["logo_props"]) => (
      <span className="grid h-4 w-4 flex-shrink-0 place-items-center">
        <Logo logo={logoProps} size={14} />
      </span>
    );

    if (Array.isArray(value)) {
      return (
        <div className="flex items-center gap-0.5">
          {value.length > 0 ? (
            value.map((projectId) => {
              const projectDetails = getProjectById(projectId);
              return projectDetails?.logo_props
                ? renderIcon(projectDetails.logo_props)
                : null;
            })
          ) : (
            <ProjectIcon className="size-3 text-tertiary" />
          )}
        </div>
      );
    } else {
      const projectDetails = getProjectById(value);
      return projectDetails?.logo_props
        ? renderIcon(projectDetails.logo_props)
        : null;
    }
  };

  const comboButton = (
    <>
      {button ? (
        <button
          ref={setReferenceElement}
          type="button"
          className={cn(
            "clickable block h-full w-full outline-none",
            buttonContainerClassName,
          )}
          onClick={handleOnClick}
          disabled={disabled}
        >
          {button}
        </button>
      ) : (
        <button
          ref={setReferenceElement}
          type="button"
          className={cn(
            "clickable block h-full max-w-full outline-none",
            {
              "cursor-not-allowed text-secondary": disabled,
              "cursor-pointer": !disabled,
            },
            buttonContainerClassName,
          )}
          onClick={handleOnClick}
          disabled={disabled}
        >
          <DropdownButton
            className={buttonClassName}
            isActive={isOpen}
            tooltipHeading="Project"
            tooltipContent={
              value?.length
                ? `${value.length} project${value.length !== 1 ? "s" : ""}`
                : placeholder
            }
            showTooltip={showTooltip}
            variant={buttonVariant}
            renderToolTipByDefault={renderByDefault}
          >
            {!hideIcon && getProjectIcon(value)}
            {BUTTON_VARIANTS_WITH_TEXT.includes(buttonVariant) && (
              <span className="max-w-40 truncate">
                {getDisplayName(value, placeholder)}
              </span>
            )}
            {dropdownArrow && (
              <ChevronDownIcon
                className={cn(
                  "h-2.5 w-2.5 flex-shrink-0",
                  dropdownArrowClassName,
                )}
                aria-hidden="true"
              />
            )}
          </DropdownButton>
        </button>
      )}
    </>
  );

  return (
    <ComboDropDown
      as="div"
      ref={dropdownRef}
      tabIndex={tabIndex}
      className={cn("h-full", className)}
      value={value}
      onChange={dropdownOnChange}
      disabled={disabled}
      onKeyDown={handleKeyDown}
      button={comboButton}
      renderByDefault={renderByDefault}
      multiple={multiple}
    >
      {isOpen && (
        <Combobox.Options className="fixed z-10" static>
          <div
            className="my-1 w-48 rounded-sm border-[0.5px] border-strong bg-surface-1 px-2 py-2.5 text-11 shadow-raised-200 focus:outline-none"
            ref={setPopperElement}
            style={styles.popper}
            {...attributes.popper}
          >
            <div className="flex items-center gap-1.5 rounded-sm border border-subtle bg-surface-2 px-2">
              <SearchIcon
                className="h-3.5 w-3.5 text-placeholder"
                strokeWidth={1.5}
              />
              <Combobox.Input
                as="input"
                ref={inputRef}
                className="w-full bg-transparent py-1 text-11 text-secondary placeholder:text-placeholder focus:outline-none"
                value={query}
                onChange={(e) => setQuery(e.target.value)}
                placeholder={t("search")}
                displayValue={(assigned: any) => assigned?.name}
                onKeyDown={searchInputKeyDown}
              />
            </div>
            <div className="mt-2 max-h-48 space-y-1 overflow-y-scroll">
              {filteredOptions ? (
                filteredOptions.length > 0 ? (
                  filteredOptions.map((option) => {
                    if (!option) return;
                    return (
                      <Combobox.Option
                        key={option.value}
                        value={option.value}
                        className={({ active, selected }) =>
                          `flex w-full cursor-pointer items-center justify-between gap-2 truncate rounded-sm px-1 py-1.5 select-none ${
                            active ? "bg-layer-transparent-hover" : ""
                          } ${selected ? "text-primary" : "text-secondary"}`
                        }
                      >
                        {({ selected }) => (
                          <>
                            <span className="flex-grow truncate">
                              {option.content}
                            </span>
                            {selected && (
                              <CheckIcon className="h-3.5 w-3.5 flex-shrink-0" />
                            )}
                          </>
                        )}
                      </Combobox.Option>
                    );
                  })
                ) : (
                  <p className="px-1.5 py-1 text-placeholder italic">
                    {t("no_matching_results")}
                  </p>
                )
              ) : (
                <p className="px-1.5 py-1 text-placeholder italic">
                  {t("loading")}
                </p>
              )}
            </div>
          </div>
        </Combobox.Options>
      )}
    </ComboDropDown>
  );
});