@xiaou66/vite-plugin-vue-mcp-next 0.0.2 → 0.0.3

package/dist/index.js

@@ -809,18 +809,39 @@ function createMcpServer(ctx, vite) {
 
 // src/mcp/transport.ts
 import { SSEServerTransport } from "@modelcontextprotocol/sdk/server/sse.js";
-function setupMcpTransport(base, server, vite) {
+import { StreamableHTTPServerTransport } from "@modelcontextprotocol/sdk/server/streamableHttp.js";
+function setupMcpTransport(base, createServer, vite) {
   const transports = /* @__PURE__ */ new Map();
   vite.middlewares.use(`${base}/sse`, (_req, res) => {
     const transport = new SSEServerTransport(`${base}/messages`, res);
-    transports.set(transport.sessionId, transport);
+    const server = createServer();
+    transports.set(transport.sessionId, { server, transport });
     res.on("close", () => {
       transports.delete(transport.sessionId);
+      void server.close();
     });
     void server.connect(transport).catch((error) => {
       res.destroy(error instanceof Error ? error : new Error(String(error)));
     });
   });
+  vite.middlewares.use(`${base}/mcp`, (req, res) => {
+    if (req.method !== "POST") {
+      res.statusCode = 405;
+      res.end("Method Not Allowed");
+      return;
+    }
+    const transport = new StreamableHTTPServerTransport({
+      sessionIdGenerator: void 0
+    });
+    const server = createServer();
+    res.on("close", () => {
+      void transport.close();
+      void server.close();
+    });
+    void server.connect(transport).then(() => transport.handleRequest(req, res)).catch((error) => {
+      res.destroy(error instanceof Error ? error : new Error(String(error)));
+    });
+  });
   vite.middlewares.use(`${base}/messages`, (req, res) => {
     if (req.method !== "POST") {
       res.statusCode = 405;
@@ -834,13 +855,13 @@ function setupMcpTransport(base, server, vite) {
       res.end("Bad Request");
       return;
     }
-    const transport = transports.get(sessionId);
-    if (!transport) {
+    const entry = transports.get(sessionId);
+    if (!entry) {
       res.statusCode = 404;
       res.end("Not Found");
       return;
     }
-    void transport.handlePostMessage(req, res).catch((error) => {
+    void entry.transport.handlePostMessage(req, res).catch((error) => {
       res.destroy(error instanceof Error ? error : new Error(String(error)));
     });
   });
@@ -1216,7 +1237,7 @@ function replaceOrAppendOwnedBlock(current, options) {
   const block = createCodexServerBlock(options);
   const matcher = createOwnedBlockMatcher(options.serverName);
   if (matcher.test(current)) {
-    return current.replace(matcher, block);
+    return ensureTrailingNewline(current);
   }
   const separator = current.trim() ? "\n\n" : "";
   return `${trimEndNewline(current)}${separator}${block}`;
@@ -1258,6 +1279,10 @@ async function readOptionalTextFile(filePath) {
 function trimEndNewline(value) {
   return value.replace(/\n+$/u, "");
 }
+function ensureTrailingNewline(value) {
+  return value.endsWith("\n") ? value : `${value}
+`;
+}
 function escapeRegExp(value) {
   return value.replace(/[.*+?^${}()|[\]\\]/g, "\\$&");
 }
@@ -1279,7 +1304,10 @@ async function updateJsonMcpClientConfig(options) {
       return;
     }
     const mcpServers = isPlainRecord(config.mcpServers) ? config.mcpServers : {};
-    mcpServers[options.serverName] = { url: options.mcpUrl };
+    if (Object.hasOwn(mcpServers, options.serverName)) {
+      return;
+    }
+    mcpServers[options.serverName] = { type: "sse", url: options.mcpUrl };
     config.mcpServers = mcpServers;
     await fs3.mkdir(path3.dirname(options.configPath), { recursive: true });
     await fs3.writeFile(
@@ -1324,7 +1352,7 @@ function isNodeError2(error) {
 }
 
 // src/plugin/mcpClientConfig/index.ts
-async function updateMcpClientConfigs(root, mcpUrl, options) {
+async function updateMcpClientConfigs(root, sseUrl, streamableHttpUrl, options) {
   const serverName = options.serverName;
   const jobs = [];
   if (options.cursor) {
@@ -1332,7 +1360,7 @@ async function updateMcpClientConfigs(root, mcpUrl, options) {
       updateJsonMcpClientConfig({
         clientName: "Cursor",
         configPath: path4.join(root, ".cursor", "mcp.json"),
-        mcpUrl,
+        mcpUrl: sseUrl,
         serverName
       })
     );
@@ -1341,7 +1369,7 @@ async function updateMcpClientConfigs(root, mcpUrl, options) {
     jobs.push(
       updateCodexMcpClientConfig({
         configPath: path4.join(root, ".codex", "config.toml"),
-        mcpUrl,
+        mcpUrl: streamableHttpUrl,
         serverName
       })
     );
@@ -1351,7 +1379,7 @@ async function updateMcpClientConfigs(root, mcpUrl, options) {
       updateJsonMcpClientConfig({
         clientName: "Claude Code",
         configPath: path4.join(root, ".mcp.json"),
-        mcpUrl,
+        mcpUrl: sseUrl,
         serverName
       })
     );
@@ -1361,7 +1389,7 @@ async function updateMcpClientConfigs(root, mcpUrl, options) {
       updateJsonMcpClientConfig({
         clientName: "Trae",
         configPath: path4.join(root, ".trae", "mcp.json"),
-        mcpUrl,
+        mcpUrl: sseUrl,
         serverName
       })
     );
@@ -1397,8 +1425,11 @@ function vueMcpNext(userOptions = {}) {
           timeout: -1
         }
       );
-      const mcpServer = createMcpServer(ctx, server);
-      setupMcpTransport(options.mcpPath, mcpServer, server);
+      setupMcpTransport(
+        options.mcpPath,
+        () => createMcpServer(ctx, server),
+        server
+      );
       server.ws.on(
         "vite-plugin-vue-mcp-next:page-connected",
         (payload) => {
@@ -1425,12 +1456,21 @@ function vueMcpNext(userOptions = {}) {
         }
       );
       const port = String(server.config.server.port || 5173);
-      const mcpUrl = `http://${options.host}:${port}${options.mcpPath}/sse`;
+      const mcpSseUrl = `http://${options.host}:${port}${options.mcpPath}/sse`;
+      const mcpStreamableHttpUrl = `http://${options.host}:${port}${options.mcpPath}/mcp`;
       const root = searchForWorkspaceRoot(server.config.root);
-      await updateMcpClientConfigs(root, mcpUrl, options.mcpClients);
+      await updateMcpClientConfigs(
+        root,
+        mcpSseUrl,
+        mcpStreamableHttpUrl,
+        options.mcpClients
+      );
       if (options.printUrl) {
         setTimeout(() => {
-          console.log(`  \u279C  MCP:     Server is running at ${mcpUrl}`);
+          console.log(`  \u279C  MCP:     SSE server is running at ${mcpSseUrl}`);
+          console.log(
+            `  \u279C  MCP:     Streamable HTTP server is running at ${mcpStreamableHttpUrl}`
+          );
         }, 300);
       }
       server.httpServer?.once("close", () => {