src/codegen/promptParts/simpleAndSafeMinify.js - main - githubmirror/english-compiler

src/codegen/promptParts/simpleAndSafeMinify.js
88 строк3 KB
Eugene Cheah
prompt parts library collection
17 фев 2023, 11:32
17 фев 2023, 11:32a26dae3
/**
 * Given two string parameters 
 * - inData : containing the code, which needs to be minified 
 * - inType : the programming language category either "cstyle" or "xml"
 * 
 * Attempt to safely minify the given data code, while stripping comments in a safe way.
 * The goal isn't to get the perfect minified code (which we will use another tool), but to safely remove whitespace and comments when possible. And reduce the overall character count.
 * 
 * @param {string} inData - The code to minify
 * @param {string} inType - The programming language category either "cstyle" or "xml"
 * 
 * @returns {string} The minified code
 */
 const simpleAndSafeMinify = (inData, inType) => {
	let outData = '';
	if (inType === 'xml') {
	  outData = simpleAndSafeMinifyXML(inData);
	} else if (inType === 'cstyle') {
	  outData = simpleAndSafeMinifyCStyle(inData);
	}
	return outData.trim();
};

/**
 * Handle minification of XML code
 * 
 * @param {string} inData - The code to minify
 * 
 * @returns {string} The minified code
 */
const simpleAndSafeMinifyXML = (inData) => {
	// Find and remove <!-- comment --> blocks
	const commentRegex = /<!--[\s\S]*?-->/g;
	let outData = inData.replace(commentRegex, '');
  
	// Find and remove whitespaces between blocks (check for ">" and "<")
	const whitespaceRegex = />\s+</g;
	outData = outData.replace(whitespaceRegex, '><');
  
	// Trim the final string
	return outData.split("\n").map((val)=>{return val.trim()}).join("\n").trim();
};

/**
 * Handle minification of cstyle code
 * 
 * @param {string} inData - The code to minify
 * 
 * @returns {string} The minified code
 */
const simpleAndSafeMinifyCStyle = (inData) => {
	// Find 1 string block which can be wrapped with either ` or ' or " quotes.
	const strRegex = /('.*'|".*"|`.*`)/;
	const strMatch = inData.match(strRegex);
  
	// If string block is found, avoi modifying the string block
	if (strMatch) {
		// Leave the string block untouched
		const strStart = strMatch.index;
		const strEnd = strStart + strMatch[0].length;
		const beforeStr = inData.slice(0, strStart);
		const afterStr = inData.slice(strEnd);
	  
		// Recursively minify the code before and after the string block
		return simpleAndSafeMinifyCStyle(beforeStr) + strMatch[0] + simpleAndSafeMinifyCStyle(afterStr);
	}
  
	// using regex, remove any trailing // comments for each line
	let outData = inData.replace( /\/\/.*\n/g, '');

	// Remove / * comment blocks * /
	outData = outData.replace( /\/\*.*?\*\//g , '');

	// Replace multiple whitespace (excluding new lines) with a single space
	outData = outData.replace( /[ \t]+/g , ' ');

	// Safely handle end of statements
	outData = outData.replace( /\;\s*/g , ';');

	// Safely handle assignments
	outData = outData.replace( / =/g , '=');
	outData = outData.replace( /= /g , '=');

	// Trim the final string
	return outData.split("\n").map((val)=>{return val.trim()}).join("\n").trim();
};

module.exports = simpleAndSafeMinify;