@agentvalet/register 0.4.10 → 0.4.11

package/dist/index.js

@@ -1619,6 +1619,7 @@ function cleanLocally() {
   }
   console.log(chalk11.green("  \u2713 Cleaned CLAUDE.md, .mcp.json, .claude/settings.local.json"));
 }
+var SSE_READ_TIMEOUT_MS = 25e3;
 async function pollForApproval(apiUrl, revocationToken, timeoutMs) {
   const streamUrl = `${apiUrl}/v1/revoke-request/stream/${revocationToken}`;
   const deadline = Date.now() + timeoutMs;
@@ -1631,47 +1632,64 @@ async function pollForApproval(apiUrl, revocationToken, timeoutMs) {
       }
     }
     function connect() {
-      if (Date.now() > deadline) {
+      if (settled || Date.now() > deadline) {
         done("timeout");
         return;
       }
-      fetch(streamUrl, { headers: { Accept: "text/event-stream" } }).then(async (res) => {
+      const controller = new AbortController();
+      let readTimer = setTimeout(() => controller.abort(), SSE_READ_TIMEOUT_MS);
+      function resetReadTimer() {
+        clearTimeout(readTimer);
+        readTimer = setTimeout(() => controller.abort(), SSE_READ_TIMEOUT_MS);
+      }
+      fetch(streamUrl, { signal: controller.signal, headers: { Accept: "text/event-stream" } }).then(async (res) => {
         if (!res.ok || !res.body) {
-          done("timeout");
+          clearTimeout(readTimer);
+          if (!settled) setTimeout(connect, 3e3);
           return;
         }
         const reader = res.body.getReader();
         const decoder = new TextDecoder();
         let buf = "";
-        while (true) {
-          const { done: streamDone, value } = await reader.read();
-          if (streamDone) break;
-          buf += decoder.decode(value, { stream: true });
-          const lines = buf.split("\n");
-          buf = lines.pop() ?? "";
-          for (const line of lines) {
-            if (!line.startsWith("data: ")) continue;
-            try {
-              const evt = JSON.parse(line.slice(6));
-              if (evt.status === "approved") {
-                done("approved");
-                return;
-              }
-              if (evt.status === "rejected") {
-                done("rejected");
-                return;
+        try {
+          while (true) {
+            const { done: streamDone, value } = await reader.read();
+            resetReadTimer();
+            if (streamDone) break;
+            buf += decoder.decode(value, { stream: true });
+            const lines = buf.split("\n");
+            buf = lines.pop() ?? "";
+            for (const line of lines) {
+              if (line.startsWith(":")) continue;
+              if (!line.startsWith("data: ")) continue;
+              try {
+                const evt = JSON.parse(line.slice(6));
+                if (evt.status === "approved") {
+                  clearTimeout(readTimer);
+                  done("approved");
+                  return;
+                }
+                if (evt.status === "rejected") {
+                  clearTimeout(readTimer);
+                  done("rejected");
+                  return;
+                }
+                if (evt.status === "timeout") {
+                  clearTimeout(readTimer);
+                  done("timeout");
+                  return;
+                }
+              } catch {
               }
-              if (evt.status === "timeout") {
-                done("timeout");
-                return;
-              }
-            } catch {
             }
           }
+        } finally {
+          clearTimeout(readTimer);
         }
         if (!settled) setTimeout(connect, 1e3);
       }).catch(() => {
-        if (!settled) setTimeout(connect, 2e3);
+        clearTimeout(readTimer);
+        if (!settled) setTimeout(connect, 1e3);
       });
     }
     connect();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@agentvalet/register",
-  "version": "0.4.10",
+  "version": "0.4.11",
   "description": "Self-register AI agents with AgentValet",
   "type": "module",
   "bin": {