@lesscap/baton-cli 0.2.6 → 0.2.7

package/dist/baton.mjs

@@ -768,10 +768,17 @@ var common = {
   },
   json: { type: "boolean", description: "output JSON" }
 };
+var resolveAuth = (env, fileToken) => {
+  const envBearer = env.BATON_TOKEN ?? env.BATON_WORKER_TOKEN;
+  if (envBearer) return { bearer: envBearer };
+  if (env.BATON_USER && env.BATON_PASS) return "cookie";
+  return fileToken ? { bearer: fileToken } : void 0;
+};
 var clientFor = (args) => {
-  const envAuth = process.env.BATON_TOKEN ?? (process.env.BATON_USER && process.env.BATON_PASS);
-  const cfgToken = envAuth ? void 0 : loadProjectConfigOrNull(projectConfigPath())?.worker?.apiToken;
-  return createClient(resolveBaseUrl(args.url), cfgToken ? { bearer: cfgToken } : void 0);
+  const baseUrl = resolveBaseUrl(args.url);
+  const fileToken = loadProjectConfigOrNull(projectConfigPath())?.worker?.apiToken;
+  const auth = resolveAuth(process.env, fileToken);
+  return createClient(baseUrl, auth && auth !== "cookie" ? auth : void 0);
 };
 var resolveProjectId = (args) => {
   if (args.project !== void 0 && args.project !== "") return Number(args.project);
@@ -2645,11 +2652,35 @@ var runWorkerDaemon = async (cfg, client, signal) => {
   const worktreeDir = defaultWorktreeDir();
   const children = /* @__PURE__ */ new Map();
   const log = (m) => console.log(`[worker #${cfg.workerId} ${cfg.name}] ${m}`);
-  const ping = () => {
-    void client.workers.heartbeat(cfg.machineId).catch((e) => log(`heartbeat failed: ${String(e)}`));
+  const HEARTBEAT_MS = 3e4;
+  const HEARTBEAT_TIMEOUT_MS = 15e3;
+  const MAX_HEARTBEAT_FAILURES = 5;
+  let heartbeatFailures = 0;
+  let fatal = false;
+  let stop = () => {
+  };
+  const ping = async () => {
+    try {
+      await Promise.race([
+        client.workers.heartbeat(cfg.machineId),
+        new Promise(
+          (_, reject) => setTimeout(() => reject(new Error("heartbeat timeout")), HEARTBEAT_TIMEOUT_MS)
+        )
+      ]);
+      heartbeatFailures = 0;
+    } catch (e) {
+      heartbeatFailures += 1;
+      log(`heartbeat failed (${heartbeatFailures}/${MAX_HEARTBEAT_FAILURES}): ${String(e)}`);
+      if (heartbeatFailures >= MAX_HEARTBEAT_FAILURES) {
+        const downSec = MAX_HEARTBEAT_FAILURES * HEARTBEAT_MS / 1e3;
+        log(`heartbeat down ~${downSec}s \u2014 exiting so the supervisor restarts a clean worker`);
+        fatal = true;
+        stop();
+      }
+    }
   };
-  ping();
-  const hb = setInterval(ping, 3e4);
+  void ping();
+  const hb = setInterval(() => void ping(), HEARTBEAT_MS);
   const killChild = (child) => {
     if (child.pid === void 0) return;
     try {
@@ -2769,12 +2800,14 @@ var runWorkerDaemon = async (cfg, client, signal) => {
   es.onerror = () => log("command stream error (eventsource will retry)");
   log(`listening for commands (server ${cfg.server}, repo ${repo})`);
   await new Promise((resolve2) => {
+    stop = resolve2;
     if (signal.aborted) return resolve2();
     signal.addEventListener("abort", () => resolve2(), { once: true });
   });
   es.close();
   clearInterval(hb);
   for (const { child } of children.values()) killChild(child);
+  if (fatal) process.exit(1);
 };
 
 // src/worker/machine-id.ts

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lesscap/baton-cli",
-  "version": "0.2.6",
+  "version": "0.2.7",
   "license": "MIT",
   "type": "module",
   "description": "baton CLI — register/run a worker, manage projects/requirements/tasks/sessions",