vite-plugin-opencode-assistant 1.1.24 → 1.1.25

package/es/endpoints/mcp-tools.mjs

@@ -0,0 +1,325 @@
+const CUSTOM_TOOLS = [
+  {
+    name: "devtools_select_page",
+    description: "\u9009\u62E9\u5F53\u524D\u9879\u76EE\u4E2D\u7684\u67D0\u4E2A\u9875\u9762\u4F5C\u4E3A\u540E\u7EED\u5DE5\u5177\u64CD\u4F5C\u7684\u76EE\u6807",
+    inputSchema: {
+      type: "object",
+      properties: {
+        pageId: { type: "number", description: "\u9875\u9762 ID\uFF08\u4ECE devtools_list_pages \u83B7\u53D6\uFF09" }
+      },
+      required: ["pageId"]
+    }
+  },
+  {
+    name: "devtools_list_pages",
+    description: "\u83B7\u53D6\u5F53\u524D\u9879\u76EE\u6240\u6709\u6253\u5F00\u7684\u9875\u9762\u5217\u8868\uFF0C\u542B active\uFF08\u7528\u6237\u6B63\u5728\u6D4F\u89C8\uFF09\u548C selected\uFF08Chrome DevTools \u5F53\u524D\u64CD\u4F5C\u76EE\u6807\uFF09\u6807\u8BB0",
+    inputSchema: {
+      type: "object",
+      properties: {}
+    }
+  },
+  {
+    name: "devtools_snapshot",
+    description: "\u83B7\u53D6\u5F53\u524D\u9875\u9762\u53EF\u8BBF\u95EE\u6027\u6811\u5FEB\u7167\uFF0C\u8FD4\u56DE\u5143\u7D20 uid\u3001\u89D2\u8272\u3001\u6587\u672C\u7B49",
+    inputSchema: {
+      type: "object",
+      properties: {
+        verbose: { type: "boolean", description: "\u662F\u5426\u83B7\u53D6\u5B8C\u6574\u8282\u70B9\u4FE1\u606F" }
+      }
+    }
+  },
+  {
+    name: "devtools_screenshot",
+    description: "\u622A\u53D6\u5F53\u524D\u9875\u9762\u5C4F\u5E55\u622A\u56FE\uFF0C\u8FD4\u56DE base64 \u56FE\u7247\u6570\u636E",
+    inputSchema: {
+      type: "object",
+      properties: {
+        format: { type: "string", description: "\u56FE\u7247\u683C\u5F0F: png \u6216 jpeg" },
+        quality: { type: "number", description: "jpeg \u8D28\u91CF 0-100" },
+        fullPage: { type: "boolean", description: "\u662F\u5426\u622A\u53D6\u5B8C\u6574\u9875\u9762" }
+      }
+    }
+  },
+  {
+    name: "devtools_evaluate",
+    description: "\u5728\u5F53\u524D\u9875\u9762\u6267\u884C JavaScript \u4EE3\u7801\u5E76\u8FD4\u56DE\u7ED3\u679C\uFF08\u8FD4\u56DE\u503C\u9700\u53EF JSON \u5E8F\u5217\u5316\uFF09",
+    inputSchema: {
+      type: "object",
+      properties: {
+        function: { type: "string", description: "JS \u51FD\u6570\u58F0\u660E\uFF0C\u5982 () => document.title" }
+      },
+      required: ["function"]
+    }
+  },
+  {
+    name: "devtools_click",
+    description: "\u70B9\u51FB\u9875\u9762\u5143\u7D20",
+    inputSchema: {
+      type: "object",
+      properties: {
+        uid: { type: "string", description: "\u5143\u7D20 uid\uFF08\u4ECE devtools_snapshot \u83B7\u53D6\uFF09" },
+        dblClick: { type: "boolean", description: "\u662F\u5426\u53CC\u51FB" }
+      },
+      required: ["uid"]
+    }
+  },
+  {
+    name: "devtools_hover",
+    description: "\u9F20\u6807\u60AC\u505C\u5728\u9875\u9762\u5143\u7D20\u4E0A",
+    inputSchema: {
+      type: "object",
+      properties: {
+        uid: { type: "string", description: "\u5143\u7D20 uid" }
+      },
+      required: ["uid"]
+    }
+  },
+  {
+    name: "devtools_drag",
+    description: "\u62D6\u62FD\u9875\u9762\u5143\u7D20",
+    inputSchema: {
+      type: "object",
+      properties: {
+        uid: { type: "string", description: "\u6E90\u5143\u7D20 uid" },
+        targetUid: { type: "string", description: "\u76EE\u6807\u5143\u7D20 uid" }
+      },
+      required: ["uid", "targetUid"]
+    }
+  },
+  {
+    name: "devtools_type",
+    description: "\u5728\u805A\u7126\u5143\u7D20\u4E2D\u8F93\u5165\u6587\u672C",
+    inputSchema: {
+      type: "object",
+      properties: {
+        text: { type: "string", description: "\u8981\u8F93\u5165\u7684\u6587\u672C" }
+      },
+      required: ["text"]
+    }
+  },
+  {
+    name: "devtools_press_key",
+    description: "\u6309\u4E0B\u952E\u76D8\u6309\u952E",
+    inputSchema: {
+      type: "object",
+      properties: {
+        key: { type: "string", description: "\u6309\u952E\u540D\u79F0\uFF0C\u5982 Enter\u3001Control+A" }
+      },
+      required: ["key"]
+    }
+  },
+  {
+    name: "devtools_fill",
+    description: "\u586B\u5199\u8F93\u5165\u6846\u503C\u5E76\u89E6\u53D1 input/change \u4E8B\u4EF6",
+    inputSchema: {
+      type: "object",
+      properties: {
+        uid: { type: "string", description: "\u8F93\u5165\u6846\u5143\u7D20 uid" },
+        value: { type: "string", description: "\u8981\u586B\u5165\u7684\u503C" }
+      },
+      required: ["uid", "value"]
+    }
+  },
+  {
+    name: "devtools_fill_form",
+    description: "\u6279\u91CF\u586B\u5199\u8868\u5355\u5B57\u6BB5",
+    inputSchema: {
+      type: "object",
+      properties: {
+        fields: {
+          type: "array",
+          items: {
+            type: "object",
+            properties: { uid: { type: "string" }, value: { type: "string" } }
+          },
+          description: "\u5B57\u6BB5\u6570\u7EC4 [{ uid, value }]"
+        }
+      },
+      required: ["fields"]
+    }
+  },
+  {
+    name: "devtools_navigate",
+    description: "\u5BFC\u822A\u9875\u9762\uFF08url/reload/back/forward\uFF09",
+    inputSchema: {
+      type: "object",
+      properties: {
+        type: { type: "string", description: "url | reload | back | forward" },
+        url: { type: "string", description: "\u76EE\u6807 URL\uFF08type=url \u65F6\uFF09" }
+      },
+      required: ["type"]
+    }
+  },
+  {
+    name: "devtools_resize_page",
+    description: "\u8C03\u6574\u9875\u9762\u89C6\u53E3\u5927\u5C0F",
+    inputSchema: {
+      type: "object",
+      properties: {
+        width: { type: "number", description: "\u89C6\u53E3\u5BBD\u5EA6" },
+        height: { type: "number", description: "\u89C6\u53E3\u9AD8\u5EA6" }
+      },
+      required: ["width", "height"]
+    }
+  },
+  {
+    name: "devtools_emulate",
+    description: "\u6A21\u62DF\u79FB\u52A8\u8BBE\u5907",
+    inputSchema: {
+      type: "object",
+      properties: {
+        device: { type: "string", description: "\u8BBE\u5907\u540D\u79F0\uFF0C\u5982 iPhone 15" }
+      },
+      required: ["device"]
+    }
+  },
+  {
+    name: "devtools_network",
+    description: "\u83B7\u53D6\u5F53\u524D\u9875\u9762\u7F51\u7EDC\u8BF7\u6C42\u5217\u8868",
+    inputSchema: { type: "object", properties: {} }
+  },
+  {
+    name: "devtools_network_request",
+    description: "\u83B7\u53D6\u67D0\u6761\u7F51\u7EDC\u8BF7\u6C42\u7684\u8BE6\u7EC6\u4FE1\u606F",
+    inputSchema: {
+      type: "object",
+      properties: {
+        reqid: { type: "number", description: "\u8BF7\u6C42 ID\uFF08\u4ECE devtools_network \u83B7\u53D6\uFF09" }
+      },
+      required: ["reqid"]
+    }
+  },
+  {
+    name: "devtools_console",
+    description: "\u83B7\u53D6\u5F53\u524D\u9875\u9762\u63A7\u5236\u53F0\u6D88\u606F",
+    inputSchema: { type: "object", properties: {} }
+  },
+  {
+    name: "devtools_console_message",
+    description: "\u83B7\u53D6\u67D0\u6761\u63A7\u5236\u53F0\u6D88\u606F\u7684\u8BE6\u7EC6\u4FE1\u606F",
+    inputSchema: {
+      type: "object",
+      properties: {
+        msgid: { type: "number", description: "\u6D88\u606F ID\uFF08\u4ECE devtools_console \u83B7\u53D6\uFF09" }
+      },
+      required: ["msgid"]
+    }
+  },
+  {
+    name: "devtools_wait_for",
+    description: "\u7B49\u5F85\u9875\u9762\u51FA\u73B0\u6307\u5B9A\u6587\u672C",
+    inputSchema: {
+      type: "object",
+      properties: {
+        text: { type: "string", description: "\u7B49\u5F85\u51FA\u73B0\u7684\u6587\u672C" }
+      },
+      required: ["text"]
+    }
+  },
+  {
+    name: "devtools_handle_dialog",
+    description: "\u5904\u7406 JavaScript \u5BF9\u8BDD\u6846\uFF08accept/dismiss/\u8F93\u5165\u6587\u672C\uFF09",
+    inputSchema: {
+      type: "object",
+      properties: {
+        action: { type: "string", description: "accept | dismiss | prompt \u8F93\u5165\u6587\u672C" }
+      },
+      required: ["action"]
+    }
+  },
+  {
+    name: "devtools_upload_file",
+    description: "\u4E0A\u4F20\u6587\u4EF6\u5230\u6587\u4EF6\u8F93\u5165\u6846",
+    inputSchema: {
+      type: "object",
+      properties: {
+        uid: { type: "string", description: "\u6587\u4EF6\u8F93\u5165\u6846\u5143\u7D20 uid" },
+        filePath: { type: "string", description: "\u6587\u4EF6\u7EDD\u5BF9\u8DEF\u5F84" }
+      },
+      required: ["uid", "filePath"]
+    }
+  },
+  {
+    name: "devtools_performance_start",
+    description: "\u5F00\u59CB\u8BB0\u5F55\u6027\u80FD trace",
+    inputSchema: {
+      type: "object",
+      properties: {
+        reload: { type: "boolean", description: "\u662F\u5426\u81EA\u52A8\u5237\u65B0" },
+        autoStop: { type: "boolean", description: "\u662F\u5426\u81EA\u52A8\u505C\u6B62" }
+      }
+    }
+  },
+  {
+    name: "devtools_performance_stop",
+    description: "\u505C\u6B62\u6027\u80FD trace \u8BB0\u5F55\u5E76\u8FD4\u56DE\u7ED3\u679C",
+    inputSchema: { type: "object", properties: {} }
+  },
+  {
+    name: "devtools_performance_insight",
+    description: "\u83B7\u53D6\u6027\u80FD\u6307\u6807\u8BE6\u7EC6\u5206\u6790",
+    inputSchema: {
+      type: "object",
+      properties: {
+        insightSetId: { type: "string", description: "\u6307\u6807\u96C6 ID" },
+        insightName: { type: "string", description: "\u6307\u6807\u540D\u79F0" }
+      },
+      required: ["insightSetId", "insightName"]
+    }
+  },
+  {
+    name: "devtools_lighthouse",
+    description: "\u8FD0\u884C Lighthouse \u5BA1\u8BA1\uFF08\u53EF\u8BBF\u95EE\u6027/SEO/\u6700\u4F73\u5B9E\u8DF5\uFF0C\u4E0D\u542B\u6027\u80FD\uFF09",
+    inputSchema: {
+      type: "object",
+      properties: {
+        mode: { type: "string", description: "navigation \u6216 snapshot" },
+        device: { type: "string", description: "desktop \u6216 mobile" }
+      }
+    }
+  },
+  {
+    name: "devtools_heapsnapshot",
+    description: "\u6355\u83B7\u5806\u5185\u5B58\u5FEB\u7167",
+    inputSchema: {
+      type: "object",
+      properties: {
+        filePath: { type: "string", description: "\u4FDD\u5B58\u8DEF\u5F84" }
+      },
+      required: ["filePath"]
+    }
+  }
+];
+const TOOL_MAP = {
+  devtools_select_page: "select_page",
+  devtools_list_pages: "list_pages",
+  devtools_snapshot: "take_snapshot",
+  devtools_screenshot: "take_screenshot",
+  devtools_evaluate: "evaluate_script",
+  devtools_click: "click",
+  devtools_hover: "hover",
+  devtools_drag: "drag",
+  devtools_type: "type_text",
+  devtools_press_key: "press_key",
+  devtools_fill: "fill",
+  devtools_fill_form: "fill_form",
+  devtools_navigate: "navigate_page",
+  devtools_resize_page: "resize_page",
+  devtools_emulate: "emulate",
+  devtools_network: "list_network_requests",
+  devtools_network_request: "get_network_request",
+  devtools_console: "list_console_messages",
+  devtools_console_message: "get_console_message",
+  devtools_wait_for: "wait_for",
+  devtools_handle_dialog: "handle_dialog",
+  devtools_upload_file: "upload_file",
+  devtools_performance_start: "performance_start_trace",
+  devtools_performance_stop: "performance_stop_trace",
+  devtools_performance_insight: "performance_analyze_insight",
+  devtools_lighthouse: "lighthouse_audit",
+  devtools_heapsnapshot: "take_heapsnapshot"
+};
+export {
+  CUSTOM_TOOLS,
+  TOOL_MAP
+};

package/es/endpoints/mcp.d.ts

@@ -0,0 +1,6 @@
+import type { ViteDevServer } from "vite";
+import type { PageContext } from "@vite-plugin-opencode-assistant/shared";
+import { MCP_API_PATH } from "@vite-plugin-opencode-assistant/shared";
+import { McpProxy } from "../core/mcp-proxy";
+export { MCP_API_PATH };
+export declare function setupMcpEndpoint(server: ViteDevServer, mcp: McpProxy, getPageContext: () => PageContext): void;

package/es/endpoints/mcp.mjs

@@ -0,0 +1,259 @@
+var __async = (__this, __arguments, generator) => {
+  return new Promise((resolve, reject) => {
+    var fulfilled = (value) => {
+      try {
+        step(generator.next(value));
+      } catch (e) {
+        reject(e);
+      }
+    };
+    var rejected = (value) => {
+      try {
+        step(generator.throw(value));
+      } catch (e) {
+        reject(e);
+      }
+    };
+    var step = (x) => x.done ? resolve(x.value) : Promise.resolve(x.value).then(fulfilled, rejected);
+    step((generator = generator.apply(__this, __arguments)).next());
+  });
+};
+import { MCP_API_PATH } from "@vite-plugin-opencode-assistant/shared";
+import { createLogger } from "@vite-plugin-opencode-assistant/shared/node";
+import { parseListPages, resolveChromePageId, getProjectOrigin } from "../endpoints/context.mjs";
+import { CUSTOM_TOOLS, TOOL_MAP } from "./mcp-tools.mjs";
+const log = createLogger("McpEndpoint");
+function setupMcpEndpoint(server, mcp, getPageContext) {
+  server.middlewares.use((req, res, next) => __async(null, null, function* () {
+    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j;
+    if (!((_a = req.url) == null ? void 0 : _a.startsWith(MCP_API_PATH))) return next();
+    const url = new URL(req.url, `http://localhost`);
+    const token = (_c = url.searchParams.get("token")) != null ? _c : (_b = req.headers["authorization"]) == null ? void 0 : _b.replace(/^Bearer /i, "");
+    if (token !== mcp.accessToken) {
+      res.writeHead(401, { "Content-Type": "application/json" });
+      res.end(JSON.stringify({ error: "Unauthorized" }));
+      return;
+    }
+    res.setHeader("Access-Control-Allow-Origin", "*");
+    res.setHeader("Access-Control-Allow-Methods", "GET, POST, DELETE, OPTIONS");
+    res.setHeader("Access-Control-Allow-Headers", "Content-Type, Mcp-Session-Id, Authorization");
+    res.setHeader("Access-Control-Expose-Headers", "Mcp-Session-Id");
+    if (req.method === "OPTIONS") {
+      res.writeHead(204);
+      res.end();
+      return;
+    }
+    if (req.method === "GET") {
+      res.writeHead(200, {
+        "Content-Type": "text/event-stream",
+        "Cache-Control": "no-cache",
+        Connection: "keep-alive",
+        "Mcp-Session-Id": mcp.sessionId
+      });
+      res.write(":ok\n\n");
+      const keepAlive = setInterval(() => res.write(":ping\n\n"), 15e3);
+      req.on("close", () => clearInterval(keepAlive));
+      return;
+    }
+    if (req.method === "DELETE") {
+      res.writeHead(200);
+      res.end();
+      return;
+    }
+    if (req.method === "POST") {
+      try {
+        const body = yield readBody(req);
+        if (!body) {
+          res.writeHead(400);
+          res.end("Empty body");
+          return;
+        }
+        const { method, id } = tryParseRequest(body);
+        log.debug("MCP request", { method, body: body.substring(0, 150) });
+        if (method === "tools/list") {
+          res.writeHead(200, {
+            "Content-Type": "application/json",
+            "Mcp-Session-Id": mcp.sessionId
+          });
+          res.end(JSON.stringify({ jsonrpc: "2.0", id, result: { tools: CUSTOM_TOOLS } }));
+          return;
+        }
+        if (method === "tools/call") {
+          const params = tryParseParams(body);
+          const toolName = params == null ? void 0 : params.name;
+          const mapped = toolName != null ? TOOL_MAP[toolName] : void 0;
+          if (!mapped) {
+            res.writeHead(200, {
+              "Content-Type": "application/json",
+              "Mcp-Session-Id": mcp.sessionId
+            });
+            res.end(
+              JSON.stringify({
+                jsonrpc: "2.0",
+                id,
+                error: { code: -32601, message: `Tool not found: ${toolName}` }
+              })
+            );
+            return;
+          }
+          if (toolName === "devtools_list_pages") {
+            const listResult = yield mcp.callChromeDevTool("list_pages", {});
+            const text = (_f = (_e = (_d = listResult == null ? void 0 : listResult.result) == null ? void 0 : _d.content) == null ? void 0 : _e[0]) == null ? void 0 : _f.text;
+            const allPages = text ? parseListPages(text) : [];
+            const pc = getPageContext();
+            const projectOrigin = getProjectOrigin(pc.url);
+            if (!projectOrigin) {
+              res.writeHead(200, {
+                "Content-Type": "application/json",
+                "Mcp-Session-Id": mcp.sessionId
+              });
+              res.end(
+                JSON.stringify({
+                  jsonrpc: "2.0",
+                  id,
+                  result: {
+                    content: [
+                      { type: "text", text: "\u6682\u65E0\u9879\u76EE\u9875\u9762\uFF0C\u8BF7\u5148\u5728\u6D4F\u89C8\u5668\u4E2D\u6253\u5F00\u672C\u5730\u5F00\u53D1\u9875\u9762" }
+                    ]
+                  }
+                })
+              );
+              return;
+            }
+            const filtered = allPages.filter((p) => p.url.startsWith(projectOrigin));
+            const chromeSelectedPageId = (_g = allPages.find((p) => p.selected)) == null ? void 0 : _g.pageId;
+            const resolved = yield resolveChromePageId(
+              mcp,
+              pc.url,
+              pc.title,
+              pc.sessionId,
+              filtered,
+              chromeSelectedPageId
+            );
+            const activePageId = resolved.ok ? resolved.pageId : null;
+            res.writeHead(200, {
+              "Content-Type": "application/json",
+              "Mcp-Session-Id": mcp.sessionId
+            });
+            res.end(
+              JSON.stringify({
+                jsonrpc: "2.0",
+                id,
+                result: {
+                  content: [
+                    {
+                      type: "text",
+                      text: JSON.stringify(
+                        filtered.map((p) => ({
+                          pageId: p.pageId,
+                          url: p.url,
+                          title: p.title,
+                          active: activePageId != null ? p.pageId === activePageId : false,
+                          selected: p.selected
+                        })),
+                        null,
+                        2
+                      )
+                    }
+                  ]
+                }
+              })
+            );
+            return;
+          }
+          if (toolName !== "devtools_select_page") {
+            const checkResult = yield mcp.callChromeDevTool("list_pages", {});
+            const checkText = (_j = (_i = (_h = checkResult == null ? void 0 : checkResult.result) == null ? void 0 : _h.content) == null ? void 0 : _i[0]) == null ? void 0 : _j.text;
+            const checkPages = checkText ? parseListPages(checkText) : [];
+            const pc = getPageContext();
+            const origin = getProjectOrigin(pc.url);
+            const hasProjectSelected = origin ? checkPages.some((p) => p.selected && p.url.startsWith(origin)) : true;
+            if (!hasProjectSelected) {
+              res.writeHead(200, {
+                "Content-Type": "application/json",
+                "Mcp-Session-Id": mcp.sessionId
+              });
+              res.end(
+                JSON.stringify({
+                  jsonrpc: "2.0",
+                  id,
+                  error: {
+                    code: -32e3,
+                    message: "Chrome DevTools \u5F53\u524D\u672A\u9009\u4E2D\u9879\u76EE\u9875\u9762\uFF0C\u8BF7\u5148\u8C03\u7528 devtools_list_pages \u67E5\u770B\u53EF\u7528\u9875\u9762\uFF0C\u518D\u8C03\u7528 devtools_select_page \u9009\u62E9\u76EE\u6807\u9875\u9762"
+                  }
+                })
+              );
+              return;
+            }
+          }
+          const forwardBody = JSON.stringify({
+            jsonrpc: "2.0",
+            id,
+            method: "tools/call",
+            params: {
+              name: mapped,
+              arguments: (params == null ? void 0 : params.arguments) || {}
+            }
+          });
+          const responseText2 = yield mcp.forward(forwardBody);
+          log.debug("MCP response", { method: toolName, response: responseText2.substring(0, 100) });
+          res.writeHead(200, {
+            "Content-Type": "application/json",
+            "Mcp-Session-Id": mcp.sessionId
+          });
+          res.end(responseText2);
+          return;
+        }
+        const responseText = yield mcp.forward(body);
+        res.writeHead(200, { "Content-Type": "application/json", "Mcp-Session-Id": mcp.sessionId });
+        res.end(responseText);
+      } catch (e) {
+        log.debug("MCP POST error", { error: e.message });
+        res.writeHead(500, { "Content-Type": "application/json" });
+        res.end(
+          JSON.stringify({
+            jsonrpc: "2.0",
+            error: { code: -32603, message: e.message }
+          })
+        );
+      }
+      return;
+    }
+    next();
+  }));
+}
+function readBody(req, maxSize = 1024 * 1024) {
+  return new Promise((resolve, reject) => {
+    let body = "";
+    req.on("data", (chunk) => {
+      body += chunk.toString();
+      if (body.length > maxSize) {
+        req.destroy();
+        reject(new Error("Request body too large"));
+      }
+    });
+    req.on("end", () => resolve(body));
+    req.on("error", reject);
+  });
+}
+function tryParseRequest(body) {
+  var _a;
+  try {
+    const data = JSON.parse(body);
+    return { method: data.method || "unknown", id: (_a = data.id) != null ? _a : null };
+  } catch (e) {
+    return { method: "unknown", id: null };
+  }
+}
+function tryParseParams(body) {
+  var _a;
+  try {
+    return (_a = JSON.parse(body).params) != null ? _a : null;
+  } catch (e) {
+    return null;
+  }
+}
+export {
+  MCP_API_PATH,
+  setupMcpEndpoint
+};