@nx/js
Version:
86 lines (85 loc) • 2.51 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", { value: true });
exports.highlight = highlight;
// Adapted from https://raw.githubusercontent.com/babel/babel/4108524/packages/babel-highlight/src/index.js
const jsTokens = require("js-tokens");
const chalk = require("chalk");
const identifiers_1 = require("./identifiers");
/**
* Chalk styles for token types.
*/
function getDefs(chalk) {
return {
keyword: chalk.cyan,
capitalized: chalk.yellow,
jsx_tag: chalk.yellow,
punctuator: chalk.yellow,
// bracket: intentionally omitted.
number: chalk.magenta,
string: chalk.green,
regex: chalk.magenta,
comment: chalk.grey,
invalid: chalk.white.bgRed.bold,
};
}
/**
* RegExp to test for newlines in terminal.
*/
const NEWLINE = /\r\n|[\n\r\u2028\u2029]/;
/**
* RegExp to test for what seems to be a JSX tag name.
*/
const JSX_TAG = /^[a-z][\w-]*$/i;
/**
* RegExp to test for the three types of brackets.
*/
const BRACKET = /^[()[\]{}]$/;
/**
* Get the type of token, specifying punctuator type.
*/
function getTokenType(match) {
const [offset, text] = match.slice(-2);
const token = jsTokens.matchToToken(match);
if (token.type === 'name') {
if ((0, identifiers_1.isKeyword)(token.value) || (0, identifiers_1.isReservedWord)(token.value)) {
return 'keyword';
}
if (JSX_TAG.test(token.value) &&
(text[offset - 1] === '<' || text.slice(offset - 2, 2) == '</')) {
return 'jsx_tag';
}
if (token.value[0] !== token.value[0].toLowerCase()) {
return 'capitalized';
}
}
if (token.type === 'punctuator' && BRACKET.test(token.value)) {
return 'bracket';
}
if (token.type === 'invalid' &&
(token.value === '@' || token.value === '#')) {
return 'punctuator';
}
return token.type;
}
/**
* Highlight `text` using the token definitions in `defs`.
*/
function highlightTokens(defs, text) {
return text.replace(jsTokens.default, function (...args) {
const type = getTokenType(args);
const colorize = defs[type];
if (colorize) {
return args[0]
.split(NEWLINE)
.map((str) => colorize(str))
.join('\n');
}
else {
return args[0];
}
});
}
function highlight(code) {
const defs = getDefs(chalk);
return highlightTokens(defs, code);
}