import { createElement, createHook, forwardRef } from "@ariakit/react-utils";
import type { Props } from "@ariakit/react-utils";
import type { ElementType } from "react";
import type { PopoverHeadingOptions } from "../popover/popover-heading.tsx";
import { usePopoverHeading } from "../popover/popover-heading.tsx";
import type { HovercardStore } from "./hovercard-store.ts";

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

/**
 * Returns props to create a `HovercardHeading` component. This hook must be
 * used in a component that's wrapped with `Hovercard` so the `aria-labelledby`
 * prop is properly set on the hovercard element.
 * @see https://ariakit.com/components/hovercard
 * @example
 * ```jsx
 * // This component must be wrapped with Hovercard
 * const props = useHovercardHeading();
 * <Role {...props}>Heading</Role>
 * ```
 */
export const useHovercardHeading = createHook<TagName, HovercardHeadingOptions>(
  function useHovercardHeading(props) {
    props = usePopoverHeading(props);
    return props;
  },
);

/**
 * Renders a heading in a hovercard. This component must be wrapped within
 * [`Hovercard`](https://ariakit.com/reference/hovercard) so the
 * `aria-labelledby` prop is properly set on the content element.
 * @see https://ariakit.com/components/hovercard
 * @example
 * ```jsx {3}
 * <HovercardProvider>
 *   <Hovercard>
 *     <HovercardHeading>Heading</HovercardHeading>
 *   </Hovercard>
 * </HovercardProvider>
 * ```
 */
export const HovercardHeading = forwardRef(function HovercardHeading(
  props: HovercardHeadingProps,
) {
  const htmlProps = useHovercardHeading(props);
  return createElement(TagName, htmlProps);
});

export interface HovercardHeadingOptions<
  T extends ElementType = TagName,
> extends PopoverHeadingOptions<T> {
  /**
   * Object returned by the
   * [`useHovercardStore`](https://ariakit.com/reference/use-hovercard-store)
   * hook.
   *
   * **Note**: This prop has no effect on this component. The heading is
   * linked to the closest
   * [`Hovercard`](https://ariakit.com/reference/hovercard) component through
   * React context, so it must be rendered inside the hovercard for the
   * `aria-labelledby` prop to be set on the hovercard element.
   */
  store?: HovercardStore;
}

export type HovercardHeadingProps<T extends ElementType = TagName> = Props<
  T,
  HovercardHeadingOptions<T>
>;