baro-ai 0.72.0 → 0.73.0

package/dist/runner.mjs

@@ -3717,6 +3717,7 @@ import { execFileSync, spawn } from "child_process";
 import { appendFileSync, mkdirSync, mkdtempSync, readFileSync, rmSync, writeFileSync } from "fs";
 import { hostname, homedir, tmpdir } from "os";
 import { join } from "path";
+import { createInterface } from "readline/promises";
 
 // ../../node_modules/ws/wrapper.mjs
 var import_stream = __toESM(require_stream(), 1);
@@ -3728,14 +3729,7 @@ var import_subprotocol = __toESM(require_subprotocol(), 1);
 var import_websocket = __toESM(require_websocket(), 1);
 var import_websocket_server = __toESM(require_websocket_server(), 1);
 
-// ../baro-orchestrator/scripts/runner.ts
-var encode = (m) => JSON.stringify(m);
-var url = process.env.CONTROL_URL ?? "wss://api.baro.jigjoy.ai";
-var token = process.env.RUNNER_TOKEN;
-var httpBase = url.replace(/^ws/, "http").replace(/\/+$/, "");
-var credsPath = join(homedir(), ".baro", "credentials.json");
-var VERSION = "0.72.0";
-var updateCachePath = join(homedir(), ".baro", "update-check.json");
+// ../baro-orchestrator/scripts/runner-helpers.ts
 function semverLt(a, b) {
   const pa = a.split(".").map(Number);
   const pb = b.split(".").map(Number);
@@ -3746,6 +3740,23 @@ function semverLt(a, b) {
   }
   return false;
 }
+function buildReexec(execPath, argv, env) {
+  return { cmd: execPath, args: argv.slice(1), env: { ...env, BARO_UPDATED: "1" } };
+}
+function buildInstallServiceArgs(opts) {
+  const args = ["connect", "--install-service", "--token", opts.token, "--workspace", opts.workspace];
+  if (opts.controlUrl) args.push("--control-url", opts.controlUrl);
+  return args;
+}
+
+// ../baro-orchestrator/scripts/runner.ts
+var encode = (m) => JSON.stringify(m);
+var url = process.env.CONTROL_URL ?? "wss://api.baro.jigjoy.ai";
+var token = process.env.RUNNER_TOKEN;
+var httpBase = url.replace(/^ws/, "http").replace(/\/+$/, "");
+var credsPath = join(homedir(), ".baro", "credentials.json");
+var VERSION = "0.73.0";
+var updateCachePath = join(homedir(), ".baro", "update-check.json");
 async function getLatest(force = false) {
   if (!force) {
     try {
@@ -3779,6 +3790,15 @@ async function selfUpdate(latest) {
     child.on("error", () => resolve(false));
   });
 }
+function reexecUpdated() {
+  const { cmd, args, env } = buildReexec(process.execPath, process.argv, process.env);
+  const child = spawn(cmd, args, { stdio: "inherit", detached: false, env });
+  child.on("exit", (code, signal) => process.exit(signal ? 0 : code ?? 0));
+  child.on("error", (e) => {
+    console.error(`[baro] could not restart into the updated runner (${e.message}) \u2014 run \`baro connect\` again`);
+    process.exit(1);
+  });
+}
 var readCliToken = () => {
   try {
     return JSON.parse(readFileSync(credsPath, "utf8")).token;
@@ -3818,6 +3838,7 @@ var workspaceDir = process.env.WORKSPACE_DIR ?? process.cwd();
 var baroBin = process.env.BARO_BIN ?? "baro";
 var runnerId = process.env.RUNNER_ID ?? hostname();
 var runOnce = process.env.BARO_RUN_ONCE === "1";
+var isService = process.env.BARO_SERVICE === "1";
 function cloneRepo(fullName, token2, emit) {
   return new Promise((resolve, reject) => {
     const dir = mkdtempSync(join(tmpdir(), "baro-clone-"));
@@ -3915,8 +3936,11 @@ async function runGoal(d, emit, signal) {
     const child = spawn(
       baroBin,
       ["--headless", d.goal, "--cwd", cwd, "--llm", d.route?.backend ?? "claude", "--parallel", String(d.parallel), "--timeout", String(d.timeoutSecs), ...d.quick ? ["--quick"] : [], ...d.followUp ? ["--continue"] : []],
-      { cwd, env, stdio: ["ignore", "pipe", "pipe"] }
+      // stdin is piped: baro --headless forwards JSON command lines
+      // (agent_message) into the orchestrator's stdin lane.
+      { cwd, env, stdio: ["pipe", "pipe", "pipe"] }
     );
+    activeChild = child;
     const started = Date.now();
     const secs = () => Math.max(1, Math.round((Date.now() - started) / 1e3));
     const stories = /* @__PURE__ */ new Set();
@@ -3958,6 +3982,7 @@ async function runGoal(d, emit, signal) {
     });
     signal.addEventListener("abort", () => child.kill("SIGTERM"));
     child.on("close", (code) => {
+      if (activeChild === child) activeChild = null;
       const ok = doneSuccess ?? (code === 0 && failed === 0 && passed > 0);
       const goalLines = new Set(d.goal.split("\n").map((s) => s.trim()).filter(Boolean));
       const isNoise = (l) => /no stdin data received|redirect stdin explicitly|proceeding without it/i.test(l);
@@ -4028,6 +4053,7 @@ ${fails || r.out.slice(-500)}`);
 var rejected;
 var currentWs = null;
 var inflight = /* @__PURE__ */ new Map();
+var activeChild = null;
 var send = (m) => {
   if (currentWs?.readyState === import_websocket.default.OPEN) currentWs.send(encode(m));
 };
@@ -4040,6 +4066,13 @@ function handleMessage(m) {
     send({ t: "pong", ts: m.ts });
   } else if (m.t === "cancel") {
     inflight.get(m.storyId)?.abort();
+  } else if (m.t === "agent_message") {
+    const { storyId, text } = m;
+    const stdin = activeChild?.stdin;
+    if (stdin && stdin.writable && !stdin.destroyed) {
+      stdin.write(`${JSON.stringify({ type: "agent_message", id: storyId, text })}
+`);
+    }
   } else if (m.t === "dispatch_run") {
     const d = m;
     if (inflight.has(d.runId)) return;
@@ -4060,6 +4093,43 @@ function handleMessage(m) {
     });
   }
 }
+function goodbyeAndExit() {
+  console.log("\n[baro] runner going offline \u2014 runs will fall back to cloud. Keep it always-on: baro connect --install-service");
+  process.exit(0);
+}
+var serviceOffered = false;
+async function maybeOfferServiceInstall() {
+  if (serviceOffered || isService || runOnce || process.env.BARO_NO_SERVICE_PROMPT === "1") return;
+  serviceOffered = true;
+  if (!process.stdin.isTTY || !process.stdout.isTTY) return;
+  await new Promise((r) => setTimeout(r, 1500));
+  if (rejected || !token || currentWs?.readyState !== import_websocket.default.OPEN) return;
+  const rl = createInterface({ input: process.stdin, output: process.stdout });
+  rl.on("SIGINT", () => {
+    rl.close();
+    goodbyeAndExit();
+  });
+  const answer = (await rl.question("\nKeep this runner online in the background (installs a login service)? [Y/n] ")).trim().toLowerCase();
+  rl.close();
+  if (answer === "n" || answer === "no") {
+    console.log("[baro] staying in the foreground \u2014 this runner goes offline when the terminal closes.\n[baro] install the service any time: baro connect --install-service --token <rt_\u2026>");
+    return;
+  }
+  console.log("[baro] installing the background service\u2026");
+  const args = buildInstallServiceArgs({ token, workspace: workspaceDir, controlUrl: process.env.CONTROL_URL });
+  const ok = await new Promise((resolve) => {
+    const ch = spawn(baroBin, args, { stdio: "inherit", shell: process.platform === "win32" });
+    ch.on("exit", (code) => resolve(code === 0));
+    ch.on("error", () => resolve(false));
+  });
+  if (ok) {
+    console.log("[baro] \u2713 service installed \u2014 the runner now stays online across terminal close, logout, and reboot.");
+    console.log("[baro] handing off to the service; this foreground runner is exiting.");
+    currentWs?.close();
+    process.exit(0);
+  }
+  console.warn("[baro] service install failed \u2014 staying in the foreground. Try manually: baro connect --install-service --token <rt_\u2026>");
+}
 function connectOnce() {
   return new Promise((resolve) => {
     const ws = new import_websocket.default(url);
@@ -4067,6 +4137,7 @@ function connectOnce() {
     ws.on("open", () => {
       ws.send(encode({ t: "register", runnerId, hostname: hostname(), token, backends: ["claude"], workspaceIds: ["default"], version: VERSION }));
       console.log(inflight.size ? `[baro] reconnected to ${url} \u2014 resuming ${inflight.size} in-flight run(s)` : `[baro] connected to ${url} \u2014 workspace ${workspaceDir}`);
+      void maybeOfferServiceInstall();
     });
     ws.on("message", (data) => {
       let m;
@@ -4093,20 +4164,37 @@ async function main() {
     await getLatest(true);
     return;
   }
-  try {
-    const latest = await getLatest();
-    if (latest && semverLt(VERSION, latest)) {
-      if (process.env.BARO_SERVICE === "1") {
+  if (process.env.BARO_UPDATED !== "1") {
+    try {
+      const latest = await getLatest(true);
+      if (latest && semverLt(VERSION, latest)) {
         if (await selfUpdate(latest)) {
-          console.log(`[baro] updated to ${latest} \u2014 restarting service\u2026`);
-          process.exit(0);
+          if (isService) {
+            console.log(`[baro] updated to ${latest} \u2014 restarting service\u2026`);
+            process.exit(0);
+          }
+          console.log(`[baro] updated to ${latest} \u2014 restarting the runner\u2026`);
+          reexecUpdated();
+          return;
         }
         console.warn(`[baro] could not self-update (likely a root-owned global install). Update manually: npm i -g baro-ai@latest`);
-      } else {
-        console.warn(`[baro] a newer baro is available (${VERSION} \u2192 ${latest}). Update: npm i -g baro-ai`);
       }
+    } catch {
     }
-  } catch {
+  }
+  if (isService) {
+    setInterval(() => {
+      void (async () => {
+        try {
+          const latest = await getLatest(true);
+          if (latest && semverLt(VERSION, latest) && inflight.size === 0 && await selfUpdate(latest)) {
+            console.log(`[baro] updated to ${latest} \u2014 restarting service\u2026`);
+            process.exit(0);
+          }
+        } catch {
+        }
+      })();
+    }, 6 * 36e5).unref();
   }
   if (!token) {
     const cli = readCliToken();
@@ -4122,6 +4210,10 @@ async function main() {
     console.error("[baro] not signed in. Run `baro login` first, or pass --token <rt_\u2026> (get one from the dashboard).");
     process.exit(1);
   }
+  if (!isService && !runOnce) {
+    process.on("SIGINT", goodbyeAndExit);
+    process.on("SIGTERM", goodbyeAndExit);
+  }
   if (runOnce) {
     setTimeout(() => {
       if (inflight.size === 0) {