Adding upstream version 124.0.1.upstream/124.0.1

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>

1 files changed, 145 insertions, 0 deletions

diff --git a/remote/test/puppeteer/packages/puppeteer-core/src/bidi/BrowserContext.ts b/remote/test/puppeteer/packages/puppeteer-core/src/bidi/BrowserContext.ts
new file mode 100644
index 0000000000..feb5e9951d
--- /dev/null
+++ b/remote/test/puppeteer/packages/puppeteer-core/src/bidi/BrowserContext.ts
@@ -0,0 +1,145 @@
+/**
+ * @license
+ * Copyright 2022 Google Inc.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import * as Bidi from 'chromium-bidi/lib/cjs/protocol/protocol.js';
+
+import type {WaitForTargetOptions} from '../api/Browser.js';
+import {BrowserContext} from '../api/BrowserContext.js';
+import type {Page} from '../api/Page.js';
+import type {Target} from '../api/Target.js';
+import {UnsupportedOperation} from '../common/Errors.js';
+import {debugError} from '../common/util.js';
+import type {Viewport} from '../common/Viewport.js';
+
+import type {BidiBrowser} from './Browser.js';
+import type {BidiConnection} from './Connection.js';
+import {UserContext} from './core/UserContext.js';
+import type {BidiPage} from './Page.js';
+
+/**
+ * @internal
+ */
+export interface BidiBrowserContextOptions {
+  defaultViewport: Viewport | null;
+}
+
+/**
+ * @internal
+ */
+export class BidiBrowserContext extends BrowserContext {
+  #browser: BidiBrowser;
+  #connection: BidiConnection;
+  #defaultViewport: Viewport | null;
+  #userContext: UserContext;
+
+  constructor(
+    browser: BidiBrowser,
+    userContext: UserContext,
+    options: BidiBrowserContextOptions
+  ) {
+    super();
+    this.#browser = browser;
+    this.#userContext = userContext;
+    this.#connection = this.#browser.connection;
+    this.#defaultViewport = options.defaultViewport;
+  }
+
+  override targets(): Target[] {
+    return this.#browser.targets().filter(target => {
+      return target.browserContext() === this;
+    });
+  }
+
+  override waitForTarget(
+    predicate: (x: Target) => boolean | Promise<boolean>,
+    options: WaitForTargetOptions = {}
+  ): Promise<Target> {
+    return this.#browser.waitForTarget(target => {
+      return target.browserContext() === this && predicate(target);
+    }, options);
+  }
+
+  get connection(): BidiConnection {
+    return this.#connection;
+  }
+
+  override async newPage(): Promise<Page> {
+    const {result} = await this.#connection.send('browsingContext.create', {
+      type: Bidi.BrowsingContext.CreateType.Tab,
+    });
+    const target = this.#browser._getTargetById(result.context);
+
+    // TODO: once BiDi has some concept matching BrowserContext, the newly
+    // created contexts should get automatically assigned to the right
+    // BrowserContext. For now, we assume that only explicitly created pages go
+    // to the current BrowserContext. Otherwise, the contexts get assigned to
+    // the default BrowserContext by the Browser.
+    target._setBrowserContext(this);
+
+    const page = await target.page();
+    if (!page) {
+      throw new Error('Page is not found');
+    }
+    if (this.#defaultViewport) {
+      try {
+        await page.setViewport(this.#defaultViewport);
+      } catch {
+        // No support for setViewport in Firefox.
+      }
+    }
+
+    return page;
+  }
+
+  override async close(): Promise<void> {
+    if (!this.isIncognito()) {
+      throw new Error('Default context cannot be closed!');
+    }
+
+    // TODO: Remove once we have adopted the new browsing contexts.
+    for (const target of this.targets()) {
+      const page = await target?.page();
+      try {
+        await page?.close();
+      } catch (error) {
+        debugError(error);
+      }
+    }
+
+    try {
+      await this.#userContext.remove();
+    } catch (error) {
+      debugError(error);
+    }
+  }
+
+  override browser(): BidiBrowser {
+    return this.#browser;
+  }
+
+  override async pages(): Promise<BidiPage[]> {
+    const results = await Promise.all(
+      [...this.targets()].map(t => {
+        return t.page();
+      })
+    );
+    return results.filter((p): p is BidiPage => {
+      return p !== null;
+    });
+  }
+
+  override isIncognito(): boolean {
+    return this.#userContext.id !== UserContext.DEFAULT;
+  }
+
+  override overridePermissions(): never {
+    throw new UnsupportedOperation();
+  }
+
+  override clearPermissionOverrides(): never {
+    throw new UnsupportedOperation();
+  }
+}