@smoothdeploy/playwright 1.57.1 → 1.58.1-beta-1770383926000
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.
Potentially problematic release.
This version of @smoothdeploy/playwright might be problematic. Click here for more details.
- package/ThirdPartyNotices.txt +1188 -65
- package/lib/agents/agentParser.js +89 -0
- package/lib/agents/generateAgents.js +27 -74
- package/lib/agents/generateAgents.js.map +7 -0
- package/lib/agents/playwright-test-planner.agent.md +1 -0
- package/lib/common/config.js +6 -4
- package/lib/common/config.js.map +7 -0
- package/lib/common/configLoader.js.map +7 -0
- package/lib/common/esmLoaderHost.js +2 -0
- package/lib/common/esmLoaderHost.js.map +7 -0
- package/lib/common/expectBundle.js +2 -17
- package/lib/common/expectBundle.js.map +7 -0
- package/lib/common/expectBundleImpl.js +132 -132
- package/lib/common/expectBundleImpl.js.map +7 -0
- package/lib/common/fixtures.js.map +7 -0
- package/lib/common/globals.js.map +7 -0
- package/lib/common/ipc.js.map +7 -0
- package/lib/common/poolBuilder.js.map +7 -0
- package/lib/common/process.js +28 -0
- package/lib/common/process.js.map +7 -0
- package/lib/common/suiteUtils.js.map +7 -0
- package/lib/common/test.js.map +7 -0
- package/lib/common/testLoader.js.map +7 -0
- package/lib/common/testType.js.map +7 -0
- package/lib/common/validators.js +10 -10
- package/lib/common/validators.js.map +7 -0
- package/lib/fsWatcher.js.map +7 -0
- package/lib/index.js +205 -12
- package/lib/index.js.map +7 -0
- package/lib/internalsForTest.js.map +7 -0
- package/lib/isomorphic/events.js.map +7 -0
- package/lib/isomorphic/folders.js.map +7 -0
- package/lib/isomorphic/stringInternPool.js.map +7 -0
- package/lib/isomorphic/teleReceiver.js +15 -2
- package/lib/isomorphic/teleReceiver.js.map +7 -0
- package/lib/isomorphic/teleSuiteUpdater.js +24 -4
- package/lib/isomorphic/teleSuiteUpdater.js.map +7 -0
- package/lib/isomorphic/testServerConnection.js.map +7 -0
- package/lib/isomorphic/testServerInterface.js.map +7 -0
- package/lib/isomorphic/testTree.js +13 -18
- package/lib/isomorphic/testTree.js.map +7 -0
- package/lib/isomorphic/types.d.js.map +7 -0
- package/lib/loader/loaderMain.js.map +7 -0
- package/lib/matchers/expect.js +2 -15
- package/lib/matchers/expect.js.map +7 -0
- package/lib/matchers/matcherHint.js +1 -44
- package/lib/matchers/matcherHint.js.map +7 -0
- package/lib/matchers/matchers.js +3 -2
- package/lib/matchers/matchers.js.map +7 -0
- package/lib/matchers/toBeTruthy.js +5 -3
- package/lib/matchers/toBeTruthy.js.map +7 -0
- package/lib/matchers/toEqual.js +4 -3
- package/lib/matchers/toEqual.js.map +7 -0
- package/lib/matchers/toHaveURL.js +5 -6
- package/lib/matchers/toHaveURL.js.map +7 -0
- package/lib/matchers/toMatchAriaSnapshot.js +5 -5
- package/lib/matchers/toMatchAriaSnapshot.js.map +7 -0
- package/lib/matchers/toMatchSnapshot.js +9 -8
- package/lib/matchers/toMatchSnapshot.js.map +7 -0
- package/lib/matchers/toMatchText.js +9 -9
- package/lib/matchers/toMatchText.js.map +7 -0
- package/lib/mcp/browser/actions.d.js.map +7 -0
- package/lib/mcp/browser/browserContextFactory.js +62 -29
- package/lib/mcp/browser/browserContextFactory.js.map +7 -0
- package/lib/mcp/browser/browserServerBackend.js +17 -9
- package/lib/mcp/browser/browserServerBackend.js.map +7 -0
- package/lib/mcp/browser/codegen.js.map +7 -0
- package/lib/mcp/browser/config.js +65 -12
- package/lib/mcp/browser/config.js.map +7 -0
- package/lib/mcp/browser/context.js +71 -94
- package/lib/mcp/browser/context.js.map +7 -0
- package/lib/mcp/browser/response.js +172 -131
- package/lib/mcp/browser/response.js.map +7 -0
- package/lib/mcp/browser/sessionLog.js +19 -104
- package/lib/mcp/browser/sessionLog.js.map +7 -0
- package/lib/mcp/browser/tab.js +92 -41
- package/lib/mcp/browser/tab.js.map +7 -0
- package/lib/mcp/browser/tools/common.js +8 -6
- package/lib/mcp/browser/tools/common.js.map +7 -0
- package/lib/mcp/browser/tools/console.js +7 -5
- package/lib/mcp/browser/tools/console.js.map +7 -0
- package/lib/mcp/browser/tools/dialogs.js +4 -4
- package/lib/mcp/browser/tools/dialogs.js.map +7 -0
- package/lib/mcp/browser/tools/evaluate.js +11 -19
- package/lib/mcp/browser/tools/evaluate.js.map +7 -0
- package/lib/mcp/browser/tools/files.js +3 -3
- package/lib/mcp/browser/tools/files.js.map +7 -0
- package/lib/mcp/browser/tools/form.js +9 -19
- package/lib/mcp/browser/tools/form.js.map +7 -0
- package/lib/mcp/browser/tools/install.js +6 -3
- package/lib/mcp/browser/tools/install.js.map +7 -0
- package/lib/mcp/browser/tools/keyboard.js +34 -11
- package/lib/mcp/browser/tools/keyboard.js.map +7 -0
- package/lib/mcp/browser/tools/mouse.js +11 -11
- package/lib/mcp/browser/tools/mouse.js.map +7 -0
- package/lib/mcp/browser/tools/navigate.js +14 -5
- package/lib/mcp/browser/tools/navigate.js.map +7 -0
- package/lib/mcp/browser/tools/network.js +20 -11
- package/lib/mcp/browser/tools/network.js.map +7 -0
- package/lib/mcp/browser/tools/open.js +57 -0
- package/lib/mcp/browser/tools/pdf.js +9 -19
- package/lib/mcp/browser/tools/pdf.js.map +7 -0
- package/lib/mcp/browser/tools/runCode.js +11 -8
- package/lib/mcp/browser/tools/runCode.js.map +7 -0
- package/lib/mcp/browser/tools/screenshot.js +16 -29
- package/lib/mcp/browser/tools/screenshot.js.map +7 -0
- package/lib/mcp/browser/tools/snapshot.js +21 -29
- package/lib/mcp/browser/tools/snapshot.js.map +7 -0
- package/lib/mcp/browser/tools/tabs.js +12 -12
- package/lib/mcp/browser/tools/tabs.js.map +7 -0
- package/lib/mcp/browser/tools/tool.js +2 -4
- package/lib/mcp/browser/tools/tool.js.map +7 -0
- package/lib/mcp/browser/tools/tracing.js +6 -6
- package/lib/mcp/browser/tools/tracing.js.map +7 -0
- package/lib/mcp/browser/tools/utils.js +49 -44
- package/lib/mcp/browser/tools/utils.js.map +7 -0
- package/lib/mcp/browser/tools/verify.js +24 -34
- package/lib/mcp/browser/tools/verify.js.map +7 -0
- package/lib/mcp/browser/tools/wait.js +6 -6
- package/lib/mcp/browser/tools/wait.js.map +7 -0
- package/lib/mcp/browser/tools.js +3 -1
- package/lib/mcp/browser/tools.js.map +7 -0
- package/lib/mcp/browser/watchdog.js.map +7 -0
- package/lib/mcp/config.d.js.map +7 -0
- package/lib/mcp/extension/cdpRelay.js +1 -1
- package/lib/mcp/extension/cdpRelay.js.map +7 -0
- package/lib/mcp/extension/extensionContextFactory.js +6 -5
- package/lib/mcp/extension/extensionContextFactory.js.map +7 -0
- package/lib/mcp/extension/protocol.js.map +7 -0
- package/lib/mcp/index.js.map +7 -0
- package/lib/mcp/log.js.map +7 -0
- package/lib/mcp/program.js +15 -20
- package/lib/mcp/program.js.map +7 -0
- package/lib/mcp/sdk/bundle.js.map +7 -0
- package/lib/mcp/sdk/exports.js +0 -2
- package/lib/mcp/sdk/exports.js.map +7 -0
- package/lib/mcp/sdk/http.js +20 -55
- package/lib/mcp/sdk/http.js.map +7 -0
- package/lib/mcp/sdk/inProcessTransport.js.map +7 -0
- package/lib/mcp/sdk/proxyBackend.js.map +7 -0
- package/lib/mcp/sdk/server.js +29 -4
- package/lib/mcp/sdk/server.js.map +7 -0
- package/lib/mcp/sdk/tool.js +2 -2
- package/lib/mcp/sdk/tool.js.map +7 -0
- package/lib/mcp/terminal/cli.js +296 -0
- package/lib/mcp/terminal/command.js +56 -0
- package/lib/mcp/terminal/commands.js +333 -0
- package/lib/mcp/terminal/daemon.js +129 -0
- package/lib/mcp/terminal/help.json +32 -0
- package/lib/mcp/terminal/helpGenerator.js +88 -0
- package/lib/mcp/terminal/socketConnection.js +80 -0
- package/lib/mcp/test/browserBackend.js +3 -13
- package/lib/mcp/test/browserBackend.js.map +7 -0
- package/lib/mcp/test/generatorTools.js +9 -9
- package/lib/mcp/test/generatorTools.js.map +7 -0
- package/lib/mcp/test/plannerTools.js +23 -22
- package/lib/mcp/test/plannerTools.js.map +7 -0
- package/lib/mcp/test/seed.js.map +7 -0
- package/lib/mcp/test/streams.js.map +7 -0
- package/lib/mcp/test/testBackend.js +6 -6
- package/lib/mcp/test/testBackend.js.map +7 -0
- package/lib/mcp/test/testContext.js +9 -3
- package/lib/mcp/test/testContext.js.map +7 -0
- package/lib/mcp/test/testTool.js.map +7 -0
- package/lib/mcp/test/testTools.js +12 -10
- package/lib/mcp/test/testTools.js.map +7 -0
- package/lib/mcpBundleImpl.js.map +7 -0
- package/lib/plugins/gitCommitInfoPlugin.js.map +7 -0
- package/lib/plugins/index.js.map +7 -0
- package/lib/plugins/webServerPlugin.js.map +7 -0
- package/lib/program.js +18 -4
- package/lib/program.js.map +7 -0
- package/lib/reporters/base.js +29 -4
- package/lib/reporters/base.js.map +7 -0
- package/lib/reporters/blob.js +3 -0
- package/lib/reporters/blob.js.map +7 -0
- package/lib/reporters/dot.js +17 -0
- package/lib/reporters/dot.js.map +7 -0
- package/lib/reporters/empty.js.map +7 -0
- package/lib/reporters/github.js.map +7 -0
- package/lib/reporters/html.js +13 -3
- package/lib/reporters/html.js.map +7 -0
- package/lib/reporters/internalReporter.js +6 -0
- package/lib/reporters/internalReporter.js.map +7 -0
- package/lib/reporters/json.js.map +7 -0
- package/lib/reporters/junit.js.map +7 -0
- package/lib/reporters/line.js +18 -0
- package/lib/reporters/line.js.map +7 -0
- package/lib/reporters/list.js +22 -0
- package/lib/reporters/list.js.map +7 -0
- package/lib/reporters/listModeReporter.js.map +7 -0
- package/lib/reporters/markdown.js.map +7 -0
- package/lib/reporters/merge.js +25 -8
- package/lib/reporters/merge.js.map +7 -0
- package/lib/reporters/multiplexer.js +8 -0
- package/lib/reporters/multiplexer.js.map +7 -0
- package/lib/reporters/reporterV2.js.map +7 -0
- package/lib/reporters/smoothdeploy.js +191 -0
- package/lib/reporters/teleEmitter.js +22 -4
- package/lib/reporters/teleEmitter.js.map +7 -0
- package/lib/reporters/versions/blobV1.js.map +7 -0
- package/lib/runner/dispatcher.js +20 -4
- package/lib/runner/dispatcher.js.map +7 -0
- package/lib/runner/failureTracker.js.map +7 -0
- package/lib/runner/lastRun.js.map +7 -0
- package/lib/runner/loadUtils.js +2 -2
- package/lib/runner/loadUtils.js.map +7 -0
- package/lib/runner/loaderHost.js.map +7 -0
- package/lib/runner/processHost.js +19 -0
- package/lib/runner/processHost.js.map +7 -0
- package/lib/runner/projectUtils.js +1 -1
- package/lib/runner/projectUtils.js.map +7 -0
- package/lib/runner/rebase.js.map +7 -0
- package/lib/runner/reporters.js +3 -1
- package/lib/runner/reporters.js.map +7 -0
- package/lib/runner/sigIntWatcher.js.map +7 -0
- package/lib/runner/storage.js +91 -0
- package/lib/runner/taskRunner.js.map +7 -0
- package/lib/runner/tasks.js.map +7 -0
- package/lib/runner/testGroups.js +14 -6
- package/lib/runner/testGroups.js.map +7 -0
- package/lib/runner/testRunner.js +13 -4
- package/lib/runner/testRunner.js.map +7 -0
- package/lib/runner/testServer.js +2 -2
- package/lib/runner/testServer.js.map +7 -0
- package/lib/runner/uiModeReporter.js.map +7 -0
- package/lib/runner/vcs.js.map +7 -0
- package/lib/runner/watchMode.js +2 -1
- package/lib/runner/watchMode.js.map +7 -0
- package/lib/runner/workerHost.js +6 -0
- package/lib/runner/workerHost.js.map +7 -0
- package/lib/third_party/pirates.js.map +7 -0
- package/lib/third_party/tsconfig-loader.js.map +7 -0
- package/lib/transform/babelBundle.js +3 -0
- package/lib/transform/babelBundle.js.map +7 -0
- package/lib/transform/babelBundleImpl.js +134 -134
- package/lib/transform/babelBundleImpl.js.map +7 -0
- package/lib/transform/compilationCache.js +2 -0
- package/lib/transform/compilationCache.js.map +7 -0
- package/lib/transform/esmLoader.js +10 -11
- package/lib/transform/esmLoader.js.map +7 -0
- package/lib/transform/md.js +221 -0
- package/lib/transform/portTransport.js.map +7 -0
- package/lib/transform/transform.js +18 -8
- package/lib/transform/transform.js.map +7 -0
- package/lib/util.js +3 -6
- package/lib/util.js.map +7 -0
- package/lib/utilsBundle.js +7 -0
- package/lib/utilsBundle.js.map +7 -0
- package/lib/utilsBundleImpl.js +51 -48
- package/lib/utilsBundleImpl.js.map +7 -0
- package/lib/worker/fixtureRunner.js +6 -2
- package/lib/worker/fixtureRunner.js.map +7 -0
- package/lib/worker/testInfo.js +39 -19
- package/lib/worker/testInfo.js.map +7 -0
- package/lib/worker/testTracing.js.map +7 -0
- package/lib/worker/timeoutManager.js.map +7 -0
- package/lib/worker/util.js.map +7 -0
- package/lib/worker/workerMain.js +17 -16
- package/lib/worker/workerMain.js.map +7 -0
- package/package.json +2 -2
- package/test.mjs +1 -0
- package/types/test.d.ts +26 -6
- package/types/testReporter.d.ts +1 -0
|
@@ -0,0 +1,7 @@
|
|
|
1
|
+
{
|
|
2
|
+
"version": 3,
|
|
3
|
+
"sources": ["../../../src/mcp/sdk/exports.ts"],
|
|
4
|
+
"sourcesContent": ["/**\n * Copyright (c) Microsoft Corporation.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nexport * from './inProcessTransport';\nexport * from './proxyBackend';\nexport * from './server';\nexport * from './tool';\nexport * from './http';\n"],
|
|
5
|
+
"mappings": ";;;;;;;;;;;;;;;AAAA;AAAA;AAgBA,4BAAc,iCAhBd;AAiBA,4BAAc,2BAjBd;AAkBA,4BAAc,qBAlBd;AAmBA,4BAAc,mBAnBd;AAoBA,4BAAc,mBApBd;",
|
|
6
|
+
"names": []
|
|
7
|
+
}
|
package/lib/mcp/sdk/http.js
CHANGED
|
@@ -28,51 +28,36 @@ var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__ge
|
|
|
28
28
|
var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
|
|
29
29
|
var http_exports = {};
|
|
30
30
|
__export(http_exports, {
|
|
31
|
-
|
|
32
|
-
|
|
33
|
-
startHttpServer: () => startHttpServer
|
|
31
|
+
addressToString: () => addressToString,
|
|
32
|
+
startMcpHttpServer: () => startMcpHttpServer
|
|
34
33
|
});
|
|
35
34
|
module.exports = __toCommonJS(http_exports);
|
|
36
35
|
var import_assert = __toESM(require("assert"));
|
|
37
|
-
var import_http = __toESM(require("http"));
|
|
38
36
|
var import_crypto = __toESM(require("crypto"));
|
|
39
37
|
var import_utilsBundle = require("playwright-core/lib/utilsBundle");
|
|
40
|
-
var mcpBundle = __toESM(require("
|
|
38
|
+
var mcpBundle = __toESM(require("playwright-core/lib/mcpBundle"));
|
|
39
|
+
var import_utils = require("playwright-core/lib/utils");
|
|
41
40
|
var mcpServer = __toESM(require("./server"));
|
|
42
41
|
const testDebug = (0, import_utilsBundle.debug)("pw:mcp:test");
|
|
43
|
-
async function
|
|
44
|
-
const
|
|
45
|
-
|
|
46
|
-
|
|
47
|
-
await new Promise((resolve, reject) => {
|
|
48
|
-
httpServer.on("error", reject);
|
|
49
|
-
abortSignal?.addEventListener("abort", () => {
|
|
50
|
-
httpServer.close();
|
|
51
|
-
reject(new Error("Aborted"));
|
|
52
|
-
});
|
|
53
|
-
httpServer.listen(port, host, () => {
|
|
54
|
-
resolve();
|
|
55
|
-
httpServer.removeListener("error", reject);
|
|
56
|
-
});
|
|
57
|
-
});
|
|
58
|
-
return httpServer;
|
|
42
|
+
async function startMcpHttpServer(config, serverBackendFactory, allowedHosts) {
|
|
43
|
+
const httpServer = (0, import_utils.createHttpServer)();
|
|
44
|
+
await (0, import_utils.startHttpServer)(httpServer, config);
|
|
45
|
+
return await installHttpTransport(httpServer, serverBackendFactory, allowedHosts);
|
|
59
46
|
}
|
|
60
|
-
function
|
|
47
|
+
function addressToString(address, options) {
|
|
61
48
|
(0, import_assert.default)(address, "Could not bind server socket");
|
|
62
49
|
if (typeof address === "string")
|
|
63
|
-
|
|
64
|
-
|
|
65
|
-
|
|
66
|
-
|
|
67
|
-
|
|
68
|
-
return `http://${resolvedHost}:${resolvedPort}`;
|
|
50
|
+
throw new Error("Unexpected address type: " + address);
|
|
51
|
+
let host = address.family === "IPv4" ? address.address : `[${address.address}]`;
|
|
52
|
+
if (options.normalizeLoopback && (host === "0.0.0.0" || host === "[::]" || host === "[::1]" || host === "127.0.0.1"))
|
|
53
|
+
host = "localhost";
|
|
54
|
+
return `${options.protocol}://${host}:${address.port}`;
|
|
69
55
|
}
|
|
70
|
-
async function installHttpTransport(httpServer, serverBackendFactory,
|
|
71
|
-
const url =
|
|
56
|
+
async function installHttpTransport(httpServer, serverBackendFactory, allowedHosts) {
|
|
57
|
+
const url = addressToString(httpServer.address(), { protocol: "http", normalizeLoopback: true });
|
|
72
58
|
const host = new URL(url).host;
|
|
73
59
|
allowedHosts = (allowedHosts || [host]).map((h) => h.toLowerCase());
|
|
74
60
|
const allowAnyHost = allowedHosts.includes("*");
|
|
75
|
-
const pathPrefix = unguessableUrl ? `/${import_crypto.default.randomUUID()}` : "";
|
|
76
61
|
const sseSessions = /* @__PURE__ */ new Map();
|
|
77
62
|
const streamableSessions = /* @__PURE__ */ new Map();
|
|
78
63
|
httpServer.on("request", async (req, res) => {
|
|
@@ -87,12 +72,7 @@ async function installHttpTransport(httpServer, serverBackendFactory, unguessabl
|
|
|
87
72
|
return res.end("Access is only allowed at " + allowedHosts.join(", "));
|
|
88
73
|
}
|
|
89
74
|
}
|
|
90
|
-
|
|
91
|
-
res.statusCode = 404;
|
|
92
|
-
return res.end("Not found");
|
|
93
|
-
}
|
|
94
|
-
const path = req.url?.slice(pathPrefix.length);
|
|
95
|
-
const url2 = new URL(`http://localhost${path}`);
|
|
75
|
+
const url2 = new URL(`http://localhost${req.url}`);
|
|
96
76
|
if (url2.pathname === "/killkillkill" && req.method === "GET") {
|
|
97
77
|
res.statusCode = 200;
|
|
98
78
|
res.end("Killing process");
|
|
@@ -104,7 +84,7 @@ async function installHttpTransport(httpServer, serverBackendFactory, unguessabl
|
|
|
104
84
|
else
|
|
105
85
|
await handleStreamable(serverBackendFactory, req, res, streamableSessions);
|
|
106
86
|
});
|
|
107
|
-
return
|
|
87
|
+
return url;
|
|
108
88
|
}
|
|
109
89
|
async function handleSSE(serverBackendFactory, req, res, url, sessions) {
|
|
110
90
|
if (req.method === "POST") {
|
|
@@ -165,23 +145,8 @@ async function handleStreamable(serverBackendFactory, req, res, sessions) {
|
|
|
165
145
|
res.statusCode = 400;
|
|
166
146
|
res.end("Invalid request");
|
|
167
147
|
}
|
|
168
|
-
function decorateServer(server) {
|
|
169
|
-
const sockets = /* @__PURE__ */ new Set();
|
|
170
|
-
server.on("connection", (socket) => {
|
|
171
|
-
sockets.add(socket);
|
|
172
|
-
socket.once("close", () => sockets.delete(socket));
|
|
173
|
-
});
|
|
174
|
-
const close = server.close;
|
|
175
|
-
server.close = (callback) => {
|
|
176
|
-
for (const socket of sockets)
|
|
177
|
-
socket.destroy();
|
|
178
|
-
sockets.clear();
|
|
179
|
-
return close.call(server, callback);
|
|
180
|
-
};
|
|
181
|
-
}
|
|
182
148
|
// Annotate the CommonJS export names for ESM import in node:
|
|
183
149
|
0 && (module.exports = {
|
|
184
|
-
|
|
185
|
-
|
|
186
|
-
startHttpServer
|
|
150
|
+
addressToString,
|
|
151
|
+
startMcpHttpServer
|
|
187
152
|
});
|
|
@@ -0,0 +1,7 @@
|
|
|
1
|
+
{
|
|
2
|
+
"version": 3,
|
|
3
|
+
"sources": ["../../../src/mcp/sdk/http.ts"],
|
|
4
|
+
"sourcesContent": ["/**\n * Copyright (c) Microsoft Corporation.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport assert from 'assert';\nimport net from 'net';\nimport http from 'http';\nimport crypto from 'crypto';\n\nimport { debug } from 'playwright-core/lib/utilsBundle';\n\nimport * as mcpBundle from './bundle';\nimport * as mcpServer from './server';\n\nimport type { ServerBackendFactory } from './server';\nimport type { SSEServerTransport } from '@modelcontextprotocol/sdk/server/sse.js';\nimport type { StreamableHTTPServerTransport } from '@modelcontextprotocol/sdk/server/streamableHttp.js';\n\nconst testDebug = debug('pw:mcp:test');\n\nexport async function startHttpServer(config: { host?: string, port?: number }, abortSignal?: AbortSignal): Promise<http.Server> {\n const { host, port } = config;\n const httpServer = http.createServer();\n decorateServer(httpServer);\n await new Promise<void>((resolve, reject) => {\n httpServer.on('error', reject);\n abortSignal?.addEventListener('abort', () => {\n httpServer.close();\n reject(new Error('Aborted'));\n });\n httpServer.listen(port, host, () => {\n resolve();\n httpServer.removeListener('error', reject);\n });\n });\n return httpServer;\n}\n\nexport function httpAddressToString(address: string | net.AddressInfo | null): string {\n assert(address, 'Could not bind server socket');\n if (typeof address === 'string')\n return address;\n const resolvedPort = address.port;\n let resolvedHost = address.family === 'IPv4' ? address.address : `[${address.address}]`;\n if (resolvedHost === '0.0.0.0' || resolvedHost === '[::]')\n resolvedHost = 'localhost';\n return `http://${resolvedHost}:${resolvedPort}`;\n}\n\nexport async function installHttpTransport(httpServer: http.Server, serverBackendFactory: ServerBackendFactory, unguessableUrl: boolean, allowedHosts?: string[]) {\n const url = httpAddressToString(httpServer.address());\n const host = new URL(url).host;\n allowedHosts = (allowedHosts || [host]).map(h => h.toLowerCase());\n const allowAnyHost = allowedHosts.includes('*');\n const pathPrefix = unguessableUrl ? `/${crypto.randomUUID()}` : '';\n\n const sseSessions = new Map();\n const streamableSessions = new Map();\n httpServer.on('request', async (req, res) => {\n if (!allowAnyHost) {\n const host = req.headers.host?.toLowerCase();\n if (!host) {\n res.statusCode = 400;\n return res.end('Missing host');\n }\n\n // Prevent DNS evil.com -> localhost rebind.\n if (!allowedHosts.includes(host)) {\n // Access from the browser is forbidden.\n res.statusCode = 403;\n return res.end('Access is only allowed at ' + allowedHosts.join(', '));\n }\n }\n\n if (!req.url?.startsWith(pathPrefix)) {\n res.statusCode = 404;\n return res.end('Not found');\n }\n\n const path = req.url?.slice(pathPrefix.length);\n const url = new URL(`http://localhost${path}`);\n if (url.pathname === '/killkillkill' && req.method === 'GET') {\n res.statusCode = 200;\n res.end('Killing process');\n // Simulate Ctrl+C in a way that works on Windows too.\n process.emit('SIGINT');\n return;\n }\n if (url.pathname.startsWith('/sse'))\n await handleSSE(serverBackendFactory, req, res, url, sseSessions);\n else\n await handleStreamable(serverBackendFactory, req, res, streamableSessions);\n });\n\n return `${url}${pathPrefix}`;\n}\n\nasync function handleSSE(serverBackendFactory: ServerBackendFactory, req: http.IncomingMessage, res: http.ServerResponse, url: URL, sessions: Map<string, SSEServerTransport>) {\n if (req.method === 'POST') {\n const sessionId = url.searchParams.get('sessionId');\n if (!sessionId) {\n res.statusCode = 400;\n return res.end('Missing sessionId');\n }\n\n const transport = sessions.get(sessionId);\n if (!transport) {\n res.statusCode = 404;\n return res.end('Session not found');\n }\n\n return await transport.handlePostMessage(req, res);\n } else if (req.method === 'GET') {\n const transport = new mcpBundle.SSEServerTransport('/sse', res);\n sessions.set(transport.sessionId, transport);\n testDebug(`create SSE session: ${transport.sessionId}`);\n await mcpServer.connect(serverBackendFactory, transport, false);\n res.on('close', () => {\n testDebug(`delete SSE session: ${transport.sessionId}`);\n sessions.delete(transport.sessionId);\n });\n return;\n }\n\n res.statusCode = 405;\n res.end('Method not allowed');\n}\n\nasync function handleStreamable(serverBackendFactory: ServerBackendFactory, req: http.IncomingMessage, res: http.ServerResponse, sessions: Map<string, StreamableHTTPServerTransport>) {\n const sessionId = req.headers['mcp-session-id'] as string | undefined;\n if (sessionId) {\n const transport = sessions.get(sessionId);\n if (!transport) {\n res.statusCode = 404;\n res.end('Session not found');\n return;\n }\n return await transport.handleRequest(req, res);\n }\n\n if (req.method === 'POST') {\n const transport = new mcpBundle.StreamableHTTPServerTransport({\n sessionIdGenerator: () => crypto.randomUUID(),\n onsessioninitialized: async sessionId => {\n testDebug(`create http session: ${transport.sessionId}`);\n await mcpServer.connect(serverBackendFactory, transport, true);\n sessions.set(sessionId, transport);\n }\n });\n\n transport.onclose = () => {\n if (!transport.sessionId)\n return;\n sessions.delete(transport.sessionId);\n testDebug(`delete http session: ${transport.sessionId}`);\n };\n\n await transport.handleRequest(req, res);\n return;\n }\n\n res.statusCode = 400;\n res.end('Invalid request');\n}\n\nfunction decorateServer(server: net.Server) {\n const sockets = new Set<net.Socket>();\n server.on('connection', socket => {\n sockets.add(socket);\n socket.once('close', () => sockets.delete(socket));\n });\n\n const close = server.close;\n server.close = (callback?: (err?: Error) => void) => {\n for (const socket of sockets)\n socket.destroy();\n sockets.clear();\n return close.call(server, callback);\n };\n}\n"],
|
|
5
|
+
"mappings": ";;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAgBA,oBAAmB;AAEnB,kBAAiB;AACjB,oBAAmB;AAEnB,yBAAsB;AAEtB,gBAA2B;AAC3B,gBAA2B;AAM3B,MAAM,gBAAY,0BAAM,aAAa;AAErC,eAAsB,gBAAgB,QAA0C,aAAiD;AAC/H,QAAM,EAAE,MAAM,KAAK,IAAI;AACvB,QAAM,aAAa,YAAAA,QAAK,aAAa;AACrC,iBAAe,UAAU;AACzB,QAAM,IAAI,QAAc,CAAC,SAAS,WAAW;AAC3C,eAAW,GAAG,SAAS,MAAM;AAC7B,iBAAa,iBAAiB,SAAS,MAAM;AAC3C,iBAAW,MAAM;AACjB,aAAO,IAAI,MAAM,SAAS,CAAC;AAAA,IAC7B,CAAC;AACD,eAAW,OAAO,MAAM,MAAM,MAAM;AAClC,cAAQ;AACR,iBAAW,eAAe,SAAS,MAAM;AAAA,IAC3C,CAAC;AAAA,EACH,CAAC;AACD,SAAO;AACT;AAEO,SAAS,oBAAoB,SAAkD;AACpF,oBAAAC,SAAO,SAAS,8BAA8B;AAC9C,MAAI,OAAO,YAAY;AACrB,WAAO;AACT,QAAM,eAAe,QAAQ;AAC7B,MAAI,eAAe,QAAQ,WAAW,SAAS,QAAQ,UAAU,IAAI,QAAQ,OAAO;AACpF,MAAI,iBAAiB,aAAa,iBAAiB;AACjD,mBAAe;AACjB,SAAO,UAAU,YAAY,IAAI,YAAY;AAC/C;AAEA,eAAsB,qBAAqB,YAAyB,sBAA4C,gBAAyB,cAAyB;AAChK,QAAM,MAAM,oBAAoB,WAAW,QAAQ,CAAC;AACpD,QAAM,OAAO,IAAI,IAAI,GAAG,EAAE;AAC1B,kBAAgB,gBAAgB,CAAC,IAAI,GAAG,IAAI,OAAK,EAAE,YAAY,CAAC;AAChE,QAAM,eAAe,aAAa,SAAS,GAAG;AAC9C,QAAM,aAAa,iBAAiB,IAAI,cAAAC,QAAO,WAAW,CAAC,KAAK;AAEhE,QAAM,cAAc,oBAAI,IAAI;AAC5B,QAAM,qBAAqB,oBAAI,IAAI;AACnC,aAAW,GAAG,WAAW,OAAO,KAAK,QAAQ;AAC3C,QAAI,CAAC,cAAc;AACjB,YAAMC,QAAO,IAAI,QAAQ,MAAM,YAAY;AAC3C,UAAI,CAACA,OAAM;AACT,YAAI,aAAa;AACjB,eAAO,IAAI,IAAI,cAAc;AAAA,MAC/B;AAGA,UAAI,CAAC,aAAa,SAASA,KAAI,GAAG;AAEhC,YAAI,aAAa;AACjB,eAAO,IAAI,IAAI,+BAA+B,aAAa,KAAK,IAAI,CAAC;AAAA,MACvE;AAAA,IACF;AAEA,QAAI,CAAC,IAAI,KAAK,WAAW,UAAU,GAAG;AACpC,UAAI,aAAa;AACjB,aAAO,IAAI,IAAI,WAAW;AAAA,IAC5B;AAEA,UAAM,OAAO,IAAI,KAAK,MAAM,WAAW,MAAM;AAC7C,UAAMC,OAAM,IAAI,IAAI,mBAAmB,IAAI,EAAE;AAC7C,QAAIA,KAAI,aAAa,mBAAmB,IAAI,WAAW,OAAO;AAC5D,UAAI,aAAa;AACjB,UAAI,IAAI,iBAAiB;AAEzB,cAAQ,KAAK,QAAQ;AACrB;AAAA,IACF;AACA,QAAIA,KAAI,SAAS,WAAW,MAAM;AAChC,YAAM,UAAU,sBAAsB,KAAK,KAAKA,MAAK,WAAW;AAAA;AAEhE,YAAM,iBAAiB,sBAAsB,KAAK,KAAK,kBAAkB;AAAA,EAC7E,CAAC;AAED,SAAO,GAAG,GAAG,GAAG,UAAU;AAC5B;AAEA,eAAe,UAAU,sBAA4C,KAA2B,KAA0B,KAAU,UAA2C;AAC7K,MAAI,IAAI,WAAW,QAAQ;AACzB,UAAM,YAAY,IAAI,aAAa,IAAI,WAAW;AAClD,QAAI,CAAC,WAAW;AACd,UAAI,aAAa;AACjB,aAAO,IAAI,IAAI,mBAAmB;AAAA,IACpC;AAEA,UAAM,YAAY,SAAS,IAAI,SAAS;AACxC,QAAI,CAAC,WAAW;AACd,UAAI,aAAa;AACjB,aAAO,IAAI,IAAI,mBAAmB;AAAA,IACpC;AAEA,WAAO,MAAM,UAAU,kBAAkB,KAAK,GAAG;AAAA,EACnD,WAAW,IAAI,WAAW,OAAO;AAC/B,UAAM,YAAY,IAAI,UAAU,mBAAmB,QAAQ,GAAG;AAC9D,aAAS,IAAI,UAAU,WAAW,SAAS;AAC3C,cAAU,uBAAuB,UAAU,SAAS,EAAE;AACtD,UAAM,UAAU,QAAQ,sBAAsB,WAAW,KAAK;AAC9D,QAAI,GAAG,SAAS,MAAM;AACpB,gBAAU,uBAAuB,UAAU,SAAS,EAAE;AACtD,eAAS,OAAO,UAAU,SAAS;AAAA,IACrC,CAAC;AACD;AAAA,EACF;AAEA,MAAI,aAAa;AACjB,MAAI,IAAI,oBAAoB;AAC9B;AAEA,eAAe,iBAAiB,sBAA4C,KAA2B,KAA0B,UAAsD;AACrL,QAAM,YAAY,IAAI,QAAQ,gBAAgB;AAC9C,MAAI,WAAW;AACb,UAAM,YAAY,SAAS,IAAI,SAAS;AACxC,QAAI,CAAC,WAAW;AACd,UAAI,aAAa;AACjB,UAAI,IAAI,mBAAmB;AAC3B;AAAA,IACF;AACA,WAAO,MAAM,UAAU,cAAc,KAAK,GAAG;AAAA,EAC/C;AAEA,MAAI,IAAI,WAAW,QAAQ;AACzB,UAAM,YAAY,IAAI,UAAU,8BAA8B;AAAA,MAC5D,oBAAoB,MAAM,cAAAF,QAAO,WAAW;AAAA,MAC5C,sBAAsB,OAAMG,eAAa;AACvC,kBAAU,wBAAwB,UAAU,SAAS,EAAE;AACvD,cAAM,UAAU,QAAQ,sBAAsB,WAAW,IAAI;AAC7D,iBAAS,IAAIA,YAAW,SAAS;AAAA,MACnC;AAAA,IACF,CAAC;AAED,cAAU,UAAU,MAAM;AACxB,UAAI,CAAC,UAAU;AACb;AACF,eAAS,OAAO,UAAU,SAAS;AACnC,gBAAU,wBAAwB,UAAU,SAAS,EAAE;AAAA,IACzD;AAEA,UAAM,UAAU,cAAc,KAAK,GAAG;AACtC;AAAA,EACF;AAEA,MAAI,aAAa;AACjB,MAAI,IAAI,iBAAiB;AAC3B;AAEA,SAAS,eAAe,QAAoB;AAC1C,QAAM,UAAU,oBAAI,IAAgB;AACpC,SAAO,GAAG,cAAc,YAAU;AAChC,YAAQ,IAAI,MAAM;AAClB,WAAO,KAAK,SAAS,MAAM,QAAQ,OAAO,MAAM,CAAC;AAAA,EACnD,CAAC;AAED,QAAM,QAAQ,OAAO;AACrB,SAAO,QAAQ,CAAC,aAAqC;AACnD,eAAW,UAAU;AACnB,aAAO,QAAQ;AACjB,YAAQ,MAAM;AACd,WAAO,MAAM,KAAK,QAAQ,QAAQ;AAAA,EACpC;AACF;",
|
|
6
|
+
"names": ["http", "assert", "crypto", "host", "url", "sessionId"]
|
|
7
|
+
}
|
|
@@ -0,0 +1,7 @@
|
|
|
1
|
+
{
|
|
2
|
+
"version": 3,
|
|
3
|
+
"sources": ["../../../src/mcp/sdk/inProcessTransport.ts"],
|
|
4
|
+
"sourcesContent": ["/**\n * Copyright (c) Microsoft Corporation.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport type { Server } from '@modelcontextprotocol/sdk/server/index.js';\nimport type { Transport, TransportSendOptions } from '@modelcontextprotocol/sdk/shared/transport.js';\nimport type { JSONRPCMessage, MessageExtraInfo } from '@modelcontextprotocol/sdk/types.js';\n\nexport class InProcessTransport implements Transport {\n private _server: Server;\n private _serverTransport: InProcessServerTransport;\n private _connected: boolean = false;\n\n constructor(server: Server) {\n this._server = server;\n this._serverTransport = new InProcessServerTransport(this);\n }\n\n async start(): Promise<void> {\n if (this._connected)\n throw new Error('InprocessTransport already started!');\n\n await this._server.connect(this._serverTransport);\n this._connected = true;\n }\n\n async send(message: JSONRPCMessage, options?: TransportSendOptions): Promise<void> {\n if (!this._connected)\n throw new Error('Transport not connected');\n\n\n this._serverTransport._receiveFromClient(message);\n }\n\n async close(): Promise<void> {\n if (this._connected) {\n this._connected = false;\n this.onclose?.();\n this._serverTransport.onclose?.();\n }\n }\n\n onclose?: (() => void) | undefined;\n onerror?: ((error: Error) => void) | undefined;\n onmessage?: ((message: JSONRPCMessage, extra?: MessageExtraInfo) => void) | undefined;\n sessionId?: string | undefined;\n setProtocolVersion?: ((version: string) => void) | undefined;\n\n _receiveFromServer(message: JSONRPCMessage, extra?: MessageExtraInfo): void {\n this.onmessage?.(message, extra);\n }\n}\n\nclass InProcessServerTransport implements Transport {\n private _clientTransport: InProcessTransport;\n\n constructor(clientTransport: InProcessTransport) {\n this._clientTransport = clientTransport;\n }\n\n async start(): Promise<void> {\n }\n\n async send(message: JSONRPCMessage, options?: TransportSendOptions): Promise<void> {\n this._clientTransport._receiveFromServer(message);\n }\n\n async close(): Promise<void> {\n this.onclose?.();\n }\n\n onclose?: (() => void) | undefined;\n onerror?: ((error: Error) => void) | undefined;\n onmessage?: ((message: JSONRPCMessage, extra?: MessageExtraInfo) => void) | undefined;\n sessionId?: string | undefined;\n setProtocolVersion?: ((version: string) => void) | undefined;\n _receiveFromClient(message: JSONRPCMessage): void {\n this.onmessage?.(message);\n }\n}\n"],
|
|
5
|
+
"mappings": ";;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAoBO,MAAM,mBAAwC;AAAA,EAKnD,YAAY,QAAgB;AAF5B,SAAQ,aAAsB;AAG5B,SAAK,UAAU;AACf,SAAK,mBAAmB,IAAI,yBAAyB,IAAI;AAAA,EAC3D;AAAA,EAEA,MAAM,QAAuB;AAC3B,QAAI,KAAK;AACP,YAAM,IAAI,MAAM,qCAAqC;AAEvD,UAAM,KAAK,QAAQ,QAAQ,KAAK,gBAAgB;AAChD,SAAK,aAAa;AAAA,EACpB;AAAA,EAEA,MAAM,KAAK,SAAyB,SAA+C;AACjF,QAAI,CAAC,KAAK;AACR,YAAM,IAAI,MAAM,yBAAyB;AAG3C,SAAK,iBAAiB,mBAAmB,OAAO;AAAA,EAClD;AAAA,EAEA,MAAM,QAAuB;AAC3B,QAAI,KAAK,YAAY;AACnB,WAAK,aAAa;AAClB,WAAK,UAAU;AACf,WAAK,iBAAiB,UAAU;AAAA,IAClC;AAAA,EACF;AAAA,EAQA,mBAAmB,SAAyB,OAAgC;AAC1E,SAAK,YAAY,SAAS,KAAK;AAAA,EACjC;AACF;AAEA,MAAM,yBAA8C;AAAA,EAGlD,YAAY,iBAAqC;AAC/C,SAAK,mBAAmB;AAAA,EAC1B;AAAA,EAEA,MAAM,QAAuB;AAAA,EAC7B;AAAA,EAEA,MAAM,KAAK,SAAyB,SAA+C;AACjF,SAAK,iBAAiB,mBAAmB,OAAO;AAAA,EAClD;AAAA,EAEA,MAAM,QAAuB;AAC3B,SAAK,UAAU;AAAA,EACjB;AAAA,EAOA,mBAAmB,SAA+B;AAChD,SAAK,YAAY,OAAO;AAAA,EAC1B;AACF;",
|
|
6
|
+
"names": []
|
|
7
|
+
}
|
|
@@ -0,0 +1,7 @@
|
|
|
1
|
+
{
|
|
2
|
+
"version": 3,
|
|
3
|
+
"sources": ["../../../src/mcp/sdk/proxyBackend.ts"],
|
|
4
|
+
"sourcesContent": ["/**\n * Copyright (c) Microsoft Corporation.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { debug } from 'playwright-core/lib/utilsBundle';\n\nimport * as mcpBundle from './bundle';\n\nimport type { ServerBackend, ClientInfo } from './server';\nimport type { Transport } from '@modelcontextprotocol/sdk/shared/transport.js';\nimport type { Tool, CallToolResult, CallToolRequest } from '@modelcontextprotocol/sdk/types.js';\nimport type { Client } from '@modelcontextprotocol/sdk/client/index.js';\n\nexport type MCPProvider = {\n name: string;\n description: string;\n connect(): Promise<Transport>;\n};\n\nconst errorsDebug = debug('pw:mcp:errors');\nconst { z, zodToJsonSchema } = mcpBundle;\n\nexport class ProxyBackend implements ServerBackend {\n private _mcpProviders: MCPProvider[];\n private _currentClient: Client | undefined;\n private _contextSwitchTool: Tool;\n private _clientInfo: ClientInfo | undefined;\n\n constructor(mcpProviders: MCPProvider[]) {\n this._mcpProviders = mcpProviders;\n this._contextSwitchTool = this._defineContextSwitchTool();\n }\n\n async initialize(clientInfo: ClientInfo): Promise<void> {\n this._clientInfo = clientInfo;\n }\n\n async listTools(): Promise<Tool[]> {\n const currentClient = await this._ensureCurrentClient();\n const response = await currentClient.listTools();\n if (this._mcpProviders.length === 1)\n return response.tools;\n return [\n ...response.tools,\n this._contextSwitchTool,\n ];\n }\n\n async callTool(name: string, args: CallToolRequest['params']['arguments']): Promise<CallToolResult> {\n if (name === this._contextSwitchTool.name)\n return this._callContextSwitchTool(args);\n const currentClient = await this._ensureCurrentClient();\n return await currentClient.callTool({\n name,\n arguments: args,\n }) as CallToolResult;\n }\n\n serverClosed?(): void {\n void this._currentClient?.close().catch(errorsDebug);\n }\n\n private async _callContextSwitchTool(params: any): Promise<CallToolResult> {\n try {\n const factory = this._mcpProviders.find(factory => factory.name === params.name);\n if (!factory)\n throw new Error('Unknown connection method: ' + params.name);\n\n await this._setCurrentClient(factory);\n return {\n content: [{ type: 'text', text: '### Result\\nSuccessfully changed connection method.\\n' }],\n };\n } catch (error) {\n return {\n content: [{ type: 'text', text: `### Result\\nError: ${error}\\n` }],\n isError: true,\n };\n }\n }\n\n private _defineContextSwitchTool(): Tool {\n return {\n name: 'browser_connect',\n description: [\n 'Connect to a browser using one of the available methods:',\n ...this._mcpProviders.map(factory => `- \"${factory.name}\": ${factory.description}`),\n ].join('\\n'),\n inputSchema: zodToJsonSchema(z.object({\n name: z.enum(this._mcpProviders.map(factory => factory.name) as [string, ...string[]]).default(this._mcpProviders[0].name).describe('The method to use to connect to the browser'),\n }), { strictUnions: true }) as Tool['inputSchema'],\n annotations: {\n title: 'Connect to a browser context',\n readOnlyHint: true,\n openWorldHint: false,\n },\n };\n }\n\n private async _ensureCurrentClient(): Promise<Client> {\n if (this._currentClient)\n return this._currentClient;\n return await this._setCurrentClient(this._mcpProviders[0]);\n }\n\n private async _setCurrentClient(factory: MCPProvider) {\n await this._currentClient?.close();\n this._currentClient = undefined;\n\n const client = new mcpBundle.Client({ name: 'Playwright MCP Proxy', version: '0.0.0' });\n client.registerCapabilities({\n roots: {\n listRoots: true,\n },\n });\n client.setRequestHandler(mcpBundle.ListRootsRequestSchema, () => ({ roots: this._clientInfo?.roots || [] }));\n client.setRequestHandler(mcpBundle.PingRequestSchema, () => ({}));\n\n const transport = await factory.connect();\n await client.connect(transport);\n this._currentClient = client;\n return client;\n }\n}\n"],
|
|
5
|
+
"mappings": ";;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAgBA,yBAAsB;AAEtB,gBAA2B;AAa3B,MAAM,kBAAc,0BAAM,eAAe;AACzC,MAAM,EAAE,GAAG,gBAAgB,IAAI;AAExB,MAAM,aAAsC;AAAA,EAMjD,YAAY,cAA6B;AACvC,SAAK,gBAAgB;AACrB,SAAK,qBAAqB,KAAK,yBAAyB;AAAA,EAC1D;AAAA,EAEA,MAAM,WAAW,YAAuC;AACtD,SAAK,cAAc;AAAA,EACrB;AAAA,EAEA,MAAM,YAA6B;AACjC,UAAM,gBAAgB,MAAM,KAAK,qBAAqB;AACtD,UAAM,WAAW,MAAM,cAAc,UAAU;AAC/C,QAAI,KAAK,cAAc,WAAW;AAChC,aAAO,SAAS;AAClB,WAAO;AAAA,MACL,GAAG,SAAS;AAAA,MACZ,KAAK;AAAA,IACP;AAAA,EACF;AAAA,EAEA,MAAM,SAAS,MAAc,MAAuE;AAClG,QAAI,SAAS,KAAK,mBAAmB;AACnC,aAAO,KAAK,uBAAuB,IAAI;AACzC,UAAM,gBAAgB,MAAM,KAAK,qBAAqB;AACtD,WAAO,MAAM,cAAc,SAAS;AAAA,MAClC;AAAA,MACA,WAAW;AAAA,IACb,CAAC;AAAA,EACH;AAAA,EAEA,eAAsB;AACpB,SAAK,KAAK,gBAAgB,MAAM,EAAE,MAAM,WAAW;AAAA,EACrD;AAAA,EAEA,MAAc,uBAAuB,QAAsC;AACzE,QAAI;AACF,YAAM,UAAU,KAAK,cAAc,KAAK,CAAAA,aAAWA,SAAQ,SAAS,OAAO,IAAI;AAC/E,UAAI,CAAC;AACH,cAAM,IAAI,MAAM,gCAAgC,OAAO,IAAI;AAE7D,YAAM,KAAK,kBAAkB,OAAO;AACpC,aAAO;AAAA,QACL,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,wDAAwD,CAAC;AAAA,MAC3F;AAAA,IACF,SAAS,OAAO;AACd,aAAO;AAAA,QACL,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM;AAAA,SAAsB,KAAK;AAAA,EAAK,CAAC;AAAA,QACjE,SAAS;AAAA,MACX;AAAA,IACF;AAAA,EACF;AAAA,EAEQ,2BAAiC;AACvC,WAAO;AAAA,MACL,MAAM;AAAA,MACN,aAAa;AAAA,QACX;AAAA,QACA,GAAG,KAAK,cAAc,IAAI,aAAW,MAAM,QAAQ,IAAI,MAAM,QAAQ,WAAW,EAAE;AAAA,MACpF,EAAE,KAAK,IAAI;AAAA,MACX,aAAa,gBAAgB,EAAE,OAAO;AAAA,QACpC,MAAM,EAAE,KAAK,KAAK,cAAc,IAAI,aAAW,QAAQ,IAAI,CAA0B,EAAE,QAAQ,KAAK,cAAc,CAAC,EAAE,IAAI,EAAE,SAAS,6CAA6C;AAAA,MACnL,CAAC,GAAG,EAAE,cAAc,KAAK,CAAC;AAAA,MAC1B,aAAa;AAAA,QACX,OAAO;AAAA,QACP,cAAc;AAAA,QACd,eAAe;AAAA,MACjB;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAc,uBAAwC;AACpD,QAAI,KAAK;AACP,aAAO,KAAK;AACd,WAAO,MAAM,KAAK,kBAAkB,KAAK,cAAc,CAAC,CAAC;AAAA,EAC3D;AAAA,EAEA,MAAc,kBAAkB,SAAsB;AACpD,UAAM,KAAK,gBAAgB,MAAM;AACjC,SAAK,iBAAiB;AAEtB,UAAM,SAAS,IAAI,UAAU,OAAO,EAAE,MAAM,wBAAwB,SAAS,QAAQ,CAAC;AACtF,WAAO,qBAAqB;AAAA,MAC1B,OAAO;AAAA,QACL,WAAW;AAAA,MACb;AAAA,IACF,CAAC;AACD,WAAO,kBAAkB,UAAU,wBAAwB,OAAO,EAAE,OAAO,KAAK,aAAa,SAAS,CAAC,EAAE,EAAE;AAC3G,WAAO,kBAAkB,UAAU,mBAAmB,OAAO,CAAC,EAAE;AAEhE,UAAM,YAAY,MAAM,QAAQ,QAAQ;AACxC,UAAM,OAAO,QAAQ,SAAS;AAC9B,SAAK,iBAAiB;AACtB,WAAO;AAAA,EACT;AACF;",
|
|
6
|
+
"names": ["factory"]
|
|
7
|
+
}
|
package/lib/mcp/sdk/server.js
CHANGED
|
@@ -28,16 +28,18 @@ var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__ge
|
|
|
28
28
|
var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
|
|
29
29
|
var server_exports = {};
|
|
30
30
|
__export(server_exports, {
|
|
31
|
+
allRootPaths: () => allRootPaths,
|
|
31
32
|
connect: () => connect,
|
|
32
33
|
createServer: () => createServer,
|
|
33
34
|
firstRootPath: () => firstRootPath,
|
|
34
35
|
start: () => start,
|
|
36
|
+
wrapInClient: () => wrapInClient,
|
|
35
37
|
wrapInProcess: () => wrapInProcess
|
|
36
38
|
});
|
|
37
39
|
module.exports = __toCommonJS(server_exports);
|
|
38
40
|
var import_url = require("url");
|
|
39
41
|
var import_utilsBundle = require("playwright-core/lib/utilsBundle");
|
|
40
|
-
var mcpBundle = __toESM(require("
|
|
42
|
+
var mcpBundle = __toESM(require("playwright-core/lib/mcpBundle"));
|
|
41
43
|
var import_http = require("./http");
|
|
42
44
|
var import_inProcessTransport = require("./inProcessTransport");
|
|
43
45
|
const serverDebug = (0, import_utilsBundle.debug)("pw:mcp:server");
|
|
@@ -46,10 +48,18 @@ async function connect(factory, transport, runHeartbeat) {
|
|
|
46
48
|
const server = createServer(factory.name, factory.version, factory.create(), runHeartbeat);
|
|
47
49
|
await server.connect(transport);
|
|
48
50
|
}
|
|
49
|
-
|
|
51
|
+
function wrapInProcess(backend) {
|
|
50
52
|
const server = createServer("Internal", "0.0.0", backend, false);
|
|
51
53
|
return new import_inProcessTransport.InProcessTransport(server);
|
|
52
54
|
}
|
|
55
|
+
async function wrapInClient(backend, options) {
|
|
56
|
+
const server = createServer("Internal", "0.0.0", backend, false);
|
|
57
|
+
const transport = new import_inProcessTransport.InProcessTransport(server);
|
|
58
|
+
const client = new mcpBundle.Client({ name: options.name, version: options.version });
|
|
59
|
+
await client.connect(transport);
|
|
60
|
+
await client.ping();
|
|
61
|
+
return client;
|
|
62
|
+
}
|
|
53
63
|
function createServer(name, version, backend, runHeartbeat) {
|
|
54
64
|
const server = new mcpBundle.Server({ name, version }, {
|
|
55
65
|
capabilities: {
|
|
@@ -141,8 +151,7 @@ async function start(serverBackendFactory, options) {
|
|
|
141
151
|
await connect(serverBackendFactory, new mcpBundle.StdioServerTransport(), false);
|
|
142
152
|
return;
|
|
143
153
|
}
|
|
144
|
-
const
|
|
145
|
-
const url = await (0, import_http.installHttpTransport)(httpServer, serverBackendFactory, false, options.allowedHosts);
|
|
154
|
+
const url = await (0, import_http.startMcpHttpServer)(options, serverBackendFactory, options.allowedHosts);
|
|
146
155
|
const mcpConfig = { mcpServers: {} };
|
|
147
156
|
mcpConfig.mcpServers[serverBackendFactory.nameInConfig] = {
|
|
148
157
|
url: `${url}/mcp`
|
|
@@ -167,6 +176,20 @@ function firstRootPath(clientInfo) {
|
|
|
167
176
|
return void 0;
|
|
168
177
|
}
|
|
169
178
|
}
|
|
179
|
+
function allRootPaths(clientInfo) {
|
|
180
|
+
const paths = [];
|
|
181
|
+
for (const root of clientInfo.roots) {
|
|
182
|
+
try {
|
|
183
|
+
const url = new URL(root.uri);
|
|
184
|
+
const path = (0, import_url.fileURLToPath)(url);
|
|
185
|
+
if (path)
|
|
186
|
+
paths.push(path);
|
|
187
|
+
} catch (error) {
|
|
188
|
+
serverDebug(error);
|
|
189
|
+
}
|
|
190
|
+
}
|
|
191
|
+
return paths;
|
|
192
|
+
}
|
|
170
193
|
function mergeTextParts(result) {
|
|
171
194
|
const content = [];
|
|
172
195
|
const testParts = [];
|
|
@@ -190,9 +213,11 @@ function mergeTextParts(result) {
|
|
|
190
213
|
}
|
|
191
214
|
// Annotate the CommonJS export names for ESM import in node:
|
|
192
215
|
0 && (module.exports = {
|
|
216
|
+
allRootPaths,
|
|
193
217
|
connect,
|
|
194
218
|
createServer,
|
|
195
219
|
firstRootPath,
|
|
196
220
|
start,
|
|
221
|
+
wrapInClient,
|
|
197
222
|
wrapInProcess
|
|
198
223
|
});
|
|
@@ -0,0 +1,7 @@
|
|
|
1
|
+
{
|
|
2
|
+
"version": 3,
|
|
3
|
+
"sources": ["../../../src/mcp/sdk/server.ts"],
|
|
4
|
+
"sourcesContent": ["/**\n * Copyright (c) Microsoft Corporation.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { fileURLToPath } from 'url';\n\nimport { debug } from 'playwright-core/lib/utilsBundle';\n\nimport * as mcpBundle from './bundle';\nimport { installHttpTransport, startHttpServer } from './http';\nimport { InProcessTransport } from './inProcessTransport';\n\nimport type { Tool, CallToolResult, CallToolRequest, Root } from '@modelcontextprotocol/sdk/types.js';\nimport type { Transport } from '@modelcontextprotocol/sdk/shared/transport.js';\nexport type { Server } from '@modelcontextprotocol/sdk/server/index.js';\nexport type { Tool, CallToolResult, CallToolRequest, Root } from '@modelcontextprotocol/sdk/types.js';\nimport type { Server } from '@modelcontextprotocol/sdk/server/index.js';\n\nconst serverDebug = debug('pw:mcp:server');\nconst serverDebugResponse = debug('pw:mcp:server:response');\n\nexport type ClientInfo = {\n name: string;\n version: string;\n roots: Root[];\n timestamp: number;\n};\n\nexport type ProgressParams = { message?: string, progress?: number, total?: number };\nexport type ProgressCallback = (params: ProgressParams) => void;\n\nexport interface ServerBackend {\n initialize?(clientInfo: ClientInfo): Promise<void>;\n listTools(): Promise<Tool[]>;\n callTool(name: string, args: CallToolRequest['params']['arguments'], progress: ProgressCallback): Promise<CallToolResult>;\n serverClosed?(server: Server): void;\n}\n\nexport type ServerBackendFactory = {\n name: string;\n nameInConfig: string;\n version: string;\n create: () => ServerBackend;\n};\n\nexport async function connect(factory: ServerBackendFactory, transport: Transport, runHeartbeat: boolean) {\n const server = createServer(factory.name, factory.version, factory.create(), runHeartbeat);\n await server.connect(transport);\n}\n\nexport async function wrapInProcess(backend: ServerBackend): Promise<Transport> {\n const server = createServer('Internal', '0.0.0', backend, false);\n return new InProcessTransport(server);\n}\n\nexport function createServer(name: string, version: string, backend: ServerBackend, runHeartbeat: boolean): Server {\n const server = new mcpBundle.Server({ name, version }, {\n capabilities: {\n tools: {},\n }\n });\n\n server.setRequestHandler(mcpBundle.ListToolsRequestSchema, async () => {\n serverDebug('listTools');\n const tools = await backend.listTools();\n return { tools };\n });\n\n let initializePromise: Promise<void> | undefined;\n server.setRequestHandler(mcpBundle.CallToolRequestSchema, async (request, extra) => {\n serverDebug('callTool', request);\n\n const progressToken = request.params._meta?.progressToken;\n let progressCounter = 0;\n\n const progress = progressToken ? (params: ProgressParams) => {\n extra.sendNotification({\n method: 'notifications/progress',\n params: {\n progressToken,\n progress: params.progress ?? ++progressCounter,\n total: params.total,\n message: params.message,\n },\n }).catch(serverDebug);\n } : () => {};\n\n try {\n if (!initializePromise)\n initializePromise = initializeServer(server, backend, runHeartbeat);\n await initializePromise;\n const toolResult = await backend.callTool(request.params.name, request.params.arguments || {}, progress);\n const mergedResult = mergeTextParts(toolResult);\n serverDebugResponse('callResult', mergedResult);\n return mergedResult;\n } catch (error) {\n return {\n content: [{ type: 'text', text: '### Result\\n' + String(error) }],\n isError: true,\n };\n }\n });\n addServerListener(server, 'close', () => backend.serverClosed?.(server));\n return server;\n}\n\nconst initializeServer = async (server: Server, backend: ServerBackend, runHeartbeat: boolean) => {\n const capabilities = server.getClientCapabilities();\n let clientRoots: Root[] = [];\n if (capabilities?.roots) {\n const { roots } = await server.listRoots().catch(e => {\n serverDebug(e);\n return { roots: [] };\n });\n clientRoots = roots;\n }\n\n const clientInfo: ClientInfo = {\n name: server.getClientVersion()?.name ?? 'unknown',\n version: server.getClientVersion()?.version ?? 'unknown',\n roots: clientRoots,\n timestamp: Date.now(),\n };\n\n await backend.initialize?.(clientInfo);\n if (runHeartbeat)\n startHeartbeat(server);\n};\n\nconst startHeartbeat = (server: Server) => {\n const beat = () => {\n Promise.race([\n server.ping(),\n new Promise((_, reject) => setTimeout(() => reject(new Error('ping timeout')), 5000)),\n ]).then(() => {\n setTimeout(beat, 3000);\n }).catch(() => {\n void server.close();\n });\n };\n\n beat();\n};\n\nfunction addServerListener(server: Server, event: 'close' | 'initialized', listener: () => void) {\n const oldListener = server[`on${event}`];\n server[`on${event}`] = () => {\n oldListener?.();\n listener();\n };\n}\n\nexport async function start(serverBackendFactory: ServerBackendFactory, options: { host?: string; port?: number, allowedHosts?: string[] }) {\n if (options.port === undefined) {\n await connect(serverBackendFactory, new mcpBundle.StdioServerTransport(), false);\n return;\n }\n\n const httpServer = await startHttpServer(options);\n const url = await installHttpTransport(httpServer, serverBackendFactory, false, options.allowedHosts);\n\n const mcpConfig: any = { mcpServers: { } };\n mcpConfig.mcpServers[serverBackendFactory.nameInConfig] = {\n url: `${url}/mcp`\n };\n const message = [\n `Listening on ${url}`,\n 'Put this in your client config:',\n JSON.stringify(mcpConfig, undefined, 2),\n 'For legacy SSE transport support, you can use the /sse endpoint instead.',\n ].join('\\n');\n // eslint-disable-next-line no-console\n console.error(message);\n}\n\nexport function firstRootPath(clientInfo: ClientInfo): string | undefined {\n if (clientInfo.roots.length === 0)\n return undefined;\n const firstRootUri = clientInfo.roots[0]?.uri;\n const url = firstRootUri ? new URL(firstRootUri) : undefined;\n try {\n return url ? fileURLToPath(url) : undefined;\n } catch (error) {\n serverDebug(error);\n return undefined;\n }\n}\n\nfunction mergeTextParts(result: CallToolResult): CallToolResult {\n const content: CallToolResult['content'] = [];\n const testParts: string[] = [];\n for (const part of result.content) {\n if (part.type === 'text') {\n testParts.push(part.text);\n continue;\n }\n if (testParts.length > 0) {\n content.push({ type: 'text', text: testParts.join('\\n') });\n testParts.length = 0;\n }\n content.push(part);\n }\n if (testParts.length > 0)\n content.push({ type: 'text', text: testParts.join('\\n') });\n return {\n ...result,\n content,\n };\n}\n"],
|
|
5
|
+
"mappings": ";;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAgBA,iBAA8B;AAE9B,yBAAsB;AAEtB,gBAA2B;AAC3B,kBAAsD;AACtD,gCAAmC;AAQnC,MAAM,kBAAc,0BAAM,eAAe;AACzC,MAAM,0BAAsB,0BAAM,wBAAwB;AA0B1D,eAAsB,QAAQ,SAA+B,WAAsB,cAAuB;AACxG,QAAM,SAAS,aAAa,QAAQ,MAAM,QAAQ,SAAS,QAAQ,OAAO,GAAG,YAAY;AACzF,QAAM,OAAO,QAAQ,SAAS;AAChC;AAEA,eAAsB,cAAc,SAA4C;AAC9E,QAAM,SAAS,aAAa,YAAY,SAAS,SAAS,KAAK;AAC/D,SAAO,IAAI,6CAAmB,MAAM;AACtC;AAEO,SAAS,aAAa,MAAc,SAAiB,SAAwB,cAA+B;AACjH,QAAM,SAAS,IAAI,UAAU,OAAO,EAAE,MAAM,QAAQ,GAAG;AAAA,IACrD,cAAc;AAAA,MACZ,OAAO,CAAC;AAAA,IACV;AAAA,EACF,CAAC;AAED,SAAO,kBAAkB,UAAU,wBAAwB,YAAY;AACrE,gBAAY,WAAW;AACvB,UAAM,QAAQ,MAAM,QAAQ,UAAU;AACtC,WAAO,EAAE,MAAM;AAAA,EACjB,CAAC;AAED,MAAI;AACJ,SAAO,kBAAkB,UAAU,uBAAuB,OAAO,SAAS,UAAU;AAClF,gBAAY,YAAY,OAAO;AAE/B,UAAM,gBAAgB,QAAQ,OAAO,OAAO;AAC5C,QAAI,kBAAkB;AAEtB,UAAM,WAAW,gBAAgB,CAAC,WAA2B;AAC3D,YAAM,iBAAiB;AAAA,QACrB,QAAQ;AAAA,QACR,QAAQ;AAAA,UACN;AAAA,UACA,UAAU,OAAO,YAAY,EAAE;AAAA,UAC/B,OAAO,OAAO;AAAA,UACd,SAAS,OAAO;AAAA,QAClB;AAAA,MACF,CAAC,EAAE,MAAM,WAAW;AAAA,IACtB,IAAI,MAAM;AAAA,IAAC;AAEX,QAAI;AACF,UAAI,CAAC;AACH,4BAAoB,iBAAiB,QAAQ,SAAS,YAAY;AACpE,YAAM;AACN,YAAM,aAAa,MAAM,QAAQ,SAAS,QAAQ,OAAO,MAAM,QAAQ,OAAO,aAAa,CAAC,GAAG,QAAQ;AACvG,YAAM,eAAe,eAAe,UAAU;AAC9C,0BAAoB,cAAc,YAAY;AAC9C,aAAO;AAAA,IACT,SAAS,OAAO;AACd,aAAO;AAAA,QACL,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,iBAAiB,OAAO,KAAK,EAAE,CAAC;AAAA,QAChE,SAAS;AAAA,MACX;AAAA,IACF;AAAA,EACF,CAAC;AACD,oBAAkB,QAAQ,SAAS,MAAM,QAAQ,eAAe,MAAM,CAAC;AACvE,SAAO;AACT;AAEA,MAAM,mBAAmB,OAAO,QAAgB,SAAwB,iBAA0B;AAChG,QAAM,eAAe,OAAO,sBAAsB;AAClD,MAAI,cAAsB,CAAC;AAC3B,MAAI,cAAc,OAAO;AACvB,UAAM,EAAE,MAAM,IAAI,MAAM,OAAO,UAAU,EAAE,MAAM,OAAK;AACpD,kBAAY,CAAC;AACb,aAAO,EAAE,OAAO,CAAC,EAAE;AAAA,IACrB,CAAC;AACD,kBAAc;AAAA,EAChB;AAEA,QAAM,aAAyB;AAAA,IAC7B,MAAM,OAAO,iBAAiB,GAAG,QAAQ;AAAA,IACzC,SAAS,OAAO,iBAAiB,GAAG,WAAW;AAAA,IAC/C,OAAO;AAAA,IACP,WAAW,KAAK,IAAI;AAAA,EACtB;AAEA,QAAM,QAAQ,aAAa,UAAU;AACrC,MAAI;AACF,mBAAe,MAAM;AACzB;AAEA,MAAM,iBAAiB,CAAC,WAAmB;AACzC,QAAM,OAAO,MAAM;AACjB,YAAQ,KAAK;AAAA,MACX,OAAO,KAAK;AAAA,MACZ,IAAI,QAAQ,CAAC,GAAG,WAAW,WAAW,MAAM,OAAO,IAAI,MAAM,cAAc,CAAC,GAAG,GAAI,CAAC;AAAA,IACtF,CAAC,EAAE,KAAK,MAAM;AACZ,iBAAW,MAAM,GAAI;AAAA,IACvB,CAAC,EAAE,MAAM,MAAM;AACb,WAAK,OAAO,MAAM;AAAA,IACpB,CAAC;AAAA,EACH;AAEA,OAAK;AACP;AAEA,SAAS,kBAAkB,QAAgB,OAAgC,UAAsB;AAC/F,QAAM,cAAc,OAAO,KAAK,KAAK,EAAE;AACvC,SAAO,KAAK,KAAK,EAAE,IAAI,MAAM;AAC3B,kBAAc;AACd,aAAS;AAAA,EACX;AACF;AAEA,eAAsB,MAAM,sBAA4C,SAAoE;AAC1I,MAAI,QAAQ,SAAS,QAAW;AAC9B,UAAM,QAAQ,sBAAsB,IAAI,UAAU,qBAAqB,GAAG,KAAK;AAC/E;AAAA,EACF;AAEA,QAAM,aAAa,UAAM,6BAAgB,OAAO;AAChD,QAAM,MAAM,UAAM,kCAAqB,YAAY,sBAAsB,OAAO,QAAQ,YAAY;AAEpG,QAAM,YAAiB,EAAE,YAAY,CAAE,EAAE;AACzC,YAAU,WAAW,qBAAqB,YAAY,IAAI;AAAA,IACxD,KAAK,GAAG,GAAG;AAAA,EACb;AACA,QAAM,UAAU;AAAA,IACd,gBAAgB,GAAG;AAAA,IACnB;AAAA,IACA,KAAK,UAAU,WAAW,QAAW,CAAC;AAAA,IACtC;AAAA,EACF,EAAE,KAAK,IAAI;AAEX,UAAQ,MAAM,OAAO;AACvB;AAEO,SAAS,cAAc,YAA4C;AACxE,MAAI,WAAW,MAAM,WAAW;AAC9B,WAAO;AACT,QAAM,eAAe,WAAW,MAAM,CAAC,GAAG;AAC1C,QAAM,MAAM,eAAe,IAAI,IAAI,YAAY,IAAI;AACnD,MAAI;AACF,WAAO,UAAM,0BAAc,GAAG,IAAI;AAAA,EACpC,SAAS,OAAO;AACd,gBAAY,KAAK;AACjB,WAAO;AAAA,EACT;AACF;AAEA,SAAS,eAAe,QAAwC;AAC9D,QAAM,UAAqC,CAAC;AAC5C,QAAM,YAAsB,CAAC;AAC7B,aAAW,QAAQ,OAAO,SAAS;AACjC,QAAI,KAAK,SAAS,QAAQ;AACxB,gBAAU,KAAK,KAAK,IAAI;AACxB;AAAA,IACF;AACA,QAAI,UAAU,SAAS,GAAG;AACxB,cAAQ,KAAK,EAAE,MAAM,QAAQ,MAAM,UAAU,KAAK,IAAI,EAAE,CAAC;AACzD,gBAAU,SAAS;AAAA,IACrB;AACA,YAAQ,KAAK,IAAI;AAAA,EACnB;AACA,MAAI,UAAU,SAAS;AACrB,YAAQ,KAAK,EAAE,MAAM,QAAQ,MAAM,UAAU,KAAK,IAAI,EAAE,CAAC;AAC3D,SAAO;AAAA,IACL,GAAG;AAAA,IACH;AAAA,EACF;AACF;",
|
|
6
|
+
"names": []
|
|
7
|
+
}
|
package/lib/mcp/sdk/tool.js
CHANGED
|
@@ -22,13 +22,13 @@ __export(tool_exports, {
|
|
|
22
22
|
toMcpTool: () => toMcpTool
|
|
23
23
|
});
|
|
24
24
|
module.exports = __toCommonJS(tool_exports);
|
|
25
|
-
var
|
|
25
|
+
var import_mcpBundle = require("playwright-core/lib/mcpBundle");
|
|
26
26
|
function toMcpTool(tool) {
|
|
27
27
|
const readOnly = tool.type === "readOnly" || tool.type === "assertion";
|
|
28
28
|
return {
|
|
29
29
|
name: tool.name,
|
|
30
30
|
description: tool.description,
|
|
31
|
-
inputSchema:
|
|
31
|
+
inputSchema: import_mcpBundle.z.toJSONSchema(tool.inputSchema),
|
|
32
32
|
annotations: {
|
|
33
33
|
title: tool.title,
|
|
34
34
|
readOnlyHint: readOnly,
|
|
@@ -0,0 +1,7 @@
|
|
|
1
|
+
{
|
|
2
|
+
"version": 3,
|
|
3
|
+
"sources": ["../../../src/mcp/sdk/tool.ts"],
|
|
4
|
+
"sourcesContent": ["/**\n * Copyright (c) Microsoft Corporation.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { zodToJsonSchema } from '../sdk/bundle';\n\nimport type { z } from 'zod';\nimport type * as mcpServer from './server';\n\nexport type ToolSchema<Input extends z.Schema> = {\n name: string;\n title: string;\n description: string;\n inputSchema: Input;\n type: 'input' | 'assertion' | 'action' | 'readOnly';\n};\n\nexport function toMcpTool(tool: ToolSchema<any>): mcpServer.Tool {\n const readOnly = tool.type === 'readOnly' || tool.type === 'assertion';\n return {\n name: tool.name,\n description: tool.description,\n inputSchema: zodToJsonSchema(tool.inputSchema, { strictUnions: true }) as mcpServer.Tool['inputSchema'],\n annotations: {\n title: tool.title,\n readOnlyHint: readOnly,\n destructiveHint: !readOnly,\n openWorldHint: true,\n },\n };\n}\n\nexport function defineToolSchema<Input extends z.Schema>(tool: ToolSchema<Input>): ToolSchema<Input> {\n return tool;\n}\n"],
|
|
5
|
+
"mappings": ";;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAgBA,oBAAiC;AAa1B,SAAS,UAAU,MAAuC;AAC/D,QAAM,WAAW,KAAK,SAAS,cAAc,KAAK,SAAS;AAC3D,SAAO;AAAA,IACL,MAAM,KAAK;AAAA,IACX,aAAa,KAAK;AAAA,IAClB,iBAAa,+BAAgB,KAAK,aAAa,EAAE,cAAc,KAAK,CAAC;AAAA,IACrE,aAAa;AAAA,MACX,OAAO,KAAK;AAAA,MACZ,cAAc;AAAA,MACd,iBAAiB,CAAC;AAAA,MAClB,eAAe;AAAA,IACjB;AAAA,EACF;AACF;AAEO,SAAS,iBAAyC,MAA4C;AACnG,SAAO;AACT;",
|
|
6
|
+
"names": []
|
|
7
|
+
}
|
|
@@ -0,0 +1,296 @@
|
|
|
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 __copyProps = (to, from, except, desc) => {
|
|
9
|
+
if (from && typeof from === "object" || typeof from === "function") {
|
|
10
|
+
for (let key of __getOwnPropNames(from))
|
|
11
|
+
if (!__hasOwnProp.call(to, key) && key !== except)
|
|
12
|
+
__defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
|
|
13
|
+
}
|
|
14
|
+
return to;
|
|
15
|
+
};
|
|
16
|
+
var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
|
|
17
|
+
// If the importer is in node compatibility mode or this is not an ESM
|
|
18
|
+
// file that has been converted to a CommonJS file using a Babel-
|
|
19
|
+
// compatible transform (i.e. "__esModule" has not been set), then set
|
|
20
|
+
// "default" to the CommonJS "module.exports" for node compatibility.
|
|
21
|
+
isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
|
|
22
|
+
mod
|
|
23
|
+
));
|
|
24
|
+
var import_child_process = require("child_process");
|
|
25
|
+
var import_crypto = __toESM(require("crypto"));
|
|
26
|
+
var import_fs = __toESM(require("fs"));
|
|
27
|
+
var import_net = __toESM(require("net"));
|
|
28
|
+
var import_os = __toESM(require("os"));
|
|
29
|
+
var import_path = __toESM(require("path"));
|
|
30
|
+
var import_utilsBundle = require("playwright-core/lib/utilsBundle");
|
|
31
|
+
var import_socketConnection = require("./socketConnection");
|
|
32
|
+
const debugCli = (0, import_utilsBundle.debug)("pw:cli");
|
|
33
|
+
const packageJSON = require("../../../package.json");
|
|
34
|
+
async function runCliCommand(sessionName, args) {
|
|
35
|
+
const session = await connectToDaemon(sessionName);
|
|
36
|
+
const result = await session.runCliCommand(args);
|
|
37
|
+
console.log(result);
|
|
38
|
+
session.dispose();
|
|
39
|
+
}
|
|
40
|
+
async function socketExists(socketPath) {
|
|
41
|
+
try {
|
|
42
|
+
const stat = await import_fs.default.promises.stat(socketPath);
|
|
43
|
+
if (stat?.isSocket())
|
|
44
|
+
return true;
|
|
45
|
+
} catch (e) {
|
|
46
|
+
}
|
|
47
|
+
return false;
|
|
48
|
+
}
|
|
49
|
+
class SocketSession {
|
|
50
|
+
constructor(connection) {
|
|
51
|
+
this._nextMessageId = 1;
|
|
52
|
+
this._callbacks = /* @__PURE__ */ new Map();
|
|
53
|
+
this._connection = connection;
|
|
54
|
+
this._connection.onmessage = (message) => this._onMessage(message);
|
|
55
|
+
this._connection.onclose = () => this.dispose();
|
|
56
|
+
}
|
|
57
|
+
async callTool(name, args) {
|
|
58
|
+
return this._send(name, args);
|
|
59
|
+
}
|
|
60
|
+
async runCliCommand(args) {
|
|
61
|
+
return await this._send("runCliCommand", { args });
|
|
62
|
+
}
|
|
63
|
+
async _send(method, params = {}) {
|
|
64
|
+
const messageId = this._nextMessageId++;
|
|
65
|
+
const message = {
|
|
66
|
+
id: messageId,
|
|
67
|
+
method,
|
|
68
|
+
params
|
|
69
|
+
};
|
|
70
|
+
await this._connection.send(message);
|
|
71
|
+
return new Promise((resolve, reject) => {
|
|
72
|
+
this._callbacks.set(messageId, { resolve, reject });
|
|
73
|
+
});
|
|
74
|
+
}
|
|
75
|
+
dispose() {
|
|
76
|
+
for (const callback of this._callbacks.values())
|
|
77
|
+
callback.reject(new Error("Disposed"));
|
|
78
|
+
this._callbacks.clear();
|
|
79
|
+
this._connection.close();
|
|
80
|
+
}
|
|
81
|
+
_onMessage(object) {
|
|
82
|
+
if (object.id && this._callbacks.has(object.id)) {
|
|
83
|
+
const callback = this._callbacks.get(object.id);
|
|
84
|
+
this._callbacks.delete(object.id);
|
|
85
|
+
if (object.error)
|
|
86
|
+
callback.reject(new Error(object.error));
|
|
87
|
+
else
|
|
88
|
+
callback.resolve(object.result);
|
|
89
|
+
} else if (object.id) {
|
|
90
|
+
throw new Error(`Unexpected message id: ${object.id}`);
|
|
91
|
+
} else {
|
|
92
|
+
throw new Error(`Unexpected message without id: ${JSON.stringify(object)}`);
|
|
93
|
+
}
|
|
94
|
+
}
|
|
95
|
+
}
|
|
96
|
+
function localCacheDir() {
|
|
97
|
+
if (process.platform === "linux")
|
|
98
|
+
return process.env.XDG_CACHE_HOME || import_path.default.join(import_os.default.homedir(), ".cache");
|
|
99
|
+
if (process.platform === "darwin")
|
|
100
|
+
return import_path.default.join(import_os.default.homedir(), "Library", "Caches");
|
|
101
|
+
if (process.platform === "win32")
|
|
102
|
+
return process.env.LOCALAPPDATA || import_path.default.join(import_os.default.homedir(), "AppData", "Local");
|
|
103
|
+
throw new Error("Unsupported platform: " + process.platform);
|
|
104
|
+
}
|
|
105
|
+
function playwrightCacheDir() {
|
|
106
|
+
return import_path.default.join(localCacheDir(), "ms-playwright");
|
|
107
|
+
}
|
|
108
|
+
function calculateSha1(buffer) {
|
|
109
|
+
const hash = import_crypto.default.createHash("sha1");
|
|
110
|
+
hash.update(buffer);
|
|
111
|
+
return hash.digest("hex");
|
|
112
|
+
}
|
|
113
|
+
function socketDirHash() {
|
|
114
|
+
return calculateSha1(__dirname);
|
|
115
|
+
}
|
|
116
|
+
function daemonSocketDir() {
|
|
117
|
+
return import_path.default.resolve(playwrightCacheDir(), "daemon", socketDirHash());
|
|
118
|
+
}
|
|
119
|
+
function daemonSocketPath(sessionName) {
|
|
120
|
+
const socketName = `${sessionName}.sock`;
|
|
121
|
+
if (import_os.default.platform() === "win32")
|
|
122
|
+
return `\\\\.\\pipe\\${socketDirHash()}-${socketName}`;
|
|
123
|
+
return import_path.default.resolve(daemonSocketDir(), socketName);
|
|
124
|
+
}
|
|
125
|
+
async function connectToDaemon(sessionName) {
|
|
126
|
+
const socketPath = daemonSocketPath(sessionName);
|
|
127
|
+
debugCli(`Connecting to daemon at ${socketPath}`);
|
|
128
|
+
if (await socketExists(socketPath)) {
|
|
129
|
+
debugCli(`Socket file exists, attempting to connect...`);
|
|
130
|
+
try {
|
|
131
|
+
return await connectToSocket(socketPath);
|
|
132
|
+
} catch (e) {
|
|
133
|
+
if (import_os.default.platform() !== "win32")
|
|
134
|
+
await import_fs.default.promises.unlink(socketPath).catch(() => {
|
|
135
|
+
});
|
|
136
|
+
}
|
|
137
|
+
}
|
|
138
|
+
const cliPath = import_path.default.join(__dirname, "../../../cli.js");
|
|
139
|
+
debugCli(`Will launch daemon process: ${cliPath}`);
|
|
140
|
+
const userDataDir = import_path.default.resolve(daemonSocketDir(), `${sessionName}-user-data`);
|
|
141
|
+
const child = (0, import_child_process.spawn)(process.execPath, [cliPath, "run-mcp-server", `--daemon=${socketPath}`, `--user-data-dir=${userDataDir}`], {
|
|
142
|
+
detached: true,
|
|
143
|
+
stdio: "ignore",
|
|
144
|
+
cwd: process.cwd()
|
|
145
|
+
// Will be used as root.
|
|
146
|
+
});
|
|
147
|
+
child.unref();
|
|
148
|
+
const maxRetries = 50;
|
|
149
|
+
const retryDelay = 100;
|
|
150
|
+
for (let i = 0; i < maxRetries; i++) {
|
|
151
|
+
await new Promise((resolve) => setTimeout(resolve, 100));
|
|
152
|
+
try {
|
|
153
|
+
return await connectToSocket(socketPath);
|
|
154
|
+
} catch (e) {
|
|
155
|
+
if (e.code !== "ENOENT")
|
|
156
|
+
throw e;
|
|
157
|
+
debugCli(`Retrying to connect to daemon at ${socketPath} (${i + 1}/${maxRetries})`);
|
|
158
|
+
}
|
|
159
|
+
}
|
|
160
|
+
throw new Error(`Failed to connect to daemon at ${socketPath} after ${maxRetries * retryDelay}ms`);
|
|
161
|
+
}
|
|
162
|
+
async function connectToSocket(socketPath) {
|
|
163
|
+
const socket = await new Promise((resolve, reject) => {
|
|
164
|
+
const socket2 = import_net.default.createConnection(socketPath, () => {
|
|
165
|
+
debugCli(`Connected to daemon at ${socketPath}`);
|
|
166
|
+
resolve(socket2);
|
|
167
|
+
});
|
|
168
|
+
socket2.on("error", reject);
|
|
169
|
+
});
|
|
170
|
+
return new SocketSession(new import_socketConnection.SocketConnection(socket));
|
|
171
|
+
}
|
|
172
|
+
function currentSessionPath() {
|
|
173
|
+
return import_path.default.resolve(daemonSocketDir(), "current-session");
|
|
174
|
+
}
|
|
175
|
+
async function getCurrentSession() {
|
|
176
|
+
try {
|
|
177
|
+
const session = await import_fs.default.promises.readFile(currentSessionPath(), "utf-8");
|
|
178
|
+
return session.trim() || "default";
|
|
179
|
+
} catch {
|
|
180
|
+
return "default";
|
|
181
|
+
}
|
|
182
|
+
}
|
|
183
|
+
async function setCurrentSession(sessionName) {
|
|
184
|
+
await import_fs.default.promises.mkdir(daemonSocketDir(), { recursive: true });
|
|
185
|
+
await import_fs.default.promises.writeFile(currentSessionPath(), sessionName);
|
|
186
|
+
}
|
|
187
|
+
async function canConnectToSocket(socketPath) {
|
|
188
|
+
return new Promise((resolve) => {
|
|
189
|
+
const socket = import_net.default.createConnection(socketPath, () => {
|
|
190
|
+
socket.destroy();
|
|
191
|
+
resolve(true);
|
|
192
|
+
});
|
|
193
|
+
socket.on("error", () => {
|
|
194
|
+
resolve(false);
|
|
195
|
+
});
|
|
196
|
+
});
|
|
197
|
+
}
|
|
198
|
+
async function listSessions() {
|
|
199
|
+
const dir = daemonSocketDir();
|
|
200
|
+
try {
|
|
201
|
+
const files = await import_fs.default.promises.readdir(dir);
|
|
202
|
+
const sessions = [];
|
|
203
|
+
for (const file of files) {
|
|
204
|
+
if (file.endsWith("-user-data")) {
|
|
205
|
+
const sessionName = file.slice(0, -"-user-data".length);
|
|
206
|
+
const socketPath = daemonSocketPath(sessionName);
|
|
207
|
+
const live = await canConnectToSocket(socketPath);
|
|
208
|
+
sessions.push({ name: sessionName, live });
|
|
209
|
+
}
|
|
210
|
+
}
|
|
211
|
+
return sessions;
|
|
212
|
+
} catch {
|
|
213
|
+
return [];
|
|
214
|
+
}
|
|
215
|
+
}
|
|
216
|
+
function resolveSessionName(args) {
|
|
217
|
+
if (args.session)
|
|
218
|
+
return args.session;
|
|
219
|
+
if (process.env.PLAYWRIGHT_CLI_SESSION)
|
|
220
|
+
return process.env.PLAYWRIGHT_CLI_SESSION;
|
|
221
|
+
return "default";
|
|
222
|
+
}
|
|
223
|
+
async function handleSessionCommand(args) {
|
|
224
|
+
const subcommand = args._[1];
|
|
225
|
+
if (!subcommand) {
|
|
226
|
+
const current = await getCurrentSession();
|
|
227
|
+
console.log(current);
|
|
228
|
+
return;
|
|
229
|
+
}
|
|
230
|
+
if (subcommand === "list") {
|
|
231
|
+
const sessions = await listSessions();
|
|
232
|
+
const current = await getCurrentSession();
|
|
233
|
+
console.log("Sessions:");
|
|
234
|
+
for (const session of sessions) {
|
|
235
|
+
const marker = session.name === current ? "->" : " ";
|
|
236
|
+
const liveMarker = session.live ? " (live)" : "";
|
|
237
|
+
console.log(`${marker} ${session.name}${liveMarker}`);
|
|
238
|
+
}
|
|
239
|
+
if (sessions.length === 0)
|
|
240
|
+
console.log(" (no sessions)");
|
|
241
|
+
return;
|
|
242
|
+
}
|
|
243
|
+
if (subcommand === "set") {
|
|
244
|
+
const sessionName = args._[2];
|
|
245
|
+
if (!sessionName) {
|
|
246
|
+
console.error("Usage: playwright-cli session set <session-name>");
|
|
247
|
+
process.exit(1);
|
|
248
|
+
}
|
|
249
|
+
await setCurrentSession(sessionName);
|
|
250
|
+
console.log(`Current session set to: ${sessionName}`);
|
|
251
|
+
return;
|
|
252
|
+
}
|
|
253
|
+
console.error(`Unknown session subcommand: ${subcommand}`);
|
|
254
|
+
process.exit(1);
|
|
255
|
+
}
|
|
256
|
+
async function main() {
|
|
257
|
+
const argv = process.argv.slice(2);
|
|
258
|
+
const args = require("minimist")(argv);
|
|
259
|
+
const help = require("./help.json");
|
|
260
|
+
const commandName = args._[0];
|
|
261
|
+
if (args.version || args.v) {
|
|
262
|
+
console.log(packageJSON.version);
|
|
263
|
+
process.exit(0);
|
|
264
|
+
}
|
|
265
|
+
if (commandName === "session") {
|
|
266
|
+
await handleSessionCommand(args);
|
|
267
|
+
return;
|
|
268
|
+
}
|
|
269
|
+
const command = help.commands[commandName];
|
|
270
|
+
if (args.help || args.h) {
|
|
271
|
+
if (command) {
|
|
272
|
+
console.log(command);
|
|
273
|
+
} else {
|
|
274
|
+
console.log("playwright-cli - run playwright mcp commands from terminal\n");
|
|
275
|
+
console.log(help.global);
|
|
276
|
+
}
|
|
277
|
+
process.exit(0);
|
|
278
|
+
}
|
|
279
|
+
if (!command) {
|
|
280
|
+
console.error(`Unknown command: ${commandName}
|
|
281
|
+
`);
|
|
282
|
+
console.log(help.global);
|
|
283
|
+
process.exit(1);
|
|
284
|
+
}
|
|
285
|
+
let sessionName = resolveSessionName(args);
|
|
286
|
+
if (sessionName === "default" && !args.session && !process.env.PLAYWRIGHT_CLI_SESSION)
|
|
287
|
+
sessionName = await getCurrentSession();
|
|
288
|
+
runCliCommand(sessionName, args).catch((e) => {
|
|
289
|
+
console.error(e.message);
|
|
290
|
+
process.exit(1);
|
|
291
|
+
});
|
|
292
|
+
}
|
|
293
|
+
main().catch((e) => {
|
|
294
|
+
console.error(e.message);
|
|
295
|
+
process.exit(1);
|
|
296
|
+
});
|