@superatomai/sdk-node 0.0.29 → 0.0.30

package/dist/index.mjs

@@ -9737,6 +9737,7 @@ init_logger();
 var SDK_VERSION = "0.0.8";
 var DEFAULT_WS_URL = "wss://ws.superatom.ai/websocket";
 var SuperatomSDK = class {
+  // 3.5 minutes (PING_INTERVAL + 30s grace)
   constructor(config) {
     this.ws = null;
     this.messageHandlers = /* @__PURE__ */ new Map();
@@ -9747,6 +9748,12 @@ var SuperatomSDK = class {
     this.collections = {};
     this.components = [];
     this.tools = [];
+    // Heartbeat properties for keeping WebSocket connection alive
+    this.pingInterval = null;
+    this.lastPong = Date.now();
+    this.PING_INTERVAL_MS = 18e4;
+    // 3 minutes
+    this.PONG_TIMEOUT_MS = 21e4;
     if (config.logLevel) {
       logger.setLogLevel(config.logLevel);
     }
@@ -9860,6 +9867,7 @@ var SuperatomSDK = class {
           this.connected = true;
           this.reconnectAttempts = 0;
           logger.info("WebSocket connected successfully");
+          this.startHeartbeat();
           resolve();
         });
         this.ws.addEventListener("message", (event) => {
@@ -9888,6 +9896,9 @@ var SuperatomSDK = class {
       const message = IncomingMessageSchema.parse(parsed);
       logger.debug("Received message:", message.type);
       switch (message.type) {
+        case "PONG":
+          this.handlePong();
+          break;
         case "DATA_REQ":
           handleDataRequest(parsed, this.collections, (msg) => this.send(msg)).catch((error) => {
             logger.error("Failed to handle data request:", error);
@@ -10010,6 +10021,7 @@ var SuperatomSDK = class {
    * Disconnect from the WebSocket service
    */
   disconnect() {
+    this.stopHeartbeat();
     if (this.ws) {
       this.ws.close();
       this.ws = null;
@@ -10020,6 +10032,7 @@ var SuperatomSDK = class {
    * Cleanup and disconnect - stops reconnection attempts and closes the connection
    */
   async destroy() {
+    this.stopHeartbeat();
     this.maxReconnectAttempts = 0;
     this.reconnectAttempts = 0;
     this.messageHandlers.clear();
@@ -10061,6 +10074,59 @@ var SuperatomSDK = class {
       logger.error("Max reconnection attempts reached");
     }
   }
+  /**
+   * Start heartbeat to keep WebSocket connection alive
+   * Sends PING every 3 minutes to prevent idle timeout from cloud infrastructure
+   */
+  startHeartbeat() {
+    this.stopHeartbeat();
+    this.lastPong = Date.now();
+    this.pingInterval = setInterval(() => {
+      if (!this.ws || this.ws.readyState !== this.ws.OPEN) {
+        logger.warn("WebSocket not open during heartbeat check, stopping heartbeat");
+        this.stopHeartbeat();
+        return;
+      }
+      const timeSinceLastPong = Date.now() - this.lastPong;
+      if (timeSinceLastPong > this.PONG_TIMEOUT_MS) {
+        logger.warn(`No PONG received for ${timeSinceLastPong}ms, connection may be dead. Reconnecting...`);
+        this.stopHeartbeat();
+        this.ws.close();
+        return;
+      }
+      try {
+        const pingMessage = {
+          type: "PING",
+          from: { type: this.type },
+          payload: { timestamp: Date.now() }
+        };
+        this.ws.send(JSON.stringify(pingMessage));
+        logger.debug("Heartbeat PING sent");
+      } catch (error) {
+        logger.error("Failed to send PING:", error);
+        this.stopHeartbeat();
+        this.ws.close();
+      }
+    }, this.PING_INTERVAL_MS);
+    logger.info(`Heartbeat started (PING every ${this.PING_INTERVAL_MS / 1e3}s)`);
+  }
+  /**
+   * Stop the heartbeat interval
+   */
+  stopHeartbeat() {
+    if (this.pingInterval) {
+      clearInterval(this.pingInterval);
+      this.pingInterval = null;
+      logger.debug("Heartbeat stopped");
+    }
+  }
+  /**
+   * Handle PONG response from server
+   */
+  handlePong() {
+    this.lastPong = Date.now();
+    logger.debug("Heartbeat PONG received");
+  }
   storeComponents(components) {
     this.components = components;
   }