1'use strict';
2const TEMPLATE_REGEX = /(?:\\(u(?:[a-f\d]{4}|\{[a-f\d]{1,6}\})|x[a-f\d]{2}|.))|(?:\{(~)?(\w+(?:\([^)]*\))?(?:\.\w+(?:\([^)]*\))?)*)(?:[ \t]|(?=\r?\n)))|(\})|((?:.|[\r\n\f])+?)/gi;
3const STYLE_REGEX = /(?:^|\.)(\w+)(?:\(([^)]*)\))?/g;
4const STRING_REGEX = /^(['"])((?:\\.|(?!\1)[^\\])*)\1$/;
5const ESCAPE_REGEX = /\\(u(?:[a-f\d]{4}|{[a-f\d]{1,6}})|x[a-f\d]{2}|.)|([^\\])/gi;
6
7const ESCAPES = new Map([
8	['n', '\n'],
9	['r', '\r'],
10	['t', '\t'],
11	['b', '\b'],
12	['f', '\f'],
13	['v', '\v'],
14	['0', '\0'],
15	['\\', '\\'],
16	['e', '\u001B'],
17	['a', '\u0007']
18]);
19
20function unescape(c) {
21	const u = c[0] === 'u';
22	const bracket = c[1] === '{';
23
24	if ((u && !bracket && c.length === 5) || (c[0] === 'x' && c.length === 3)) {
25		return String.fromCharCode(parseInt(c.slice(1), 16));
26	}
27
28	if (u && bracket) {
29		return String.fromCodePoint(parseInt(c.slice(2, -1), 16));
30	}
31
32	return ESCAPES.get(c) || c;
33}
34
35function parseArguments(name, arguments_) {
36	const results = [];
37	const chunks = arguments_.trim().split(/\s*,\s*/g);
38	let matches;
39
40	for (const chunk of chunks) {
41		const number = Number(chunk);
42		if (!Number.isNaN(number)) {
43			results.push(number);
44		} else if ((matches = chunk.match(STRING_REGEX))) {
45			results.push(matches[2].replace(ESCAPE_REGEX, (m, escape, character) => escape ? unescape(escape) : character));
46		} else {
47			throw new Error(`Invalid Chalk template style argument: ${chunk} (in style '${name}')`);
48		}
49	}
50
51	return results;
52}
53
54function parseStyle(style) {
55	STYLE_REGEX.lastIndex = 0;
56
57	const results = [];
58	let matches;
59
60	while ((matches = STYLE_REGEX.exec(style)) !== null) {
61		const name = matches[1];
62
63		if (matches[2]) {
64			const args = parseArguments(name, matches[2]);
65			results.push([name].concat(args));
66		} else {
67			results.push([name]);
68		}
69	}
70
71	return results;
72}
73
74function buildStyle(chalk, styles) {
75	const enabled = {};
76
77	for (const layer of styles) {
78		for (const style of layer.styles) {
79			enabled[style[0]] = layer.inverse ? null : style.slice(1);
80		}
81	}
82
83	let current = chalk;
84	for (const [styleName, styles] of Object.entries(enabled)) {
85		if (!Array.isArray(styles)) {
86			continue;
87		}
88
89		if (!(styleName in current)) {
90			throw new Error(`Unknown Chalk style: ${styleName}`);
91		}
92
93		current = styles.length > 0 ? current[styleName](...styles) : current[styleName];
94	}
95
96	return current;
97}
98
99module.exports = (chalk, temporary) => {
100	const styles = [];
101	const chunks = [];
102	let chunk = [];
103
104	// eslint-disable-next-line max-params
105	temporary.replace(TEMPLATE_REGEX, (m, escapeCharacter, inverse, style, close, character) => {
106		if (escapeCharacter) {
107			chunk.push(unescape(escapeCharacter));
108		} else if (style) {
109			const string = chunk.join('');
110			chunk = [];
111			chunks.push(styles.length === 0 ? string : buildStyle(chalk, styles)(string));
112			styles.push({inverse, styles: parseStyle(style)});
113		} else if (close) {
114			if (styles.length === 0) {
115				throw new Error('Found extraneous } in Chalk template literal');
116			}
117
118			chunks.push(buildStyle(chalk, styles)(chunk.join('')));
119			chunk = [];
120			styles.pop();
121		} else {
122			chunk.push(character);
123		}
124	});
125
126	chunks.push(chunk.join(''));
127
128	if (styles.length > 0) {
129		const errMessage = `Chalk template literal is missing ${styles.length} closing bracket${styles.length === 1 ? '' : 's'} (\`}\`)`;
130		throw new Error(errMessage);
131	}
132
133	return chunks.join('');
134};
135