@treegress.com/treegress-browser-core 1.59.0-treegress.3

package/lib/protocol/validatorPrimitives.js

@@ -0,0 +1,193 @@
+"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 validatorPrimitives_exports = {};
+__export(validatorPrimitives_exports, {
+  ValidationError: () => ValidationError,
+  createMetadataValidator: () => createMetadataValidator,
+  findValidator: () => findValidator,
+  maybeFindValidator: () => maybeFindValidator,
+  scheme: () => scheme,
+  tAny: () => tAny,
+  tArray: () => tArray,
+  tBinary: () => tBinary,
+  tBoolean: () => tBoolean,
+  tChannel: () => tChannel,
+  tEnum: () => tEnum,
+  tFloat: () => tFloat,
+  tInt: () => tInt,
+  tObject: () => tObject,
+  tOptional: () => tOptional,
+  tString: () => tString,
+  tType: () => tType,
+  tUndefined: () => tUndefined
+});
+module.exports = __toCommonJS(validatorPrimitives_exports);
+class ValidationError extends Error {
+}
+const scheme = {};
+function findValidator(type, method, kind) {
+  const validator = maybeFindValidator(type, method, kind);
+  if (!validator)
+    throw new ValidationError(`Unknown scheme for ${kind}: ${type}.${method}`);
+  return validator;
+}
+function maybeFindValidator(type, method, kind) {
+  const schemeName = type + (kind === "Initializer" ? "" : method[0].toUpperCase() + method.substring(1)) + kind;
+  return scheme[schemeName];
+}
+function createMetadataValidator() {
+  return tOptional(scheme["Metadata"]);
+}
+const tFloat = (arg, path, context) => {
+  if (arg instanceof Number)
+    return arg.valueOf();
+  if (typeof arg === "number")
+    return arg;
+  throw new ValidationError(`${path}: expected float, got ${typeof arg}`);
+};
+const tInt = (arg, path, context) => {
+  let value;
+  if (arg instanceof Number)
+    value = arg.valueOf();
+  else if (typeof arg === "number")
+    value = arg;
+  else
+    throw new ValidationError(`${path}: expected integer, got ${typeof arg}`);
+  if (!Number.isInteger(value))
+    throw new ValidationError(`${path}: expected integer, got float ${value}`);
+  return value;
+};
+const tBoolean = (arg, path, context) => {
+  if (arg instanceof Boolean)
+    return arg.valueOf();
+  if (typeof arg === "boolean")
+    return arg;
+  throw new ValidationError(`${path}: expected boolean, got ${typeof arg}`);
+};
+const tString = (arg, path, context) => {
+  if (arg instanceof String)
+    return arg.valueOf();
+  if (typeof arg === "string")
+    return arg;
+  throw new ValidationError(`${path}: expected string, got ${typeof arg}`);
+};
+const tBinary = (arg, path, context) => {
+  if (context.binary === "fromBase64") {
+    if (arg instanceof String)
+      return Buffer.from(arg.valueOf(), "base64");
+    if (typeof arg === "string")
+      return Buffer.from(arg, "base64");
+    throw new ValidationError(`${path}: expected base64-encoded buffer, got ${typeof arg}`);
+  }
+  if (context.binary === "toBase64") {
+    if (!(arg instanceof Buffer))
+      throw new ValidationError(`${path}: expected Buffer, got ${typeof arg}`);
+    return arg.toString("base64");
+  }
+  if (context.binary === "buffer") {
+    if (!(arg instanceof Buffer) && !(arg instanceof Object))
+      throw new ValidationError(`${path}: expected Buffer, got ${typeof arg}`);
+    return arg;
+  }
+  throw new ValidationError(`Unsupported binary behavior "${context.binary}"`);
+};
+const tUndefined = (arg, path, context) => {
+  if (Object.is(arg, void 0))
+    return arg;
+  throw new ValidationError(`${path}: expected undefined, got ${typeof arg}`);
+};
+const tAny = (arg, path, context) => {
+  return arg;
+};
+const tOptional = (v) => {
+  return (arg, path, context) => {
+    if (Object.is(arg, void 0))
+      return arg;
+    return v(arg, path, context);
+  };
+};
+const tArray = (v) => {
+  return (arg, path, context) => {
+    if (!Array.isArray(arg))
+      throw new ValidationError(`${path}: expected array, got ${typeof arg}`);
+    return arg.map((x, index) => v(x, path + "[" + index + "]", context));
+  };
+};
+const tObject = (s) => {
+  return (arg, path, context) => {
+    if (Object.is(arg, null))
+      throw new ValidationError(`${path}: expected object, got null`);
+    if (typeof arg !== "object")
+      throw new ValidationError(`${path}: expected object, got ${typeof arg}`);
+    const result = {};
+    for (const [key, v] of Object.entries(s)) {
+      const value = v(arg[key], path ? path + "." + key : key, context);
+      if (!Object.is(value, void 0))
+        result[key] = value;
+    }
+    if (context.isUnderTest()) {
+      for (const [key, value] of Object.entries(arg)) {
+        if (key.startsWith("__testHook"))
+          result[key] = value;
+      }
+    }
+    return result;
+  };
+};
+const tEnum = (e) => {
+  return (arg, path, context) => {
+    if (!e.includes(arg))
+      throw new ValidationError(`${path}: expected one of (${e.join("|")})`);
+    return arg;
+  };
+};
+const tChannel = (names) => {
+  return (arg, path, context) => {
+    return context.tChannelImpl(names, arg, path, context);
+  };
+};
+const tType = (name) => {
+  return (arg, path, context) => {
+    const v = scheme[name];
+    if (!v)
+      throw new ValidationError(path + ': unknown type "' + name + '"');
+    return v(arg, path, context);
+  };
+};
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  ValidationError,
+  createMetadataValidator,
+  findValidator,
+  maybeFindValidator,
+  scheme,
+  tAny,
+  tArray,
+  tBinary,
+  tBoolean,
+  tChannel,
+  tEnum,
+  tFloat,
+  tInt,
+  tObject,
+  tOptional,
+  tString,
+  tType,
+  tUndefined
+});

package/lib/remote/playwrightConnection.js

@@ -0,0 +1,129 @@
+"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 playwrightConnection_exports = {};
+__export(playwrightConnection_exports, {
+  PlaywrightConnection: () => PlaywrightConnection
+});
+module.exports = __toCommonJS(playwrightConnection_exports);
+var import_server = require("../server");
+var import_android = require("../server/android/android");
+var import_browser = require("../server/browser");
+var import_debugControllerDispatcher = require("../server/dispatchers/debugControllerDispatcher");
+var import_profiler = require("../server/utils/profiler");
+var import_utils = require("../utils");
+var import_debugLogger = require("../server/utils/debugLogger");
+class PlaywrightConnection {
+  constructor(semaphore, transport, controller, playwright, initialize, id) {
+    this._cleanups = [];
+    this._disconnected = false;
+    this._transport = transport;
+    this._semaphore = semaphore;
+    this._id = id;
+    this._profileName = (/* @__PURE__ */ new Date()).toISOString();
+    const lock = this._semaphore.acquire();
+    this._dispatcherConnection = new import_server.DispatcherConnection();
+    this._dispatcherConnection.onmessage = async (message) => {
+      await lock;
+      if (!transport.isClosed()) {
+        const messageString = JSON.stringify(message);
+        if (import_debugLogger.debugLogger.isEnabled("server:channel"))
+          import_debugLogger.debugLogger.log("server:channel", `[${this._id}] ${(0, import_utils.monotonicTime)() * 1e3} SEND \u25BA ${messageString}`);
+        if (import_debugLogger.debugLogger.isEnabled("server:metadata"))
+          this.logServerMetadata(message, messageString, "SEND");
+        transport.send(messageString);
+      }
+    };
+    transport.on("message", async (message) => {
+      await lock;
+      const messageString = Buffer.from(message).toString();
+      const jsonMessage = JSON.parse(messageString);
+      if (import_debugLogger.debugLogger.isEnabled("server:channel"))
+        import_debugLogger.debugLogger.log("server:channel", `[${this._id}] ${(0, import_utils.monotonicTime)() * 1e3} \u25C0 RECV ${messageString}`);
+      if (import_debugLogger.debugLogger.isEnabled("server:metadata"))
+        this.logServerMetadata(jsonMessage, messageString, "RECV");
+      this._dispatcherConnection.dispatch(jsonMessage);
+    });
+    transport.on("close", () => this._onDisconnect());
+    transport.on("error", (error) => this._onDisconnect(error));
+    if (controller) {
+      import_debugLogger.debugLogger.log("server", `[${this._id}] engaged reuse controller mode`);
+      this._root = new import_debugControllerDispatcher.DebugControllerDispatcher(this._dispatcherConnection, playwright.debugController);
+      return;
+    }
+    this._root = new import_server.RootDispatcher(this._dispatcherConnection, async (scope, params) => {
+      await (0, import_profiler.startProfiling)();
+      const options = await initialize();
+      if (options.preLaunchedBrowser) {
+        const browser = options.preLaunchedBrowser;
+        browser.options.sdkLanguage = params.sdkLanguage;
+        browser.on(import_browser.Browser.Events.Disconnected, () => {
+          this.close({ code: 1001, reason: "Browser closed" });
+        });
+      }
+      if (options.preLaunchedAndroidDevice) {
+        const androidDevice = options.preLaunchedAndroidDevice;
+        androidDevice.on(import_android.AndroidDevice.Events.Close, () => {
+          this.close({ code: 1001, reason: "Android device disconnected" });
+        });
+      }
+      if (options.dispose)
+        this._cleanups.push(options.dispose);
+      const dispatcher = new import_server.PlaywrightDispatcher(scope, playwright, options);
+      this._cleanups.push(() => dispatcher.cleanup());
+      return dispatcher;
+    });
+  }
+  async _onDisconnect(error) {
+    this._disconnected = true;
+    import_debugLogger.debugLogger.log("server", `[${this._id}] disconnected. error: ${error}`);
+    await this._root.stopPendingOperations(new Error("Disconnected")).catch(() => {
+    });
+    this._root._dispose();
+    import_debugLogger.debugLogger.log("server", `[${this._id}] starting cleanup`);
+    for (const cleanup of this._cleanups)
+      await cleanup().catch(() => {
+      });
+    await (0, import_profiler.stopProfiling)(this._profileName);
+    this._semaphore.release();
+    import_debugLogger.debugLogger.log("server", `[${this._id}] finished cleanup`);
+  }
+  logServerMetadata(message, messageString, direction) {
+    const serverLogMetadata = {
+      wallTime: Date.now(),
+      id: message.id,
+      guid: message.guid,
+      method: message.method,
+      payloadSizeInBytes: Buffer.byteLength(messageString, "utf-8")
+    };
+    import_debugLogger.debugLogger.log("server:metadata", (direction === "SEND" ? "SEND \u25BA " : "\u25C0 RECV ") + JSON.stringify(serverLogMetadata));
+  }
+  async close(reason) {
+    if (this._disconnected)
+      return;
+    import_debugLogger.debugLogger.log("server", `[${this._id}] force closing connection: ${reason?.reason || ""} (${reason?.code || 0})`);
+    try {
+      this._transport.close(reason);
+    } catch (e) {
+    }
+  }
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  PlaywrightConnection
+});

package/lib/remote/playwrightPipeServer.js

@@ -0,0 +1,100 @@
+"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 playwrightPipeServer_exports = {};
+__export(playwrightPipeServer_exports, {
+  PlaywrightPipeServer: () => PlaywrightPipeServer
+});
+module.exports = __toCommonJS(playwrightPipeServer_exports);
+var import_net = __toESM(require("net"));
+var import_fs = __toESM(require("fs"));
+var import_playwrightConnection = require("./playwrightConnection");
+var import_serverTransport = require("./serverTransport");
+var import_debugLogger = require("../server/utils/debugLogger");
+var import_browser = require("../server/browser");
+var import_utils = require("../utils");
+class PlaywrightPipeServer {
+  constructor(browser) {
+    this._connections = /* @__PURE__ */ new Set();
+    this._connectionId = 0;
+    this._browser = browser;
+    browser.on(import_browser.Browser.Events.Disconnected, () => this.close());
+  }
+  async listen(pipeName) {
+    if (!pipeName.startsWith("\\\\.\\pipe\\")) {
+      try {
+        import_fs.default.unlinkSync(pipeName);
+      } catch {
+      }
+    }
+    this._server = import_net.default.createServer((socket) => {
+      const id = String(++this._connectionId);
+      import_debugLogger.debugLogger.log("server", `[${id}] pipe client connected`);
+      const transport = new import_serverTransport.SocketServerTransport(socket);
+      const connection = new import_playwrightConnection.PlaywrightConnection(
+        new import_utils.Semaphore(1),
+        transport,
+        false,
+        this._browser.attribution.playwright,
+        () => this._initPreLaunchedBrowserMode(id),
+        id
+      );
+      this._connections.add(connection);
+      transport.on("close", () => this._connections.delete(connection));
+    });
+    (0, import_utils.decorateServer)(this._server);
+    await new Promise((resolve, reject) => {
+      this._server.listen(pipeName, () => resolve());
+      this._server.on("error", reject);
+    });
+    import_debugLogger.debugLogger.log("server", `Pipe server listening at ${pipeName}`);
+  }
+  async _initPreLaunchedBrowserMode(id) {
+    import_debugLogger.debugLogger.log("server", `[${id}] engaged pre-launched (browser) pipe mode`);
+    return {
+      preLaunchedBrowser: this._browser,
+      sharedBrowser: true,
+      denyLaunch: true
+    };
+  }
+  async close() {
+    if (!this._server)
+      return;
+    import_debugLogger.debugLogger.log("server", "closing pipe server");
+    for (const connection of this._connections)
+      await connection.close({ code: 1001, reason: "Server closing" });
+    this._connections.clear();
+    await new Promise((f) => this._server.close(() => f()));
+    this._server = void 0;
+    import_debugLogger.debugLogger.log("server", "closed pipe server");
+  }
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  PlaywrightPipeServer
+});