vite-plugin-swagger-mcp 0.0.16 → 0.0.17

package/dist/cjs/index.d.ts

@@ -42,7 +42,7 @@ export declare class SwaggerMcpServer {
         method: string;
     }>;
 }
-export default function vitePluginSwaggerMcp({ swaggerUrl, token, }: {
+export default function vitePluginSwaggerMcp(options: {
     swaggerUrl: string;
     token?: string;
 }): Plugin;

package/dist/cjs/index.js

@@ -1,6 +1,8 @@
+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)
@@ -14,6 +16,14 @@ 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
@@ -24,9 +34,10 @@ __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_node_crypto = require("node:crypto");
-var import_zod = require("zod");
+var z = __toESM(require("zod"));
+var import_streamableHttp = require("@modelcontextprotocol/sdk/server/streamableHttp");
+var import_types = require("@modelcontextprotocol/sdk/types");
 var SwaggerMcpServer = class {
   constructor(swaggerUrl, token) {
     this.swaggerUrl = swaggerUrl;
@@ -134,34 +145,22 @@ var SwaggerMcpServer = class {
     };
   }
 };
-var transports = {};
-function vitePluginSwaggerMcp({
-  swaggerUrl,
-  token
-}) {
+function vitePluginSwaggerMcp(options) {
+  const transports = {};
   return {
     name: "vite-plugin-swagger-mcp",
-    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 mcpServer = new import_mcp.McpServer({
+    configureServer(vite) {
+      var _a;
+      const swaggerServer = new SwaggerMcpServer(
+        options.swaggerUrl,
+        options.token
+      );
+      const createMcpServer = () => {
+        const server = new import_mcp.McpServer({
           name: "swagger-mcp-server",
           version: "0.1.0"
         });
-        mcpServer.tool("getModules", "获取模块列表", async () => {
+        server.tool("getModules", "获取模块列表", async () => {
           const res = await swaggerServer.getModules();
           return {
             content: [
@@ -172,33 +171,10 @@ function vitePluginSwaggerMcp({
             ]
           };
         });
-        mcpServer.tool(
-          "getModuleApis",
-          "获取特定模块下的所有接口及返回值类型",
-          {
-            module: import_zod.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) }]
-            };
-          }
-        );
-        mcpServer.tool(
+        server.tool(
           "getApiTypes",
           "获取特定接口的参数及返回值类型",
-          { path: import_zod.z.string(), method: import_zod.z.string() },
+          { path: z.string(), method: z.string() },
           async (args) => ({
             content: [
               {
@@ -210,64 +186,77 @@ function vitePluginSwaggerMcp({
             ]
           })
         );
-        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;
-          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;
+        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))
               }
-              if (!sessionId) {
-                await transport.handleRequest(req, res, json);
-                const sessionId2 = transport.sessionId;
-                if (sessionId2) {
-                  transports[sessionId2] = transport;
+            ]
+          })
+        );
+        return server;
+      };
+      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 {
+            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)(),
+                onsessioninitialized(id) {
+                  transports[id] = transport;
                 }
-                return;
-              }
+              });
+              transport.onclose = () => {
+                if (transport.sessionId) {
+                  delete transports[transport.sessionId];
+                }
+              };
+              const mcpServer = createMcpServer();
+              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("Bad Request");
             }
+            await transport.handleRequest(req, res, json);
+          } catch (err) {
+            console.error(err);
+            res.statusCode = 500;
+            res.end("Internal Error");
           }
-          if (req.method === "GET") {
-            if (!((_a2 = req.headers.accept) == null ? void 0 : _a2.includes("text/event-stream"))) {
-              res.writeHead(405);
-              return res.end();
-            }
-            res.writeHead(200, {
-              "Content-Type": "text/event-stream",
-              "Cache-Control": "no-cache",
-              Connection: "keep-alive"
-            });
-            transport.onmessage = (message) => {
-              res.write(message);
-            };
-            return;
+          return;
+        }
+        if (req.method === "GET") {
+          if (!sessionId || !transports[sessionId]) {
+            res.statusCode = 400;
+            return res.end("Missing session");
           }
-        });
-      } catch (error) {
-        console.log("MCP server error", 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`
+      );
     }
   };
 }

package/dist/esm/index.d.ts

@@ -42,7 +42,7 @@ export declare class SwaggerMcpServer {
         method: string;
     }>;
 }
-export default function vitePluginSwaggerMcp({ swaggerUrl, token, }: {
+export default function vitePluginSwaggerMcp(options: {
     swaggerUrl: string;
     token?: string;
 }): Plugin;

package/dist/esm/index.js

@@ -13,9 +13,13 @@ function _toPrimitive(t, r) { if ("object" != _typeof(t) || !t) return t; var e
 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 { randomUUID } from "node:crypto";
-import { z } from "zod";
+import * as z from "zod";
+import { StreamableHTTPServerTransport } from "@modelcontextprotocol/sdk/server/streamableHttp";
+import { isInitializeRequest } from "@modelcontextprotocol/sdk/types";
+/* =========================
+ * Swagger MCP Server
+ * ========================= */
 export var SwaggerMcpServer = /*#__PURE__*/function () {
   function SwaggerMcpServer(swaggerUrl, token) {
     _classCallCheck(this, SwaggerMcpServer);
@@ -228,292 +232,269 @@ export var SwaggerMcpServer = /*#__PURE__*/function () {
   return SwaggerMcpServer;
 }();
 
-// Map to store transports by session ID
-var transports = {};
-export default function vitePluginSwaggerMcp(_ref) {
-  var swaggerUrl = _ref.swaggerUrl,
-    token = _ref.token;
+/* =========================
+ * Vite Plugin
+ * ========================= */
+
+export default function vitePluginSwaggerMcp(options) {
+  var transports = {};
   return {
     name: "vite-plugin-swagger-mcp",
-    enforce: "pre",
-    configureServer: function configureServer(server) {
-      return _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee9() {
-        var _server$config, transport, swaggerServer, mcpServer;
-        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;
+    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)
+                  }]
+                });
+              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
+                  };
+                  _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
+                  };
+                  _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;
+      };
+
+      /* =========================
+       * 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;
                 }
-              }); // Clean up transport when closed
-              transport.onclose = function () {
-                if (transport.sessionId) {
-                  delete transports[transport.sessionId];
+                _context8.next = 26;
+                return _iterator.return();
+              case 26:
+                _context8.prev = 26;
+                if (!_didIteratorError) {
+                  _context8.next = 29;
+                  break;
                 }
-              };
-              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();
+                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];
                   }
-                }, _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;
-                        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", {
-                          content: [{
-                            type: "text",
-                            text: JSON.stringify(res)
-                          }]
-                        });
-                      case 7:
-                      case "end":
-                        return _context6.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;
-                  return _regeneratorRuntime().wrap(function _callee8$(_context8) {
-                    while (1) switch (_context8.prev = _context8.next) {
-                      case 0:
-                        if (!(req.method === 'POST')) {
-                          _context8.next = 57;
-                          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;
-                          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);
-                        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;
-                        return transport.handleRequest(req, res, json);
-                      case 39:
-                        return _context8.abrupt("return");
-                      case 40:
-                        if (_sessionId) {
-                          _context8.next = 46;
-                          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;
-                        }
-                        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.");
-                        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;
-                        }
-                        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 "end":
-                        return _context8.stop();
-                    }
-                  }, _callee8, null, [[5, 17, 21, 31], [22,, 26, 30], [34, 51]]);
-                }));
-                return function (_x6, _x7) {
-                  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:
-            case "end":
-              return _context9.stop();
-          }
-        }, _callee9, null, [[0, 14]]);
-      }))();
+                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"));
     }
   };
 }

package/package.json

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