@stackmemoryai/stackmemory 1.5.4 → 1.5.6

package/dist/src/cli/commands/orchestrate.js

@@ -60,6 +60,55 @@ function formatElapsed(ms) {
   const days = Math.floor(hours / 24);
   return `${days}d ago`;
 }
+function fmtTokens(n) {
+  if (n >= 1e6) return `${(n / 1e6).toFixed(1)}M`;
+  if (n >= 1e3) return `${(n / 1e3).toFixed(1)}K`;
+  return String(n);
+}
+function budgetBar(pct, width = 30) {
+  const filled = Math.min(Math.round(pct / 100 * width), width);
+  const empty = width - filled;
+  const color = pct >= 75 ? "\x1B[31m" : pct >= 50 ? "\x1B[33m" : "\x1B[32m";
+  const dim = "\x1B[2m";
+  const rst = "\x1B[0m";
+  return `${color}${"\u2588".repeat(filled)}${dim}${"\u2591".repeat(empty)}${rst} ${String(pct).padStart(3)}%`;
+}
+function fmtMinutes(m) {
+  if (m < 0) return "N/A";
+  if (m >= 60) return `${Math.floor(m / 60)}h ${m % 60}m`;
+  return `${m}m`;
+}
+function printUsageSummary(u) {
+  const totalTokens = u.totalTokens || 0;
+  const inputTokens = u.inputTokens || 0;
+  const outputTokens = u.outputTokens || 0;
+  const estMessages = u.estimatedMessages || 0;
+  const tokensPerMin = u.tokensPerMin || 0;
+  const budgetPct5x = u.budgetPct5x || 0;
+  const budgetPct20x = u.budgetPct20x || 0;
+  const mins5x = u.minutesRemaining5x ?? -1;
+  const mins20x = u.minutesRemaining20x ?? -1;
+  const cacheHitRate = u.cacheHitRate || 0;
+  const b = "\x1B[1m";
+  const d = "\x1B[2m";
+  const w = "\x1B[37m";
+  const r = "\x1B[0m";
+  console.log(`${b}Token Usage${r}`);
+  console.log(
+    `  Input  ${w}${fmtTokens(inputTokens)}${r}  ${d}|${r}  Output  ${w}${fmtTokens(outputTokens)}${r}  ${d}|${r}  Total  ${w}${fmtTokens(totalTokens)}${r}`
+  );
+  console.log(
+    `  Rate   ${w}${fmtTokens(tokensPerMin)}/min${r}  ${d}|${r}  Messages  ${w}${estMessages}${r}  ${d}|${r}  Cache hit  ${w}${cacheHitRate}%${r}`
+  );
+  console.log("");
+  console.log(`${b}Budget (Max plan, 5h window)${r}`);
+  console.log(
+    `  5x  (225 msgs)  ${budgetBar(budgetPct5x)}  ${d}~${fmtMinutes(mins5x)} left${r}`
+  );
+  console.log(
+    `  20x (900 msgs)  ${budgetBar(budgetPct20x)}  ${d}~${fmtMinutes(mins20x)} left${r}`
+  );
+}
 function createConductorCommands() {
   const cmd = new Command("conductor");
   cmd.description("Conductor \u2014 autonomous agent orchestration via Linear");
@@ -334,6 +383,54 @@ function createConductorCommands() {
       process.on("SIGTERM", forward);
     });
   });
+  cmd.command("usage").description("Show token usage, budget, and time-to-exhaustion").option("--json", "Output as JSON", false).option("--scan", "Scan Claude Code JSONL logs for historical data", false).action(async (options) => {
+    const statusPath = join(
+      process.cwd(),
+      ".stackmemory",
+      "conductor-status.json"
+    );
+    if (options.scan) {
+      const conductor = new Conductor({ repoRoot: process.cwd() });
+      await conductor.scanUsageLogs();
+      const summary = conductor.getUsageSummary();
+      if (options.json) {
+        console.log(JSON.stringify(summary, null, 2));
+        return;
+      }
+      printUsageSummary(summary);
+      return;
+    }
+    if (!existsSync(statusPath)) {
+      console.log(
+        "No conductor-status.json found. Run with --scan to check JSONL logs, or start the conductor first."
+      );
+      return;
+    }
+    try {
+      const data = JSON.parse(readFileSync(statusPath, "utf-8"));
+      const usage = data.usage || {};
+      if (options.json) {
+        console.log(JSON.stringify(usage, null, 2));
+        return;
+      }
+      printUsageSummary(usage);
+      const rl = data.rateLimit;
+      if (rl) {
+        console.log("");
+        if (rl.inBackoff) {
+          console.log(
+            `Rate limit: \x1B[31mBACKOFF\x1B[0m (${rl.backoffRemainingSec}s remaining, ${rl.totalHits} total hits)`
+          );
+        } else {
+          console.log(
+            `Rate limit: \x1B[32mOK\x1B[0m (${rl.totalHits} total hits)`
+          );
+        }
+      }
+    } catch (err) {
+      console.error("Failed to read status file:", err.message);
+    }
+  });
   cmd.command("start").description("Start the orchestrator daemon").option("--team <id>", "Linear team ID").option(
     "--states <states>",
     "Comma-separated issue states to pick up",
@@ -367,6 +464,305 @@ function createConductorCommands() {
     });
     await conductor.start();
   });
+  cmd.command("monitor").description("Interactive TUI dashboard for conductor monitoring").option("--interval <seconds>", "Auto-refresh interval in seconds", "10").option(
+    "--phase <phase>",
+    "Filter by phase (reading, planning, implementing, testing, committing)"
+  ).option("--no-interactive", "Disable interactive keys (CI/pipe mode)").action(async (options) => {
+    const interval = parseInt(options.interval, 10) * 1e3;
+    const interactive = options.interactive !== false;
+    let currentMode = "dashboard";
+    let paused = false;
+    let refreshInterval = interval;
+    let phaseFilter = options.phase || null;
+    const b = "\x1B[1m";
+    const d = "\x1B[2m";
+    const cyan = "\x1B[36m";
+    const green = "\x1B[32m";
+    const yellow = "\x1B[33m";
+    const red = "\x1B[31m";
+    const r = "\x1B[0m";
+    function readStatuses() {
+      const agentsDir = join(
+        homedir(),
+        ".stackmemory",
+        "conductor",
+        "agents"
+      );
+      if (!existsSync(agentsDir)) return [];
+      const entries = readdirSync(agentsDir, { withFileTypes: true });
+      const statuses = [];
+      for (const entry of entries) {
+        if (!entry.isDirectory()) continue;
+        const statusPath = join(agentsDir, entry.name, "status.json");
+        if (!existsSync(statusPath)) continue;
+        try {
+          statuses.push(JSON.parse(readFileSync(statusPath, "utf-8")));
+        } catch {
+        }
+      }
+      statuses.sort(
+        (a, b2) => new Date(b2.lastUpdate).getTime() - new Date(a.lastUpdate).getTime()
+      );
+      if (phaseFilter) {
+        return statuses.filter((s) => s.phase === phaseFilter);
+      }
+      return statuses;
+    }
+    function printStatusTable(statuses) {
+      if (statuses.length === 0) {
+        const filterNote = phaseFilter ? ` (filter: ${phaseFilter})` : "";
+        console.log(`  No active agents${filterNote}`);
+        return;
+      }
+      const header = `${"Issue".padEnd(12)}${"Phase".padEnd(16)}${"Tools".padStart(7)}${"Files".padStart(7)}${"Tokens".padStart(9)}   Last Update`;
+      console.log(header);
+      for (const s of statuses) {
+        const elapsed = Date.now() - new Date(s.lastUpdate).getTime();
+        const line = `${s.issue.padEnd(12)}${s.phase.padEnd(16)}${String(s.toolCalls).padStart(7)}${String(s.filesModified).padStart(7)}${String(s.tokensUsed).padStart(9)}   ${formatElapsed(elapsed)}`;
+        console.log(line);
+      }
+    }
+    function getWorktreeFiles(workspacePath) {
+      if (!workspacePath || !existsSync(workspacePath)) return [];
+      try {
+        const output = execSync("git status --short 2>/dev/null", {
+          cwd: workspacePath,
+          timeout: 5e3,
+          encoding: "utf-8"
+        });
+        return output.trim().split("\n").filter((l) => l.length > 0);
+      } catch {
+        return [];
+      }
+    }
+    function printFilesView(statuses) {
+      if (statuses.length === 0) {
+        const filterNote = phaseFilter ? ` (filter: ${phaseFilter})` : "";
+        console.log(`  No active agents${filterNote}`);
+        return;
+      }
+      for (const s of statuses) {
+        const elapsed = Date.now() - new Date(s.lastUpdate).getTime();
+        const phaseColor = s.phase === "committing" ? green : s.phase === "testing" ? yellow : s.phase === "implementing" ? cyan : "";
+        console.log(
+          `${b}${s.issue}${r}  ${phaseColor}${s.phase}${r}  ${d}${formatElapsed(elapsed)}${r}`
+        );
+        const files = getWorktreeFiles(s.workspacePath || "");
+        if (files.length === 0) {
+          console.log(`  ${d}(no file changes detected)${r}`);
+        } else {
+          for (const f of files) {
+            const status = f.substring(0, 2);
+            const path = f.substring(3);
+            let color = "";
+            if (status.includes("M")) color = yellow;
+            else if (status.includes("A") || status.includes("?"))
+              color = green;
+            else if (status.includes("D")) color = red;
+            console.log(`  ${color}${status}${r} ${path}`);
+          }
+        }
+        console.log("");
+      }
+    }
+    async function getUsage() {
+      const conductor = new Conductor({ repoRoot: process.cwd() });
+      await conductor.scanUsageLogs();
+      return conductor.getUsageSummary();
+    }
+    async function render() {
+      process.stdout.write("\x1B[2J\x1B[H");
+      const pauseTag = paused ? " [PAUSED]" : "";
+      const intervalSec = Math.round(refreshInterval / 1e3);
+      console.log(
+        `${b}\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550${r}`
+      );
+      console.log(
+        `${b}  Conductor Monitor${r}  ${(/* @__PURE__ */ new Date()).toLocaleTimeString()}${pauseTag}`
+      );
+      console.log(
+        `  Mode: ${cyan}${currentMode}${r}  |  Refresh: ${intervalSec}s`
+      );
+      const filterNote = phaseFilter ? `  Filter: ${cyan}${phaseFilter}${r}` : "";
+      if (interactive) {
+        console.log(
+          `  ${d}[s]tatus [u]sage [f]iles [d]ashboard [j]son [l]ogs [r]efresh [p]ause [1-5]phase [0]clear [+/-] [q]uit${r}`
+        );
+      }
+      if (filterNote) console.log(filterNote);
+      console.log(
+        `${b}\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550${r}`
+      );
+      console.log("");
+      const statuses = readStatuses();
+      switch (currentMode) {
+        case "dashboard": {
+          printStatusTable(statuses);
+          console.log("");
+          const usage = await getUsage();
+          printUsageSummary(usage);
+          break;
+        }
+        case "status":
+          printStatusTable(statuses);
+          break;
+        case "usage": {
+          const usage = await getUsage();
+          printUsageSummary(usage);
+          break;
+        }
+        case "json": {
+          const usage = await getUsage();
+          console.log(JSON.stringify(usage, null, 2));
+          break;
+        }
+        case "files":
+          printFilesView(statuses);
+          break;
+      }
+      console.log("");
+      console.log(
+        `${d}\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500${r}`
+      );
+    }
+    await render();
+    if (!interactive) {
+      const timer = setInterval(async () => {
+        if (!paused) await render();
+      }, refreshInterval);
+      await new Promise((resolve) => {
+        process.on("SIGINT", () => {
+          clearInterval(timer);
+          resolve();
+        });
+        process.on("SIGTERM", () => {
+          clearInterval(timer);
+          resolve();
+        });
+      });
+      return;
+    }
+    if (process.stdin.isTTY) {
+      process.stdin.setRawMode(true);
+    }
+    process.stdin.resume();
+    process.stdin.setEncoding("utf-8");
+    let refreshTimer = null;
+    function scheduleRefresh() {
+      if (refreshTimer) clearTimeout(refreshTimer);
+      refreshTimer = setTimeout(async () => {
+        if (!paused) await render();
+        scheduleRefresh();
+      }, refreshInterval);
+    }
+    scheduleRefresh();
+    process.stdin.on("data", async (key) => {
+      switch (key) {
+        case "s":
+          currentMode = "status";
+          await render();
+          break;
+        case "u":
+          currentMode = "usage";
+          await render();
+          break;
+        case "d":
+          currentMode = "dashboard";
+          await render();
+          break;
+        case "f":
+          currentMode = "files";
+          await render();
+          break;
+        case "j":
+          currentMode = "json";
+          await render();
+          break;
+        case "1":
+          phaseFilter = "reading";
+          await render();
+          break;
+        case "2":
+          phaseFilter = "planning";
+          await render();
+          break;
+        case "3":
+          phaseFilter = "implementing";
+          await render();
+          break;
+        case "4":
+          phaseFilter = "testing";
+          await render();
+          break;
+        case "5":
+          phaseFilter = "committing";
+          await render();
+          break;
+        case "0":
+          phaseFilter = null;
+          await render();
+          break;
+        case "l": {
+          process.stdout.write("\x1B[2J\x1B[H");
+          const statuses = readStatuses();
+          if (statuses.length === 0) {
+            console.log("No active agents to show logs for.");
+          } else {
+            console.log("Active issues:");
+            console.log("");
+            for (const s of statuses) {
+              console.log(`  ${s.issue}  (${s.phase})`);
+            }
+            console.log("");
+            console.log("Use: stackmemory conductor logs <ISSUE-ID> -f");
+          }
+          console.log("\nPress any key to return...");
+          await new Promise((resolve) => {
+            process.stdin.once("data", () => resolve());
+          });
+          await render();
+          break;
+        }
+        case "r":
+          await render();
+          break;
+        case "p":
+          paused = !paused;
+          await render();
+          break;
+        case "+":
+        case "=":
+          refreshInterval += 5e3;
+          await render();
+          scheduleRefresh();
+          break;
+        case "-":
+        case "_":
+          if (refreshInterval > 5e3) refreshInterval -= 5e3;
+          await render();
+          scheduleRefresh();
+          break;
+        case "q":
+        case "":
+          if (refreshTimer) clearTimeout(refreshTimer);
+          if (process.stdin.isTTY) process.stdin.setRawMode(false);
+          process.exit(0);
+          break;
+      }
+    });
+    await new Promise((resolve) => {
+      process.on("SIGINT", () => {
+        if (refreshTimer) clearTimeout(refreshTimer);
+        if (process.stdin.isTTY) process.stdin.setRawMode(false);
+        resolve();
+      });
+      process.on("SIGTERM", () => {
+        if (refreshTimer) clearTimeout(refreshTimer);
+        if (process.stdin.isTTY) process.stdin.setRawMode(false);
+        resolve();
+      });
+    });
+  });
   return cmd;
 }
 export {