@stackmemoryai/stackmemory 0.5.64 → 0.5.67

package/dist/src/cli/claude-sm.js

@@ -27,6 +27,22 @@ import {
 } from "../core/models/model-router.js";
 import { launchWrapper } from "../features/sweep/pty-wrapper.js";
 import { FallbackMonitor } from "../core/models/fallback-monitor.js";
+import {
+  ensureWorkerStateDir,
+  saveRegistry,
+  loadRegistry,
+  clearRegistry
+} from "../features/workers/worker-registry.js";
+import {
+  isTmuxAvailable,
+  createTmuxSession,
+  sendToPane,
+  killTmuxSession,
+  attachToSession,
+  listPanes,
+  sendCtrlC,
+  sessionExists
+} from "../features/workers/tmux-manager.js";
 const DEFAULT_SM_CONFIG = {
   defaultWorktree: false,
   defaultSandbox: false,
@@ -156,11 +172,15 @@ class ClaudeSM {
     }
     return null;
   }
-  gepaProcess = null;
+  gepaProcesses = [];
   startGEPAWatcher() {
-    const claudeMdPath = fs.existsSync(path.join(process.cwd(), "CLAUDE.md")) ? path.join(process.cwd(), "CLAUDE.md") : null;
-    if (!claudeMdPath) {
-      console.log(chalk.gray("   Prompt Forge: disabled (no CLAUDE.md found)"));
+    const watchFiles = ["CLAUDE.md", "AGENT.md", "AGENTS.md"].map((f) => path.join(process.cwd(), f)).filter((p) => fs.existsSync(p));
+    if (watchFiles.length === 0) {
+      console.log(
+        chalk.gray(
+          "   Prompt Forge: disabled (no CLAUDE.md, AGENT.md, or AGENTS.md found)"
+        )
+      );
       return;
     }
     const gepaPaths = [
@@ -193,29 +213,31 @@ class ClaudeSM {
       console.log(chalk.gray("   Prompt Forge: disabled (scripts not found)"));
       return;
     }
-    this.gepaProcess = spawn("node", [gepaScript, "watch", claudeMdPath], {
-      detached: true,
-      stdio: ["ignore", "pipe", "pipe"],
-      env: { ...process.env, GEPA_SILENT: "1" }
-    });
-    this.gepaProcess.unref();
-    this.gepaProcess.stdout?.on("data", (data) => {
-      const output = data.toString().trim();
-      if (output && !output.includes("Watching")) {
-        console.log(chalk.magenta(`[GEPA] ${output}`));
-      }
-    });
+    for (const filePath of watchFiles) {
+      const gepaProcess = spawn("node", [gepaScript, "watch", filePath], {
+        detached: true,
+        stdio: ["ignore", "pipe", "pipe"],
+        env: { ...process.env, GEPA_SILENT: "1" }
+      });
+      gepaProcess.unref();
+      this.gepaProcesses.push(gepaProcess);
+      gepaProcess.stdout?.on("data", (data) => {
+        const output = data.toString().trim();
+        if (output && !output.includes("Watching")) {
+          console.log(chalk.magenta(`[GEPA] ${output}`));
+        }
+      });
+    }
+    const fileNames = watchFiles.map((f) => path.basename(f)).join(", ");
     console.log(
-      chalk.cyan(
-        `   Prompt Forge: watching ${path.basename(claudeMdPath)} for optimization`
-      )
+      chalk.cyan(`   Prompt Forge: watching ${fileNames} for optimization`)
     );
   }
   stopGEPAWatcher() {
-    if (this.gepaProcess) {
-      this.gepaProcess.kill("SIGTERM");
-      this.gepaProcess = null;
+    for (const proc of this.gepaProcesses) {
+      proc.kill("SIGTERM");
     }
+    this.gepaProcesses = [];
   }
   ensureGreptileMcp() {
     const apiKey = process.env["GREPTILE_API_KEY"];
@@ -822,11 +844,17 @@ class ClaudeSM {
           claudeBin: claudeBin2,
           claudeArgs
         });
+        return;
       } catch (error) {
-        console.error(chalk.red(error.message));
-        process.exit(1);
+        const msg = error.message || "Unknown PTY error";
+        console.error(chalk.yellow(`[Sweep disabled] ${msg}`));
+        console.log(
+          chalk.gray(
+            "Falling back to direct Claude launch (no prediction bar)..."
+          )
+        );
+        this.config.useSweep = false;
       }
-      return;
     }
     console.log(chalk.gray("Starting Claude..."));
     console.log(chalk.gray("\u2500".repeat(42)));
@@ -1224,6 +1252,115 @@ GREPTILE_API_KEY=${key.trim()}
   console.log(chalk.gray(`
 Saved to ${getConfigPath()}`));
 });
+program.command("spawn <count>").description("Spawn N parallel Claude workers in tmux panes").option("-t, --task <desc>", "Task description for each worker").option("--worktree", "Create isolated git worktrees per worker").option("--no-sweep", "Disable Sweep predictions").option("--no-attach", "Do not attach to tmux session after creation").action(async (countStr, opts) => {
+  const count = parseInt(countStr, 10);
+  if (isNaN(count) || count < 1 || count > 8) {
+    console.error(chalk.red("Worker count must be between 1 and 8"));
+    process.exit(1);
+  }
+  if (!isTmuxAvailable()) {
+    console.error(chalk.red("tmux is required for parallel workers."));
+    console.log(chalk.gray("Install with: brew install tmux"));
+    process.exit(1);
+  }
+  const sessionId = uuidv4().substring(0, 8);
+  const sessionName = `claude-sm-${sessionId}`;
+  console.log(
+    chalk.blue(`Spawning ${count} workers in tmux session: ${sessionName}`)
+  );
+  createTmuxSession(sessionName, count);
+  const workers = [];
+  const panes = listPanes(sessionName);
+  for (let i = 0; i < count; i++) {
+    const workerId = `w${i}-${uuidv4().substring(0, 6)}`;
+    const stateDir = ensureWorkerStateDir(workerId);
+    const pane = panes[i] || String(i);
+    const parts = ["claude-sm"];
+    if (opts["sweep"] === false) parts.push("--no-sweep");
+    if (opts["worktree"]) parts.push("--worktree");
+    if (opts["task"]) parts.push("--task", `"${opts["task"]}"`);
+    const cmd = parts.join(" ");
+    sendToPane(
+      sessionName,
+      pane,
+      `export SWEEP_INSTANCE_ID=${workerId} SWEEP_STATE_DIR=${stateDir} && ${cmd}`
+    );
+    workers.push({
+      id: workerId,
+      pane,
+      cwd: process.cwd(),
+      startedAt: (/* @__PURE__ */ new Date()).toISOString(),
+      stateDir,
+      task: opts["task"]
+    });
+    console.log(
+      chalk.gray(
+        `  Worker ${i}: ${workerId} (pane ${pane}, state: ${stateDir})`
+      )
+    );
+  }
+  const session = {
+    sessionName,
+    workers,
+    createdAt: (/* @__PURE__ */ new Date()).toISOString()
+  };
+  saveRegistry(session);
+  console.log(chalk.green(`
+Registry saved (${workers.length} workers)`));
+  if (opts["attach"] !== false) {
+    console.log(chalk.gray("Attaching to tmux session..."));
+    attachToSession(sessionName);
+  } else {
+    console.log(
+      chalk.gray(`Attach later with: tmux attach -t ${sessionName}`)
+    );
+  }
+});
+const workersCmd = program.command("workers").description("List active workers (default) or manage them");
+workersCmd.command("list", { isDefault: true }).description("List active workers").action(() => {
+  const session = loadRegistry();
+  if (!session) {
+    console.log(chalk.gray("No active worker session."));
+    return;
+  }
+  const alive = sessionExists(session.sessionName);
+  const status = alive ? chalk.green("ACTIVE") : chalk.red("DEAD");
+  console.log(chalk.blue(`Session: ${session.sessionName} [${status}]`));
+  console.log(chalk.gray(`Created: ${session.createdAt}`));
+  console.log();
+  for (const w of session.workers) {
+    const taskLabel = w.task ? ` task="${w.task}"` : "";
+    console.log(`  ${chalk.cyan(w.id)} pane=${w.pane}${taskLabel}`);
+    console.log(chalk.gray(`    state: ${w.stateDir}`));
+  }
+});
+workersCmd.command("kill [id]").description("Kill entire session or send Ctrl-C to a specific worker").action((id) => {
+  const session = loadRegistry();
+  if (!session) {
+    console.log(chalk.gray("No active worker session."));
+    return;
+  }
+  if (id) {
+    const worker = session.workers.find((w) => w.id === id);
+    if (!worker) {
+      console.error(chalk.red(`Worker ${id} not found.`));
+      process.exit(1);
+    }
+    sendCtrlC(session.sessionName, worker.pane);
+    console.log(
+      chalk.yellow(`Sent Ctrl-C to worker ${id} (pane ${worker.pane})`)
+    );
+  } else {
+    if (sessionExists(session.sessionName)) {
+      killTmuxSession(session.sessionName);
+      console.log(
+        chalk.yellow(`Killed tmux session: ${session.sessionName}`)
+      );
+    }
+    clearRegistry();
+    console.log(chalk.gray("Registry cleared."));
+  }
+});
 program.option("-w, --worktree", "Create isolated worktree for this instance").option("-W, --no-worktree", "Disable worktree (override default)").option("-r, --remote", "Enable remote mode (WhatsApp for all questions)").option("--no-remote", "Disable remote mode (override default)").option("-n, --notify-done", "Send WhatsApp notification when session ends").option("--no-notify-done", "Disable notification when session ends").option(
   "--whatsapp",
   "Enable WhatsApp mode (auto-start webhook + ngrok + notifications)"