|
|
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; |
|
|
|
|
|
|
|
|
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; |
|
|
} |
|
|
} |
|
|
|