@attson/atwebpilot-mcp 0.0.23 → 0.0.25

package/dist/index.js

@@ -771,10 +771,12 @@ var Coordinator = class {
 // src/loopback-ws-hub.ts
 import { WebSocketServer, WebSocket } from "ws";
 var HEARTBEAT_INTERVAL_MS = 2e4;
+var DEFAULT_KEEPALIVE_INTERVAL_MS = 15e3;
 var LoopbackWSHub = class {
   constructor(opts) {
     this.opts = opts;
     this.execTimeoutMs = opts.execTimeoutMs ?? 3e4;
+    this.keepaliveIntervalMs = opts.keepaliveIntervalMs ?? DEFAULT_KEEPALIVE_INTERVAL_MS;
     this.wss = new WebSocketServer({
       port: opts.port,
       path: "/worker",
@@ -790,6 +792,8 @@ var LoopbackWSHub = class {
   msgHandlers = [];
   disconnectHandlers = [];
   execTimeoutMs;
+  keepaliveIntervalMs;
+  keepaliveTimers = /* @__PURE__ */ new Map();
   ready() {
     return new Promise((resolve) => {
       const addr = this.wss.address();
@@ -806,6 +810,8 @@ var LoopbackWSHub = class {
       p.reject(new Error("hub closing"));
     }
     this.pending.clear();
+    for (const timer of this.keepaliveTimers.values()) clearInterval(timer);
+    this.keepaliveTimers.clear();
     for (const socket of this.wss.clients) {
       socket.terminate();
     }
@@ -859,6 +865,7 @@ var LoopbackWSHub = class {
         server_time: this.opts.clock.now(),
         heartbeat_interval_ms: HEARTBEAT_INTERVAL_MS
       });
+      this.startKeepalive(socket);
       for (const h of this.msgHandlers) h(msg.worker_id, msg);
       return;
     }
@@ -886,6 +893,7 @@ var LoopbackWSHub = class {
     }
   }
   onSocketClose(socket) {
+    this.stopKeepalive(socket);
     const wid = this.workerOf.get(socket);
     if (!wid) return;
     this.workerOf.delete(socket);
@@ -899,6 +907,30 @@ var LoopbackWSHub = class {
     }
     for (const h of this.disconnectHandlers) h(wid);
   }
+  startKeepalive(socket) {
+    if (this.keepaliveIntervalMs <= 0) return;
+    if (this.keepaliveTimers.has(socket)) return;
+    const timer = setInterval(() => {
+      if (socket.readyState !== WebSocket.OPEN) return;
+      this.rawSend(socket, {
+        type: "PONG",
+        nonce: this.opts.idGen.next("nonce"),
+        ts: this.opts.clock.now(),
+        protocol_version: PROTOCOL_VERSION,
+        echo_nonce: "server-keepalive"
+      });
+    }, this.keepaliveIntervalMs);
+    if (typeof timer.unref === "function") {
+      timer.unref();
+    }
+    this.keepaliveTimers.set(socket, timer);
+  }
+  stopKeepalive(socket) {
+    const timer = this.keepaliveTimers.get(socket);
+    if (!timer) return;
+    clearInterval(timer);
+    this.keepaliveTimers.delete(socket);
+  }
   rawSend(socket, msg) {
     const r = ServerToClientSchema.safeParse(msg);
     if (!r.success) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@attson/atwebpilot-mcp",
-  "version": "0.0.23",
+  "version": "0.0.25",
   "license": "Apache-2.0",
   "description": "MCP server for the atwebpilot browser extension — let Claude Code drive your browser.",
   "homepage": "https://github.com/attson/atwebpilot#readme",