vite-plugin-swagger-mcp 0.0.10 → 0.0.12

package/dist/cjs/index.js

@@ -215,38 +215,49 @@ function vitePluginSwaggerMcp({
           "MCP server connected:",
           `http://localhost:${(_b = (_a = server.config) == null ? void 0 : _a.server) == null ? void 0 : _b.port}/_mcp/sse/swagger`
         );
-        const SSE_PATH = "/_mcp/sse/swagger";
-        const MESSAGES_PATH = "/_mcp/sse/swagger/messages";
-        server.middlewares.use(
-          async (req, res, next) => {
-            var _a2;
-            const url = (_a2 = req.url) == null ? void 0 : _a2.split("?")[0];
-            if (url === SSE_PATH || url === MESSAGES_PATH) {
-              if (req.method === "GET") {
-                if (req.headers["accept"] !== "text/event-stream") {
-                  res.statusCode = 405;
-                  res.end("Method Not Allowed - Expected Accept: text/event-stream");
-                  return;
-                }
-                res.setHeader("Content-Type", "text/event-stream");
-                res.setHeader("Cache-Control", "no-cache");
-                res.setHeader("Connection", "keep-alive");
-                res.setHeader("X-Accel-Buffering", "no");
-              }
-              try {
-                await transport.handleRequest(req, res);
-              } catch (err) {
-                console.error("MCP Transport Error:", err);
-                if (!res.headersSent) {
-                  res.statusCode = 500;
-                  res.end("Internal MCP Error");
-                }
-              }
+        server.middlewares.use("/_mcp/sse/swagger", async (req, res) => {
+          var _a2;
+          if (req.method === "POST") {
+            let body = "";
+            for await (const chunk of req)
+              body += chunk;
+            const json = JSON.parse(body);
+            try {
+              const result = await transport.handleRequest(req, res, json);
+              res.writeHead(200, {
+                "Content-Type": "application/json"
+              });
+              res.end(JSON.stringify(result));
               return;
+            } catch (err) {
+              res.writeHead(500);
+              res.end(
+                JSON.stringify({
+                  jsonrpc: "2.0",
+                  id: (json == null ? void 0 : json.id) ?? null,
+                  error: { message: err == null ? void 0 : err.message }
+                })
+              );
             }
-            next();
           }
-        );
+          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;
+          }
+          res.statusCode = 405;
+          res.end();
+        });
       } 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";
@@ -236,7 +238,7 @@ export default function vitePluginSwaggerMcp(_ref) {
     enforce: "pre",
     configureServer: function configureServer(server) {
       return _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee9() {
-        var _server$config, transport, swaggerServer, mcpServer, SSE_PATH, MESSAGES_PATH;
+        var _server$config, transport, swaggerServer, mcpServer;
         return _regeneratorRuntime().wrap(function _callee9$(_context9) {
           while (1) switch (_context9.prev = _context9.next) {
             case 0:
@@ -372,80 +374,132 @@ export default function vitePluginSwaggerMcp(_ref) {
               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"));
-
-              // 2. 定义 SSE 和消息路径
-              SSE_PATH = "/_mcp/sse/swagger";
-              MESSAGES_PATH = "/_mcp/sse/swagger/messages"; // 规范建议分开或明确指定
-              server.middlewares.use( /*#__PURE__*/function () {
-                var _ref6 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee8(req, res, next) {
-                  var _req$url;
-                  var url;
+              server.middlewares.use('/_mcp/sse/swagger', /*#__PURE__*/function () {
+                var _ref6 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee8(req, res) {
+                  var body, _iteratorAbruptCompletion, _didIteratorError, _iteratorError, _iterator, _step, chunk, json, result, _json$id, _req$headers$accept;
                   return _regeneratorRuntime().wrap(function _callee8$(_context8) {
                     while (1) switch (_context8.prev = _context8.next) {
                       case 0:
-                        url = (_req$url = req.url) === null || _req$url === void 0 ? void 0 : _req$url.split("?")[0]; // 处理 SSE 建立连接 (GET) 和 消息接收 (POST)
-                        if (!(url === SSE_PATH || url === MESSAGES_PATH)) {
-                          _context8.next = 21;
-                          break;
-                        }
-                        if (!(req.method === "GET")) {
-                          _context8.next = 11;
+                        if (!(req.method === 'POST')) {
+                          _context8.next = 44;
                           break;
                         }
-                        if (!(req.headers["accept"] !== "text/event-stream")) {
-                          _context8.next = 7;
+                        body = '';
+                        _iteratorAbruptCompletion = false;
+                        _didIteratorError = false;
+                        _context8.prev = 4;
+                        _iterator = _asyncIterator(req);
+                      case 6:
+                        _context8.next = 8;
+                        return _iterator.next();
+                      case 8:
+                        if (!(_iteratorAbruptCompletion = !(_step = _context8.sent).done)) {
+                          _context8.next = 14;
                           break;
                         }
-                        res.statusCode = 405;
-                        res.end("Method Not Allowed - Expected Accept: text/event-stream");
-                        return _context8.abrupt("return");
-                      case 7:
-                        // 对应文档 2.2.3: 设置正确的 Content-Type
-                        res.setHeader("Content-Type", "text/event-stream");
-                        res.setHeader("Cache-Control", "no-cache");
-                        res.setHeader("Connection", "keep-alive");
-                        // 禁用某些代理的缓冲
-                        res.setHeader("X-Accel-Buffering", "no");
+                        chunk = _step.value;
+                        body += chunk;
                       case 11:
-                        _context8.prev = 11;
-                        _context8.next = 14;
-                        return transport.handleRequest(req, res);
+                        _iteratorAbruptCompletion = false;
+                        _context8.next = 6;
+                        break;
                       case 14:
                         _context8.next = 20;
                         break;
                       case 16:
                         _context8.prev = 16;
-                        _context8.t0 = _context8["catch"](11);
-                        console.error("MCP Transport Error:", _context8.t0);
-                        if (!res.headersSent) {
-                          res.statusCode = 500;
-                          res.end("Internal MCP Error");
-                        }
+                        _context8.t0 = _context8["catch"](4);
+                        _didIteratorError = true;
+                        _iteratorError = _context8.t0;
                       case 20:
+                        _context8.prev = 20;
+                        _context8.prev = 21;
+                        if (!(_iteratorAbruptCompletion && _iterator.return != null)) {
+                          _context8.next = 25;
+                          break;
+                        }
+                        _context8.next = 25;
+                        return _iterator.return();
+                      case 25:
+                        _context8.prev = 25;
+                        if (!_didIteratorError) {
+                          _context8.next = 28;
+                          break;
+                        }
+                        throw _iteratorError;
+                      case 28:
+                        return _context8.finish(25);
+                      case 29:
+                        return _context8.finish(20);
+                      case 30:
+                        json = JSON.parse(body);
+                        _context8.prev = 31;
+                        _context8.next = 34;
+                        return transport.handleRequest(req, res, json);
+                      case 34:
+                        result = _context8.sent;
+                        // 规范：POST 一定返回 response
+                        res.writeHead(200, {
+                          'Content-Type': 'application/json'
+                        });
+                        res.end(JSON.stringify(result));
                         return _context8.abrupt("return");
-                      case 21:
-                        next();
-                      case 22:
+                      case 40:
+                        _context8.prev = 40;
+                        _context8.t1 = _context8["catch"](31);
+                        res.writeHead(500);
+                        res.end(JSON.stringify({
+                          jsonrpc: '2.0',
+                          id: (_json$id = json === null || json === void 0 ? void 0 : json.id) !== null && _json$id !== void 0 ? _json$id : null,
+                          error: {
+                            message: _context8.t1 === null || _context8.t1 === void 0 ? void 0 : _context8.t1.message
+                          }
+                        }));
+                      case 44:
+                        if (!(req.method === 'GET')) {
+                          _context8.next = 51;
+                          break;
+                        }
+                        if ((_req$headers$accept = req.headers.accept) !== null && _req$headers$accept !== void 0 && _req$headers$accept.includes('text/event-stream')) {
+                          _context8.next = 48;
+                          break;
+                        }
+                        res.writeHead(405);
+                        return _context8.abrupt("return", res.end());
+                      case 48:
+                        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 51:
+                        res.statusCode = 405;
+                        res.end();
+                      case 53:
                       case "end":
                         return _context8.stop();
                     }
-                  }, _callee8, null, [[11, 16]]);
+                  }, _callee8, null, [[4, 16, 20, 30], [21,, 25, 29], [31, 40]]);
                 }));
-                return function (_x6, _x7, _x8) {
+                return function (_x6, _x7) {
                   return _ref6.apply(this, arguments);
                 };
               }());
-              _context9.next = 19;
+              _context9.next = 17;
               break;
-            case 16:
-              _context9.prev = 16;
+            case 14:
+              _context9.prev = 14;
               _context9.t0 = _context9["catch"](0);
               console.log("MCP server error", _context9.t0);
-            case 19:
+            case 17:
             case "end":
               return _context9.stop();
           }
-        }, _callee9, null, [[0, 16]]);
+        }, _callee9, null, [[0, 14]]);
       }))();
     }
   };

package/package.json

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