@tontoko/fast-playwright-mcp 0.0.4
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.
- package/LICENSE +202 -0
- package/README.md +1047 -0
- package/cli.js +18 -0
- package/config.d.ts +124 -0
- package/index.d.ts +25 -0
- package/index.js +18 -0
- package/lib/actions.d.js +0 -0
- package/lib/batch/batch-executor.js +137 -0
- package/lib/browser-context-factory.js +252 -0
- package/lib/browser-server-backend.js +139 -0
- package/lib/config/constants.js +80 -0
- package/lib/config.js +405 -0
- package/lib/context.js +274 -0
- package/lib/diagnostics/common/diagnostic-base.js +63 -0
- package/lib/diagnostics/common/error-enrichment-utils.js +212 -0
- package/lib/diagnostics/common/index.js +56 -0
- package/lib/diagnostics/common/initialization-manager.js +210 -0
- package/lib/diagnostics/common/performance-tracker.js +132 -0
- package/lib/diagnostics/diagnostic-error.js +140 -0
- package/lib/diagnostics/diagnostic-level.js +123 -0
- package/lib/diagnostics/diagnostic-thresholds.js +347 -0
- package/lib/diagnostics/element-discovery.js +441 -0
- package/lib/diagnostics/enhanced-error-handler.js +376 -0
- package/lib/diagnostics/error-enrichment.js +157 -0
- package/lib/diagnostics/frame-reference-manager.js +179 -0
- package/lib/diagnostics/page-analyzer.js +639 -0
- package/lib/diagnostics/parallel-page-analyzer.js +129 -0
- package/lib/diagnostics/resource-manager.js +134 -0
- package/lib/diagnostics/smart-config.js +482 -0
- package/lib/diagnostics/smart-handle.js +118 -0
- package/lib/diagnostics/unified-system.js +717 -0
- package/lib/extension/cdp-relay.js +486 -0
- package/lib/extension/extension-context-factory.js +74 -0
- package/lib/extension/main.js +41 -0
- package/lib/file-utils.js +42 -0
- package/lib/generate-keys.js +75 -0
- package/lib/http-server.js +50 -0
- package/lib/in-process-client.js +64 -0
- package/lib/index.js +48 -0
- package/lib/javascript.js +90 -0
- package/lib/log.js +33 -0
- package/lib/loop/loop-claude.js +247 -0
- package/lib/loop/loop-open-ai.js +222 -0
- package/lib/loop/loop.js +174 -0
- package/lib/loop/main.js +46 -0
- package/lib/loopTools/context.js +76 -0
- package/lib/loopTools/main.js +65 -0
- package/lib/loopTools/perform.js +40 -0
- package/lib/loopTools/snapshot.js +37 -0
- package/lib/loopTools/tool.js +26 -0
- package/lib/manual-promise.js +125 -0
- package/lib/mcp/in-process-transport.js +91 -0
- package/lib/mcp/proxy-backend.js +127 -0
- package/lib/mcp/server.js +123 -0
- package/lib/mcp/transport.js +159 -0
- package/lib/package.js +28 -0
- package/lib/program.js +82 -0
- package/lib/response.js +493 -0
- package/lib/schemas/expectation.js +152 -0
- package/lib/session-log.js +210 -0
- package/lib/tab.js +417 -0
- package/lib/tools/base-tool-handler.js +141 -0
- package/lib/tools/batch-execute.js +150 -0
- package/lib/tools/common.js +65 -0
- package/lib/tools/console.js +60 -0
- package/lib/tools/diagnose/diagnose-analysis-runner.js +101 -0
- package/lib/tools/diagnose/diagnose-config-handler.js +130 -0
- package/lib/tools/diagnose/diagnose-report-builder.js +394 -0
- package/lib/tools/diagnose.js +147 -0
- package/lib/tools/dialogs.js +57 -0
- package/lib/tools/evaluate.js +67 -0
- package/lib/tools/files.js +53 -0
- package/lib/tools/find-elements.js +307 -0
- package/lib/tools/install.js +60 -0
- package/lib/tools/keyboard.js +93 -0
- package/lib/tools/mouse.js +110 -0
- package/lib/tools/navigate.js +82 -0
- package/lib/tools/network.js +50 -0
- package/lib/tools/pdf.js +46 -0
- package/lib/tools/screenshot.js +113 -0
- package/lib/tools/snapshot.js +158 -0
- package/lib/tools/tabs.js +97 -0
- package/lib/tools/tool.js +47 -0
- package/lib/tools/utils.js +131 -0
- package/lib/tools/wait.js +64 -0
- package/lib/tools.js +65 -0
- package/lib/types/batch.js +47 -0
- package/lib/types/diff.js +0 -0
- package/lib/types/performance.js +0 -0
- package/lib/types/threshold-base.js +0 -0
- package/lib/utils/array-utils.js +44 -0
- package/lib/utils/code-deduplication-utils.js +141 -0
- package/lib/utils/common-formatters.js +252 -0
- package/lib/utils/console-filter.js +64 -0
- package/lib/utils/diagnostic-report-utils.js +178 -0
- package/lib/utils/diff-formatter.js +126 -0
- package/lib/utils/disposable-manager.js +135 -0
- package/lib/utils/error-handler-middleware.js +77 -0
- package/lib/utils/image-processor.js +137 -0
- package/lib/utils/index.js +92 -0
- package/lib/utils/report-builder.js +189 -0
- package/lib/utils/request-logger.js +82 -0
- package/lib/utils/response-diff-detector.js +150 -0
- package/lib/utils/section-builder.js +62 -0
- package/lib/utils/tool-patterns.js +153 -0
- package/lib/utils.js +46 -0
- package/package.json +77 -0
|
@@ -0,0 +1,125 @@
|
|
|
1
|
+
import { createRequire } from "node:module";
|
|
2
|
+
var __create = Object.create;
|
|
3
|
+
var __getProtoOf = Object.getPrototypeOf;
|
|
4
|
+
var __defProp = Object.defineProperty;
|
|
5
|
+
var __getOwnPropNames = Object.getOwnPropertyNames;
|
|
6
|
+
var __hasOwnProp = Object.prototype.hasOwnProperty;
|
|
7
|
+
var __toESM = (mod, isNodeMode, target) => {
|
|
8
|
+
target = mod != null ? __create(__getProtoOf(mod)) : {};
|
|
9
|
+
const to = isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target;
|
|
10
|
+
for (let key of __getOwnPropNames(mod))
|
|
11
|
+
if (!__hasOwnProp.call(to, key))
|
|
12
|
+
__defProp(to, key, {
|
|
13
|
+
get: () => mod[key],
|
|
14
|
+
enumerable: true
|
|
15
|
+
});
|
|
16
|
+
return to;
|
|
17
|
+
};
|
|
18
|
+
var __require = /* @__PURE__ */ createRequire(import.meta.url);
|
|
19
|
+
|
|
20
|
+
// src/manual-promise.ts
|
|
21
|
+
class ManualPromise extends Promise {
|
|
22
|
+
_resolve;
|
|
23
|
+
_reject;
|
|
24
|
+
_isDone;
|
|
25
|
+
constructor() {
|
|
26
|
+
let resolve;
|
|
27
|
+
let reject;
|
|
28
|
+
super((res, rej) => {
|
|
29
|
+
resolve = res;
|
|
30
|
+
reject = rej;
|
|
31
|
+
});
|
|
32
|
+
this._isDone = false;
|
|
33
|
+
this._resolve = resolve;
|
|
34
|
+
this._reject = reject;
|
|
35
|
+
}
|
|
36
|
+
isDone() {
|
|
37
|
+
return this._isDone;
|
|
38
|
+
}
|
|
39
|
+
resolve(t) {
|
|
40
|
+
this._isDone = true;
|
|
41
|
+
this._resolve(t);
|
|
42
|
+
}
|
|
43
|
+
reject(e) {
|
|
44
|
+
this._isDone = true;
|
|
45
|
+
this._reject(e);
|
|
46
|
+
}
|
|
47
|
+
static get [Symbol.species]() {
|
|
48
|
+
return Promise;
|
|
49
|
+
}
|
|
50
|
+
get [Symbol.toStringTag]() {
|
|
51
|
+
return "ManualPromise";
|
|
52
|
+
}
|
|
53
|
+
}
|
|
54
|
+
|
|
55
|
+
class LongStandingScope {
|
|
56
|
+
_terminateError;
|
|
57
|
+
_closeError;
|
|
58
|
+
_terminatePromises = new Map;
|
|
59
|
+
_isClosed = false;
|
|
60
|
+
reject(error) {
|
|
61
|
+
this._isClosed = true;
|
|
62
|
+
this._terminateError = error;
|
|
63
|
+
for (const p of this._terminatePromises.keys()) {
|
|
64
|
+
p.resolve(error);
|
|
65
|
+
}
|
|
66
|
+
}
|
|
67
|
+
close(error) {
|
|
68
|
+
this._isClosed = true;
|
|
69
|
+
this._closeError = error;
|
|
70
|
+
for (const [p, frames] of this._terminatePromises) {
|
|
71
|
+
p.resolve(cloneError(error, frames));
|
|
72
|
+
}
|
|
73
|
+
}
|
|
74
|
+
isClosed() {
|
|
75
|
+
return this._isClosed;
|
|
76
|
+
}
|
|
77
|
+
static raceMultiple(scopes, promise) {
|
|
78
|
+
return Promise.race(scopes.map((s) => s.race(promise)));
|
|
79
|
+
}
|
|
80
|
+
race(promise) {
|
|
81
|
+
return this._race(Array.isArray(promise) ? promise : [promise], false);
|
|
82
|
+
}
|
|
83
|
+
safeRace(promise, defaultValue) {
|
|
84
|
+
return this._race([promise], true, defaultValue);
|
|
85
|
+
}
|
|
86
|
+
async _race(promises, safe, defaultValue) {
|
|
87
|
+
const terminatePromise = new ManualPromise;
|
|
88
|
+
const frames = captureRawStack();
|
|
89
|
+
if (this._terminateError) {
|
|
90
|
+
terminatePromise.resolve(this._terminateError);
|
|
91
|
+
}
|
|
92
|
+
if (this._closeError) {
|
|
93
|
+
terminatePromise.resolve(cloneError(this._closeError, frames));
|
|
94
|
+
}
|
|
95
|
+
this._terminatePromises.set(terminatePromise, frames);
|
|
96
|
+
try {
|
|
97
|
+
return await Promise.race([
|
|
98
|
+
terminatePromise.then((e) => safe ? defaultValue : Promise.reject(e)),
|
|
99
|
+
...promises
|
|
100
|
+
]);
|
|
101
|
+
} finally {
|
|
102
|
+
this._terminatePromises.delete(terminatePromise);
|
|
103
|
+
}
|
|
104
|
+
}
|
|
105
|
+
}
|
|
106
|
+
function cloneError(error, frames) {
|
|
107
|
+
const clone = new Error(error.message);
|
|
108
|
+
clone.name = error.name;
|
|
109
|
+
clone.stack = [`${error.name}:${error.message}`, ...frames].join(`
|
|
110
|
+
`);
|
|
111
|
+
return clone;
|
|
112
|
+
}
|
|
113
|
+
function captureRawStack() {
|
|
114
|
+
const stackTraceLimit = Error.stackTraceLimit;
|
|
115
|
+
Error.stackTraceLimit = 50;
|
|
116
|
+
const error = new Error("Stack trace capture");
|
|
117
|
+
const stack = error.stack ?? "";
|
|
118
|
+
Error.stackTraceLimit = stackTraceLimit;
|
|
119
|
+
return stack.split(`
|
|
120
|
+
`);
|
|
121
|
+
}
|
|
122
|
+
export {
|
|
123
|
+
ManualPromise,
|
|
124
|
+
LongStandingScope
|
|
125
|
+
};
|
|
@@ -0,0 +1,91 @@
|
|
|
1
|
+
import { createRequire } from "node:module";
|
|
2
|
+
var __create = Object.create;
|
|
3
|
+
var __getProtoOf = Object.getPrototypeOf;
|
|
4
|
+
var __defProp = Object.defineProperty;
|
|
5
|
+
var __getOwnPropNames = Object.getOwnPropertyNames;
|
|
6
|
+
var __hasOwnProp = Object.prototype.hasOwnProperty;
|
|
7
|
+
var __toESM = (mod, isNodeMode, target) => {
|
|
8
|
+
target = mod != null ? __create(__getProtoOf(mod)) : {};
|
|
9
|
+
const to = isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target;
|
|
10
|
+
for (let key of __getOwnPropNames(mod))
|
|
11
|
+
if (!__hasOwnProp.call(to, key))
|
|
12
|
+
__defProp(to, key, {
|
|
13
|
+
get: () => mod[key],
|
|
14
|
+
enumerable: true
|
|
15
|
+
});
|
|
16
|
+
return to;
|
|
17
|
+
};
|
|
18
|
+
var __require = /* @__PURE__ */ createRequire(import.meta.url);
|
|
19
|
+
|
|
20
|
+
// src/mcp/in-process-transport.ts
|
|
21
|
+
class InProcessTransport {
|
|
22
|
+
_server;
|
|
23
|
+
_serverTransport;
|
|
24
|
+
_connected = false;
|
|
25
|
+
constructor(server) {
|
|
26
|
+
this._server = server;
|
|
27
|
+
this._serverTransport = new InProcessServerTransport(this);
|
|
28
|
+
}
|
|
29
|
+
async start() {
|
|
30
|
+
if (this._connected) {
|
|
31
|
+
throw new Error("InprocessTransport already started!");
|
|
32
|
+
}
|
|
33
|
+
await this._server.connect(this._serverTransport);
|
|
34
|
+
this._connected = true;
|
|
35
|
+
}
|
|
36
|
+
send(message, _options) {
|
|
37
|
+
if (!this._connected) {
|
|
38
|
+
throw new Error("Transport not connected");
|
|
39
|
+
}
|
|
40
|
+
this._serverTransport._receiveFromClient(message);
|
|
41
|
+
return Promise.resolve();
|
|
42
|
+
}
|
|
43
|
+
close() {
|
|
44
|
+
if (this._connected) {
|
|
45
|
+
this._connected = false;
|
|
46
|
+
this.onclose?.();
|
|
47
|
+
this._serverTransport.onclose?.();
|
|
48
|
+
}
|
|
49
|
+
return Promise.resolve();
|
|
50
|
+
}
|
|
51
|
+
onclose;
|
|
52
|
+
onerror;
|
|
53
|
+
onmessage;
|
|
54
|
+
sessionId;
|
|
55
|
+
setProtocolVersion;
|
|
56
|
+
_receiveFromServer(message, extra) {
|
|
57
|
+
this.onmessage?.(message, extra);
|
|
58
|
+
}
|
|
59
|
+
}
|
|
60
|
+
|
|
61
|
+
class InProcessServerTransport {
|
|
62
|
+
_clientTransport;
|
|
63
|
+
constructor(clientTransport) {
|
|
64
|
+
this._clientTransport = clientTransport;
|
|
65
|
+
}
|
|
66
|
+
start() {
|
|
67
|
+
if (!this._clientTransport) {
|
|
68
|
+
throw new Error("InProcessServerTransport: Client transport not available");
|
|
69
|
+
}
|
|
70
|
+
return Promise.resolve();
|
|
71
|
+
}
|
|
72
|
+
send(message, _options) {
|
|
73
|
+
this._clientTransport._receiveFromServer(message);
|
|
74
|
+
return Promise.resolve();
|
|
75
|
+
}
|
|
76
|
+
close() {
|
|
77
|
+
this.onclose?.();
|
|
78
|
+
return Promise.resolve();
|
|
79
|
+
}
|
|
80
|
+
onclose;
|
|
81
|
+
onerror;
|
|
82
|
+
onmessage;
|
|
83
|
+
sessionId;
|
|
84
|
+
setProtocolVersion;
|
|
85
|
+
_receiveFromClient(message, extra) {
|
|
86
|
+
this.onmessage?.(message, extra);
|
|
87
|
+
}
|
|
88
|
+
}
|
|
89
|
+
export {
|
|
90
|
+
InProcessTransport
|
|
91
|
+
};
|
|
@@ -0,0 +1,127 @@
|
|
|
1
|
+
import { createRequire } from "node:module";
|
|
2
|
+
var __create = Object.create;
|
|
3
|
+
var __getProtoOf = Object.getPrototypeOf;
|
|
4
|
+
var __defProp = Object.defineProperty;
|
|
5
|
+
var __getOwnPropNames = Object.getOwnPropertyNames;
|
|
6
|
+
var __hasOwnProp = Object.prototype.hasOwnProperty;
|
|
7
|
+
var __toESM = (mod, isNodeMode, target) => {
|
|
8
|
+
target = mod != null ? __create(__getProtoOf(mod)) : {};
|
|
9
|
+
const to = isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target;
|
|
10
|
+
for (let key of __getOwnPropNames(mod))
|
|
11
|
+
if (!__hasOwnProp.call(to, key))
|
|
12
|
+
__defProp(to, key, {
|
|
13
|
+
get: () => mod[key],
|
|
14
|
+
enumerable: true
|
|
15
|
+
});
|
|
16
|
+
return to;
|
|
17
|
+
};
|
|
18
|
+
var __require = /* @__PURE__ */ createRequire(import.meta.url);
|
|
19
|
+
|
|
20
|
+
// src/mcp/proxy-backend.ts
|
|
21
|
+
import { z } from "zod";
|
|
22
|
+
import { logUnhandledError } from "../log.js";
|
|
23
|
+
import { packageJSON } from "../package.js";
|
|
24
|
+
import { defineTool } from "../tools/tool.js";
|
|
25
|
+
|
|
26
|
+
class ProxyBackend {
|
|
27
|
+
name = "Playwright MCP Client Switcher";
|
|
28
|
+
version = packageJSON.version;
|
|
29
|
+
_clientFactories;
|
|
30
|
+
_currentClient;
|
|
31
|
+
_contextSwitchTool;
|
|
32
|
+
_tools = [];
|
|
33
|
+
_server;
|
|
34
|
+
constructor(clientFactories) {
|
|
35
|
+
this._clientFactories = clientFactories;
|
|
36
|
+
this._contextSwitchTool = this._defineContextSwitchTool();
|
|
37
|
+
}
|
|
38
|
+
async initialize(server) {
|
|
39
|
+
this._server = server;
|
|
40
|
+
await this._setCurrentClient(this._clientFactories[0]);
|
|
41
|
+
}
|
|
42
|
+
tools() {
|
|
43
|
+
if (this._clientFactories.length === 1) {
|
|
44
|
+
return this._tools;
|
|
45
|
+
}
|
|
46
|
+
return [...this._tools, this._contextSwitchTool.schema];
|
|
47
|
+
}
|
|
48
|
+
async callTool(schema, rawArguments) {
|
|
49
|
+
if (schema.name === this._contextSwitchTool.schema.name) {
|
|
50
|
+
return this._callContextSwitchTool(rawArguments);
|
|
51
|
+
}
|
|
52
|
+
const result = await this._currentClient?.callTool({
|
|
53
|
+
name: schema.name,
|
|
54
|
+
arguments: rawArguments
|
|
55
|
+
});
|
|
56
|
+
return result;
|
|
57
|
+
}
|
|
58
|
+
serverClosed() {
|
|
59
|
+
this._currentClient?.close().catch(logUnhandledError);
|
|
60
|
+
}
|
|
61
|
+
async _callContextSwitchTool(params) {
|
|
62
|
+
try {
|
|
63
|
+
const contextParams = params;
|
|
64
|
+
const factory = this._clientFactories.find((f) => f.name === contextParams.name);
|
|
65
|
+
if (!factory) {
|
|
66
|
+
throw new Error(`Unknown connection method: ${contextParams.name}`);
|
|
67
|
+
}
|
|
68
|
+
await this._setCurrentClient(factory);
|
|
69
|
+
return {
|
|
70
|
+
content: [
|
|
71
|
+
{
|
|
72
|
+
type: "text",
|
|
73
|
+
text: `### Result
|
|
74
|
+
Successfully changed connection method.
|
|
75
|
+
`
|
|
76
|
+
}
|
|
77
|
+
]
|
|
78
|
+
};
|
|
79
|
+
} catch (error) {
|
|
80
|
+
return {
|
|
81
|
+
content: [{ type: "text", text: `### Result
|
|
82
|
+
Error: ${error}
|
|
83
|
+
` }],
|
|
84
|
+
isError: true
|
|
85
|
+
};
|
|
86
|
+
}
|
|
87
|
+
}
|
|
88
|
+
_defineContextSwitchTool() {
|
|
89
|
+
return defineTool({
|
|
90
|
+
capability: "core",
|
|
91
|
+
schema: {
|
|
92
|
+
name: "browser_connect",
|
|
93
|
+
title: "Connect to a browser context",
|
|
94
|
+
description: [
|
|
95
|
+
"Connect to a browser using one of the available methods:",
|
|
96
|
+
...this._clientFactories.map((factory) => `- "${factory.name}": ${factory.description}`)
|
|
97
|
+
].join(`
|
|
98
|
+
`),
|
|
99
|
+
inputSchema: z.object({
|
|
100
|
+
name: z.enum(this._clientFactories.map((factory) => factory.name)).describe("The method to use to connect to the browser")
|
|
101
|
+
}),
|
|
102
|
+
type: "readOnly"
|
|
103
|
+
},
|
|
104
|
+
handle() {
|
|
105
|
+
throw new Error("Unreachable");
|
|
106
|
+
}
|
|
107
|
+
});
|
|
108
|
+
}
|
|
109
|
+
async _setCurrentClient(factory) {
|
|
110
|
+
await this._currentClient?.close();
|
|
111
|
+
if (!this._server) {
|
|
112
|
+
throw new Error("Server not initialized");
|
|
113
|
+
}
|
|
114
|
+
this._currentClient = await factory.create(this._server);
|
|
115
|
+
const tools = await this._currentClient.listTools();
|
|
116
|
+
this._tools = tools.tools.map((tool) => ({
|
|
117
|
+
name: tool.name,
|
|
118
|
+
title: tool.title ?? "",
|
|
119
|
+
description: tool.description ?? "",
|
|
120
|
+
inputSchema: tool.inputSchema ? z.any() : z.object({}),
|
|
121
|
+
type: tool.annotations?.readOnlyHint ? "readOnly" : "destructive"
|
|
122
|
+
}));
|
|
123
|
+
}
|
|
124
|
+
}
|
|
125
|
+
export {
|
|
126
|
+
ProxyBackend
|
|
127
|
+
};
|
|
@@ -0,0 +1,123 @@
|
|
|
1
|
+
import { createRequire } from "node:module";
|
|
2
|
+
var __create = Object.create;
|
|
3
|
+
var __getProtoOf = Object.getPrototypeOf;
|
|
4
|
+
var __defProp = Object.defineProperty;
|
|
5
|
+
var __getOwnPropNames = Object.getOwnPropertyNames;
|
|
6
|
+
var __hasOwnProp = Object.prototype.hasOwnProperty;
|
|
7
|
+
var __toESM = (mod, isNodeMode, target) => {
|
|
8
|
+
target = mod != null ? __create(__getProtoOf(mod)) : {};
|
|
9
|
+
const to = isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target;
|
|
10
|
+
for (let key of __getOwnPropNames(mod))
|
|
11
|
+
if (!__hasOwnProp.call(to, key))
|
|
12
|
+
__defProp(to, key, {
|
|
13
|
+
get: () => mod[key],
|
|
14
|
+
enumerable: true
|
|
15
|
+
});
|
|
16
|
+
return to;
|
|
17
|
+
};
|
|
18
|
+
var __require = /* @__PURE__ */ createRequire(import.meta.url);
|
|
19
|
+
|
|
20
|
+
// src/mcp/server.ts
|
|
21
|
+
import { Server } from "@modelcontextprotocol/sdk/server/index.js";
|
|
22
|
+
import {
|
|
23
|
+
CallToolRequestSchema,
|
|
24
|
+
ListToolsRequestSchema
|
|
25
|
+
} from "@modelcontextprotocol/sdk/types.js";
|
|
26
|
+
import debug from "debug";
|
|
27
|
+
import { z } from "zod";
|
|
28
|
+
import { zodToJsonSchema } from "zod-to-json-schema";
|
|
29
|
+
import { logUnhandledError } from "../log.js";
|
|
30
|
+
import { ManualPromise } from "../manual-promise.js";
|
|
31
|
+
import { logRequest } from "../utils/request-logger.js";
|
|
32
|
+
var serverDebug = debug("pw:mcp:server");
|
|
33
|
+
async function connect(serverBackendFactory, transport, runHeartbeat) {
|
|
34
|
+
const backend = serverBackendFactory();
|
|
35
|
+
const server = createServer(backend, runHeartbeat);
|
|
36
|
+
await server.connect(transport);
|
|
37
|
+
}
|
|
38
|
+
function createServer(backend, runHeartbeat) {
|
|
39
|
+
const initializedPromise = new ManualPromise;
|
|
40
|
+
const server = new Server({ name: backend.name, version: backend.version }, {
|
|
41
|
+
capabilities: {
|
|
42
|
+
tools: {}
|
|
43
|
+
}
|
|
44
|
+
});
|
|
45
|
+
server.setRequestHandler(ListToolsRequestSchema, () => {
|
|
46
|
+
const tools = backend.tools();
|
|
47
|
+
return {
|
|
48
|
+
tools: tools.map((tool) => ({
|
|
49
|
+
name: tool.name,
|
|
50
|
+
description: tool.description,
|
|
51
|
+
inputSchema: tool.inputSchema instanceof z.ZodType ? zodToJsonSchema(tool.inputSchema) : tool.inputSchema,
|
|
52
|
+
annotations: {
|
|
53
|
+
title: tool.title,
|
|
54
|
+
readOnlyHint: tool.type === "readOnly",
|
|
55
|
+
destructiveHint: tool.type === "destructive",
|
|
56
|
+
openWorldHint: true
|
|
57
|
+
}
|
|
58
|
+
}))
|
|
59
|
+
};
|
|
60
|
+
});
|
|
61
|
+
let heartbeatRunning = false;
|
|
62
|
+
server.setRequestHandler(CallToolRequestSchema, async (request) => {
|
|
63
|
+
await initializedPromise;
|
|
64
|
+
if (runHeartbeat && !heartbeatRunning) {
|
|
65
|
+
heartbeatRunning = true;
|
|
66
|
+
startHeartbeat(server);
|
|
67
|
+
}
|
|
68
|
+
const errorResult = (...messages) => ({
|
|
69
|
+
content: [{ type: "text", text: `### Result
|
|
70
|
+
${messages.join(`
|
|
71
|
+
`)}` }],
|
|
72
|
+
isError: true
|
|
73
|
+
});
|
|
74
|
+
const tools = backend.tools();
|
|
75
|
+
const tool = tools.find((t) => t.name === request.params.name);
|
|
76
|
+
if (!tool) {
|
|
77
|
+
return errorResult(`Error: Tool "${request.params.name}" not found`);
|
|
78
|
+
}
|
|
79
|
+
try {
|
|
80
|
+
logRequest(request.params.name, request.params.arguments ?? {});
|
|
81
|
+
return await backend.callTool(tool, request.params.arguments || {});
|
|
82
|
+
} catch (error) {
|
|
83
|
+
return errorResult(String(error));
|
|
84
|
+
}
|
|
85
|
+
});
|
|
86
|
+
addServerListener(server, "initialized", () => {
|
|
87
|
+
backend.initialize?.(server).then(() => initializedPromise.resolve()).catch(logUnhandledError);
|
|
88
|
+
});
|
|
89
|
+
addServerListener(server, "close", () => backend.serverClosed?.());
|
|
90
|
+
return server;
|
|
91
|
+
}
|
|
92
|
+
var startHeartbeat = (server) => {
|
|
93
|
+
const beat = async () => {
|
|
94
|
+
try {
|
|
95
|
+
await Promise.race([
|
|
96
|
+
server.ping(),
|
|
97
|
+
new Promise((_, reject) => setTimeout(() => reject(new Error("ping timeout")), 5000))
|
|
98
|
+
]);
|
|
99
|
+
setTimeout(beat, 3000);
|
|
100
|
+
} catch (error) {
|
|
101
|
+
serverDebug("Heartbeat ping failed:", error);
|
|
102
|
+
try {
|
|
103
|
+
await server.close();
|
|
104
|
+
} catch (closeError) {
|
|
105
|
+
serverDebug("Failed to close server after heartbeat failure:", closeError);
|
|
106
|
+
}
|
|
107
|
+
}
|
|
108
|
+
};
|
|
109
|
+
beat().catch((error) => {
|
|
110
|
+
serverDebug("Heartbeat initialization failed:", error);
|
|
111
|
+
});
|
|
112
|
+
};
|
|
113
|
+
function addServerListener(server, event, listener) {
|
|
114
|
+
const oldListener = server[`on${event}`];
|
|
115
|
+
server[`on${event}`] = () => {
|
|
116
|
+
oldListener?.();
|
|
117
|
+
listener();
|
|
118
|
+
};
|
|
119
|
+
}
|
|
120
|
+
export {
|
|
121
|
+
createServer,
|
|
122
|
+
connect
|
|
123
|
+
};
|
|
@@ -0,0 +1,159 @@
|
|
|
1
|
+
import { createRequire } from "node:module";
|
|
2
|
+
var __create = Object.create;
|
|
3
|
+
var __getProtoOf = Object.getPrototypeOf;
|
|
4
|
+
var __defProp = Object.defineProperty;
|
|
5
|
+
var __getOwnPropNames = Object.getOwnPropertyNames;
|
|
6
|
+
var __hasOwnProp = Object.prototype.hasOwnProperty;
|
|
7
|
+
var __toESM = (mod, isNodeMode, target) => {
|
|
8
|
+
target = mod != null ? __create(__getProtoOf(mod)) : {};
|
|
9
|
+
const to = isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target;
|
|
10
|
+
for (let key of __getOwnPropNames(mod))
|
|
11
|
+
if (!__hasOwnProp.call(to, key))
|
|
12
|
+
__defProp(to, key, {
|
|
13
|
+
get: () => mod[key],
|
|
14
|
+
enumerable: true
|
|
15
|
+
});
|
|
16
|
+
return to;
|
|
17
|
+
};
|
|
18
|
+
var __require = /* @__PURE__ */ createRequire(import.meta.url);
|
|
19
|
+
|
|
20
|
+
// src/mcp/transport.ts
|
|
21
|
+
import crypto from "node:crypto";
|
|
22
|
+
import { SSEServerTransport } from "@modelcontextprotocol/sdk/server/sse.js";
|
|
23
|
+
import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
|
|
24
|
+
import { StreamableHTTPServerTransport } from "@modelcontextprotocol/sdk/server/streamableHttp.js";
|
|
25
|
+
import debug from "debug";
|
|
26
|
+
import { httpAddressToString, startHttpServer } from "../http-server.js";
|
|
27
|
+
import { connect } from "./server.js";
|
|
28
|
+
async function start(serverBackendFactory, options) {
|
|
29
|
+
if (options.port !== undefined) {
|
|
30
|
+
const httpServer = await startHttpServer(options);
|
|
31
|
+
startHttpTransport(httpServer, serverBackendFactory);
|
|
32
|
+
} else {
|
|
33
|
+
await startStdioTransport(serverBackendFactory);
|
|
34
|
+
}
|
|
35
|
+
}
|
|
36
|
+
async function startStdioTransport(serverBackendFactory) {
|
|
37
|
+
await connect(serverBackendFactory, new StdioServerTransport, false);
|
|
38
|
+
}
|
|
39
|
+
var testDebug = debug("pw:mcp:test");
|
|
40
|
+
var transportDebug = debug("pw:mcp:transport");
|
|
41
|
+
async function handleSSE(serverBackendFactory, req, res, url, sessions) {
|
|
42
|
+
if (req.method === "POST") {
|
|
43
|
+
const sessionId = url.searchParams.get("sessionId");
|
|
44
|
+
if (!sessionId) {
|
|
45
|
+
res.statusCode = 400;
|
|
46
|
+
return res.end("Missing sessionId");
|
|
47
|
+
}
|
|
48
|
+
const transport = sessions.get(sessionId);
|
|
49
|
+
if (!transport) {
|
|
50
|
+
res.statusCode = 404;
|
|
51
|
+
return res.end("Session not found");
|
|
52
|
+
}
|
|
53
|
+
return await transport.handlePostMessage(req, res);
|
|
54
|
+
}
|
|
55
|
+
if (req.method === "GET") {
|
|
56
|
+
const transport = new SSEServerTransport("/sse", res);
|
|
57
|
+
sessions.set(transport.sessionId, transport);
|
|
58
|
+
testDebug(`create SSE session: ${transport.sessionId}`);
|
|
59
|
+
try {
|
|
60
|
+
await connect(serverBackendFactory, transport, false);
|
|
61
|
+
} catch (error) {
|
|
62
|
+
testDebug(`SSE session connection failed: ${transport.sessionId}`, error);
|
|
63
|
+
sessions.delete(transport.sessionId);
|
|
64
|
+
if (!res.headersSent) {
|
|
65
|
+
res.statusCode = 500;
|
|
66
|
+
res.end("Connection failed");
|
|
67
|
+
}
|
|
68
|
+
return;
|
|
69
|
+
}
|
|
70
|
+
res.on("close", () => {
|
|
71
|
+
testDebug(`delete SSE session: ${transport.sessionId}`);
|
|
72
|
+
sessions.delete(transport.sessionId);
|
|
73
|
+
});
|
|
74
|
+
res.on("error", (error) => {
|
|
75
|
+
testDebug(`SSE session error: ${transport.sessionId}`, error);
|
|
76
|
+
sessions.delete(transport.sessionId);
|
|
77
|
+
});
|
|
78
|
+
return;
|
|
79
|
+
}
|
|
80
|
+
res.statusCode = 405;
|
|
81
|
+
res.end("Method not allowed");
|
|
82
|
+
}
|
|
83
|
+
async function handleStreamable(serverBackendFactory, req, res, sessions) {
|
|
84
|
+
const sessionId = req.headers["mcp-session-id"];
|
|
85
|
+
if (sessionId) {
|
|
86
|
+
const transport = sessions.get(sessionId);
|
|
87
|
+
if (!transport) {
|
|
88
|
+
res.statusCode = 404;
|
|
89
|
+
res.end("Session not found");
|
|
90
|
+
return;
|
|
91
|
+
}
|
|
92
|
+
return await transport.handleRequest(req, res);
|
|
93
|
+
}
|
|
94
|
+
if (req.method === "POST") {
|
|
95
|
+
const transport = new StreamableHTTPServerTransport({
|
|
96
|
+
sessionIdGenerator: () => crypto.randomUUID(),
|
|
97
|
+
onsessioninitialized: async (_httpSessionId) => {
|
|
98
|
+
testDebug(`create http session: ${transport.sessionId}`);
|
|
99
|
+
try {
|
|
100
|
+
await connect(serverBackendFactory, transport, true);
|
|
101
|
+
if (transport.sessionId) {
|
|
102
|
+
sessions.set(transport.sessionId, transport);
|
|
103
|
+
}
|
|
104
|
+
} catch (error) {
|
|
105
|
+
testDebug(`HTTP session initialization failed: ${transport.sessionId}`, error);
|
|
106
|
+
}
|
|
107
|
+
}
|
|
108
|
+
});
|
|
109
|
+
transport.onclose = () => {
|
|
110
|
+
if (!transport.sessionId) {
|
|
111
|
+
return;
|
|
112
|
+
}
|
|
113
|
+
sessions.delete(transport.sessionId);
|
|
114
|
+
testDebug(`delete http session: ${transport.sessionId}`);
|
|
115
|
+
};
|
|
116
|
+
try {
|
|
117
|
+
await transport.handleRequest(req, res);
|
|
118
|
+
} catch (error) {
|
|
119
|
+
testDebug("HTTP transport request handling failed", error);
|
|
120
|
+
if (!res.headersSent) {
|
|
121
|
+
res.statusCode = 500;
|
|
122
|
+
res.end("Request handling failed");
|
|
123
|
+
}
|
|
124
|
+
}
|
|
125
|
+
return;
|
|
126
|
+
}
|
|
127
|
+
res.statusCode = 400;
|
|
128
|
+
res.end("Invalid request");
|
|
129
|
+
}
|
|
130
|
+
function startHttpTransport(httpServer, serverBackendFactory) {
|
|
131
|
+
const sseSessions = new Map;
|
|
132
|
+
const streamableSessions = new Map;
|
|
133
|
+
httpServer.on("request", async (req, res) => {
|
|
134
|
+
const url2 = new URL(`http://localhost${req.url}`);
|
|
135
|
+
if (url2.pathname.startsWith("/sse")) {
|
|
136
|
+
await handleSSE(serverBackendFactory, req, res, url2, sseSessions);
|
|
137
|
+
} else {
|
|
138
|
+
await handleStreamable(serverBackendFactory, req, res, streamableSessions);
|
|
139
|
+
}
|
|
140
|
+
});
|
|
141
|
+
const url = httpAddressToString(httpServer.address());
|
|
142
|
+
const message = [
|
|
143
|
+
`Listening on ${url}`,
|
|
144
|
+
"Put this in your client config:",
|
|
145
|
+
JSON.stringify({
|
|
146
|
+
mcpServers: {
|
|
147
|
+
playwright: {
|
|
148
|
+
url: `${url}/mcp`
|
|
149
|
+
}
|
|
150
|
+
}
|
|
151
|
+
}, undefined, 2),
|
|
152
|
+
"For legacy SSE transport support, you can use the /sse endpoint instead."
|
|
153
|
+
].join(`
|
|
154
|
+
`);
|
|
155
|
+
transportDebug("Server listening:", message);
|
|
156
|
+
}
|
|
157
|
+
export {
|
|
158
|
+
start
|
|
159
|
+
};
|
package/lib/package.js
ADDED
|
@@ -0,0 +1,28 @@
|
|
|
1
|
+
import { createRequire } from "node:module";
|
|
2
|
+
var __create = Object.create;
|
|
3
|
+
var __getProtoOf = Object.getPrototypeOf;
|
|
4
|
+
var __defProp = Object.defineProperty;
|
|
5
|
+
var __getOwnPropNames = Object.getOwnPropertyNames;
|
|
6
|
+
var __hasOwnProp = Object.prototype.hasOwnProperty;
|
|
7
|
+
var __toESM = (mod, isNodeMode, target) => {
|
|
8
|
+
target = mod != null ? __create(__getProtoOf(mod)) : {};
|
|
9
|
+
const to = isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target;
|
|
10
|
+
for (let key of __getOwnPropNames(mod))
|
|
11
|
+
if (!__hasOwnProp.call(to, key))
|
|
12
|
+
__defProp(to, key, {
|
|
13
|
+
get: () => mod[key],
|
|
14
|
+
enumerable: true
|
|
15
|
+
});
|
|
16
|
+
return to;
|
|
17
|
+
};
|
|
18
|
+
var __require = /* @__PURE__ */ createRequire(import.meta.url);
|
|
19
|
+
|
|
20
|
+
// src/package.ts
|
|
21
|
+
import fs from "node:fs";
|
|
22
|
+
import path from "node:path";
|
|
23
|
+
import url from "node:url";
|
|
24
|
+
var __filename2 = url.fileURLToPath(import.meta.url);
|
|
25
|
+
var packageJSON = JSON.parse(fs.readFileSync(path.join(path.dirname(__filename2), "..", "package.json"), "utf8"));
|
|
26
|
+
export {
|
|
27
|
+
packageJSON
|
|
28
|
+
};
|