1 | "use strict";
|
2 |
|
3 | Object.defineProperty(exports, "__esModule", {
|
4 | value: true
|
5 | });
|
6 | exports.shouldHighlight = shouldHighlight;
|
7 | exports.getChalk = getChalk;
|
8 | exports.default = highlight;
|
9 |
|
10 | var _jsTokens = require("js-tokens");
|
11 |
|
12 | var _helperValidatorIdentifier = require("@babel/helper-validator-identifier");
|
13 |
|
14 | var _chalk = require("chalk");
|
15 |
|
16 | const sometimesKeywords = new Set(["as", "async", "from", "get", "of", "set"]);
|
17 |
|
18 | function getDefs(chalk) {
|
19 | return {
|
20 | keyword: chalk.cyan,
|
21 | capitalized: chalk.yellow,
|
22 | jsxIdentifier: chalk.yellow,
|
23 | punctuator: chalk.yellow,
|
24 | number: chalk.magenta,
|
25 | string: chalk.green,
|
26 | regex: chalk.magenta,
|
27 | comment: chalk.grey,
|
28 | invalid: chalk.white.bgRed.bold
|
29 | };
|
30 | }
|
31 |
|
32 | const NEWLINE = /\r\n|[\n\r\u2028\u2029]/;
|
33 | const BRACKET = /^[()[\]{}]$/;
|
34 | let tokenize;
|
35 | {
|
36 | const JSX_TAG = /^[a-z][\w-]*$/i;
|
37 |
|
38 | const getTokenType = function (token, offset, text) {
|
39 | if (token.type === "name") {
|
40 | if ((0, _helperValidatorIdentifier.isKeyword)(token.value) || (0, _helperValidatorIdentifier.isStrictReservedWord)(token.value, true) || sometimesKeywords.has(token.value)) {
|
41 | return "keyword";
|
42 | }
|
43 |
|
44 | if (JSX_TAG.test(token.value) && (text[offset - 1] === "<" || text.substr(offset - 2, 2) == "</")) {
|
45 | return "jsxIdentifier";
|
46 | }
|
47 |
|
48 | if (token.value[0] !== token.value[0].toLowerCase()) {
|
49 | return "capitalized";
|
50 | }
|
51 | }
|
52 |
|
53 | if (token.type === "punctuator" && BRACKET.test(token.value)) {
|
54 | return "bracket";
|
55 | }
|
56 |
|
57 | if (token.type === "invalid" && (token.value === "@" || token.value === "#")) {
|
58 | return "punctuator";
|
59 | }
|
60 |
|
61 | return token.type;
|
62 | };
|
63 |
|
64 | tokenize = function* (text) {
|
65 | let match;
|
66 |
|
67 | while (match = _jsTokens.default.exec(text)) {
|
68 | const token = _jsTokens.matchToToken(match);
|
69 |
|
70 | yield {
|
71 | type: getTokenType(token, match.index, text),
|
72 | value: token.value
|
73 | };
|
74 | }
|
75 | };
|
76 | }
|
77 |
|
78 | function highlightTokens(defs, text) {
|
79 | let highlighted = "";
|
80 |
|
81 | for (const {
|
82 | type,
|
83 | value
|
84 | } of tokenize(text)) {
|
85 | const colorize = defs[type];
|
86 |
|
87 | if (colorize) {
|
88 | highlighted += value.split(NEWLINE).map(str => colorize(str)).join("\n");
|
89 | } else {
|
90 | highlighted += value;
|
91 | }
|
92 | }
|
93 |
|
94 | return highlighted;
|
95 | }
|
96 |
|
97 | function shouldHighlight(options) {
|
98 | return !!_chalk.supportsColor || options.forceColor;
|
99 | }
|
100 |
|
101 | function getChalk(options) {
|
102 | return options.forceColor ? new _chalk.constructor({
|
103 | enabled: true,
|
104 | level: 1
|
105 | }) : _chalk;
|
106 | }
|
107 |
|
108 | function highlight(code, options = {}) {
|
109 | if (shouldHighlight(options)) {
|
110 | const chalk = getChalk(options);
|
111 | const defs = getDefs(chalk);
|
112 | return highlightTokens(defs, code);
|
113 | } else {
|
114 | return code;
|
115 | }
|
116 | } |
\ | No newline at end of file |