import {
  useWrapElement,
  createElement,
  createHook,
  forwardRef,
} from "@ariakit/react-utils";
import type { Props } from "@ariakit/react-utils";
import type { ElementType } from "react";
import { CompositeScopedContextProvider } from "../composite/composite-context.tsx";
import type { FormGroupOptions } from "./form-group.tsx";
import { useFormGroup } from "./form-group.tsx";

const TagName = "div" satisfies ElementType;
type TagName = typeof TagName;

/**
 * Returns props to create a `FormRadioGroup` component.
 * @see https://ariakit.com/components/form
 * @example
 * ```jsx
 * const store = useFormStore({ defaultValues: { color: "red" } });
 * const props = useFormRadioGroup({ store });
 * <Form store={store}>
 *   <Role {...props}>
 *     <FormGroupLabel>Favorite color</FormGroupLabel>
 *     <FormRadio name={store.names.color} value="red" />
 *     <FormRadio name={store.names.color} value="blue" />
 *     <FormRadio name={store.names.color} value="green" />
 *   </Role>
 * </Form>
 * ```
 */
export const useFormRadioGroup = createHook<TagName, FormRadioGroupOptions>(
  function useFormRadioGroup({ store, ...props }) {
    // Reset the composite context so FormRadio items inside this group
    // don't register to an unrelated ancestor composite store (e.g.,
    // TabStore). Form controls explicitly pass their store to
    // useCollectionItem, so they're not affected by this reset.
    props = useWrapElement(
      props,
      (element) => (
        <CompositeScopedContextProvider value={undefined}>
          {element}
        </CompositeScopedContextProvider>
      ),
      [],
    );

    props = { role: "radiogroup", ...props };
    props = useFormGroup(props);
    return props;
  },
);

/**
 * Renders a group element for
 * [`FormRadio`](https://ariakit.com/reference/form-radio) elements. The
 * [`FormGroupLabel`](https://ariakit.com/reference/form-group-label) component
 * can be used inside this component so the `aria-labelledby` prop is properly
 * set on the group element.
 * @see https://ariakit.com/components/form
 * @example
 * ```jsx {8-13}
 * const form = useFormStore({
 *   defaultValues: {
 *     color: "red",
 *   },
 * });
 *
 * <Form store={form}>
 *   <FormRadioGroup>
 *     <FormGroupLabel>Favorite color</FormGroupLabel>
 *     <FormRadio name={form.names.color} value="red" />
 *     <FormRadio name={form.names.color} value="blue" />
 *     <FormRadio name={form.names.color} value="green" />
 *   </FormRadioGroup>
 * </Form>
 * ```
 */
export const FormRadioGroup = forwardRef(function FormRadioGroup(
  props: FormRadioGroupProps,
) {
  const htmlProps = useFormRadioGroup(props);
  return createElement(TagName, htmlProps);
});

export type FormRadioGroupOptions<T extends ElementType = TagName> =
  FormGroupOptions<T>;

export type FormRadioGroupProps<T extends ElementType = TagName> = Props<
  T,
  FormRadioGroupOptions<T>
>;