@checkly/playwright-core 1.47.12 → 1.48.10-alpha

package/lib/server/bidi/bidiFirefox.js

@@ -10,6 +10,7 @@ var _utils = require("../../utils");
 var _browserType = require("../browserType");
 var _bidiBrowser = require("./bidiBrowser");
 var _bidiConnection = require("./bidiConnection");
+var _firefoxPrefs = require("./third_party/firefoxPrefs");
 function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
 /**
  * Copyright (c) Microsoft Corporation.
@@ -44,6 +45,12 @@ class BidiFirefox extends _browserType.BrowserType {
   }
   amendEnvironment(env, userDataDir, executable, browserArguments) {
     if (!_path.default.isAbsolute(_os.default.homedir())) throw new Error(`Cannot launch Firefox with relative home directory. Did you set ${_os.default.platform() === 'win32' ? 'USERPROFILE' : 'HOME'} to a relative path?`);
+    env = {
+      ...env,
+      'MOZ_CRASHREPORTER': '1',
+      'MOZ_CRASHREPORTER_NO_REPORT': '1',
+      'MOZ_CRASHREPORTER_SHUTDOWN': '1'
+    };
     if (_os.default.platform() === 'linux') {
       // Always remove SNAP_NAME and SNAP_INSTANCE_NAME env variables since they
       // confuse Firefox: in our case, builds never come from SNAP.
@@ -63,6 +70,12 @@ class BidiFirefox extends _browserType.BrowserType {
       id: _bidiConnection.kBrowserCloseMessageId
     });
   }
+  async prepareUserDataDir(options, userDataDir) {
+    await (0, _firefoxPrefs.createProfile)({
+      path: userDataDir,
+      preferences: options.firefoxUserPrefs || {}
+    });
+  }
   defaultArgs(options, isPersistent, userDataDir) {
     const {
       args = [],
@@ -74,37 +87,18 @@ class BidiFirefox extends _browserType.BrowserType {
     if (headless) firefoxArguments.push('--headless');else firefoxArguments.push('--foreground');
     firefoxArguments.push(`--profile`, userDataDir);
     firefoxArguments.push(...args);
-    // TODO: make ephemeral context work without this argument.
-    firefoxArguments.push('about:blank');
-    // if (isPersistent)
-    //   firefoxArguments.push('about:blank');
-    // else
-    //   firefoxArguments.push('-silent');
     return firefoxArguments;
   }
   readyState(options) {
     (0, _utils.assert)(options.useWebSocket);
-    return new BidiReadyState();
+    return new FirefoxReadyState();
   }
 }
 exports.BidiFirefox = BidiFirefox;
-class BidiReadyState {
-  constructor() {
-    this._wsEndpoint = new _utils.ManualPromise();
-  }
+class FirefoxReadyState extends _browserType.BrowserReadyState {
   onBrowserOutput(message) {
     // Bidi WebSocket in Firefox.
     const match = message.match(/WebDriver BiDi listening on (ws:\/\/.*)$/);
     if (match) this._wsEndpoint.resolve(match[1] + '/session');
   }
-  onBrowserExit() {
-    // Unblock launch when browser prematurely exits.
-    this._wsEndpoint.resolve(undefined);
-  }
-  async waitUntilReady() {
-    const wsEndpoint = await this._wsEndpoint;
-    return {
-      wsEndpoint
-    };
-  }
 }

package/lib/server/bidi/bidiInput.js

@@ -82,7 +82,7 @@ class RawMouseImpl {
     this._session = session;
   }
   async move(x, y, button, buttons, modifiers, forClick) {
-    // TODO: bidi throws when x/y are not integers.
+    // Bidi throws when x/y are not integers.
     x = Math.round(x);
     y = Math.round(y);
     await this._performActions([{
@@ -103,41 +103,25 @@ class RawMouseImpl {
       button: toBidiButton(button)
     }]);
   }
-  async click(x, y, options = {}) {
+  async wheel(x, y, buttons, modifiers, deltaX, deltaY) {
+    // Bidi throws when x/y are not integers.
     x = Math.round(x);
     y = Math.round(y);
-    const button = toBidiButton(options.button || 'left');
-    const {
-      delay = null,
-      clickCount = 1
-    } = options;
-    const actions = [];
-    actions.push({
-      type: 'pointerMove',
-      x,
-      y
+    await this._session.send('input.performActions', {
+      context: this._session.sessionId,
+      actions: [{
+        type: 'wheel',
+        id: 'pw_mouse_wheel',
+        actions: [{
+          type: 'scroll',
+          x,
+          y,
+          deltaX,
+          deltaY
+        }]
+      }]
     });
-    for (let cc = 1; cc <= clickCount; ++cc) {
-      actions.push({
-        type: 'pointerDown',
-        button
-      });
-      if (delay) actions.push({
-        type: 'pause',
-        duration: delay
-      });
-      actions.push({
-        type: 'pointerUp',
-        button
-      });
-      if (delay && cc < clickCount) actions.push({
-        type: 'pause',
-        duration: delay
-      });
-    }
-    await this._performActions(actions);
   }
-  async wheel(x, y, buttons, modifiers, deltaX, deltaY) {}
   async _performActions(actions) {
     await this._session.send('input.performActions', {
       context: this._session.sessionId,

package/lib/server/bidi/bidiNetworkManager.js

@@ -8,6 +8,7 @@ exports.bidiBytesValueToString = bidiBytesValueToString;
 var _eventsHelper = require("../../utils/eventsHelper");
 var network = _interopRequireWildcard(require("../network"));
 var bidi = _interopRequireWildcard(require("./third_party/bidiProtocol"));
+var _cookieStore = require("../cookieStore");
 function _getRequireWildcardCache(e) { if ("function" != typeof WeakMap) return null; var r = new WeakMap(), t = new WeakMap(); return (_getRequireWildcardCache = function (e) { return e ? t : r; })(e); }
 function _interopRequireWildcard(e, r) { if (!r && e && e.__esModule) return e; if (null === e || "object" != typeof e && "function" != typeof e) return { default: e }; var t = _getRequireWildcardCache(r); if (t && t.has(e)) return t.get(e); var n = { __proto__: null }, a = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var u in e) if ("default" !== u && Object.prototype.hasOwnProperty.call(e, u)) { var i = a ? Object.getOwnPropertyDescriptor(e, u) : null; i && (i.get || i.set) ? Object.defineProperty(n, u, i) : n[u] = e[u]; } return n.default = e, t && t.set(e, n), n; }
 /**
@@ -59,7 +60,7 @@ class BidiNetworkManager {
         var _redirectedFrom$_orig;
         this._session.sendMayFail('network.continueRequest', {
           request: param.request.request,
-          headers: (_redirectedFrom$_orig = redirectedFrom._originalRequestRoute) === null || _redirectedFrom$_orig === void 0 ? void 0 : _redirectedFrom$_orig._alreadyContinuedHeaders
+          ...(((_redirectedFrom$_orig = redirectedFrom._originalRequestRoute) === null || _redirectedFrom$_orig === void 0 ? void 0 : _redirectedFrom$_orig._alreadyContinuedHeaders) || {})
         });
       } else {
         route = new BidiRouteImpl(this._session, param.request.request);
@@ -242,13 +243,12 @@ class BidiRouteImpl {
         return header;
       });
     }
-    this._alreadyContinuedHeaders = toBidiHeaders(headers);
+    this._alreadyContinuedHeaders = headers;
     await this._session.sendMayFail('network.continueRequest', {
       request: this._requestId,
       url: overrides.url,
       method: overrides.method,
-      // TODO: cookies!
-      headers: this._alreadyContinuedHeaders,
+      ...toBidiRequestHeaders(this._alreadyContinuedHeaders),
       body: overrides.postData ? {
         type: 'base64',
         value: Buffer.from(overrides.postData).toString('base64')
@@ -261,7 +261,7 @@ class BidiRouteImpl {
       request: this._requestId,
       statusCode: response.status,
       reasonPhrase: network.statusText(response.status),
-      headers: toBidiHeaders(response.headers),
+      ...toBidiResponseHeaders(response.headers),
       body: {
         type: 'base64',
         value: base64body
@@ -285,6 +285,34 @@ function fromBidiHeaders(bidiHeaders) {
   });
   return result;
 }
+function toBidiRequestHeaders(allHeaders) {
+  const bidiHeaders = toBidiHeaders(allHeaders);
+  const cookies = bidiHeaders.filter(h => h.name.toLowerCase() === 'cookie');
+  const headers = bidiHeaders.filter(h => h.name.toLowerCase() !== 'cookie');
+  return {
+    cookies,
+    headers
+  };
+}
+function toBidiResponseHeaders(headers) {
+  const setCookieHeaders = headers.filter(h => h.name.toLowerCase() === 'set-cookie');
+  const otherHeaders = headers.filter(h => h.name.toLowerCase() !== 'set-cookie');
+  const rawCookies = setCookieHeaders.map(h => (0, _cookieStore.parseRawCookie)(h.value));
+  const cookies = rawCookies.filter(Boolean).map(c => {
+    return {
+      ...c,
+      value: {
+        type: 'string',
+        value: c.value
+      },
+      sameSite: toBidiSameSite(c.sameSite)
+    };
+  });
+  return {
+    cookies,
+    headers: toBidiHeaders(otherHeaders)
+  };
+}
 function toBidiHeaders(headers) {
   return headers.map(({
     name,
@@ -301,4 +329,10 @@ function bidiBytesValueToString(value) {
   if (value.type === 'string') return value.value;
   if (value.type === 'base64') return Buffer.from(value.type, 'base64').toString('binary');
   return 'unknown value type: ' + value.type;
+}
+function toBidiSameSite(sameSite) {
+  if (!sameSite) return undefined;
+  if (sameSite === 'Strict') return bidi.Network.SameSite.Strict;
+  if (sameSite === 'Lax') return bidi.Network.SameSite.Lax;
+  return bidi.Network.SameSite.None;
 }

package/lib/server/bidi/bidiOverCdp.js

@@ -0,0 +1,103 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.connectBidiOverCdp = connectBidiOverCdp;
+var bidiMapper = _interopRequireWildcard(require("chromium-bidi/lib/cjs/bidiMapper/BidiMapper"));
+var bidiCdpConnection = _interopRequireWildcard(require("chromium-bidi/lib/cjs/cdp/CdpConnection"));
+var _debugLogger = require("../../utils/debugLogger");
+function _getRequireWildcardCache(e) { if ("function" != typeof WeakMap) return null; var r = new WeakMap(), t = new WeakMap(); return (_getRequireWildcardCache = function (e) { return e ? t : r; })(e); }
+function _interopRequireWildcard(e, r) { if (!r && e && e.__esModule) return e; if (null === e || "object" != typeof e && "function" != typeof e) return { default: e }; var t = _getRequireWildcardCache(r); if (t && t.has(e)) return t.get(e); var n = { __proto__: null }, a = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var u in e) if ("default" !== u && Object.prototype.hasOwnProperty.call(e, u)) { var i = a ? Object.getOwnPropertyDescriptor(e, u) : null; i && (i.get || i.set) ? Object.defineProperty(n, u, i) : n[u] = e[u]; } return n.default = e, t && t.set(e, n), n; }
+/**
+ * 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.
+ */
+
+const bidiServerLogger = (prefix, ...args) => {
+  _debugLogger.debugLogger.log(prefix, args);
+};
+async function connectBidiOverCdp(cdp) {
+  let server = undefined;
+  const bidiTransport = new BidiTransportImpl();
+  const bidiConnection = new BidiConnection(bidiTransport, () => {
+    var _server;
+    return (_server = server) === null || _server === void 0 ? void 0 : _server.close();
+  });
+  const cdpTransportImpl = new CdpTransportImpl(cdp);
+  const cdpConnection = new bidiCdpConnection.MapperCdpConnection(cdpTransportImpl, bidiServerLogger);
+  // Make sure onclose event is propagated.
+  cdp.onclose = () => {
+    var _bidiConnection$onclo;
+    return (_bidiConnection$onclo = bidiConnection.onclose) === null || _bidiConnection$onclo === void 0 ? void 0 : _bidiConnection$onclo.call(bidiConnection);
+  };
+  server = await bidiMapper.BidiServer.createAndStart(bidiTransport, cdpConnection, await cdpConnection.createBrowserSession(), /* selfTargetId= */'', undefined, bidiServerLogger);
+  return bidiConnection;
+}
+class BidiTransportImpl {
+  constructor() {
+    this._handler = void 0;
+    this._bidiConnection = void 0;
+  }
+  setOnMessage(handler) {
+    this._handler = handler;
+  }
+  sendMessage(message) {
+    var _this$_bidiConnection, _this$_bidiConnection2;
+    return (_this$_bidiConnection = (_this$_bidiConnection2 = this._bidiConnection).onmessage) === null || _this$_bidiConnection === void 0 ? void 0 : _this$_bidiConnection.call(_this$_bidiConnection2, message);
+  }
+  close() {
+    var _this$_bidiConnection3, _this$_bidiConnection4;
+    (_this$_bidiConnection3 = (_this$_bidiConnection4 = this._bidiConnection).onclose) === null || _this$_bidiConnection3 === void 0 || _this$_bidiConnection3.call(_this$_bidiConnection4);
+  }
+}
+class BidiConnection {
+  constructor(bidiTransport, closeCallback) {
+    this._bidiTransport = void 0;
+    this._closeCallback = void 0;
+    this.onmessage = void 0;
+    this.onclose = void 0;
+    this._bidiTransport = bidiTransport;
+    this._bidiTransport._bidiConnection = this;
+    this._closeCallback = closeCallback;
+  }
+  send(s) {
+    var _this$_bidiTransport$, _this$_bidiTransport;
+    (_this$_bidiTransport$ = (_this$_bidiTransport = this._bidiTransport)._handler) === null || _this$_bidiTransport$ === void 0 || _this$_bidiTransport$.call(_this$_bidiTransport, s);
+  }
+  close() {
+    this._closeCallback();
+  }
+}
+class CdpTransportImpl {
+  constructor(connection) {
+    this._connection = void 0;
+    this._handler = void 0;
+    this._bidiConnection = void 0;
+    this._connection = connection;
+    this._connection.onmessage = message => {
+      var _this$_handler;
+      (_this$_handler = this._handler) === null || _this$_handler === void 0 || _this$_handler.call(this, JSON.stringify(message));
+    };
+  }
+  setOnMessage(handler) {
+    this._handler = handler;
+  }
+  sendMessage(message) {
+    return this._connection.send(JSON.parse(message));
+  }
+  close() {
+    this._connection.close();
+  }
+}

package/lib/server/bidi/bidiPage.js

@@ -14,6 +14,7 @@ var bidi = _interopRequireWildcard(require("./third_party/bidiProtocol"));
 var _bidiExecutionContext = require("./bidiExecutionContext");
 var _bidiNetworkManager = require("./bidiNetworkManager");
 var _browserContext = require("../browserContext");
+var _bidiPdf = require("./bidiPdf");
 function _getRequireWildcardCache(e) { if ("function" != typeof WeakMap) return null; var r = new WeakMap(), t = new WeakMap(); return (_getRequireWildcardCache = function (e) { return e ? t : r; })(e); }
 function _interopRequireWildcard(e, r) { if (!r && e && e.__esModule) return e; if (null === e || "object" != typeof e && "function" != typeof e) return { default: e }; var t = _getRequireWildcardCache(r); if (t && t.has(e)) return t.get(e); var n = { __proto__: null }, a = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var u in e) if ("default" !== u && Object.prototype.hasOwnProperty.call(e, u)) { var i = a ? Object.getOwnPropertyDescriptor(e, u) : null; i && (i.get || i.set) ? Object.defineProperty(n, u, i) : n[u] = e[u]; } return n.default = e, t && t.set(e, n), n; }
 /**
@@ -33,6 +34,7 @@ function _interopRequireWildcard(e, r) { if (!r && e && e.__esModule) return e;
  */
 
 const UTILITY_WORLD_NAME = '__playwright_utility_world__';
+const kPlaywrightBindingChannel = 'playwrightChannel';
 class BidiPage {
   constructor(browserContext, bidiSession, opener) {
     this.rawMouse = void 0;
@@ -46,7 +48,9 @@ class BidiPage {
     this._sessionListeners = [];
     this._browserContext = void 0;
     this._networkManager = void 0;
+    this._pdf = void 0;
     this._initializedPage = null;
+    this._initScriptIds = [];
     this._session = bidiSession;
     this._opener = opener;
     this.rawKeyboard = new _bidiInput.RawKeyboardImpl(bidiSession);
@@ -56,8 +60,9 @@ class BidiPage {
     this._page = new _page.Page(this, browserContext);
     this._browserContext = browserContext;
     this._networkManager = new _bidiNetworkManager.BidiNetworkManager(this._session, this._page, this._onNavigationResponseStarted.bind(this));
+    this._pdf = new _bidiPdf.BidiPDF(this._session);
     this._page.on(_page.Page.Events.FrameDetached, frame => this._removeContextsForFrame(frame, false));
-    this._sessionListeners = [_eventsHelper.eventsHelper.addEventListener(bidiSession, 'script.realmCreated', this._onRealmCreated.bind(this)), _eventsHelper.eventsHelper.addEventListener(bidiSession, 'browsingContext.contextDestroyed', this._onBrowsingContextDestroyed.bind(this)), _eventsHelper.eventsHelper.addEventListener(bidiSession, 'browsingContext.navigationStarted', this._onNavigationStarted.bind(this)), _eventsHelper.eventsHelper.addEventListener(bidiSession, 'browsingContext.navigationAborted', this._onNavigationAborted.bind(this)), _eventsHelper.eventsHelper.addEventListener(bidiSession, 'browsingContext.navigationFailed', this._onNavigationFailed.bind(this)), _eventsHelper.eventsHelper.addEventListener(bidiSession, 'browsingContext.fragmentNavigated', this._onFragmentNavigated.bind(this)), _eventsHelper.eventsHelper.addEventListener(bidiSession, 'browsingContext.domContentLoaded', this._onDomContentLoaded.bind(this)), _eventsHelper.eventsHelper.addEventListener(bidiSession, 'browsingContext.load', this._onLoad.bind(this)), _eventsHelper.eventsHelper.addEventListener(bidiSession, 'browsingContext.userPromptOpened', this._onUserPromptOpened.bind(this)), _eventsHelper.eventsHelper.addEventListener(bidiSession, 'log.entryAdded', this._onLogEntryAdded.bind(this))];
+    this._sessionListeners = [_eventsHelper.eventsHelper.addEventListener(bidiSession, 'script.realmCreated', this._onRealmCreated.bind(this)), _eventsHelper.eventsHelper.addEventListener(bidiSession, 'script.message', this._onScriptMessage.bind(this)), _eventsHelper.eventsHelper.addEventListener(bidiSession, 'browsingContext.contextDestroyed', this._onBrowsingContextDestroyed.bind(this)), _eventsHelper.eventsHelper.addEventListener(bidiSession, 'browsingContext.navigationStarted', this._onNavigationStarted.bind(this)), _eventsHelper.eventsHelper.addEventListener(bidiSession, 'browsingContext.navigationAborted', this._onNavigationAborted.bind(this)), _eventsHelper.eventsHelper.addEventListener(bidiSession, 'browsingContext.navigationFailed', this._onNavigationFailed.bind(this)), _eventsHelper.eventsHelper.addEventListener(bidiSession, 'browsingContext.fragmentNavigated', this._onFragmentNavigated.bind(this)), _eventsHelper.eventsHelper.addEventListener(bidiSession, 'browsingContext.domContentLoaded', this._onDomContentLoaded.bind(this)), _eventsHelper.eventsHelper.addEventListener(bidiSession, 'browsingContext.load', this._onLoad.bind(this)), _eventsHelper.eventsHelper.addEventListener(bidiSession, 'browsingContext.userPromptOpened', this._onUserPromptOpened.bind(this)), _eventsHelper.eventsHelper.addEventListener(bidiSession, 'log.entryAdded', this._onLogEntryAdded.bind(this))];
 
     // Initialize main frame.
     this._pagePromise = this._initialize().finally(async () => {
@@ -72,18 +77,12 @@ class BidiPage {
     });
   }
   async _initialize() {
-    const {
-      contexts
-    } = await this._session.send('browsingContext.getTree', {
-      root: this._session.sessionId
-    });
-    this._handleFrameTree(contexts[0]);
-    await Promise.all([this.updateHttpCredentials(), this.updateRequestInterception(), this._updateViewport()]);
+    // Initialize main frame.
+    this._onFrameAttached(this._session.sessionId, null);
+    await Promise.all([this.updateHttpCredentials(), this.updateRequestInterception(), this._updateViewport(), this._installMainBinding(), this._addAllInitScripts()]);
   }
-  _handleFrameTree(frameTree) {
-    this._onFrameAttached(frameTree.context, frameTree.parent || null);
-    if (!frameTree.children) return;
-    for (const child of frameTree.children) this._handleFrameTree(child);
+  async _addAllInitScripts() {
+    return Promise.all(this._page.allInitScripts().map(initScript => this.addInitScript(initScript)));
   }
   potentiallyUninitializedPage() {
     return this._page;
@@ -236,7 +235,11 @@ class BidiPage {
     await this._updateViewport();
   }
   async updateUserAgent() {}
-  async bringToFront() {}
+  async bringToFront() {
+    await this._session.send('browsingContext.activate', {
+      context: this._session.sessionId
+    });
+  }
   async _updateViewport() {
     const options = this._browserContext._options;
     const deviceSize = this._page.emulatedSize();
@@ -266,20 +269,72 @@ class BidiPage {
       wait: bidi.BrowsingContext.ReadinessState.Interactive
     });
   }
-  goBack() {
-    throw new Error('Method not implemented.');
+  async goBack() {
+    return await this._session.send('browsingContext.traverseHistory', {
+      context: this._session.sessionId,
+      delta: -1
+    }).then(() => true).catch(() => false);
+  }
+  async goForward() {
+    return await this._session.send('browsingContext.traverseHistory', {
+      context: this._session.sessionId,
+      delta: +1
+    }).then(() => true).catch(() => false);
   }
-  goForward() {
+  async requestGC() {
     throw new Error('Method not implemented.');
   }
+
+  // TODO: consider calling this only when bindings are added.
+  async _installMainBinding() {
+    const functionDeclaration = addMainBinding.toString();
+    const args = [{
+      type: 'channel',
+      value: {
+        channel: kPlaywrightBindingChannel,
+        ownership: bidi.Script.ResultOwnership.Root
+      }
+    }];
+    const promises = [];
+    promises.push(this._session.send('script.addPreloadScript', {
+      functionDeclaration,
+      arguments: args
+    }));
+    promises.push(this._session.send('script.callFunction', {
+      functionDeclaration,
+      arguments: args,
+      target: toBidiExecutionContext(await this._page.mainFrame()._mainContext())._target,
+      awaitPromise: false,
+      userActivation: false
+    }));
+    await Promise.all(promises);
+  }
+  async _onScriptMessage(event) {
+    if (event.channel !== kPlaywrightBindingChannel) return;
+    const pageOrError = await this.pageOrError();
+    if (pageOrError instanceof Error) return;
+    const context = this._realmToContext.get(event.source.realm);
+    if (!context) return;
+    if (event.data.type !== 'string') return;
+    await this._page._onBindingCalled(event.data.value, context);
+  }
   async addInitScript(initScript) {
-    await this._updateBootstrapScript();
+    const {
+      script
+    } = await this._session.send('script.addPreloadScript', {
+      // TODO: remove function call from the source.
+      functionDeclaration: `() => { return ${initScript.source} }`,
+      // TODO: push to iframes?
+      contexts: [this._session.sessionId]
+    });
+    if (!initScript.internal) this._initScriptIds.push(script);
   }
   async removeNonInternalInitScripts() {
-    await this._updateBootstrapScript();
-  }
-  async _updateBootstrapScript() {
-    throw new Error('Method not implemented.');
+    const promises = this._initScriptIds.map(script => this._session.send('script.removePreloadScript', {
+      script
+    }));
+    this._initScriptIds = [];
+    await Promise.all(promises);
   }
   async closePage(runBeforeUnload) {
     await this._session.send('browsingContext.close', {
@@ -289,7 +344,22 @@ class BidiPage {
   }
   async setBackgroundColor(color) {}
   async takeScreenshot(progress, format, documentRect, viewportRect, quality, fitsViewport, scale) {
-    throw new Error('Method not implemented.');
+    const rect = documentRect || viewportRect;
+    const {
+      data
+    } = await this._session.send('browsingContext.captureScreenshot', {
+      context: this._session.sessionId,
+      format: {
+        type: `image/${format === 'png' ? 'png' : 'jpeg'}`,
+        quality: quality || 80
+      },
+      origin: documentRect ? 'document' : 'viewport',
+      clip: {
+        type: 'box',
+        ...rect
+      }
+    });
+    return Buffer.from(data, 'base64');
   }
   async getContentFrame(handle) {
     const executionContext = toBidiExecutionContext(handle._context);
@@ -419,6 +489,9 @@ class BidiPage {
   }
   async inputActionEpilogue() {}
   async resetForReuse() {}
+  async pdf(options) {
+    return this._pdf.generate(options);
+  }
   async getFrameElement(frame) {
     const parent = frame.parentFrame();
     if (!parent) throw new Error('Frame has been detached.');
@@ -445,11 +518,11 @@ class BidiPage {
   shouldToggleStyleSheetToSyncAnimations() {
     return true;
   }
-  useMainWorldForSetContent() {
-    return true;
-  }
 }
 exports.BidiPage = BidiPage;
+function addMainBinding(callback) {
+  globalThis['__playwright__binding__'] = callback;
+}
 function toBidiExecutionContext(executionContext) {
   return executionContext[contextDelegateSymbol];
 }

package/lib/server/bidi/bidiPdf.js

@@ -0,0 +1,140 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.BidiPDF = void 0;
+var _utils = require("../../utils");
+/**
+ * Copyright 2017 Google Inc. All rights reserved.
+ * Modifications 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.
+ */
+
+const PagePaperFormats = {
+  letter: {
+    width: 8.5,
+    height: 11
+  },
+  legal: {
+    width: 8.5,
+    height: 14
+  },
+  tabloid: {
+    width: 11,
+    height: 17
+  },
+  ledger: {
+    width: 17,
+    height: 11
+  },
+  a0: {
+    width: 33.1,
+    height: 46.8
+  },
+  a1: {
+    width: 23.4,
+    height: 33.1
+  },
+  a2: {
+    width: 16.54,
+    height: 23.4
+  },
+  a3: {
+    width: 11.7,
+    height: 16.54
+  },
+  a4: {
+    width: 8.27,
+    height: 11.7
+  },
+  a5: {
+    width: 5.83,
+    height: 8.27
+  },
+  a6: {
+    width: 4.13,
+    height: 5.83
+  }
+};
+const unitToPixels = {
+  'px': 1,
+  'in': 96,
+  'cm': 37.8,
+  'mm': 3.78
+};
+function convertPrintParameterToInches(text) {
+  if (text === undefined) return undefined;
+  let unit = text.substring(text.length - 2).toLowerCase();
+  let valueText = '';
+  if (unitToPixels.hasOwnProperty(unit)) {
+    valueText = text.substring(0, text.length - 2);
+  } else {
+    // In case of unknown unit try to parse the whole parameter as number of pixels.
+    // This is consistent with phantom's paperSize behavior.
+    unit = 'px';
+    valueText = text;
+  }
+  const value = Number(valueText);
+  (0, _utils.assert)(!isNaN(value), 'Failed to parse parameter value: ' + text);
+  const pixels = value * unitToPixels[unit];
+  return pixels / 96;
+}
+class BidiPDF {
+  constructor(session) {
+    this._session = void 0;
+    this._session = session;
+  }
+  async generate(options) {
+    const {
+      scale = 1,
+      printBackground = false,
+      landscape = false,
+      pageRanges = '',
+      margin = {}
+    } = options;
+    let paperWidth = 8.5;
+    let paperHeight = 11;
+    if (options.format) {
+      const format = PagePaperFormats[options.format.toLowerCase()];
+      (0, _utils.assert)(format, 'Unknown paper format: ' + options.format);
+      paperWidth = format.width;
+      paperHeight = format.height;
+    } else {
+      paperWidth = convertPrintParameterToInches(options.width) || paperWidth;
+      paperHeight = convertPrintParameterToInches(options.height) || paperHeight;
+    }
+    const {
+      data
+    } = await this._session.send('browsingContext.print', {
+      context: this._session.sessionId,
+      background: printBackground,
+      margin: {
+        bottom: convertPrintParameterToInches(margin.bottom) || 0,
+        left: convertPrintParameterToInches(margin.left) || 0,
+        right: convertPrintParameterToInches(margin.right) || 0,
+        top: convertPrintParameterToInches(margin.top) || 0
+      },
+      orientation: landscape ? 'landscape' : 'portrait',
+      page: {
+        width: paperWidth,
+        height: paperHeight
+      },
+      pageRanges: pageRanges ? pageRanges.split(',').map(r => r.trim()) : undefined,
+      scale
+    });
+    return Buffer.from(data, 'base64');
+  }
+}
+exports.BidiPDF = BidiPDF;