@gaberrb/polypus 0.2.2 → 0.3.0

package/dist/index.js

@@ -177,6 +177,15 @@ var en = {
   "swarm.mergeConflict": "  conflict merging {branch}",
   "swarm.summary": "Summary:",
   "swarm.allMerged": "\u2713 All committed branches merged cleanly.",
+  "swarm.view.header": "Swarm \xB7 orchestrator [{lead}]",
+  "swarm.view.decomposing": "splitting the task\u2026",
+  "swarm.view.pending": "queued",
+  "swarm.view.running": "running",
+  "swarm.view.done": "done",
+  "swarm.view.stopped": "stopped",
+  "swarm.view.conflict": "conflict",
+  "swarm.view.step": "step {n}",
+  "swarm.view.steps": "{n} steps",
   "swarm.conflictsHeader": "\u26A0 {n} branch(es) had merge conflicts (kept for inspection):",
   "swarm.statusDone": "done",
   "swarm.statusIncomplete": "incomplete",
@@ -373,6 +382,15 @@ var ptBR = {
   "swarm.mergeConflict": "  conflito ao mesclar {branch}",
   "swarm.summary": "Resumo:",
   "swarm.allMerged": "\u2713 Todos os branches commitados foram mesclados sem conflito.",
+  "swarm.view.header": "Swarm \xB7 orquestrador [{lead}]",
+  "swarm.view.decomposing": "dividindo a tarefa\u2026",
+  "swarm.view.pending": "na fila",
+  "swarm.view.running": "executando",
+  "swarm.view.done": "conclu\xEDdo",
+  "swarm.view.stopped": "parado",
+  "swarm.view.conflict": "conflito",
+  "swarm.view.step": "passo {n}",
+  "swarm.view.steps": "{n} passos",
   "swarm.conflictsHeader": "\u26A0 {n} branch(es) tiveram conflitos de merge (mantidos para inspe\xE7\xE3o):",
   "swarm.statusDone": "ok",
   "swarm.statusIncomplete": "incompleta",
@@ -2202,9 +2220,9 @@ var TENTACLE_FRAMES = [
 ];
 var WIDTH = Math.max(...[...ART].map((l) => [...l].length));
 function center(line) {
-  const pad = WIDTH - [...line].length;
-  const left = Math.floor(pad / 2);
-  return " ".repeat(left) + line + " ".repeat(pad - left);
+  const pad2 = WIDTH - [...line].length;
+  const left = Math.floor(pad2 / 2);
+  return " ".repeat(left) + line + " ".repeat(pad2 - left);
 }
 var GLYPHS = {
   P: ["\u2588\u2588\u2588\u2588\u2588\u2588", "\u2588\u2588  \u2588\u2588", "\u2588\u2588\u2588\u2588\u2588\u2588", "\u2588\u2588    ", "\u2588\u2588    "],
@@ -2835,6 +2853,169 @@ function extractJsonArray(text2) {
   }
 }
 
+// src/ui/swarm-view.ts
+var RESET3 = "\x1B[0m";
+var FRAMES2 = ["\u280B", "\u2819", "\u2839", "\u2838", "\u283C", "\u2834", "\u2826", "\u2827", "\u2807", "\u280F"];
+function describeToolCall(call) {
+  const raw = call.name === "run_command" ? call.arguments.command : call.arguments.path;
+  const arg = typeof raw === "string" ? raw : "";
+  const short = arg.length > 40 ? arg.slice(0, 39) + "\u2026" : arg;
+  return short ? `${call.name} ${short}` : call.name;
+}
+var SwarmView = class {
+  constructor(leadName, opts = {}) {
+    this.leadName = leadName;
+    this.tty = opts.tty ?? (Boolean(process.stdout.isTTY) && !process.env.NO_COLOR);
+    this.color = opts.color ?? this.tty;
+    this.write = opts.sink ?? ((s) => process.stdout.write(s));
+  }
+  leadName;
+  tty;
+  color;
+  write;
+  workers = /* @__PURE__ */ new Map();
+  order = [];
+  phase = "decomposing";
+  frame = 0;
+  lastLines = 0;
+  timer;
+  start() {
+    if (!this.tty) {
+      this.write(`\u{1F419} ${t("swarm.view.header", { lead: this.leadName })} \u2014 ${t("swarm.view.decomposing")}
+`);
+      return;
+    }
+    this.flush();
+    this.timer = setInterval(() => {
+      this.frame = (this.frame + 1) % FRAMES2.length;
+      this.flush();
+    }, 110);
+    this.timer.unref?.();
+  }
+  setSubtasks(subtasks) {
+    this.phase = "running";
+    for (const s of subtasks) {
+      this.workers.set(s.id, { id: s.id, title: s.title, agent: "", status: "pending", action: "", steps: 0 });
+      this.order.push(s.id);
+    }
+    if (!this.tty) {
+      this.write(`  ${t("swarm.decomposed", { n: subtasks.length })}
+`);
+      for (const s of subtasks) this.write(`  ${s.id}: ${s.title}
+`);
+    }
+    this.flush();
+  }
+  workerStart(id, agent) {
+    const w = this.workers.get(id);
+    if (!w) return;
+    w.agent = agent;
+    w.status = "running";
+    if (!this.tty) this.write(`  \u25B6 ${id} [${agent}] ${w.title}
+`);
+    this.flush();
+  }
+  workerAction(id, action) {
+    const w = this.workers.get(id);
+    if (!w) return;
+    w.action = action;
+    this.flush();
+  }
+  workerStep(id, n) {
+    const w = this.workers.get(id);
+    if (!w) return;
+    w.steps = n;
+    this.flush();
+  }
+  workerDone(o) {
+    const w = this.workers.get(o.subtask.id);
+    if (!w) return;
+    w.status = o.finished ? "done" : "stopped";
+    w.steps = o.steps;
+    w.branch = o.branch;
+    w.action = "";
+    if (!this.tty) {
+      const tag = o.finished ? "\u2713" : "\u25A0";
+      const changes = o.committed ? t("swarm.changesCommitted") : t("swarm.noChanges");
+      this.write(`  ${tag} ${o.subtask.id} (${t("swarm.view.steps", { n: o.steps })}, ${changes})
+`);
+    }
+    this.flush();
+  }
+  merge(r) {
+    for (const w of this.workers.values()) {
+      if (w.branch === r.branch) w.merge = r.ok ? "ok" : "conflict";
+    }
+    if (!this.tty) {
+      this.write(r.ok ? `  \u2935 ${t("swarm.merged", { branch: r.branch })}
+` : `  \u2717 ${t("swarm.mergeConflict", { branch: r.branch })}
+`);
+    }
+    this.flush();
+  }
+  stop() {
+    this.phase = "done";
+    if (this.timer) {
+      clearInterval(this.timer);
+      this.timer = void 0;
+    }
+    this.flush();
+  }
+  /** Content lines of the dashboard (no cursor control). Exposed for tests. */
+  frameLines() {
+    const spin = this.dim(FRAMES2[this.frame]);
+    const lead = `\u{1F419} ${t("swarm.view.header", { lead: this.leadName })}`;
+    const lines = [];
+    if (this.phase === "decomposing") {
+      lines.push(`${spin} ${lead}`);
+      lines.push("  " + this.dim(t("swarm.view.decomposing")));
+      return lines;
+    }
+    lines.push(`${this.phase === "running" ? spin : "  "} ${lead}`);
+    lines.push("");
+    for (const id of this.order) {
+      const w = this.workers.get(id);
+      lines.push(this.row(w, spin));
+    }
+    return lines;
+  }
+  // -------------------------------------------------------------------------
+  row(w, spin) {
+    const icon = w.status === "running" ? spin : w.status === "done" ? this.c("\u2713", "32") : w.status === "stopped" ? this.c("\u25A0", "33") : this.dim("\xB7");
+    const status = this.statusLabel(w);
+    const meta = w.steps > 0 ? this.dim(" \xB7 " + (w.status === "running" ? t("swarm.view.step", { n: w.steps }) : t("swarm.view.steps", { n: w.steps }))) : "";
+    const action = w.action ? w.action : this.dim("\u2014");
+    return `  ${icon} ${pad(w.id, 4)} ${pad(status, 12)} ${pad(`[${w.agent}]`, 14)} ${action}${meta}`;
+  }
+  statusLabel(w) {
+    if (w.merge === "conflict") return this.c(t("swarm.view.conflict"), "31");
+    if (w.status === "running") return this.c(t("swarm.view.running"), "36");
+    if (w.status === "done") return this.c(t("swarm.view.done"), "32");
+    if (w.status === "stopped") return this.c(t("swarm.view.stopped"), "33");
+    return this.dim(t("swarm.view.pending"));
+  }
+  /** Redraw the block in place (TTY) by clearing the previous frame first. */
+  flush() {
+    if (!this.tty) return;
+    const lines = this.frameLines();
+    let s = "";
+    if (this.lastLines > 0) s += `\x1B[${this.lastLines}A`;
+    s += "\x1B[0J";
+    s += lines.join("\n") + "\n";
+    this.write(s);
+    this.lastLines = lines.length;
+  }
+  c(s, code) {
+    return this.color ? `\x1B[${code}m${s}${RESET3}` : s;
+  }
+  dim(s) {
+    return this.color ? `\x1B[2m${s}${RESET3}` : s;
+  }
+};
+function pad(s, n) {
+  return s.length >= n ? s : s + " ".repeat(n - s.length);
+}
+
 // src/cli/commands/swarm.ts
 async function swarm(task, opts) {
   const config = await loadConfig();
@@ -2851,15 +3032,32 @@ async function swarm(task, opts) {
     pc8.dim(t("swarm.status", { agents: resolved.map((a) => a.config.name).join(", "), workspace: process.cwd() }))
   );
   console.log(pc8.yellow(t("swarm.bypassNote") + "\n"));
-  const result = await runSwarm({
-    task,
-    workspace: process.cwd(),
-    agents: resolved,
-    allow: config.permissions.allow,
-    deny: config.permissions.deny,
-    maxSubtasks: opts.maxSubtasks ? Number(opts.maxSubtasks) : void 0,
-    events: renderSwarmEvents()
-  });
+  const view = new SwarmView(resolved[0].config.name);
+  view.start();
+  let result;
+  try {
+    result = await runSwarm({
+      task,
+      workspace: process.cwd(),
+      agents: resolved,
+      allow: config.permissions.allow,
+      deny: config.permissions.deny,
+      maxSubtasks: opts.maxSubtasks ? Number(opts.maxSubtasks) : void 0,
+      events: {
+        onDecomposed: (subtasks) => view.setSubtasks(subtasks),
+        onWorkerStart: (subtask, agentName) => view.workerStart(subtask.id, agentName),
+        onWorkerDone: (outcome) => view.workerDone(outcome),
+        onMerge: (merge) => view.merge(merge),
+        workerEvents: (subtask) => ({
+          onToolCall: (call) => view.workerAction(subtask.id, describeToolCall(call)),
+          onStep: (step) => view.workerStep(subtask.id, step)
+        })
+      }
+    });
+  } finally {
+    view.stop();
+  }
+  console.log("");
   console.log(pc8.bold("\n" + t("swarm.summary")));
   for (const o of result.outcomes) {
     const status = o.finished ? pc8.green(t("swarm.statusDone")) : pc8.yellow(t("swarm.statusIncomplete"));
@@ -2876,26 +3074,6 @@ async function swarm(task, opts) {
     console.log(pc8.green("\n" + t("swarm.allMerged")));
   }
 }
-function renderSwarmEvents() {
-  return {
-    onDecomposed(subtasks) {
-      console.log(pc8.bold(t("swarm.decomposed", { n: subtasks.length })));
-      for (const s of subtasks) console.log(pc8.dim(`  ${s.id}: ${s.title}`));
-      console.log("");
-    },
-    onWorkerStart(subtask, agentName) {
-      console.log(pc8.cyan(t("swarm.workerStart", { id: subtask.id, agent: agentName })));
-    },
-    onWorkerDone(o) {
-      const head = o.finished ? pc8.green(t("swarm.workerDone", { id: o.subtask.id })) : pc8.yellow(t("swarm.workerStopped", { id: o.subtask.id }));
-      const changes = o.committed ? t("swarm.changesCommitted") : t("swarm.noChanges");
-      console.log(head + pc8.dim(t("swarm.workerMeta", { steps: o.steps, changes })));
-    },
-    onMerge(m) {
-      console.log(m.ok ? pc8.dim(t("swarm.merged", { branch: m.branch })) : pc8.red(t("swarm.mergeConflict", { branch: m.branch })));
-    }
-  };
-}
 
 // src/cli/commands/models.ts
 import pc9 from "picocolors";
@@ -2991,12 +3169,18 @@ ${comments}` : "",
     "## Riscos e alternativas"
   ].join("\n");
 }
-async function generatePrd(issue, provider) {
+async function generatePrd(issue, provider, projectContext) {
+  const messages = [{ role: "system", content: SYSTEM }];
+  if (projectContext) {
+    messages.push({
+      role: "system",
+      content: `Project context (for grounding; do not restate verbatim):
+${projectContext}`
+    });
+  }
+  messages.push({ role: "user", content: buildPrdPrompt(issue) });
   const res = await provider.chat({
-    messages: [
-      { role: "system", content: SYSTEM },
-      { role: "user", content: buildPrdPrompt(issue) }
-    ],
+    messages,
     params: { maxTokens: 2e3, temperature: 0.2 }
   });
   const text2 = res.content.trim();
@@ -3041,6 +3225,23 @@ async function withRetry(fn, opts = {}) {
 }
 
 // src/cli/commands/cli-io.ts
+import { readFileSync, existsSync as existsSync2 } from "fs";
+import { resolve as resolve7 } from "path";
+var GUIDE_MAX = 12e3;
+function readProjectGuide(files) {
+  const parts = [];
+  for (const file of files) {
+    try {
+      const path = resolve7(process.cwd(), file);
+      if (existsSync2(path)) parts.push(`# ${file}
+${readFileSync(path, "utf8").trim()}`);
+    } catch {
+    }
+  }
+  if (parts.length === 0) return void 0;
+  const joined = parts.join("\n\n");
+  return joined.length > GUIDE_MAX ? joined.slice(0, GUIDE_MAX) + "\n\u2026(truncated)" : joined;
+}
 function numericRef(ref) {
   const num = ref.replace(/^#/, "");
   if (!/^\d+$/.test(num)) throw new Error(t("cli.invalidRef", { ref }));
@@ -3061,7 +3262,8 @@ var exec2 = promisify2(execFile);
 async function prd(issueRef, opts) {
   const issue = await loadIssue(issueRef, opts.input);
   const { provider } = resolveFreeProvider(opts.model ?? DEFAULT_PRD_MODEL);
-  const markdown = await withRetry(() => generatePrd(issue, provider));
+  const guide = readProjectGuide(["context.md"]);
+  const markdown = await withRetry(() => generatePrd(issue, provider, guide));
   if (opts.out) {
     await writeFile4(opts.out, markdown + "\n", "utf8");
     console.error(pc10.green(t("prd.wrote", { path: opts.out })));
@@ -3129,13 +3331,19 @@ function buildReviewPrompt(diff, meta) {
     "Liste os achados agrupados por severidade (\u{1F534} bug, \u{1F7E1} aten\xE7\xE3o, \u{1F7E2} sugest\xE3o)."
   ].join("\n");
 }
-async function reviewDiff(diff, meta, provider) {
+async function reviewDiff(diff, meta, provider, projectGuide) {
   if (!diff.trim()) return "_Sem altera\xE7\xF5es no diff para revisar._";
+  const messages = [{ role: "system", content: SYSTEM2 }];
+  if (projectGuide) {
+    messages.push({
+      role: "system",
+      content: `Project context and conventions to review against:
+${projectGuide}`
+    });
+  }
+  messages.push({ role: "user", content: buildReviewPrompt(diff, meta) });
   const res = await provider.chat({
-    messages: [
-      { role: "system", content: SYSTEM2 },
-      { role: "user", content: buildReviewPrompt(diff, meta) }
-    ],
+    messages,
     params: { maxTokens: 1500, temperature: 0.2 }
   });
   const text2 = res.content.trim();
@@ -3150,7 +3358,8 @@ async function review(prRef, opts) {
   const diff = await loadDiff(num, opts.input);
   const meta = await loadMeta(num, opts.input);
   const { provider } = resolveFreeProvider(opts.model ?? DEFAULT_REVIEW_MODEL);
-  const markdown = await withRetry(() => reviewDiff(diff, meta, provider));
+  const guide = readProjectGuide(["rules.md", "context.md"]);
+  const markdown = await withRetry(() => reviewDiff(diff, meta, provider, guide));
   if (opts.out) {
     await writeFile5(opts.out, markdown + "\n", "utf8");
     console.error(pc11.green(t("review.wrote", { path: opts.out })));
@@ -3174,13 +3383,13 @@ async function loadMeta(num, input) {
 import { join as join3 } from "path";
 
 // src/core/config/dotenv.ts
-import { existsSync as existsSync2, readFileSync } from "fs";
+import { existsSync as existsSync3, readFileSync as readFileSync2 } from "fs";
 function loadDotenv(paths) {
   for (const path of paths) {
-    if (!existsSync2(path)) continue;
+    if (!existsSync3(path)) continue;
     let text2;
     try {
-      text2 = readFileSync(path, "utf8");
+      text2 = readFileSync2(path, "utf8");
     } catch {
       continue;
     }