@mjasnikovs/pi-task 0.16.0 → 0.16.1

package/dist/remote/server.js

@@ -109,14 +109,38 @@ export async function startServer(onMessage, getHtml) {
         }
     });
     const wss = new WebSocketServer({ server: httpServer, path: '/ws' });
+    // Track every accepted TCP socket so stop() can forcibly destroy lingering
+    // keep-alive / WebSocket connections. Without this, httpServer.close() only
+    // stops accepting new connections and waits for existing ones to drain — an
+    // open browser tab never drains, so the listening server and its sockets
+    // stay as active event-loop handles forever. In headless print mode the host
+    // exits by natural event-loop drain (it sets process.exitCode and returns,
+    // with no process.exit()), so these lingering handles keep the pi process
+    // alive indefinitely — which blocks `docker stop` / OS shutdown until the
+    // SIGKILL grace timeout fires. A/B-proven: terminating clients + destroying
+    // sockets here makes the loop drain in ~2ms instead of never.
+    const sockets = new Set();
+    httpServer.on('connection', s => {
+        sockets.add(s);
+        s.on('close', () => sockets.delete(s));
+    });
     const handle = {
         port,
         ip,
         ips,
         onFirstConnect: null,
         stop() {
+            // Terminate WebSocket clients (immediate close, no drain), then
+            // destroy any remaining raw sockets, then close the servers so the
+            // event loop has no lingering handles holding the process open.
+            for (const ws of wss.clients)
+                ws.terminate();
             wss.close();
+            for (const s of sockets)
+                s.destroy();
+            sockets.clear();
             httpServer.close();
+            httpServer.closeAllConnections?.();
         }
     };
     wss.on('connection', ws => {

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@mjasnikovs/pi-task",
-    "version": "0.16.0",
+    "version": "0.16.1",
     "description": "Deterministic spec-orchestration for local models, with a bundled real-time remote web view and web/docs/fetch/worker subagent tools.",
     "type": "module",
     "main": "./dist/index.js",