vite-plugin-swagger-mcp 0.0.17 → 0.0.20

package/dist/cjs/index.d.ts

@@ -1,5 +1,6 @@
 import type { Plugin } from "vite";
 import { SchemaObject } from "./type";
+export declare const isInitializeRequest: (value: unknown) => boolean;
 export declare class SwaggerMcpServer {
     private swaggerUrl;
     private token?;
@@ -42,7 +43,7 @@ export declare class SwaggerMcpServer {
         method: string;
     }>;
 }
-export default function vitePluginSwaggerMcp(options: {
+export default function vitePluginSwaggerMcp({ swaggerUrl, token, }: {
     swaggerUrl: string;
     token?: string;
 }): Plugin;

package/dist/cjs/index.js

@@ -1,8 +1,6 @@
-var __create = Object.create;
 var __defProp = Object.defineProperty;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
 var __getOwnPropNames = Object.getOwnPropertyNames;
-var __getProtoOf = Object.getPrototypeOf;
 var __hasOwnProp = Object.prototype.hasOwnProperty;
 var __export = (target, all) => {
   for (var name in all)
@@ -16,28 +14,28 @@ var __copyProps = (to, from, except, desc) => {
   }
   return to;
 };
-var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
-  // If the importer is in node compatibility mode or this is not an ESM
-  // file that has been converted to a CommonJS file using a Babel-
-  // compatible transform (i.e. "__esModule" has not been set), then set
-  // "default" to the CommonJS "module.exports" for node compatibility.
-  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
-  mod
-));
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
 
 // src/index.ts
 var src_exports = {};
 __export(src_exports, {
   SwaggerMcpServer: () => SwaggerMcpServer,
-  default: () => vitePluginSwaggerMcp
+  default: () => vitePluginSwaggerMcp,
+  isInitializeRequest: () => isInitializeRequest
 });
 module.exports = __toCommonJS(src_exports);
 var import_mcp = require("@modelcontextprotocol/sdk/server/mcp.js");
-var import_node_crypto = require("node:crypto");
-var z = __toESM(require("zod"));
-var import_streamableHttp = require("@modelcontextprotocol/sdk/server/streamableHttp");
+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_v4 = require("zod/v4");
+async function readJsonBody(req) {
+  let body = "";
+  for await (const chunk of req)
+    body += chunk;
+  return JSON.parse(body || "{}");
+}
+var isInitializeRequest = (value) => import_types.InitializeRequestParamsSchema.safeParse(value).success;
 var SwaggerMcpServer = class {
   constructor(swaggerUrl, token) {
     this.swaggerUrl = swaggerUrl;
@@ -145,122 +143,105 @@ var SwaggerMcpServer = class {
     };
   }
 };
-function vitePluginSwaggerMcp(options) {
-  const transports = {};
+var transports = {};
+function vitePluginSwaggerMcp({
+  swaggerUrl,
+  token
+}) {
   return {
     name: "vite-plugin-swagger-mcp",
-    configureServer(vite) {
-      var _a;
-      const swaggerServer = new SwaggerMcpServer(
-        options.swaggerUrl,
-        options.token
-      );
-      const createMcpServer = () => {
-        const server = new import_mcp.McpServer({
+    enforce: "pre",
+    async configureServer(server) {
+      var _a, _b;
+      const swaggerServer = new SwaggerMcpServer(swaggerUrl, token);
+      const createNewServer = () => {
+        const mcpServer = new import_mcp.McpServer({
           name: "swagger-mcp-server",
           version: "0.1.0"
         });
-        server.tool("getModules", "获取模块列表", async () => {
+        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) }] };
         });
-        server.tool(
+        mcpServer.tool(
+          "getModuleApis",
+          "获取特定模块下的所有接口",
+          { module: import_v4.z.string().describe("模块名称") },
+          async ({ module: module2 }) => {
+            const res = await swaggerServer.getModuleApis(module2);
+            return { content: [{ type: "text", text: JSON.stringify(res) }] };
+          }
+        );
+        mcpServer.tool(
           "getApiTypes",
           "获取特定接口的参数及返回值类型",
-          { path: z.string(), method: 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)) }]
           })
         );
-        server.registerTool(
-          "getApiTypes",
-          {
-            description: "获取接口类型定义",
-            inputSchema: {
-              path: z.string(),
-              method: z.string()
-            }
-          },
-          async ({ path, method }) => ({
-            content: [
-              {
-                type: "text",
-                text: JSON.stringify(await swaggerServer.getApiTypes(path, method))
-              }
-            ]
-          })
-        );
-        return server;
+        return mcpServer;
       };
-      vite.middlewares.use("/_mcp/swagger", async (req, res) => {
-        const sessionId = req.headers["mcp-session-id"];
-        if (req.method === "POST") {
-          let body = "";
-          for await (const chunk of req)
-            body += chunk;
-          const json = JSON.parse(body);
-          try {
+      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 json = await readJsonBody(req);
             let transport;
             if (sessionId && transports[sessionId]) {
               transport = transports[sessionId];
-            } else if (!sessionId && (0, import_types.isInitializeRequest)(json)) {
+            } else if (!sessionId && isInitializeRequest(json)) {
               transport = new import_streamableHttp.StreamableHTTPServerTransport({
                 sessionIdGenerator: () => (0, import_node_crypto.randomUUID)(),
-                onsessioninitialized(id) {
-                  transports[id] = transport;
+                // 如果你的编辑器不支持 SSE 只能接收 JSON 响应，可以设为 true
+                // 但大多数现代编辑器建议保持默认（false）以支持标准流
+                enableJsonResponse: false,
+                onsessioninitialized: (sid) => {
+                  transports[sid] = transport;
+                  console.log(`[MCP] Session initialized: ${sid}`);
                 }
               });
               transport.onclose = () => {
-                if (transport.sessionId) {
+                if (transport.sessionId)
                   delete transports[transport.sessionId];
-                }
+                console.log(`[MCP] Session closed: ${transport.sessionId}`);
               };
-              const mcpServer = createMcpServer();
+              const mcpServer = createNewServer();
               await mcpServer.connect(transport);
             } else {
               res.statusCode = 400;
-              return res.end("Bad Request");
+              return res.end(JSON.stringify({ error: "Invalid session or not an initialize request" }));
             }
             await transport.handleRequest(req, res, json);
-          } catch (err) {
-            console.error(err);
-            res.statusCode = 500;
-            res.end("Internal Error");
+            return;
           }
-          return;
-        }
-        if (req.method === "GET") {
-          if (!sessionId || !transports[sessionId]) {
-            res.statusCode = 400;
-            return res.end("Missing session");
+          if (req.method === "GET") {
+            if (!sessionId || !transports[sessionId]) {
+              const initTransport = new import_streamableHttp.StreamableHTTPServerTransport();
+              await initTransport.handleRequest(req, res);
+              return;
+            }
+            await transports[sessionId].handleRequest(req, res);
+            return;
+          }
+          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");
           }
-          await transports[sessionId].handleRequest(req, res);
-          return;
         }
-        res.statusCode = 405;
-        res.end();
       });
-      console.log(
-        `Swagger MCP ready: http://localhost:${(_a = vite.config.server) == null ? void 0 : _a.port}/_mcp/swagger`
-      );
+      console.log(`✅ MCP Swagger Server mounted at http://localhost:${(_b = (_a = server.config) == null ? void 0 : _a.server) == null ? void 0 : _b.port}${ENDPOINT}`);
     }
   };
 }
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
-  SwaggerMcpServer
+  SwaggerMcpServer,
+  isInitializeRequest
 });

package/dist/esm/index.d.ts

@@ -1,5 +1,6 @@
 import type { Plugin } from "vite";
 import { SchemaObject } from "./type";
+export declare const isInitializeRequest: (value: unknown) => boolean;
 export declare class SwaggerMcpServer {
     private swaggerUrl;
     private token?;
@@ -42,7 +43,7 @@ export declare class SwaggerMcpServer {
         method: string;
     }>;
 }
-export default function vitePluginSwaggerMcp(options: {
+export default function vitePluginSwaggerMcp({ swaggerUrl, token, }: {
     swaggerUrl: string;
     token?: string;
 }): Plugin;

package/dist/esm/index.js

@@ -2,24 +2,91 @@ 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 { InitializeRequestParamsSchema } from "@modelcontextprotocol/sdk/types";
 import { randomUUID } from "node:crypto";
-import * as z from "zod";
-import { StreamableHTTPServerTransport } from "@modelcontextprotocol/sdk/server/streamableHttp";
-import { isInitializeRequest } from "@modelcontextprotocol/sdk/types";
-/* =========================
- * Swagger MCP Server
- * ========================= */
+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 isInitializeRequest = function isInitializeRequest(value) {
+  return InitializeRequestParamsSchema.safeParse(value).success;
+};
 export var SwaggerMcpServer = /*#__PURE__*/function () {
   function SwaggerMcpServer(swaggerUrl, token) {
     _classCallCheck(this, SwaggerMcpServer);
@@ -184,7 +251,7 @@ export var SwaggerMcpServer = /*#__PURE__*/function () {
           }
         }, _callee3, this);
       }));
-      function getModuleApis(_x) {
+      function getModuleApis(_x2) {
         return _getModuleApis.apply(this, arguments);
       }
       return getModuleApis;
@@ -223,7 +290,7 @@ export var SwaggerMcpServer = /*#__PURE__*/function () {
           }
         }, _callee4, this);
       }));
-      function getApiTypes(_x2, _x3) {
+      function getApiTypes(_x3, _x4) {
         return _getApiTypes.apply(this, arguments);
       }
       return getApiTypes;
@@ -232,269 +299,229 @@ export var SwaggerMcpServer = /*#__PURE__*/function () {
   return SwaggerMcpServer;
 }();
 
-/* =========================
- * Vite Plugin
- * ========================= */
-
-export default function vitePluginSwaggerMcp(options) {
-  var transports = {};
+// 存储会话 ID 对应的 Transport 实例
+var transports = {};
+export default function vitePluginSwaggerMcp(_ref) {
+  var swaggerUrl = _ref.swaggerUrl,
+    token = _ref.token;
   return {
     name: "vite-plugin-swagger-mcp",
-    configureServer: function configureServer(vite) {
-      var _vite$config$server;
-      var swaggerServer = new SwaggerMcpServer(options.swaggerUrl, options.token);
-
-      /** 创建 MCP Server（每个 session 共享一个 definition） */
-      var createMcpServer = function createMcpServer() {
-        var server = new McpServer({
-          name: "swagger-mcp-server",
-          version: "0.1.0"
-        });
-
-        /***
-        * 获取模块列表
-        */
-        server.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)
-                  }]
+    enforce: "pre",
+    configureServer: function configureServer(server) {
+      return _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee9() {
+        var _server$config;
+        var swaggerServer, createNewServer, ENDPOINT;
+        return _regeneratorRuntime().wrap(function _callee9$(_context9) {
+          while (1) switch (_context9.prev = _context9.next) {
+            case 0:
+              swaggerServer = new SwaggerMcpServer(swaggerUrl, token); // --- 辅助函数：创建并配置一个新的 MCP Server 实例 ---
+              createNewServer = function createNewServer() {
+                var mcpServer = new McpServer({
+                  name: "swagger-mcp-server",
+                  version: "0.1.0"
                 });
-              case 4:
-              case "end":
-                return _context5.stop();
-            }
-          }, _callee5);
-        })));
 
-        /***
-         * 获取特定接口的参数及返回值类型
-         */
-        server.tool("getApiTypes", "获取特定接口的参数及返回值类型", {
-          path: z.string(),
-          method: z.string()
-        }, /*#__PURE__*/function () {
-          var _ref2 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee6(args) {
-            return _regeneratorRuntime().wrap(function _callee6$(_context6) {
-              while (1) switch (_context6.prev = _context6.next) {
-                case 0:
-                  _context6.t0 = JSON;
-                  _context6.next = 3;
-                  return swaggerServer.getApiTypes(args.path, args.method);
-                case 3:
-                  _context6.t1 = _context6.sent;
-                  _context6.t2 = _context6.t0.stringify.call(_context6.t0, _context6.t1);
-                  _context6.t3 = {
-                    type: "text",
-                    text: _context6.t2
+                // 注册所有工具
+                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);
+                })));
+                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);
                   };
-                  _context6.t4 = [_context6.t3];
-                  return _context6.abrupt("return", {
-                    content: _context6.t4
-                  });
-                case 8:
-                case "end":
-                  return _context6.stop();
-              }
-            }, _callee6);
-          }));
-          return function (_x4) {
-            return _ref2.apply(this, arguments);
-          };
-        }());
-        server.registerTool("getApiTypes", {
-          description: "获取接口类型定义",
-          inputSchema: {
-            path: z.string(),
-            method: z.string()
-          }
-        }, /*#__PURE__*/function () {
-          var _ref4 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee7(_ref3) {
-            var path, method;
-            return _regeneratorRuntime().wrap(function _callee7$(_context7) {
-              while (1) switch (_context7.prev = _context7.next) {
-                case 0:
-                  path = _ref3.path, method = _ref3.method;
-                  _context7.t0 = JSON;
-                  _context7.next = 4;
-                  return swaggerServer.getApiTypes(path, method);
-                case 4:
-                  _context7.t1 = _context7.sent;
-                  _context7.t2 = _context7.t0.stringify.call(_context7.t0, _context7.t1);
-                  _context7.t3 = {
-                    type: "text",
-                    text: _context7.t2
+                }());
+                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);
                   };
-                  _context7.t4 = [_context7.t3];
-                  return _context7.abrupt("return", {
-                    content: _context7.t4
-                  });
-                case 9:
-                case "end":
-                  return _context7.stop();
-              }
-            }, _callee7);
-          }));
-          return function (_x5) {
-            return _ref4.apply(this, arguments);
-          };
-        }());
-        return server;
-      };
+                }());
+                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:
+                        _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;
+                        }
+                        _context8.next = 6;
+                        return readJsonBody(req);
+                      case 6:
+                        json = _context8.sent;
+                        if (!(_sessionId && transports[_sessionId])) {
+                          _context8.next = 11;
+                          break;
+                        }
+                        transport = transports[_sessionId];
+                        _context8.next = 21;
+                        break;
+                      case 11:
+                        if (!(!_sessionId && isInitializeRequest(json))) {
+                          _context8.next = 19;
+                          break;
+                        }
+                        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));
+                        };
 
-      /* =========================
-       * MCP HTTP Endpoint
-       * ========================= */
-
-      vite.middlewares.use("/_mcp/swagger", /*#__PURE__*/function () {
-        var _ref5 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee8(req, res) {
-          var sessionId, body, _iteratorAbruptCompletion, _didIteratorError, _iteratorError, _iterator, _step, chunk, json, transport, mcpServer;
-          return _regeneratorRuntime().wrap(function _callee8$(_context8) {
-            while (1) switch (_context8.prev = _context8.next) {
-              case 0:
-                sessionId = req.headers["mcp-session-id"];
-                /* ---------- POST (JSON-RPC) ---------- */
-                if (!(req.method === "POST")) {
-                  _context8.next = 57;
-                  break;
-                }
-                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;
-                  break;
-                }
-                chunk = _step.value;
-                body += chunk;
-              case 12:
-                _iteratorAbruptCompletion = false;
-                _context8.next = 7;
-                break;
-              case 15:
-                _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;
-                  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);
-                _context8.prev = 32;
-                if (!(sessionId && transports[sessionId])) {
-                  _context8.next = 37;
-                  break;
-                }
-                transport = transports[sessionId];
-                _context8.next = 47;
-                break;
-              case 37:
-                if (!(!sessionId && isInitializeRequest(json))) {
-                  _context8.next = 45;
-                  break;
-                }
-                transport = new StreamableHTTPServerTransport({
-                  sessionIdGenerator: function sessionIdGenerator() {
-                    return randomUUID();
-                  },
-                  onsessioninitialized: function onsessioninitialized(id) {
-                    transports[id] = transport;
-                  }
-                });
-                transport.onclose = function () {
-                  if (transport.sessionId) {
-                    delete transports[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 23:
+                        return _context8.abrupt("return");
+                      case 24:
+                        if (!(req.method === "GET")) {
+                          _context8.next = 33;
+                          break;
+                        }
+                        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 30:
+                        _context8.next = 32;
+                        return transports[_sessionId].handleRequest(req, res);
+                      case 32:
+                        return _context8.abrupt("return");
+                      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");
+                        }
+                      case 41:
+                      case "end":
+                        return _context8.stop();
+                    }
+                  }, _callee8, null, [[0, 37]]);
+                }));
+                return function (_x7, _x8, _x9) {
+                  return _ref6.apply(this, arguments);
                 };
-                mcpServer = createMcpServer();
-                _context8.next = 43;
-                return mcpServer.connect(transport);
-              case 43:
-                _context8.next = 47;
-                break;
-              case 45:
-                res.statusCode = 400;
-                return _context8.abrupt("return", res.end("Bad Request"));
-              case 47:
-                _context8.next = 49;
-                return transport.handleRequest(req, res, json);
-              case 49:
-                _context8.next = 56;
-                break;
-              case 51:
-                _context8.prev = 51;
-                _context8.t1 = _context8["catch"](32);
-                console.error(_context8.t1);
-                res.statusCode = 500;
-                res.end("Internal Error");
-              case 56:
-                return _context8.abrupt("return");
-              case 57:
-                if (!(req.method === "GET")) {
-                  _context8.next = 64;
-                  break;
-                }
-                if (!(!sessionId || !transports[sessionId])) {
-                  _context8.next = 61;
-                  break;
-                }
-                res.statusCode = 400;
-                return _context8.abrupt("return", res.end("Missing session"));
-              case 61:
-                _context8.next = 63;
-                return transports[sessionId].handleRequest(req, res);
-              case 63:
-                return _context8.abrupt("return");
-              case 64:
-                res.statusCode = 405;
-                res.end();
-              case 66:
-              case "end":
-                return _context8.stop();
-            }
-          }, _callee8, null, [[5, 17, 21, 31], [22,, 26, 30], [32, 51]]);
-        }));
-        return function (_x6, _x7) {
-          return _ref5.apply(this, arguments);
-        };
-      }());
-      console.log("Swagger MCP ready: http://localhost:".concat((_vite$config$server = vite.config.server) === null || _vite$config$server === void 0 ? void 0 : _vite$config$server.port, "/_mcp/swagger"));
+              }());
+              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);
+      }))();
     }
   };
 }

package/package.json

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