/** * 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. */ import * as types from './types'; import { BrowserContext, BrowserContextBase } from './browserContext'; import { Page } from './page'; import { EventEmitter } from 'events'; import { Download } from './download'; import { Events } from './events'; import { ProxySettings } from './types'; import { ChildProcess } from 'child_process'; export interface BrowserProcess { onclose: ((exitCode: number | null, signal: string | null) => void) | undefined; process: ChildProcess; kill(): Promise; close(): Promise; } export type BrowserOptions = { name: string, downloadsPath?: string, headful?: boolean, persistent?: types.BrowserContextOptions, // Undefined means no persistent context. slowMo?: number, browserProcess: BrowserProcess, proxy?: ProxySettings, }; export interface Browser extends EventEmitter { newContext(options?: types.BrowserContextOptions): Promise; contexts(): BrowserContext[]; newPage(options?: types.BrowserContextOptions): Promise; isConnected(): boolean; close(): Promise; version(): string; } export abstract class BrowserBase extends EventEmitter implements Browser { readonly _options: BrowserOptions; private _downloads = new Map(); _defaultContext: BrowserContextBase | null = null; private _startedClosing = false; constructor(options: BrowserOptions) { super(); this._options = options; } abstract newContext(options?: types.BrowserContextOptions): Promise; abstract contexts(): BrowserContext[]; abstract isConnected(): boolean; abstract version(): string; async newPage(options?: types.BrowserContextOptions): Promise { const context = await this.newContext(options); const page = await context.newPage(); page._ownedContext = context; return page; } _downloadCreated(page: Page, uuid: string, url: string, suggestedFilename?: string) { const download = new Download(page, this._options.downloadsPath || '', uuid, url, suggestedFilename); this._downloads.set(uuid, download); } _downloadFilenameSuggested(uuid: string, suggestedFilename: string) { const download = this._downloads.get(uuid); if (!download) return; download._filenameSuggested(suggestedFilename); } _downloadFinished(uuid: string, error?: string) { const download = this._downloads.get(uuid); if (!download) return; download._reportFinished(error); this._downloads.delete(uuid); } _didClose() { for (const context of this.contexts()) (context as BrowserContextBase)._browserClosed(); if (this._defaultContext) this._defaultContext._browserClosed(); this.emit(Events.Browser.Disconnected); } async close() { if (!this._startedClosing) { this._startedClosing = true; await this._options.browserProcess.close(); } if (this.isConnected()) await new Promise(x => this.once(Events.Browser.Disconnected, x)); } }