thunderstrike/puppeteer
thunderstrike/puppeteer
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: avoid double subscription to frame manager in Page (#11091)

Browse Source
This commit is contained in:
Alex Rudenko 2023-10-06 12:38:44 +02:00 committed by GitHub
parent 9c0c299f0b
commit 5887649891
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 20 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
packages/puppeteer-core/src/cdp/Page.ts
View File

@ -246,7 +246,16 @@ export class CdpPage extends Page {
this.#coverage = new Coverage(client);
this.#viewport = null;
this.#setupEventListeners();
for (const [eventName, handler] of this.#frameManagerHandlers) {
this.#frameManager.on(eventName, handler);
}
for (const [eventName, handler] of this.#networkManagerHandlers) {
// TODO: Remove any.
this.#frameManager.networkManager.on(eventName, handler as any);
}
this.#setupPrimaryTargetListeners();
this.#tabSession?.on(CDPSessionEvent.Swapped, async newSession => {
this.#client = newSession;
@ -264,7 +273,7 @@ export class CdpPage extends Page {
this.#tracing.updateClient(newSession);
this.#coverage.updateClient(newSession);
await this.#frameManager.swapFrameTree(newSession);
this.#setupEventListeners();
this.#setupPrimaryTargetListeners();
});
this.#tabSession?.on(CDPSessionEvent.Ready, session => {
assert(session instanceof CdpCDPSession);
@ -278,27 +287,22 @@ export class CdpPage extends Page {
});
}
#setupEventListeners() {
/**
* Sets up listeners for the primary target. The primary target can change
* during a navigation to a prerended page.
*/
#setupPrimaryTargetListeners() {
this.#client.on(CDPSessionEvent.Ready, this.#onAttachedToTarget);
this.#target
._targetManager()
.on(TargetManagerEvent.TargetGone, this.#onDetachedFromTarget);
for (const [eventName, handler] of this.#frameManagerHandlers) {
this.#frameManager.on(eventName, handler);
}
for (const [eventName, handler] of this.#networkManagerHandlers) {
// TODO: Remove any.
this.#frameManager.networkManager.on(eventName, handler as any);
}
for (const [eventName, handler] of this.#sessionHandlers) {
// TODO: Remove any.
this.#client.on(eventName, handler as any);
}
this.#target
._targetManager()
.on(TargetManagerEvent.TargetGone, this.#onDetachedFromTarget);
this.#target._isClosedDeferred
.valueOrThrow()
.then(() => {