@gaberrb/polypus 0.4.1 → 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.",
@@ -1435,6 +1443,116 @@ function clamp(s) {
   return s.length > MAX_OUTPUT ? s.slice(0, MAX_OUTPUT) + "\n\u2026[truncated]" : s;
 }
 
+// src/core/tools/search-file.ts
+import { readdir as readdir2, readFile as readFile4, stat } from "fs/promises";
+import { join as join2, resolve as resolve5 } from "path";
+import { z as z6 } from "zod";
+var Args5 = z6.object({
+  query: z6.string().min(1),
+  path: z6.string().optional(),
+  glob: z6.string().optional(),
+  max_results: z6.number().int().positive().max(1e3).optional()
+});
+var DEFAULT_MAX_RESULTS = 50;
+var MAX_OUTPUT2 = 2e4;
+var MAX_FILE_BYTES = 2e6;
+var SNIPPET_CHARS = 200;
+var NUL = String.fromCharCode(0);
+var SKIP_DIRS = /* @__PURE__ */ new Set(["node_modules", ".git", "dist", ".next", "coverage", ".turbo"]);
+var searchTool = {
+  mutating: false,
+  spec: {
+    name: "search",
+    description: "Search file contents by regular expression across the workspace (like grep/ripgrep). Returns matches as 'path:line: snippet'. Respects the allow/deny-list and skips node_modules/.git. Use this to find where a symbol is defined or used instead of reading files blindly.",
+    parameters: {
+      type: "object",
+      properties: {
+        query: { type: "string", description: "Regular expression to match against each line" },
+        path: { type: "string", description: "Workspace-relative directory to search in (default '.')" },
+        glob: {
+          type: "string",
+          description: "Optional glob to limit files, e.g. 'src/**/*.ts'"
+        },
+        max_results: {
+          type: "number",
+          description: `Maximum number of matches to return (default ${DEFAULT_MAX_RESULTS})`
+        }
+      },
+      required: ["query"]
+    }
+  },
+  async run(rawArgs, ctx) {
+    const parsed = Args5.safeParse(rawArgs);
+    if (!parsed.success) return { ok: false, output: "Invalid args: 'query' is required." };
+    const { query, path = ".", glob, max_results } = parsed.data;
+    let regex;
+    try {
+      regex = new RegExp(query);
+    } catch (err) {
+      return { ok: false, output: `Invalid regular expression: ${err.message}` };
+    }
+    if (path !== ".") {
+      const decision = ctx.permissions.authorizeRead(path);
+      if (!decision.allowed) return { ok: false, output: `Search denied: ${decision.reason}` };
+    }
+    const globRe = glob ? globToRegExp(glob) : void 0;
+    const limit = max_results ?? DEFAULT_MAX_RESULTS;
+    const root = resolve5(ctx.workspace, path);
+    const matches = [];
+    let truncated = false;
+    const walk = async (dir) => {
+      if (matches.length >= limit) return;
+      let entries;
+      try {
+        entries = await readdir2(dir, { withFileTypes: true });
+      } catch {
+        return;
+      }
+      for (const entry of entries) {
+        if (matches.length >= limit) return;
+        const abs = join2(dir, entry.name);
+        const rel = toPosix(abs.slice(ctx.workspace.length + 1));
+        if (entry.isDirectory()) {
+          if (SKIP_DIRS.has(entry.name)) continue;
+          await walk(abs);
+          continue;
+        }
+        if (!entry.isFile()) continue;
+        if (globRe && !globRe.test(rel)) continue;
+        if (!ctx.permissions.authorizeRead(rel).allowed) continue;
+        try {
+          const info = await stat(abs);
+          if (info.size > MAX_FILE_BYTES) continue;
+          const content = await readFile4(abs, "utf8");
+          if (content.includes(NUL)) continue;
+          const lines = content.split("\n");
+          for (let i = 0; i < lines.length; i++) {
+            if (matches.length >= limit) {
+              truncated = true;
+              return;
+            }
+            if (regex.test(lines[i])) {
+              const snippet = lines[i].trim().slice(0, SNIPPET_CHARS);
+              matches.push(`${rel}:${i + 1}: ${snippet}`);
+            }
+          }
+        } catch {
+        }
+      }
+    };
+    await walk(root);
+    if (matches.length === 0) {
+      return { ok: true, output: `No matches for /${query}/${glob ? ` in ${glob}` : ""}.` };
+    }
+    const header = `${matches.length}${truncated ? "+" : ""} match(es) for /${query}/:`;
+    const body = [header, ...matches].join("\n");
+    return {
+      ok: true,
+      output: body.length > MAX_OUTPUT2 ? body.slice(0, MAX_OUTPUT2) + "\n\u2026[truncated]" : body
+    };
+  }
+};
+
 // src/core/tools/types.ts
 var FINISH_TOOL = {
   name: "finish",
@@ -1450,9 +1568,9 @@ var FINISH_TOOL = {
 
 // src/core/tools/write-file.ts
 import { mkdir as mkdir2, writeFile as writeFile3 } from "fs/promises";
-import { dirname, resolve as resolve5 } from "path";
-import { z as z6 } from "zod";
-var Args5 = z6.object({ path: z6.string().min(1), content: z6.string() });
+import { dirname, resolve as resolve6 } from "path";
+import { z as z7 } from "zod";
+var Args6 = z7.object({ path: z7.string().min(1), content: z7.string() });
 var writeFileTool = {
   mutating: true,
   spec: {
@@ -1468,7 +1586,7 @@ var writeFileTool = {
     }
   },
   async run(rawArgs, ctx) {
-    const args = Args5.safeParse(rawArgs);
+    const args = Args6.safeParse(rawArgs);
     if (!args.success) {
       const got = Object.keys(rawArgs ?? {});
       return {
@@ -1480,7 +1598,7 @@ var writeFileTool = {
     const decision = await ctx.permissions.authorizeWrite(args.data.path, preview);
     if (!decision.allowed) return { ok: false, output: `Write denied: ${decision.reason}` };
     try {
-      const abs = resolve5(ctx.workspace, args.data.path);
+      const abs = resolve6(ctx.workspace, args.data.path);
       await mkdir2(dirname(abs), { recursive: true });
       await writeFile3(abs, args.data.content, "utf8");
       const lines = args.data.content.split("\n").length;
@@ -1499,6 +1617,7 @@ function previewContent(content) {
 var TOOLS = {
   [readFileTool.spec.name]: readFileTool,
   [listDirTool.spec.name]: listDirTool,
+  [searchTool.spec.name]: searchTool,
   [writeFileTool.spec.name]: writeFileTool,
   [editFileTool.spec.name]: editFileTool,
   [runCommandTool.spec.name]: runCommandTool
@@ -1511,8 +1630,8 @@ function getTool(name) {
 }
 
 // src/core/agent/correction.ts
-import { readFile as readFile4, readdir as readdir2 } from "fs/promises";
-import { dirname as dirname2, resolve as resolve6 } from "path";
+import { readFile as readFile5, readdir as readdir3 } from "fs/promises";
+import { dirname as dirname2, resolve as resolve7 } from "path";
 function truncationGuidance(toolName) {
   const fileHint = toolName === "write_file" || toolName === "edit_file" ? " Write large files in parts: create the file with the first chunk via write_file, then append the rest with edit_file in the next steps." : "";
   return [
@@ -1612,7 +1731,7 @@ ${text2}` : null;
 }
 async function readWorkspaceFile(workspace, path) {
   try {
-    return await readFile4(resolve6(workspace, path), "utf8");
+    return await readFile5(resolve7(workspace, path), "utf8");
   } catch {
     return null;
   }
@@ -1670,11 +1789,11 @@ async function occurrenceLines(workspace, path, search) {
   return out;
 }
 async function listNearest(workspace, path) {
-  let dir = dirname2(resolve6(workspace, path));
+  let dir = dirname2(resolve7(workspace, path));
   for (let i = 0; i < 8; i++) {
     try {
-      const entries = await readdir2(dir, { withFileTypes: true });
-      const rel = dir === resolve6(workspace) ? "." : dir;
+      const entries = await readdir3(dir, { withFileTypes: true });
+      const rel = dir === resolve7(workspace) ? "." : dir;
       const names = entries.slice(0, 40).map((e) => e.isDirectory() ? `${e.name}/` : e.name);
       return `${rel}:
   ${names.join("  ") || "(empty)"}`;
@@ -1696,14 +1815,14 @@ function formatSchema(spec) {
 }
 
 // src/core/agent/project-context.ts
-import { readFile as readFile5 } from "fs/promises";
-import { join as join2 } from "path";
-var INSTRUCTION_FILES = [join2(".poly", "agents.md"), "AGENTS.md"];
+import { readFile as readFile6 } from "fs/promises";
+import { join as join3 } from "path";
+var INSTRUCTION_FILES = [join3(".poly", "agents.md"), "AGENTS.md"];
 var MAX_CHARS2 = 8e3;
 async function loadProjectInstructions(workspace) {
   for (const rel of INSTRUCTION_FILES) {
     try {
-      const raw = (await readFile5(join2(workspace, rel), "utf8")).trim();
+      const raw = (await readFile6(join3(workspace, rel), "utf8")).trim();
       if (!raw) continue;
       return raw.length > MAX_CHARS2 ? raw.slice(0, MAX_CHARS2) + "\n\u2026(truncated)" : raw;
     } catch {
@@ -1863,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";
 
@@ -2501,10 +2670,10 @@ async function readLineTTY(prompt) {
   stdin.resume();
   stdin.on("data", onData);
   try {
-    const line = await new Promise((resolve8) => {
-      rl.question(prompt).then(resolve8, () => resolve8(null));
-      rl.on("SIGINT", () => resolve8(null));
-      rl.on("close", () => resolve8(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 {
@@ -2649,7 +2818,7 @@ import pc7 from "picocolors";
 // src/core/git/worktree.ts
 import { mkdtemp } from "fs/promises";
 import { tmpdir } from "os";
-import { join as join3 } from "path";
+import { join as join4 } from "path";
 import { simpleGit } from "simple-git";
 async function ensureRepo(workspace) {
   const git = simpleGit(workspace);
@@ -2670,7 +2839,7 @@ async function identityArgs(git) {
 }
 async function createWorktree(git, label) {
   const branch = `polypus/${label}-${Date.now().toString(36)}`;
-  const path = await mkdtemp(join3(tmpdir(), "polypus-wt-"));
+  const path = await mkdtemp(join4(tmpdir(), "polypus-wt-"));
   await git.raw(["worktree", "add", "-b", branch, path, "HEAD"]);
   return { path, branch };
 }
@@ -3153,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);
@@ -3288,7 +3466,7 @@ import pc9 from "picocolors";
 
 // src/core/scaffold/init.ts
 import { mkdir as mkdir3, writeFile as writeFile4, access } from "fs/promises";
-import { dirname as dirname3, join as join4 } from "path";
+import { dirname as dirname3, join as join5 } from "path";
 
 // src/core/scaffold/templates.ts
 function polyTemplates(locale) {
@@ -3531,7 +3709,7 @@ async function scaffoldPoly(workspace, opts) {
   const skipped = [];
   for (const [rel, content] of Object.entries(templates)) {
     const display = `.poly/${rel}`;
-    const abs = join4(workspace, ".poly", ...rel.split("/"));
+    const abs = join5(workspace, ".poly", ...rel.split("/"));
     if (!opts.force && await exists(abs)) {
       skipped.push(display);
       continue;
@@ -3631,7 +3809,7 @@ async function resolveOpenRouterKey() {
 }
 
 // src/cli/commands/prd.ts
-import { writeFile as writeFile5, readFile as readFile6 } 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";
@@ -3723,13 +3901,13 @@ async function withRetry(fn, opts = {}) {
 
 // src/cli/commands/cli-io.ts
 import { readFileSync, existsSync as existsSync2 } from "fs";
-import { resolve as resolve7 } 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 = resolve7(process.cwd(), file);
+      const path = resolve9(process.cwd(), file);
       if (existsSync2(path)) parts.push(`# ${file}
 ${readFileSync(path, "utf8").trim()}`);
     } catch {
@@ -3770,7 +3948,7 @@ async function prd(issueRef, opts) {
 }
 async function loadIssue(issueRef, input) {
   if (input) {
-    const raw = input === "-" ? await readStdin() : await readFile6(input, "utf8");
+    const raw = input === "-" ? await readStdin() : await readFile8(input, "utf8");
     return normalize2(JSON.parse(stripBom(raw)));
   }
   const num = numericRef(issueRef);
@@ -3789,7 +3967,7 @@ function normalize2(raw) {
 }
 
 // src/cli/commands/review.ts
-import { writeFile as writeFile6, readFile as readFile7 } 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";
@@ -3865,7 +4043,7 @@ async function review(prRef, opts) {
   }
 }
 async function loadDiff(num, input) {
-  if (input) return input === "-" ? readStdin() : readFile7(input, "utf8");
+  if (input) return input === "-" ? readStdin() : readFile9(input, "utf8");
   const { stdout: stdout2 } = await exec3("gh", ["pr", "diff", num]);
   return stdout2;
 }
@@ -3877,7 +4055,7 @@ async function loadMeta(num, input) {
 }
 
 // src/cli/index.ts
-import { join as join5 } from "path";
+import { join as join6 } from "path";
 
 // src/core/config/dotenv.ts
 import { existsSync as existsSync3, readFileSync as readFileSync2 } from "fs";
@@ -3946,7 +4124,7 @@ function buildProgram() {
 }
 async function main() {
   try {
-    loadDotenv([join5(configDir(), ".env"), join5(process.cwd(), ".env")]);
+    loadDotenv([join6(configDir(), ".env"), join6(process.cwd(), ".env")]);
     await resolveLocale();
     await buildProgram().parseAsync(process.argv);
   } catch (err) {