claude-code-controller 0.5.0 → 0.6.0

package/dist/api/index.cjs

@@ -131,9 +131,11 @@ var ActionTracker = class {
 var import_hono = require("hono");
 
 // src/controller.ts
-var import_node_events = require("events");
+var import_node_events2 = require("events");
 var import_node_child_process3 = require("child_process");
 var import_node_crypto2 = require("crypto");
+var import_node_fs5 = require("fs");
+var import_node_path4 = require("path");
 
 // src/team-manager.ts
 var import_promises = require("fs/promises");
@@ -466,6 +468,8 @@ if pid == 0:
 else:
     signal.signal(signal.SIGTERM, lambda *a: (os.kill(pid, signal.SIGTERM), sys.exit(0)))
     signal.signal(signal.SIGINT, lambda *a: (os.kill(pid, signal.SIGTERM), sys.exit(0)))
+    buf = b""
+    trust_sent = False
     try:
         while True:
             r, _, _ = select.select([fd, 0], [], [], 1.0)
@@ -475,6 +479,12 @@ else:
                     if not data:
                         break
                     os.write(1, data)
+                    if not trust_sent:
+                        buf += data
+                        if b"Yes" in buf and b"trust" in buf:
+                            os.write(fd, b"\\r")
+                            trust_sent = True
+                            buf = b""
                 except OSError:
                     break
             if 0 in r:
@@ -814,6 +824,135 @@ function createLogger(level = "info") {
   };
 }
 
+// src/statusline-capture.ts
+var import_node_events = require("events");
+var import_node_fs4 = require("fs");
+var import_node_path3 = require("path");
+function statusLineDir(teamName) {
+  return (0, import_node_path3.join)(teamDir(teamName), "statusline");
+}
+function statusLineLogPath(teamName, agentName) {
+  return (0, import_node_path3.join)(statusLineDir(teamName), `${agentName}.jsonl`);
+}
+function buildStatusLineCommand(logFilePath) {
+  const escapedPath = logFilePath.replace(/\\/g, "\\\\").replace(/'/g, "\\'");
+  return [
+    "python3 -c",
+    `'import sys,json;`,
+    `d=json.load(sys.stdin);`,
+    `open("${escapedPath}","a").write(json.dumps(d)+"\\n");`,
+    `print(d.get("model",{}).get("display_name",""))'`
+  ].join(" ");
+}
+function buildStatusLineSettings(logFilePath) {
+  return {
+    statusLine: {
+      type: "command",
+      command: buildStatusLineCommand(logFilePath)
+    }
+  };
+}
+var StatusLineWatcher = class extends import_node_events.EventEmitter {
+  teamName;
+  log;
+  watchers = /* @__PURE__ */ new Map();
+  fileOffsets = /* @__PURE__ */ new Map();
+  stopped = false;
+  constructor(teamName, logger) {
+    super();
+    this.teamName = teamName;
+    this.log = logger;
+  }
+  /**
+   * Ensure the statusline directory exists.
+   */
+  ensureDir() {
+    const dir = statusLineDir(this.teamName);
+    if (!(0, import_node_fs4.existsSync)(dir)) {
+      (0, import_node_fs4.mkdirSync)(dir, { recursive: true });
+    }
+  }
+  /**
+   * Start watching a specific agent's statusLine log file.
+   */
+  watchAgent(agentName) {
+    if (this.stopped) return;
+    const filePath = statusLineLogPath(this.teamName, agentName);
+    if (!(0, import_node_fs4.existsSync)(filePath)) {
+      (0, import_node_fs4.writeFileSync)(filePath, "");
+    }
+    try {
+      const stats = (0, import_node_fs4.readFileSync)(filePath);
+      this.fileOffsets.set(filePath, stats.length);
+    } catch {
+      this.fileOffsets.set(filePath, 0);
+    }
+    try {
+      const watcher = (0, import_node_fs4.watch)(filePath, (eventType) => {
+        if (eventType === "change") {
+          this.readNewLines(agentName, filePath);
+        }
+      });
+      this.watchers.set(agentName, watcher);
+      this.log.debug(`Watching statusLine for agent "${agentName}" at ${filePath}`);
+    } catch (err) {
+      this.log.error(`Failed to watch statusLine for "${agentName}": ${err}`);
+    }
+  }
+  /**
+   * Stop watching a specific agent.
+   */
+  unwatchAgent(agentName) {
+    const watcher = this.watchers.get(agentName);
+    if (watcher) {
+      watcher.close();
+      this.watchers.delete(agentName);
+    }
+    const filePath = statusLineLogPath(this.teamName, agentName);
+    this.fileOffsets.delete(filePath);
+  }
+  /**
+   * Stop all watchers.
+   */
+  stop() {
+    this.stopped = true;
+    for (const [, watcher] of this.watchers) {
+      watcher.close();
+    }
+    this.watchers.clear();
+    this.fileOffsets.clear();
+  }
+  /**
+   * Read new lines appended to the log file since last read.
+   */
+  readNewLines(agentName, filePath) {
+    try {
+      const content = (0, import_node_fs4.readFileSync)(filePath, "utf-8");
+      const offset = this.fileOffsets.get(filePath) ?? 0;
+      const newContent = content.slice(offset);
+      this.fileOffsets.set(filePath, content.length);
+      if (!newContent.trim()) return;
+      const lines = newContent.trim().split("\n");
+      for (const line of lines) {
+        if (!line.trim()) continue;
+        try {
+          const data = JSON.parse(line);
+          const event = {
+            agentName,
+            data,
+            timestamp: (/* @__PURE__ */ new Date()).toISOString()
+          };
+          this.emit("update", event);
+        } catch (parseErr) {
+          this.log.debug(`Failed to parse statusLine JSON: ${line}`);
+        }
+      }
+    } catch (err) {
+      this.log.debug(`Error reading statusLine file: ${err}`);
+    }
+  }
+};
+
 // src/controller.ts
 var PROTOCOL_ONLY_TYPES = /* @__PURE__ */ new Set([
   "shutdown_approved",
@@ -833,12 +972,13 @@ var AGENT_COLORS = [
   "#FF69B4",
   "#7B68EE"
 ];
-var ClaudeCodeController = class extends import_node_events.EventEmitter {
+var ClaudeCodeController = class extends import_node_events2.EventEmitter {
   teamName;
   team;
   tasks;
   processes;
   poller;
+  statusLineWatcher;
   log;
   cwd;
   claudeBinary;
@@ -860,7 +1000,11 @@ var ClaudeCodeController = class extends import_node_events.EventEmitter {
       "controller",
       this.log
     );
+    this.statusLineWatcher = new StatusLineWatcher(this.teamName, this.log);
     this.poller.onMessages((events) => this.handlePollEvents(events));
+    this.statusLineWatcher.on("update", (event) => {
+      this.emit("agent:statusline", event.agentName, event.data);
+    });
   }
   // ─── Lifecycle ───────────────────────────────────────────────────────
   /**
@@ -871,6 +1015,7 @@ var ClaudeCodeController = class extends import_node_events.EventEmitter {
     if (this.initialized) return this;
     await this.team.create({ cwd: this.cwd });
     await this.tasks.init();
+    this.statusLineWatcher.ensureDir();
     this.poller.start();
     this.initialized = true;
     this.log.info(
@@ -911,6 +1056,7 @@ var ClaudeCodeController = class extends import_node_events.EventEmitter {
     }
     await this.processes.killAll();
     this.poller.stop();
+    this.statusLineWatcher.stop();
     await this.team.destroy();
     this.initialized = false;
     this.log.info("Controller shut down");
@@ -939,6 +1085,8 @@ var ClaudeCodeController = class extends import_node_events.EventEmitter {
       subscriptions: []
     };
     await this.team.addMember(member);
+    this.ensureWorkspaceTrusted(cwd, opts.name);
+    this.statusLineWatcher.watchAgent(opts.name);
     const env = Object.keys(this.defaultEnv).length > 0 || opts.env ? { ...this.defaultEnv, ...opts.env } : void 0;
     const proc = this.processes.spawn({
       teamName: this.teamName,
@@ -1145,6 +1293,7 @@ var ClaudeCodeController = class extends import_node_events.EventEmitter {
    * Kill a specific agent.
    */
   async killAgent(name) {
+    this.statusLineWatcher.unwatchAgent(name);
     await this.processes.kill(name);
     await this.team.removeMember(name);
   }
@@ -1206,6 +1355,32 @@ var ClaudeCodeController = class extends import_node_events.EventEmitter {
       }
     }
   }
+  /**
+   * Ensure the agent's cwd has a .claude/settings.local.json so the
+   * CLI skips the interactive workspace trust prompt.
+   * Also injects statusLine capture configuration.
+   */
+  ensureWorkspaceTrusted(cwd, agentName) {
+    const claudeDir = (0, import_node_path4.join)(cwd, ".claude");
+    const settingsPath = (0, import_node_path4.join)(claudeDir, "settings.local.json");
+    (0, import_node_fs5.mkdirSync)(claudeDir, { recursive: true });
+    let settings = {};
+    if ((0, import_node_fs5.existsSync)(settingsPath)) {
+      try {
+        settings = JSON.parse((0, import_node_fs5.readFileSync)(settingsPath, "utf-8"));
+      } catch {
+        settings = {};
+      }
+    }
+    if (agentName && !settings.statusLine) {
+      const logPath = statusLineLogPath(this.teamName, agentName);
+      const statusLineSettings = buildStatusLineSettings(logPath);
+      settings = { ...settings, ...statusLineSettings };
+      this.log.debug(`Injected statusLine capture for "${agentName}"`);
+    }
+    (0, import_node_fs5.writeFileSync)(settingsPath, JSON.stringify(settings, null, 2) + "\n");
+    this.log.debug(`Updated ${settingsPath}`);
+  }
   ensureInitialized() {
     if (!this.initialized) {
       throw new Error(
@@ -1219,7 +1394,7 @@ function sleep2(ms) {
 }
 
 // src/claude.ts
-var import_node_events2 = require("events");
+var import_node_events3 = require("events");
 var import_node_crypto3 = require("crypto");
 Symbol.asyncDispose ??= /* @__PURE__ */ Symbol("Symbol.asyncDispose");
 function buildEnv(opts) {
@@ -1292,7 +1467,7 @@ function waitForReady(controller, agentName, timeoutMs = 15e3) {
     controller.on("agent:exited", onExit);
   });
 }
-var Agent = class _Agent extends import_node_events2.EventEmitter {
+var Agent = class _Agent extends import_node_events3.EventEmitter {
   controller;
   handle;
   ownsController;
@@ -1385,14 +1560,22 @@ var Agent = class _Agent extends import_node_events2.EventEmitter {
     this.ensureNotDisposed();
     const timeout = opts?.timeout ?? 12e4;
     const responsePromise = new Promise((resolve, reject) => {
+      let gotMessage = false;
       const timer = setTimeout(() => {
         cleanup();
         reject(new Error(`Timeout (${timeout}ms) waiting for response`));
       }, timeout);
       const onMsg = (text) => {
+        gotMessage = true;
         cleanup();
         resolve(text);
       };
+      const onIdle = () => {
+        if (!gotMessage) {
+          cleanup();
+          resolve("");
+        }
+      };
       const onExit = (code) => {
         cleanup();
         reject(new Error(`Agent exited (code=${code}) before responding`));
@@ -1400,9 +1583,11 @@ var Agent = class _Agent extends import_node_events2.EventEmitter {
       const cleanup = () => {
         clearTimeout(timer);
         this.removeListener("message", onMsg);
+        this.removeListener("idle", onIdle);
         this.removeListener("exit", onExit);
       };
       this.on("message", onMsg);
+      this.on("idle", onIdle);
       this.on("exit", onExit);
     });
     const wrapped = `${question}
@@ -1421,14 +1606,22 @@ IMPORTANT: You MUST send your complete answer back using the SendMessage tool. D
     this.ensureNotDisposed();
     const timeout = opts?.timeout ?? 12e4;
     return new Promise((resolve, reject) => {
+      let gotMessage = false;
       const timer = setTimeout(() => {
         cleanup();
         reject(new Error(`Timeout (${timeout}ms) waiting for response`));
       }, timeout);
       const onMsg = (text) => {
+        gotMessage = true;
         cleanup();
         resolve(text);
       };
+      const onIdle = () => {
+        if (!gotMessage) {
+          cleanup();
+          resolve("");
+        }
+      };
       const onExit = (code) => {
         cleanup();
         reject(new Error(`Agent exited (code=${code}) before responding`));
@@ -1436,9 +1629,11 @@ IMPORTANT: You MUST send your complete answer back using the SendMessage tool. D
       const cleanup = () => {
         clearTimeout(timer);
         this.removeListener("message", onMsg);
+        this.removeListener("idle", onIdle);
         this.removeListener("exit", onExit);
       };
       this.on("message", onMsg);
+      this.on("idle", onIdle);
       this.on("exit", onExit);
     });
   }
@@ -1472,6 +1667,9 @@ IMPORTANT: You MUST send your complete answer back using the SendMessage tool. D
     const onIdle = (name, _details) => {
       if (name === agentName) this.emit("idle");
     };
+    const onStatusLine = (name, data) => {
+      if (name === agentName) this.emit("statusline", data);
+    };
     const onPermission = (name, parsed) => {
       if (name !== agentName) return;
       let handled = false;
@@ -1520,6 +1718,7 @@ IMPORTANT: You MUST send your complete answer back using the SendMessage tool. D
     };
     this.controller.on("message", onMessage);
     this.controller.on("idle", onIdle);
+    this.controller.on("agent:statusline", onStatusLine);
     this.controller.on("permission:request", onPermission);
     this.controller.on("plan:approval_request", onPlan);
     this.controller.on("agent:exited", onExit);
@@ -1527,6 +1726,7 @@ IMPORTANT: You MUST send your complete answer back using the SendMessage tool. D
     this.boundListeners = [
       { event: "message", fn: onMessage },
       { event: "idle", fn: onIdle },
+      { event: "agent:statusline", fn: onStatusLine },
       { event: "permission:request", fn: onPermission },
       { event: "plan:approval_request", fn: onPlan },
       { event: "agent:exited", fn: onExit },