67 lines
2.3 KiB
JavaScript
67 lines
2.3 KiB
JavaScript
|
import jsTokens from "js-tokens";
|
||
|
|
//#region src/js-tokens.ts
|
||
|
|
const FILL_COMMENT = " ";
|
||
|
|
function stripLiteralFromToken(token, fillChar, filter) {
|
||
|
|
if (token.type === "SingleLineComment") return FILL_COMMENT.repeat(token.value.length);
|
||
|
|
if (token.type === "MultiLineComment") return token.value.replace(/[^\n]/g, FILL_COMMENT);
|
||
|
|
if (token.type === "StringLiteral") {
|
||
|
|
if (!token.closed) return token.value;
|
||
|
|
const body = token.value.slice(1, -1);
|
||
|
|
if (filter(body)) return token.value[0] + fillChar.repeat(body.length) + token.value[token.value.length - 1];
|
||
|
|
}
|
||
|
|
if (token.type === "NoSubstitutionTemplate") {
|
||
|
|
const body = token.value.slice(1, -1);
|
||
|
|
if (filter(body)) return `\`${body.replace(/[^\n]/g, fillChar)}\``;
|
||
|
|
}
|
||
|
|
if (token.type === "RegularExpressionLiteral") {
|
||
|
|
const body = token.value;
|
||
|
|
if (filter(body)) return body.replace(/\/(.*)\/(\w?)$/g, (_, $1, $2) => `/${fillChar.repeat($1.length)}/${$2}`);
|
||
|
|
}
|
||
|
|
if (token.type === "TemplateHead") {
|
||
|
|
const body = token.value.slice(1, -2);
|
||
|
|
if (filter(body)) return `\`${body.replace(/[^\n]/g, fillChar)}\${`;
|
||
|
|
}
|
||
|
|
if (token.type === "TemplateTail") {
|
||
|
|
const body = token.value.slice(0, -2);
|
||
|
|
if (filter(body)) return `}${body.replace(/[^\n]/g, fillChar)}\``;
|
||
|
|
}
|
||
|
|
if (token.type === "TemplateMiddle") {
|
||
|
|
const body = token.value.slice(1, -2);
|
||
|
|
if (filter(body)) return `}${body.replace(/[^\n]/g, fillChar)}\${`;
|
||
|
|
}
|
||
|
|
return token.value;
|
||
|
|
}
|
||
|
|
function optionsWithDefaults(options) {
|
||
|
|
return {
|
||
|
|
fillChar: options?.fillChar ?? " ",
|
||
|
|
filter: options?.filter ?? (() => true)
|
||
|
|
};
|
||
|
|
}
|
||
|
|
/**
|
||
|
|
* Strip literal from code.
|
||
|
|
*/
|
||
|
|
function stripLiteral(code, options) {
|
||
|
|
let result = "";
|
||
|
|
const _options = optionsWithDefaults(options);
|
||
|
|
for (const token of jsTokens(code, { jsx: false })) result += stripLiteralFromToken(token, _options.fillChar, _options.filter);
|
||
|
|
return result;
|
||
|
|
}
|
||
|
|
/**
|
||
|
|
* Strip literal from code, return more detailed information.
|
||
|
|
*/
|
||
|
|
function stripLiteralDetailed(code, options) {
|
||
|
|
let result = "";
|
||
|
|
const tokens = [];
|
||
|
|
const _options = optionsWithDefaults(options);
|
||
|
|
for (const token of jsTokens(code, { jsx: false })) {
|
||
|
|
tokens.push(token);
|
||
|
|
result += stripLiteralFromToken(token, _options.fillChar, _options.filter);
|
||
|
|
}
|
||
|
|
return {
|
||
|
|
result,
|
||
|
|
tokens
|
||
|
|
};
|
||
|
|
}
|
||
|
|
//#endregion
|
||
|
|
export { stripLiteral, stripLiteralDetailed, stripLiteralDetailed as stripLiteralJsTokens };
|