@gaberrb/polypus 0.4.2 → 0.4.3

package/dist/index.js

@@ -285,7 +285,11 @@ var en = {
   "welcome.firstRun": "No agents configured yet \u2014 let's set you up.",
   // agent system prompt
   "prompt.language": "Communicate with the user in {language}.",
-  "prompt.projectInstructions": "Project-specific operating instructions follow, loaded from `.poly/agents.md`. Treat them as authoritative for how to work in THIS repo. Paths they reference (e.g. skills/*.md, ../context.md, ../rules.md) are relative to the `.poly/` directory \u2014 read those files when relevant before acting:"
+  "prompt.projectInstructions": "Project-specific operating instructions follow, loaded from `.poly/agents.md`. Treat them as authoritative for how to work in THIS repo. Paths they reference (e.g. skills/*.md, ../context.md, ../rules.md) are relative to the `.poly/` directory \u2014 read those files when relevant before acting:",
+  // @-mentions
+  "mentions.injectedHeader": "Referenced files (@-mentions)",
+  "mentions.dirHeader": "@{path} (directory listing)",
+  "mentions.notFound": "(could not resolve @{path}: not found or outside the allow-list)"
 };
 var ptBR = {
   "common.default": "padr\xE3o",
@@ -478,6 +482,10 @@ var ptBR = {
   "wizard.keyPrompt": "Chave de API (armazenada em texto puro no arquivo de config)",
   "prompt.language": "Comunique-se com o usu\xE1rio em {language}.",
   "prompt.projectInstructions": "Seguem instru\xE7\xF5es operacionais espec\xEDficas do projeto, carregadas de `.poly/agents.md`. Trate-as como autoritativas para trabalhar NESTE reposit\xF3rio. Os caminhos que elas citam (ex.: skills/*.md, ../context.md, ../rules.md) s\xE3o relativos \xE0 pasta `.poly/` \u2014 leia esses arquivos quando relevante antes de agir:",
+  // @-mentions
+  "mentions.injectedHeader": "Arquivos referenciados (@-mentions)",
+  "mentions.dirHeader": "@{path} (conte\xFAdo do diret\xF3rio)",
+  "mentions.notFound": "(n\xE3o foi poss\xEDvel resolver @{path}: n\xE3o encontrado ou fora da allow-list)",
   "models.fetching": "Buscando modelos do OpenRouter\u2026",
   "models.fetchError": "N\xE3o foi poss\xEDvel buscar modelos: {msg}",
   "models.none": "Nenhum modelo corresponde aos filtros.",
@@ -1974,6 +1982,56 @@ ${guidance}`;
   return { finished: false, reason: "maxsteps", steps: maxSteps, messages, usage };
 }
 
+// src/core/context/mentions.ts
+import { readdir as readdir4, readFile as readFile7, stat as stat2 } from "fs/promises";
+import { resolve as resolve8 } from "path";
+var MAX_FILE_CHARS = 1e4;
+var MENTION_RE = /(?:^|\s)@([\w./-]+)/g;
+async function resolveMentions(task, policy) {
+  const tokens2 = [...task.matchAll(MENTION_RE)].map((m) => m[1]);
+  const unique = [...new Set(tokens2)];
+  if (unique.length === 0) return { task, injected: [] };
+  const blocks = [];
+  const injected = [];
+  for (const token of unique) {
+    const decision = checkPath(policy, token);
+    if (!decision.allowed) {
+      blocks.push(`## @${token}
+${t("mentions.notFound", { path: token })}`);
+      continue;
+    }
+    const abs = resolve8(policy.workspace, token);
+    try {
+      const info = await stat2(abs);
+      if (info.isDirectory()) {
+        const entries = await readdir4(abs, { withFileTypes: true });
+        const listing = entries.map((e) => e.isDirectory() ? `${e.name}/` : e.name).sort().join("\n");
+        blocks.push(`## ${t("mentions.dirHeader", { path: decision.rel })}
+${listing || "(empty)"}`);
+        injected.push(decision.rel);
+      } else {
+        const raw = await readFile7(abs, "utf8");
+        const content = raw.length > MAX_FILE_CHARS ? raw.slice(0, MAX_FILE_CHARS) + "\n\u2026[truncated]" : raw;
+        blocks.push(`## @${decision.rel}
+\`\`\`
+${content}
+\`\`\``);
+        injected.push(decision.rel);
+      }
+    } catch {
+      blocks.push(`## @${token}
+${t("mentions.notFound", { path: token })}`);
+    }
+  }
+  if (injected.length === 0) return { task, injected: [] };
+  const augmented = `${task}
+
+--- ${t("mentions.injectedHeader")} ---
+
+${blocks.join("\n\n")}`;
+  return { task: augmented, injected };
+}
+
 // src/ui/repl.ts
 import pc6 from "picocolors";
 
@@ -2612,10 +2670,10 @@ async function readLineTTY(prompt) {
   stdin.resume();
   stdin.on("data", onData);
   try {
-    const line = await new Promise((resolve9) => {
-      rl.question(prompt).then(resolve9, () => resolve9(null));
-      rl.on("SIGINT", () => resolve9(null));
-      rl.on("close", () => resolve9(null));
+    const line = await new Promise((resolve10) => {
+      rl.question(prompt).then(resolve10, () => resolve10(null));
+      rl.on("SIGINT", () => resolve10(null));
+      rl.on("close", () => resolve10(null));
     });
     return line === null ? null : store.expand(line);
   } finally {
@@ -3264,6 +3322,15 @@ async function run(task, opts) {
   await startRepl(ctx);
 }
 async function executeTask(task, resolved, workspace, session) {
+  const mention = await resolveMentions(task, {
+    workspace,
+    allow: session.allow,
+    deny: session.deny
+  });
+  if (mention.injected.length > 0) {
+    task = mention.task;
+    console.log(pc8.dim(`\u21B3 @ ${mention.injected.join(", ")}`));
+  }
   const spinner3 = new Spinner();
   const controller = new AbortController();
   const cancel2 = listenForCancel(controller);
@@ -3742,7 +3809,7 @@ async function resolveOpenRouterKey() {
 }
 
 // src/cli/commands/prd.ts
-import { writeFile as writeFile5, readFile as readFile7 } from "fs/promises";
+import { writeFile as writeFile5, readFile as readFile8 } from "fs/promises";
 import { execFile } from "child_process";
 import { promisify as promisify2 } from "util";
 import pc11 from "picocolors";
@@ -3834,13 +3901,13 @@ async function withRetry(fn, opts = {}) {
 
 // src/cli/commands/cli-io.ts
 import { readFileSync, existsSync as existsSync2 } from "fs";
-import { resolve as resolve8 } from "path";
+import { resolve as resolve9 } from "path";
 var GUIDE_MAX = 12e3;
 function readProjectGuide(files) {
   const parts = [];
   for (const file of files) {
     try {
-      const path = resolve8(process.cwd(), file);
+      const path = resolve9(process.cwd(), file);
       if (existsSync2(path)) parts.push(`# ${file}
 ${readFileSync(path, "utf8").trim()}`);
     } catch {
@@ -3881,7 +3948,7 @@ async function prd(issueRef, opts) {
 }
 async function loadIssue(issueRef, input) {
   if (input) {
-    const raw = input === "-" ? await readStdin() : await readFile7(input, "utf8");
+    const raw = input === "-" ? await readStdin() : await readFile8(input, "utf8");
     return normalize2(JSON.parse(stripBom(raw)));
   }
   const num = numericRef(issueRef);
@@ -3900,7 +3967,7 @@ function normalize2(raw) {
 }
 
 // src/cli/commands/review.ts
-import { writeFile as writeFile6, readFile as readFile8 } from "fs/promises";
+import { writeFile as writeFile6, readFile as readFile9 } from "fs/promises";
 import { execFile as execFile2 } from "child_process";
 import { promisify as promisify3 } from "util";
 import pc12 from "picocolors";
@@ -3976,7 +4043,7 @@ async function review(prRef, opts) {
   }
 }
 async function loadDiff(num, input) {
-  if (input) return input === "-" ? readStdin() : readFile8(input, "utf8");
+  if (input) return input === "-" ? readStdin() : readFile9(input, "utf8");
   const { stdout: stdout2 } = await exec3("gh", ["pr", "diff", num]);
   return stdout2;
 }