patchright-bun 1.58.1 → 1.58.2

package/lib/mcp/terminal/socketConnection.js

@@ -1,80 +1,80 @@
-"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 socketConnection_exports = {};
-__export(socketConnection_exports, {
-  SocketConnection: () => SocketConnection
-});
-module.exports = __toCommonJS(socketConnection_exports);
-var import_utilsBundle = require("patchright-core/lib/utilsBundle");
-const daemonDebug = (0, import_utilsBundle.debug)("pw:daemon");
-class SocketConnection {
-  constructor(socket) {
-    this._pendingBuffers = [];
-    this._socket = socket;
-    socket.on("data", (buffer) => this._onData(buffer));
-    socket.on("close", () => {
-      this.onclose?.();
-    });
-    socket.on("error", (e) => daemonDebug(`error: ${e.message}`));
-  }
-  async send(message) {
-    await new Promise((resolve, reject) => {
-      this._socket.write(`${JSON.stringify(message)}
-`, (error) => {
-        if (error)
-          reject(error);
-        else
-          resolve(void 0);
-      });
-    });
-  }
-  close() {
-    this._socket.destroy();
-  }
-  _onData(buffer) {
-    let end = buffer.indexOf("\n");
-    if (end === -1) {
-      this._pendingBuffers.push(buffer);
-      return;
-    }
-    this._pendingBuffers.push(buffer.slice(0, end));
-    const message = Buffer.concat(this._pendingBuffers).toString();
-    this._dispatchMessage(message);
-    let start = end + 1;
-    end = buffer.indexOf("\n", start);
-    while (end !== -1) {
-      const message2 = buffer.toString(void 0, start, end);
-      this._dispatchMessage(message2);
-      start = end + 1;
-      end = buffer.indexOf("\n", start);
-    }
-    this._pendingBuffers = [buffer.slice(start)];
-  }
-  _dispatchMessage(message) {
-    try {
-      this.onmessage?.(JSON.parse(message));
-    } catch (e) {
-      daemonDebug("failed to dispatch message", e);
-    }
-  }
-}
-// Annotate the CommonJS export names for ESM import in node:
-0 && (module.exports = {
-  SocketConnection
-});
+"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 socketConnection_exports = {};
+__export(socketConnection_exports, {
+  SocketConnection: () => SocketConnection
+});
+module.exports = __toCommonJS(socketConnection_exports);
+var import_utilsBundle = require("patchright-bun-core/lib/utilsBundle");
+const daemonDebug = (0, import_utilsBundle.debug)("pw:daemon");
+class SocketConnection {
+  constructor(socket) {
+    this._pendingBuffers = [];
+    this._socket = socket;
+    socket.on("data", (buffer) => this._onData(buffer));
+    socket.on("close", () => {
+      this.onclose?.();
+    });
+    socket.on("error", (e) => daemonDebug(`error: ${e.message}`));
+  }
+  async send(message) {
+    await new Promise((resolve, reject) => {
+      this._socket.write(`${JSON.stringify(message)}
+`, (error) => {
+        if (error)
+          reject(error);
+        else
+          resolve(void 0);
+      });
+    });
+  }
+  close() {
+    this._socket.destroy();
+  }
+  _onData(buffer) {
+    let end = buffer.indexOf("\n");
+    if (end === -1) {
+      this._pendingBuffers.push(buffer);
+      return;
+    }
+    this._pendingBuffers.push(buffer.slice(0, end));
+    const message = Buffer.concat(this._pendingBuffers).toString();
+    this._dispatchMessage(message);
+    let start = end + 1;
+    end = buffer.indexOf("\n", start);
+    while (end !== -1) {
+      const message2 = buffer.toString(void 0, start, end);
+      this._dispatchMessage(message2);
+      start = end + 1;
+      end = buffer.indexOf("\n", start);
+    }
+    this._pendingBuffers = [buffer.slice(start)];
+  }
+  _dispatchMessage(message) {
+    try {
+      this.onmessage?.(JSON.parse(message));
+    } catch (e) {
+      daemonDebug("failed to dispatch message", e);
+    }
+  }
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  SocketConnection
+});

package/lib/mcp/test/browserBackend.js

@@ -1,98 +1,98 @@
-"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 browserBackend_exports = {};
-__export(browserBackend_exports, {
-  createCustomMessageHandler: () => createCustomMessageHandler
-});
-module.exports = __toCommonJS(browserBackend_exports);
-var import_config = require("../browser/config");
-var import_browserServerBackend = require("../browser/browserServerBackend");
-var import_tab = require("../browser/tab");
-var import_util = require("../../util");
-var import_browserContextFactory = require("../browser/browserContextFactory");
-function createCustomMessageHandler(testInfo, context) {
-  let backend;
-  return async (data) => {
-    if (data.initialize) {
-      if (backend)
-        throw new Error("MCP backend is already initialized");
-      backend = new import_browserServerBackend.BrowserServerBackend({ ...import_config.defaultConfig, capabilities: ["testing"] }, (0, import_browserContextFactory.identityBrowserContextFactory)(context));
-      await backend.initialize(data.initialize.clientInfo);
-      const pausedMessage = await generatePausedMessage(testInfo, context);
-      return { initialize: { pausedMessage } };
-    }
-    if (data.listTools) {
-      if (!backend)
-        throw new Error("MCP backend is not initialized");
-      return { listTools: await backend.listTools() };
-    }
-    if (data.callTool) {
-      if (!backend)
-        throw new Error("MCP backend is not initialized");
-      return { callTool: await backend.callTool(data.callTool.name, data.callTool.arguments) };
-    }
-    if (data.close) {
-      backend?.serverClosed();
-      backend = void 0;
-      return { close: {} };
-    }
-    throw new Error("Unknown MCP request");
-  };
-}
-async function generatePausedMessage(testInfo, context) {
-  const lines = [];
-  if (testInfo.errors.length) {
-    lines.push(`### Paused on error:`);
-    for (const error of testInfo.errors)
-      lines.push((0, import_util.stripAnsiEscapes)(error.message || ""));
-  } else {
-    lines.push(`### Paused at end of test. ready for interaction`);
-  }
-  for (let i = 0; i < context.pages().length; i++) {
-    const page = context.pages()[i];
-    const stateSuffix = context.pages().length > 1 ? i + 1 + " of " + context.pages().length : "state";
-    lines.push(
-      "",
-      `### Page ${stateSuffix}`,
-      `- Page URL: ${page.url()}`,
-      `- Page Title: ${await page.title()}`.trim()
-    );
-    let console = testInfo.errors.length ? await import_tab.Tab.collectConsoleMessages(page) : [];
-    console = console.filter((msg) => msg.type === "error");
-    if (console.length) {
-      lines.push("- Console Messages:");
-      for (const message of console)
-        lines.push(`  - ${message.toString()}`);
-    }
-    lines.push(
-      `- Page Snapshot:`,
-      "```yaml",
-      (await page._snapshotForAI()).full,
-      "```"
-    );
-  }
-  lines.push("");
-  if (testInfo.errors.length)
-    lines.push(`### Task`, `Try recovering from the error prior to continuing`);
-  return lines.join("\n");
-}
-// Annotate the CommonJS export names for ESM import in node:
-0 && (module.exports = {
-  createCustomMessageHandler
-});
+"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 browserBackend_exports = {};
+__export(browserBackend_exports, {
+  createCustomMessageHandler: () => createCustomMessageHandler
+});
+module.exports = __toCommonJS(browserBackend_exports);
+var import_config = require("../browser/config");
+var import_browserServerBackend = require("../browser/browserServerBackend");
+var import_tab = require("../browser/tab");
+var import_util = require("../../util");
+var import_browserContextFactory = require("../browser/browserContextFactory");
+function createCustomMessageHandler(testInfo, context) {
+  let backend;
+  return async (data) => {
+    if (data.initialize) {
+      if (backend)
+        throw new Error("MCP backend is already initialized");
+      backend = new import_browserServerBackend.BrowserServerBackend({ ...import_config.defaultConfig, capabilities: ["testing"] }, (0, import_browserContextFactory.identityBrowserContextFactory)(context));
+      await backend.initialize(data.initialize.clientInfo);
+      const pausedMessage = await generatePausedMessage(testInfo, context);
+      return { initialize: { pausedMessage } };
+    }
+    if (data.listTools) {
+      if (!backend)
+        throw new Error("MCP backend is not initialized");
+      return { listTools: await backend.listTools() };
+    }
+    if (data.callTool) {
+      if (!backend)
+        throw new Error("MCP backend is not initialized");
+      return { callTool: await backend.callTool(data.callTool.name, data.callTool.arguments) };
+    }
+    if (data.close) {
+      backend?.serverClosed();
+      backend = void 0;
+      return { close: {} };
+    }
+    throw new Error("Unknown MCP request");
+  };
+}
+async function generatePausedMessage(testInfo, context) {
+  const lines = [];
+  if (testInfo.errors.length) {
+    lines.push(`### Paused on error:`);
+    for (const error of testInfo.errors)
+      lines.push((0, import_util.stripAnsiEscapes)(error.message || ""));
+  } else {
+    lines.push(`### Paused at end of test. ready for interaction`);
+  }
+  for (let i = 0; i < context.pages().length; i++) {
+    const page = context.pages()[i];
+    const stateSuffix = context.pages().length > 1 ? i + 1 + " of " + context.pages().length : "state";
+    lines.push(
+      "",
+      `### Page ${stateSuffix}`,
+      `- Page URL: ${page.url()}`,
+      `- Page Title: ${await page.title()}`.trim()
+    );
+    let console = testInfo.errors.length ? await import_tab.Tab.collectConsoleMessages(page) : [];
+    console = console.filter((msg) => msg.type === "error");
+    if (console.length) {
+      lines.push("- Console Messages:");
+      for (const message of console)
+        lines.push(`  - ${message.toString()}`);
+    }
+    lines.push(
+      `- Page Snapshot:`,
+      "```yaml",
+      (await page._snapshotForAI()).full,
+      "```"
+    );
+  }
+  lines.push("");
+  if (testInfo.errors.length)
+    lines.push(`### Task`, `Try recovering from the error prior to continuing`);
+  return lines.join("\n");
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  createCustomMessageHandler
+});

package/lib/mcp/test/generatorTools.js

@@ -1,122 +1,122 @@
-"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 generatorTools_exports = {};
-__export(generatorTools_exports, {
-  generatorReadLog: () => generatorReadLog,
-  generatorWriteTest: () => generatorWriteTest,
-  setupPage: () => setupPage
-});
-module.exports = __toCommonJS(generatorTools_exports);
-var import_fs = __toESM(require("fs"));
-var import_path = __toESM(require("path"));
-var import_mcpBundle = require("patchright-core/lib/mcpBundle");
-var import_testTool = require("./testTool");
-var import_testContext = require("./testContext");
-const setupPage = (0, import_testTool.defineTestTool)({
-  schema: {
-    name: "generator_setup_page",
-    title: "Setup generator page",
-    description: "Setup the page for test.",
-    inputSchema: import_mcpBundle.z.object({
-      plan: import_mcpBundle.z.string().describe("The plan for the test. This should be the actual test plan with all the steps."),
-      project: import_mcpBundle.z.string().optional().describe('Project to use for setup. For example: "chromium", if no project is provided uses the first project in the config.'),
-      seedFile: import_mcpBundle.z.string().optional().describe('A seed file contains a single test that is used to setup the page for testing, for example: "tests/seed.spec.ts". If no seed file is provided, a default seed file is created.')
-    }),
-    type: "readOnly"
-  },
-  handle: async (context, params) => {
-    const seed = await context.getOrCreateSeedFile(params.seedFile, params.project);
-    context.generatorJournal = new import_testContext.GeneratorJournal(context.rootPath, params.plan, seed);
-    const { output, status } = await context.runSeedTest(seed.file, seed.projectName);
-    return { content: [{ type: "text", text: output }], isError: status !== "paused" };
-  }
-});
-const generatorReadLog = (0, import_testTool.defineTestTool)({
-  schema: {
-    name: "generator_read_log",
-    title: "Retrieve test log",
-    description: "Retrieve the performed test log",
-    inputSchema: import_mcpBundle.z.object({}),
-    type: "readOnly"
-  },
-  handle: async (context) => {
-    if (!context.generatorJournal)
-      throw new Error(`Please setup page using "${setupPage.schema.name}" first.`);
-    const result = context.generatorJournal.journal();
-    return { content: [{
-      type: "text",
-      text: result
-    }] };
-  }
-});
-const generatorWriteTest = (0, import_testTool.defineTestTool)({
-  schema: {
-    name: "generator_write_test",
-    title: "Write test",
-    description: "Write the generated test to the test file",
-    inputSchema: import_mcpBundle.z.object({
-      fileName: import_mcpBundle.z.string().describe("The file to write the test to"),
-      code: import_mcpBundle.z.string().describe("The generated test code")
-    }),
-    type: "readOnly"
-  },
-  handle: async (context, params) => {
-    if (!context.generatorJournal)
-      throw new Error(`Please setup page using "${setupPage.schema.name}" first.`);
-    const testRunner = context.existingTestRunner();
-    if (!testRunner)
-      throw new Error("No test runner found, please setup page and perform actions first.");
-    const config = await testRunner.loadConfig();
-    const dirs = [];
-    for (const project of config.projects) {
-      const testDir = import_path.default.relative(context.rootPath, project.project.testDir).replace(/\\/g, "/");
-      const fileName = params.fileName.replace(/\\/g, "/");
-      if (fileName.startsWith(testDir)) {
-        const resolvedFile = import_path.default.resolve(context.rootPath, fileName);
-        await import_fs.default.promises.mkdir(import_path.default.dirname(resolvedFile), { recursive: true });
-        await import_fs.default.promises.writeFile(resolvedFile, params.code);
-        return {
-          content: [{
-            type: "text",
-            text: `### Result
-Test written to ${params.fileName}`
-          }]
-        };
-      }
-      dirs.push(testDir);
-    }
-    throw new Error(`Test file did not match any of the test dirs: ${dirs.join(", ")}`);
-  }
-});
-// Annotate the CommonJS export names for ESM import in node:
-0 && (module.exports = {
-  generatorReadLog,
-  generatorWriteTest,
-  setupPage
-});
+"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 generatorTools_exports = {};
+__export(generatorTools_exports, {
+  generatorReadLog: () => generatorReadLog,
+  generatorWriteTest: () => generatorWriteTest,
+  setupPage: () => setupPage
+});
+module.exports = __toCommonJS(generatorTools_exports);
+var import_fs = __toESM(require("fs"));
+var import_path = __toESM(require("path"));
+var import_mcpBundle = require("patchright-bun-core/lib/mcpBundle");
+var import_testTool = require("./testTool");
+var import_testContext = require("./testContext");
+const setupPage = (0, import_testTool.defineTestTool)({
+  schema: {
+    name: "generator_setup_page",
+    title: "Setup generator page",
+    description: "Setup the page for test.",
+    inputSchema: import_mcpBundle.z.object({
+      plan: import_mcpBundle.z.string().describe("The plan for the test. This should be the actual test plan with all the steps."),
+      project: import_mcpBundle.z.string().optional().describe('Project to use for setup. For example: "chromium", if no project is provided uses the first project in the config.'),
+      seedFile: import_mcpBundle.z.string().optional().describe('A seed file contains a single test that is used to setup the page for testing, for example: "tests/seed.spec.ts". If no seed file is provided, a default seed file is created.')
+    }),
+    type: "readOnly"
+  },
+  handle: async (context, params) => {
+    const seed = await context.getOrCreateSeedFile(params.seedFile, params.project);
+    context.generatorJournal = new import_testContext.GeneratorJournal(context.rootPath, params.plan, seed);
+    const { output, status } = await context.runSeedTest(seed.file, seed.projectName);
+    return { content: [{ type: "text", text: output }], isError: status !== "paused" };
+  }
+});
+const generatorReadLog = (0, import_testTool.defineTestTool)({
+  schema: {
+    name: "generator_read_log",
+    title: "Retrieve test log",
+    description: "Retrieve the performed test log",
+    inputSchema: import_mcpBundle.z.object({}),
+    type: "readOnly"
+  },
+  handle: async (context) => {
+    if (!context.generatorJournal)
+      throw new Error(`Please setup page using "${setupPage.schema.name}" first.`);
+    const result = context.generatorJournal.journal();
+    return { content: [{
+      type: "text",
+      text: result
+    }] };
+  }
+});
+const generatorWriteTest = (0, import_testTool.defineTestTool)({
+  schema: {
+    name: "generator_write_test",
+    title: "Write test",
+    description: "Write the generated test to the test file",
+    inputSchema: import_mcpBundle.z.object({
+      fileName: import_mcpBundle.z.string().describe("The file to write the test to"),
+      code: import_mcpBundle.z.string().describe("The generated test code")
+    }),
+    type: "readOnly"
+  },
+  handle: async (context, params) => {
+    if (!context.generatorJournal)
+      throw new Error(`Please setup page using "${setupPage.schema.name}" first.`);
+    const testRunner = context.existingTestRunner();
+    if (!testRunner)
+      throw new Error("No test runner found, please setup page and perform actions first.");
+    const config = await testRunner.loadConfig();
+    const dirs = [];
+    for (const project of config.projects) {
+      const testDir = import_path.default.relative(context.rootPath, project.project.testDir).replace(/\\/g, "/");
+      const fileName = params.fileName.replace(/\\/g, "/");
+      if (fileName.startsWith(testDir)) {
+        const resolvedFile = import_path.default.resolve(context.rootPath, fileName);
+        await import_fs.default.promises.mkdir(import_path.default.dirname(resolvedFile), { recursive: true });
+        await import_fs.default.promises.writeFile(resolvedFile, params.code);
+        return {
+          content: [{
+            type: "text",
+            text: `### Result
+Test written to ${params.fileName}`
+          }]
+        };
+      }
+      dirs.push(testDir);
+    }
+    throw new Error(`Test file did not match any of the test dirs: ${dirs.join(", ")}`);
+  }
+});
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  generatorReadLog,
+  generatorWriteTest,
+  setupPage
+});