You can not select more than 25 topics
			Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
		
		
		
		
		
			
		
			
				
					135 lines
				
				3.3 KiB
			
		
		
			
		
	
	
					135 lines
				
				3.3 KiB
			| 
											3 years ago
										 | 'use strict'; | ||
|  | const 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; | ||
|  | const STYLE_REGEX = /(?:^|\.)(\w+)(?:\(([^)]*)\))?/g; | ||
|  | const STRING_REGEX = /^(['"])((?:\\.|(?!\1)[^\\])*)\1$/; | ||
|  | const ESCAPE_REGEX = /\\(u(?:[a-f\d]{4}|{[a-f\d]{1,6}})|x[a-f\d]{2}|.)|([^\\])/gi; | ||
|  | 
 | ||
|  | const ESCAPES = new Map([ | ||
|  | 	['n', '\n'], | ||
|  | 	['r', '\r'], | ||
|  | 	['t', '\t'], | ||
|  | 	['b', '\b'], | ||
|  | 	['f', '\f'], | ||
|  | 	['v', '\v'], | ||
|  | 	['0', '\0'], | ||
|  | 	['\\', '\\'], | ||
|  | 	['e', '\u001B'], | ||
|  | 	['a', '\u0007'] | ||
|  | ]); | ||
|  | 
 | ||
|  | function unescape(c) { | ||
|  | 	const u = c[0] === 'u'; | ||
|  | 	const bracket = c[1] === '{'; | ||
|  | 
 | ||
|  | 	if ((u && !bracket && c.length === 5) || (c[0] === 'x' && c.length === 3)) { | ||
|  | 		return String.fromCharCode(parseInt(c.slice(1), 16)); | ||
|  | 	} | ||
|  | 
 | ||
|  | 	if (u && bracket) { | ||
|  | 		return String.fromCodePoint(parseInt(c.slice(2, -1), 16)); | ||
|  | 	} | ||
|  | 
 | ||
|  | 	return ESCAPES.get(c) || c; | ||
|  | } | ||
|  | 
 | ||
|  | function parseArguments(name, arguments_) { | ||
|  | 	const results = []; | ||
|  | 	const chunks = arguments_.trim().split(/\s*,\s*/g); | ||
|  | 	let matches; | ||
|  | 
 | ||
|  | 	for (const chunk of chunks) { | ||
|  | 		const number = Number(chunk); | ||
|  | 		if (!Number.isNaN(number)) { | ||
|  | 			results.push(number); | ||
|  | 		} else if ((matches = chunk.match(STRING_REGEX))) { | ||
|  | 			results.push(matches[2].replace(ESCAPE_REGEX, (m, escape, character) => escape ? unescape(escape) : character)); | ||
|  | 		} else { | ||
|  | 			throw new Error(`Invalid Chalk template style argument: ${chunk} (in style '${name}')`); | ||
|  | 		} | ||
|  | 	} | ||
|  | 
 | ||
|  | 	return results; | ||
|  | } | ||
|  | 
 | ||
|  | function parseStyle(style) { | ||
|  | 	STYLE_REGEX.lastIndex = 0; | ||
|  | 
 | ||
|  | 	const results = []; | ||
|  | 	let matches; | ||
|  | 
 | ||
|  | 	while ((matches = STYLE_REGEX.exec(style)) !== null) { | ||
|  | 		const name = matches[1]; | ||
|  | 
 | ||
|  | 		if (matches[2]) { | ||
|  | 			const args = parseArguments(name, matches[2]); | ||
|  | 			results.push([name].concat(args)); | ||
|  | 		} else { | ||
|  | 			results.push([name]); | ||
|  | 		} | ||
|  | 	} | ||
|  | 
 | ||
|  | 	return results; | ||
|  | } | ||
|  | 
 | ||
|  | function buildStyle(chalk, styles) { | ||
|  | 	const enabled = {}; | ||
|  | 
 | ||
|  | 	for (const layer of styles) { | ||
|  | 		for (const style of layer.styles) { | ||
|  | 			enabled[style[0]] = layer.inverse ? null : style.slice(1); | ||
|  | 		} | ||
|  | 	} | ||
|  | 
 | ||
|  | 	let current = chalk; | ||
|  | 	for (const [styleName, styles] of Object.entries(enabled)) { | ||
|  | 		if (!Array.isArray(styles)) { | ||
|  | 			continue; | ||
|  | 		} | ||
|  | 
 | ||
|  | 		if (!(styleName in current)) { | ||
|  | 			throw new Error(`Unknown Chalk style: ${styleName}`); | ||
|  | 		} | ||
|  | 
 | ||
|  | 		current = styles.length > 0 ? current[styleName](...styles) : current[styleName]; | ||
|  | 	} | ||
|  | 
 | ||
|  | 	return current; | ||
|  | } | ||
|  | 
 | ||
|  | module.exports = (chalk, temporary) => { | ||
|  | 	const styles = []; | ||
|  | 	const chunks = []; | ||
|  | 	let chunk = []; | ||
|  | 
 | ||
|  | 	// eslint-disable-next-line max-params
 | ||
|  | 	temporary.replace(TEMPLATE_REGEX, (m, escapeCharacter, inverse, style, close, character) => { | ||
|  | 		if (escapeCharacter) { | ||
|  | 			chunk.push(unescape(escapeCharacter)); | ||
|  | 		} else if (style) { | ||
|  | 			const string = chunk.join(''); | ||
|  | 			chunk = []; | ||
|  | 			chunks.push(styles.length === 0 ? string : buildStyle(chalk, styles)(string)); | ||
|  | 			styles.push({inverse, styles: parseStyle(style)}); | ||
|  | 		} else if (close) { | ||
|  | 			if (styles.length === 0) { | ||
|  | 				throw new Error('Found extraneous } in Chalk template literal'); | ||
|  | 			} | ||
|  | 
 | ||
|  | 			chunks.push(buildStyle(chalk, styles)(chunk.join(''))); | ||
|  | 			chunk = []; | ||
|  | 			styles.pop(); | ||
|  | 		} else { | ||
|  | 			chunk.push(character); | ||
|  | 		} | ||
|  | 	}); | ||
|  | 
 | ||
|  | 	chunks.push(chunk.join('')); | ||
|  | 
 | ||
|  | 	if (styles.length > 0) { | ||
|  | 		const errMessage = `Chalk template literal is missing ${styles.length} closing bracket${styles.length === 1 ? '' : 's'} (\`}\`)`; | ||
|  | 		throw new Error(errMessage); | ||
|  | 	} | ||
|  | 
 | ||
|  | 	return chunks.join(''); | ||
|  | }; |