Innovenergy_trunk/frontend/node_modules/highlight-es/index.js

46 lines
1.2 KiB
JavaScript
Raw Normal View History

var chalk = require('chalk');
var jsTokens = require('js-tokens');
var isES2016Keyword = require('is-es2016-keyword');
var jsTokensRe = jsTokens.default;
var matchToToken = jsTokens.matchToToken;
var NEWLINE = /\r\n|[\n\r\u2028\u2029]/;
var defaultRenderer = {
string: chalk.green,
punctuator: chalk.grey,
keyword: chalk.cyan,
number: chalk.magenta,
regex: chalk.magenta,
comment: chalk.grey.bold,
invalid: chalk.inverse
};
module.exports = function highlight (src, renderer) {
renderer = renderer || defaultRenderer;
return src.replace(jsTokensRe, function (value) {
var match = [];
for (var i = 0; i < arguments.length; i++)
match.push(arguments[i]);
var token = matchToToken(match);
if (token.type === 'name' && isES2016Keyword(value, true))
token.type = 'keyword';
if (renderer[token.type]) {
return value
.split(NEWLINE)
.map(function (str) {
return renderer[token.type](str);
})
.join('\n');
}
return value;
});
};