@inferencesh/sdk 0.1.0

package/dist/stream.js

@@ -0,0 +1,120 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.StreamManager = void 0;
+class StreamManager {
+    constructor(options) {
+        this.eventSource = null;
+        this.reconnectTimeout = null;
+        this.stopTimeout = null;
+        this.initialConnectionAttempts = 0;
+        this.isConnected = false;
+        this.isStopped = false;
+        this.options = {
+            autoReconnect: true,
+            maxReconnects: 5,
+            reconnectDelayMs: 1000,
+            ...options,
+        };
+    }
+    clearTimeouts() {
+        if (this.reconnectTimeout) {
+            clearTimeout(this.reconnectTimeout);
+            this.reconnectTimeout = null;
+        }
+        if (this.stopTimeout) {
+            clearTimeout(this.stopTimeout);
+            this.stopTimeout = null;
+        }
+    }
+    closeEventSource() {
+        if (this.eventSource) {
+            this.eventSource.close();
+            this.eventSource = null;
+        }
+    }
+    cleanup() {
+        this.clearTimeouts();
+        this.closeEventSource();
+        this.isConnected = false;
+        this.options.onStop?.();
+    }
+    stopAfter(delayMs) {
+        this.clearTimeouts();
+        this.stopTimeout = setTimeout(() => this.stop(), delayMs);
+    }
+    clearStopTimeout() {
+        if (this.stopTimeout) {
+            clearTimeout(this.stopTimeout);
+            this.stopTimeout = null;
+        }
+    }
+    stop() {
+        this.isStopped = true;
+        this.cleanup();
+    }
+    scheduleReconnect() {
+        if (!this.options.autoReconnect || this.isStopped)
+            return false;
+        // If we had a successful connection before, always reconnect
+        // Otherwise, check if we've exceeded initial connection attempts
+        if (!this.isConnected && this.initialConnectionAttempts >= (this.options.maxReconnects ?? 5)) {
+            return false;
+        }
+        this.reconnectTimeout = setTimeout(() => {
+            if (!this.isStopped) {
+                if (!this.isConnected) {
+                    this.initialConnectionAttempts++;
+                }
+                this.connect();
+            }
+        }, this.options.reconnectDelayMs);
+        return true;
+    }
+    async connect() {
+        if (this.isStopped)
+            return;
+        this.cleanup();
+        this.isStopped = false;
+        try {
+            const source = await this.options.createEventSource();
+            if (!source || this.isStopped)
+                return;
+            this.eventSource = source;
+            this.isConnected = true;
+            this.options.onStart?.();
+            source.onmessage = (e) => {
+                if (this.isStopped)
+                    return;
+                try {
+                    const parsed = JSON.parse(e.data);
+                    this.options.onData?.(parsed);
+                    this.options.onYield?.(parsed);
+                }
+                catch (err) {
+                    const error = err instanceof Error ? err : new Error("Invalid JSON");
+                    this.options.onError?.(error);
+                }
+            };
+            source.onerror = (evt) => {
+                if (this.isStopped)
+                    return;
+                const error = new Error("Event stream error: " + evt);
+                this.options.onError?.(error);
+                this.cleanup();
+                if (!this.scheduleReconnect()) {
+                    this.stop();
+                }
+            };
+        }
+        catch (err) {
+            if (this.isStopped)
+                return;
+            const error = err instanceof Error ? err : new Error("createEventSource failed");
+            this.options.onError?.(error);
+            if (!this.scheduleReconnect()) {
+                this.stop();
+            }
+        }
+    }
+}
+exports.StreamManager = StreamManager;