vite-plugin-opencode-assistant 1.1.24 → 1.1.25

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();