@stablyai/internal-playwright-core 0.1.0

package/lib/server/bidi/bidiConnection.js

@@ -0,0 +1,212 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var bidiConnection_exports = {};
+__export(bidiConnection_exports, {
+  BidiConnection: () => BidiConnection,
+  BidiSession: () => BidiSession,
+  kBrowserCloseMessageId: () => kBrowserCloseMessageId,
+  kShutdownSessionNewMessageId: () => kShutdownSessionNewMessageId
+});
+module.exports = __toCommonJS(bidiConnection_exports);
+var import_events = require("events");
+var import_debugLogger = require("../utils/debugLogger");
+var import_helper = require("../helper");
+var import_protocolError = require("../protocolError");
+const kBrowserCloseMessageId = Number.MAX_SAFE_INTEGER - 1;
+const kShutdownSessionNewMessageId = kBrowserCloseMessageId - 1;
+class BidiConnection {
+  constructor(transport, onDisconnect, protocolLogger, browserLogsCollector) {
+    this._lastId = 0;
+    this._closed = false;
+    this._browsingContextToSession = /* @__PURE__ */ new Map();
+    this._realmToBrowsingContext = /* @__PURE__ */ new Map();
+    // TODO: shared/service workers might have multiple owner realms.
+    this._realmToOwnerRealm = /* @__PURE__ */ new Map();
+    this._transport = transport;
+    this._onDisconnect = onDisconnect;
+    this._protocolLogger = protocolLogger;
+    this._browserLogsCollector = browserLogsCollector;
+    this.browserSession = new BidiSession(this, "", (message) => {
+      this.rawSend(message);
+    });
+    this._transport.onmessage = this._dispatchMessage.bind(this);
+    this._transport.onclose = this._onClose.bind(this);
+  }
+  nextMessageId() {
+    return ++this._lastId;
+  }
+  rawSend(message) {
+    this._protocolLogger("send", message);
+    this._transport.send(message);
+  }
+  _dispatchMessage(message) {
+    this._protocolLogger("receive", message);
+    const object = message;
+    if (object.type === "event") {
+      if (object.method === "script.realmCreated") {
+        if ("context" in object.params)
+          this._realmToBrowsingContext.set(object.params.realm, object.params.context);
+        if (object.params.type === "dedicated-worker")
+          this._realmToOwnerRealm.set(object.params.realm, object.params.owners[0]);
+      } else if (object.method === "script.realmDestroyed") {
+        this._realmToBrowsingContext.delete(object.params.realm);
+        this._realmToOwnerRealm.delete(object.params.realm);
+      }
+      let context;
+      let realm;
+      if ("context" in object.params) {
+        context = object.params.context;
+      } else if (object.method === "log.entryAdded" || object.method === "script.message") {
+        context = object.params.source?.context;
+        realm = object.params.source?.realm;
+      } else if (object.method === "script.realmCreated" && object.params.type === "dedicated-worker") {
+        realm = object.params.owners[0];
+      }
+      if (!context && realm) {
+        while (this._realmToOwnerRealm.get(realm))
+          realm = this._realmToOwnerRealm.get(realm);
+        context = this._realmToBrowsingContext.get(realm);
+      }
+      if (context) {
+        const session = this._browsingContextToSession.get(context);
+        if (session) {
+          session.dispatchMessage(message);
+          return;
+        }
+      }
+    } else if (message.id) {
+      for (const session of this._browsingContextToSession.values()) {
+        if (session.hasCallback(message.id)) {
+          session.dispatchMessage(message);
+          return;
+        }
+      }
+    }
+    this.browserSession.dispatchMessage(message);
+  }
+  _onClose(reason) {
+    this._closed = true;
+    this._transport.onmessage = void 0;
+    this._transport.onclose = void 0;
+    this._browserDisconnectedLogs = import_helper.helper.formatBrowserLogs(this._browserLogsCollector.recentLogs(), reason);
+    this.browserSession.dispose();
+    this._onDisconnect();
+  }
+  isClosed() {
+    return this._closed;
+  }
+  close() {
+    if (!this._closed)
+      this._transport.close();
+  }
+  createMainFrameBrowsingContextSession(bowsingContextId) {
+    const result = new BidiSession(this, bowsingContextId, (message) => this.rawSend(message));
+    this._browsingContextToSession.set(bowsingContextId, result);
+    return result;
+  }
+}
+class BidiSession extends import_events.EventEmitter {
+  constructor(connection, sessionId, rawSend) {
+    super();
+    this._disposed = false;
+    this._callbacks = /* @__PURE__ */ new Map();
+    this._crashed = false;
+    this._browsingContexts = /* @__PURE__ */ new Set();
+    this.setMaxListeners(0);
+    this.connection = connection;
+    this.sessionId = sessionId;
+    this._rawSend = rawSend;
+    this.on = super.on;
+    this.off = super.removeListener;
+    this.addListener = super.addListener;
+    this.removeListener = super.removeListener;
+    this.once = super.once;
+  }
+  addFrameBrowsingContext(context) {
+    this._browsingContexts.add(context);
+    this.connection._browsingContextToSession.set(context, this);
+  }
+  removeFrameBrowsingContext(context) {
+    this._browsingContexts.delete(context);
+    this.connection._browsingContextToSession.delete(context);
+  }
+  async send(method, params) {
+    if (this._crashed || this._disposed || this.connection._browserDisconnectedLogs)
+      throw new import_protocolError.ProtocolError(this._crashed ? "crashed" : "closed", void 0, this.connection._browserDisconnectedLogs);
+    const id = this.connection.nextMessageId();
+    const messageObj = { id, method, params };
+    this._rawSend(messageObj);
+    return new Promise((resolve, reject) => {
+      this._callbacks.set(id, { resolve, reject, error: new import_protocolError.ProtocolError("error", method) });
+    });
+  }
+  sendMayFail(method, params) {
+    return this.send(method, params).catch((error) => import_debugLogger.debugLogger.log("error", error));
+  }
+  markAsCrashed() {
+    this._crashed = true;
+  }
+  isDisposed() {
+    return this._disposed;
+  }
+  dispose() {
+    this._disposed = true;
+    this.connection._browsingContextToSession.delete(this.sessionId);
+    for (const context of this._browsingContexts)
+      this.connection._browsingContextToSession.delete(context);
+    this._browsingContexts.clear();
+    for (const callback of this._callbacks.values()) {
+      callback.error.type = this._crashed ? "crashed" : "closed";
+      callback.error.logs = this.connection._browserDisconnectedLogs;
+      callback.reject(callback.error);
+    }
+    this._callbacks.clear();
+  }
+  hasCallback(id) {
+    return this._callbacks.has(id);
+  }
+  dispatchMessage(message) {
+    const object = message;
+    if (object.id === kBrowserCloseMessageId || object.id === kShutdownSessionNewMessageId)
+      return;
+    if (object.id && this._callbacks.has(object.id)) {
+      const callback = this._callbacks.get(object.id);
+      this._callbacks.delete(object.id);
+      if (object.type === "error") {
+        callback.error.setMessage(object.error + "\nMessage: " + object.message);
+        callback.reject(callback.error);
+      } else if (object.type === "success") {
+        callback.resolve(object.result);
+      } else {
+        callback.error.setMessage("Internal error, unexpected response type: " + JSON.stringify(object));
+        callback.reject(callback.error);
+      }
+    } else if (object.id) {
+    } else {
+      Promise.resolve().then(() => this.emit(object.method, object.params));
+    }
+  }
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  BidiConnection,
+  BidiSession,
+  kBrowserCloseMessageId,
+  kShutdownSessionNewMessageId
+});

package/lib/server/bidi/bidiExecutionContext.js

@@ -0,0 +1,221 @@
+"use strict";
+var __create = Object.create;
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getProtoOf = Object.getPrototypeOf;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
+  // If the importer is in node compatibility mode or this is not an ESM
+  // file that has been converted to a CommonJS file using a Babel-
+  // compatible transform (i.e. "__esModule" has not been set), then set
+  // "default" to the CommonJS "module.exports" for node compatibility.
+  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
+  mod
+));
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var bidiExecutionContext_exports = {};
+__export(bidiExecutionContext_exports, {
+  BidiExecutionContext: () => BidiExecutionContext,
+  createHandle: () => createHandle
+});
+module.exports = __toCommonJS(bidiExecutionContext_exports);
+var import_utils = require("../../utils");
+var import_utilityScriptSerializers = require("../../utils/isomorphic/utilityScriptSerializers");
+var js = __toESM(require("../javascript"));
+var dom = __toESM(require("../dom"));
+var import_bidiDeserializer = require("./third_party/bidiDeserializer");
+var bidi = __toESM(require("./third_party/bidiProtocol"));
+var import_bidiSerializer = require("./third_party/bidiSerializer");
+class BidiExecutionContext {
+  constructor(session, realmInfo) {
+    this._session = session;
+    if (realmInfo.type === "window") {
+      this._target = {
+        context: realmInfo.context,
+        sandbox: realmInfo.sandbox
+      };
+    } else {
+      this._target = {
+        realm: realmInfo.realm
+      };
+    }
+  }
+  async rawEvaluateJSON(expression) {
+    const response = await this._session.send("script.evaluate", {
+      expression,
+      target: this._target,
+      serializationOptions: {
+        maxObjectDepth: 10,
+        maxDomDepth: 10
+      },
+      awaitPromise: true,
+      userActivation: true
+    });
+    if (response.type === "success")
+      return import_bidiDeserializer.BidiDeserializer.deserialize(response.result);
+    if (response.type === "exception")
+      throw new js.JavaScriptErrorInEvaluate(response.exceptionDetails.text);
+    throw new js.JavaScriptErrorInEvaluate("Unexpected response type: " + JSON.stringify(response));
+  }
+  async rawEvaluateHandle(context, expression) {
+    const response = await this._session.send("script.evaluate", {
+      expression,
+      target: this._target,
+      resultOwnership: bidi.Script.ResultOwnership.Root,
+      // Necessary for the handle to be returned.
+      serializationOptions: { maxObjectDepth: 0, maxDomDepth: 0 },
+      awaitPromise: true,
+      userActivation: true
+    });
+    if (response.type === "success") {
+      if ("handle" in response.result)
+        return createHandle(context, response.result);
+      throw new js.JavaScriptErrorInEvaluate("Cannot get handle: " + JSON.stringify(response.result));
+    }
+    if (response.type === "exception")
+      throw new js.JavaScriptErrorInEvaluate(response.exceptionDetails.text);
+    throw new js.JavaScriptErrorInEvaluate("Unexpected response type: " + JSON.stringify(response));
+  }
+  async evaluateWithArguments(functionDeclaration, returnByValue, utilityScript, values, handles) {
+    const response = await this._session.send("script.callFunction", {
+      functionDeclaration,
+      target: this._target,
+      arguments: [
+        { handle: utilityScript._objectId },
+        ...values.map(import_bidiSerializer.BidiSerializer.serialize),
+        ...handles.map((handle) => ({ handle: handle._objectId }))
+      ],
+      resultOwnership: returnByValue ? void 0 : bidi.Script.ResultOwnership.Root,
+      // Necessary for the handle to be returned.
+      serializationOptions: returnByValue ? {} : { maxObjectDepth: 0, maxDomDepth: 0 },
+      awaitPromise: true,
+      userActivation: true
+    });
+    if (response.type === "exception")
+      throw new js.JavaScriptErrorInEvaluate(response.exceptionDetails.text);
+    if (response.type === "success") {
+      if (returnByValue)
+        return (0, import_utilityScriptSerializers.parseEvaluationResultValue)(import_bidiDeserializer.BidiDeserializer.deserialize(response.result));
+      return createHandle(utilityScript._context, response.result);
+    }
+    throw new js.JavaScriptErrorInEvaluate("Unexpected response type: " + JSON.stringify(response));
+  }
+  async getProperties(handle) {
+    const names = await handle.evaluate((object) => {
+      const names2 = [];
+      const descriptors = Object.getOwnPropertyDescriptors(object);
+      for (const name in descriptors) {
+        if (descriptors[name]?.enumerable)
+          names2.push(name);
+      }
+      return names2;
+    });
+    const values = await Promise.all(names.map((name) => handle.evaluateHandle((object, name2) => object[name2], name)));
+    const map = /* @__PURE__ */ new Map();
+    for (let i = 0; i < names.length; i++)
+      map.set(names[i], values[i]);
+    return map;
+  }
+  async releaseHandle(handle) {
+    if (!handle._objectId)
+      return;
+    await this._session.send("script.disown", {
+      target: this._target,
+      handles: [handle._objectId]
+    });
+  }
+  async nodeIdForElementHandle(handle) {
+    const shared = await this._remoteValueForReference({ handle: handle._objectId });
+    if (!("sharedId" in shared))
+      throw new Error("Element is not a node");
+    return {
+      sharedId: shared.sharedId
+    };
+  }
+  async remoteObjectForNodeId(context, nodeId) {
+    const result = await this._remoteValueForReference(nodeId, true);
+    if (!("handle" in result))
+      throw new Error("Can't get remote object for nodeId");
+    return createHandle(context, result);
+  }
+  async contentFrameIdForFrame(handle) {
+    const contentWindow = await this._rawCallFunction("e => e.contentWindow", { handle: handle._objectId });
+    if (contentWindow?.type === "window")
+      return contentWindow.value.context;
+    return null;
+  }
+  async frameIdForWindowHandle(handle) {
+    if (!handle._objectId)
+      throw new Error("JSHandle is not a DOM node handle");
+    const contentWindow = await this._remoteValueForReference({ handle: handle._objectId });
+    if (contentWindow.type === "window")
+      return contentWindow.value.context;
+    return null;
+  }
+  async _remoteValueForReference(reference, createHandle2) {
+    return await this._rawCallFunction("e => e", reference, createHandle2);
+  }
+  async _rawCallFunction(functionDeclaration, arg, createHandle2) {
+    const response = await this._session.send("script.callFunction", {
+      functionDeclaration,
+      target: this._target,
+      arguments: [arg],
+      // "Root" is necessary for the handle to be returned.
+      resultOwnership: createHandle2 ? bidi.Script.ResultOwnership.Root : bidi.Script.ResultOwnership.None,
+      serializationOptions: { maxObjectDepth: 0, maxDomDepth: 0 },
+      awaitPromise: true,
+      userActivation: true
+    });
+    if (response.type === "exception")
+      throw new js.JavaScriptErrorInEvaluate(response.exceptionDetails.text);
+    if (response.type === "success")
+      return response.result;
+    throw new js.JavaScriptErrorInEvaluate("Unexpected response type: " + JSON.stringify(response));
+  }
+}
+function renderPreview(remoteObject) {
+  if (remoteObject.type === "undefined")
+    return "undefined";
+  if (remoteObject.type === "null")
+    return "null";
+  if ("value" in remoteObject)
+    return String(remoteObject.value);
+  return `<${remoteObject.type}>`;
+}
+function remoteObjectValue(remoteObject) {
+  if (remoteObject.type === "undefined")
+    return void 0;
+  if (remoteObject.type === "null")
+    return null;
+  if (remoteObject.type === "number" && typeof remoteObject.value === "string")
+    return js.parseUnserializableValue(remoteObject.value);
+  if ("value" in remoteObject)
+    return remoteObject.value;
+  return void 0;
+}
+function createHandle(context, remoteObject) {
+  if (remoteObject.type === "node") {
+    (0, import_utils.assert)(context instanceof dom.FrameExecutionContext);
+    return new dom.ElementHandle(context, remoteObject.handle);
+  }
+  const objectId = "handle" in remoteObject ? remoteObject.handle : void 0;
+  return new js.JSHandle(context, remoteObject.type, renderPreview(remoteObject), objectId, remoteObjectValue(remoteObject));
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  BidiExecutionContext,
+  createHandle
+});

package/lib/server/bidi/bidiFirefox.js

@@ -0,0 +1,130 @@
+"use strict";
+var __create = Object.create;
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getProtoOf = Object.getPrototypeOf;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
+  // If the importer is in node compatibility mode or this is not an ESM
+  // file that has been converted to a CommonJS file using a Babel-
+  // compatible transform (i.e. "__esModule" has not been set), then set
+  // "default" to the CommonJS "module.exports" for node compatibility.
+  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
+  mod
+));
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var bidiFirefox_exports = {};
+__export(bidiFirefox_exports, {
+  BidiFirefox: () => BidiFirefox
+});
+module.exports = __toCommonJS(bidiFirefox_exports);
+var import_os = __toESM(require("os"));
+var import_path = __toESM(require("path"));
+var import_ascii = require("../utils/ascii");
+var import_browserType = require("../browserType");
+var import_bidiBrowser = require("./bidiBrowser");
+var import_bidiConnection = require("./bidiConnection");
+var import_firefoxPrefs = require("./third_party/firefoxPrefs");
+var import_manualPromise = require("../../utils/isomorphic/manualPromise");
+class BidiFirefox extends import_browserType.BrowserType {
+  constructor(parent) {
+    super(parent, "firefox");
+  }
+  executablePath() {
+    return "";
+  }
+  async connectToTransport(transport, options) {
+    return import_bidiBrowser.BidiBrowser.connect(this.attribution.playwright, transport, options);
+  }
+  doRewriteStartupLog(error) {
+    if (!error.logs)
+      return error;
+    if (error.logs.includes(`as root in a regular user's session is not supported.`))
+      error.logs = "\n" + (0, import_ascii.wrapInASCIIBox)(`Firefox is unable to launch if the $HOME folder isn't owned by the current user.
+Workaround: Set the HOME=/root environment variable${process.env.GITHUB_ACTION ? " in your GitHub Actions workflow file" : ""} when running Playwright.`, 1);
+    if (error.logs.includes("no DISPLAY environment variable specified"))
+      error.logs = "\n" + (0, import_ascii.wrapInASCIIBox)(import_browserType.kNoXServerRunningError, 1);
+    return error;
+  }
+  amendEnvironment(env) {
+    if (!import_path.default.isAbsolute(import_os.default.homedir()))
+      throw new Error(`Cannot launch Firefox with relative home directory. Did you set ${import_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 (import_os.default.platform() === "linux") {
+      return { ...env, SNAP_NAME: void 0, SNAP_INSTANCE_NAME: void 0 };
+    }
+    return env;
+  }
+  attemptToGracefullyCloseBrowser(transport) {
+    this._attemptToGracefullyCloseBrowser(transport).catch(() => {
+    });
+  }
+  async _attemptToGracefullyCloseBrowser(transport) {
+    if (!transport.onmessage) {
+      transport.send({ method: "session.new", params: { capabilities: {} }, id: import_bidiConnection.kShutdownSessionNewMessageId });
+      await new Promise((resolve) => {
+        transport.onmessage = (message) => {
+          if (message.id === import_bidiConnection.kShutdownSessionNewMessageId)
+            resolve(true);
+        };
+      });
+    }
+    transport.send({ method: "browser.close", params: {}, id: import_bidiConnection.kBrowserCloseMessageId });
+  }
+  supportsPipeTransport() {
+    return false;
+  }
+  async prepareUserDataDir(options, userDataDir) {
+    await (0, import_firefoxPrefs.createProfile)({
+      path: userDataDir,
+      preferences: options.firefoxUserPrefs || {}
+    });
+  }
+  async defaultArgs(options, isPersistent, userDataDir) {
+    const { args = [], headless } = options;
+    const userDataDirArg = args.find((arg) => arg.startsWith("-profile") || arg.startsWith("--profile"));
+    if (userDataDirArg)
+      throw this._createUserDataDirArgMisuseError("--profile");
+    if (args.find((arg) => !arg.startsWith("-")))
+      throw new Error("Arguments can not specify page to be opened");
+    const firefoxArguments = ["--remote-debugging-port=0"];
+    if (headless)
+      firefoxArguments.push("--headless");
+    else
+      firefoxArguments.push("--foreground");
+    firefoxArguments.push(`--profile`, userDataDir);
+    firefoxArguments.push(...args);
+    return firefoxArguments;
+  }
+  async waitForReadyState(options, browserLogsCollector) {
+    const result = new import_manualPromise.ManualPromise();
+    browserLogsCollector.onMessage((message) => {
+      const match = message.match(/WebDriver BiDi listening on (ws:\/\/.*)$/);
+      if (match)
+        result.resolve({ wsEndpoint: match[1] + "/session" });
+    });
+    return result;
+  }
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  BidiFirefox
+});