import type { PickRequired } from "@ariakit/utils";
import type { ReactElement, ReactNode } from "react";
import { ComboboxContextProvider } from "./combobox-context.tsx";
import type {
  ComboboxStoreProps,
  ComboboxStoreSelectedValue,
} from "./combobox-store.ts";
import { useComboboxStore } from "./combobox-store.ts";

type Value = ComboboxStoreSelectedValue;

/**
 * Provides a combobox store that controls the state of
 * [Combobox](https://ariakit.com/components/combobox) components.
 * @see https://ariakit.com/components/combobox
 * @example
 * ```jsx
 * <ComboboxProvider defaultInputValue="">
 *   <Combobox />
 *   <ComboboxPopover>
 *     <ComboboxItem value="Apple" />
 *     <ComboboxItem value="Banana" />
 *     <ComboboxItem value="Orange" />
 *   </ComboboxPopover>
 * </ComboboxProvider>
 * ```
 */
export function ComboboxProvider<T extends Value = Value>(
  props: PickRequired<
    ComboboxProviderProps<T>,
    "selectedValue" | "defaultSelectedValue"
  >,
): ReactElement;

export function ComboboxProvider(props?: ComboboxProviderProps): ReactElement;

export function ComboboxProvider(props: ComboboxProviderProps = {}) {
  const store = useComboboxStore(props);
  return (
    <ComboboxContextProvider value={store}>
      {props.children}
    </ComboboxContextProvider>
  );
}

export interface ComboboxProviderProps<
  T extends Value = Value,
> extends ComboboxStoreProps<T> {
  children?: ReactNode;
}