vite-plugin-swagger-mcp 0.0.16 → 0.0.19

package/dist/cjs/index.js

@@ -25,8 +25,15 @@ __export(src_exports, {
 module.exports = __toCommonJS(src_exports);
 var import_mcp = require("@modelcontextprotocol/sdk/server/mcp.js");
 var import_streamableHttp = require("@modelcontextprotocol/sdk/server/streamableHttp.js");
+var import_types = require("@modelcontextprotocol/sdk/types");
 var import_node_crypto = require("node:crypto");
-var import_zod = require("zod");
+var import_v4 = require("zod/v4");
+async function readJsonBody(req) {
+  let body = "";
+  for await (const chunk of req)
+    body += chunk;
+  return JSON.parse(body || "{}");
+}
 var SwaggerMcpServer = class {
   constructor(swaggerUrl, token) {
     this.swaggerUrl = swaggerUrl;
@@ -144,130 +151,90 @@ function vitePluginSwaggerMcp({
     enforce: "pre",
     async configureServer(server) {
       var _a, _b;
-      try {
-        let transport = new import_streamableHttp.StreamableHTTPServerTransport({
-          sessionIdGenerator: () => (0, import_node_crypto.randomUUID)(),
-          onsessioninitialized: (sessionId) => {
-            transports[sessionId] = transport;
-          }
-        });
-        transport.onclose = () => {
-          if (transport.sessionId) {
-            delete transports[transport.sessionId];
-          }
-        };
-        const swaggerServer = new SwaggerMcpServer(swaggerUrl, token);
+      const swaggerServer = new SwaggerMcpServer(swaggerUrl, token);
+      const createNewServer = () => {
         const mcpServer = new import_mcp.McpServer({
           name: "swagger-mcp-server",
           version: "0.1.0"
         });
         mcpServer.tool("getModules", "获取模块列表", async () => {
           const res = await swaggerServer.getModules();
-          return {
-            content: [
-              {
-                type: "text",
-                text: JSON.stringify(res)
-              }
-            ]
-          };
+          return { content: [{ type: "text", text: JSON.stringify(res) }] };
         });
         mcpServer.tool(
           "getModuleApis",
-          "获取特定模块下的所有接口及返回值类型",
-          {
-            module: import_zod.z.string().describe("模块名称")
-          },
+          "获取特定模块下的所有接口",
+          { module: import_v4.z.string().describe("模块名称") },
           async ({ module: module2 }) => {
-            if (!module2) {
-              return {
-                content: [
-                  {
-                    type: "text",
-                    text: JSON.stringify({ error: "模块名称不能为空" })
-                  }
-                ]
-              };
-            }
             const res = await swaggerServer.getModuleApis(module2);
-            return {
-              content: [{ type: "text", text: JSON.stringify(res) }]
-            };
+            return { content: [{ type: "text", text: JSON.stringify(res) }] };
           }
         );
         mcpServer.tool(
           "getApiTypes",
           "获取特定接口的参数及返回值类型",
-          { path: import_zod.z.string(), method: import_zod.z.string() },
+          { path: import_v4.z.string(), method: import_v4.z.string() },
           async (args) => ({
-            content: [
-              {
-                type: "text",
-                text: JSON.stringify(
-                  await swaggerServer.getApiTypes(args.path, args.method)
-                )
-              }
-            ]
+            content: [{ type: "text", text: JSON.stringify(await swaggerServer.getApiTypes(args.path, args.method)) }]
           })
         );
-        await mcpServer.connect(transport);
-        console.log(
-          "MCP server connected:",
-          `http://localhost:${(_b = (_a = server.config) == null ? void 0 : _a.server) == null ? void 0 : _b.port}/_mcp/sse/swagger`
-        );
-        server.middlewares.use("/_mcp/sse/swagger", async (req, res) => {
-          var _a2;
+        return mcpServer;
+      };
+      const ENDPOINT = "/_mcp/sse/swagger";
+      server.middlewares.use(ENDPOINT, async (req, res, next) => {
+        try {
+          const url = new URL(req.url || "", `http://${req.headers.host}`);
+          const sessionId = req.headers["mcp-session-id"] || url.searchParams.get("sessionId");
           if (req.method === "POST") {
-            const sessionId = req.headers["mcp-session-id"];
-            let body = "";
-            for await (const chunk of req)
-              body += chunk;
-            const json = JSON.parse(body);
-            console.log("post request received");
-            console.log("body: ", json);
-            try {
-              if (sessionId && transports[sessionId]) {
-                transport = transports[sessionId];
-                await transport.handleRequest(req, res, json);
-                return;
-              }
-              if (!sessionId) {
-                await transport.handleRequest(req, res, json);
-                const sessionId2 = transport.sessionId;
-                if (sessionId2) {
-                  transports[sessionId2] = transport;
+            const json = await readJsonBody(req);
+            let transport;
+            if (sessionId && transports[sessionId]) {
+              transport = transports[sessionId];
+            } else if (!sessionId && (0, import_types.isInitializeRequest)(json)) {
+              transport = new import_streamableHttp.StreamableHTTPServerTransport({
+                sessionIdGenerator: () => (0, import_node_crypto.randomUUID)(),
+                // 如果你的编辑器不支持 SSE 只能接收 JSON 响应，可以设为 true
+                // 但大多数现代编辑器建议保持默认（false）以支持标准流
+                enableJsonResponse: false,
+                onsessioninitialized: (sid) => {
+                  transports[sid] = transport;
+                  console.log(`[MCP] Session initialized: ${sid}`);
                 }
-                return;
-              }
+              });
+              transport.onclose = () => {
+                if (transport.sessionId)
+                  delete transports[transport.sessionId];
+                console.log(`[MCP] Session closed: ${transport.sessionId}`);
+              };
+              const mcpServer = createNewServer();
+              await mcpServer.connect(transport);
+            } else {
               res.statusCode = 400;
-              res.end("Bad Request: invalid session ID or method.");
-              return;
-            } catch (error) {
-              console.error("Error handling MCP request:", error);
-              res.statusCode = 500;
-              res.end("Internal server error.");
-              return;
+              return res.end(JSON.stringify({ error: "Invalid session or not an initialize request" }));
             }
+            await transport.handleRequest(req, res, json);
+            return;
           }
           if (req.method === "GET") {
-            if (!((_a2 = req.headers.accept) == null ? void 0 : _a2.includes("text/event-stream"))) {
-              res.writeHead(405);
-              return res.end();
+            if (!sessionId || !transports[sessionId]) {
+              const initTransport = new import_streamableHttp.StreamableHTTPServerTransport();
+              await initTransport.handleRequest(req, res);
+              return;
             }
-            res.writeHead(200, {
-              "Content-Type": "text/event-stream",
-              "Cache-Control": "no-cache",
-              Connection: "keep-alive"
-            });
-            transport.onmessage = (message) => {
-              res.write(message);
-            };
+            await transports[sessionId].handleRequest(req, res);
             return;
           }
-        });
-      } catch (error) {
-        console.log("MCP server error", error);
-      }
+          res.statusCode = 405;
+          res.end("Method Not Allowed");
+        } catch (error) {
+          console.error("[MCP Error]", error);
+          if (!res.headersSent) {
+            res.statusCode = 500;
+            res.end("Internal Server Error");
+          }
+        }
+      });
+      console.log(`✅ MCP Swagger Server mounted at http://localhost:${(_b = (_a = server.config) == null ? void 0 : _a.server) == null ? void 0 : _b.port}${ENDPOINT}`);
     }
   };
 }

package/dist/esm/index.js

@@ -2,20 +2,88 @@ function _typeof(o) { "@babel/helpers - typeof"; return _typeof = "function" ==
 function ownKeys(e, r) { var t = Object.keys(e); if (Object.getOwnPropertySymbols) { var o = Object.getOwnPropertySymbols(e); r && (o = o.filter(function (r) { return Object.getOwnPropertyDescriptor(e, r).enumerable; })), t.push.apply(t, o); } return t; }
 function _objectSpread(e) { for (var r = 1; r < arguments.length; r++) { var t = null != arguments[r] ? arguments[r] : {}; r % 2 ? ownKeys(Object(t), !0).forEach(function (r) { _defineProperty(e, r, t[r]); }) : Object.getOwnPropertyDescriptors ? Object.defineProperties(e, Object.getOwnPropertyDescriptors(t)) : ownKeys(Object(t)).forEach(function (r) { Object.defineProperty(e, r, Object.getOwnPropertyDescriptor(t, r)); }); } return e; }
 function _regeneratorRuntime() { "use strict"; /*! regenerator-runtime -- Copyright (c) 2014-present, Facebook, Inc. -- license (MIT): https://github.com/facebook/regenerator/blob/main/LICENSE */ _regeneratorRuntime = function _regeneratorRuntime() { return e; }; var t, e = {}, r = Object.prototype, n = r.hasOwnProperty, o = Object.defineProperty || function (t, e, r) { t[e] = r.value; }, i = "function" == typeof Symbol ? Symbol : {}, a = i.iterator || "@@iterator", c = i.asyncIterator || "@@asyncIterator", u = i.toStringTag || "@@toStringTag"; function define(t, e, r) { return Object.defineProperty(t, e, { value: r, enumerable: !0, configurable: !0, writable: !0 }), t[e]; } try { define({}, ""); } catch (t) { define = function define(t, e, r) { return t[e] = r; }; } function wrap(t, e, r, n) { var i = e && e.prototype instanceof Generator ? e : Generator, a = Object.create(i.prototype), c = new Context(n || []); return o(a, "_invoke", { value: makeInvokeMethod(t, r, c) }), a; } function tryCatch(t, e, r) { try { return { type: "normal", arg: t.call(e, r) }; } catch (t) { return { type: "throw", arg: t }; } } e.wrap = wrap; var h = "suspendedStart", l = "suspendedYield", f = "executing", s = "completed", y = {}; function Generator() {} function GeneratorFunction() {} function GeneratorFunctionPrototype() {} var p = {}; define(p, a, function () { return this; }); var d = Object.getPrototypeOf, v = d && d(d(values([]))); v && v !== r && n.call(v, a) && (p = v); var g = GeneratorFunctionPrototype.prototype = Generator.prototype = Object.create(p); function defineIteratorMethods(t) { ["next", "throw", "return"].forEach(function (e) { define(t, e, function (t) { return this._invoke(e, t); }); }); } function AsyncIterator(t, e) { function invoke(r, o, i, a) { var c = tryCatch(t[r], t, o); if ("throw" !== c.type) { var u = c.arg, h = u.value; return h && "object" == _typeof(h) && n.call(h, "__await") ? e.resolve(h.__await).then(function (t) { invoke("next", t, i, a); }, function (t) { invoke("throw", t, i, a); }) : e.resolve(h).then(function (t) { u.value = t, i(u); }, function (t) { return invoke("throw", t, i, a); }); } a(c.arg); } var r; o(this, "_invoke", { value: function value(t, n) { function callInvokeWithMethodAndArg() { return new e(function (e, r) { invoke(t, n, e, r); }); } return r = r ? r.then(callInvokeWithMethodAndArg, callInvokeWithMethodAndArg) : callInvokeWithMethodAndArg(); } }); } function makeInvokeMethod(e, r, n) { var o = h; return function (i, a) { if (o === f) throw new Error("Generator is already running"); if (o === s) { if ("throw" === i) throw a; return { value: t, done: !0 }; } for (n.method = i, n.arg = a;;) { var c = n.delegate; if (c) { var u = maybeInvokeDelegate(c, n); if (u) { if (u === y) continue; return u; } } if ("next" === n.method) n.sent = n._sent = n.arg;else if ("throw" === n.method) { if (o === h) throw o = s, n.arg; n.dispatchException(n.arg); } else "return" === n.method && n.abrupt("return", n.arg); o = f; var p = tryCatch(e, r, n); if ("normal" === p.type) { if (o = n.done ? s : l, p.arg === y) continue; return { value: p.arg, done: n.done }; } "throw" === p.type && (o = s, n.method = "throw", n.arg = p.arg); } }; } function maybeInvokeDelegate(e, r) { var n = r.method, o = e.iterator[n]; if (o === t) return r.delegate = null, "throw" === n && e.iterator.return && (r.method = "return", r.arg = t, maybeInvokeDelegate(e, r), "throw" === r.method) || "return" !== n && (r.method = "throw", r.arg = new TypeError("The iterator does not provide a '" + n + "' method")), y; var i = tryCatch(o, e.iterator, r.arg); if ("throw" === i.type) return r.method = "throw", r.arg = i.arg, r.delegate = null, y; var a = i.arg; return a ? a.done ? (r[e.resultName] = a.value, r.next = e.nextLoc, "return" !== r.method && (r.method = "next", r.arg = t), r.delegate = null, y) : a : (r.method = "throw", r.arg = new TypeError("iterator result is not an object"), r.delegate = null, y); } function pushTryEntry(t) { var e = { tryLoc: t[0] }; 1 in t && (e.catchLoc = t[1]), 2 in t && (e.finallyLoc = t[2], e.afterLoc = t[3]), this.tryEntries.push(e); } function resetTryEntry(t) { var e = t.completion || {}; e.type = "normal", delete e.arg, t.completion = e; } function Context(t) { this.tryEntries = [{ tryLoc: "root" }], t.forEach(pushTryEntry, this), this.reset(!0); } function values(e) { if (e || "" === e) { var r = e[a]; if (r) return r.call(e); if ("function" == typeof e.next) return e; if (!isNaN(e.length)) { var o = -1, i = function next() { for (; ++o < e.length;) if (n.call(e, o)) return next.value = e[o], next.done = !1, next; return next.value = t, next.done = !0, next; }; return i.next = i; } } throw new TypeError(_typeof(e) + " is not iterable"); } return GeneratorFunction.prototype = GeneratorFunctionPrototype, o(g, "constructor", { value: GeneratorFunctionPrototype, configurable: !0 }), o(GeneratorFunctionPrototype, "constructor", { value: GeneratorFunction, configurable: !0 }), GeneratorFunction.displayName = define(GeneratorFunctionPrototype, u, "GeneratorFunction"), e.isGeneratorFunction = function (t) { var e = "function" == typeof t && t.constructor; return !!e && (e === GeneratorFunction || "GeneratorFunction" === (e.displayName || e.name)); }, e.mark = function (t) { return Object.setPrototypeOf ? Object.setPrototypeOf(t, GeneratorFunctionPrototype) : (t.__proto__ = GeneratorFunctionPrototype, define(t, u, "GeneratorFunction")), t.prototype = Object.create(g), t; }, e.awrap = function (t) { return { __await: t }; }, defineIteratorMethods(AsyncIterator.prototype), define(AsyncIterator.prototype, c, function () { return this; }), e.AsyncIterator = AsyncIterator, e.async = function (t, r, n, o, i) { void 0 === i && (i = Promise); var a = new AsyncIterator(wrap(t, r, n, o), i); return e.isGeneratorFunction(r) ? a : a.next().then(function (t) { return t.done ? t.value : a.next(); }); }, defineIteratorMethods(g), define(g, u, "Generator"), define(g, a, function () { return this; }), define(g, "toString", function () { return "[object Generator]"; }), e.keys = function (t) { var e = Object(t), r = []; for (var n in e) r.push(n); return r.reverse(), function next() { for (; r.length;) { var t = r.pop(); if (t in e) return next.value = t, next.done = !1, next; } return next.done = !0, next; }; }, e.values = values, Context.prototype = { constructor: Context, reset: function reset(e) { if (this.prev = 0, this.next = 0, this.sent = this._sent = t, this.done = !1, this.delegate = null, this.method = "next", this.arg = t, this.tryEntries.forEach(resetTryEntry), !e) for (var r in this) "t" === r.charAt(0) && n.call(this, r) && !isNaN(+r.slice(1)) && (this[r] = t); }, stop: function stop() { this.done = !0; var t = this.tryEntries[0].completion; if ("throw" === t.type) throw t.arg; return this.rval; }, dispatchException: function dispatchException(e) { if (this.done) throw e; var r = this; function handle(n, o) { return a.type = "throw", a.arg = e, r.next = n, o && (r.method = "next", r.arg = t), !!o; } for (var o = this.tryEntries.length - 1; o >= 0; --o) { var i = this.tryEntries[o], a = i.completion; if ("root" === i.tryLoc) return handle("end"); if (i.tryLoc <= this.prev) { var c = n.call(i, "catchLoc"), u = n.call(i, "finallyLoc"); if (c && u) { if (this.prev < i.catchLoc) return handle(i.catchLoc, !0); if (this.prev < i.finallyLoc) return handle(i.finallyLoc); } else if (c) { if (this.prev < i.catchLoc) return handle(i.catchLoc, !0); } else { if (!u) throw new Error("try statement without catch or finally"); if (this.prev < i.finallyLoc) return handle(i.finallyLoc); } } } }, abrupt: function abrupt(t, e) { for (var r = this.tryEntries.length - 1; r >= 0; --r) { var o = this.tryEntries[r]; if (o.tryLoc <= this.prev && n.call(o, "finallyLoc") && this.prev < o.finallyLoc) { var i = o; break; } } i && ("break" === t || "continue" === t) && i.tryLoc <= e && e <= i.finallyLoc && (i = null); var a = i ? i.completion : {}; return a.type = t, a.arg = e, i ? (this.method = "next", this.next = i.finallyLoc, y) : this.complete(a); }, complete: function complete(t, e) { if ("throw" === t.type) throw t.arg; return "break" === t.type || "continue" === t.type ? this.next = t.arg : "return" === t.type ? (this.rval = this.arg = t.arg, this.method = "return", this.next = "end") : "normal" === t.type && e && (this.next = e), y; }, finish: function finish(t) { for (var e = this.tryEntries.length - 1; e >= 0; --e) { var r = this.tryEntries[e]; if (r.finallyLoc === t) return this.complete(r.completion, r.afterLoc), resetTryEntry(r), y; } }, catch: function _catch(t) { for (var e = this.tryEntries.length - 1; e >= 0; --e) { var r = this.tryEntries[e]; if (r.tryLoc === t) { var n = r.completion; if ("throw" === n.type) { var o = n.arg; resetTryEntry(r); } return o; } } throw new Error("illegal catch attempt"); }, delegateYield: function delegateYield(e, r, n) { return this.delegate = { iterator: values(e), resultName: r, nextLoc: n }, "next" === this.method && (this.arg = t), y; } }, e; }
-function asyncGeneratorStep(gen, resolve, reject, _next, _throw, key, arg) { try { var info = gen[key](arg); var value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { Promise.resolve(value).then(_next, _throw); } }
-function _asyncToGenerator(fn) { return function () { var self = this, args = arguments; return new Promise(function (resolve, reject) { var gen = fn.apply(self, args); function _next(value) { asyncGeneratorStep(gen, resolve, reject, _next, _throw, "next", value); } function _throw(err) { asyncGeneratorStep(gen, resolve, reject, _next, _throw, "throw", err); } _next(undefined); }); }; }
 function _classCallCheck(instance, Constructor) { if (!(instance instanceof Constructor)) { throw new TypeError("Cannot call a class as a function"); } }
 function _defineProperties(target, props) { for (var i = 0; i < props.length; i++) { var descriptor = props[i]; descriptor.enumerable = descriptor.enumerable || false; descriptor.configurable = true; if ("value" in descriptor) descriptor.writable = true; Object.defineProperty(target, _toPropertyKey(descriptor.key), descriptor); } }
 function _createClass(Constructor, protoProps, staticProps) { if (protoProps) _defineProperties(Constructor.prototype, protoProps); if (staticProps) _defineProperties(Constructor, staticProps); Object.defineProperty(Constructor, "prototype", { writable: false }); return Constructor; }
 function _defineProperty(obj, key, value) { key = _toPropertyKey(key); if (key in obj) { Object.defineProperty(obj, key, { value: value, enumerable: true, configurable: true, writable: true }); } else { obj[key] = value; } return obj; }
 function _toPropertyKey(t) { var i = _toPrimitive(t, "string"); return "symbol" == _typeof(i) ? i : String(i); }
 function _toPrimitive(t, r) { if ("object" != _typeof(t) || !t) return t; var e = t[Symbol.toPrimitive]; if (void 0 !== e) { var i = e.call(t, r || "default"); if ("object" != _typeof(i)) return i; throw new TypeError("@@toPrimitive must return a primitive value."); } return ("string" === r ? String : Number)(t); }
+function asyncGeneratorStep(gen, resolve, reject, _next, _throw, key, arg) { try { var info = gen[key](arg); var value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { Promise.resolve(value).then(_next, _throw); } }
+function _asyncToGenerator(fn) { return function () { var self = this, args = arguments; return new Promise(function (resolve, reject) { var gen = fn.apply(self, args); function _next(value) { asyncGeneratorStep(gen, resolve, reject, _next, _throw, "next", value); } function _throw(err) { asyncGeneratorStep(gen, resolve, reject, _next, _throw, "throw", err); } _next(undefined); }); }; }
 function _asyncIterator(r) { var n, t, o, e = 2; for ("undefined" != typeof Symbol && (t = Symbol.asyncIterator, o = Symbol.iterator); e--;) { if (t && null != (n = r[t])) return n.call(r); if (o && null != (n = r[o])) return new AsyncFromSyncIterator(n.call(r)); t = "@@asyncIterator", o = "@@iterator"; } throw new TypeError("Object is not async iterable"); }
 function AsyncFromSyncIterator(r) { function AsyncFromSyncIteratorContinuation(r) { if (Object(r) !== r) return Promise.reject(new TypeError(r + " is not an object.")); var n = r.done; return Promise.resolve(r.value).then(function (r) { return { value: r, done: n }; }); } return AsyncFromSyncIterator = function AsyncFromSyncIterator(r) { this.s = r, this.n = r.next; }, AsyncFromSyncIterator.prototype = { s: null, n: null, next: function next() { return AsyncFromSyncIteratorContinuation(this.n.apply(this.s, arguments)); }, return: function _return(r) { var n = this.s.return; return void 0 === n ? Promise.resolve({ value: r, done: !0 }) : AsyncFromSyncIteratorContinuation(n.apply(this.s, arguments)); }, throw: function _throw(r) { var n = this.s.return; return void 0 === n ? Promise.reject(r) : AsyncFromSyncIteratorContinuation(n.apply(this.s, arguments)); } }, new AsyncFromSyncIterator(r); }
 import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
 import { StreamableHTTPServerTransport } from "@modelcontextprotocol/sdk/server/streamableHttp.js";
+import { isInitializeRequest } from "@modelcontextprotocol/sdk/types";
 import { randomUUID } from "node:crypto";
-import { z } from "zod";
+import { z } from "zod/v4";
+// --- 辅助函数：解析 JSON Body ---
+function readJsonBody(_x) {
+  return _readJsonBody.apply(this, arguments);
+}
+function _readJsonBody() {
+  _readJsonBody = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee10(req) {
+    var body, _iteratorAbruptCompletion, _didIteratorError, _iteratorError, _iterator, _step, chunk;
+    return _regeneratorRuntime().wrap(function _callee10$(_context10) {
+      while (1) switch (_context10.prev = _context10.next) {
+        case 0:
+          body = "";
+          _iteratorAbruptCompletion = false;
+          _didIteratorError = false;
+          _context10.prev = 3;
+          _iterator = _asyncIterator(req);
+        case 5:
+          _context10.next = 7;
+          return _iterator.next();
+        case 7:
+          if (!(_iteratorAbruptCompletion = !(_step = _context10.sent).done)) {
+            _context10.next = 13;
+            break;
+          }
+          chunk = _step.value;
+          body += chunk;
+        case 10:
+          _iteratorAbruptCompletion = false;
+          _context10.next = 5;
+          break;
+        case 13:
+          _context10.next = 19;
+          break;
+        case 15:
+          _context10.prev = 15;
+          _context10.t0 = _context10["catch"](3);
+          _didIteratorError = true;
+          _iteratorError = _context10.t0;
+        case 19:
+          _context10.prev = 19;
+          _context10.prev = 20;
+          if (!(_iteratorAbruptCompletion && _iterator.return != null)) {
+            _context10.next = 24;
+            break;
+          }
+          _context10.next = 24;
+          return _iterator.return();
+        case 24:
+          _context10.prev = 24;
+          if (!_didIteratorError) {
+            _context10.next = 27;
+            break;
+          }
+          throw _iteratorError;
+        case 27:
+          return _context10.finish(24);
+        case 28:
+          return _context10.finish(19);
+        case 29:
+          return _context10.abrupt("return", JSON.parse(body || "{}"));
+        case 30:
+        case "end":
+          return _context10.stop();
+      }
+    }, _callee10, null, [[3, 15, 19, 29], [20,, 24, 28]]);
+  }));
+  return _readJsonBody.apply(this, arguments);
+}
 export var SwaggerMcpServer = /*#__PURE__*/function () {
   function SwaggerMcpServer(swaggerUrl, token) {
     _classCallCheck(this, SwaggerMcpServer);
@@ -180,7 +248,7 @@ export var SwaggerMcpServer = /*#__PURE__*/function () {
           }
         }, _callee3, this);
       }));
-      function getModuleApis(_x) {
+      function getModuleApis(_x2) {
         return _getModuleApis.apply(this, arguments);
       }
       return getModuleApis;
@@ -219,7 +287,7 @@ export var SwaggerMcpServer = /*#__PURE__*/function () {
           }
         }, _callee4, this);
       }));
-      function getApiTypes(_x2, _x3) {
+      function getApiTypes(_x3, _x4) {
         return _getApiTypes.apply(this, arguments);
       }
       return getApiTypes;
@@ -228,7 +296,7 @@ export var SwaggerMcpServer = /*#__PURE__*/function () {
   return SwaggerMcpServer;
 }();
 
-// Map to store transports by session ID
+// 存储会话 ID 对应的 Transport 实例
 var transports = {};
 export default function vitePluginSwaggerMcp(_ref) {
   var swaggerUrl = _ref.swaggerUrl,
@@ -238,281 +306,218 @@ export default function vitePluginSwaggerMcp(_ref) {
     enforce: "pre",
     configureServer: function configureServer(server) {
       return _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee9() {
-        var _server$config, transport, swaggerServer, mcpServer;
+        var _server$config;
+        var swaggerServer, createNewServer, ENDPOINT;
         return _regeneratorRuntime().wrap(function _callee9$(_context9) {
           while (1) switch (_context9.prev = _context9.next) {
             case 0:
-              _context9.prev = 0;
-              transport = new StreamableHTTPServerTransport({
-                sessionIdGenerator: function sessionIdGenerator() {
-                  return randomUUID();
-                },
-                onsessioninitialized: function onsessioninitialized(sessionId) {
-                  // Store the transport by session ID
-                  transports[sessionId] = transport;
-                }
-              }); // Clean up transport when closed
-              transport.onclose = function () {
-                if (transport.sessionId) {
-                  delete transports[transport.sessionId];
-                }
-              };
-              swaggerServer = new SwaggerMcpServer(swaggerUrl, token); // 实例化 MCP Server
-              mcpServer = new McpServer({
-                name: "swagger-mcp-server",
-                version: "0.1.0"
-              }); // 注册工具
-              /***
-               * 获取模块列表
-               */
-              mcpServer.tool("getModules", "获取模块列表", /*#__PURE__*/_asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee5() {
-                var res;
-                return _regeneratorRuntime().wrap(function _callee5$(_context5) {
-                  while (1) switch (_context5.prev = _context5.next) {
-                    case 0:
-                      _context5.next = 2;
-                      return swaggerServer.getModules();
-                    case 2:
-                      res = _context5.sent;
-                      return _context5.abrupt("return", {
-                        content: [{
-                          type: "text",
-                          text: JSON.stringify(res)
-                        }]
-                      });
-                    case 4:
-                    case "end":
-                      return _context5.stop();
-                  }
-                }, _callee5);
-              })));
+              swaggerServer = new SwaggerMcpServer(swaggerUrl, token); // --- 辅助函数：创建并配置一个新的 MCP Server 实例 ---
+              createNewServer = function createNewServer() {
+                var mcpServer = new McpServer({
+                  name: "swagger-mcp-server",
+                  version: "0.1.0"
+                });
 
-              /***
-               * 获取特定模块下的所有接口及返回值类型
-               */
-              mcpServer.tool("getModuleApis", "获取特定模块下的所有接口及返回值类型", {
-                module: z.string().describe("模块名称")
-              }, /*#__PURE__*/function () {
-                var _ref4 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee6(_ref3) {
-                  var module, res;
-                  return _regeneratorRuntime().wrap(function _callee6$(_context6) {
-                    while (1) switch (_context6.prev = _context6.next) {
+                // 注册所有工具
+                mcpServer.tool("getModules", "获取模块列表", /*#__PURE__*/_asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee5() {
+                  var res;
+                  return _regeneratorRuntime().wrap(function _callee5$(_context5) {
+                    while (1) switch (_context5.prev = _context5.next) {
                       case 0:
-                        module = _ref3.module;
-                        if (module) {
-                          _context6.next = 3;
-                          break;
-                        }
-                        return _context6.abrupt("return", {
-                          content: [{
-                            type: "text",
-                            text: JSON.stringify({
-                              error: "模块名称不能为空"
-                            })
-                          }]
-                        });
-                      case 3:
-                        _context6.next = 5;
-                        return swaggerServer.getModuleApis(module);
-                      case 5:
-                        res = _context6.sent;
-                        return _context6.abrupt("return", {
+                        _context5.next = 2;
+                        return swaggerServer.getModules();
+                      case 2:
+                        res = _context5.sent;
+                        return _context5.abrupt("return", {
                           content: [{
                             type: "text",
                             text: JSON.stringify(res)
                           }]
                         });
-                      case 7:
+                      case 4:
                       case "end":
-                        return _context6.stop();
+                        return _context5.stop();
                     }
-                  }, _callee6);
-                }));
-                return function (_x4) {
-                  return _ref4.apply(this, arguments);
-                };
-              }());
-
-              /***
-               * 获取特定接口的参数及返回值类型
-               */
-              mcpServer.tool("getApiTypes", "获取特定接口的参数及返回值类型", {
-                path: z.string(),
-                method: z.string()
-              }, /*#__PURE__*/function () {
-                var _ref5 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee7(args) {
-                  return _regeneratorRuntime().wrap(function _callee7$(_context7) {
-                    while (1) switch (_context7.prev = _context7.next) {
-                      case 0:
-                        _context7.t0 = JSON;
-                        _context7.next = 3;
-                        return swaggerServer.getApiTypes(args.path, args.method);
-                      case 3:
-                        _context7.t1 = _context7.sent;
-                        _context7.t2 = _context7.t0.stringify.call(_context7.t0, _context7.t1);
-                        _context7.t3 = {
-                          type: "text",
-                          text: _context7.t2
-                        };
-                        _context7.t4 = [_context7.t3];
-                        return _context7.abrupt("return", {
-                          content: _context7.t4
-                        });
-                      case 8:
-                      case "end":
-                        return _context7.stop();
-                    }
-                  }, _callee7);
-                }));
-                return function (_x5) {
-                  return _ref5.apply(this, arguments);
-                };
-              }());
-              _context9.next = 10;
-              return mcpServer.connect(transport);
-            case 10:
-              // Connect to the MCP mcpServer
-              console.log("MCP server connected:", "http://localhost:".concat((_server$config = server.config) === null || _server$config === void 0 || (_server$config = _server$config.server) === null || _server$config === void 0 ? void 0 : _server$config.port, "/_mcp/sse/swagger"));
-              server.middlewares.use('/_mcp/sse/swagger', /*#__PURE__*/function () {
-                var _ref6 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee8(req, res) {
-                  var _sessionId, body, _iteratorAbruptCompletion, _didIteratorError, _iteratorError, _iterator, _step, chunk, json, _sessionId2, _req$headers$accept;
+                  }, _callee5);
+                })));
+                mcpServer.tool("getModuleApis", "获取特定模块下的所有接口", {
+                  module: z.string().describe("模块名称")
+                }, /*#__PURE__*/function () {
+                  var _ref4 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee6(_ref3) {
+                    var module, res;
+                    return _regeneratorRuntime().wrap(function _callee6$(_context6) {
+                      while (1) switch (_context6.prev = _context6.next) {
+                        case 0:
+                          module = _ref3.module;
+                          _context6.next = 3;
+                          return swaggerServer.getModuleApis(module);
+                        case 3:
+                          res = _context6.sent;
+                          return _context6.abrupt("return", {
+                            content: [{
+                              type: "text",
+                              text: JSON.stringify(res)
+                            }]
+                          });
+                        case 5:
+                        case "end":
+                          return _context6.stop();
+                      }
+                    }, _callee6);
+                  }));
+                  return function (_x5) {
+                    return _ref4.apply(this, arguments);
+                  };
+                }());
+                mcpServer.tool("getApiTypes", "获取特定接口的参数及返回值类型", {
+                  path: z.string(),
+                  method: z.string()
+                }, /*#__PURE__*/function () {
+                  var _ref5 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee7(args) {
+                    return _regeneratorRuntime().wrap(function _callee7$(_context7) {
+                      while (1) switch (_context7.prev = _context7.next) {
+                        case 0:
+                          _context7.t0 = JSON;
+                          _context7.next = 3;
+                          return swaggerServer.getApiTypes(args.path, args.method);
+                        case 3:
+                          _context7.t1 = _context7.sent;
+                          _context7.t2 = _context7.t0.stringify.call(_context7.t0, _context7.t1);
+                          _context7.t3 = {
+                            type: "text",
+                            text: _context7.t2
+                          };
+                          _context7.t4 = [_context7.t3];
+                          return _context7.abrupt("return", {
+                            content: _context7.t4
+                          });
+                        case 8:
+                        case "end":
+                          return _context7.stop();
+                      }
+                    }, _callee7);
+                  }));
+                  return function (_x6) {
+                    return _ref5.apply(this, arguments);
+                  };
+                }());
+                return mcpServer;
+              };
+              ENDPOINT = "/_mcp/sse/swagger";
+              server.middlewares.use(ENDPOINT, /*#__PURE__*/function () {
+                var _ref6 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee8(req, res, next) {
+                  var url, _sessionId, json, transport, mcpServer, initTransport;
                   return _regeneratorRuntime().wrap(function _callee8$(_context8) {
                     while (1) switch (_context8.prev = _context8.next) {
                       case 0:
-                        if (!(req.method === 'POST')) {
-                          _context8.next = 57;
+                        _context8.prev = 0;
+                        url = new URL(req.url || "", "http://".concat(req.headers.host));
+                        _sessionId = req.headers["mcp-session-id"] || url.searchParams.get("sessionId");
+                        /* ================= POST 请求处理 ================= */
+                        if (!(req.method === "POST")) {
+                          _context8.next = 24;
                           break;
                         }
-                        _sessionId = req.headers["mcp-session-id"];
-                        body = '';
-                        _iteratorAbruptCompletion = false;
-                        _didIteratorError = false;
-                        _context8.prev = 5;
-                        _iterator = _asyncIterator(req);
-                      case 7:
-                        _context8.next = 9;
-                        return _iterator.next();
-                      case 9:
-                        if (!(_iteratorAbruptCompletion = !(_step = _context8.sent).done)) {
-                          _context8.next = 15;
+                        _context8.next = 6;
+                        return readJsonBody(req);
+                      case 6:
+                        json = _context8.sent;
+                        if (!(_sessionId && transports[_sessionId])) {
+                          _context8.next = 11;
                           break;
                         }
-                        chunk = _step.value;
-                        body += chunk;
-                      case 12:
-                        _iteratorAbruptCompletion = false;
-                        _context8.next = 7;
-                        break;
-                      case 15:
+                        transport = transports[_sessionId];
                         _context8.next = 21;
                         break;
-                      case 17:
-                        _context8.prev = 17;
-                        _context8.t0 = _context8["catch"](5);
-                        _didIteratorError = true;
-                        _iteratorError = _context8.t0;
-                      case 21:
-                        _context8.prev = 21;
-                        _context8.prev = 22;
-                        if (!(_iteratorAbruptCompletion && _iterator.return != null)) {
-                          _context8.next = 26;
+                      case 11:
+                        if (!(!_sessionId && isInitializeRequest(json))) {
+                          _context8.next = 19;
                           break;
                         }
-                        _context8.next = 26;
-                        return _iterator.return();
-                      case 26:
-                        _context8.prev = 26;
-                        if (!_didIteratorError) {
-                          _context8.next = 29;
-                          break;
-                        }
-                        throw _iteratorError;
-                      case 29:
-                        return _context8.finish(26);
-                      case 30:
-                        return _context8.finish(21);
-                      case 31:
-                        json = JSON.parse(body);
-                        console.log("post request received");
-                        console.log("body: ", json);
-                        _context8.prev = 34;
-                        if (!(_sessionId && transports[_sessionId])) {
-                          _context8.next = 40;
-                          break;
-                        }
-                        transport = transports[_sessionId];
-                        _context8.next = 39;
+                        transport = new StreamableHTTPServerTransport({
+                          sessionIdGenerator: function sessionIdGenerator() {
+                            return randomUUID();
+                          },
+                          // 如果你的编辑器不支持 SSE 只能接收 JSON 响应，可以设为 true
+                          // 但大多数现代编辑器建议保持默认（false）以支持标准流
+                          enableJsonResponse: false,
+                          onsessioninitialized: function onsessioninitialized(sid) {
+                            transports[sid] = transport;
+                            console.log("[MCP] Session initialized: ".concat(sid));
+                          }
+                        });
+                        transport.onclose = function () {
+                          if (transport.sessionId) delete transports[transport.sessionId];
+                          console.log("[MCP] Session closed: ".concat(transport.sessionId));
+                        };
+
+                        // 为这个 Transport 绑定一个新的 Server 实例，防止 "Already initialized"
+                        mcpServer = createNewServer();
+                        _context8.next = 17;
+                        return mcpServer.connect(transport);
+                      case 17:
+                        _context8.next = 21;
+                        break;
+                      case 19:
+                        res.statusCode = 400;
+                        return _context8.abrupt("return", res.end(JSON.stringify({
+                          error: "Invalid session or not an initialize request"
+                        })));
+                      case 21:
+                        _context8.next = 23;
                         return transport.handleRequest(req, res, json);
-                      case 39:
+                      case 23:
                         return _context8.abrupt("return");
-                      case 40:
-                        if (_sessionId) {
-                          _context8.next = 46;
+                      case 24:
+                        if (!(req.method === "GET")) {
+                          _context8.next = 33;
                           break;
                         }
-                        _context8.next = 43;
-                        return transport.handleRequest(req, res, json);
-                      case 43:
-                        // session ID will only be available (if in not Stateless-Mode)
-                        // after handling the first request
-                        _sessionId2 = transport.sessionId;
-                        if (_sessionId2) {
-                          transports[_sessionId2] = transport;
+                        if (!(!_sessionId || !transports[_sessionId])) {
+                          _context8.next = 30;
+                          break;
                         }
+                        // 如果是初次连接请求且没有 sessionId，让 transport 处理以发送 endpoint 事件
+                        // 这里创建一个临时的 transport 来引导客户端
+                        initTransport = new StreamableHTTPServerTransport();
+                        _context8.next = 29;
+                        return initTransport.handleRequest(req, res);
+                      case 29:
                         return _context8.abrupt("return");
-                      case 46:
-                        res.statusCode = 400;
-                        res.end("Bad Request: invalid session ID or method.");
-                        return _context8.abrupt("return");
-                      case 51:
-                        _context8.prev = 51;
-                        _context8.t1 = _context8["catch"](34);
-                        console.error('Error handling MCP request:', _context8.t1);
-                        res.statusCode = 500;
-                        res.end("Internal server error.");
+                      case 30:
+                        _context8.next = 32;
+                        return transports[_sessionId].handleRequest(req, res);
+                      case 32:
                         return _context8.abrupt("return");
-                      case 57:
-                        if (!(req.method === 'GET')) {
-                          _context8.next = 64;
-                          break;
-                        }
-                        if ((_req$headers$accept = req.headers.accept) !== null && _req$headers$accept !== void 0 && _req$headers$accept.includes('text/event-stream')) {
-                          _context8.next = 61;
-                          break;
+                      case 33:
+                        res.statusCode = 405;
+                        res.end("Method Not Allowed");
+                        _context8.next = 41;
+                        break;
+                      case 37:
+                        _context8.prev = 37;
+                        _context8.t0 = _context8["catch"](0);
+                        console.error("[MCP Error]", _context8.t0);
+                        if (!res.headersSent) {
+                          res.statusCode = 500;
+                          res.end("Internal Server Error");
                         }
-                        res.writeHead(405);
-                        return _context8.abrupt("return", res.end());
-                      case 61:
-                        res.writeHead(200, {
-                          'Content-Type': 'text/event-stream',
-                          'Cache-Control': 'no-cache',
-                          Connection: 'keep-alive'
-                        });
-                        transport.onmessage = function (message) {
-                          res.write(message);
-                        };
-                        return _context8.abrupt("return");
-                      case 64:
+                      case 41:
                       case "end":
                         return _context8.stop();
                     }
-                  }, _callee8, null, [[5, 17, 21, 31], [22,, 26, 30], [34, 51]]);
+                  }, _callee8, null, [[0, 37]]);
                 }));
-                return function (_x6, _x7) {
+                return function (_x7, _x8, _x9) {
                   return _ref6.apply(this, arguments);
                 };
               }());
-              _context9.next = 17;
-              break;
-            case 14:
-              _context9.prev = 14;
-              _context9.t0 = _context9["catch"](0);
-              console.log("MCP server error", _context9.t0);
-            case 17:
+              console.log("\u2705 MCP Swagger Server mounted at http://localhost:".concat((_server$config = server.config) === null || _server$config === void 0 || (_server$config = _server$config.server) === null || _server$config === void 0 ? void 0 : _server$config.port).concat(ENDPOINT));
+            case 5:
             case "end":
               return _context9.stop();
           }
-        }, _callee9, null, [[0, 14]]);
+        }, _callee9);
       }))();
     }
   };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "vite-plugin-swagger-mcp",
-  "version": "0.0.16",
+  "version": "0.0.19",
   "description": "vite plugin for swagger mcp",
   "homepage": "https://github.com/mmdctjj/vite-plugin-swagger-mcp",
   "repository": {