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.
		
		
		
		
		
			
		
			
				
					116 lines
				
				2.7 KiB
			
		
		
			
		
	
	
					116 lines
				
				2.7 KiB
			| 
								 
											3 years ago
										 
									 | 
							
								"use strict";
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								Object.defineProperty(exports, "__esModule", {
							 | 
						||
| 
								 | 
							
								  value: true
							 | 
						||
| 
								 | 
							
								});
							 | 
						||
| 
								 | 
							
								exports.default = highlight;
							 | 
						||
| 
								 | 
							
								exports.getChalk = getChalk;
							 | 
						||
| 
								 | 
							
								exports.shouldHighlight = shouldHighlight;
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								var _jsTokens = require("js-tokens");
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								var _helperValidatorIdentifier = require("@babel/helper-validator-identifier");
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								var _chalk = require("chalk");
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								const sometimesKeywords = new Set(["as", "async", "from", "get", "of", "set"]);
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								function getDefs(chalk) {
							 | 
						||
| 
								 | 
							
								  return {
							 | 
						||
| 
								 | 
							
								    keyword: chalk.cyan,
							 | 
						||
| 
								 | 
							
								    capitalized: chalk.yellow,
							 | 
						||
| 
								 | 
							
								    jsxIdentifier: chalk.yellow,
							 | 
						||
| 
								 | 
							
								    punctuator: chalk.yellow,
							 | 
						||
| 
								 | 
							
								    number: chalk.magenta,
							 | 
						||
| 
								 | 
							
								    string: chalk.green,
							 | 
						||
| 
								 | 
							
								    regex: chalk.magenta,
							 | 
						||
| 
								 | 
							
								    comment: chalk.grey,
							 | 
						||
| 
								 | 
							
								    invalid: chalk.white.bgRed.bold
							 | 
						||
| 
								 | 
							
								  };
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								const NEWLINE = /\r\n|[\n\r\u2028\u2029]/;
							 | 
						||
| 
								 | 
							
								const BRACKET = /^[()[\]{}]$/;
							 | 
						||
| 
								 | 
							
								let tokenize;
							 | 
						||
| 
								 | 
							
								{
							 | 
						||
| 
								 | 
							
								  const JSX_TAG = /^[a-z][\w-]*$/i;
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  const getTokenType = function (token, offset, text) {
							 | 
						||
| 
								 | 
							
								    if (token.type === "name") {
							 | 
						||
| 
								 | 
							
								      if ((0, _helperValidatorIdentifier.isKeyword)(token.value) || (0, _helperValidatorIdentifier.isStrictReservedWord)(token.value, true) || sometimesKeywords.has(token.value)) {
							 | 
						||
| 
								 | 
							
								        return "keyword";
							 | 
						||
| 
								 | 
							
								      }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								      if (JSX_TAG.test(token.value) && (text[offset - 1] === "<" || text.slice(offset - 2, offset) == "</")) {
							 | 
						||
| 
								 | 
							
								        return "jsxIdentifier";
							 | 
						||
| 
								 | 
							
								      }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								      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;
							 | 
						||
| 
								 | 
							
								  };
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  tokenize = function* (text) {
							 | 
						||
| 
								 | 
							
								    let match;
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								    while (match = _jsTokens.default.exec(text)) {
							 | 
						||
| 
								 | 
							
								      const token = _jsTokens.matchToToken(match);
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								      yield {
							 | 
						||
| 
								 | 
							
								        type: getTokenType(token, match.index, text),
							 | 
						||
| 
								 | 
							
								        value: token.value
							 | 
						||
| 
								 | 
							
								      };
							 | 
						||
| 
								 | 
							
								    }
							 | 
						||
| 
								 | 
							
								  };
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								function highlightTokens(defs, text) {
							 | 
						||
| 
								 | 
							
								  let highlighted = "";
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  for (const {
							 | 
						||
| 
								 | 
							
								    type,
							 | 
						||
| 
								 | 
							
								    value
							 | 
						||
| 
								 | 
							
								  } of tokenize(text)) {
							 | 
						||
| 
								 | 
							
								    const colorize = defs[type];
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								    if (colorize) {
							 | 
						||
| 
								 | 
							
								      highlighted += value.split(NEWLINE).map(str => colorize(str)).join("\n");
							 | 
						||
| 
								 | 
							
								    } else {
							 | 
						||
| 
								 | 
							
								      highlighted += value;
							 | 
						||
| 
								 | 
							
								    }
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  return highlighted;
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								function shouldHighlight(options) {
							 | 
						||
| 
								 | 
							
								  return !!_chalk.supportsColor || options.forceColor;
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								function getChalk(options) {
							 | 
						||
| 
								 | 
							
								  return options.forceColor ? new _chalk.constructor({
							 | 
						||
| 
								 | 
							
								    enabled: true,
							 | 
						||
| 
								 | 
							
								    level: 1
							 | 
						||
| 
								 | 
							
								  }) : _chalk;
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								function highlight(code, options = {}) {
							 | 
						||
| 
								 | 
							
								  if (code !== "" && shouldHighlight(options)) {
							 | 
						||
| 
								 | 
							
								    const chalk = getChalk(options);
							 | 
						||
| 
								 | 
							
								    const defs = getDefs(chalk);
							 | 
						||
| 
								 | 
							
								    return highlightTokens(defs, code);
							 | 
						||
| 
								 | 
							
								  } else {
							 | 
						||
| 
								 | 
							
								    return code;
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								}
							 |