agent-yes 1.85.0 → 1.87.0

package/ts/subcommands.ts

@@ -12,10 +12,12 @@
  */
 
 import { appendFile, mkdir, open, readFile, stat, writeFile } from "fs/promises";
+import ms from "ms";
 import { homedir } from "os";
 import path from "path";
 import { type GlobalPidRecord, readGlobalPids } from "./globalPidIndex.ts";
 import yargs from "yargs";
+import { type ResolvedRemote, readRemotes, resolveRemoteSpec } from "./remotes.ts";
 
 // ---------------------------------------------------------------------------
 // notes store  (~/.agent-yes/notes.jsonl)
@@ -26,7 +28,7 @@ function notesPath(): string {
   return path.join(dir, "notes.jsonl");
 }
 
-async function readNotes(): Promise<Map<number, string>> {
+export async function readNotes(): Promise<Map<number, string>> {
   let raw: string;
   try {
     raw = await readFile(notesPath(), "utf-8");
@@ -136,6 +138,8 @@ const SUBCOMMANDS = new Set([
   "send",
   "restart",
   "note",
+  "serve",
+  "remote",
 ]);
 
 const IDLE_THRESHOLD_MS = 60 * 1000;
@@ -175,6 +179,14 @@ export async function runSubcommand(argv: string[]): Promise<number | null> {
         return await cmdRestart(rest);
       case "note":
         return await cmdNote(rest);
+      case "serve": {
+        const { cmdServe } = await import("./serve.ts");
+        return cmdServe(rest);
+      }
+      case "remote": {
+        const { cmdRemote } = await import("./remotes.ts");
+        return cmdRemote(rest);
+      }
       default:
         return null;
     }
@@ -189,7 +201,7 @@ export async function runSubcommand(argv: string[]): Promise<number | null> {
 // shared helpers
 // ---------------------------------------------------------------------------
 
-interface CommonOpts {
+export interface CommonOpts {
   all: boolean;
   active: boolean;
   cwdScope: string | null;
@@ -211,7 +223,7 @@ export function matchKeyword(record: GlobalPidRecord, keyword: string): boolean
   return false;
 }
 
-async function listRecords(
+export async function listRecords(
   keyword: string | undefined,
   opts: CommonOpts,
 ): Promise<GlobalPidRecord[]> {
@@ -240,7 +252,7 @@ async function listRecords(
   return records;
 }
 
-function isPidAlive(pid: number): boolean {
+export function isPidAlive(pid: number): boolean {
   try {
     process.kill(pid, 0);
     return true;
@@ -249,7 +261,10 @@ function isPidAlive(pid: number): boolean {
   }
 }
 
-async function resolveOne(keyword: string | undefined, opts: CommonOpts): Promise<GlobalPidRecord> {
+export async function resolveOne(
+  keyword: string | undefined,
+  opts: CommonOpts,
+): Promise<GlobalPidRecord> {
   if (!keyword) {
     throw new Error("keyword required (pid, cwd substring, cli name, or prompt substring)");
   }
@@ -268,6 +283,319 @@ async function resolveOne(keyword: string | undefined, opts: CommonOpts): Promis
   );
 }
 
+// ---------------------------------------------------------------------------
+// remote routing helpers
+// ---------------------------------------------------------------------------
+
+async function remoteGet(remote: ResolvedRemote, pathname: string): Promise<Response> {
+  return fetch(`${remote.url}${pathname}`, {
+    headers: { Authorization: `Bearer ${remote.token}` },
+  });
+}
+
+async function remotePost(
+  remote: ResolvedRemote,
+  pathname: string,
+  body: unknown,
+): Promise<Response> {
+  return fetch(`${remote.url}${pathname}`, {
+    method: "POST",
+    headers: { Authorization: `Bearer ${remote.token}`, "Content-Type": "application/json" },
+    body: JSON.stringify(body),
+  });
+}
+
+async function runRemoteLs(
+  remote: ResolvedRemote,
+  opts: { all: boolean; active: boolean },
+): Promise<number> {
+  const params = new URLSearchParams();
+  if (remote.keyword) params.set("keyword", remote.keyword);
+  if (opts.all) params.set("all", "1");
+  if (opts.active) params.set("active", "1");
+  const res = await remoteGet(remote, `/api/ls?${params}`);
+  if (!res.ok) {
+    process.stderr.write(`remote error ${res.status}: ${await res.text()}\n`);
+    return 1;
+  }
+  const records = (await res.json()) as any[];
+  if (records.length === 0) {
+    process.stderr.write(
+      remote.keyword
+        ? `no agents matched "${remote.keyword}" on ${remote.url}\n`
+        : `no running agents on ${remote.url}\n`,
+    );
+    return 0;
+  }
+  process.stderr.write(`[remote ${remote.url}]\n`);
+  const termWidth = (process.stdout as any).columns ?? 120;
+  const widths = {
+    pid: Math.max(3, ...records.map((r: any) => String(r.pid).length)),
+    cli: Math.max(3, ...records.map((r: any) => String(r.cli).length)),
+    status: Math.max(6, ...records.map((r: any) => String(r.status).length)),
+    cwd: Math.max(3, ...records.map((r: any) => String(r.cwd).length)),
+  };
+  const fixedWidth = widths.pid + widths.cli + widths.status + widths.cwd + 4 * 2;
+  const promptBudget = Math.max(20, termWidth - fixedWidth - 1);
+  const header =
+    [
+      "PID".padEnd(widths.pid),
+      "CLI".padEnd(widths.cli),
+      "STATUS".padEnd(widths.status),
+      "CWD".padEnd(widths.cwd),
+      "PROMPT",
+    ].join("  ") + "\n";
+  process.stdout.write(header);
+  for (const r of records) {
+    const label = r.prompt ? truncate(`→ ${r.prompt}`, promptBudget) : "";
+    process.stdout.write(
+      [
+        String(r.pid).padEnd(widths.pid),
+        String(r.cli).padEnd(widths.cli),
+        String(r.status).padEnd(widths.status),
+        String(r.cwd).padEnd(widths.cwd),
+        label,
+      ].join("  ") + "\n",
+    );
+  }
+  return 0;
+}
+
+async function runRemoteRead(
+  remote: ResolvedRemote,
+  mode: "cat" | "tail" | "head",
+  follow: boolean,
+  n: number,
+  reconnectTimeoutMs = 120_000,
+): Promise<number> {
+  const keyword = remote.keyword ?? "";
+  if (!keyword) {
+    process.stderr.write(
+      "remote tail/cat/head requires a keyword (e.g. token@host:port:keyword)\n",
+    );
+    return 1;
+  }
+
+  if (mode === "tail" && follow) {
+    const ac = new AbortController();
+    process.on("SIGINT", () => ac.abort());
+    const deadline = Date.now() + reconnectTimeoutMs;
+    let delay = 1_000;
+    let attempt = 0;
+
+    process.stderr.write(
+      `[remote ${remote.url}  ${keyword}]\nfollowing... (Ctrl-C to stop, timeout: ${Math.round(reconnectTimeoutMs / 1000)}s)\n`,
+    );
+
+    while (!ac.signal.aborted) {
+      try {
+        const res = await fetch(`${remote.url}/api/tail/${encodeURIComponent(keyword)}`, {
+          headers: { Authorization: `Bearer ${remote.token}`, Accept: "text/event-stream" },
+          signal: ac.signal,
+        });
+        if (!res.ok) {
+          // 401/404 are permanent failures — no point retrying
+          if (res.status === 401 || res.status === 404) {
+            process.stderr.write(`remote error ${res.status}: ${await res.text()}\n`);
+            return 1;
+          }
+          throw new Error(`HTTP ${res.status}`);
+        }
+
+        if (attempt > 0) process.stderr.write("remote: reconnected\n");
+        delay = 1_000; // reset backoff on successful connect
+
+        const reader = res.body!.getReader();
+        const dec = new TextDecoder();
+        let buf = "";
+        while (true) {
+          const { done, value } = await reader.read();
+          if (done) break;
+          buf += dec.decode(value, { stream: true });
+          const lines = buf.split("\n");
+          buf = lines.pop() ?? "";
+          for (const line of lines) {
+            if (!line.startsWith("data: ")) continue;
+            try {
+              const text = JSON.parse(line.slice(6)) as string;
+              process.stdout.write(text);
+              if (!text.endsWith("\n")) process.stdout.write("\n");
+            } catch {
+              /* skip non-JSON */
+            }
+          }
+        }
+        break; // stream ended cleanly
+      } catch (e: any) {
+        if (e.name === "AbortError" || ac.signal.aborted) return 0;
+        if (Date.now() >= deadline) {
+          process.stderr.write(
+            `remote: timeout after ${Math.round(reconnectTimeoutMs / 1000)}s, giving up\n`,
+          );
+          return 1;
+        }
+        process.stderr.write(
+          `remote: disconnected (${e.message}), retrying in ${delay / 1000}s…\n`,
+        );
+        await new Promise<void>((resolve, reject) => {
+          const t = setTimeout(resolve, delay);
+          ac.signal.addEventListener("abort", () => {
+            clearTimeout(t);
+            reject(new Error("abort"));
+          });
+        }).catch(() => {});
+        if (ac.signal.aborted) return 0;
+        delay = Math.min(delay * 2, 30_000);
+        attempt++;
+      }
+    }
+    return 0;
+  }
+
+  // Static read (cat/head/tail without -f)
+  const params = new URLSearchParams({ mode, n: String(n) });
+  const res = await remoteGet(remote, `/api/read/${encodeURIComponent(keyword)}?${params}`);
+  if (!res.ok) {
+    process.stderr.write(`remote error ${res.status}: ${await res.text()}\n`);
+    return 1;
+  }
+  const text = await res.text();
+  process.stderr.write(`[remote ${remote.url}  ${keyword}]\n`);
+  process.stdout.write(text);
+  if (!text.endsWith("\n")) process.stdout.write("\n");
+  return 0;
+}
+
+async function runRemoteSend(remote: ResolvedRemote, msg: string, code: string): Promise<number> {
+  const keyword = remote.keyword ?? "";
+  if (!keyword) {
+    process.stderr.write("remote send requires a keyword (e.g. token@host:port:keyword)\n");
+    return 1;
+  }
+  const res = await remotePost(remote, "/api/send", { keyword, msg, code });
+  if (!res.ok) {
+    process.stderr.write(`remote error ${res.status}: ${await res.text()}\n`);
+    return 1;
+  }
+  const data = (await res.json()) as any;
+  process.stdout.write(`sent to remote pid ${data.pid} (${remote.url}  ${keyword})\n`);
+  return 0;
+}
+
+async function runRemoteStatus(remote: ResolvedRemote): Promise<number> {
+  const keyword = remote.keyword ?? "";
+  if (!keyword) {
+    process.stderr.write("remote status requires a keyword (e.g. token@host:port:keyword)\n");
+    return 1;
+  }
+  const res = await remoteGet(remote, `/api/status/${encodeURIComponent(keyword)}`);
+  if (!res.ok) {
+    process.stderr.write(`remote error ${res.status}: ${await res.text()}\n`);
+    return 1;
+  }
+  process.stdout.write(JSON.stringify(await res.json(), null, 2) + "\n");
+  return 0;
+}
+
+// ---------------------------------------------------------------------------
+// --all-remotes helpers
+// ---------------------------------------------------------------------------
+
+async function fetchRemoteRecordsRaw(
+  url: string,
+  token: string,
+  opts: { all: boolean; active: boolean; keyword?: string },
+): Promise<any[]> {
+  const params = new URLSearchParams();
+  if (opts.all) params.set("all", "1");
+  if (opts.active) params.set("active", "1");
+  if (opts.keyword) params.set("keyword", opts.keyword);
+  try {
+    const res = await fetch(`${url}/api/ls?${params}`, {
+      headers: { Authorization: `Bearer ${token}` },
+      signal: AbortSignal.timeout(5000),
+    });
+    if (!res.ok) return [];
+    return (await res.json()) as any[];
+  } catch {
+    return [];
+  }
+}
+
+async function runAllRemotesLs(opts: {
+  all: boolean;
+  active: boolean;
+  keyword?: string;
+}): Promise<number> {
+  const remotes = await readRemotes();
+  const localOpts: CommonOpts = {
+    all: opts.all,
+    active: opts.active,
+    json: true,
+    latest: false,
+    cwdScope: null,
+  };
+
+  const [localResult, ...remoteResults] = await Promise.allSettled([
+    listRecords(opts.keyword, localOpts).then((recs) => ({
+      host: "local",
+      records: recs as any[],
+    })),
+    ...Array.from(remotes.entries()).map(([alias, cfg]) =>
+      fetchRemoteRecordsRaw(cfg.url, cfg.token, opts).then((records) => ({ host: alias, records })),
+    ),
+  ]);
+
+  type HostedRow = { host: string; rec: any };
+  const rows: HostedRow[] = [];
+  if (localResult.status === "fulfilled") {
+    for (const r of localResult.value.records) rows.push({ host: "local", rec: r });
+  }
+  for (const res of remoteResults) {
+    if (res.status === "fulfilled") {
+      for (const r of res.value.records) rows.push({ host: res.value.host, rec: r });
+    }
+  }
+
+  if (rows.length === 0) {
+    process.stderr.write("no running agents\n");
+    return 0;
+  }
+
+  const termWidth = (process.stdout as any).columns ?? 120;
+  const hostW = Math.max(4, ...rows.map((r) => r.host.length));
+  const pidW = Math.max(3, ...rows.map((r) => String(r.rec.pid).length));
+  const cliW = Math.max(3, ...rows.map((r) => String(r.rec.cli).length));
+  const statusW = Math.max(6, ...rows.map((r) => String(r.rec.status).length));
+  const cwdW = Math.max(3, ...rows.map((r) => shortenPath(String(r.rec.cwd)).length));
+  const promptBudget = Math.max(20, termWidth - hostW - pidW - cliW - statusW - cwdW - 5 * 2 - 1);
+
+  process.stdout.write(
+    [
+      "HOST".padEnd(hostW),
+      "PID".padEnd(pidW),
+      "CLI".padEnd(cliW),
+      "STATUS".padEnd(statusW),
+      "CWD".padEnd(cwdW),
+      "PROMPT",
+    ].join("  ") + "\n",
+  );
+  for (const { host, rec } of rows) {
+    const label = rec.prompt ? truncate(`→ ${rec.prompt}`, promptBudget) : "";
+    process.stdout.write(
+      [
+        host.padEnd(hostW),
+        String(rec.pid).padEnd(pidW),
+        String(rec.cli).padEnd(cliW),
+        String(rec.status).padEnd(statusW),
+        shortenPath(String(rec.cwd)).padEnd(cwdW),
+        label,
+      ].join("  ") + "\n",
+    );
+  }
+  return 0;
+}
+
 // ---------------------------------------------------------------------------
 // ay ls
 // ---------------------------------------------------------------------------
@@ -297,8 +625,14 @@ async function cmdLs(rest: string[]): Promise<number> {
       description: "Show only the most recent agent",
     })
     .option("cwd", { type: "string", description: "Restrict to agents whose cwd starts with dir" })
+    .option("all-remotes", {
+      type: "boolean",
+      default: false,
+      description: "Include agents from all configured remotes (remotes.yaml)",
+    })
     .option("help", { alias: "h", type: "boolean", default: false, description: "Show this help" })
     .example("ay ls", "list running agents")
+    .example("ay ls --all-remotes", "include all configured remote machines")
     .example("ay ls --all", "include exited agents")
     .example("ay ls --json", "machine-readable output")
     .example("ay ls symval", "filter by cwd/prompt keyword")
@@ -313,7 +647,19 @@ async function cmdLs(rest: string[]): Promise<number> {
     return 0;
   }
 
+  if (argv["all-remotes"]) {
+    return runAllRemotesLs({
+      all: argv.all,
+      active: argv.active,
+      keyword: argv._[0] !== undefined ? String(argv._[0]) : undefined,
+    });
+  }
+
   const keyword = argv._[0] !== undefined ? String(argv._[0]) : undefined;
+  if (keyword) {
+    const remote = await resolveRemoteSpec(keyword);
+    if (remote) return runRemoteLs(remote, { all: argv.all, active: argv.active });
+  }
   const opts: CommonOpts = {
     all: argv.all,
     active: argv.active,
@@ -422,6 +768,7 @@ async function cmdLs(rest: string[]): Promise<number> {
     if (alive) {
       hints.push(`  ay status ${alive.pid}                # JSON status snapshot\n`);
       hints.push(`  ay status ${alive.pid} --watch        # stream changes as JSON\n`);
+      hints.push(`  ay status ${alive.pid} --wait-idle    # block until state == idle\n`);
       hints.push(`  ay tail ${alive.pid}                  # view latest output\n`);
       hints.push(`  ay tail -f ${alive.pid}               # follow live output\n`);
       hints.push(
@@ -491,6 +838,11 @@ async function cmdRead(rest: string[], { mode }: ReadOpts): Promise<number> {
       description: "Use most recent match when multiple match",
     })
     .option("cwd", { type: "string", description: "Restrict to agents under this dir" })
+    .option("reconnect-timeout", {
+      type: "number",
+      default: 120,
+      description: "Seconds before giving up reconnecting remote SSE (default: 120)",
+    })
     .help(false)
     .version(false)
     .exitProcess(false);
@@ -504,6 +856,18 @@ async function cmdRead(rest: string[], { mode }: ReadOpts): Promise<number> {
     cwdScope: typeof argv.cwd === "string" ? path.resolve(argv.cwd) : null,
   };
   const keyword = argv._[0] !== undefined ? String(argv._[0]) : undefined;
+  if (keyword) {
+    const remote = await resolveRemoteSpec(keyword);
+    const nFlag2 = argv.n;
+    const n2 =
+      nFlag2 !== undefined && Number.isFinite(nFlag2) && nFlag2 > 0
+        ? Math.floor(nFlag2)
+        : mode === "cat"
+          ? 0
+          : 96;
+    const reconnectTimeoutMs = ((argv["reconnect-timeout"] as number) ?? 120) * 1000;
+    if (remote) return runRemoteRead(remote, mode, argv.follow, n2, reconnectTimeoutMs);
+  }
   const follow = argv.follow;
   const nFlag = argv.n;
   const n =
@@ -579,7 +943,7 @@ async function cmdRead(rest: string[], { mode }: ReadOpts): Promise<number> {
  * Feed the raw PTY bytes through @xterm/headless and emit plain text.
  * Same approach as koho's renderTerminalBuffer + agent-yes's XtermProxy.
  */
-async function renderRawLog(
+export async function renderRawLog(
   buf: Uint8Array,
   { mode, n }: { mode: "cat" | "tail" | "head"; n: number },
 ): Promise<string> {
@@ -773,6 +1137,10 @@ async function cmdSend(rest: string[]): Promise<number> {
     throw new Error("usage: ay send <keyword> <msg|-> [--code=enter|esc|ctrl-c|ctrl-y|tab|none]");
 
   const codeName = argv.code.toLowerCase();
+  {
+    const remote = await resolveRemoteSpec(keyword);
+    if (remote) return runRemoteSend(remote, rawMessage, codeName);
+  }
   const trailing = controlCodeFromName(codeName);
 
   const record = await resolveOne(keyword, opts);
@@ -851,7 +1219,7 @@ export function controlCodeFromName(name: string): string {
   }
 }
 
-async function writeToIpc(ipcPath: string, payload: string): Promise<void> {
+export async function writeToIpc(ipcPath: string, payload: string): Promise<void> {
   if (process.platform === "win32") {
     const { connect } = await import("net");
     await new Promise<void>((resolve, reject) => {
@@ -974,7 +1342,7 @@ async function cmdNote(rest: string[]): Promise<number> {
 // ay status
 // ---------------------------------------------------------------------------
 
-interface StatusSnapshot {
+export interface StatusSnapshot {
   pid: number;
   cli: string;
   cwd: string;
@@ -989,7 +1357,7 @@ interface StatusSnapshot {
   log_file: string | null;
 }
 
-async function snapshotStatus(record: GlobalPidRecord): Promise<StatusSnapshot> {
+export async function snapshotStatus(record: GlobalPidRecord): Promise<StatusSnapshot> {
   const alive = isPidAlive(record.pid);
   let state: "active" | "idle" | "stopped";
   let logMtimeMs: number | null = null;
@@ -1032,6 +1400,15 @@ async function cmdStatus(rest: string[]): Promise<number> {
       default: false,
       description: "Stream changes as JSON",
     })
+    .option("wait-idle", {
+      type: "boolean",
+      default: false,
+      description: "Block until state == idle. Exit 0 idle, 1 stopped, 2 timeout",
+    })
+    .option("timeout", {
+      type: "string",
+      description: "Timeout for --wait-idle (e.g. 30s, 5m). Default: no timeout",
+    })
     .option("interval", { type: "number", default: 2, description: "Poll interval in seconds" })
     .option("latest", { type: "boolean", default: false, description: "Use most recent match" })
     .option("cwd", { type: "string", description: "Restrict to agents under this dir" })
@@ -1049,11 +1426,25 @@ async function cmdStatus(rest: string[]): Promise<number> {
   };
   const keyword = argv._[0] !== undefined ? String(argv._[0]) : undefined;
 
-  if (!keyword) throw new Error("usage: ay status <keyword> [--watch] [--interval=N]");
+  if (!keyword)
+    throw new Error("usage: ay status <keyword> [--watch | --wait-idle] [--timeout=Ns]");
+
+  {
+    const remote = await resolveRemoteSpec(keyword);
+    if (remote) return runRemoteStatus(remote);
+  }
 
   const watch = argv.watch;
+  const waitIdle = argv["wait-idle"];
   const intervalFlag = argv.interval;
   const intervalMs = Math.max(500, (Number.isFinite(intervalFlag) ? intervalFlag : 2) * 1000);
+  const timeoutMs =
+    typeof argv.timeout === "string" && argv.timeout.length > 0
+      ? (ms(argv.timeout) ?? Number.NaN)
+      : null;
+  if (timeoutMs !== null && !Number.isFinite(timeoutMs)) {
+    throw new Error(`invalid --timeout value: ${argv.timeout}`);
+  }
 
   const record = await resolveOne(keyword, opts);
 
@@ -1062,6 +1453,26 @@ async function cmdStatus(rest: string[]): Promise<number> {
     process.stdout.write(JSON.stringify(out) + "\n");
   };
 
+  if (waitIdle) {
+    const startedAt = Date.now();
+    for (;;) {
+      const snap = await snapshotStatus(record);
+      if (snap.state === "idle") {
+        emit(snap);
+        return 0;
+      }
+      if (snap.state === "stopped") {
+        emit(snap);
+        return 1;
+      }
+      if (timeoutMs !== null && Date.now() - startedAt >= timeoutMs) {
+        emit(snap);
+        return 2;
+      }
+      await new Promise((r) => setTimeout(r, intervalMs));
+    }
+  }
+
   if (!watch) {
     emit(await snapshotStatus(record));
     return 0;