agentfeed 0.1.1 → 0.1.3

package/dist/sse-client.js

@@ -1,60 +1,100 @@
+import http from "node:http";
+import https from "node:https";
 export function connectSSE(url, apiKey, onEvent, onError) {
     let aborted = false;
-    const controller = new AbortController();
-    const run = async () => {
-        while (!aborted) {
-            try {
-                const res = await fetch(url, {
-                    headers: { Authorization: `Bearer ${apiKey}` },
-                    signal: controller.signal,
-                });
-                if (!res.ok || !res.body) {
-                    throw new Error(`SSE connect failed: ${res.status}`);
-                }
-                const reader = res.body.getReader();
-                const decoder = new TextDecoder();
-                let buffer = "";
-                let currentEvent = { type: "message", data: "" };
-                while (!aborted) {
-                    const { done, value } = await reader.read();
-                    if (done)
-                        break;
-                    buffer += decoder.decode(value, { stream: true });
-                    const lines = buffer.split("\n");
-                    buffer = lines.pop() ?? "";
-                    for (const line of lines) {
-                        if (line.startsWith("event: ")) {
-                            currentEvent.type = line.slice(7);
-                        }
-                        else if (line.startsWith("data: ")) {
-                            currentEvent.data = line.slice(6);
-                        }
-                        else if (line.startsWith("id: ")) {
-                            currentEvent.id = line.slice(4);
-                        }
-                        else if (line === "") {
-                            if (currentEvent.data) {
-                                onEvent({ ...currentEvent });
-                            }
-                            currentEvent = { type: "message", data: "" };
+    let reconnecting = false;
+    let currentReq = null;
+    const connect = () => {
+        if (aborted)
+            return;
+        reconnecting = false;
+        const parsed = new URL(url);
+        const isHttps = parsed.protocol === "https:";
+        const options = {
+            hostname: parsed.hostname,
+            port: parsed.port || (isHttps ? 443 : 80),
+            path: parsed.pathname + parsed.search,
+            method: "GET",
+            headers: {
+                Authorization: `Bearer ${apiKey}`,
+                Accept: "text/event-stream",
+                "Cache-Control": "no-cache",
+                Connection: "keep-alive",
+            },
+            // Disable connection pooling to avoid agent timeouts
+            agent: false,
+        };
+        const mod = isHttps ? https : http;
+        const req = mod.request(options, (res) => {
+            if (res.statusCode !== 200) {
+                res.resume();
+                onError(new Error(`SSE connect failed: ${res.statusCode}`));
+                scheduleReconnect();
+                return;
+            }
+            // Keep TCP connection alive
+            res.socket?.setKeepAlive(true, 10000);
+            res.socket?.setTimeout(0);
+            res.setEncoding("utf-8");
+            let buffer = "";
+            let currentEvent = { type: "message", data: "" };
+            res.on("data", (chunk) => {
+                buffer += chunk;
+                const lines = buffer.split("\n");
+                buffer = lines.pop() ?? "";
+                for (const line of lines) {
+                    if (line.startsWith("event: ")) {
+                        currentEvent.type = line.slice(7);
+                    }
+                    else if (line.startsWith("data: ")) {
+                        currentEvent.data = line.slice(6);
+                    }
+                    else if (line.startsWith("id: ")) {
+                        currentEvent.id = line.slice(4);
+                    }
+                    else if (line === "") {
+                        if (currentEvent.data || currentEvent.type === "heartbeat") {
+                            onEvent({ ...currentEvent });
                         }
+                        currentEvent = { type: "message", data: "" };
                     }
                 }
+            });
+            res.on("end", () => {
+                if (!aborted) {
+                    console.log("SSE stream ended, reconnecting...");
+                    scheduleReconnect();
+                }
+            });
+            res.on("error", (err) => {
+                if (!aborted) {
+                    onError(err);
+                    scheduleReconnect();
+                }
+            });
+        });
+        req.on("error", (err) => {
+            if (!aborted) {
+                onError(err);
+                scheduleReconnect();
             }
-            catch (err) {
-                if (aborted)
-                    return;
-                onError(err instanceof Error ? err : new Error(String(err)));
-                // Reconnect after delay
-                await new Promise((r) => setTimeout(r, 5000));
-            }
-        }
+        });
+        // Disable request timeout
+        req.setTimeout(0);
+        req.end();
+        currentReq = req;
+    };
+    const scheduleReconnect = () => {
+        if (aborted || reconnecting)
+            return;
+        reconnecting = true;
+        setTimeout(connect, 5000);
     };
-    run();
+    connect();
     return {
         close: () => {
             aborted = true;
-            controller.abort();
+            currentReq?.destroy();
         },
     };
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "agentfeed",
-  "version": "0.1.1",
+  "version": "0.1.3",
   "description": "Worker daemon for AgentFeed - watches feeds and wakes AI agents via claude -p",
   "type": "module",
   "bin": {