UNPKG

puppeteer-core

Version:

A high-level API to control headless Chrome over the DevTools Protocol

1,874 lines (1,795 loc) 68.4 kB
/** * Copyright 2017 Google Inc. All rights reserved. * * 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. */ const fs = require('fs'); const EventEmitter = require('events'); const mime = require('mime'); const {Events} = require('./Events'); const {Connection} = require('./Connection'); const {NetworkManager} = require('./NetworkManager'); const {Dialog} = require('./Dialog'); const {EmulationManager} = require('./EmulationManager'); const {FrameManager} = require('./FrameManager'); const {Keyboard, Mouse, Touchscreen} = require('./Input'); const Tracing = require('./Tracing'); const {helper, debugError, assert} = require('./helper'); const {Coverage} = require('./Coverage'); const {Worker} = require('./Worker'); const {createJSHandle} = require('./JSHandle'); const {Accessibility} = require('./Accessibility'); const {TimeoutSettings} = require('./TimeoutSettings'); const writeFileAsync = helper.promisify(fs.writeFile); class Page extends EventEmitter { /** * @param {!Puppeteer.CDPSession} client * @param {!Puppeteer.Target} target * @param {boolean} ignoreHTTPSErrors * @param {?Puppeteer.Viewport} defaultViewport * @param {!Puppeteer.TaskQueue} screenshotTaskQueue * @return {!Promise<!Page>} */ static /* async */ create(client, target, ignoreHTTPSErrors, defaultViewport, screenshotTaskQueue) {return (fn => { const gen = fn.call(this); return new Promise((resolve, reject) => { function step(key, arg) { let info, value; try { info = gen[key](arg); value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { return Promise.resolve(value).then( value => { step('next', value); }, err => { step('throw', err); }); } } return step('next'); }); })(function*(){ (yield client.send('Page.enable')); const {frameTree} = (yield client.send('Page.getFrameTree')); const page = new Page(client, target, frameTree, ignoreHTTPSErrors, screenshotTaskQueue); (yield Promise.all([ client.send('Target.setAutoAttach', {autoAttach: true, waitForDebuggerOnStart: false, flatten: true}), client.send('Page.setLifecycleEventsEnabled', { enabled: true }), client.send('Network.enable', {}), client.send('Runtime.enable', {}).then(() => page._frameManager.ensureSecondaryDOMWorld()), client.send('Security.enable', {}), client.send('Performance.enable', {}), client.send('Log.enable', {}), ])); if (ignoreHTTPSErrors) (yield client.send('Security.setOverrideCertificateErrors', {override: true})); // Initialize default page size. if (defaultViewport) (yield page.setViewport(defaultViewport)); return page; });} /** * @param {!Puppeteer.CDPSession} client * @param {!Puppeteer.Target} target * @param {!Protocol.Page.FrameTree} frameTree * @param {boolean} ignoreHTTPSErrors * @param {!Puppeteer.TaskQueue} screenshotTaskQueue */ constructor(client, target, frameTree, ignoreHTTPSErrors, screenshotTaskQueue) { super(); this._closed = false; this._client = client; this._target = target; this._keyboard = new Keyboard(client); this._mouse = new Mouse(client, this._keyboard); this._timeoutSettings = new TimeoutSettings(); this._touchscreen = new Touchscreen(client, this._keyboard); this._accessibility = new Accessibility(client); this._networkManager = new NetworkManager(client); /** @type {!FrameManager} */ this._frameManager = new FrameManager(client, frameTree, this, this._networkManager, this._timeoutSettings); this._networkManager.setFrameManager(this._frameManager); this._emulationManager = new EmulationManager(client); this._tracing = new Tracing(client); /** @type {!Map<string, Function>} */ this._pageBindings = new Map(); this._ignoreHTTPSErrors = ignoreHTTPSErrors; this._coverage = new Coverage(client); this._javascriptEnabled = true; /** @type {?Puppeteer.Viewport} */ this._viewport = null; this._screenshotTaskQueue = screenshotTaskQueue; /** @type {!Map<string, Worker>} */ this._workers = new Map(); client.on('Target.attachedToTarget', event => { if (event.targetInfo.type !== 'worker') { // If we don't detach from service workers, they will never die. client.send('Target.detachFromTarget', { sessionId: event.sessionId }).catch(debugError); return; } const session = Connection.fromSession(client).session(event.sessionId); const worker = new Worker(session, event.targetInfo.url, this._addConsoleMessage.bind(this), this._handleException.bind(this)); this._workers.set(event.sessionId, worker); this.emit(Events.Page.WorkerCreated, worker); }); client.on('Target.detachedFromTarget', event => { const worker = this._workers.get(event.sessionId); if (!worker) return; this.emit(Events.Page.WorkerDestroyed, worker); this._workers.delete(event.sessionId); }); this._frameManager.on(Events.FrameManager.FrameAttached, event => this.emit(Events.Page.FrameAttached, event)); this._frameManager.on(Events.FrameManager.FrameDetached, event => this.emit(Events.Page.FrameDetached, event)); this._frameManager.on(Events.FrameManager.FrameNavigated, event => this.emit(Events.Page.FrameNavigated, event)); this._networkManager.on(Events.NetworkManager.Request, event => this.emit(Events.Page.Request, event)); this._networkManager.on(Events.NetworkManager.Response, event => this.emit(Events.Page.Response, event)); this._networkManager.on(Events.NetworkManager.RequestFailed, event => this.emit(Events.Page.RequestFailed, event)); this._networkManager.on(Events.NetworkManager.RequestFinished, event => this.emit(Events.Page.RequestFinished, event)); client.on('Page.domContentEventFired', event => this.emit(Events.Page.DOMContentLoaded)); client.on('Page.loadEventFired', event => this.emit(Events.Page.Load)); client.on('Runtime.consoleAPICalled', event => this._onConsoleAPI(event)); client.on('Runtime.bindingCalled', event => this._onBindingCalled(event)); client.on('Page.javascriptDialogOpening', event => this._onDialog(event)); client.on('Runtime.exceptionThrown', exception => this._handleException(exception.exceptionDetails)); client.on('Security.certificateError', event => this._onCertificateError(event)); client.on('Inspector.targetCrashed', event => this._onTargetCrashed()); client.on('Performance.metrics', event => this._emitMetrics(event)); client.on('Log.entryAdded', event => this._onLogEntryAdded(event)); this._target._isClosedPromise.then(() => { this.emit(Events.Page.Close); this._closed = true; }); } /** * @param {!{longitude: number, latitude: number, accuracy: (number|undefined)}} options */ /* async */ setGeolocation(options) {return (fn => { const gen = fn.call(this); return new Promise((resolve, reject) => { function step(key, arg) { let info, value; try { info = gen[key](arg); value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { return Promise.resolve(value).then( value => { step('next', value); }, err => { step('throw', err); }); } } return step('next'); }); })(function*(){ const { longitude, latitude, accuracy = 0} = options; if (longitude < -180 || longitude > 180) throw new Error(`Invalid longitude "${longitude}": precondition -180 <= LONGITUDE <= 180 failed.`); if (latitude < -90 || latitude > 90) throw new Error(`Invalid latitude "${latitude}": precondition -90 <= LATITUDE <= 90 failed.`); if (accuracy < 0) throw new Error(`Invalid accuracy "${accuracy}": precondition 0 <= ACCURACY failed.`); (yield this._client.send('Emulation.setGeolocationOverride', {longitude, latitude, accuracy})); });} /** * @return {!Puppeteer.Target} */ target() { return this._target; } /** * @return {!Puppeteer.Browser} */ browser() { return this._target.browser(); } /** * @return {!Puppeteer.BrowserContext} */ browserContext() { return this._target.browserContext(); } _onTargetCrashed() { this.emit('error', new Error('Page crashed!')); } /** * @param {!Protocol.Log.entryAddedPayload} event */ _onLogEntryAdded(event) { const {level, text, args, source, url, lineNumber} = event.entry; if (args) args.map(arg => helper.releaseObject(this._client, arg)); if (source !== 'worker') this.emit(Events.Page.Console, new ConsoleMessage(level, text, [], {url, lineNumber})); } /** * @return {!Puppeteer.Frame} */ mainFrame() { return this._frameManager.mainFrame(); } /** * @return {!Keyboard} */ get keyboard() { return this._keyboard; } /** * @return {!Touchscreen} */ get touchscreen() { return this._touchscreen; } /** * @return {!Coverage} */ get coverage() { return this._coverage; } /** * @return {!Tracing} */ get tracing() { return this._tracing; } /** * @return {!Accessibility} */ get accessibility() { return this._accessibility; } /** * @return {!Array<Puppeteer.Frame>} */ frames() { return this._frameManager.frames(); } /** * @return {!Array<!Worker>} */ workers() { return Array.from(this._workers.values()); } /** * @param {boolean} value */ /* async */ setRequestInterception(value) {return (fn => { const gen = fn.call(this); return new Promise((resolve, reject) => { function step(key, arg) { let info, value; try { info = gen[key](arg); value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { return Promise.resolve(value).then( value => { step('next', value); }, err => { step('throw', err); }); } } return step('next'); }); })(function*(){ return this._networkManager.setRequestInterception(value); });} /** * @param {boolean} enabled */ setOfflineMode(enabled) { return this._networkManager.setOfflineMode(enabled); } /** * @param {number} timeout */ setDefaultNavigationTimeout(timeout) { this._timeoutSettings.setDefaultNavigationTimeout(timeout); } /** * @param {number} timeout */ setDefaultTimeout(timeout) { this._timeoutSettings.setDefaultTimeout(timeout); } /** * @param {!Protocol.Security.certificateErrorPayload} event */ _onCertificateError(event) { if (!this._ignoreHTTPSErrors) return; this._client.send('Security.handleCertificateError', { eventId: event.eventId, action: 'continue' }).catch(debugError); } /** * @param {string} selector * @return {!Promise<?Puppeteer.ElementHandle>} */ /* async */ $(selector) {return (fn => { const gen = fn.call(this); return new Promise((resolve, reject) => { function step(key, arg) { let info, value; try { info = gen[key](arg); value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { return Promise.resolve(value).then( value => { step('next', value); }, err => { step('throw', err); }); } } return step('next'); }); })(function*(){ return this.mainFrame().$(selector); });} /** * @param {Function|string} pageFunction * @param {!Array<*>} args * @return {!Promise<!Puppeteer.JSHandle>} */ /* async */ evaluateHandle(pageFunction, ...args) {return (fn => { const gen = fn.call(this); return new Promise((resolve, reject) => { function step(key, arg) { let info, value; try { info = gen[key](arg); value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { return Promise.resolve(value).then( value => { step('next', value); }, err => { step('throw', err); }); } } return step('next'); }); })(function*(){ const context = (yield this.mainFrame().executionContext()); return context.evaluateHandle(pageFunction, ...args); });} /** * @param {!Puppeteer.JSHandle} prototypeHandle * @return {!Promise<!Puppeteer.JSHandle>} */ /* async */ queryObjects(prototypeHandle) {return (fn => { const gen = fn.call(this); return new Promise((resolve, reject) => { function step(key, arg) { let info, value; try { info = gen[key](arg); value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { return Promise.resolve(value).then( value => { step('next', value); }, err => { step('throw', err); }); } } return step('next'); }); })(function*(){ const context = (yield this.mainFrame().executionContext()); return context.queryObjects(prototypeHandle); });} /** * @param {string} selector * @param {Function|string} pageFunction * @param {!Array<*>} args * @return {!Promise<(!Object|undefined)>} */ /* async */ $eval(selector, pageFunction, ...args) {return (fn => { const gen = fn.call(this); return new Promise((resolve, reject) => { function step(key, arg) { let info, value; try { info = gen[key](arg); value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { return Promise.resolve(value).then( value => { step('next', value); }, err => { step('throw', err); }); } } return step('next'); }); })(function*(){ return this.mainFrame().$eval(selector, pageFunction, ...args); });} /** * @param {string} selector * @param {Function|string} pageFunction * @param {!Array<*>} args * @return {!Promise<(!Object|undefined)>} */ /* async */ $$eval(selector, pageFunction, ...args) {return (fn => { const gen = fn.call(this); return new Promise((resolve, reject) => { function step(key, arg) { let info, value; try { info = gen[key](arg); value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { return Promise.resolve(value).then( value => { step('next', value); }, err => { step('throw', err); }); } } return step('next'); }); })(function*(){ return this.mainFrame().$$eval(selector, pageFunction, ...args); });} /** * @param {string} selector * @return {!Promise<!Array<!Puppeteer.ElementHandle>>} */ /* async */ $$(selector) {return (fn => { const gen = fn.call(this); return new Promise((resolve, reject) => { function step(key, arg) { let info, value; try { info = gen[key](arg); value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { return Promise.resolve(value).then( value => { step('next', value); }, err => { step('throw', err); }); } } return step('next'); }); })(function*(){ return this.mainFrame().$$(selector); });} /** * @param {string} expression * @return {!Promise<!Array<!Puppeteer.ElementHandle>>} */ /* async */ $x(expression) {return (fn => { const gen = fn.call(this); return new Promise((resolve, reject) => { function step(key, arg) { let info, value; try { info = gen[key](arg); value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { return Promise.resolve(value).then( value => { step('next', value); }, err => { step('throw', err); }); } } return step('next'); }); })(function*(){ return this.mainFrame().$x(expression); });} /** * @param {!Array<string>} urls * @return {!Promise<!Array<Network.Cookie>>} */ /* async */ cookies(...urls) {return (fn => { const gen = fn.call(this); return new Promise((resolve, reject) => { function step(key, arg) { let info, value; try { info = gen[key](arg); value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { return Promise.resolve(value).then( value => { step('next', value); }, err => { step('throw', err); }); } } return step('next'); }); })(function*(){ return ((yield this._client.send('Network.getCookies', { urls: urls.length ? urls : [this.url()] }))).cookies; });} /** * @param {Array<Protocol.Network.deleteCookiesParameters>} cookies */ /* async */ deleteCookie(...cookies) {return (fn => { const gen = fn.call(this); return new Promise((resolve, reject) => { function step(key, arg) { let info, value; try { info = gen[key](arg); value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { return Promise.resolve(value).then( value => { step('next', value); }, err => { step('throw', err); }); } } return step('next'); }); })(function*(){ const pageURL = this.url(); for (const cookie of cookies) { const item = Object.assign({}, cookie); if (!cookie.url && pageURL.startsWith('http')) item.url = pageURL; (yield this._client.send('Network.deleteCookies', item)); } });} /** * @param {Array<Network.CookieParam>} cookies */ /* async */ setCookie(...cookies) {return (fn => { const gen = fn.call(this); return new Promise((resolve, reject) => { function step(key, arg) { let info, value; try { info = gen[key](arg); value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { return Promise.resolve(value).then( value => { step('next', value); }, err => { step('throw', err); }); } } return step('next'); }); })(function*(){ const pageURL = this.url(); const startsWithHTTP = pageURL.startsWith('http'); const items = cookies.map(cookie => { const item = Object.assign({}, cookie); if (!item.url && startsWithHTTP) item.url = pageURL; assert( item.url !== 'about:blank', `Blank page can not have cookie "${item.name}"` ); assert( !String.prototype.startsWith.call(item.url || '', 'data:'), `Data URL page can not have cookie "${item.name}"` ); return item; }); (yield this.deleteCookie(...items)); if (items.length) (yield this._client.send('Network.setCookies', { cookies: items })); });} /** * @param {!{url?: string, path?: string, content?: string, type?: string}} options * @return {!Promise<!Puppeteer.ElementHandle>} */ /* async */ addScriptTag(options) {return (fn => { const gen = fn.call(this); return new Promise((resolve, reject) => { function step(key, arg) { let info, value; try { info = gen[key](arg); value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { return Promise.resolve(value).then( value => { step('next', value); }, err => { step('throw', err); }); } } return step('next'); }); })(function*(){ return this.mainFrame().addScriptTag(options); });} /** * @param {!{url?: string, path?: string, content?: string}} options * @return {!Promise<!Puppeteer.ElementHandle>} */ /* async */ addStyleTag(options) {return (fn => { const gen = fn.call(this); return new Promise((resolve, reject) => { function step(key, arg) { let info, value; try { info = gen[key](arg); value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { return Promise.resolve(value).then( value => { step('next', value); }, err => { step('throw', err); }); } } return step('next'); }); })(function*(){ return this.mainFrame().addStyleTag(options); });} /** * @param {string} name * @param {Function} puppeteerFunction */ /* async */ exposeFunction(name, puppeteerFunction) {return (fn => { const gen = fn.call(this); return new Promise((resolve, reject) => { function step(key, arg) { let info, value; try { info = gen[key](arg); value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { return Promise.resolve(value).then( value => { step('next', value); }, err => { step('throw', err); }); } } return step('next'); }); })(function*(){ if (this._pageBindings.has(name)) throw new Error(`Failed to add page binding with name ${name}: window['${name}'] already exists!`); this._pageBindings.set(name, puppeteerFunction); const expression = helper.evaluationString(addPageBinding, name); (yield this._client.send('Runtime.addBinding', {name: name})); (yield this._client.send('Page.addScriptToEvaluateOnNewDocument', {source: expression})); (yield Promise.all(this.frames().map(frame => frame.evaluate(expression).catch(debugError)))); function addPageBinding(bindingName) { const binding = window[bindingName]; window[bindingName] = /* async */(...args) => {return (fn => { const gen = fn.call(this); return new Promise((resolve, reject) => { function step(key, arg) { let info, value; try { info = gen[key](arg); value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { return Promise.resolve(value).then( value => { step('next', value); }, err => { step('throw', err); }); } } return step('next'); }); })(function*(){ const me = window[bindingName]; let callbacks = me['callbacks']; if (!callbacks) { callbacks = new Map(); me['callbacks'] = callbacks; } const seq = (me['lastSeq'] || 0) + 1; me['lastSeq'] = seq; const promise = new Promise((resolve, reject) => callbacks.set(seq, {resolve, reject})); binding(JSON.stringify({name: bindingName, seq, args})); return promise; });}; } });} /** * @param {?{username: string, password: string}} credentials */ /* async */ authenticate(credentials) {return (fn => { const gen = fn.call(this); return new Promise((resolve, reject) => { function step(key, arg) { let info, value; try { info = gen[key](arg); value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { return Promise.resolve(value).then( value => { step('next', value); }, err => { step('throw', err); }); } } return step('next'); }); })(function*(){ return this._networkManager.authenticate(credentials); });} /** * @param {!Object<string, string>} headers */ /* async */ setExtraHTTPHeaders(headers) {return (fn => { const gen = fn.call(this); return new Promise((resolve, reject) => { function step(key, arg) { let info, value; try { info = gen[key](arg); value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { return Promise.resolve(value).then( value => { step('next', value); }, err => { step('throw', err); }); } } return step('next'); }); })(function*(){ return this._networkManager.setExtraHTTPHeaders(headers); });} /** * @param {string} userAgent */ /* async */ setUserAgent(userAgent) {return (fn => { const gen = fn.call(this); return new Promise((resolve, reject) => { function step(key, arg) { let info, value; try { info = gen[key](arg); value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { return Promise.resolve(value).then( value => { step('next', value); }, err => { step('throw', err); }); } } return step('next'); }); })(function*(){ return this._networkManager.setUserAgent(userAgent); });} /** * @return {!Promise<!Metrics>} */ /* async */ metrics() {return (fn => { const gen = fn.call(this); return new Promise((resolve, reject) => { function step(key, arg) { let info, value; try { info = gen[key](arg); value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { return Promise.resolve(value).then( value => { step('next', value); }, err => { step('throw', err); }); } } return step('next'); }); })(function*(){ const response = (yield this._client.send('Performance.getMetrics')); return this._buildMetricsObject(response.metrics); });} /** * @param {!Protocol.Performance.metricsPayload} event */ _emitMetrics(event) { this.emit(Events.Page.Metrics, { title: event.title, metrics: this._buildMetricsObject(event.metrics) }); } /** * @param {?Array<!Protocol.Performance.Metric>} metrics * @return {!Metrics} */ _buildMetricsObject(metrics) { const result = {}; for (const metric of metrics || []) { if (supportedMetrics.has(metric.name)) result[metric.name] = metric.value; } return result; } /** * @param {!Protocol.Runtime.ExceptionDetails} exceptionDetails */ _handleException(exceptionDetails) { const message = helper.getExceptionMessage(exceptionDetails); const err = new Error(message); err.stack = ''; // Don't report clientside error with a node stack attached this.emit(Events.Page.PageError, err); } /** * @param {!Protocol.Runtime.consoleAPICalledPayload} event */ /* async */ _onConsoleAPI(event) {return (fn => { const gen = fn.call(this); return new Promise((resolve, reject) => { function step(key, arg) { let info, value; try { info = gen[key](arg); value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { return Promise.resolve(value).then( value => { step('next', value); }, err => { step('throw', err); }); } } return step('next'); }); })(function*(){ if (event.executionContextId === 0) { // DevTools protocol stores the last 1000 console messages. These // messages are always reported even for removed execution contexts. In // this case, they are marked with executionContextId = 0 and are // reported upon enabling Runtime agent. // // Ignore these messages since: // - there's no execution context we can use to operate with message // arguments // - these messages are reported before Puppeteer clients can subscribe // to the 'console' // page event. // // @see https://github.com/GoogleChrome/puppeteer/issues/3865 return; } const context = this._frameManager.executionContextById(event.executionContextId); const values = event.args.map(arg => createJSHandle(context, arg)); this._addConsoleMessage(event.type, values, event.stackTrace); });} /** * @param {!Protocol.Runtime.bindingCalledPayload} event */ /* async */ _onBindingCalled(event) {return (fn => { const gen = fn.call(this); return new Promise((resolve, reject) => { function step(key, arg) { let info, value; try { info = gen[key](arg); value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { return Promise.resolve(value).then( value => { step('next', value); }, err => { step('throw', err); }); } } return step('next'); }); })(function*(){ const {name, seq, args} = JSON.parse(event.payload); let expression = null; try { const result = (yield this._pageBindings.get(name)(...args)); expression = helper.evaluationString(deliverResult, name, seq, result); } catch (error) { if (error instanceof Error) expression = helper.evaluationString(deliverError, name, seq, error.message, error.stack); else expression = helper.evaluationString(deliverErrorValue, name, seq, error); } this._client.send('Runtime.evaluate', { expression, contextId: event.executionContextId }).catch(debugError); /** * @param {string} name * @param {number} seq * @param {*} result */ function deliverResult(name, seq, result) { window[name]['callbacks'].get(seq).resolve(result); window[name]['callbacks'].delete(seq); } /** * @param {string} name * @param {number} seq * @param {string} message * @param {string} stack */ function deliverError(name, seq, message, stack) { const error = new Error(message); error.stack = stack; window[name]['callbacks'].get(seq).reject(error); window[name]['callbacks'].delete(seq); } /** * @param {string} name * @param {number} seq * @param {*} value */ function deliverErrorValue(name, seq, value) { window[name]['callbacks'].get(seq).reject(value); window[name]['callbacks'].delete(seq); } });} /** * @param {string} type * @param {!Array<!Puppeteer.JSHandle>} args * @param {Protocol.Runtime.StackTrace=} stackTrace */ _addConsoleMessage(type, args, stackTrace) { if (!this.listenerCount(Events.Page.Console)) { args.forEach(arg => arg.dispose()); return; } const textTokens = []; for (const arg of args) { const remoteObject = arg._remoteObject; if (remoteObject.objectId) textTokens.push(arg.toString()); else textTokens.push(helper.valueFromRemoteObject(remoteObject)); } const location = stackTrace && stackTrace.callFrames.length ? { url: stackTrace.callFrames[0].url, lineNumber: stackTrace.callFrames[0].lineNumber, columnNumber: stackTrace.callFrames[0].columnNumber, } : {}; const message = new ConsoleMessage(type, textTokens.join(' '), args, location); this.emit(Events.Page.Console, message); } _onDialog(event) { let dialogType = null; if (event.type === 'alert') dialogType = Dialog.Type.Alert; else if (event.type === 'confirm') dialogType = Dialog.Type.Confirm; else if (event.type === 'prompt') dialogType = Dialog.Type.Prompt; else if (event.type === 'beforeunload') dialogType = Dialog.Type.BeforeUnload; assert(dialogType, 'Unknown javascript dialog type: ' + event.type); const dialog = new Dialog(this._client, dialogType, event.message, event.defaultPrompt); this.emit(Events.Page.Dialog, dialog); } /** * @return {!string} */ url() { return this.mainFrame().url(); } /** * @return {!Promise<string>} */ /* async */ content() {return (fn => { const gen = fn.call(this); return new Promise((resolve, reject) => { function step(key, arg) { let info, value; try { info = gen[key](arg); value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { return Promise.resolve(value).then( value => { step('next', value); }, err => { step('throw', err); }); } } return step('next'); }); })(function*(){ return (yield this._frameManager.mainFrame().content()); });} /** * @param {string} html * @param {!{timeout?: number, waitUntil?: string|!Array<string>}=} options */ /* async */ setContent(html, options) {return (fn => { const gen = fn.call(this); return new Promise((resolve, reject) => { function step(key, arg) { let info, value; try { info = gen[key](arg); value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { return Promise.resolve(value).then( value => { step('next', value); }, err => { step('throw', err); }); } } return step('next'); }); })(function*(){ (yield this._frameManager.mainFrame().setContent(html, options)); });} /** * @param {string} url * @param {!{referer?: string, timeout?: number, waitUntil?: string|!Array<string>}=} options * @return {!Promise<?Puppeteer.Response>} */ /* async */ goto(url, options) {return (fn => { const gen = fn.call(this); return new Promise((resolve, reject) => { function step(key, arg) { let info, value; try { info = gen[key](arg); value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { return Promise.resolve(value).then( value => { step('next', value); }, err => { step('throw', err); }); } } return step('next'); }); })(function*(){ return (yield this._frameManager.mainFrame().goto(url, options)); });} /** * @param {!{timeout?: number, waitUntil?: string|!Array<string>}=} options * @return {!Promise<?Puppeteer.Response>} */ /* async */ reload(options) {return (fn => { const gen = fn.call(this); return new Promise((resolve, reject) => { function step(key, arg) { let info, value; try { info = gen[key](arg); value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { return Promise.resolve(value).then( value => { step('next', value); }, err => { step('throw', err); }); } } return step('next'); }); })(function*(){ const [response] = (yield Promise.all([ this.waitForNavigation(options), this._client.send('Page.reload') ])); return response; });} /** * @param {!{timeout?: number, waitUntil?: string|!Array<string>}=} options * @return {!Promise<?Puppeteer.Response>} */ /* async */ waitForNavigation(options = {}) {return (fn => { const gen = fn.call(this); return new Promise((resolve, reject) => { function step(key, arg) { let info, value; try { info = gen[key](arg); value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { return Promise.resolve(value).then( value => { step('next', value); }, err => { step('throw', err); }); } } return step('next'); }); })(function*(){ return (yield this._frameManager.mainFrame().waitForNavigation(options)); });} /** * @param {(string|Function)} urlOrPredicate * @param {!{timeout?: number}=} options * @return {!Promise<!Puppeteer.Request>} */ /* async */ waitForRequest(urlOrPredicate, options = {}) {return (fn => { const gen = fn.call(this); return new Promise((resolve, reject) => { function step(key, arg) { let info, value; try { info = gen[key](arg); value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { return Promise.resolve(value).then( value => { step('next', value); }, err => { step('throw', err); }); } } return step('next'); }); })(function*(){ const { timeout = this._timeoutSettings.timeout(), } = options; return helper.waitForEvent(this._networkManager, Events.NetworkManager.Request, request => { if (helper.isString(urlOrPredicate)) return (urlOrPredicate === request.url()); if (typeof urlOrPredicate === 'function') return !!(urlOrPredicate(request)); return false; }, timeout); });} /** * @param {(string|Function)} urlOrPredicate * @param {!{timeout?: number}=} options * @return {!Promise<!Puppeteer.Response>} */ /* async */ waitForResponse(urlOrPredicate, options = {}) {return (fn => { const gen = fn.call(this); return new Promise((resolve, reject) => { function step(key, arg) { let info, value; try { info = gen[key](arg); value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { return Promise.resolve(value).then( value => { step('next', value); }, err => { step('throw', err); }); } } return step('next'); }); })(function*(){ const { timeout = this._timeoutSettings.timeout(), } = options; return helper.waitForEvent(this._networkManager, Events.NetworkManager.Response, response => { if (helper.isString(urlOrPredicate)) return (urlOrPredicate === response.url()); if (typeof urlOrPredicate === 'function') return !!(urlOrPredicate(response)); return false; }, timeout); });} /** * @param {!{timeout?: number, waitUntil?: string|!Array<string>}=} options * @return {!Promise<?Puppeteer.Response>} */ /* async */ goBack(options) {return (fn => { const gen = fn.call(this); return new Promise((resolve, reject) => { function step(key, arg) { let info, value; try { info = gen[key](arg); value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { return Promise.resolve(value).then( value => { step('next', value); }, err => { step('throw', err); }); } } return step('next'); }); })(function*(){ return this._go(-1, options); });} /** * @param {!{timeout?: number, waitUntil?: string|!Array<string>}=} options * @return {!Promise<?Puppeteer.Response>} */ /* async */ goForward(options) {return (fn => { const gen = fn.call(this); return new Promise((resolve, reject) => { function step(key, arg) { let info, value; try { info = gen[key](arg); value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { return Promise.resolve(value).then( value => { step('next', value); }, err => { step('throw', err); }); } } return step('next'); }); })(function*(){ return this._go(+1, options); });} /** * @param {!{timeout?: number, waitUntil?: string|!Array<string>}=} options * @return {!Promise<?Puppeteer.Response>} */ /* async */ _go(delta, options) {return (fn => { const gen = fn.call(this); return new Promise((resolve, reject) => { function step(key, arg) { let info, value; try { info = gen[key](arg); value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { return Promise.resolve(value).then( value => { step('next', value); }, err => { step('throw', err); }); } } return step('next'); }); })(function*(){ const history = (yield this._client.send('Page.getNavigationHistory')); const entry = history.entries[history.currentIndex + delta]; if (!entry) return null; const [response] = (yield Promise.all([ this.waitForNavigation(options), this._client.send('Page.navigateToHistoryEntry', {entryId: entry.id}), ])); return response; });} /* async */ bringToFront() {return (fn => { const gen = fn.call(this); return new Promise((resolve, reject) => { function step(key, arg) { let info, value; try { info = gen[key](arg); value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { return Promise.resolve(value).then( value => { step('next', value); }, err => { step('throw', err); }); } } return step('next'); }); })(function*(){ (yield this._client.send('Page.bringToFront')); });} /** * @param {!{viewport: !Puppeteer.Viewport, userAgent: string}} options */ /* async */ emulate(options) {return (fn => { const gen = fn.call(this); return new Promise((resolve, reject) => { function step(key, arg) { let info, value; try { info = gen[key](arg); value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { return Promise.resolve(value).then( value => { step('next', value); }, err => { step('throw', err); }); } } return step('next'); }); })(function*(){ (yield Promise.all([ this.setViewport(options.viewport), this.setUserAgent(options.userAgent) ])); });} /** * @param {boolean} enabled */ /* async */ setJavaScriptEnabled(enabled) {return (fn => { const gen = fn.call(this); return new Promise((resolve, reject) => { function step(key, arg) { let info, value; try { info = gen[key](arg); value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { return Promise.resolve(value).then( value => { step('next', value); }, err => { step('throw', err); }); } } return step('next'); }); })(function*(){ if (this._javascriptEnabled === enabled) return; this._javascriptEnabled = enabled; (yield this._client.send('Emulation.setScriptExecutionDisabled', { value: !enabled })); });} /** * @param {boolean} enabled */ /* async */ setBypassCSP(enabled) {return (fn => { const gen = fn.call(this); return new Promise((resolve, reject) => { function step(key, arg) { let info, value; try { info = gen[key](arg); value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { return Promise.resolve(value).then( value => { step('next', value); }, err => { step('throw', err); }); } } return step('next'); }); })(function*(){ (yield this._client.send('Page.setBypassCSP', { enabled })); });} /** * @param {?string} mediaType */ /* async */ emulateMedia(mediaType) {return (fn => { const gen = fn.call(this); return new Promise((resolve, reject) => { function step(key, arg) { let info, value; try { info = gen[key](arg); value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { return Promise.resolve(value).then( value => { step('next', value); }, err => { step('throw', err); }); } } return step('next'); }); })(function*(){ assert(mediaType === 'screen' || mediaType === 'print' || mediaType === null, 'Unsupported media type: ' + mediaType); (yield this._client.send('Emulation.setEmulatedMedia', {media: mediaType || ''})); });} /** * @param {!Puppeteer.Viewport} viewport */ /* async */ setViewport(viewport) {return (fn => { const gen = fn.call(this); return new Promise((resolve, reject) => { function step(key, arg) { let info, value; try { info = gen[key](arg); value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { return Promise.resolve(value).then( value => { step('next', value); }, err => { step('throw', err); }); } } return step('next'); }); })(function*(){ const needsReload = (yield this._emulationManager.emulateViewport(viewport)); this._viewport = viewport; if (needsReload) (yield this.reload()); });} /** * @return {?Puppeteer.Viewport} */ viewport() { return this._viewport; } /** * @param {Function|string} pageFunction * @param {!Array<*>} args * @return {!Promise<*>} */ /* async */ evaluate(pageFunction, ...args) {return (fn => { const gen = fn.call(this); return new Promise((resolve, reject) => { function step(key, arg) { let info, value; try { info = gen[key](arg); value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { return Promise.resolve(value).then( value => { step('next', value); }, err => { step('throw', err); }); } } return step('next'); }); })(function*(){ return this._frameManager.mainFrame().evaluate(pageFunction, ...args); });} /** * @param {Function|string} pageFunction * @param {!Array<*>} args */ /* async */ evaluateOnNewDocument(pageFunction, ...args) {return (fn => { const gen = fn.call(this); return new Promise((resolve, reject) => { function step(key, arg) { let info, value; try { info = gen[key](arg); value = info.value; } catch (error) { reject(error); return; }