refactor: move to flatten protocol (#3827)

DevTools protocol is dropping nested targets and switching to
flatten protocol. This patch adopts the new scheme.

Once this change lands, tip-of-tree Puppeteer will be incompatible
with Chromium below 72.0.3606.0. Chromium 72 goes stable on [Jan, 29](https://www.chromestatus.com/features/schedule) - the same time we release the
next version of Puppeteer, so this change won't hurt those clients who try using
tip-of-tree Puppeteer with stable chrome. 

For the record: the previous attempt to land this was https://github.com/GoogleChrome/puppeteer/pull/3524.
This commit is contained in:
Andrey Lushnikov 2019-01-22 18:10:11 -05:00 committed by GitHub
parent 678b8e85ad
commit 89a5c396bf
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 48 additions and 71 deletions

View File

@ -16,7 +16,6 @@
const {helper, assert} = require('./helper'); const {helper, assert} = require('./helper');
const {Events} = require('./Events'); const {Events} = require('./Events');
const debugProtocol = require('debug')('puppeteer:protocol'); const debugProtocol = require('debug')('puppeteer:protocol');
const debugSession = require('debug')('puppeteer:session');
const EventEmitter = require('events'); const EventEmitter = require('events');
class Connection extends EventEmitter { class Connection extends EventEmitter {
@ -46,11 +45,15 @@ class Connection extends EventEmitter {
* @return {!Connection} * @return {!Connection}
*/ */
static fromSession(session) { static fromSession(session) {
let connection = session._connection; return session._connection;
// TODO(lushnikov): move to flatten protocol to avoid this. }
while (connection instanceof CDPSession)
connection = connection._connection; /**
return connection; * @param {string} sessionId
* @return {?CDPSession}
*/
session(sessionId) {
return this._sessions.get(sessionId) || null;
} }
/** /**
@ -66,15 +69,24 @@ class Connection extends EventEmitter {
* @return {!Promise<?Object>} * @return {!Promise<?Object>}
*/ */
send(method, params = {}) { send(method, params = {}) {
const id = ++this._lastId; const id = this._rawSend({method, params});
const message = JSON.stringify({id, method, params});
debugProtocol('SEND ► ' + message);
this._transport.send(message);
return new Promise((resolve, reject) => { return new Promise((resolve, reject) => {
this._callbacks.set(id, {resolve, reject, error: new Error(), method}); this._callbacks.set(id, {resolve, reject, error: new Error(), method});
}); });
} }
/**
* @param {*} message
* @return {number}
*/
_rawSend(message) {
const id = ++this._lastId;
message = JSON.stringify(Object.assign({}, message, {id}));
debugProtocol('SEND ► ' + message);
this._transport.send(message);
return id;
}
/** /**
* @param {string} message * @param {string} message
*/ */
@ -83,7 +95,22 @@ class Connection extends EventEmitter {
await new Promise(f => setTimeout(f, this._delay)); await new Promise(f => setTimeout(f, this._delay));
debugProtocol('◀ RECV ' + message); debugProtocol('◀ RECV ' + message);
const object = JSON.parse(message); const object = JSON.parse(message);
if (object.id) { if (object.method === 'Target.attachedToTarget') {
const sessionId = object.params.sessionId;
const session = new CDPSession(this, object.params.targetInfo.type, sessionId);
this._sessions.set(sessionId, session);
} else if (object.method === 'Target.detachedFromTarget') {
const session = this._sessions.get(object.params.sessionId);
if (session) {
session._onClosed();
this._sessions.delete(object.params.sessionId);
}
}
if (object.sessionId) {
const session = this._sessions.get(object.sessionId);
if (session)
session._onMessage(object);
} else if (object.id) {
const callback = this._callbacks.get(object.id); const callback = this._callbacks.get(object.id);
// Callbacks could be all rejected if someone has called `.dispose()`. // Callbacks could be all rejected if someone has called `.dispose()`.
if (callback) { if (callback) {
@ -93,21 +120,10 @@ class Connection extends EventEmitter {
else else
callback.resolve(object.result); callback.resolve(object.result);
} }
} else {
if (object.method === 'Target.receivedMessageFromTarget') {
const session = this._sessions.get(object.params.sessionId);
if (session)
session._onMessage(object.params.message);
} else if (object.method === 'Target.detachedFromTarget') {
const session = this._sessions.get(object.params.sessionId);
if (session)
session._onClosed();
this._sessions.delete(object.params.sessionId);
} else { } else {
this.emit(object.method, object.params); this.emit(object.method, object.params);
} }
} }
}
_onClose() { _onClose() {
if (this._closed) if (this._closed)
@ -134,30 +150,24 @@ class Connection extends EventEmitter {
* @return {!Promise<!CDPSession>} * @return {!Promise<!CDPSession>}
*/ */
async createSession(targetInfo) { async createSession(targetInfo) {
const {sessionId} = await this.send('Target.attachToTarget', {targetId: targetInfo.targetId}); const {sessionId} = await this.send('Target.attachToTarget', {targetId: targetInfo.targetId, flatten: true});
const session = new CDPSession(this, targetInfo.type, sessionId); return this._sessions.get(sessionId);
this._sessions.set(sessionId, session);
return session;
} }
} }
class CDPSession extends EventEmitter { class CDPSession extends EventEmitter {
/** /**
* @param {!Connection|!CDPSession} connection * @param {!Connection} connection
* @param {string} targetType * @param {string} targetType
* @param {string} sessionId * @param {string} sessionId
*/ */
constructor(connection, targetType, sessionId) { constructor(connection, targetType, sessionId) {
super(); super();
this._lastId = 0;
/** @type {!Map<number, {resolve: function, reject: function, error: !Error, method: string}>}*/ /** @type {!Map<number, {resolve: function, reject: function, error: !Error, method: string}>}*/
this._callbacks = new Map(); this._callbacks = new Map();
/** @type {null|Connection|CDPSession} */
this._connection = connection; this._connection = connection;
this._targetType = targetType; this._targetType = targetType;
this._sessionId = sessionId; this._sessionId = sessionId;
/** @type {!Map<string, !CDPSession>}*/
this._sessions = new Map();
} }
/** /**
@ -168,28 +178,16 @@ class CDPSession extends EventEmitter {
send(method, params = {}) { send(method, params = {}) {
if (!this._connection) if (!this._connection)
return Promise.reject(new Error(`Protocol error (${method}): Session closed. Most likely the ${this._targetType} has been closed.`)); return Promise.reject(new Error(`Protocol error (${method}): Session closed. Most likely the ${this._targetType} has been closed.`));
const id = ++this._lastId; const id = this._connection._rawSend({sessionId: this._sessionId, method, params});
const message = JSON.stringify({id, method, params});
debugSession('SEND ► ' + message);
this._connection.send('Target.sendMessageToTarget', {sessionId: this._sessionId, message}).catch(e => {
// The response from target might have been already dispatched.
if (!this._callbacks.has(id))
return;
const callback = this._callbacks.get(id);
this._callbacks.delete(id);
callback.reject(rewriteError(callback.error, e && e.message));
});
return new Promise((resolve, reject) => { return new Promise((resolve, reject) => {
this._callbacks.set(id, {resolve, reject, error: new Error(), method}); this._callbacks.set(id, {resolve, reject, error: new Error(), method});
}); });
} }
/** /**
* @param {string} message * @param {{id?: number, method: string, params: Object, error: {message: string, data: any}, result?: *}} object
*/ */
_onMessage(message) { _onMessage(object) {
debugSession('◀ RECV ' + message);
const object = JSON.parse(message);
if (object.id && this._callbacks.has(object.id)) { if (object.id && this._callbacks.has(object.id)) {
const callback = this._callbacks.get(object.id); const callback = this._callbacks.get(object.id);
this._callbacks.delete(object.id); this._callbacks.delete(object.id);
@ -198,17 +196,6 @@ class CDPSession extends EventEmitter {
else else
callback.resolve(object.result); callback.resolve(object.result);
} else { } else {
if (object.method === 'Target.receivedMessageFromTarget') {
const session = this._sessions.get(object.params.sessionId);
if (session)
session._onMessage(object.params.message);
} else if (object.method === 'Target.detachedFromTarget') {
const session = this._sessions.get(object.params.sessionId);
if (session) {
session._onClosed();
this._sessions.delete(object.params.sessionId);
}
}
assert(!object.id); assert(!object.id);
this.emit(object.method, object.params); this.emit(object.method, object.params);
} }
@ -227,16 +214,6 @@ class CDPSession extends EventEmitter {
this._connection = null; this._connection = null;
this.emit(Events.CDPSession.Disconnected); this.emit(Events.CDPSession.Disconnected);
} }
/**
* @param {string} targetType
* @param {string} sessionId
*/
_createSession(targetType, sessionId) {
const session = new CDPSession(this, targetType, sessionId);
this._sessions.set(sessionId, session);
return session;
}
} }
helper.tracePublicAPI(CDPSession); helper.tracePublicAPI(CDPSession);

View File

@ -18,6 +18,7 @@ const fs = require('fs');
const EventEmitter = require('events'); const EventEmitter = require('events');
const mime = require('mime'); const mime = require('mime');
const {Events} = require('./Events'); const {Events} = require('./Events');
const {Connection} = require('./Connection');
const {NetworkManager} = require('./NetworkManager'); const {NetworkManager} = require('./NetworkManager');
const {Dialog} = require('./Dialog'); const {Dialog} = require('./Dialog');
const {EmulationManager} = require('./EmulationManager'); const {EmulationManager} = require('./EmulationManager');
@ -47,7 +48,7 @@ class Page extends EventEmitter {
const page = new Page(client, target, frameTree, ignoreHTTPSErrors, screenshotTaskQueue); const page = new Page(client, target, frameTree, ignoreHTTPSErrors, screenshotTaskQueue);
await Promise.all([ await Promise.all([
client.send('Target.setAutoAttach', {autoAttach: true, waitForDebuggerOnStart: false}), client.send('Target.setAutoAttach', {autoAttach: true, waitForDebuggerOnStart: false, flatten: true}),
client.send('Page.setLifecycleEventsEnabled', { enabled: true }), client.send('Page.setLifecycleEventsEnabled', { enabled: true }),
client.send('Network.enable', {}), client.send('Network.enable', {}),
client.send('Runtime.enable', {}).then(() => page._frameManager.ensureSecondaryDOMWorld()), client.send('Runtime.enable', {}).then(() => page._frameManager.ensureSecondaryDOMWorld()),
@ -106,11 +107,10 @@ class Page extends EventEmitter {
}).catch(debugError); }).catch(debugError);
return; return;
} }
const session = client._createSession(event.targetInfo.type, event.sessionId); const session = Connection.fromSession(client).session(event.sessionId);
const worker = new Worker(session, event.targetInfo.url, this._addConsoleMessage.bind(this), this._handleException.bind(this)); const worker = new Worker(session, event.targetInfo.url, this._addConsoleMessage.bind(this), this._handleException.bind(this));
this._workers.set(event.sessionId, worker); this._workers.set(event.sessionId, worker);
this.emit(Events.Page.WorkerCreated, worker); this.emit(Events.Page.WorkerCreated, worker);
}); });
client.on('Target.detachedFromTarget', event => { client.on('Target.detachedFromTarget', event => {
const worker = this._workers.get(event.sessionId); const worker = this._workers.get(event.sessionId);