asrai-mcp 0.5.0 → 0.5.1

package/package.json

@@ -1,8 +1,15 @@
 {
   "name": "asrai-mcp",
-  "version": "0.5.0",
+  "version": "0.5.1",
   "description": "Asrai crypto analysis MCP server — pay-per-use via x402 on Base. Zero install: just npx.",
-  "keywords": ["mcp", "crypto", "asrai", "x402", "trading", "signals"],
+  "keywords": [
+    "mcp",
+    "crypto",
+    "asrai",
+    "x402",
+    "trading",
+    "signals"
+  ],
   "homepage": "https://asrai.me/agents",
   "repository": {
     "type": "git",

package/src/sse-server.js

@@ -23,6 +23,7 @@ import { randomBytes, randomUUID } from "node:crypto";
 import { privateKeyToAccount } from "viem/accounts";
 import { SSEServerTransport } from "@modelcontextprotocol/sdk/server/sse.js";
 import { StreamableHTTPServerTransport } from "@modelcontextprotocol/sdk/server/streamableHttp.js";
+import { isInitializeRequest } from "@modelcontextprotocol/sdk/types.js";
 import { createServer } from "./server.js";
 import { connectionStorage } from "./tools.js";
 
@@ -30,6 +31,7 @@ const app = express();
 app.use(express.json());
 
 // Track active SSE transports by session ID (for POST /messages routing)
+// Stores { transport, key } so the key is available when POST /messages arrives
 const sseTransports = {};
 
 // Track active streamable HTTP transports by session ID
@@ -59,7 +61,7 @@ app.get("/sse", async (req, res) => {
   if (!key) return;
 
   const transport = new SSEServerTransport("/messages", res);
-  sseTransports[transport.sessionId] = transport;
+  sseTransports[transport.sessionId] = { transport, key };
 
   res.on("close", () => {
     delete sseTransports[transport.sessionId];
@@ -73,9 +75,11 @@ app.get("/sse", async (req, res) => {
 
 app.post("/messages", async (req, res) => {
   const sessionId = req.query.sessionId;
-  const transport = sseTransports[sessionId];
-  if (transport) {
-    await transport.handlePostMessage(req, res);
+  const session = sseTransports[sessionId];
+  if (session) {
+    await connectionStorage.run({ key: session.key, spend: 0 }, async () => {
+      await session.transport.handlePostMessage(req, res);
+    });
   } else {
     res.status(400).send("No active session found.");
   }
@@ -83,10 +87,19 @@ app.post("/messages", async (req, res) => {
 
 // ── HTTP Streamable transport (recommended) ───────────────────────────────────
 
-app.all("/mcp", async (req, res) => {
+// POST /mcp — initialize new session or handle existing session message
+app.post("/mcp", async (req, res) => {
   const sessionId = req.headers["mcp-session-id"];
 
-  if (!sessionId) {
+  if (sessionId) {
+    // Existing session — route to stored transport
+    const session = streamableTransports[sessionId];
+    if (!session) return res.status(404).json({ jsonrpc: "2.0", error: { code: -32000, message: "Session not found" }, id: null });
+
+    await connectionStorage.run({ key: session.key, spend: 0 }, async () => {
+      await session.transport.handleRequest(req, res, req.body);
+    });
+  } else if (isInitializeRequest(req.body)) {
     // New session — extract and validate key
     const key = extractKey(req, res, "/mcp");
     if (!key) return;
@@ -98,26 +111,42 @@ app.all("/mcp", async (req, res) => {
       },
     });
 
-    res.on("close", () => {
+    transport.onclose = () => {
       if (transport.sessionId) delete streamableTransports[transport.sessionId];
-    });
+    };
 
     await connectionStorage.run({ key, spend: 0 }, async () => {
       const server = createServer();
       await server.connect(transport);
-      await transport.handleRequest(req, res);
+      await transport.handleRequest(req, res, req.body);
     });
   } else {
-    // Existing session — look up stored transport + key
-    const session = streamableTransports[sessionId];
-    if (!session) return res.status(404).send("Session not found.");
-
-    await connectionStorage.run({ key: session.key, spend: 0 }, async () => {
-      await session.transport.handleRequest(req, res);
-    });
+    res.status(400).json({ jsonrpc: "2.0", error: { code: -32000, message: "No valid session ID" }, id: null });
   }
 });
 
+// GET /mcp — SSE stream for server-initiated messages (existing sessions only)
+app.get("/mcp", async (req, res) => {
+  const sessionId = req.headers["mcp-session-id"];
+  const session = streamableTransports[sessionId];
+  if (!session) return res.status(400).send("Invalid or missing session ID");
+
+  await connectionStorage.run({ key: session.key, spend: 0 }, async () => {
+    await session.transport.handleRequest(req, res);
+  });
+});
+
+// DELETE /mcp — session termination
+app.delete("/mcp", async (req, res) => {
+  const sessionId = req.headers["mcp-session-id"];
+  const session = streamableTransports[sessionId];
+  if (!session) return res.status(400).send("Invalid or missing session ID");
+
+  await connectionStorage.run({ key: session.key, spend: 0 }, async () => {
+    await session.transport.handleRequest(req, res, req.body);
+  });
+});
+
 // ── Utility routes ────────────────────────────────────────────────────────────
 
 app.post("/generate-wallet", (_req, res) => {