@discomedia/utils 1.0.54 → 1.0.55

package/dist/index.mjs

@@ -2517,7 +2517,7 @@ const safeJSON = (text) => {
 // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 const sleep = (ms) => new Promise((resolve) => setTimeout(resolve, ms));
 
-const VERSION = '6.15.0'; // x-release-please-version
+const VERSION = '6.16.0'; // x-release-please-version
 
 // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 const isRunningInBrowser = () => {
@@ -16901,6 +16901,11 @@ class AlpacaMarketDataAPI extends EventEmitter {
     optionWs = null;
     stockSubscriptions = { trades: [], quotes: [], bars: [] };
     optionSubscriptions = { trades: [], quotes: [], bars: [] };
+    autoReconnect = true;
+    stockReconnectAttempts = 0;
+    optionReconnectAttempts = 0;
+    stockReconnectTimeout = null;
+    optionReconnectTimeout = null;
     setMode(mode = 'production') {
         if (mode === 'sandbox') {
             // sandbox mode
@@ -16929,6 +16934,13 @@ class AlpacaMarketDataAPI extends EventEmitter {
             return 'production';
         }
     }
+    setAutoReconnect(enabled) {
+        this.autoReconnect = enabled;
+        log(`Auto-reconnect ${enabled ? 'enabled' : 'disabled'}`);
+    }
+    getAutoReconnect() {
+        return this.autoReconnect;
+    }
     constructor() {
         super();
         this.dataURL = 'https://data.alpaca.markets/v2';
@@ -16951,6 +16963,36 @@ class AlpacaMarketDataAPI extends EventEmitter {
         }
         return AlpacaMarketDataAPI.instance;
     }
+    getReconnectDelay(attempt) {
+        // 0: instant (0ms), 1: 5s, 2: 10s, 3: 15s, 4+: 60s
+        const delays = [0, 5000, 10000, 15000, 60000];
+        return attempt < delays.length ? delays[attempt] : 60000;
+    }
+    scheduleReconnect(streamType) {
+        if (!this.autoReconnect) {
+            log(`Auto-reconnect disabled, not reconnecting ${streamType} stream`);
+            return;
+        }
+        const attempts = streamType === 'stock' ? this.stockReconnectAttempts : this.optionReconnectAttempts;
+        const delay = this.getReconnectDelay(attempts);
+        log(`Scheduling ${streamType} stream reconnect (attempt ${attempts + 1}) in ${delay}ms`);
+        const timeout = setTimeout(() => {
+            log(`Attempting to reconnect ${streamType} stream (attempt ${attempts + 1})`);
+            if (streamType === 'stock') {
+                this.stockReconnectAttempts++;
+            }
+            else {
+                this.optionReconnectAttempts++;
+            }
+            this.connect(streamType);
+        }, delay);
+        if (streamType === 'stock') {
+            this.stockReconnectTimeout = timeout;
+        }
+        else {
+            this.optionReconnectTimeout = timeout;
+        }
+    }
     on(event, listener) {
         return super.on(event, listener);
     }
@@ -16968,6 +17010,21 @@ class AlpacaMarketDataAPI extends EventEmitter {
         }
         ws.on('open', () => {
             log(`${streamType} stream connected`);
+            // Reset reconnect attempts on successful connection
+            if (streamType === 'stock') {
+                this.stockReconnectAttempts = 0;
+                if (this.stockReconnectTimeout) {
+                    clearTimeout(this.stockReconnectTimeout);
+                    this.stockReconnectTimeout = null;
+                }
+            }
+            else {
+                this.optionReconnectAttempts = 0;
+                if (this.optionReconnectTimeout) {
+                    clearTimeout(this.optionReconnectTimeout);
+                    this.optionReconnectTimeout = null;
+                }
+            }
             const authMessage = {
                 action: 'auth',
                 key: process.env.ALPACA_API_KEY,
@@ -16992,20 +17049,42 @@ class AlpacaMarketDataAPI extends EventEmitter {
                 }
             }
         });
-        ws.on('close', () => {
-            log(`${streamType} stream disconnected`, { type: 'warn' });
+        ws.on('close', (code, reason) => {
+            const reasonStr = reason.toString() || 'No reason provided';
+            const codeDesc = this.getCloseCodeDescription(code);
+            log(`${streamType} stream disconnected - Code: ${code} (${codeDesc}), Reason: ${reasonStr}`, { type: 'warn' });
             if (streamType === 'stock') {
                 this.stockWs = null;
             }
             else {
                 this.optionWs = null;
             }
-            // Optional: implement reconnect logic
+            this.scheduleReconnect(streamType);
         });
         ws.on('error', (error) => {
             log(`${streamType} stream error: ${error.message}`, { type: 'error' });
         });
     }
+    getCloseCodeDescription(code) {
+        const codes = {
+            1000: 'Normal Closure',
+            1001: 'Going Away',
+            1002: 'Protocol Error',
+            1003: 'Unsupported Data',
+            1005: 'No Status Received',
+            1006: 'Abnormal Closure',
+            1007: 'Invalid Frame Payload Data',
+            1008: 'Policy Violation',
+            1009: 'Message Too Big',
+            1010: 'Mandatory Extension',
+            1011: 'Internal Server Error',
+            1012: 'Service Restart',
+            1013: 'Try Again Later',
+            1014: 'Bad Gateway',
+            1015: 'TLS Handshake',
+        };
+        return codes[code] || 'Unknown';
+    }
     sendSubscription(streamType) {
         const ws = streamType === 'stock' ? this.stockWs : this.optionWs;
         const subscriptions = streamType === 'stock' ? this.stockSubscriptions : this.optionSubscriptions;
@@ -17040,11 +17119,21 @@ class AlpacaMarketDataAPI extends EventEmitter {
         }
     }
     disconnectStockStream() {
+        if (this.stockReconnectTimeout) {
+            clearTimeout(this.stockReconnectTimeout);
+            this.stockReconnectTimeout = null;
+        }
+        this.stockReconnectAttempts = 0;
         if (this.stockWs) {
             this.stockWs.close();
         }
     }
     disconnectOptionStream() {
+        if (this.optionReconnectTimeout) {
+            clearTimeout(this.optionReconnectTimeout);
+            this.optionReconnectTimeout = null;
+        }
+        this.optionReconnectAttempts = 0;
         if (this.optionWs) {
             this.optionWs.close();
         }