import { createStore } from "@ariakit/store";
import type { Store, StoreOptions, StoreProps } from "@ariakit/store";
import { defaultValue } from "@ariakit/utils";
import type { SetState } from "@ariakit/utils";
import type {
  CompositeStoreFunctions,
  CompositeStoreOptions,
  CompositeStoreState,
} from "../composite/composite-store.ts";
import { createCompositeStore } from "../composite/composite-store.ts";

/**
 * Creates a radio store.
 */
export function createRadioStore(props: RadioStoreProps = {}): RadioStore {
  const syncState = props.store?.getState();

  const composite = createCompositeStore({
    ...props,
    focusLoop: defaultValue(props.focusLoop, syncState?.focusLoop, true),
  });

  const initialState: RadioStoreState = {
    ...composite.getState(),
    value: defaultValue(
      props.value,
      syncState?.value,
      props.defaultValue,
      null,
    ),
  };

  const radio = createStore(initialState, composite, props.store);

  return {
    ...composite,
    ...radio,
    setValue: (value) => radio.setState("value", value),
  };
}

export interface RadioStoreState extends CompositeStoreState {
  /** @default true */
  focusLoop: CompositeStoreState["focusLoop"];
  /**
   * The value of the radio group.
   * @default null
   */
  value: string | number | null;
}

export interface RadioStoreFunctions extends CompositeStoreFunctions {
  /**
   * Sets the [`value`](https://ariakit.com/reference/radio-provider#value)
   * state.
   * @example
   * store.setValue("apple");
   * store.setValue((value) => value === "apple" ? "orange" : "apple");
   */
  setValue: SetState<RadioStoreState["value"]>;
}

export interface RadioStoreOptions
  extends
    StoreOptions<RadioStoreState, "focusLoop" | "value">,
    CompositeStoreOptions {
  /**
   * The default value of the radio group.
   * @default null
   */
  defaultValue?: RadioStoreState["value"];
}

export interface RadioStoreProps
  extends RadioStoreOptions, StoreProps<RadioStoreState> {}

export interface RadioStore
  extends RadioStoreFunctions, Store<RadioStoreState> {}