vibemon 1.5.1 → 1.6.0

package/modules/validators.cjs

@@ -10,9 +10,9 @@ const TOOL_MAX_LENGTH = 50;
 const MODEL_MAX_LENGTH = 50;
 const TERMINAL_ID_MAX_LENGTH = 100;
 // Memory is now a number (0-100), not a string
-// iTerm2: w0t0p0:UUID format, Ghostty: numeric PID
-const ITERM2_SESSION_PATTERN = /^w\d+t\d+p\d+:[0-9A-Fa-f-]{36}$/;
-const GHOSTTY_PID_PATTERN = /^\d{1,10}$/;
+// iTerm2: iterm2:w0t0p0:UUID format, Ghostty: ghostty:PID format
+const ITERM2_SESSION_PATTERN = /^iterm2:w\d+t\d+p\d+:[0-9A-Fa-f-]{36}$/;
+const GHOSTTY_PID_PATTERN = /^ghostty:\d{1,10}$/;
 
 /**
  * Validate state value

package/modules/ws-client.cjs

@@ -12,6 +12,9 @@ const RECONNECT_INITIAL_DELAY = 5000;   // 5 seconds
 const RECONNECT_MAX_DELAY = 30000;      // 30 seconds
 const RECONNECT_MULTIPLIER = 1.5;
 
+// Heartbeat configuration
+const PING_INTERVAL = 30000;            // Ping every 30 seconds
+
 class WsClient {
   constructor() {
     this.ws = null;
@@ -21,6 +24,8 @@ class WsClient {
     this.isConnecting = false;
     this.isConnected = false;
     this.shouldReconnect = true;
+    this.pingTimer = null;
+    this.pongReceived = true;
 
     // Persistent storage for token
     this.store = new Store({
@@ -158,9 +163,14 @@ class WsClient {
         this.isConnecting = false;
         this.isConnected = true;
         this.reconnectDelay = RECONNECT_INITIAL_DELAY;
+        this.startHeartbeat();
         this.notifyConnectionChange();
       });
 
+      this.ws.on('pong', () => {
+        this.pongReceived = true;
+      });
+
       this.ws.on('message', (data) => {
         this.handleMessage(data);
       });
@@ -236,6 +246,38 @@ class WsClient {
     }
   }
 
+  /**
+   * Start heartbeat ping to detect stale connections
+   */
+  startHeartbeat() {
+    this.stopHeartbeat();
+    this.pongReceived = true;
+
+    this.pingTimer = setInterval(() => {
+      if (!this.ws || this.ws.readyState !== WebSocket.OPEN) return;
+
+      if (!this.pongReceived) {
+        // Server didn't respond to last ping - connection is stale
+        console.log('WebSocket heartbeat timeout, reconnecting...');
+        this.ws.terminate();
+        return;
+      }
+
+      this.pongReceived = false;
+      this.ws.ping();
+    }, PING_INTERVAL);
+  }
+
+  /**
+   * Stop heartbeat ping
+   */
+  stopHeartbeat() {
+    if (this.pingTimer) {
+      clearInterval(this.pingTimer);
+      this.pingTimer = null;
+    }
+  }
+
   /**
    * Handle disconnection and schedule reconnect
    */
@@ -243,6 +285,7 @@ class WsClient {
     this.isConnecting = false;
     this.isConnected = false;
     this.ws = null;
+    this.stopHeartbeat();
     this.notifyConnectionChange();
 
     if (this.shouldReconnect) {
@@ -286,6 +329,7 @@ class WsClient {
    */
   disconnect() {
     this.shouldReconnect = false;
+    this.stopHeartbeat();
 
     if (this.reconnectTimer) {
       clearTimeout(this.reconnectTimer);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "vibemon",
-  "version": "1.5.1",
+  "version": "1.6.0",
   "description": "AI assistant status monitor",
   "main": "main.js",
   "bin": {