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 { ComboboxSide } from "../positioner/positioner-context";
import { useComboboxRootContext } from "../root/combobox-root-context";

export interface ComboboxInputGroupState {
	open: boolean;
	disabled: boolean;
	readOnly: boolean;
	popupSide: ComboboxSide | null;
	listEmpty: boolean;
	placeholder: boolean;
}

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

/**
 * A wrapper for the input and its associated controls. Renders a
 * `<div role="group">`.
 */
export function ComboboxInputGroup(
	props: ComboboxInputGroupProps,
): JSX.Element {
	const root = useComboboxRootContext();

	const listEmpty = createMemo<boolean>(
		() => root.flatFilteredItems().length === 0,
	);
	const placeholder = createMemo<boolean>(() => {
		if (root.selectionMode() === "none") return false;
		const sv = root.selectedValue();
		if (sv == null) return true;
		if (Array.isArray(sv)) return sv.length === 0;
		return false;
	});
	const popupSide = createMemo<ComboboxSide | null>(() => {
		if (!root.mounted() || root.positionerElement() == null) return null;
		return root.popupSide();
	});

	const state = createMemo<ComboboxInputGroupState>(() => ({
		open: root.open(),
		disabled: root.disabled(),
		readOnly: root.readOnly(),
		popupSide: popupSide(),
		listEmpty: listEmpty(),
		placeholder: placeholder(),
	}));

	const internal = {
		role: "group" as const,
		// triggerStateAttributesMapping projection.
		get "data-popup-open"() {
			return state().open ? "" : undefined;
		},
		get "data-pressed"() {
			return state().open ? "" : undefined;
		},
		get "data-disabled"() {
			return state().disabled ? "" : undefined;
		},
		get "data-readonly"() {
			return state().readOnly ? "" : undefined;
		},
		get "data-list-empty"() {
			return state().listEmpty ? "" : undefined;
		},
		get "data-placeholder"() {
			return state().placeholder ? "" : undefined;
		},
		get "data-popup-side"() {
			return state().popupSide ?? undefined;
		},
	};

	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 Tag = dynamic(() => props.as ?? "div");
	return <Tag {...merged} ref={composeRefs(userRef(props))} />;
}