@yunazgr/pi-companion 0.2.3 → 0.2.4

package/README.md

@@ -167,6 +167,8 @@ Paired devices (name, browser user agent, pairing and last-seen times, credentia
 
 A phone pairs once with the daemon. Individual Pi sessions still opt in using /companion.
 
+Network loss and daemon restarts reconnect automatically with bounded backoff. Heartbeats detect silent connections, and returning to the foreground or restoring network access retries immediately. Pairing credentials and unsent drafts survive outages; session snapshots, pending questions, and previously loaded upload lists refresh on reconnect. Missed activity is not replayed, and prompts/answers are never automatically resent. A deliberate **Disconnect** waits for the device’s explicit retry; **Revoke** removes access and requires pairing again.
+
 ## Settings
 
 The **Settings** page (local console only) covers:

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@yunazgr/pi-companion",
-  "version": "0.2.3",
+  "version": "0.2.4",
   "description": "Local-first remote control for Pi sessions: live activity feed, steering, git diff, file drop and phone pairing from a single Rust daemon.",
   "keywords": [
     "pi-package",
@@ -71,7 +71,7 @@
     "highlight.js": "^11.12.0",
     "jsqr": "^1.4.0",
     "marked": "^18.1.0",
-    "mermaid": "^12.1.0",
+    "mermaid": "^10.9.8",
     "svelte": "^5.57.2",
     "svelte-check": "^4.7.6",
     "typebox": "^1.0.13",
@@ -79,7 +79,10 @@
     "vite": "^8.0.12"
   },
   "overrides": {
-    "node-domexception": "file:./tools/shims/node-domexception"
+    "node-domexception": "file:./tools/shims/node-domexception",
+    "mermaid": {
+      "katex": "^0.18.2"
+    }
   },
   "allowScripts": {
     "esbuild": true,

package/src/bridge.ts

@@ -16,6 +16,7 @@ export class CompanionBridge implements AskChannel {
   private ws?: WebSocket;
   private ctx?: ExtensionContext;
   private reconnect?: NodeJS.Timeout;
+  private heartbeat?: NodeJS.Timeout;
   private closed = false;
   private activated = false;
   private restoreDialogs?: () => void;
@@ -80,6 +81,7 @@ export class CompanionBridge implements AskChannel {
     this.pendingDeletes.clear();
     if (this.reconnect) clearTimeout(this.reconnect);
     this.reconnect = undefined;
+    if (this.heartbeat) clearTimeout(this.heartbeat);
     const ws = this.ws;
     this.ws = undefined;
     ws?.close();
@@ -115,6 +117,9 @@ export class CompanionBridge implements AskChannel {
       if (allowSpawn) await ensureDaemon((message, level = "info") => this.log(message, level));
       if (this.closed || !this.snapshot.remoteEnabled) return;
       this.open();
+    } catch (error) {
+      this.log("Could not reconnect to Pi Companion: " + String(error), "warning");
+      this.scheduleReconnect();
     } finally {
       this.connecting = false;
     }
@@ -122,10 +127,18 @@ export class CompanionBridge implements AskChannel {
 
   private open() {
     const base = process.env.PI_COMPANION_URL ?? "ws://127.0.0.1:43721";
-    const ws = new WebSocket(base.replace(/\/$/, "") + "/ws/bridge/" + this.sessionId);
+    const ws = new WebSocket(base.replace(/\/$/, "") + "/ws/bridge/" + this.sessionId, { handshakeTimeout: 15_000 });
     this.ws = ws;
+    const alive = () => {
+      if (this.ws !== ws || this.closed) return;
+      if (this.heartbeat) clearTimeout(this.heartbeat);
+      this.heartbeat = setTimeout(() => ws.terminate(), 45_000);
+      this.heartbeat.unref();
+    };
+    ws.on("ping", alive); // ws automatically pongs; the watchdog detects silent loss.
     ws.on("open", () => {
-      if (this.ws !== ws || this.closed || !this.snapshot.remoteEnabled) { ws.close(); return; }
+      if (this.ws !== ws || this.closed || !this.snapshot.remoteEnabled) return ws.terminate();
+      alive();
       this.everConnected = true;
       this.downSince = 0;
       this.retryDelay = 1000;
@@ -133,11 +146,14 @@ export class CompanionBridge implements AskChannel {
     });
     ws.on("message", raw => {
       if (this.ws !== ws || this.closed || !this.snapshot.remoteEnabled) return;
-      try { void this.handle(JSON.parse(raw.toString()) as ServerMessage); }
-      catch (error) { this.send({ type: "error", message: "Invalid server message: " + String(error) }); }
+      alive();
+      const failure = (error: unknown) => this.send({ type: "error", message: "Invalid server message: " + String(error) });
+      try { void this.handle(JSON.parse(raw.toString()) as ServerMessage).catch(failure); }
+      catch (error) { failure(error); }
     });
     ws.on("close", () => {
       if (this.ws !== ws) return;
+      if (this.heartbeat) clearTimeout(this.heartbeat);
       if (!this.downSince) this.downSince = Date.now();
       this.scheduleReconnect();
     });
@@ -156,6 +172,7 @@ export class CompanionBridge implements AskChannel {
     this.restoreDialogs = undefined;
     for (const entry of [...this.asks.values()]) entry.settle(null);
     if (this.reconnect) clearTimeout(this.reconnect);
+    if (this.heartbeat) clearTimeout(this.heartbeat);
     this.ws?.close();
   }