123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305 |
- "use strict";
- Object.defineProperty(exports, "__esModule", {
- value: true
- });
- exports.JavaScriptErrorInEvaluate = exports.JSHandle = exports.ExecutionContext = void 0;
- exports.evaluate = evaluate;
- exports.evaluateExpression = evaluateExpression;
- exports.isJavaScriptErrorInEvaluate = isJavaScriptErrorInEvaluate;
- exports.normalizeEvaluationExpression = normalizeEvaluationExpression;
- exports.parseUnserializableValue = parseUnserializableValue;
- exports.sparseArrayToString = sparseArrayToString;
- var utilityScriptSource = _interopRequireWildcard(require("../generated/utilityScriptSource"));
- var _utilityScriptSerializers = require("./isomorphic/utilityScriptSerializers");
- var _instrumentation = require("./instrumentation");
- var _manualPromise = require("../utils/manualPromise");
- function _getRequireWildcardCache(nodeInterop) { if (typeof WeakMap !== "function") return null; var cacheBabelInterop = new WeakMap(); var cacheNodeInterop = new WeakMap(); return (_getRequireWildcardCache = function (nodeInterop) { return nodeInterop ? cacheNodeInterop : cacheBabelInterop; })(nodeInterop); }
- function _interopRequireWildcard(obj, nodeInterop) { if (!nodeInterop && obj && obj.__esModule) { return obj; } if (obj === null || typeof obj !== "object" && typeof obj !== "function") { return { default: obj }; } var cache = _getRequireWildcardCache(nodeInterop); if (cache && cache.has(obj)) { return cache.get(obj); } var newObj = {}; var hasPropertyDescriptor = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var key in obj) { if (key !== "default" && Object.prototype.hasOwnProperty.call(obj, key)) { var desc = hasPropertyDescriptor ? Object.getOwnPropertyDescriptor(obj, key) : null; if (desc && (desc.get || desc.set)) { Object.defineProperty(newObj, key, desc); } else { newObj[key] = obj[key]; } } } newObj.default = obj; if (cache) { cache.set(obj, newObj); } return newObj; }
- /**
- * Copyright (c) Microsoft Corporation.
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
- class ExecutionContext extends _instrumentation.SdkObject {
- constructor(parent, delegate, worldNameForTest) {
- super(parent, 'execution-context');
- this._delegate = void 0;
- this._utilityScriptPromise = void 0;
- this._contextDestroyedScope = new _manualPromise.LongStandingScope();
- this.worldNameForTest = void 0;
- this.worldNameForTest = worldNameForTest;
- this._delegate = delegate;
- }
- contextDestroyed(reason) {
- this._contextDestroyedScope.close(new Error(reason));
- }
- async _raceAgainstContextDestroyed(promise) {
- return this._contextDestroyedScope.race(promise);
- }
- rawEvaluateJSON(expression) {
- return this._raceAgainstContextDestroyed(this._delegate.rawEvaluateJSON(expression));
- }
- rawEvaluateHandle(expression) {
- return this._raceAgainstContextDestroyed(this._delegate.rawEvaluateHandle(expression));
- }
- rawCallFunctionNoReply(func, ...args) {
- this._delegate.rawCallFunctionNoReply(func, ...args);
- }
- evaluateWithArguments(expression, returnByValue, utilityScript, values, objectIds) {
- return this._raceAgainstContextDestroyed(this._delegate.evaluateWithArguments(expression, returnByValue, utilityScript, values, objectIds));
- }
- getProperties(context, objectId) {
- return this._raceAgainstContextDestroyed(this._delegate.getProperties(context, objectId));
- }
- createHandle(remoteObject) {
- return this._delegate.createHandle(this, remoteObject);
- }
- releaseHandle(objectId) {
- return this._delegate.releaseHandle(objectId);
- }
- adoptIfNeeded(handle) {
- return null;
- }
- utilityScript() {
- if (!this._utilityScriptPromise) {
- const source = `
- (() => {
- const module = {};
- ${utilityScriptSource.source}
- return new (module.exports.UtilityScript())();
- })();`;
- this._utilityScriptPromise = this._raceAgainstContextDestroyed(this._delegate.rawEvaluateHandle(source).then(objectId => new JSHandle(this, 'object', 'UtilityScript', objectId)));
- }
- return this._utilityScriptPromise;
- }
- async objectCount(objectId) {
- return this._delegate.objectCount(objectId);
- }
- async doSlowMo() {
- // overridden in FrameExecutionContext
- }
- }
- exports.ExecutionContext = ExecutionContext;
- class JSHandle extends _instrumentation.SdkObject {
- constructor(context, type, preview, objectId, value) {
- super(context, 'handle');
- this.__jshandle = true;
- this._context = void 0;
- this._disposed = false;
- this._objectId = void 0;
- this._value = void 0;
- this._objectType = void 0;
- this._preview = void 0;
- this._previewCallback = void 0;
- this._context = context;
- this._objectId = objectId;
- this._value = value;
- this._objectType = type;
- this._preview = this._objectId ? preview || `JSHandle@${this._objectType}` : String(value);
- if (this._objectId && globalThis.leakedJSHandles) globalThis.leakedJSHandles.set(this, new Error('Leaked JSHandle'));
- }
- callFunctionNoReply(func, arg) {
- this._context.rawCallFunctionNoReply(func, this, arg);
- }
- async evaluate(pageFunction, arg) {
- return evaluate(this._context, true /* returnByValue */, pageFunction, this, arg);
- }
- async evaluateHandle(pageFunction, arg) {
- return evaluate(this._context, false /* returnByValue */, pageFunction, this, arg);
- }
- async evaluateExpression(expression, options, arg) {
- const value = await evaluateExpression(this._context, expression, {
- ...options,
- returnByValue: true
- }, this, arg);
- await this._context.doSlowMo();
- return value;
- }
- async evaluateExpressionHandle(expression, options, arg) {
- const value = await evaluateExpression(this._context, expression, {
- ...options,
- returnByValue: false
- }, this, arg);
- await this._context.doSlowMo();
- return value;
- }
- async getProperty(propertyName) {
- const objectHandle = await this.evaluateHandle((object, propertyName) => {
- const result = {
- __proto__: null
- };
- result[propertyName] = object[propertyName];
- return result;
- }, propertyName);
- const properties = await objectHandle.getProperties();
- const result = properties.get(propertyName);
- objectHandle.dispose();
- return result;
- }
- async getProperties() {
- if (!this._objectId) return new Map();
- return this._context.getProperties(this._context, this._objectId);
- }
- rawValue() {
- return this._value;
- }
- async jsonValue() {
- if (!this._objectId) return this._value;
- const utilityScript = await this._context.utilityScript();
- const script = `(utilityScript, ...args) => utilityScript.jsonValue(...args)`;
- return this._context.evaluateWithArguments(script, true, utilityScript, [true], [this._objectId]);
- }
- asElement() {
- return null;
- }
- dispose() {
- if (this._disposed) return;
- this._disposed = true;
- if (this._objectId) {
- this._context.releaseHandle(this._objectId).catch(e => {});
- if (globalThis.leakedJSHandles) globalThis.leakedJSHandles.delete(this);
- }
- }
- toString() {
- return this._preview;
- }
- _setPreviewCallback(callback) {
- this._previewCallback = callback;
- }
- preview() {
- return this._preview;
- }
- worldNameForTest() {
- return this._context.worldNameForTest;
- }
- _setPreview(preview) {
- this._preview = preview;
- if (this._previewCallback) this._previewCallback(preview);
- }
- async objectCount() {
- if (!this._objectId) throw new Error('Can only count objects for a handle that points to the constructor prototype');
- return this._context.objectCount(this._objectId);
- }
- }
- exports.JSHandle = JSHandle;
- async function evaluate(context, returnByValue, pageFunction, ...args) {
- return evaluateExpression(context, String(pageFunction), {
- returnByValue,
- isFunction: typeof pageFunction === 'function'
- }, ...args);
- }
- async function evaluateExpression(context, expression, options, ...args) {
- const utilityScript = await context.utilityScript();
- expression = normalizeEvaluationExpression(expression, options.isFunction);
- const handles = [];
- const toDispose = [];
- const pushHandle = handle => {
- handles.push(handle);
- return handles.length - 1;
- };
- args = args.map(arg => (0, _utilityScriptSerializers.serializeAsCallArgument)(arg, handle => {
- if (handle instanceof JSHandle) {
- if (!handle._objectId) return {
- fallThrough: handle._value
- };
- if (handle._disposed) throw new JavaScriptErrorInEvaluate('JSHandle is disposed!');
- const adopted = context.adoptIfNeeded(handle);
- if (adopted === null) return {
- h: pushHandle(Promise.resolve(handle))
- };
- toDispose.push(adopted);
- return {
- h: pushHandle(adopted)
- };
- }
- return {
- fallThrough: handle
- };
- }));
- const utilityScriptObjectIds = [];
- for (const handle of await Promise.all(handles)) {
- if (handle._context !== context) throw new JavaScriptErrorInEvaluate('JSHandles can be evaluated only in the context they were created!');
- utilityScriptObjectIds.push(handle._objectId);
- }
- // See UtilityScript for arguments.
- const utilityScriptValues = [options.isFunction, options.returnByValue, options.exposeUtilityScript, expression, args.length, ...args];
- const script = `(utilityScript, ...args) => utilityScript.evaluate(...args)`;
- try {
- return await context.evaluateWithArguments(script, options.returnByValue || false, utilityScript, utilityScriptValues, utilityScriptObjectIds);
- } finally {
- toDispose.map(handlePromise => handlePromise.then(handle => handle.dispose()));
- }
- }
- function parseUnserializableValue(unserializableValue) {
- if (unserializableValue === 'NaN') return NaN;
- if (unserializableValue === 'Infinity') return Infinity;
- if (unserializableValue === '-Infinity') return -Infinity;
- if (unserializableValue === '-0') return -0;
- }
- function normalizeEvaluationExpression(expression, isFunction) {
- expression = expression.trim();
- if (isFunction) {
- try {
- new Function('(' + expression + ')');
- } catch (e1) {
- // This means we might have a function shorthand. Try another
- // time prefixing 'function '.
- if (expression.startsWith('async ')) expression = 'async function ' + expression.substring('async '.length);else expression = 'function ' + expression;
- try {
- new Function('(' + expression + ')');
- } catch (e2) {
- // We tried hard to serialize, but there's a weird beast here.
- throw new Error('Passed function is not well-serializable!');
- }
- }
- }
- if (/^(async)?\s*function(\s|\()/.test(expression)) expression = '(' + expression + ')';
- return expression;
- }
- // Error inside the expression evaluation as opposed to a protocol error.
- class JavaScriptErrorInEvaluate extends Error {}
- exports.JavaScriptErrorInEvaluate = JavaScriptErrorInEvaluate;
- function isJavaScriptErrorInEvaluate(error) {
- return error instanceof JavaScriptErrorInEvaluate;
- }
- function sparseArrayToString(entries) {
- const arrayEntries = [];
- for (const {
- name,
- value
- } of entries) {
- const index = +name;
- if (isNaN(index) || index < 0) continue;
- arrayEntries.push({
- index,
- value
- });
- }
- arrayEntries.sort((a, b) => a.index - b.index);
- let lastIndex = -1;
- const tokens = [];
- for (const {
- index,
- value
- } of arrayEntries) {
- const emptyItems = index - lastIndex - 1;
- if (emptyItems === 1) tokens.push(`empty`);else if (emptyItems > 1) tokens.push(`empty x ${emptyItems}`);
- tokens.push(String(value));
- lastIndex = index;
- }
- return '[' + tokens.join(', ') + ']';
- }
|