import valueParser from 'postcss-value-parser';

import { assert, isRegExp, isString } from './validateTypes.mjs';
import { isValueFunction } from './typeGuards.mjs';

/** @typedef {(expression: string, expressionIndex: number, funcNode: valueParser.FunctionNode, parsedValue: valueParser.ParsedValue) => void} Callback */

/**
 * Search a CSS string for functions by name.
 * For every match, invoke the callback, passing the function's
 * "argument(s) string" (whatever is inside the parentheses)
 * as an argument.
 *
 * Callback will be called once for every matching function found,
 * with the function's "argument(s) string" and its starting index
 * as the arguments.
 *
 * @param {string} source
 * @param {string | RegExp} functionName
 * @param {Callback} callback
 * @returns {valueParser.ParsedValue}
 */
export default function functionArgumentsSearch(source, functionName, callback) {
	const parsedValue = valueParser(source);

	return parsedValue.walk((node) => {
		if (!isValueFunction(node)) return;

		const { value } = node;

		if (isString(functionName) && value !== functionName) return;

		if (isRegExp(functionName) && !functionName.test(node.value)) return;

		const parenLength = 1; // == '('
		const expressionIndex = node.sourceIndex + value.length + parenLength;

		let expression;

		if (node.value.toLowerCase() === 'url') {
			// Re-parse url tokens as component values to support interpolation patterns in non-standard CSS
			const [reParsedValue] = valueParser(
				// Slice the source value at the opening parenthesis `url(foo) trailing` -> `(foo) trailing`
				// When parsed as a list of component values the first node will be a simple block
				source.slice(node.sourceIndex + value.length),
			).nodes;

			assert(reParsedValue);

			const endIndex =
				expressionIndex +
				reParsedValue.sourceEndIndex -
				1 -
				(source[expressionIndex + reParsedValue.sourceEndIndex - 2] === ')' ? parenLength : 0);

			expression = source.slice(expressionIndex, endIndex);
		} else {
			const endIndex =
				node.sourceEndIndex - (source[node.sourceEndIndex - 1] === ')' ? parenLength : 0);

			expression = source.slice(expressionIndex, endIndex);
		}

		callback(expression, expressionIndex, node, parsedValue);
	});
}
