vite-plugin-opencode-assistant 1.1.24 → 1.1.26

package/es/endpoints/mcp.mjs

@@ -0,0 +1,290 @@
+var __defProp = Object.defineProperty;
+var __getOwnPropSymbols = Object.getOwnPropertySymbols;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __propIsEnum = Object.prototype.propertyIsEnumerable;
+var __defNormalProp = (obj, key, value) => key in obj ? __defProp(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
+var __spreadValues = (a, b) => {
+  for (var prop in b || (b = {}))
+    if (__hasOwnProp.call(b, prop))
+      __defNormalProp(a, prop, b[prop]);
+  if (__getOwnPropSymbols)
+    for (var prop of __getOwnPropSymbols(b)) {
+      if (__propIsEnum.call(b, prop))
+        __defNormalProp(a, prop, b[prop]);
+    }
+  return a;
+};
+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,
+  getProjectOrigins,
+  isProjectPage
+} from "../core/mcp-chrome.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;
+    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) {
+      sendMcpJson(res, 401, { 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") {
+      handleGetSse(req, res, mcp);
+      return;
+    }
+    if (req.method === "DELETE") {
+      res.writeHead(200);
+      res.end();
+      return;
+    }
+    if (req.method === "POST") {
+      yield handlePost(req, res, mcp, getProjectOrigins(server), getPageContext);
+      return;
+    }
+    next();
+  }));
+}
+function handleGetSse(req, res, mcp) {
+  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));
+}
+function handlePost(req, res, mcp, projectOrigins, getPageContext) {
+  return __async(this, null, function* () {
+    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) });
+      switch (method) {
+        case "tools/list":
+          return handleToolsList(res, id, mcp.sessionId);
+        case "tools/call":
+          return handleToolsCall(res, id, body, mcp, projectOrigins, getPageContext);
+        default:
+          return handleForward(res, body, mcp);
+      }
+    } catch (e) {
+      log.debug("MCP POST error", { error: e.message });
+      sendMcpJson(res, 500, {
+        jsonrpc: "2.0",
+        error: { code: -32603, message: e.message }
+      });
+    }
+  });
+}
+function handleToolsList(res, id, sessionId) {
+  sendMcpJson(res, 200, { jsonrpc: "2.0", id, result: { tools: CUSTOM_TOOLS } }, sessionId);
+}
+function handleToolsCall(res, id, body, mcp, projectOrigins, getPageContext) {
+  const params = tryParseParams(body);
+  const toolName = params == null ? void 0 : params.name;
+  const mapped = toolName != null ? TOOL_MAP[toolName] : void 0;
+  if (!mapped) {
+    sendMcpError(res, id, -32601, `Tool not found: ${toolName}`, mcp.sessionId);
+    return;
+  }
+  switch (toolName) {
+    case "devtools_list_pages":
+      return handleListPages(res, id, mcp, projectOrigins, getPageContext);
+    default:
+      return handleDevTool(res, id, mcp, mapped, (params == null ? void 0 : params.arguments) || {}, projectOrigins);
+  }
+}
+function handleListPages(res, id, mcp, projectOrigins, getPageContext) {
+  return __async(this, null, function* () {
+    var _a, _b, _c, _d;
+    const listResult = yield mcp.callChromeDevTool("list_pages", {});
+    const text = (_c = (_b = (_a = listResult == null ? void 0 : listResult.result) == null ? void 0 : _a.content) == null ? void 0 : _b[0]) == null ? void 0 : _c.text;
+    const allPages = text ? parseListPages(text) : [];
+    log.debug("Chrome pages", { total: allPages.length, urls: allPages.map((p) => p.url) });
+    log.debug("project origins", { origins: projectOrigins });
+    const filtered = allPages.filter((p) => isProjectPage(p.url, projectOrigins));
+    log.debug("filtered pages", { count: filtered.length, pageIds: filtered.map((p) => p.pageId) });
+    if (filtered.length === 0) {
+      sendMcpResult(res, id, "\u6682\u65E0\u9879\u76EE\u9875\u9762\uFF0C\u8BF7\u5148\u5728\u6D4F\u89C8\u5668\u4E2D\u6253\u5F00\u672C\u5730\u5F00\u53D1\u9875\u9762", mcp.sessionId);
+      return;
+    }
+    const pc = getPageContext();
+    const chromeSelectedPageId = (_d = allPages.find((p) => p.selected)) == null ? void 0 : _d.pageId;
+    const resolved = yield resolveChromePageId(
+      mcp,
+      pc.url,
+      pc.title,
+      projectOrigins,
+      pc.sessionId,
+      filtered,
+      chromeSelectedPageId
+    );
+    const activePageId = resolved.ok ? resolved.pageId : null;
+    const pageList = filtered.map((p) => ({
+      pageId: p.pageId,
+      url: p.url,
+      title: p.title,
+      active: activePageId != null ? p.pageId === activePageId : false,
+      selected: p.selected
+    }));
+    sendMcpResult(res, id, JSON.stringify(pageList, null, 2), mcp.sessionId);
+  });
+}
+function handleDevTool(res, id, mcp, mapped, args, projectOrigins) {
+  return __async(this, null, function* () {
+    var _a, _b, _c;
+    const pageId = args["pageId"];
+    if (typeof pageId !== "number") {
+      sendMcpError(
+        res,
+        id,
+        -32e3,
+        "\u7F3A\u5C11 pageId \u53C2\u6570\uFF0C\u8BF7\u5148\u8C03\u7528 devtools_list_pages \u83B7\u53D6\u53EF\u7528\u9875\u9762\u5217\u8868",
+        mcp.sessionId
+      );
+      return;
+    }
+    const checkResult = yield mcp.callChromeDevTool("list_pages", {});
+    const checkText = (_c = (_b = (_a = checkResult == null ? void 0 : checkResult.result) == null ? void 0 : _a.content) == null ? void 0 : _b[0]) == null ? void 0 : _c.text;
+    const checkPages = checkText ? parseListPages(checkText) : [];
+    const projectPages = checkPages.filter((p) => isProjectPage(p.url, projectOrigins));
+    const isValid = projectPages.some((p) => p.pageId === pageId);
+    log.debug("handleDevTool validation", {
+      pageId,
+      totalChromePages: checkPages.length,
+      projectPages: projectPages.length,
+      projectPageIds: projectPages.map((p) => p.pageId),
+      origins: projectOrigins,
+      isValid
+    });
+    if (!isValid) {
+      sendMcpError(res, id, -32e3, "pageId \u65E0\u6548\u6216\u975E\u9879\u76EE\u9875\u9762", mcp.sessionId);
+      return;
+    }
+    yield mcp.callChromeDevTool("select_page", { pageId, bringToFront: false });
+    const forwardArgs = __spreadValues({}, args);
+    delete forwardArgs["pageId"];
+    const forwardBody = JSON.stringify({
+      jsonrpc: "2.0",
+      id,
+      method: "tools/call",
+      params: { name: mapped, arguments: forwardArgs }
+    });
+    const responseText = yield mcp.forward(forwardBody);
+    log.debug("MCP response", { mapped, response: responseText.substring(0, 100) });
+    sendMcpJson(res, 200, responseText, mcp.sessionId);
+  });
+}
+function handleForward(res, body, mcp) {
+  return __async(this, null, function* () {
+    const responseText = yield mcp.forward(body);
+    sendMcpJson(res, 200, responseText, mcp.sessionId);
+  });
+}
+function sendMcpResult(res, id, text, sessionId) {
+  sendMcpJson(
+    res,
+    200,
+    {
+      jsonrpc: "2.0",
+      id,
+      result: { content: [{ type: "text", text }] }
+    },
+    sessionId
+  );
+}
+function sendMcpError(res, id, code, message, sessionId) {
+  sendMcpJson(
+    res,
+    200,
+    {
+      jsonrpc: "2.0",
+      id,
+      error: { code, message }
+    },
+    sessionId
+  );
+}
+function sendMcpJson(res, statusCode, body, sessionId) {
+  const headers = { "Content-Type": "application/json" };
+  if (sessionId) headers["Mcp-Session-Id"] = sessionId;
+  res.writeHead(statusCode, headers);
+  res.end(typeof body === "string" ? body : JSON.stringify(body));
+}
+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
+};

package/es/index.mjs

@@ -47,7 +47,9 @@ import { setupMiddlewares, LOGS_API_PATH } from "./endpoints/index.mjs";
 import { injectWidget } from "./core/injector.mjs";
 import { OpenCodeAPI } from "./core/api.mjs";
 import { OpenCodeService } from "./core/service.mjs";
+import { McpProxy } from "./core/mcp-proxy.mjs";
 import { resolveWidgetPath, resolveWidgetStylePath } from "./utils/paths.mjs";
+import { findGitRoot } from "./utils/system.mjs";
 function opencodePlugin(options = {}) {
   const plugins = [];
   plugins.push(
@@ -75,6 +77,7 @@ function createOpenCodePlugin(options = {}) {
   let activeTabId = DEFAULT_TAB;
   const serviceInstanceId = crypto.randomUUID();
   const sseClients = /* @__PURE__ */ new Set();
+  const mcpProxy = new McpProxy();
   const api = new OpenCodeAPI(
     config.hostname,
     () => actualWebPort,
@@ -93,6 +96,7 @@ function createOpenCodePlugin(options = {}) {
       actualProxyPort = port;
     }
   );
+  service.workspaceRoot = findGitRoot(process.cwd());
   return {
     name: "vite-plugin-opencode",
     apply(_viteConfig, env) {
@@ -106,56 +110,60 @@ function createOpenCodePlugin(options = {}) {
         projectRoot = server.config.root;
         let viteOrigin = "";
         const getViteOrigin = () => viteOrigin;
-        setupMiddlewares(server, {
-          get webUrl() {
-            return actualWebPort ? `http://${config.hostname}:${actualWebPort}` : null;
+        setupMiddlewares(
+          server,
+          {
+            get webUrl() {
+              return actualWebPort ? `http://${config.hostname}:${actualWebPort}` : null;
+            },
+            get sseClients() {
+              return sseClients;
+            },
+            getPageContext() {
+              return pageContexts.get(activeTabId) || pageContexts.get(DEFAULT_TAB) || { url: "", title: "" };
+            },
+            setPageContext(tabId, ctx) {
+              pageContexts.set(tabId || DEFAULT_TAB, ctx);
+            },
+            setActiveTabId(tabId) {
+              activeTabId = tabId;
+            },
+            clearSelectedElements() {
+              const ctx = pageContexts.get(activeTabId);
+              if (ctx) {
+                ctx.selectedElements = [];
+                pageContexts.set(activeTabId, ctx);
+              }
+              const defaultCtx = pageContexts.get(DEFAULT_TAB);
+              if (defaultCtx) {
+                defaultCtx.selectedElements = [];
+              }
+            },
+            get isServiceStarted() {
+              return service.isStarted;
+            },
+            get currentTask() {
+              return service.currentTask;
+            },
+            get actualProxyPort() {
+              return actualProxyPort;
+            },
+            get actualWebPort() {
+              return actualWebPort;
+            },
+            get serviceInstanceId() {
+              return serviceInstanceId;
+            },
+            getSessions: () => api.getSessions(service.workspaceRoot),
+            createSession: () => api.createSession(service.workspaceRoot),
+            deleteSession: (id) => api.deleteSession(id),
+            resolveWidgetPath,
+            resolveWidgetStylePath,
+            getAvailableModels: () => service.getAvailableModels(),
+            retryWarmupChromeMcp: (selectedModel) => service.retryWarmupChromeMcp(getViteOrigin(), selectedModel)
           },
-          get sseClients() {
-            return sseClients;
-          },
-          getPageContext() {
-            return pageContexts.get(activeTabId) || pageContexts.get(DEFAULT_TAB) || { url: "", title: "" };
-          },
-          setPageContext(tabId, ctx) {
-            pageContexts.set(tabId || DEFAULT_TAB, ctx);
-          },
-          setActiveTabId(tabId) {
-            activeTabId = tabId;
-          },
-          clearSelectedElements() {
-            const ctx = pageContexts.get(activeTabId);
-            if (ctx) {
-              ctx.selectedElements = [];
-              pageContexts.set(activeTabId, ctx);
-            }
-            const defaultCtx = pageContexts.get(DEFAULT_TAB);
-            if (defaultCtx) {
-              defaultCtx.selectedElements = [];
-            }
-          },
-          get isServiceStarted() {
-            return service.isStarted;
-          },
-          get currentTask() {
-            return service.currentTask;
-          },
-          get actualProxyPort() {
-            return actualProxyPort;
-          },
-          get actualWebPort() {
-            return actualWebPort;
-          },
-          get serviceInstanceId() {
-            return serviceInstanceId;
-          },
-          getSessions: () => api.getSessions(service.workspaceRoot),
-          createSession: () => api.createSession(service.workspaceRoot),
-          deleteSession: (id) => api.deleteSession(id),
-          resolveWidgetPath,
-          resolveWidgetStylePath,
-          getAvailableModels: () => service.getAvailableModels(),
-          retryWarmupChromeMcp: (selectedModel) => service.retryWarmupChromeMcp(getViteOrigin(), selectedModel)
-        });
+          mcpProxy
+        );
         (_a2 = server.httpServer) == null ? void 0 : _a2.on("listening", () => __async(null, null, function* () {
           var _a3;
           log.debug("Vite server listening event fired");
@@ -186,7 +194,16 @@ function createOpenCodePlugin(options = {}) {
             logsApiUrl
           });
           try {
-            yield service.start([viteOrigin], contextApiUrl, logsApiUrl, viteOrigin);
+            yield mcpProxy.start();
+            yield service.start(
+              mcpProxy.accessToken,
+              vitePort,
+              [viteOrigin],
+              contextApiUrl,
+              logsApiUrl,
+              viteOrigin,
+              mcpProxy
+            );
           } catch (e) {
             log.error("Failed to start services", { error: e });
           }
@@ -194,9 +211,11 @@ function createOpenCodePlugin(options = {}) {
         (_b2 = server.httpServer) == null ? void 0 : _b2.on("close", () => {
           log.debug("HTTP server closing");
           service.stop();
+          mcpProxy.stop();
         });
         const cleanup = () => __async(null, null, function* () {
           log.debug("Process cleanup triggered");
+          mcpProxy.stop();
           yield service.stop();
           process.exit(0);
         });
@@ -223,23 +242,9 @@ function createOpenCodePlugin(options = {}) {
       const titleInject = `<script>
         (function () {
           var KEY = "_opencode_pk";
-          var prefix = sessionStorage.getItem(KEY);
-          if (!prefix) {
-            prefix = "[" + Math.random().toString(36).slice(2, 5) + "]";
-            sessionStorage.setItem(KEY, prefix);
-          }
-          var applied = false;
-          function apply() {
-            if (applied) return;
-            var title = document.title;
-            if (title.indexOf(prefix) === 0) return;
-            applied = true;
-            document.title = prefix + title.replace(prefix, "");
-            applied = false;
+          if (!sessionStorage.getItem(KEY)) {
+            sessionStorage.setItem(KEY, "[" + Math.random().toString(36).slice(2, 5) + "]");
           }
-          apply();
-          var target = document.querySelector("title") || document.head;
-          new MutationObserver(apply).observe(target, { childList: true });
         })();
       </script>`;
       timer.end();