import * as Core from "@ariakit/components/composite/composite-overflow-store";
import { useStore } from "@ariakit/react-store";
import type { Store } from "@ariakit/react-store";
import type {
  PopoverStoreFunctions,
  PopoverStoreOptions,
  PopoverStoreState,
} from "../popover/popover-store.ts";
import { usePopoverStoreProps } from "../popover/popover-store.ts";

export function useCompositeOverflowStoreProps<
  T extends Core.CompositeOverflowStore,
>(store: T, update: () => void, props: CompositeOverflowStoreProps) {
  return usePopoverStoreProps(store, update, props);
}

/**
 * Creates a composite overflow store.
 * @see https://ariakit.com/components/composite
 * @example
 * ```jsx
 * const composite = useCompositeStore();
 * const overflow = useCompositeOverflowStore();
 * <Composite store={composite}>
 *   <CompositeItem>Item 1</CompositeItem>
 *   <CompositeItem>Item 2</CompositeItem>
 *   <CompositeOverflowDisclosure store={overflow}>
 *     +2 items
 *   </CompositeOverflowDisclosure>
 *   <CompositeOverflow store={overflow}>
 *     <CompositeItem>Item 3</CompositeItem>
 *     <CompositeItem>Item 4</CompositeItem>
 *   </CompositeOverflow>
 * </Composite>
 * ```
 */
export function useCompositeOverflowStore(
  props: CompositeOverflowStoreProps = {},
): CompositeOverflowStore {
  const [store, update] = useStore(Core.createCompositeOverflowStore, props);
  return useCompositeOverflowStoreProps(store, update, props);
}

export interface CompositeOverflowStoreState
  extends Core.CompositeOverflowStoreState, PopoverStoreState {}

export interface CompositeOverflowStoreFunctions
  extends
    Omit<Core.CompositeOverflowStoreFunctions, "disclosure">,
    PopoverStoreFunctions {}

export interface CompositeOverflowStoreOptions
  extends
    Omit<Core.CompositeOverflowStoreOptions, "disclosure">,
    PopoverStoreOptions {}

export type CompositeOverflowStoreProps = CompositeOverflowStoreOptions &
  Omit<Core.CompositeOverflowStoreProps, "disclosure">;

export type CompositeOverflowStore = CompositeOverflowStoreFunctions &
  Omit<Store<Core.CompositeOverflowStore>, "disclosure">;