@mcp-ts/sdk 1.3.1 → 1.3.3

package/dist/server/index.mjs

@@ -1002,18 +1002,15 @@ var MCPClient = class _MCPClient {
     __publicField(this, "serverUrl");
     __publicField(this, "callbackUrl");
     __publicField(this, "onRedirect");
-    __publicField(this, "tokens");
-    __publicField(this, "tokenExpiresAt");
-    __publicField(this, "clientInformation");
     __publicField(this, "clientId");
     __publicField(this, "clientSecret");
-    __publicField(this, "onSaveTokens");
     __publicField(this, "headers");
     /** OAuth Client Metadata */
     __publicField(this, "clientName");
     __publicField(this, "clientUri");
     __publicField(this, "logoUri");
     __publicField(this, "policyUri");
+    __publicField(this, "createdAt");
     /** Event emitters for connection lifecycle */
     __publicField(this, "_onConnectionEvent", new Emitter());
     __publicField(this, "onConnectionEvent", this._onConnectionEvent.event);
@@ -1028,12 +1025,8 @@ var MCPClient = class _MCPClient {
     this.serverId = options.serverId;
     this.sessionId = options.sessionId;
     this.transportType = options.transportType;
-    this.tokens = options.tokens;
-    this.tokenExpiresAt = options.tokenExpiresAt;
-    this.clientInformation = options.clientInformation;
     this.clientId = options.clientId;
     this.clientSecret = options.clientSecret;
-    this.onSaveTokens = options.onSaveTokens;
     this.headers = options.headers;
     this.clientName = options.clientName;
     this.clientUri = options.clientUri;
@@ -1053,6 +1046,8 @@ var MCPClient = class _MCPClient {
       sessionId: this.sessionId,
       serverId: this.serverId,
       serverName: this.serverName || this.serverId,
+      serverUrl: this.serverUrl || "",
+      createdAt: this.createdAt,
       state: newState,
       previousState,
       timestamp: Date.now()
@@ -1173,6 +1168,7 @@ var MCPClient = class _MCPClient {
       this.serverName = this.serverName || sessionData.serverName;
       this.serverId = this.serverId || sessionData.serverId || "unknown";
       this.headers = this.headers || sessionData.headers;
+      this.createdAt = sessionData.createdAt;
     }
     if (!this.serverUrl || !this.callbackUrl || !this.serverId) {
       throw new Error("Missing required connection metadata");
@@ -1226,6 +1222,7 @@ var MCPClient = class _MCPClient {
     }
     const existingSession = await storage.getSession(this.identity, this.sessionId);
     if (!existingSession && this.serverId && this.serverUrl && this.callbackUrl) {
+      this.createdAt = Date.now();
       console.log(`[MCPClient] Creating initial session ${this.sessionId} for OAuth flow`);
       await storage.createSession({
         sessionId: this.sessionId,
@@ -1235,7 +1232,8 @@ var MCPClient = class _MCPClient {
         serverUrl: this.serverUrl,
         callbackUrl: this.callbackUrl,
         transportType: this.transportType || "streamable_http",
-        createdAt: Date.now()
+        createdAt: this.createdAt,
+        active: false
       }, Math.floor(STATE_EXPIRATION_MS / 1e3));
     }
   }
@@ -1243,9 +1241,10 @@ var MCPClient = class _MCPClient {
    * Saves current session state to storage
    * Creates new session if it doesn't exist, updates if it does
    * @param ttl - Time-to-live in seconds (defaults to 12hr for connected sessions)
+   * @param active - Session status marker used to avoid unnecessary TTL rewrites
    * @private
    */
-  async saveSession(ttl = SESSION_TTL_SECONDS) {
+  async saveSession(ttl = SESSION_TTL_SECONDS, active = true) {
     if (!this.sessionId || !this.serverId || !this.serverUrl || !this.callbackUrl) {
       return;
     }
@@ -1257,7 +1256,8 @@ var MCPClient = class _MCPClient {
       serverUrl: this.serverUrl,
       callbackUrl: this.callbackUrl,
       transportType: this.transportType || "streamable_http",
-      createdAt: Date.now()
+      createdAt: this.createdAt || Date.now(),
+      active
     };
     const existingSession = await storage.getSession(this.identity, this.sessionId);
     if (existingSession) {
@@ -1331,15 +1331,17 @@ var MCPClient = class _MCPClient {
       this.emitStateChange("CONNECTED");
       this.emitProgress("Connected successfully");
       const existingSession = await storage.getSession(this.identity, this.sessionId);
-      if (!existingSession || existingSession.transportType !== this.transportType) {
-        console.log(`[MCPClient] Saving session ${this.sessionId} (new or transport changed)`);
-        await this.saveSession(SESSION_TTL_SECONDS);
+      const needsTransportUpdate = !existingSession || existingSession.transportType !== this.transportType;
+      const needsTtlPromotion = !existingSession || existingSession.active !== true;
+      if (needsTransportUpdate || needsTtlPromotion) {
+        console.log(`[MCPClient] Saving session ${this.sessionId} with 12hr TTL (connect success)`);
+        await this.saveSession(SESSION_TTL_SECONDS, true);
       }
     } catch (error) {
       if (error instanceof UnauthorizedError$1 || error instanceof Error && error.message.toLowerCase().includes("unauthorized")) {
         this.emitStateChange("AUTHENTICATING");
         console.log(`[MCPClient] Saving session ${this.sessionId} with 10min TTL (OAuth pending)`);
-        await this.saveSession(Math.floor(STATE_EXPIRATION_MS / 1e3));
+        await this.saveSession(Math.floor(STATE_EXPIRATION_MS / 1e3), false);
         let authUrl = "";
         if (this.oauthProvider) {
           authUrl = this.oauthProvider.authUrl || "";
@@ -1417,7 +1419,7 @@ var MCPClient = class _MCPClient {
         await this.client.connect(this.transport);
         this.emitStateChange("CONNECTED");
         console.log(`[MCPClient] Updating session ${this.sessionId} to 12hr TTL (OAuth complete)`);
-        await this.saveSession(SESSION_TTL_SECONDS);
+        await this.saveSession(SESSION_TTL_SECONDS, true);
         return;
       } catch (error) {
         lastError = error;
@@ -2081,7 +2083,9 @@ var SSEConnectionManager = class {
         serverId: s.serverId,
         serverName: s.serverName,
         serverUrl: s.serverUrl,
-        transport: s.transportType
+        transport: s.transportType,
+        createdAt: s.createdAt,
+        active: s.active !== false
       }))
     };
   }
@@ -2096,6 +2100,13 @@ var SSEConnectionManager = class {
       (s) => s.serverId === serverId || s.serverUrl === serverUrl
     );
     if (duplicate) {
+      if (duplicate.active === false) {
+        await this.restoreSession({ sessionId: duplicate.sessionId });
+        return {
+          sessionId: duplicate.sessionId,
+          success: true
+        };
+      }
       throw new Error(`Connection already exists for server: ${duplicate.serverUrl || duplicate.serverId} (${duplicate.serverName})`);
     }
     const sessionId = await storage.generateSessionId();
@@ -2104,6 +2115,7 @@ var SSEConnectionManager = class {
       sessionId,
       serverId,
       serverName,
+      serverUrl,
       state: "CONNECTING",
       previousState: "DISCONNECTED",
       timestamp: Date.now()
@@ -2236,6 +2248,7 @@ var SSEConnectionManager = class {
       sessionId,
       serverId: session.serverId ?? "unknown",
       serverName: session.serverName ?? "Unknown",
+      serverUrl: session.serverUrl,
       state: "VALIDATING",
       previousState: "DISCONNECTED",
       timestamp: Date.now()
@@ -2287,6 +2300,7 @@ var SSEConnectionManager = class {
       sessionId,
       serverId: session.serverId ?? "unknown",
       serverName: session.serverName ?? "Unknown",
+      serverUrl: session.serverUrl,
       state: "AUTHENTICATING",
       previousState: "DISCONNECTED",
       timestamp: Date.now()
@@ -2418,7 +2432,9 @@ function writeSSEEvent(res, event, data) {
 }
 
 // src/server/handlers/nextjs-handler.ts
-var managers = /* @__PURE__ */ new Map();
+function isRpcResponseEvent(event) {
+  return "id" in event && ("result" in event || "error" in event);
+}
 function createNextMcpHandler(options = {}) {
   const {
     getIdentity = (request) => new URL(request.url).searchParams.get("identity"),
@@ -2431,72 +2447,29 @@ function createNextMcpHandler(options = {}) {
     clientDefaults,
     getClientMetadata
   } = options;
-  async function GET(request) {
-    const identity = getIdentity(request);
-    const authToken = getAuthToken(request);
-    if (!identity) {
-      return new Response("Missing identity", { status: 400 });
-    }
-    const isAuthorized = await authenticate(identity, authToken);
-    if (!isAuthorized) {
-      return new Response("Unauthorized", { status: 401 });
-    }
-    const stream = new TransformStream();
-    const writer = stream.writable.getWriter();
-    const encoder = new TextEncoder();
-    const sendSSE = (event, data) => {
-      const message = `event: ${event}
-data: ${JSON.stringify(data)}
-
-`;
-      writer.write(encoder.encode(message)).catch(() => {
-      });
-    };
-    const previousManager = managers.get(identity);
-    if (previousManager) {
-      previousManager.dispose();
-    }
-    const resolvedClientMetadata = getClientMetadata ? await getClientMetadata(request) : clientDefaults;
-    const manager = new SSEConnectionManager(
+  const toManagerOptions = (identity, resolvedClientMetadata) => ({
+    identity,
+    heartbeatInterval,
+    clientDefaults: resolvedClientMetadata
+  });
+  async function resolveClientMetadata(request) {
+    return getClientMetadata ? await getClientMetadata(request) : clientDefaults;
+  }
+  async function GET() {
+    return Response.json(
       {
-        identity,
-        heartbeatInterval,
-        clientDefaults: resolvedClientMetadata
-        // Pass resolved metadata
-      },
-      (event) => {
-        if ("id" in event) {
-          sendSSE("rpc-response", event);
-        } else if ("type" in event && "sessionId" in event) {
-          sendSSE("connection", event);
-        } else {
-          sendSSE("observability", event);
+        error: {
+          code: "METHOD_NOT_ALLOWED",
+          message: "Use POST /api/mcp. For streaming use Accept: text/event-stream."
         }
-      }
+      },
+      { status: 405 }
     );
-    managers.set(identity, manager);
-    sendSSE("connected", { timestamp: Date.now() });
-    const abortController = new AbortController();
-    request.signal?.addEventListener("abort", () => {
-      manager.dispose();
-      managers.delete(identity);
-      writer.close().catch(() => {
-      });
-      abortController.abort();
-    });
-    return new Response(stream.readable, {
-      status: 200,
-      headers: {
-        "Content-Type": "text/event-stream",
-        "Cache-Control": "no-cache, no-transform",
-        "Connection": "keep-alive",
-        "X-Accel-Buffering": "no"
-      }
-    });
   }
   async function POST(request) {
     const identity = getIdentity(request);
     const authToken = getAuthToken(request);
+    const acceptsEventStream = (request.headers.get("accept") || "").toLowerCase().includes("text/event-stream");
     if (!identity) {
       return Response.json({ error: { code: "MISSING_IDENTITY", message: "Missing identity" } }, { status: 400 });
     }
@@ -2504,28 +2477,103 @@ data: ${JSON.stringify(data)}
     if (!isAuthorized) {
       return Response.json({ error: { code: "UNAUTHORIZED", message: "Unauthorized" } }, { status: 401 });
     }
+    let rawBody = "";
     try {
-      const body = await request.json();
-      const manager = managers.get(identity);
-      if (!manager) {
+      rawBody = await request.text();
+      const body = rawBody ? JSON.parse(rawBody) : null;
+      if (!body || typeof body !== "object") {
         return Response.json(
           {
             error: {
-              code: "NO_CONNECTION",
-              message: "No SSE connection found. Please establish SSE connection first."
+              code: "INVALID_REQUEST",
+              message: "Invalid JSON-RPC request body"
             }
           },
           { status: 400 }
         );
       }
-      const response = await manager.handleRequest(body);
-      return Response.json(response);
+      const resolvedClientMetadata = await resolveClientMetadata(request);
+      if (!acceptsEventStream) {
+        const manager2 = new SSEConnectionManager(
+          toManagerOptions(identity, resolvedClientMetadata),
+          () => {
+          }
+        );
+        try {
+          const response = await manager2.handleRequest(body);
+          return Response.json(response);
+        } finally {
+          manager2.dispose();
+        }
+      }
+      const stream = new TransformStream();
+      const writer = stream.writable.getWriter();
+      const encoder = new TextEncoder();
+      let streamWritable = true;
+      const sendSSE = (event, data) => {
+        if (!streamWritable) return;
+        const message = `event: ${event}
+data: ${JSON.stringify(data)}
+
+`;
+        writer.write(encoder.encode(message)).catch(() => {
+          streamWritable = false;
+        });
+      };
+      const manager = new SSEConnectionManager(
+        toManagerOptions(identity, resolvedClientMetadata),
+        (event) => {
+          if (isRpcResponseEvent(event)) {
+            sendSSE("rpc-response", event);
+          } else if ("type" in event && "sessionId" in event) {
+            sendSSE("connection", event);
+          } else {
+            sendSSE("observability", event);
+          }
+        }
+      );
+      sendSSE("connected", { timestamp: Date.now() });
+      void (async () => {
+        try {
+          await manager.handleRequest(body);
+        } catch (error) {
+          const err = error instanceof Error ? error : new Error("Unknown error");
+          sendSSE("rpc-response", {
+            id: body.id || "unknown",
+            error: {
+              code: "EXECUTION_ERROR",
+              message: err.message
+            }
+          });
+        } finally {
+          streamWritable = false;
+          manager.dispose();
+          writer.close().catch(() => {
+          });
+        }
+      })();
+      return new Response(stream.readable, {
+        status: 200,
+        headers: {
+          "Content-Type": "text/event-stream",
+          "Cache-Control": "no-cache, no-transform",
+          "Connection": "keep-alive",
+          "X-Accel-Buffering": "no"
+        }
+      });
     } catch (error) {
+      const err = error instanceof Error ? error : new Error("Unknown error");
+      console.error("[MCP Next Handler] Failed to handle RPC", {
+        identity,
+        message: err.message,
+        stack: err.stack,
+        rawBody: rawBody.slice(0, 500)
+      });
       return Response.json(
         {
           error: {
             code: "EXECUTION_ERROR",
-            message: error instanceof Error ? error.message : "Unknown error"
+            message: err.message
           }
         },
         { status: 500 }