|
1
|
"use strict"; |
|
2
|
var __defProp = Object.defineProperty; |
|
3
|
var __getOwnPropDesc = Object.getOwnPropertyDescriptor; |
|
4
|
var __getOwnPropNames = Object.getOwnPropertyNames; |
|
5
|
var __hasOwnProp = Object.prototype.hasOwnProperty; |
|
6
|
var __export = (target, all) => { |
|
7
|
for (var name in all) |
|
8
|
__defProp(target, name, { get: all[name], enumerable: true }); |
|
9
|
}; |
|
10
|
var __copyProps = (to, from, except, desc) => { |
|
11
|
if (from && typeof from === "object" || typeof from === "function") { |
|
12
|
for (let key of __getOwnPropNames(from)) |
|
13
|
if (!__hasOwnProp.call(to, key) && key !== except) |
|
14
|
__defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable }); |
|
15
|
} |
|
16
|
return to; |
|
17
|
}; |
|
18
|
var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod); |
|
19
|
var browserBackend_exports = {}; |
|
20
|
__export(browserBackend_exports, { |
|
21
|
createCustomMessageHandler: () => createCustomMessageHandler |
|
22
|
}); |
|
23
|
module.exports = __toCommonJS(browserBackend_exports); |
|
24
|
var import_config = require("../browser/config"); |
|
25
|
var import_browserServerBackend = require("../browser/browserServerBackend"); |
|
26
|
var import_tab = require("../browser/tab"); |
|
27
|
var import_util = require("../../util"); |
|
28
|
var import_browserContextFactory = require("../browser/browserContextFactory"); |
|
29
|
function createCustomMessageHandler(testInfo, context) { |
|
30
|
let backend; |
|
31
|
return async (data) => { |
|
32
|
if (data.initialize) { |
|
33
|
if (backend) |
|
34
|
throw new Error("MCP backend is already initialized"); |
|
35
|
backend = new import_browserServerBackend.BrowserServerBackend({ ...import_config.defaultConfig, capabilities: ["testing"] }, (0, import_browserContextFactory.identityBrowserContextFactory)(context)); |
|
36
|
await backend.initialize(data.initialize.clientInfo); |
|
37
|
const pausedMessage = await generatePausedMessage(testInfo, context); |
|
38
|
return { initialize: { pausedMessage } }; |
|
39
|
} |
|
40
|
if (data.listTools) { |
|
41
|
if (!backend) |
|
42
|
throw new Error("MCP backend is not initialized"); |
|
43
|
return { listTools: await backend.listTools() }; |
|
44
|
} |
|
45
|
if (data.callTool) { |
|
46
|
if (!backend) |
|
47
|
throw new Error("MCP backend is not initialized"); |
|
48
|
return { callTool: await backend.callTool(data.callTool.name, data.callTool.arguments) }; |
|
49
|
} |
|
50
|
if (data.close) { |
|
51
|
backend?.serverClosed(); |
|
52
|
backend = void 0; |
|
53
|
return { close: {} }; |
|
54
|
} |
|
55
|
throw new Error("Unknown MCP request"); |
|
56
|
}; |
|
57
|
} |
|
58
|
async function generatePausedMessage(testInfo, context) { |
|
59
|
const lines = []; |
|
60
|
if (testInfo.errors.length) { |
|
61
|
lines.push(`### Paused on error:`); |
|
62
|
for (const error of testInfo.errors) |
|
63
|
lines.push((0, import_util.stripAnsiEscapes)(error.message || "")); |
|
64
|
} else { |
|
65
|
lines.push(`### Paused at end of test. ready for interaction`); |
|
66
|
} |
|
67
|
for (let i = 0; i < context.pages().length; i++) { |
|
68
|
const page = context.pages()[i]; |
|
69
|
const stateSuffix = context.pages().length > 1 ? i + 1 + " of " + context.pages().length : "state"; |
|
70
|
lines.push( |
|
71
|
"", |
|
72
|
`### Page ${stateSuffix}`, |
|
73
|
`- Page URL: ${page.url()}`, |
|
74
|
`- Page Title: ${await page.title()}`.trim() |
|
75
|
); |
|
76
|
let console = testInfo.errors.length ? await import_tab.Tab.collectConsoleMessages(page) : []; |
|
77
|
console = console.filter((msg) => msg.type === "error"); |
|
78
|
if (console.length) { |
|
79
|
lines.push("- Console Messages:"); |
|
80
|
for (const message of console) |
|
81
|
lines.push(` - ${message.toString()}`); |
|
82
|
} |
|
83
|
lines.push( |
|
84
|
`- Page Snapshot:`, |
|
85
|
"```yaml", |
|
86
|
(await page._snapshotForAI()).full, |
|
87
|
"```" |
|
88
|
); |
|
89
|
} |
|
90
|
lines.push(""); |
|
91
|
if (testInfo.errors.length) |
|
92
|
lines.push(`### Task`, `Try recovering from the error prior to continuing`); |
|
93
|
return lines.join("\n"); |
|
94
|
} |
|
95
|
// Annotate the CommonJS export names for ESM import in node: |
|
96
|
0 && (module.exports = { |
|
97
|
createCustomMessageHandler |
|
98
|
}); |
|
99
|
|