puppeteer-core
Version:
A high-level API to control headless Chrome over the DevTools Protocol
76 lines • 2.27 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", { value: true });
exports.interpolateFunction = exports.createFunction = void 0;
exports.stringifyFunction = stringifyFunction;
/**
* @license
* Copyright 2023 Google Inc.
* SPDX-License-Identifier: Apache-2.0
*/
const createdFunctions = new Map();
/**
* Creates a function from a string.
*
* @internal
*/
const createFunction = (functionValue) => {
let fn = createdFunctions.get(functionValue);
if (fn) {
return fn;
}
fn = new Function(`return ${functionValue}`)();
createdFunctions.set(functionValue, fn);
return fn;
};
exports.createFunction = createFunction;
/**
* @internal
*/
function stringifyFunction(fn) {
let value = fn.toString();
if (value.match(/^(async )*function(\(|\s)/) ||
value.match(/^(async )*function\s*\*\s*/)) {
return value;
}
const isArrow = value.startsWith('(') ||
value.match(/^async\s*\(/) ||
value.match(/^(async)*\s*(?:[$_\p{ID_Start}])(?:[$\u200C\u200D\p{ID_Continue}])*\s*=>/u);
if (isArrow) {
return value;
}
// This means we might have a function shorthand (e.g. `test(){}`). Let's
// try prefixing.
let prefix = 'function ';
if (value.startsWith('async ')) {
prefix = `async ${prefix}`;
value = value.substring('async '.length);
}
return `${prefix}${value}`;
}
/**
* Replaces `PLACEHOLDER`s with the given replacements.
*
* All replacements must be valid JS code.
*
* @example
*
* ```ts
* interpolateFunction(() => PLACEHOLDER('test'), {test: 'void 0'});
* // Equivalent to () => void 0
* ```
*
* @internal
*/
const interpolateFunction = (fn, replacements) => {
let value = stringifyFunction(fn);
for (const [name, jsValue] of Object.entries(replacements)) {
value = value.replace(new RegExp(`PLACEHOLDER\\(\\s*(?:'${name}'|"${name}")\\s*\\)`, 'g'),
// Wrapping this ensures tersers that accidentally inline PLACEHOLDER calls
// are still valid. Without, we may get calls like ()=>{...}() which is
// not valid.
`(${jsValue})`);
}
return (0, exports.createFunction)(value);
};
exports.interpolateFunction = interpolateFunction;
//# sourceMappingURL=Function.js.map