@onklave/agent-cli 0.1.55 → 0.1.56

package/main.js

@@ -1536,6 +1536,11 @@ var HeartbeatService = class {
             `heartbeat failed: ${response.status} ${response.statusText}`
           )
         );
+        return;
+      }
+      const respBody = await response.json().catch(() => null);
+      if (respBody?.pendingUpgrade) {
+        this.opts.onPendingUpgrade?.({ targetVersion: respBody.targetVersion });
       }
     } catch (err) {
       this.onError(err instanceof Error ? err : new Error(String(err)));
@@ -5226,13 +5231,48 @@ async function daemonStart() {
   let sigintTimer = null;
   let runtimePublisher = null;
   let updateCheckTimer = null;
+  const upgradeService = new DaemonUpgradeService({
+    allowAutoUpgrade: !!process.env["ONKLAVE_ALLOW_AUTO_UPGRADE"],
+    currentVersion: readPackageVersion() ?? "0.0.0",
+    runInstall: async () => {
+      const { execFileSync: execFileSync3 } = __require("child_process");
+      execFileSync3("npm", ["install", "-g", "@onklave/agent-cli@latest"], {
+        stdio: "pipe",
+        timeout: 18e4
+      });
+      const v = execFileSync3(
+        "npm",
+        ["view", "@onklave/agent-cli@latest", "version"],
+        { stdio: "pipe", timeout: 3e4 }
+      ).toString().trim();
+      return v || "latest";
+    },
+    requestRestart: () => void drainAndExit("upgrade"),
+    audit: (action, details) => auditStreamer.record({
+      sessionId: creds.machineId,
+      type: "daemon_lifecycle" /* DAEMON_LIFECYCLE */,
+      action,
+      details,
+      outcome: action === "upgrade_failed" ? "failure" : "success"
+    }),
+    log: (m) => console.log(m)
+  });
+  let upgradeInFlight = false;
+  const triggerUpgrade = () => {
+    if (upgradeInFlight) return;
+    upgradeInFlight = true;
+    void upgradeService.handle().finally(() => {
+      upgradeInFlight = false;
+    });
+  };
   const heartbeat = new HeartbeatService({
     platformUrl,
     deviceToken: creds.deviceToken,
     machineId: creds.machineId,
     getActiveSessionCount: () => claimHandler.getActiveSessionCount(),
     cliVersion: readPackageVersion() ?? "0.0.0",
-    posture: collectPosture()
+    posture: collectPosture(),
+    onPendingUpgrade: () => triggerUpgrade()
   });
   const drainAndExit = async (reason) => {
     if (shuttingDown) return;
@@ -5295,34 +5335,8 @@ async function daemonStart() {
   }
   console.log("Connected.");
   claimHandler.attachToSocket(comms.inner());
-  const upgradeService = new DaemonUpgradeService({
-    allowAutoUpgrade: !!process.env["ONKLAVE_ALLOW_AUTO_UPGRADE"],
-    currentVersion: readPackageVersion() ?? "0.0.0",
-    runInstall: async () => {
-      const { execFileSync: execFileSync3 } = __require("child_process");
-      execFileSync3("npm", ["install", "-g", "@onklave/agent-cli@latest"], {
-        stdio: "pipe",
-        timeout: 18e4
-      });
-      const v = execFileSync3(
-        "npm",
-        ["view", "@onklave/agent-cli@latest", "version"],
-        { stdio: "pipe", timeout: 3e4 }
-      ).toString().trim();
-      return v || "latest";
-    },
-    requestRestart: () => void drainAndExit("upgrade"),
-    audit: (action, details) => auditStreamer.record({
-      sessionId: creds.machineId,
-      type: "daemon_lifecycle" /* DAEMON_LIFECYCLE */,
-      action,
-      details,
-      outcome: action === "upgrade_failed" ? "failure" : "success"
-    }),
-    log: (m) => console.log(m)
-  });
   comms.inner().onUpgradeRequest(() => {
-    void upgradeService.handle();
+    triggerUpgrade();
   });
   await stateService.transition("online");
   heartbeat.start();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@onklave/agent-cli",
-  "version": "0.1.55",
+  "version": "0.1.56",
   "description": "Onklave Agent CLI — local agent runner with cloud orchestration",
   "bin": {
     "onklave": "./main.js"