kmx.io/prism.js/tests/helper/util.js

Branch - branch -master

• Show log

  Commit

• Hash : 5333e281
  Author :
  Date : 2021-10-19T19:34:41
  

  Added regex coverage (#3138)

Download

• tests/helper/util.js

• const { RegExpParser } = require('regexpp');
  
  
  /**
   * @typedef {import("regexpp/ast").Pattern} Pattern
   * @typedef {import("regexpp/ast").Flags} Flags
   * @typedef {{ pattern: Pattern, flags: Flags }} LiteralAST
   */
  
  
  const parser = new RegExpParser();
  /** @type {Map<string, LiteralAST>} */
  const astCache = new Map();
  
  
  module.exports = {
  
  	/**
  	 * Performs a breadth-first search on the given start element.
  	 *
  	 * @param {any} start
  	 * @param {(path: { key: string, value: any }[], obj: Record<string, any>) => void} callback
  	 */
  	BFS(start, callback) {
  		const visited = new Set();
  		/** @type {{ key: string, value: any }[][]} */
  		let toVisit = [
  			[{ key: null, value: start }]
  		];
  
  		while (toVisit.length > 0) {
  			/** @type {{ key: string, value: any }[][]} */
  			const newToVisit = [];
  
  			for (const path of toVisit) {
  				const obj = path[path.length - 1].value;
  				if (!visited.has(obj)) {
  					visited.add(obj);
  
  					for (const key in obj) {
  						const value = obj[key];
  
  						path.push({ key, value });
  						callback(path, obj);
  
  						if (Array.isArray(value) || Object.prototype.toString.call(value) == '[object Object]') {
  							newToVisit.push([...path]);
  						}
  
  						path.pop();
  					}
  				}
  			}
  
  			toVisit = newToVisit;
  		}
  	},
  
  	/**
  	 * Given the `BFS` path given to `BFS` callbacks, this will return the Prism language token path of the current
  	 * value (e.g. `Prism.languages.xml.tag.pattern`).
  	 *
  	 * @param {readonly{ key: string, value: any }[]} path
  	 * @param {string} [root]
  	 * @returns {string}
  	 */
  	BFSPathToPrismTokenPath(path, root = 'Prism.languages') {
  		let tokenPath = root;
  		for (const { key } of path) {
  			if (!key) {
  				// do nothing
  			} else if (/^\d+$/.test(key)) {
  				tokenPath += `[${key}]`;
  			} else if (/^[a-z]\w*$/i.test(key)) {
  				tokenPath += `.${key}`;
  			} else {
  				tokenPath += `[${JSON.stringify(key)}]`;
  			}
  		}
  		return tokenPath;
  	},
  
  	/**
  	 * Returns the AST of a given pattern.
  	 *
  	 * @param {RegExp} regex
  	 * @returns {LiteralAST}
  	 */
  	parseRegex(regex) {
  		const key = regex.toString();
  		let literal = astCache.get(key);
  		if (literal === undefined) {
  			const flags = parser.parseFlags(regex.flags, undefined);
  			const pattern = parser.parsePattern(regex.source, undefined, undefined, flags.unicode);
  			literal = { pattern, flags };
  			astCache.set(key, literal);
  		}
  		return literal;
  	}
  
  };
  

Download