import type { JSX, ValidComponent } from "@solidjs/web";
import { dynamic } from "@solidjs/web";
import { createMemo, omit } from "solid-js";
import { composeRefs, userRef } from "../../internals/compose-refs";
import { mergeRootProps } from "../../internals/merge-root-props";
import { resolveClassStyle } from "../../internals/resolve-class-style";
import type { BaseUIComponentProps } from "../../internals/types";
import {
	type ComboboxAlign,
	type ComboboxSide,
	useComboboxPositionerContext,
} from "../positioner/positioner-context";
import { useComboboxRootContext } from "../root/combobox-root-context";

export interface ComboboxArrowState {
	open: boolean;
	side: ComboboxSide;
	align: ComboboxAlign;
	anchorHidden: boolean;
}

export interface ComboboxArrowProps
	extends BaseUIComponentProps<"div", ComboboxArrowState> {
	as?: ValidComponent;
}

/** Displays an element positioned against the anchor. Renders a `<div>`. */
export function ComboboxArrow(props: ComboboxArrowProps): JSX.Element {
	const root = useComboboxRootContext();
	const positioner = useComboboxPositionerContext();

	const state = createMemo<ComboboxArrowState>(() => ({
		open: root.open(),
		side: positioner.side(),
		align: positioner.align(),
		anchorHidden: positioner.anchorHidden(),
	}));

	const internal = {
		"aria-hidden": true,
		get style() {
			return positioner.arrowStyles();
		},
		// P3: enumerate every data-* derived from state so Solid's spread
		// installs reactive bindings up front.
		get "data-open"() {
			return state().open ? "" : undefined;
		},
		get "data-closed"() {
			return state().open ? undefined : "";
		},
		get "data-anchor-hidden"() {
			return state().anchorHidden ? "" : undefined;
		},
		get "data-side"() {
			return state().side;
		},
		get "data-align"() {
			return state().align;
		},
	};

	const userClassStyle = resolveClassStyle(props, () => state());
	const userProps = omit(props, "as", "class", "style", "attrs");

	let merged: Record<string, unknown> = mergeRootProps(
		internal,
		userClassStyle,
	);
	merged = mergeRootProps(merged, userProps);

	const setArrowRef = (el: HTMLElement | null): void => {
		positioner.arrowRef.current = el;
	};

	const Tag = dynamic(() => props.as ?? "div");
	return <Tag {...merged} ref={composeRefs(setArrowRef, userRef(props))} />;
}