import { parse } from "css-tree";

function mapSelectors(selector: string): string[] {
	if (!selector) {
		return [];
	}

	return selector
		.split(/\s*(?![^(]*\)),\s*/)
		.map((s) => s.replace(/\u200C/g, ","));
}

function mapPosition(node, css) {
	let res: any = {
		start: {
			line: node.loc.start.line,
			column: node.loc.start.column,
		},
		end: {
			line: node.loc.end.line,
			column: node.loc.end.column,
		},
		content: css,
	};

	if (node.loc.source && node.loc.source !== "<unknown>") {
		res.source = node.loc.source;
	}

	return res;
}

function transformAst(node, css, type = null) {
	if (!node) {
		return;
	}

	if (node.type === "StyleSheet") {
		return {
			type: "stylesheet",
			stylesheet: {
				rules: node.children
					.map((child) => transformAst(child, css))
					.filter((child) => child !== null)
					.toArray(),
				parsingErrors: [],
			},
		};
	}

	if (node.type === "Atrule") {
		let atrule: any = {
			type: node.name,
		};

		if (node.name === "supports" || node.name === "media") {
			atrule[node.name] = node.prelude.value;
			atrule.rules = transformAst(node.block, css);
		} else if (node.name === "page") {
			atrule.selectors = node.prelude
				? mapSelectors(node.prelude.value)
				: [];
			atrule.declarations = transformAst(node.block, css);
		} else if (node.name === "document") {
			atrule.document = node.prelude ? node.prelude.value : "";
			atrule.vendor = "";
			atrule.rules = transformAst(node.block, css);
		} else if (node.name === "font-face") {
			atrule.declarations = transformAst(node.block, css);
		} else if (
			node.name === "import" ||
			node.name === "charset" ||
			node.name === "namespace"
		) {
			atrule[node.name] = node.prelude ? node.prelude.value : "";
		} else if (node.name === "keyframes") {
			atrule.name = node.prelude ? node.prelude.value : "";
			atrule.keyframes = transformAst(node.block, css, "keyframe");
			atrule.vendor = undefined;
		} else {
			atrule.rules = transformAst(node.block, css);
		}

		atrule.position = mapPosition(node, css);

		return atrule;
	}

	if (node.type === "Block") {
		return node.children
			.map((child) => transformAst(child, css, type))
			.filter((child) => child !== null)
			.toArray();
	}

	if (node.type === "Rule") {
		let value = node.prelude.value;

		let res: any = {
			type: type != null ? type : "rule",
			declarations: transformAst(node.block, css),
			position: mapPosition(node, css),
		};

		if (type === "keyframe") {
			res.values = mapSelectors(value);
		} else {
			res.selectors = mapSelectors(value);
		}

		return res;
	}

	if (node.type === "Comment") {
		return {
			type: "comment",
			comment: node.value,
			position: mapPosition(node, css),
		};
	}

	if (node.type === "Declaration") {
		return {
			type: "declaration",
			property: node.property,
			value: node.value.value ? node.value.value.trim() : "",
			position: mapPosition(node, css),
		};
	}

	if (node.type === "Raw") {
		return null;
	}

	throw Error(`Unknown node type ${node.type}`);
}

export function cssTreeParse(css, source): any {
	let errors = [];
	let ast = parse(css, {
		parseValue: false,
		parseAtrulePrelude: false,
		parseRulePrelude: false,
		positions: true,
		filename: source,
		onParseError: (error) => {
			errors.push(
				`${source}:${error.line}:${error.column}: ${error.formattedMessage}`
			);
		},
	});

	if (errors.length > 0) {
		throw new Error(errors[0]);
	}

	return transformAst(ast, css);
}