@teamlearners/clawops 0.5.4 → 0.5.5

package/dist/agent/index.js

@@ -335,6 +335,7 @@ function createPipelineLogger(parent) {
 // src/agent/control-ws.ts
 var INITIAL_RECONNECT_DELAY = 1e3;
 var MAX_RECONNECT_DELAY = 3e4;
+var PING_TIMEOUT = 6e4;
 function buildControlWsUrl(options) {
   const scheme = options.baseUrl.startsWith("https") ? "wss" : "ws";
   const host = options.baseUrl.replace(/^https?:\/\//, "").replace(/\/$/, "");
@@ -360,6 +361,7 @@ var ControlWebSocket = class {
   _connectedResolve = null;
   _connectedPromise;
   _log = NOOP_LOGGER;
+  _pingTimer = null;
   setLogger(logger) {
     this._log = logger;
   }
@@ -390,6 +392,7 @@ var ControlWebSocket = class {
   /** Close the WebSocket and stop reconnecting. */
   close() {
     this._closed = true;
+    this._clearPingTimer();
     if (this._ws) {
       this._ws.close();
       this._ws = null;
@@ -406,12 +409,16 @@ var ControlWebSocket = class {
     this._ws = ws;
     ws.on("open", () => {
       this._reconnectDelay = INITIAL_RECONNECT_DELAY;
+      this._resetPingTimer();
       if (this._connectedResolve) {
         this._connectedResolve();
         this._connectedResolve = null;
       }
       this._log.info("Control WS connected: %s", this._url);
     });
+    ws.on("ping", () => {
+      this._resetPingTimer();
+    });
     ws.on("message", (data) => {
       try {
         const msg = JSON.parse(data.toString());
@@ -421,6 +428,7 @@ var ControlWebSocket = class {
       }
     });
     ws.on("close", () => {
+      this._clearPingTimer();
       if (!this._closed) {
         this._scheduleReconnect();
       }
@@ -446,6 +454,21 @@ var ControlWebSocket = class {
       }
     }
   }
+  _resetPingTimer() {
+    this._clearPingTimer();
+    this._pingTimer = setTimeout(() => {
+      this._log.warn("Control WS ping timeout, closing connection");
+      if (this._ws) {
+        this._ws.terminate();
+      }
+    }, PING_TIMEOUT);
+  }
+  _clearPingTimer() {
+    if (this._pingTimer) {
+      clearTimeout(this._pingTimer);
+      this._pingTimer = null;
+    }
+  }
   _scheduleReconnect() {
     const delay = this._reconnectDelay;
     this._reconnectDelay = Math.min(this._reconnectDelay * 2, MAX_RECONNECT_DELAY);