claude-code-controller 0.4.0 → 0.5.1

package/dist/index.js

@@ -6,6 +6,8 @@ import { randomUUID as randomUUID3 } from "crypto";
 import { EventEmitter } from "events";
 import { execSync as execSync2 } from "child_process";
 import { randomUUID as randomUUID2 } from "crypto";
+import { mkdirSync, existsSync as existsSync4, writeFileSync } from "fs";
+import { join as join2 } from "path";
 
 // src/team-manager.ts
 import { readFile, writeFile, mkdir, rm } from "fs/promises";
@@ -338,6 +340,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)
@@ -347,6 +351,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:
@@ -372,6 +382,7 @@ else:
       stdio: ["pipe", "pipe", "pipe"],
       env: {
         ...process.env,
+        CLAUDECODE: "1",
         CLAUDE_CODE_EXPERIMENTAL_AGENT_TEAMS: "1",
         ...opts.env
       }
@@ -705,7 +716,10 @@ var silentLogger = {
 var PROTOCOL_ONLY_TYPES = /* @__PURE__ */ new Set([
   "shutdown_approved",
   "plan_approval_response",
-  "permission_response"
+  "permission_response",
+  "task_completed",
+  "sandbox_permission_request",
+  "sandbox_permission_response"
 ]);
 var AGENT_COLORS = [
   "#00FF00",
@@ -813,12 +827,17 @@ var ClaudeCodeController = class extends EventEmitter {
       name: opts.name,
       agentType: opts.type || "general-purpose",
       model: opts.model,
+      prompt: opts.prompt,
+      color,
+      planModeRequired: false,
+      backendType: "in-process",
       joinedAt: Date.now(),
       tmuxPaneId: "",
       cwd,
       subscriptions: []
     };
     await this.team.addMember(member);
+    this.ensureWorkspaceTrusted(cwd);
     const env = Object.keys(this.defaultEnv).length > 0 || opts.env ? { ...this.defaultEnv, ...opts.env } : void 0;
     const proc = this.processes.spawn({
       teamName: this.teamName,
@@ -1055,7 +1074,7 @@ var ClaudeCodeController = class extends EventEmitter {
       const { raw, parsed } = event;
       switch (parsed.type) {
         case "idle_notification":
-          this.emit("idle", raw.from);
+          this.emit("idle", raw.from, parsed);
           break;
         case "shutdown_approved":
           this.log.info(
@@ -1075,6 +1094,9 @@ var ClaudeCodeController = class extends EventEmitter {
           );
           this.emit("permission:request", raw.from, parsed);
           break;
+        case "task_completed":
+          this.emit("message", raw.from, raw);
+          break;
         case "plain_text":
           this.emit("message", raw.from, raw);
           break;
@@ -1083,6 +1105,19 @@ var ClaudeCodeController = class extends EventEmitter {
       }
     }
   }
+  /**
+   * Ensure the agent's cwd has a .claude/settings.local.json so the
+   * CLI skips the interactive workspace trust prompt.
+   */
+  ensureWorkspaceTrusted(cwd) {
+    const claudeDir = join2(cwd, ".claude");
+    const settingsPath = join2(claudeDir, "settings.local.json");
+    if (!existsSync4(settingsPath)) {
+      mkdirSync(claudeDir, { recursive: true });
+      writeFileSync(settingsPath, "{}\n");
+      this.log.debug(`Created ${settingsPath} for workspace trust`);
+    }
+  }
   ensureInitialized() {
     if (!this.initialized) {
       throw new Error(
@@ -1129,7 +1164,7 @@ function waitForReady(controller, agentName, timeoutMs = 15e3) {
         )
       );
     }, timeoutMs);
-    const onReady = (name) => {
+    const onReady = (name, ..._rest) => {
       if (name === agentName && !settled) {
         settled = true;
         cleanup();
@@ -1260,14 +1295,22 @@ var Agent = class _Agent extends EventEmitter2 {
     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`));
@@ -1275,9 +1318,11 @@ var Agent = class _Agent extends EventEmitter2 {
       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}
@@ -1296,14 +1341,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`));
@@ -1311,9 +1364,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);
     });
   }
@@ -1344,7 +1399,7 @@ IMPORTANT: You MUST send your complete answer back using the SendMessage tool. D
     const onMessage = (name, msg) => {
       if (name === agentName) this.emit("message", msg.text);
     };
-    const onIdle = (name) => {
+    const onIdle = (name, _details) => {
       if (name === agentName) this.emit("idle");
     };
     const onPermission = (name, parsed) => {