vite-plugin-swagger-mcp 0.0.13 → 0.0.15

package/dist/cjs/index.js

@@ -210,21 +210,60 @@ 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) => {
-          try {
-            await transport.handleRequest(req, res);
-          } catch (err) {
-            console.error("[MCP]", err);
-            if (!res.headersSent) {
+          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;
+              }
+              if (!sessionId) {
+                await mcpServer.connect(transport);
+                await transport.handleRequest(req, res, json);
+                const sessionId2 = transport.sessionId;
+                if (sessionId2) {
+                  transports[sessionId2] = transport;
+                }
+                return;
+              }
+              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();
+              res.end("Internal server error.");
+              return;
             }
           }
+          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;
+          }
         });
       } catch (error) {
         console.log("MCP server error", error);

package/dist/esm/index.js

@@ -10,6 +10,8 @@ function _createClass(Constructor, protoProps, staticProps) { if (protoProps) _d
 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 _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";
@@ -240,178 +242,275 @@ export default function vitePluginSwaggerMcp(_ref) {
         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();
+              try {
+                transport = new StreamableHTTPServerTransport({
+                  sessionIdGenerator: function sessionIdGenerator() {
+                    return randomUUID();
+                  },
+                  onsessioninitialized: function onsessioninitialized(sessionId) {
+                    // Store the transport by session ID
+                    transports[sessionId] = transport;
                   }
-                }, _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) {
+                }); // 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:
-                        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);
-                };
-              }());
+                  }, _callee5);
+                })));
 
-              /***
-               * 获取特定接口的参数及返回值类型
-               */
-              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);
-                };
-              }());
+                /***
+                 * 获取特定模块下的所有接口及返回值类型
+                 */
+                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);
+                  };
+                }());
 
-              // Connect to the MCP mcpServer
-              _context9.next = 10;
-              return mcpServer.connect(transport);
-            case 10:
-              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) {
-                  return _regeneratorRuntime().wrap(function _callee8$(_context8) {
-                    while (1) switch (_context8.prev = _context8.next) {
-                      case 0:
-                        _context8.prev = 0;
-                        _context8.next = 3;
-                        return transport.handleRequest(req, res);
-                      case 3:
-                        _context8.next = 9;
-                        break;
-                      case 5:
-                        _context8.prev = 5;
-                        _context8.t0 = _context8["catch"](0);
-                        console.error('[MCP]', _context8.t0);
-                        if (!res.headersSent) {
+                /***
+                 * 获取特定接口的参数及返回值类型
+                 */
+                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);
+                  };
+                }());
+
+                // 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 = 59;
+                            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 = 48;
+                            break;
+                          }
+                          _context8.next = 43;
+                          return mcpServer.connect(transport);
+                        case 43:
+                          _context8.next = 45;
+                          return transport.handleRequest(req, res, json);
+                        case 45:
+                          // 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 48:
+                          res.statusCode = 400;
+                          res.end("Bad Request: invalid session ID or method.");
+                          return _context8.abrupt("return");
+                        case 53:
+                          _context8.prev = 53;
+                          _context8.t1 = _context8["catch"](34);
+                          console.error('Error handling MCP request:', _context8.t1);
                           res.statusCode = 500;
-                          res.end();
-                        }
-                      case 9:
-                      case "end":
-                        return _context8.stop();
-                    }
-                  }, _callee8, null, [[0, 5]]);
-                }));
-                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:
+                          res.end("Internal server error.");
+                          return _context8.abrupt("return");
+                        case 59:
+                          if (!(req.method === 'GET')) {
+                            _context8.next = 66;
+                            break;
+                          }
+                          if ((_req$headers$accept = req.headers.accept) !== null && _req$headers$accept !== void 0 && _req$headers$accept.includes('text/event-stream')) {
+                            _context8.next = 63;
+                            break;
+                          }
+                          res.writeHead(405);
+                          return _context8.abrupt("return", res.end());
+                        case 63:
+                          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 66:
+                        case "end":
+                          return _context8.stop();
+                      }
+                    }, _callee8, null, [[5, 17, 21, 31], [22,, 26, 30], [34, 53]]);
+                  }));
+                  return function (_x6, _x7) {
+                    return _ref6.apply(this, arguments);
+                  };
+                }());
+              } catch (error) {
+                console.log("MCP server error", error);
+              }
+            case 1:
             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.13",
+  "version": "0.0.15",
   "description": "vite plugin for swagger mcp",
   "homepage": "https://github.com/mmdctjj/vite-plugin-swagger-mcp",
   "repository": {