import * as Core from "@ariakit/components/combobox/combobox-store";
import { useStore, useStoreProps } from "@ariakit/react-store";
import type { Store } from "@ariakit/react-store";
import { useUpdateEffect } from "@ariakit/react-utils";
import type { BivariantCallback, PickRequired } from "@ariakit/utils";
import type {
  CompositeStoreFunctions,
  CompositeStoreOptions,
  CompositeStoreState,
} from "../composite/composite-store.ts";
import {
  useCompositeStoreOptions,
  useCompositeStoreProps,
} from "../composite/composite-store.ts";
import type {
  PopoverStoreFunctions,
  PopoverStoreOptions,
  PopoverStoreState,
} from "../popover/popover-store.ts";
import { usePopoverStoreProps } from "../popover/popover-store.ts";
import { useTagContext } from "../tag/tag-context.tsx";
import type { TagStore } from "../tag/tag-store.ts";

export function useComboboxStoreOptions<T extends Core.ComboboxStoreOptions>(
  props: T,
) {
  const tag = useTagContext();
  props = {
    ...props,
    tag: props.tag !== undefined ? props.tag : tag,
  };
  return useCompositeStoreOptions(props);
}

export function useComboboxStoreProps<T extends Core.ComboboxStore>(
  store: T,
  update: () => void,
  props: ComboboxStoreProps,
) {
  useUpdateEffect(update, [props.tag]);

  const inputValueProps = {
    inputValue: props.inputValue ?? props.value,
    setInputValue: props.setInputValue ?? props.setValue,
  };
  useStoreProps(store, inputValueProps, "inputValue", "setInputValue");
  useStoreProps(store, props, "selectedValue", "setSelectedValue");
  useStoreProps(store, props, "selectOnMove");
  useStoreProps(store, props, "resetValueOnHide");
  useStoreProps(store, props, "resetValueOnSelect");

  return Object.assign(
    useCompositeStoreProps(
      usePopoverStoreProps(store, update, props),
      update,
      props,
    ),
    { tag: props.tag },
  );
}

/**
 * Creates a combobox store to control the state of
 * [Combobox](https://ariakit.com/components/combobox) components.
 * @see https://ariakit.com/components/combobox
 * @example
 * ```jsx
 * const combobox = useComboboxStore();
 *
 * <ComboboxProvider store={combobox}>
 *   <Combobox />
 *   <ComboboxPopover>
 *     <ComboboxItem value="Apple" />
 *     <ComboboxItem value="Banana" />
 *     <ComboboxItem value="Orange" />
 *   </ComboboxPopover>
 * </ComboboxProvider>
 * ```
 */
export function useComboboxStore<
  T extends ComboboxStoreSelectedValue = ComboboxStoreSelectedValue,
>(
  props: PickRequired<
    ComboboxStoreProps<T>,
    "selectedValue" | "defaultSelectedValue"
  >,
): ComboboxStore<T>;

export function useComboboxStore(props?: ComboboxStoreProps): ComboboxStore;

export function useComboboxStore(
  props: ComboboxStoreProps = {},
): ComboboxStore {
  props = useComboboxStoreOptions(props);
  const [store, update] = useStore(Core.createComboboxStore, props);
  return useComboboxStoreProps(store, update, props);
}

export type ComboboxStoreSelectedValue = Core.ComboboxStoreSelectedValue;

export interface ComboboxStoreItem extends Core.ComboboxStoreItem {}

export interface ComboboxStoreState<
  T extends ComboboxStoreSelectedValue = ComboboxStoreSelectedValue,
>
  extends
    Core.ComboboxStoreState<T>,
    CompositeStoreState<ComboboxStoreItem>,
    PopoverStoreState {}

export interface ComboboxStoreFunctions<
  T extends ComboboxStoreSelectedValue = ComboboxStoreSelectedValue,
>
  extends
    Pick<ComboboxStoreOptions<T>, "tag">,
    Omit<Core.ComboboxStoreFunctions<T>, "tag" | "disclosure">,
    CompositeStoreFunctions<ComboboxStoreItem>,
    PopoverStoreFunctions {}

export interface ComboboxStoreOptions<
  T extends ComboboxStoreSelectedValue = ComboboxStoreSelectedValue,
>
  extends
    Omit<Core.ComboboxStoreOptions<T>, "tag" | "disclosure">,
    CompositeStoreOptions<ComboboxStoreItem>,
    PopoverStoreOptions {
  /**
   * A callback that gets called when the
   * [`inputValue`](https://ariakit.com/reference/combobox-provider#inputvalue)
   * state changes.
   *
   * Live examples:
   * - [Combobox with integrated
   *   filter](https://ariakit.com/examples/combobox-filtering-integrated)
   * - [ComboboxGroup](https://ariakit.com/examples/combobox-group)
   * - [Combobox with links](https://ariakit.com/examples/combobox-links)
   * - [Multi-selectable
   *   Combobox](https://ariakit.com/examples/combobox-multiple)
   * - [Menu with Combobox](https://ariakit.com/examples/menu-combobox)
   * - [Select with Combobox](https://ariakit.com/examples/select-combobox)
   */
  setInputValue?: (value: ComboboxStoreState<T>["inputValue"]) => void;
  /**
   * A callback that gets called when the
   * [`inputValue`](https://ariakit.com/reference/combobox-provider#inputvalue)
   * state changes.
   * @deprecated Use
   * [`setInputValue`](https://ariakit.com/reference/combobox-provider#setinputvalue)
   * instead.
   */
  setValue?: (value: ComboboxStoreState<T>["value"]) => void;
  /**
   * A callback that's invoked when the
   * [`selectedValue`](https://ariakit.com/reference/combobox-provider#selectedvalue)
   * state changes, typically when the user selects an item. This can be used to
   * implement behavior like `onSelect` or `onItemSelect`.
   *
   * Live examples:
   * - [Multi-selectable
   *   Combobox](https://ariakit.com/examples/combobox-multiple)
   */
  setSelectedValue?: BivariantCallback<
    (value: ComboboxStoreState<T>["selectedValue"]) => void
  >;
  /**
   * A reference to a [tag store](https://ariakit.com/apis/use-tag-store). It's
   * automatically set when rendering a combobox within a tag list.
   */
  tag?: TagStore | null;
}

export interface ComboboxStoreProps<
  T extends ComboboxStoreSelectedValue = ComboboxStoreSelectedValue,
>
  extends
    ComboboxStoreOptions<T>,
    Omit<Core.ComboboxStoreProps<T>, "tag" | "disclosure"> {}

export interface ComboboxStore<
  T extends ComboboxStoreSelectedValue = ComboboxStoreSelectedValue,
>
  extends
    ComboboxStoreFunctions<T>,
    Omit<Store<Core.ComboboxStore<T>>, "tag" | "disclosure"> {}