mirror of
https://github.com/puppeteer/puppeteer
synced 2024-06-14 14:02:48 +00:00
102 lines
2.8 KiB
TypeScript
102 lines
2.8 KiB
TypeScript
/**
|
|
* Copyright 2022 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.
|
|
*/
|
|
|
|
import * as Bidi from 'chromium-bidi/lib/cjs/protocol/protocol.js';
|
|
|
|
import {BrowserContext as BrowserContextBase} from '../../api/BrowserContext.js';
|
|
import {Page as PageBase} from '../../api/Page.js';
|
|
import {Viewport} from '../PuppeteerViewport.js';
|
|
|
|
import {Browser} from './Browser.js';
|
|
import {Connection} from './Connection.js';
|
|
import {Page} from './Page.js';
|
|
import {debugError} from './utils.js';
|
|
|
|
interface BrowserContextOptions {
|
|
defaultViewport: Viewport | null;
|
|
}
|
|
|
|
/**
|
|
* @internal
|
|
*/
|
|
export class BrowserContext extends BrowserContextBase {
|
|
#browser: Browser;
|
|
#connection: Connection;
|
|
#defaultViewport: Viewport | null;
|
|
#pages = new Map<string, Page>();
|
|
#onContextDestroyedBind = this.#onContextDestroyed.bind(this);
|
|
|
|
constructor(browser: Browser, options: BrowserContextOptions) {
|
|
super();
|
|
this.#browser = browser;
|
|
this.#connection = this.#browser.connection;
|
|
this.#defaultViewport = options.defaultViewport;
|
|
this.#connection.on(
|
|
'browsingContext.contextDestroyed',
|
|
this.#onContextDestroyedBind
|
|
);
|
|
}
|
|
|
|
get connection(): Connection {
|
|
return this.#connection;
|
|
}
|
|
|
|
async #onContextDestroyed(
|
|
event: Bidi.BrowsingContext.ContextDestroyedEvent['params']
|
|
) {
|
|
const page = this.#pages.get(event.context);
|
|
await page?.close().catch(error => {
|
|
debugError(error);
|
|
});
|
|
this.#pages.delete(event.context);
|
|
}
|
|
|
|
override async newPage(): Promise<PageBase> {
|
|
const {result} = await this.#connection.send('browsingContext.create', {
|
|
type: 'tab',
|
|
});
|
|
const page = new Page(this, result);
|
|
if (this.#defaultViewport) {
|
|
try {
|
|
await page.setViewport(this.#defaultViewport);
|
|
} catch {
|
|
// No support for setViewport in Firefox.
|
|
}
|
|
}
|
|
|
|
this.#pages.set(result.context, page);
|
|
|
|
return page;
|
|
}
|
|
|
|
override async close(): Promise<void> {
|
|
for (const page of this.#pages.values()) {
|
|
await page?.close().catch(error => {
|
|
debugError(error);
|
|
});
|
|
}
|
|
this.#pages.clear();
|
|
}
|
|
|
|
override browser(): Browser {
|
|
return this.#browser;
|
|
}
|
|
|
|
override async pages(): Promise<PageBase[]> {
|
|
return [...this.#pages.values()];
|
|
}
|
|
}
|