@proagentstore/cli 0.2.9 → 0.2.10

package/dist/index.js

@@ -932,20 +932,16 @@ async function connectViaRelay(instanceIds, localUrl, runnerToken, opts) {
   writeLine(`  Agents:   ${instanceIds.length} instance${instanceIds.length === 1 ? "" : "s"}`);
   writeLine("  No cloudflared needed. Ctrl+C to disconnect.");
   writeLine("\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550");
-  let stopped = false;
-  const heartbeatLoop = async () => {
-    while (!stopped) {
-      await new Promise((r) => setTimeout(r, 3e4));
-      if (stopped) break;
+  const heartbeat = () => {
+    const timer = setTimeout(async () => {
       for (const id of instanceIds) {
         await requestPags("POST", `/v1/instances/${apiPathSegment(id)}/runtime/heartbeat`, opts, {}).catch(() => void 0);
       }
-    }
+      heartbeat();
+    }, 3e4);
+    timer.unref();
   };
-  void heartbeatLoop();
-  process.once("beforeExit", () => {
-    stopped = true;
-  });
+  heartbeat();
 }
 function openRelaySocket(instanceId, wsBase, pagsToken, localUrl, runnerToken) {
   let backoffMs = 1e3;
@@ -959,6 +955,13 @@ function openRelaySocket(instanceId, wsBase, pagsToken, localUrl, runnerToken) {
     };
     ws.onmessage = async (event) => {
       const text = typeof event.data === "string" ? event.data : String(event.data);
+      if (text === "ping") {
+        try {
+          ws.send("pong");
+        } catch {
+        }
+        return;
+      }
       let cmd;
       try {
         cmd = JSON.parse(text);
@@ -978,16 +981,29 @@ function openRelaySocket(instanceId, wsBase, pagsToken, localUrl, runnerToken) {
           headers,
           body: hasBody ? JSON.stringify(cmd.body) : void 0
         });
-        const result = await res.json().catch(() => ({}));
-        ws.send(JSON.stringify({ id: cmd.id, status: res.status, result }));
+        const text2 = await res.text().catch(() => "");
+        let result;
+        try {
+          result = text2 ? JSON.parse(text2) : {};
+        } catch {
+          result = { raw: text2.slice(0, 500) };
+        }
+        try {
+          ws.send(JSON.stringify({ id: cmd.id, status: res.status, result }));
+        } catch {
+        }
       } catch (err) {
-        ws.send(JSON.stringify({ id: cmd.id, status: 500, error: err instanceof Error ? err.message : String(err) }));
+        try {
+          ws.send(JSON.stringify({ id: cmd.id, status: 500, error: err instanceof Error ? err.message : String(err) }));
+        } catch {
+        }
       }
     };
-    ws.onclose = () => {
+    ws.onclose = (ev) => {
       if (reconnecting) return;
       reconnecting = true;
-      writeLine(`Relay disconnected: ${instanceId.slice(0, 8)}\u2026 \u2014 reconnecting in ${Math.round(backoffMs / 1e3)}s`);
+      const reason = ev.code === 4401 || ev.code === 1008 ? " (token expired \u2014 run `pags login` then `pags up`)" : "";
+      writeLine(`Relay disconnected: ${instanceId.slice(0, 8)}\u2026${reason} \u2014 reconnecting in ${Math.round(backoffMs / 1e3)}s`);
       setTimeout(() => {
         reconnecting = false;
         connect();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@proagentstore/cli",
-  "version": "0.2.9",
+  "version": "0.2.10",
   "description": "CLI for creating, publishing, and running ProAgentStore agents",
   "license": "MIT",
   "type": "module",