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