mirror of
https://github.com/Sevichecc/Urara-Blog.git
synced 2025-05-02 03:39:30 +08:00
109 lines
3 KiB
Text
Executable file
109 lines
3 KiB
Text
Executable file
import MagicString from 'magic-string';
|
|
import { createFilter } from '@rollup/pluginutils';
|
|
|
|
function escape(str) {
|
|
return str.replace(/[-[\]/{}()*+?.\\^$|]/g, '\\$&');
|
|
}
|
|
|
|
function ensureFunction(functionOrValue) {
|
|
if (typeof functionOrValue === 'function') return functionOrValue;
|
|
return () => functionOrValue;
|
|
}
|
|
|
|
function longest(a, b) {
|
|
return b.length - a.length;
|
|
}
|
|
|
|
function getReplacements(options) {
|
|
if (options.values) {
|
|
return Object.assign({}, options.values);
|
|
}
|
|
const values = Object.assign({}, options);
|
|
delete values.delimiters;
|
|
delete values.include;
|
|
delete values.exclude;
|
|
delete values.sourcemap;
|
|
delete values.sourceMap;
|
|
return values;
|
|
}
|
|
|
|
function mapToFunctions(object) {
|
|
return Object.keys(object).reduce((fns, key) => {
|
|
const functions = Object.assign({}, fns);
|
|
functions[key] = ensureFunction(object[key]);
|
|
return functions;
|
|
}, {});
|
|
}
|
|
|
|
export default function replace(options = {}) {
|
|
const filter = createFilter(options.include, options.exclude);
|
|
const { delimiters, preventAssignment } = options;
|
|
const functionValues = mapToFunctions(getReplacements(options));
|
|
const keys = Object.keys(functionValues).sort(longest).map(escape);
|
|
const lookahead = preventAssignment ? '(?!\\s*=[^=])' : '';
|
|
const pattern = delimiters
|
|
? new RegExp(
|
|
`${escape(delimiters[0])}(${keys.join('|')})${escape(delimiters[1])}${lookahead}`,
|
|
'g'
|
|
)
|
|
: new RegExp(`\\b(${keys.join('|')})\\b${lookahead}`, 'g');
|
|
|
|
return {
|
|
name: 'replace',
|
|
|
|
buildStart() {
|
|
if (![true, false].includes(preventAssignment)) {
|
|
this.warn({
|
|
message:
|
|
"@rollup/plugin-replace: 'preventAssignment' currently defaults to false. It is recommended to set this option to `true`, as the next major version will default this option to `true`."
|
|
});
|
|
}
|
|
},
|
|
|
|
renderChunk(code, chunk) {
|
|
const id = chunk.fileName;
|
|
if (!keys.length) return null;
|
|
if (!filter(id)) return null;
|
|
return executeReplacement(code, id);
|
|
},
|
|
|
|
transform(code, id) {
|
|
if (!keys.length) return null;
|
|
if (!filter(id)) return null;
|
|
return executeReplacement(code, id);
|
|
}
|
|
};
|
|
|
|
function executeReplacement(code, id) {
|
|
const magicString = new MagicString(code);
|
|
if (!codeHasReplacements(code, id, magicString)) {
|
|
return null;
|
|
}
|
|
|
|
const result = { code: magicString.toString() };
|
|
if (isSourceMapEnabled()) {
|
|
result.map = magicString.generateMap({ hires: true });
|
|
}
|
|
return result;
|
|
}
|
|
|
|
function codeHasReplacements(code, id, magicString) {
|
|
let result = false;
|
|
let match;
|
|
|
|
// eslint-disable-next-line no-cond-assign
|
|
while ((match = pattern.exec(code))) {
|
|
result = true;
|
|
|
|
const start = match.index;
|
|
const end = start + match[0].length;
|
|
const replacement = String(functionValues[match[1]](id));
|
|
magicString.overwrite(start, end, replacement);
|
|
}
|
|
return result;
|
|
}
|
|
|
|
function isSourceMapEnabled() {
|
|
return options.sourceMap !== false && options.sourcemap !== false;
|
|
}
|
|
}
|