patchright-bun 1.58.1 → 1.58.2

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (180) hide show
  1. package/README.md +225 -225
  2. package/ThirdPartyNotices.txt +5041 -5041
  3. package/cli.js +19 -19
  4. package/index.d.ts +17 -17
  5. package/index.js +17 -17
  6. package/index.mjs +18 -18
  7. package/jsx-runtime.js +42 -42
  8. package/jsx-runtime.mjs +21 -21
  9. package/lib/agents/agentParser.js +89 -89
  10. package/lib/agents/copilot-setup-steps.yml +34 -34
  11. package/lib/agents/generateAgents.js +348 -348
  12. package/lib/agents/playwright-test-coverage.prompt.md +31 -31
  13. package/lib/agents/playwright-test-generate.prompt.md +8 -8
  14. package/lib/agents/playwright-test-generator.agent.md +88 -88
  15. package/lib/agents/playwright-test-heal.prompt.md +6 -6
  16. package/lib/agents/playwright-test-healer.agent.md +55 -55
  17. package/lib/agents/playwright-test-plan.prompt.md +9 -9
  18. package/lib/agents/playwright-test-planner.agent.md +73 -73
  19. package/lib/common/config.js +282 -282
  20. package/lib/common/configLoader.js +344 -344
  21. package/lib/common/esmLoaderHost.js +104 -104
  22. package/lib/common/expectBundle.js +28 -28
  23. package/lib/common/expectBundleImpl.js +407 -407
  24. package/lib/common/fixtures.js +302 -302
  25. package/lib/common/globals.js +58 -58
  26. package/lib/common/ipc.js +60 -60
  27. package/lib/common/poolBuilder.js +85 -85
  28. package/lib/common/process.js +132 -132
  29. package/lib/common/suiteUtils.js +140 -140
  30. package/lib/common/test.js +321 -321
  31. package/lib/common/testLoader.js +101 -101
  32. package/lib/common/testType.js +298 -298
  33. package/lib/common/validators.js +68 -68
  34. package/lib/fsWatcher.js +67 -67
  35. package/lib/index.js +726 -726
  36. package/lib/internalsForTest.js +42 -42
  37. package/lib/isomorphic/events.js +77 -77
  38. package/lib/isomorphic/folders.js +30 -30
  39. package/lib/isomorphic/stringInternPool.js +69 -69
  40. package/lib/isomorphic/teleReceiver.js +521 -521
  41. package/lib/isomorphic/teleSuiteUpdater.js +157 -157
  42. package/lib/isomorphic/testServerConnection.js +225 -225
  43. package/lib/isomorphic/testServerInterface.js +16 -16
  44. package/lib/isomorphic/testTree.js +329 -329
  45. package/lib/isomorphic/types.d.js +16 -16
  46. package/lib/loader/loaderMain.js +59 -59
  47. package/lib/matchers/expect.js +311 -311
  48. package/lib/matchers/matcherHint.js +44 -44
  49. package/lib/matchers/matchers.js +383 -383
  50. package/lib/matchers/toBeTruthy.js +75 -75
  51. package/lib/matchers/toEqual.js +100 -100
  52. package/lib/matchers/toHaveURL.js +101 -101
  53. package/lib/matchers/toMatchAriaSnapshot.js +159 -159
  54. package/lib/matchers/toMatchSnapshot.js +342 -342
  55. package/lib/matchers/toMatchText.js +99 -99
  56. package/lib/mcp/browser/browserContextFactory.js +329 -329
  57. package/lib/mcp/browser/browserServerBackend.js +84 -84
  58. package/lib/mcp/browser/config.js +421 -421
  59. package/lib/mcp/browser/context.js +244 -244
  60. package/lib/mcp/browser/response.js +278 -278
  61. package/lib/mcp/browser/sessionLog.js +75 -75
  62. package/lib/mcp/browser/tab.js +343 -343
  63. package/lib/mcp/browser/tools/common.js +65 -65
  64. package/lib/mcp/browser/tools/console.js +46 -46
  65. package/lib/mcp/browser/tools/dialogs.js +60 -60
  66. package/lib/mcp/browser/tools/evaluate.js +61 -61
  67. package/lib/mcp/browser/tools/files.js +58 -58
  68. package/lib/mcp/browser/tools/form.js +63 -63
  69. package/lib/mcp/browser/tools/install.js +72 -72
  70. package/lib/mcp/browser/tools/keyboard.js +107 -107
  71. package/lib/mcp/browser/tools/mouse.js +107 -107
  72. package/lib/mcp/browser/tools/navigate.js +71 -71
  73. package/lib/mcp/browser/tools/network.js +63 -63
  74. package/lib/mcp/browser/tools/open.js +57 -57
  75. package/lib/mcp/browser/tools/pdf.js +49 -49
  76. package/lib/mcp/browser/tools/runCode.js +78 -78
  77. package/lib/mcp/browser/tools/screenshot.js +93 -93
  78. package/lib/mcp/browser/tools/snapshot.js +173 -173
  79. package/lib/mcp/browser/tools/tabs.js +67 -67
  80. package/lib/mcp/browser/tools/tool.js +47 -47
  81. package/lib/mcp/browser/tools/tracing.js +74 -74
  82. package/lib/mcp/browser/tools/utils.js +94 -94
  83. package/lib/mcp/browser/tools/verify.js +143 -143
  84. package/lib/mcp/browser/tools/wait.js +63 -63
  85. package/lib/mcp/browser/tools.js +84 -84
  86. package/lib/mcp/browser/watchdog.js +44 -44
  87. package/lib/mcp/config.d.js +16 -16
  88. package/lib/mcp/extension/cdpRelay.js +351 -351
  89. package/lib/mcp/extension/extensionContextFactory.js +76 -76
  90. package/lib/mcp/extension/protocol.js +28 -28
  91. package/lib/mcp/index.js +61 -61
  92. package/lib/mcp/log.js +35 -35
  93. package/lib/mcp/program.js +111 -111
  94. package/lib/mcp/sdk/exports.js +28 -28
  95. package/lib/mcp/sdk/http.js +152 -152
  96. package/lib/mcp/sdk/inProcessTransport.js +71 -71
  97. package/lib/mcp/sdk/server.js +223 -223
  98. package/lib/mcp/sdk/tool.js +47 -47
  99. package/lib/mcp/terminal/cli.js +296 -296
  100. package/lib/mcp/terminal/command.js +56 -56
  101. package/lib/mcp/terminal/commands.js +333 -333
  102. package/lib/mcp/terminal/daemon.js +129 -129
  103. package/lib/mcp/terminal/help.json +31 -31
  104. package/lib/mcp/terminal/helpGenerator.js +88 -88
  105. package/lib/mcp/terminal/socketConnection.js +80 -80
  106. package/lib/mcp/test/browserBackend.js +98 -98
  107. package/lib/mcp/test/generatorTools.js +122 -122
  108. package/lib/mcp/test/plannerTools.js +145 -145
  109. package/lib/mcp/test/seed.js +82 -82
  110. package/lib/mcp/test/streams.js +44 -44
  111. package/lib/mcp/test/testBackend.js +99 -99
  112. package/lib/mcp/test/testContext.js +285 -285
  113. package/lib/mcp/test/testTool.js +30 -30
  114. package/lib/mcp/test/testTools.js +108 -108
  115. package/lib/plugins/gitCommitInfoPlugin.js +198 -198
  116. package/lib/plugins/index.js +28 -28
  117. package/lib/plugins/webServerPlugin.js +237 -237
  118. package/lib/program.js +417 -417
  119. package/lib/reporters/base.js +634 -634
  120. package/lib/reporters/blob.js +138 -138
  121. package/lib/reporters/dot.js +99 -99
  122. package/lib/reporters/empty.js +32 -32
  123. package/lib/reporters/github.js +128 -128
  124. package/lib/reporters/html.js +633 -633
  125. package/lib/reporters/internalReporter.js +138 -138
  126. package/lib/reporters/json.js +254 -254
  127. package/lib/reporters/junit.js +232 -232
  128. package/lib/reporters/line.js +131 -131
  129. package/lib/reporters/list.js +253 -253
  130. package/lib/reporters/listModeReporter.js +69 -69
  131. package/lib/reporters/markdown.js +144 -144
  132. package/lib/reporters/merge.js +558 -558
  133. package/lib/reporters/multiplexer.js +112 -112
  134. package/lib/reporters/reporterV2.js +102 -102
  135. package/lib/reporters/teleEmitter.js +317 -317
  136. package/lib/reporters/versions/blobV1.js +16 -16
  137. package/lib/runner/dispatcher.js +530 -530
  138. package/lib/runner/failureTracker.js +72 -72
  139. package/lib/runner/lastRun.js +77 -77
  140. package/lib/runner/loadUtils.js +334 -334
  141. package/lib/runner/loaderHost.js +89 -89
  142. package/lib/runner/processHost.js +180 -180
  143. package/lib/runner/projectUtils.js +241 -241
  144. package/lib/runner/rebase.js +189 -189
  145. package/lib/runner/reporters.js +138 -138
  146. package/lib/runner/sigIntWatcher.js +96 -96
  147. package/lib/runner/storage.js +91 -91
  148. package/lib/runner/taskRunner.js +127 -127
  149. package/lib/runner/tasks.js +410 -410
  150. package/lib/runner/testGroups.js +125 -125
  151. package/lib/runner/testRunner.js +398 -398
  152. package/lib/runner/testServer.js +269 -269
  153. package/lib/runner/uiModeReporter.js +30 -30
  154. package/lib/runner/vcs.js +72 -72
  155. package/lib/runner/watchMode.js +396 -396
  156. package/lib/runner/workerHost.js +104 -104
  157. package/lib/third_party/pirates.js +62 -62
  158. package/lib/third_party/tsconfig-loader.js +103 -103
  159. package/lib/transform/babelBundle.js +46 -46
  160. package/lib/transform/babelBundleImpl.js +461 -461
  161. package/lib/transform/compilationCache.js +274 -274
  162. package/lib/transform/esmLoader.js +103 -103
  163. package/lib/transform/md.js +221 -221
  164. package/lib/transform/portTransport.js +67 -67
  165. package/lib/transform/transform.js +303 -303
  166. package/lib/util.js +400 -400
  167. package/lib/utilsBundle.js +50 -50
  168. package/lib/utilsBundleImpl.js +103 -103
  169. package/lib/worker/fixtureRunner.js +262 -262
  170. package/lib/worker/testInfo.js +536 -536
  171. package/lib/worker/testTracing.js +345 -345
  172. package/lib/worker/timeoutManager.js +174 -174
  173. package/lib/worker/util.js +31 -31
  174. package/lib/worker/workerMain.js +530 -530
  175. package/package.json +2 -2
  176. package/test.d.ts +18 -18
  177. package/test.js +24 -24
  178. package/test.mjs +34 -34
  179. package/types/test.d.ts +10251 -10251
  180. package/types/testReporter.d.ts +822 -822
@@ -1,351 +1,351 @@
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 cdpRelay_exports = {};
30
- __export(cdpRelay_exports, {
31
- CDPRelayServer: () => CDPRelayServer
32
- });
33
- module.exports = __toCommonJS(cdpRelay_exports);
34
- var import_child_process = require("child_process");
35
- var import_utilsBundle = require("patchright-core/lib/utilsBundle");
36
- var import_registry = require("patchright-core/lib/server/registry/index");
37
- var import_utils = require("patchright-core/lib/utils");
38
- var import_http2 = require("../sdk/http");
39
- var import_log = require("../log");
40
- var protocol = __toESM(require("./protocol"));
41
- const debugLogger = (0, import_utilsBundle.debug)("pw:mcp:relay");
42
- class CDPRelayServer {
43
- constructor(server, browserChannel, userDataDir, executablePath) {
44
- this._playwrightConnection = null;
45
- this._extensionConnection = null;
46
- this._nextSessionId = 1;
47
- this._wsHost = (0, import_http2.addressToString)(server.address(), { protocol: "ws" });
48
- this._browserChannel = browserChannel;
49
- this._userDataDir = userDataDir;
50
- this._executablePath = executablePath;
51
- const uuid = crypto.randomUUID();
52
- this._cdpPath = `/cdp/${uuid}`;
53
- this._extensionPath = `/extension/${uuid}`;
54
- this._resetExtensionConnection();
55
- this._wss = new import_utilsBundle.wsServer({ server });
56
- this._wss.on("connection", this._onConnection.bind(this));
57
- }
58
- cdpEndpoint() {
59
- return `${this._wsHost}${this._cdpPath}`;
60
- }
61
- extensionEndpoint() {
62
- return `${this._wsHost}${this._extensionPath}`;
63
- }
64
- async ensureExtensionConnectionForMCPContext(clientInfo, abortSignal, toolName) {
65
- debugLogger("Ensuring extension connection for MCP context");
66
- if (this._extensionConnection)
67
- return;
68
- this._connectBrowser(clientInfo, toolName);
69
- debugLogger("Waiting for incoming extension connection");
70
- await Promise.race([
71
- this._extensionConnectionPromise,
72
- new Promise((_, reject) => setTimeout(() => {
73
- reject(new Error(`Extension connection timeout. Make sure the "Playwright MCP Bridge" extension is installed. See https://github.com/microsoft/playwright-mcp/blob/main/extension/README.md for installation instructions.`));
74
- }, process.env.PWMCP_TEST_CONNECTION_TIMEOUT ? parseInt(process.env.PWMCP_TEST_CONNECTION_TIMEOUT, 10) : 5e3)),
75
- new Promise((_, reject) => abortSignal.addEventListener("abort", reject))
76
- ]);
77
- debugLogger("Extension connection established");
78
- }
79
- _connectBrowser(clientInfo, toolName) {
80
- const mcpRelayEndpoint = `${this._wsHost}${this._extensionPath}`;
81
- const url = new URL("chrome-extension://jakfalbnbhgkpmoaakfflhflbfpkailf/connect.html");
82
- url.searchParams.set("mcpRelayUrl", mcpRelayEndpoint);
83
- const client = {
84
- name: clientInfo.name,
85
- version: clientInfo.version
86
- };
87
- url.searchParams.set("client", JSON.stringify(client));
88
- url.searchParams.set("protocolVersion", process.env.PWMCP_TEST_PROTOCOL_VERSION ?? protocol.VERSION.toString());
89
- if (toolName)
90
- url.searchParams.set("newTab", String(toolName === "browser_navigate"));
91
- const token = process.env.PLAYWRIGHT_MCP_EXTENSION_TOKEN;
92
- if (token)
93
- url.searchParams.set("token", token);
94
- const href = url.toString();
95
- let executablePath = this._executablePath;
96
- if (!executablePath) {
97
- const executableInfo = import_registry.registry.findExecutable(this._browserChannel);
98
- if (!executableInfo)
99
- throw new Error(`Unsupported channel: "${this._browserChannel}"`);
100
- executablePath = executableInfo.executablePath("javascript");
101
- if (!executablePath)
102
- throw new Error(`"${this._browserChannel}" executable not found. Make sure it is installed at a standard location.`);
103
- }
104
- const args = [];
105
- if (this._userDataDir)
106
- args.push(`--user-data-dir=${this._userDataDir}`);
107
- args.push(href);
108
- (0, import_child_process.spawn)(executablePath, args, {
109
- windowsHide: true,
110
- detached: true,
111
- shell: false,
112
- stdio: "ignore"
113
- });
114
- }
115
- stop() {
116
- this.closeConnections("Server stopped");
117
- this._wss.close();
118
- }
119
- closeConnections(reason) {
120
- this._closePlaywrightConnection(reason);
121
- this._closeExtensionConnection(reason);
122
- }
123
- _onConnection(ws2, request) {
124
- const url = new URL(`http://localhost${request.url}`);
125
- debugLogger(`New connection to ${url.pathname}`);
126
- if (url.pathname === this._cdpPath) {
127
- this._handlePlaywrightConnection(ws2);
128
- } else if (url.pathname === this._extensionPath) {
129
- this._handleExtensionConnection(ws2);
130
- } else {
131
- debugLogger(`Invalid path: ${url.pathname}`);
132
- ws2.close(4004, "Invalid path");
133
- }
134
- }
135
- _handlePlaywrightConnection(ws2) {
136
- if (this._playwrightConnection) {
137
- debugLogger("Rejecting second Playwright connection");
138
- ws2.close(1e3, "Another CDP client already connected");
139
- return;
140
- }
141
- this._playwrightConnection = ws2;
142
- ws2.on("message", async (data) => {
143
- try {
144
- const message = JSON.parse(data.toString());
145
- await this._handlePlaywrightMessage(message);
146
- } catch (error) {
147
- debugLogger(`Error while handling Playwright message
148
- ${data.toString()}
149
- `, error);
150
- }
151
- });
152
- ws2.on("close", () => {
153
- if (this._playwrightConnection !== ws2)
154
- return;
155
- this._playwrightConnection = null;
156
- this._closeExtensionConnection("Playwright client disconnected");
157
- debugLogger("Playwright WebSocket closed");
158
- });
159
- ws2.on("error", (error) => {
160
- debugLogger("Playwright WebSocket error:", error);
161
- });
162
- debugLogger("Playwright MCP connected");
163
- }
164
- _closeExtensionConnection(reason) {
165
- this._extensionConnection?.close(reason);
166
- this._extensionConnectionPromise.reject(new Error(reason));
167
- this._resetExtensionConnection();
168
- }
169
- _resetExtensionConnection() {
170
- this._connectedTabInfo = void 0;
171
- this._extensionConnection = null;
172
- this._extensionConnectionPromise = new import_utils.ManualPromise();
173
- void this._extensionConnectionPromise.catch(import_log.logUnhandledError);
174
- }
175
- _closePlaywrightConnection(reason) {
176
- if (this._playwrightConnection?.readyState === import_utilsBundle.ws.OPEN)
177
- this._playwrightConnection.close(1e3, reason);
178
- this._playwrightConnection = null;
179
- }
180
- _handleExtensionConnection(ws2) {
181
- if (this._extensionConnection) {
182
- ws2.close(1e3, "Another extension connection already established");
183
- return;
184
- }
185
- this._extensionConnection = new ExtensionConnection(ws2);
186
- this._extensionConnection.onclose = (c, reason) => {
187
- debugLogger("Extension WebSocket closed:", reason, c === this._extensionConnection);
188
- if (this._extensionConnection !== c)
189
- return;
190
- this._resetExtensionConnection();
191
- this._closePlaywrightConnection(`Extension disconnected: ${reason}`);
192
- };
193
- this._extensionConnection.onmessage = this._handleExtensionMessage.bind(this);
194
- this._extensionConnectionPromise.resolve();
195
- }
196
- _handleExtensionMessage(method, params) {
197
- switch (method) {
198
- case "forwardCDPEvent":
199
- const sessionId = params.sessionId || this._connectedTabInfo?.sessionId;
200
- this._sendToPlaywright({
201
- sessionId,
202
- method: params.method,
203
- params: params.params
204
- });
205
- break;
206
- }
207
- }
208
- async _handlePlaywrightMessage(message) {
209
- debugLogger("\u2190 Playwright:", `${message.method} (id=${message.id})`);
210
- const { id, sessionId, method, params } = message;
211
- try {
212
- const result = await this._handleCDPCommand(method, params, sessionId);
213
- this._sendToPlaywright({ id, sessionId, result });
214
- } catch (e) {
215
- debugLogger("Error in the extension:", e);
216
- this._sendToPlaywright({
217
- id,
218
- sessionId,
219
- error: { message: e.message }
220
- });
221
- }
222
- }
223
- async _handleCDPCommand(method, params, sessionId) {
224
- switch (method) {
225
- case "Browser.getVersion": {
226
- return {
227
- protocolVersion: "1.3",
228
- product: "Chrome/Extension-Bridge",
229
- userAgent: "CDP-Bridge-Server/1.0.0"
230
- };
231
- }
232
- case "Browser.setDownloadBehavior": {
233
- return {};
234
- }
235
- case "Target.setAutoAttach": {
236
- if (sessionId)
237
- break;
238
- const { targetInfo } = await this._extensionConnection.send("attachToTab", {});
239
- this._connectedTabInfo = {
240
- targetInfo,
241
- sessionId: `pw-tab-${this._nextSessionId++}`
242
- };
243
- debugLogger("Simulating auto-attach");
244
- this._sendToPlaywright({
245
- method: "Target.attachedToTarget",
246
- params: {
247
- sessionId: this._connectedTabInfo.sessionId,
248
- targetInfo: {
249
- ...this._connectedTabInfo.targetInfo,
250
- attached: true
251
- },
252
- waitingForDebugger: false
253
- }
254
- });
255
- return {};
256
- }
257
- case "Target.getTargetInfo": {
258
- return this._connectedTabInfo?.targetInfo;
259
- }
260
- }
261
- return await this._forwardToExtension(method, params, sessionId);
262
- }
263
- async _forwardToExtension(method, params, sessionId) {
264
- if (!this._extensionConnection)
265
- throw new Error("Extension not connected");
266
- if (this._connectedTabInfo?.sessionId === sessionId)
267
- sessionId = void 0;
268
- return await this._extensionConnection.send("forwardCDPCommand", { sessionId, method, params });
269
- }
270
- _sendToPlaywright(message) {
271
- debugLogger("\u2192 Playwright:", `${message.method ?? `response(id=${message.id})`}`);
272
- this._playwrightConnection?.send(JSON.stringify(message));
273
- }
274
- }
275
- class ExtensionConnection {
276
- constructor(ws2) {
277
- this._callbacks = /* @__PURE__ */ new Map();
278
- this._lastId = 0;
279
- this._ws = ws2;
280
- this._ws.on("message", this._onMessage.bind(this));
281
- this._ws.on("close", this._onClose.bind(this));
282
- this._ws.on("error", this._onError.bind(this));
283
- }
284
- async send(method, params) {
285
- if (this._ws.readyState !== import_utilsBundle.ws.OPEN)
286
- throw new Error(`Unexpected WebSocket state: ${this._ws.readyState}`);
287
- const id = ++this._lastId;
288
- this._ws.send(JSON.stringify({ id, method, params }));
289
- const error = new Error(`Protocol error: ${method}`);
290
- return new Promise((resolve, reject) => {
291
- this._callbacks.set(id, { resolve, reject, error });
292
- });
293
- }
294
- close(message) {
295
- debugLogger("closing extension connection:", message);
296
- if (this._ws.readyState === import_utilsBundle.ws.OPEN)
297
- this._ws.close(1e3, message);
298
- }
299
- _onMessage(event) {
300
- const eventData = event.toString();
301
- let parsedJson;
302
- try {
303
- parsedJson = JSON.parse(eventData);
304
- } catch (e) {
305
- debugLogger(`<closing ws> Closing websocket due to malformed JSON. eventData=${eventData} e=${e?.message}`);
306
- this._ws.close();
307
- return;
308
- }
309
- try {
310
- this._handleParsedMessage(parsedJson);
311
- } catch (e) {
312
- debugLogger(`<closing ws> Closing websocket due to failed onmessage callback. eventData=${eventData} e=${e?.message}`);
313
- this._ws.close();
314
- }
315
- }
316
- _handleParsedMessage(object) {
317
- if (object.id && this._callbacks.has(object.id)) {
318
- const callback = this._callbacks.get(object.id);
319
- this._callbacks.delete(object.id);
320
- if (object.error) {
321
- const error = callback.error;
322
- error.message = object.error;
323
- callback.reject(error);
324
- } else {
325
- callback.resolve(object.result);
326
- }
327
- } else if (object.id) {
328
- debugLogger("\u2190 Extension: unexpected response", object);
329
- } else {
330
- this.onmessage?.(object.method, object.params);
331
- }
332
- }
333
- _onClose(event) {
334
- debugLogger(`<ws closed> code=${event.code} reason=${event.reason}`);
335
- this._dispose();
336
- this.onclose?.(this, event.reason);
337
- }
338
- _onError(event) {
339
- debugLogger(`<ws error> message=${event.message} type=${event.type} target=${event.target}`);
340
- this._dispose();
341
- }
342
- _dispose() {
343
- for (const callback of this._callbacks.values())
344
- callback.reject(new Error("WebSocket closed"));
345
- this._callbacks.clear();
346
- }
347
- }
348
- // Annotate the CommonJS export names for ESM import in node:
349
- 0 && (module.exports = {
350
- CDPRelayServer
351
- });
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 cdpRelay_exports = {};
30
+ __export(cdpRelay_exports, {
31
+ CDPRelayServer: () => CDPRelayServer
32
+ });
33
+ module.exports = __toCommonJS(cdpRelay_exports);
34
+ var import_child_process = require("child_process");
35
+ var import_utilsBundle = require("patchright-bun-core/lib/utilsBundle");
36
+ var import_registry = require("patchright-bun-core/lib/server/registry/index");
37
+ var import_utils = require("patchright-bun-core/lib/utils");
38
+ var import_http2 = require("../sdk/http");
39
+ var import_log = require("../log");
40
+ var protocol = __toESM(require("./protocol"));
41
+ const debugLogger = (0, import_utilsBundle.debug)("pw:mcp:relay");
42
+ class CDPRelayServer {
43
+ constructor(server, browserChannel, userDataDir, executablePath) {
44
+ this._playwrightConnection = null;
45
+ this._extensionConnection = null;
46
+ this._nextSessionId = 1;
47
+ this._wsHost = (0, import_http2.addressToString)(server.address(), { protocol: "ws" });
48
+ this._browserChannel = browserChannel;
49
+ this._userDataDir = userDataDir;
50
+ this._executablePath = executablePath;
51
+ const uuid = crypto.randomUUID();
52
+ this._cdpPath = `/cdp/${uuid}`;
53
+ this._extensionPath = `/extension/${uuid}`;
54
+ this._resetExtensionConnection();
55
+ this._wss = new import_utilsBundle.wsServer({ server });
56
+ this._wss.on("connection", this._onConnection.bind(this));
57
+ }
58
+ cdpEndpoint() {
59
+ return `${this._wsHost}${this._cdpPath}`;
60
+ }
61
+ extensionEndpoint() {
62
+ return `${this._wsHost}${this._extensionPath}`;
63
+ }
64
+ async ensureExtensionConnectionForMCPContext(clientInfo, abortSignal, toolName) {
65
+ debugLogger("Ensuring extension connection for MCP context");
66
+ if (this._extensionConnection)
67
+ return;
68
+ this._connectBrowser(clientInfo, toolName);
69
+ debugLogger("Waiting for incoming extension connection");
70
+ await Promise.race([
71
+ this._extensionConnectionPromise,
72
+ new Promise((_, reject) => setTimeout(() => {
73
+ reject(new Error(`Extension connection timeout. Make sure the "Playwright MCP Bridge" extension is installed. See https://github.com/microsoft/playwright-mcp/blob/main/extension/README.md for installation instructions.`));
74
+ }, process.env.PWMCP_TEST_CONNECTION_TIMEOUT ? parseInt(process.env.PWMCP_TEST_CONNECTION_TIMEOUT, 10) : 5e3)),
75
+ new Promise((_, reject) => abortSignal.addEventListener("abort", reject))
76
+ ]);
77
+ debugLogger("Extension connection established");
78
+ }
79
+ _connectBrowser(clientInfo, toolName) {
80
+ const mcpRelayEndpoint = `${this._wsHost}${this._extensionPath}`;
81
+ const url = new URL("chrome-extension://jakfalbnbhgkpmoaakfflhflbfpkailf/connect.html");
82
+ url.searchParams.set("mcpRelayUrl", mcpRelayEndpoint);
83
+ const client = {
84
+ name: clientInfo.name,
85
+ version: clientInfo.version
86
+ };
87
+ url.searchParams.set("client", JSON.stringify(client));
88
+ url.searchParams.set("protocolVersion", process.env.PWMCP_TEST_PROTOCOL_VERSION ?? protocol.VERSION.toString());
89
+ if (toolName)
90
+ url.searchParams.set("newTab", String(toolName === "browser_navigate"));
91
+ const token = process.env.PLAYWRIGHT_MCP_EXTENSION_TOKEN;
92
+ if (token)
93
+ url.searchParams.set("token", token);
94
+ const href = url.toString();
95
+ let executablePath = this._executablePath;
96
+ if (!executablePath) {
97
+ const executableInfo = import_registry.registry.findExecutable(this._browserChannel);
98
+ if (!executableInfo)
99
+ throw new Error(`Unsupported channel: "${this._browserChannel}"`);
100
+ executablePath = executableInfo.executablePath("javascript");
101
+ if (!executablePath)
102
+ throw new Error(`"${this._browserChannel}" executable not found. Make sure it is installed at a standard location.`);
103
+ }
104
+ const args = [];
105
+ if (this._userDataDir)
106
+ args.push(`--user-data-dir=${this._userDataDir}`);
107
+ args.push(href);
108
+ (0, import_child_process.spawn)(executablePath, args, {
109
+ windowsHide: true,
110
+ detached: true,
111
+ shell: false,
112
+ stdio: "ignore"
113
+ });
114
+ }
115
+ stop() {
116
+ this.closeConnections("Server stopped");
117
+ this._wss.close();
118
+ }
119
+ closeConnections(reason) {
120
+ this._closePlaywrightConnection(reason);
121
+ this._closeExtensionConnection(reason);
122
+ }
123
+ _onConnection(ws2, request) {
124
+ const url = new URL(`http://localhost${request.url}`);
125
+ debugLogger(`New connection to ${url.pathname}`);
126
+ if (url.pathname === this._cdpPath) {
127
+ this._handlePlaywrightConnection(ws2);
128
+ } else if (url.pathname === this._extensionPath) {
129
+ this._handleExtensionConnection(ws2);
130
+ } else {
131
+ debugLogger(`Invalid path: ${url.pathname}`);
132
+ ws2.close(4004, "Invalid path");
133
+ }
134
+ }
135
+ _handlePlaywrightConnection(ws2) {
136
+ if (this._playwrightConnection) {
137
+ debugLogger("Rejecting second Playwright connection");
138
+ ws2.close(1e3, "Another CDP client already connected");
139
+ return;
140
+ }
141
+ this._playwrightConnection = ws2;
142
+ ws2.on("message", async (data) => {
143
+ try {
144
+ const message = JSON.parse(data.toString());
145
+ await this._handlePlaywrightMessage(message);
146
+ } catch (error) {
147
+ debugLogger(`Error while handling Playwright message
148
+ ${data.toString()}
149
+ `, error);
150
+ }
151
+ });
152
+ ws2.on("close", () => {
153
+ if (this._playwrightConnection !== ws2)
154
+ return;
155
+ this._playwrightConnection = null;
156
+ this._closeExtensionConnection("Playwright client disconnected");
157
+ debugLogger("Playwright WebSocket closed");
158
+ });
159
+ ws2.on("error", (error) => {
160
+ debugLogger("Playwright WebSocket error:", error);
161
+ });
162
+ debugLogger("Playwright MCP connected");
163
+ }
164
+ _closeExtensionConnection(reason) {
165
+ this._extensionConnection?.close(reason);
166
+ this._extensionConnectionPromise.reject(new Error(reason));
167
+ this._resetExtensionConnection();
168
+ }
169
+ _resetExtensionConnection() {
170
+ this._connectedTabInfo = void 0;
171
+ this._extensionConnection = null;
172
+ this._extensionConnectionPromise = new import_utils.ManualPromise();
173
+ void this._extensionConnectionPromise.catch(import_log.logUnhandledError);
174
+ }
175
+ _closePlaywrightConnection(reason) {
176
+ if (this._playwrightConnection?.readyState === import_utilsBundle.ws.OPEN)
177
+ this._playwrightConnection.close(1e3, reason);
178
+ this._playwrightConnection = null;
179
+ }
180
+ _handleExtensionConnection(ws2) {
181
+ if (this._extensionConnection) {
182
+ ws2.close(1e3, "Another extension connection already established");
183
+ return;
184
+ }
185
+ this._extensionConnection = new ExtensionConnection(ws2);
186
+ this._extensionConnection.onclose = (c, reason) => {
187
+ debugLogger("Extension WebSocket closed:", reason, c === this._extensionConnection);
188
+ if (this._extensionConnection !== c)
189
+ return;
190
+ this._resetExtensionConnection();
191
+ this._closePlaywrightConnection(`Extension disconnected: ${reason}`);
192
+ };
193
+ this._extensionConnection.onmessage = this._handleExtensionMessage.bind(this);
194
+ this._extensionConnectionPromise.resolve();
195
+ }
196
+ _handleExtensionMessage(method, params) {
197
+ switch (method) {
198
+ case "forwardCDPEvent":
199
+ const sessionId = params.sessionId || this._connectedTabInfo?.sessionId;
200
+ this._sendToPlaywright({
201
+ sessionId,
202
+ method: params.method,
203
+ params: params.params
204
+ });
205
+ break;
206
+ }
207
+ }
208
+ async _handlePlaywrightMessage(message) {
209
+ debugLogger("\u2190 Playwright:", `${message.method} (id=${message.id})`);
210
+ const { id, sessionId, method, params } = message;
211
+ try {
212
+ const result = await this._handleCDPCommand(method, params, sessionId);
213
+ this._sendToPlaywright({ id, sessionId, result });
214
+ } catch (e) {
215
+ debugLogger("Error in the extension:", e);
216
+ this._sendToPlaywright({
217
+ id,
218
+ sessionId,
219
+ error: { message: e.message }
220
+ });
221
+ }
222
+ }
223
+ async _handleCDPCommand(method, params, sessionId) {
224
+ switch (method) {
225
+ case "Browser.getVersion": {
226
+ return {
227
+ protocolVersion: "1.3",
228
+ product: "Chrome/Extension-Bridge",
229
+ userAgent: "CDP-Bridge-Server/1.0.0"
230
+ };
231
+ }
232
+ case "Browser.setDownloadBehavior": {
233
+ return {};
234
+ }
235
+ case "Target.setAutoAttach": {
236
+ if (sessionId)
237
+ break;
238
+ const { targetInfo } = await this._extensionConnection.send("attachToTab", {});
239
+ this._connectedTabInfo = {
240
+ targetInfo,
241
+ sessionId: `pw-tab-${this._nextSessionId++}`
242
+ };
243
+ debugLogger("Simulating auto-attach");
244
+ this._sendToPlaywright({
245
+ method: "Target.attachedToTarget",
246
+ params: {
247
+ sessionId: this._connectedTabInfo.sessionId,
248
+ targetInfo: {
249
+ ...this._connectedTabInfo.targetInfo,
250
+ attached: true
251
+ },
252
+ waitingForDebugger: false
253
+ }
254
+ });
255
+ return {};
256
+ }
257
+ case "Target.getTargetInfo": {
258
+ return this._connectedTabInfo?.targetInfo;
259
+ }
260
+ }
261
+ return await this._forwardToExtension(method, params, sessionId);
262
+ }
263
+ async _forwardToExtension(method, params, sessionId) {
264
+ if (!this._extensionConnection)
265
+ throw new Error("Extension not connected");
266
+ if (this._connectedTabInfo?.sessionId === sessionId)
267
+ sessionId = void 0;
268
+ return await this._extensionConnection.send("forwardCDPCommand", { sessionId, method, params });
269
+ }
270
+ _sendToPlaywright(message) {
271
+ debugLogger("\u2192 Playwright:", `${message.method ?? `response(id=${message.id})`}`);
272
+ this._playwrightConnection?.send(JSON.stringify(message));
273
+ }
274
+ }
275
+ class ExtensionConnection {
276
+ constructor(ws2) {
277
+ this._callbacks = /* @__PURE__ */ new Map();
278
+ this._lastId = 0;
279
+ this._ws = ws2;
280
+ this._ws.on("message", this._onMessage.bind(this));
281
+ this._ws.on("close", this._onClose.bind(this));
282
+ this._ws.on("error", this._onError.bind(this));
283
+ }
284
+ async send(method, params) {
285
+ if (this._ws.readyState !== import_utilsBundle.ws.OPEN)
286
+ throw new Error(`Unexpected WebSocket state: ${this._ws.readyState}`);
287
+ const id = ++this._lastId;
288
+ this._ws.send(JSON.stringify({ id, method, params }));
289
+ const error = new Error(`Protocol error: ${method}`);
290
+ return new Promise((resolve, reject) => {
291
+ this._callbacks.set(id, { resolve, reject, error });
292
+ });
293
+ }
294
+ close(message) {
295
+ debugLogger("closing extension connection:", message);
296
+ if (this._ws.readyState === import_utilsBundle.ws.OPEN)
297
+ this._ws.close(1e3, message);
298
+ }
299
+ _onMessage(event) {
300
+ const eventData = event.toString();
301
+ let parsedJson;
302
+ try {
303
+ parsedJson = JSON.parse(eventData);
304
+ } catch (e) {
305
+ debugLogger(`<closing ws> Closing websocket due to malformed JSON. eventData=${eventData} e=${e?.message}`);
306
+ this._ws.close();
307
+ return;
308
+ }
309
+ try {
310
+ this._handleParsedMessage(parsedJson);
311
+ } catch (e) {
312
+ debugLogger(`<closing ws> Closing websocket due to failed onmessage callback. eventData=${eventData} e=${e?.message}`);
313
+ this._ws.close();
314
+ }
315
+ }
316
+ _handleParsedMessage(object) {
317
+ if (object.id && this._callbacks.has(object.id)) {
318
+ const callback = this._callbacks.get(object.id);
319
+ this._callbacks.delete(object.id);
320
+ if (object.error) {
321
+ const error = callback.error;
322
+ error.message = object.error;
323
+ callback.reject(error);
324
+ } else {
325
+ callback.resolve(object.result);
326
+ }
327
+ } else if (object.id) {
328
+ debugLogger("\u2190 Extension: unexpected response", object);
329
+ } else {
330
+ this.onmessage?.(object.method, object.params);
331
+ }
332
+ }
333
+ _onClose(event) {
334
+ debugLogger(`<ws closed> code=${event.code} reason=${event.reason}`);
335
+ this._dispose();
336
+ this.onclose?.(this, event.reason);
337
+ }
338
+ _onError(event) {
339
+ debugLogger(`<ws error> message=${event.message} type=${event.type} target=${event.target}`);
340
+ this._dispose();
341
+ }
342
+ _dispose() {
343
+ for (const callback of this._callbacks.values())
344
+ callback.reject(new Error("WebSocket closed"));
345
+ this._callbacks.clear();
346
+ }
347
+ }
348
+ // Annotate the CommonJS export names for ESM import in node:
349
+ 0 && (module.exports = {
350
+ CDPRelayServer
351
+ });