import plugin from "tailwindcss/plugin";

/*
   This plugin is completely replicated from https://github.com/tailwindlabs/tailwindcss-container-queries
   Except altered to support @container(min-height: 100px) and (min-width: 100px) to be written as @[height:100px|width:100px]

 */
export default plugin(
	function containerQueries({ matchUtilities, matchVariant, theme }) {
		let values: Record<string, string> = theme("containers") ?? {};

		function parseValue(value: string) {
			let numericValue = value.match(/^(\d+\.\d+|\d+|\.\d+)\D+/)?.[1] ?? null;
			if (numericValue === null) return null;
			return parseFloat(value);
		}

		matchUtilities(
			{
				"@container": (value, { modifier }) => {
					return {
						"container-type": value,
						"container-name": modifier,
					};
				},
			},
			{
				values: {
					DEFAULT: "inline-size",
					normal: "normal",
				},
				modifiers: "any",
			},
		);

		matchVariant(
			"@",
			(value = "", { modifier }) => {
				const [width, height] = (() => {
					if (!value.includes("width:") && value.includes("height")) return [null, value.replace("height:", "")];
					return value.replace("mwidth:", "").replace("mheight:", "").replace("width:", "").replace("height:", "").split("|");
				})();
				const isHeight = value.includes("height");
				const isWidth = value.includes("width");
				const isMaxHeight = value.includes("mheight");
				const isMaxWidth = value.includes("mwidth");
				let pwidth = width ? parseValue(width) : null;
				let pheight = height ? parseValue(height) : null;
				const selector = (() => {
					const w = isMaxWidth ? "max-width" : "min-width";
					const h = isMaxHeight ? "max-height" : "min-height";
					if (!isHeight && !isWidth) return `(${w}: ${width})`;
					if (isHeight && isWidth) return `(${w}:${width}) and (${h}:${height})`;
					if (isHeight) return `(${h}:${height})`;
					return `(${w}:${width})`;
				})();
				return pwidth !== null || pheight !== null ? `@container ${modifier ?? ""} ${selector}` : [];
			},
			{
				values,
				sort(aVariant, zVariant) {
					let a = parseFloat(aVariant.value);
					let z = parseFloat(zVariant.value);

					if (a === null || z === null) return 0;

					// Sort values themselves regardless of unit
					if (a - z !== 0) return a - z;

					let aLabel = aVariant.modifier ?? "";
					let zLabel = zVariant.modifier ?? "";

					// Explicitly move empty labels to the end
					if (aLabel === "" && zLabel !== "") {
						return 1;
					} else if (aLabel !== "" && zLabel === "") {
						return -1;
					}

					// Sort labels alphabetically in the English locale
					// We are intentionally overriding the locale because we do not want the sort to
					// be affected by the machine's locale (be it a developer or CI environment)
					return aLabel.localeCompare(zLabel, "en", { numeric: true });
				},
			},
		);
	},
	{
		theme: {
			containers: {
				xs: "20rem",
				sm: "24rem",
				md: "28rem",
				lg: "32rem",
				xl: "36rem",
				"2xl": "42rem",
				"3xl": "48rem",
				"4xl": "56rem",
				"5xl": "64rem",
				"6xl": "72rem",
				"7xl": "80rem",
			},
		},
	},
);