@mcp-use/client 2.4.4 → 2.4.5-canary.1

package/dist/index.js

@@ -1659,7 +1659,7 @@ init_connector_telemetry();
 init_logging();
 
 // src/utils/version.ts
-var VERSION = "2.4.4";
+var VERSION = "2.4.5-canary.1";
 function getPackageVersion() {
   return VERSION;
 }
@@ -2438,7 +2438,11 @@ var OAuthSessionStore = class _OAuthSessionStore {
     return (await this.readCredential("tokens", ctx))?.value;
   }
   async saveTokens(tokens, ctx) {
-    const serialized = JSON.stringify(tokens);
+    const stored = typeof tokens.expires_in === "number" ? {
+      ...tokens,
+      mcp_use_expires_at: Date.now() + tokens.expires_in * 1e3
+    } : tokens;
+    const serialized = JSON.stringify(stored);
     await this.store.set(this.credentialKey("tokens", ctx), serialized);
     if (ctx) await this.store.set(this.credentialKey("tokens"), serialized);
     await this.store.remove(this.getKey("code_verifier"));
@@ -2999,30 +3003,38 @@ var NodeOAuthClientProvider = class _NodeOAuthClientProvider {
       throw err;
     }
   }
-  stopLoopback() {
+  stopLoopback(callbackSocket) {
     if (this.pendingTimer) {
       clearTimeout(this.pendingTimer);
       this.pendingTimer = null;
     }
     if (this.server) {
-      this.server.close();
+      const server = this.server;
       this.server = null;
+      server.close();
+      if (callbackSocket && !callbackSocket.destroyed) {
+        callbackSocket.once("close", () => server.closeAllConnections());
+      } else {
+        server.closeAllConnections();
+      }
     }
     this.authorizationUrl = null;
   }
-  resolvePending(response) {
+  resolvePending(response, callbackSocket) {
     const p = this.pending;
     this.pending = null;
-    this.stopLoopback();
+    this.stopLoopback(callbackSocket);
     p?.resolve(response);
   }
-  rejectPending(err) {
+  rejectPending(err, callbackSocket) {
     const p = this.pending;
     this.pending = null;
-    this.stopLoopback();
+    this.stopLoopback(callbackSocket);
     p?.reject(err);
   }
   handleCallback(rawUrl, res) {
+    const callbackSocket = res.socket ?? void 0;
+    res.setHeader("connection", "close");
     const url = new URL(rawUrl, `http://127.0.0.1:${this.port}`);
     if (url.pathname === "/authorize") {
       if (this.authorizationUrl === null || this.pending === null) {
@@ -3051,7 +3063,7 @@ var NodeOAuthClientProvider = class _NodeOAuthClientProvider {
       res.statusCode = 400;
       res.setHeader("content-type", "text/html; charset=utf-8");
       res.end(FAILURE_HTML(err, errDesc));
-      this.rejectPending(new OAuthFlowError(err, errDesc));
+      this.rejectPending(new OAuthFlowError(err, errDesc), callbackSocket);
       return;
     }
     if (!code || !state) {
@@ -3062,7 +3074,10 @@ var NodeOAuthClientProvider = class _NodeOAuthClientProvider {
     res.statusCode = 200;
     res.setHeader("content-type", "text/html; charset=utf-8");
     res.end(SUCCESS_HTML);
-    this.resolvePending({ code, ...iss !== void 0 ? { iss } : {} });
+    this.resolvePending(
+      { code, ...iss !== void 0 ? { iss } : {} },
+      callbackSocket
+    );
   }
 };
 async function defaultOpener(url) {