playwright/src/rpc/server/browserDispatcher.ts

/**
 * 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 { BrowserBase, Browser } from '../../browser';
import { BrowserContextBase } from '../../browserContext';
import * as types from '../../types';
import { BrowserContextDispatcher } from './browserContextDispatcher';
import { BrowserChannel, BrowserContextChannel, PageChannel, BrowserInitializer } from '../channels';
import { Dispatcher, DispatcherScope, lookupDispatcher } from '../dispatcher';
import { PageDispatcher } from './pageDispatcher';
import { Events } from '../../events';

export class BrowserDispatcher extends Dispatcher<Browser, BrowserInitializer> implements BrowserChannel {
  constructor(scope: DispatcherScope, browser: BrowserBase) {
    super(scope, browser, 'browser', {});
    browser.on(Events.Browser.Disconnected, () => this._dispatchEvent('close'));
  }

  async newContext(params: { options?: types.BrowserContextOptions }): Promise<BrowserContextChannel> {
    return new BrowserContextDispatcher(this._scope, await this._object.newContext(params.options) as BrowserContextBase);
  }

  async newPage(params: { options?: types.BrowserContextOptions }): Promise<PageChannel> {
    return lookupDispatcher<PageDispatcher>(await this._object.newPage(params.options))!;
  }

  async close(): Promise<void> {
    await this._object.close();
  }
}