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 { TransitionStatus } from "../../internals/use-transition-status/use-transition-status";
import {
	type TooltipAlign,
	type TooltipSide,
	useTooltipPositionerContext,
} from "../tooltip-positioner-context";
import {
	type TooltipInstantType,
	useTooltipRootContext,
} from "../tooltip-root-context";

export interface TooltipPopupState {
	open: boolean;
	side: TooltipSide;
	align: TooltipAlign;
	instant: TooltipInstantType;
	transitionStatus: TransitionStatus;
}

export interface TooltipPopupProps
	extends BaseUIComponentProps<"div", TooltipPopupState> {
	/**
	 * Tag/component to render. Defaults to `"div"`. Replaces the
	 * pre-spike `render` prop (axis G).
	 */
	as?: ValidComponent;
}

/**
 * A container for the tooltip contents. Renders a `<div>` element. Mirrors
 * Base UI React's `<Tooltip.Popup>`.
 *
 * Solid 2.0-native rewire (§1 wave 2): replaces `renderElement` with
 * `dynamic(() => props.as ?? "div")` and composes
 * `floatingProps` + state-attr getters via `mergeRootProps`.
 */
export function TooltipPopup(props: TooltipPopupProps): JSX.Element {
	const rootContext = useTooltipRootContext();
	const positionerContext = useTooltipPositionerContext();

	const floatingProps = rootContext.interactions.getFloatingProps();

	const state = createMemo<TooltipPopupState>(() => ({
		open: rootContext.open(),
		side: positionerContext.side(),
		align: positionerContext.align(),
		instant: rootContext.instantType(),
		transitionStatus: rootContext.transitionStatus(),
	}));

	const popupRef = (el: HTMLElement | null) => {
		rootContext.setPopupElement(el);
	};

	const stateAttrs = {
		get "data-open"() {
			return state().open ? "" : undefined;
		},
		get "data-closed"() {
			return state().open ? undefined : "";
		},
		get "data-side"() {
			return state().side;
		},
		get "data-align"() {
			return state().align;
		},
		get "data-instant"() {
			const v = state().instant;
			return v ? v : undefined;
		},
		get "data-starting-style"() {
			return state().transitionStatus === "starting" ? "" : undefined;
		},
		get "data-ending-style"() {
			return state().transitionStatus === "ending" ? "" : undefined;
		},
	};

	const userClassStyle = resolveClassStyle(props, () => state());

	const userProps = omit(props, "as", "class", "style", "attrs");

	let merged: Record<string, unknown> = mergeRootProps(
		floatingProps ?? {},
		stateAttrs,
	);
	merged = mergeRootProps(merged, userClassStyle);
	merged = mergeRootProps(merged, userProps);

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