extensionContextFactory.js 3.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475
  1. "use strict";
  2. var __create = Object.create;
  3. var __defProp = Object.defineProperty;
  4. var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
  5. var __getOwnPropNames = Object.getOwnPropertyNames;
  6. var __getProtoOf = Object.getPrototypeOf;
  7. var __hasOwnProp = Object.prototype.hasOwnProperty;
  8. var __export = (target, all) => {
  9. for (var name in all)
  10. __defProp(target, name, { get: all[name], enumerable: true });
  11. };
  12. var __copyProps = (to, from, except, desc) => {
  13. if (from && typeof from === "object" || typeof from === "function") {
  14. for (let key of __getOwnPropNames(from))
  15. if (!__hasOwnProp.call(to, key) && key !== except)
  16. __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
  17. }
  18. return to;
  19. };
  20. var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
  21. // If the importer is in node compatibility mode or this is not an ESM
  22. // file that has been converted to a CommonJS file using a Babel-
  23. // compatible transform (i.e. "__esModule" has not been set), then set
  24. // "default" to the CommonJS "module.exports" for node compatibility.
  25. isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
  26. mod
  27. ));
  28. var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
  29. var extensionContextFactory_exports = {};
  30. __export(extensionContextFactory_exports, {
  31. ExtensionContextFactory: () => ExtensionContextFactory
  32. });
  33. module.exports = __toCommonJS(extensionContextFactory_exports);
  34. var playwright = __toESM(require("playwright-core"));
  35. var import_utilsBundle = require("playwright-core/lib/utilsBundle");
  36. var import_http = require("../sdk/http");
  37. var import_cdpRelay = require("./cdpRelay");
  38. const debugLogger = (0, import_utilsBundle.debug)("pw:mcp:relay");
  39. class ExtensionContextFactory {
  40. constructor(browserChannel, userDataDir, executablePath) {
  41. this._browserChannel = browserChannel;
  42. this._userDataDir = userDataDir;
  43. this._executablePath = executablePath;
  44. }
  45. async createContext(clientInfo, abortSignal, toolName) {
  46. const browser = await this._obtainBrowser(clientInfo, abortSignal, toolName);
  47. return {
  48. browserContext: browser.contexts()[0],
  49. close: async () => {
  50. debugLogger("close() called for browser context");
  51. await browser.close();
  52. }
  53. };
  54. }
  55. async _obtainBrowser(clientInfo, abortSignal, toolName) {
  56. const relay = await this._startRelay(abortSignal);
  57. await relay.ensureExtensionConnectionForMCPContext(clientInfo, abortSignal, toolName);
  58. return await playwright.chromium.connectOverCDP(relay.cdpEndpoint());
  59. }
  60. async _startRelay(abortSignal) {
  61. const httpServer = await (0, import_http.startHttpServer)({});
  62. if (abortSignal.aborted) {
  63. httpServer.close();
  64. throw new Error(abortSignal.reason);
  65. }
  66. const cdpRelayServer = new import_cdpRelay.CDPRelayServer(httpServer, this._browserChannel, this._userDataDir, this._executablePath);
  67. abortSignal.addEventListener("abort", () => cdpRelayServer.stop());
  68. debugLogger(`CDP relay server started, extension endpoint: ${cdpRelayServer.extensionEndpoint()}.`);
  69. return cdpRelayServer;
  70. }
  71. }
  72. // Annotate the CommonJS export names for ESM import in node:
  73. 0 && (module.exports = {
  74. ExtensionContextFactory
  75. });