import * as Core from "@ariakit/components/radio/radio-store";
import { useStore, useStoreProps } from "@ariakit/react-store";
import type { Store } from "@ariakit/react-store";
import type {
  CompositeStoreFunctions,
  CompositeStoreOptions,
  CompositeStoreState,
} from "../composite/composite-store.ts";
import {
  useCompositeStoreOptions,
  useCompositeStoreProps,
} from "../composite/composite-store.ts";

export function useRadioStoreProps<T extends Core.RadioStore>(
  store: T,
  update: () => void,
  props: RadioStoreProps,
) {
  store = useCompositeStoreProps(store, update, props);
  useStoreProps(store, props, "value", "setValue");
  return store;
}

/**
 * Creates a radio store to control the state of
 * [Radio](https://ariakit.com/components/radio) components.
 * @see https://ariakit.com/components/radio
 * @example
 * ```jsx
 * const radio = useRadioStore();
 *
 * <RadioGroup store={radio}>
 *   <Radio value="Apple" />
 *   <Radio value="Orange" />
 * </RadioGroup>
 * ```
 */
export function useRadioStore(props: RadioStoreProps = {}): RadioStore {
  props = useCompositeStoreOptions(props);
  const [store, update] = useStore(Core.createRadioStore, props);
  return useRadioStoreProps(store, update, props);
}

export interface RadioStoreState
  extends Core.RadioStoreState, CompositeStoreState {}

export interface RadioStoreFunctions
  extends Core.RadioStoreFunctions, CompositeStoreFunctions {}

export interface RadioStoreOptions
  extends Core.RadioStoreOptions, CompositeStoreOptions {
  /**
   * Function that will be called when the
   * [`value`](https://ariakit.com/reference/radio-provider#value) state
   * changes.
   */
  setValue?: (value: RadioStoreState["value"]) => void;
}

export interface RadioStoreProps
  extends RadioStoreOptions, Core.RadioStoreProps {}

export interface RadioStore
  extends RadioStoreFunctions, Store<Core.RadioStore> {}