@bagelink/sdk 1.20.54 → 1.20.58

package/src/openAPITools/StreamController.js

@@ -0,0 +1,62 @@
+/**
+ * StreamController - Simple, reliable SSE stream management
+ * Provides a type-safe API for consuming Server-Sent Events
+ */
+/**
+ * StreamController - Simple, reliable event emitter for SSE streams
+ * @template TEventMap - Map of event names to their data types
+ *
+ * @example
+ * const stream = api.messages.stream(params)
+ *   .on('token', (data) => console.log(data))
+ *   .on('done', (data) => console.log('Complete!'))
+ *
+ * stream.close()
+ */
+export class StreamController {
+    cleanup;
+    handlers = new Map();
+    _closed = false;
+    constructor(cleanup) {
+        this.cleanup = cleanup;
+    }
+    /**
+     * Register an event handler (fully typed!)
+     * @param event - Event type to listen for
+     * @param handler - Handler function (data type inferred from event)
+     * @returns this (for chaining)
+     */
+    on(event, handler) {
+        if (!this.handlers.has(event)) {
+            this.handlers.set(event, new Set());
+        }
+        this.handlers.get(event).add(handler);
+        return this;
+    }
+    /**
+     * Emit an event to all registered handlers
+     * @internal
+     */
+    emit(event, data) {
+        const handlers = this.handlers.get(event);
+        if (handlers) {
+            handlers.forEach((handler) => {
+                try {
+                    handler(data);
+                }
+                catch (error) {
+                    console.error(`Error in ${event} handler:`, error);
+                }
+            });
+        }
+    }
+    /**
+     * Close the stream
+     */
+    close() {
+        if (!this._closed) {
+            this.cleanup();
+            this._closed = true;
+        }
+    }
+}