@astralyn/pi 1.0.4 → 1.1.0

package/dist/extensions/search/workspace.js

@@ -0,0 +1,245 @@
+/**
+ * The local side of code_search: the planner runs remotely and names commands against a virtual
+ * `/codebase`; this module executes them read-only, confined to one directory.
+ *
+ * Security boundary: every model-supplied path must resolve inside the root, both lexically and
+ * after following symlinks. Only `rg`, `readfile`, and `tree` exist; there is no shell and no write.
+ */
+import { existsSync, readdirSync, readFileSync, realpathSync, statSync } from "node:fs";
+import { basename, dirname, isAbsolute, join, relative, resolve, sep } from "node:path";
+import { createGrepToolDefinition } from "../../core/tools/grep.js";
+export const VIRTUAL_ROOT = "/codebase";
+/** Per-command result cap. Results are sent back to Devin, whose request limit is ~320KB. */
+const RESULT_MAX_LINES = 40;
+const LINE_MAX_CHARS = 200;
+/** The repo map's share of that limit; deeper trees fall back to shallower ones until they fit. */
+const MAP_MAX_BYTES = 64 * 1024;
+const MAP_MAX_DEPTH = 3;
+const NOISE_DIRS = [
+    ".git",
+    "node_modules",
+    "dist",
+    "build",
+    "out",
+    "target",
+    "coverage",
+    ".venv",
+    "venv",
+    "__pycache__",
+    ".cache",
+    ".next",
+    ".nuxt",
+    ".turbo",
+    ".idea",
+    "vendor",
+    "third_party",
+];
+/** Pi's own grep tool: gitignore-aware, with its bundled ripgrep. It takes one glob, so only the first include applies. */
+export function piGrep(root) {
+    const tool = createGrepToolDefinition(root);
+    const execute = tool.execute;
+    return async (pattern, path, glob, signal) => {
+        const result = await execute("code_search", { pattern, path, glob, limit: RESULT_MAX_LINES }, signal);
+        return result.content?.find((c) => c.type === "text")?.text ?? "";
+    };
+}
+function within(base, path) {
+    const rel = relative(base, path);
+    return rel === "" || (!rel.startsWith(`..${sep}`) && rel !== ".." && !isAbsolute(rel));
+}
+/** realpath of the longest existing prefix, so a not-yet-existing path still resolves through symlinks. */
+function realpathOf(path) {
+    let current = resolve(path);
+    const tail = [];
+    while (!existsSync(current)) {
+        const parent = dirname(current);
+        if (parent === current)
+            return resolve(path);
+        tail.unshift(basename(current));
+        current = parent;
+    }
+    try {
+        return resolve(realpathSync(current), ...tail);
+    }
+    catch {
+        return resolve(path);
+    }
+}
+/** Whether `path` lies inside `root`, both as written and after following symlinks. */
+export function isInside(root, path) {
+    const abs = resolve(path);
+    return within(resolve(root), abs) && within(realpathOf(root), realpathOf(abs));
+}
+function capLines(text) {
+    const lines = text.split("\n");
+    const kept = lines
+        .slice(0, RESULT_MAX_LINES)
+        .map((l) => (l.length > LINE_MAX_CHARS ? l.slice(0, LINE_MAX_CHARS) : l));
+    if (lines.length > RESULT_MAX_LINES)
+        kept.push(`... (${lines.length - RESULT_MAX_LINES} more lines)`);
+    return kept.join("\n");
+}
+export class Workspace {
+    constructor(root, grep = piGrep(root)) {
+        this.root = resolve(root);
+        this.grep = grep;
+        this.noise = new Set([...NOISE_DIRS, ...gitignoredNames(this.root)]);
+    }
+    contains(path) {
+        return isInside(this.root, path);
+    }
+    /** `/codebase/src/a.ts` or `src/a.ts` → an absolute path inside the root, or null. */
+    toReal(virtual) {
+        if (typeof virtual !== "string" || !virtual)
+            return null;
+        const normalized = virtual.replace(/\\/g, "/");
+        let rel;
+        if (normalized === VIRTUAL_ROOT || normalized.startsWith(`${VIRTUAL_ROOT}/`)) {
+            rel = normalized.slice(VIRTUAL_ROOT.length);
+        }
+        else if (isAbsolute(virtual) || normalized.startsWith("/") || /^[a-zA-Z]:/.test(normalized)) {
+            // A drive-letter path is another drive on Windows and a stray filename elsewhere; refuse it everywhere.
+            return null;
+        }
+        else {
+            rel = normalized;
+        }
+        const candidate = resolve(this.root, rel.replace(/^\/+/, ""));
+        return this.contains(candidate) ? candidate : null;
+    }
+    toVirtual(real) {
+        const rel = relative(this.root, real).split(sep).join("/");
+        return rel ? `${VIRTUAL_ROOT}/${rel}` : VIRTUAL_ROOT;
+    }
+    /** The deepest tree of the root, up to three levels, that fits the map budget. */
+    repoMap() {
+        for (let depth = MAP_MAX_DEPTH; depth > 1; depth--) {
+            const tree = this.renderTree(this.root, depth);
+            if (Buffer.byteLength(tree) <= MAP_MAX_BYTES)
+                return { depth, tree };
+        }
+        return { depth: 1, tree: this.renderTree(this.root, 1).slice(0, MAP_MAX_BYTES / 4) };
+    }
+    async run(command, signal) {
+        switch (command?.type) {
+            case "rg":
+                return this.rg(command, signal);
+            case "readfile":
+                return this.readfile(command);
+            case "tree":
+                return this.tree(command);
+            default:
+                return `Error: unknown command type '${String(command?.type)}'; use rg, readfile, or tree`;
+        }
+    }
+    async rg(command, signal) {
+        if (typeof command.pattern !== "string" || !command.pattern)
+            return "Error: missing pattern";
+        const real = this.toReal(command.path ?? VIRTUAL_ROOT);
+        if (!real)
+            return `Error: path outside project root: ${String(command.path)}`;
+        if (!existsSync(real))
+            return `Error: path does not exist: ${String(command.path)}`;
+        const include = Array.isArray(command.include) && typeof command.include[0] === "string" ? command.include[0] : undefined;
+        let raw;
+        try {
+            raw = await this.grep(command.pattern, real, include, signal);
+        }
+        catch (e) {
+            return `Error: ${e instanceof Error ? e.message : String(e)}`;
+        }
+        // Pi's grep prints paths relative to the searched directory, or the bare file name for a file.
+        const isFile = statSync(real).isFile();
+        const base = this.toVirtual(real);
+        const remapped = raw
+            .split("\n")
+            .map((line) => {
+            const m = line.match(/^(.+?):(\d+):(.*)$/);
+            if (!m)
+                return line;
+            return `${isFile ? base : `${base}/${m[1].replace(/\\/g, "/")}`}:${m[2]}:${m[3]}`;
+        })
+            .join("\n");
+        return capLines(remapped.trim() || "(no matches)");
+    }
+    readfile(command) {
+        const real = this.toReal(command.file);
+        if (!real)
+            return `Error: path outside project root: ${String(command.file)}`;
+        let content;
+        try {
+            if (!statSync(real).isFile())
+                return `Error: not a file: ${String(command.file)}`;
+            content = readFileSync(real, "utf-8");
+        }
+        catch {
+            return `Error: file not found: ${String(command.file)}`;
+        }
+        const lines = content.split("\n");
+        const start = Math.max(1, Number(command.start_line) || 1);
+        const end = Math.min(lines.length, Number(command.end_line) || lines.length);
+        return capLines(lines
+            .slice(start - 1, end)
+            .map((line, i) => `${start + i}:${line}`)
+            .join("\n"));
+    }
+    tree(command) {
+        const real = this.toReal(command.path ?? VIRTUAL_ROOT);
+        if (!real)
+            return `Error: path outside project root: ${String(command.path)}`;
+        try {
+            if (!statSync(real).isDirectory())
+                return `Error: not a directory: ${String(command.path)}`;
+        }
+        catch {
+            return `Error: directory not found: ${String(command.path)}`;
+        }
+        const levels = Math.min(Math.max(Number(command.levels) || 2, 1), 4);
+        return capLines(this.renderTree(real, levels));
+    }
+    renderTree(dir, maxDepth) {
+        const lines = [this.toVirtual(dir)];
+        const walk = (current, depth, prefix) => {
+            if (depth >= maxDepth)
+                return;
+            let entries;
+            try {
+                entries = readdirSync(current)
+                    .filter((name) => !this.noise.has(name))
+                    .sort();
+            }
+            catch {
+                return;
+            }
+            entries.forEach((name, i) => {
+                const last = i === entries.length - 1;
+                lines.push(`${prefix}${last ? "└── " : "├── "}${name}`);
+                const path = join(current, name);
+                let isDir = false;
+                try {
+                    isDir = statSync(path).isDirectory() && this.contains(path);
+                }
+                catch { }
+                if (isDir)
+                    walk(path, depth + 1, `${prefix}${last ? "    " : "│   "}`);
+            });
+        };
+        walk(dir, 0, "");
+        return lines.join("\n");
+    }
+}
+/** Plain directory names from `.gitignore`; patterns with slashes or globs are left to grep. */
+function gitignoredNames(root) {
+    let text;
+    try {
+        text = readFileSync(join(root, ".gitignore"), "utf-8");
+    }
+    catch {
+        return [];
+    }
+    return text
+        .split("\n")
+        .map((line) => line.trim().replace(/^\/+|\/+$/g, ""))
+        .filter((name) => name && !name.startsWith("#") && !name.startsWith("!") && !/[/*?[\]]/.test(name));
+}
+//# sourceMappingURL=workspace.js.map

package/dist/extensions/search/workspace.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"workspace.js","sourceRoot":"","sources":["../../../src/extensions/search/workspace.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AACH,OAAO,EAAE,UAAU,EAAE,WAAW,EAAE,YAAY,EAAE,YAAY,EAAE,QAAQ,EAAE,MAAM,SAAS,CAAC;AACxF,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,UAAU,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,GAAG,EAAE,MAAM,WAAW,CAAC;AACxF,OAAO,EAAE,wBAAwB,EAAE,MAAM,0BAA0B,CAAC;AAEpE,MAAM,CAAC,MAAM,YAAY,GAAG,WAAW,CAAC;AAExC,6FAA6F;AAC7F,MAAM,gBAAgB,GAAG,EAAE,CAAC;AAC5B,MAAM,cAAc,GAAG,GAAG,CAAC;AAC3B,mGAAmG;AACnG,MAAM,aAAa,GAAG,EAAE,GAAG,IAAI,CAAC;AAChC,MAAM,aAAa,GAAG,CAAC,CAAC;AAExB,MAAM,UAAU,GAAG;IAClB,MAAM;IACN,cAAc;IACd,MAAM;IACN,OAAO;IACP,KAAK;IACL,QAAQ;IACR,UAAU;IACV,OAAO;IACP,MAAM;IACN,aAAa;IACb,QAAQ;IACR,OAAO;IACP,OAAO;IACP,QAAQ;IACR,OAAO;IACP,QAAQ;IACR,aAAa;CACb,CAAC;AAKF,2HAA2H;AAC3H,MAAM,UAAU,MAAM,CAAC,IAAY;IAClC,MAAM,IAAI,GAAG,wBAAwB,CAAC,IAAI,CAAC,CAAC;IAC5C,MAAM,OAAO,GAAG,IAAI,CAAC,OAI8C,CAAC;IACpE,OAAO,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE;QAC5C,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,aAAa,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,gBAAgB,EAAE,EAAE,MAAM,CAAC,CAAC;QACtG,OAAO,MAAM,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,EAAE,IAAI,IAAI,EAAE,CAAC;IACnE,CAAC,CAAC;AACH,CAAC;AAED,SAAS,MAAM,CAAC,IAAY,EAAE,IAAY;IACzC,MAAM,GAAG,GAAG,QAAQ,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;IACjC,OAAO,GAAG,KAAK,EAAE,IAAI,CAAC,CAAC,GAAG,CAAC,UAAU,CAAC,KAAK,GAAG,EAAE,CAAC,IAAI,GAAG,KAAK,IAAI,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC;AACxF,CAAC;AAED,2GAA2G;AAC3G,SAAS,UAAU,CAAC,IAAY;IAC/B,IAAI,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC5B,MAAM,IAAI,GAAa,EAAE,CAAC;IAC1B,OAAO,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;QAC7B,MAAM,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;QAChC,IAAI,MAAM,KAAK,OAAO;YAAE,OAAO,OAAO,CAAC,IAAI,CAAC,CAAC;QAC7C,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC;QAChC,OAAO,GAAG,MAAM,CAAC;IAClB,CAAC;IACD,IAAI,CAAC;QACJ,OAAO,OAAO,CAAC,YAAY,CAAC,OAAO,CAAC,EAAE,GAAG,IAAI,CAAC,CAAC;IAChD,CAAC;IAAC,MAAM,CAAC;QACR,OAAO,OAAO,CAAC,IAAI,CAAC,CAAC;IACtB,CAAC;AACF,CAAC;AAED,uFAAuF;AACvF,MAAM,UAAU,QAAQ,CAAC,IAAY,EAAE,IAAY;IAClD,MAAM,GAAG,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC1B,OAAO,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,GAAG,CAAC,IAAI,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC;AAChF,CAAC;AAED,SAAS,QAAQ,CAAC,IAAY;IAC7B,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAC/B,MAAM,IAAI,GAAG,KAAK;SAChB,KAAK,CAAC,CAAC,EAAE,gBAAgB,CAAC;SAC1B,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,MAAM,GAAG,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAC3E,IAAI,KAAK,CAAC,MAAM,GAAG,gBAAgB;QAAE,IAAI,CAAC,IAAI,CAAC,QAAQ,KAAK,CAAC,MAAM,GAAG,gBAAgB,cAAc,CAAC,CAAC;IACtG,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AACxB,CAAC;AAaD,MAAM,OAAO,SAAS;IAKrB,YAAY,IAAY,EAAE,OAAa,MAAM,CAAC,IAAI,CAAC;QAClD,IAAI,CAAC,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;QAC1B,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,KAAK,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,UAAU,EAAE,GAAG,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACtE,CAAC;IAED,QAAQ,CAAC,IAAY;QACpB,OAAO,QAAQ,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;IAClC,CAAC;IAED,sFAAsF;IACtF,MAAM,CAAC,OAAgB;QACtB,IAAI,OAAO,OAAO,KAAK,QAAQ,IAAI,CAAC,OAAO;YAAE,OAAO,IAAI,CAAC;QACzD,MAAM,UAAU,GAAG,OAAO,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;QAC/C,IAAI,GAAW,CAAC;QAChB,IAAI,UAAU,KAAK,YAAY,IAAI,UAAU,CAAC,UAAU,CAAC,GAAG,YAAY,GAAG,CAAC,EAAE,CAAC;YAC9E,GAAG,GAAG,UAAU,CAAC,KAAK,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;QAC7C,CAAC;aAAM,IAAI,UAAU,CAAC,OAAO,CAAC,IAAI,UAAU,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,YAAY,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC;YAC/F,wGAAwG;YACxG,OAAO,IAAI,CAAC;QACb,CAAC;aAAM,CAAC;YACP,GAAG,GAAG,UAAU,CAAC;QAClB,CAAC;QACD,MAAM,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC,IAAI,EAAE,GAAG,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,CAAC;QAC9D,OAAO,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC;IACpD,CAAC;IAED,SAAS,CAAC,IAAY;QACrB,MAAM,GAAG,GAAG,QAAQ,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC3D,OAAO,GAAG,CAAC,CAAC,CAAC,GAAG,YAAY,IAAI,GAAG,EAAE,CAAC,CAAC,CAAC,YAAY,CAAC;IACtD,CAAC;IAED,kFAAkF;IAClF,OAAO;QACN,KAAK,IAAI,KAAK,GAAG,aAAa,EAAE,KAAK,GAAG,CAAC,EAAE,KAAK,EAAE,EAAE,CAAC;YACpD,MAAM,IAAI,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;YAC/C,IAAI,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,aAAa;gBAAE,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;QACtE,CAAC;QACD,OAAO,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,aAAa,GAAG,CAAC,CAAC,EAAE,CAAC;IACtF,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,OAAgB,EAAE,MAAoB;QAC/C,QAAQ,OAAO,EAAE,IAAI,EAAE,CAAC;YACvB,KAAK,IAAI;gBACR,OAAO,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;YACjC,KAAK,UAAU;gBACd,OAAO,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;YAC/B,KAAK,MAAM;gBACV,OAAO,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAC3B;gBACC,OAAO,gCAAgC,MAAM,CAAC,OAAO,EAAE,IAAI,CAAC,8BAA8B,CAAC;QAC7F,CAAC;IACF,CAAC;IAEO,KAAK,CAAC,EAAE,CAAC,OAAgB,EAAE,MAAoB;QACtD,IAAI,OAAO,OAAO,CAAC,OAAO,KAAK,QAAQ,IAAI,CAAC,OAAO,CAAC,OAAO;YAAE,OAAO,wBAAwB,CAAC;QAC7F,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,IAAI,YAAY,CAAC,CAAC;QACvD,IAAI,CAAC,IAAI;YAAE,OAAO,qCAAqC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;QAC9E,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC;YAAE,OAAO,+BAA+B,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;QACpF,MAAM,OAAO,GACZ,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,OAAO,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAC3G,IAAI,GAAW,CAAC;QAChB,IAAI,CAAC;YACJ,GAAG,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;QAC/D,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACZ,OAAO,UAAU,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC;QAC/D,CAAC;QACD,+FAA+F;QAC/F,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,CAAC;QACvC,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;QAClC,MAAM,QAAQ,GAAG,GAAG;aAClB,KAAK,CAAC,IAAI,CAAC;aACX,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;YACb,MAAM,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,oBAAoB,CAAC,CAAC;YAC3C,IAAI,CAAC,CAAC;gBAAE,OAAO,IAAI,CAAC;YACpB,OAAO,GAAG,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,IAAI,IAAI,CAAC,CAAC,CAAC,CAAE,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QACpF,CAAC,CAAC;aACD,IAAI,CAAC,IAAI,CAAC,CAAC;QACb,OAAO,QAAQ,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,cAAc,CAAC,CAAC;IACpD,CAAC;IAEO,QAAQ,CAAC,OAAgB;QAChC,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QACvC,IAAI,CAAC,IAAI;YAAE,OAAO,qCAAqC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;QAC9E,IAAI,OAAe,CAAC;QACpB,IAAI,CAAC;YACJ,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE;gBAAE,OAAO,sBAAsB,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;YAClF,OAAO,GAAG,YAAY,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;QACvC,CAAC;QAAC,MAAM,CAAC;YACR,OAAO,0BAA0B,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;QACzD,CAAC;QACD,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAClC,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC;QAC3D,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,EAAE,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,CAAC;QAC7E,OAAO,QAAQ,CACd,KAAK;aACH,KAAK,CAAC,KAAK,GAAG,CAAC,EAAE,GAAG,CAAC;aACrB,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,KAAK,GAAG,CAAC,IAAI,IAAI,EAAE,CAAC;aACxC,IAAI,CAAC,IAAI,CAAC,CACZ,CAAC;IACH,CAAC;IAEO,IAAI,CAAC,OAAgB;QAC5B,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,IAAI,YAAY,CAAC,CAAC;QACvD,IAAI,CAAC,IAAI;YAAE,OAAO,qCAAqC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;QAC9E,IAAI,CAAC;YACJ,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,WAAW,EAAE;gBAAE,OAAO,2BAA2B,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;QAC7F,CAAC;QAAC,MAAM,CAAC;YACR,OAAO,+BAA+B,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;QAC9D,CAAC;QACD,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QACrE,OAAO,QAAQ,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,CAAC;IAChD,CAAC;IAEO,UAAU,CAAC,GAAW,EAAE,QAAgB;QAC/C,MAAM,KAAK,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC;QACpC,MAAM,IAAI,GAAG,CAAC,OAAe,EAAE,KAAa,EAAE,MAAc,EAAQ,EAAE;YACrE,IAAI,KAAK,IAAI,QAAQ;gBAAE,OAAO;YAC9B,IAAI,OAAiB,CAAC;YACtB,IAAI,CAAC;gBACJ,OAAO,GAAG,WAAW,CAAC,OAAO,CAAC;qBAC5B,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;qBACvC,IAAI,EAAE,CAAC;YACV,CAAC;YAAC,MAAM,CAAC;gBACR,OAAO;YACR,CAAC;YACD,OAAO,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE;gBAC3B,MAAM,IAAI,GAAG,CAAC,KAAK,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC;gBACtC,KAAK,CAAC,IAAI,CAAC,GAAG,MAAM,GAAG,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,GAAG,IAAI,EAAE,CAAC,CAAC;gBACxD,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;gBACjC,IAAI,KAAK,GAAG,KAAK,CAAC;gBAClB,IAAI,CAAC;oBACJ,KAAK,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,WAAW,EAAE,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;gBAC7D,CAAC;gBAAC,MAAM,CAAC,CAAA,CAAC;gBACV,IAAI,KAAK;oBAAE,IAAI,CAAC,IAAI,EAAE,KAAK,GAAG,CAAC,EAAE,GAAG,MAAM,GAAG,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;YACxE,CAAC,CAAC,CAAC;QACJ,CAAC,CAAC;QACF,IAAI,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;QACjB,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACzB,CAAC;CACD;AAED,gGAAgG;AAChG,SAAS,eAAe,CAAC,IAAY;IACpC,IAAI,IAAY,CAAC;IACjB,IAAI,CAAC;QACJ,IAAI,GAAG,YAAY,CAAC,IAAI,CAAC,IAAI,EAAE,YAAY,CAAC,EAAE,OAAO,CAAC,CAAC;IACxD,CAAC;IAAC,MAAM,CAAC;QACR,OAAO,EAAE,CAAC;IACX,CAAC;IACD,OAAO,IAAI;SACT,KAAK,CAAC,IAAI,CAAC;SACX,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,OAAO,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC;SACpD,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;AACtG,CAAC","sourcesContent":["/**\n * The local side of code_search: the planner runs remotely and names commands against a virtual\n * `/codebase`; this module executes them read-only, confined to one directory.\n *\n * Security boundary: every model-supplied path must resolve inside the root, both lexically and\n * after following symlinks. Only `rg`, `readfile`, and `tree` exist; there is no shell and no write.\n */\nimport { existsSync, readdirSync, readFileSync, realpathSync, statSync } from \"node:fs\";\nimport { basename, dirname, isAbsolute, join, relative, resolve, sep } from \"node:path\";\nimport { createGrepToolDefinition } from \"../../core/tools/grep.ts\";\n\nexport const VIRTUAL_ROOT = \"/codebase\";\n\n/** Per-command result cap. Results are sent back to Devin, whose request limit is ~320KB. */\nconst RESULT_MAX_LINES = 40;\nconst LINE_MAX_CHARS = 200;\n/** The repo map's share of that limit; deeper trees fall back to shallower ones until they fit. */\nconst MAP_MAX_BYTES = 64 * 1024;\nconst MAP_MAX_DEPTH = 3;\n\nconst NOISE_DIRS = [\n\t\".git\",\n\t\"node_modules\",\n\t\"dist\",\n\t\"build\",\n\t\"out\",\n\t\"target\",\n\t\"coverage\",\n\t\".venv\",\n\t\"venv\",\n\t\"__pycache__\",\n\t\".cache\",\n\t\".next\",\n\t\".nuxt\",\n\t\".turbo\",\n\t\".idea\",\n\t\"vendor\",\n\t\"third_party\",\n];\n\n/** `rg` backend: pattern, directory or file, optional glob → `path:line: text` lines. */\nexport type Grep = (pattern: string, path: string, glob: string | undefined, signal?: AbortSignal) => Promise<string>;\n\n/** Pi's own grep tool: gitignore-aware, with its bundled ripgrep. It takes one glob, so only the first include applies. */\nexport function piGrep(root: string): Grep {\n\tconst tool = createGrepToolDefinition(root);\n\tconst execute = tool.execute as unknown as (\n\t\tid: string,\n\t\tparams: { pattern: string; path: string; glob?: string; limit: number },\n\t\tsignal?: AbortSignal,\n\t) => Promise<{ content?: Array<{ type?: string; text?: string }> }>;\n\treturn async (pattern, path, glob, signal) => {\n\t\tconst result = await execute(\"code_search\", { pattern, path, glob, limit: RESULT_MAX_LINES }, signal);\n\t\treturn result.content?.find((c) => c.type === \"text\")?.text ?? \"\";\n\t};\n}\n\nfunction within(base: string, path: string): boolean {\n\tconst rel = relative(base, path);\n\treturn rel === \"\" || (!rel.startsWith(`..${sep}`) && rel !== \"..\" && !isAbsolute(rel));\n}\n\n/** realpath of the longest existing prefix, so a not-yet-existing path still resolves through symlinks. */\nfunction realpathOf(path: string): string {\n\tlet current = resolve(path);\n\tconst tail: string[] = [];\n\twhile (!existsSync(current)) {\n\t\tconst parent = dirname(current);\n\t\tif (parent === current) return resolve(path);\n\t\ttail.unshift(basename(current));\n\t\tcurrent = parent;\n\t}\n\ttry {\n\t\treturn resolve(realpathSync(current), ...tail);\n\t} catch {\n\t\treturn resolve(path);\n\t}\n}\n\n/** Whether `path` lies inside `root`, both as written and after following symlinks. */\nexport function isInside(root: string, path: string): boolean {\n\tconst abs = resolve(path);\n\treturn within(resolve(root), abs) && within(realpathOf(root), realpathOf(abs));\n}\n\nfunction capLines(text: string): string {\n\tconst lines = text.split(\"\\n\");\n\tconst kept = lines\n\t\t.slice(0, RESULT_MAX_LINES)\n\t\t.map((l) => (l.length > LINE_MAX_CHARS ? l.slice(0, LINE_MAX_CHARS) : l));\n\tif (lines.length > RESULT_MAX_LINES) kept.push(`... (${lines.length - RESULT_MAX_LINES} more lines)`);\n\treturn kept.join(\"\\n\");\n}\n\nexport interface Command {\n\ttype?: unknown;\n\tpattern?: unknown;\n\tpath?: unknown;\n\tinclude?: unknown;\n\tfile?: unknown;\n\tstart_line?: unknown;\n\tend_line?: unknown;\n\tlevels?: unknown;\n}\n\nexport class Workspace {\n\treadonly root: string;\n\tprivate readonly grep: Grep;\n\tprivate readonly noise: Set<string>;\n\n\tconstructor(root: string, grep: Grep = piGrep(root)) {\n\t\tthis.root = resolve(root);\n\t\tthis.grep = grep;\n\t\tthis.noise = new Set([...NOISE_DIRS, ...gitignoredNames(this.root)]);\n\t}\n\n\tcontains(path: string): boolean {\n\t\treturn isInside(this.root, path);\n\t}\n\n\t/** `/codebase/src/a.ts` or `src/a.ts` → an absolute path inside the root, or null. */\n\ttoReal(virtual: unknown): string | null {\n\t\tif (typeof virtual !== \"string\" || !virtual) return null;\n\t\tconst normalized = virtual.replace(/\\\\/g, \"/\");\n\t\tlet rel: string;\n\t\tif (normalized === VIRTUAL_ROOT || normalized.startsWith(`${VIRTUAL_ROOT}/`)) {\n\t\t\trel = normalized.slice(VIRTUAL_ROOT.length);\n\t\t} else if (isAbsolute(virtual) || normalized.startsWith(\"/\") || /^[a-zA-Z]:/.test(normalized)) {\n\t\t\t// A drive-letter path is another drive on Windows and a stray filename elsewhere; refuse it everywhere.\n\t\t\treturn null;\n\t\t} else {\n\t\t\trel = normalized;\n\t\t}\n\t\tconst candidate = resolve(this.root, rel.replace(/^\\/+/, \"\"));\n\t\treturn this.contains(candidate) ? candidate : null;\n\t}\n\n\ttoVirtual(real: string): string {\n\t\tconst rel = relative(this.root, real).split(sep).join(\"/\");\n\t\treturn rel ? `${VIRTUAL_ROOT}/${rel}` : VIRTUAL_ROOT;\n\t}\n\n\t/** The deepest tree of the root, up to three levels, that fits the map budget. */\n\trepoMap(): { depth: number; tree: string } {\n\t\tfor (let depth = MAP_MAX_DEPTH; depth > 1; depth--) {\n\t\t\tconst tree = this.renderTree(this.root, depth);\n\t\t\tif (Buffer.byteLength(tree) <= MAP_MAX_BYTES) return { depth, tree };\n\t\t}\n\t\treturn { depth: 1, tree: this.renderTree(this.root, 1).slice(0, MAP_MAX_BYTES / 4) };\n\t}\n\n\tasync run(command: Command, signal?: AbortSignal): Promise<string> {\n\t\tswitch (command?.type) {\n\t\t\tcase \"rg\":\n\t\t\t\treturn this.rg(command, signal);\n\t\t\tcase \"readfile\":\n\t\t\t\treturn this.readfile(command);\n\t\t\tcase \"tree\":\n\t\t\t\treturn this.tree(command);\n\t\t\tdefault:\n\t\t\t\treturn `Error: unknown command type '${String(command?.type)}'; use rg, readfile, or tree`;\n\t\t}\n\t}\n\n\tprivate async rg(command: Command, signal?: AbortSignal): Promise<string> {\n\t\tif (typeof command.pattern !== \"string\" || !command.pattern) return \"Error: missing pattern\";\n\t\tconst real = this.toReal(command.path ?? VIRTUAL_ROOT);\n\t\tif (!real) return `Error: path outside project root: ${String(command.path)}`;\n\t\tif (!existsSync(real)) return `Error: path does not exist: ${String(command.path)}`;\n\t\tconst include =\n\t\t\tArray.isArray(command.include) && typeof command.include[0] === \"string\" ? command.include[0] : undefined;\n\t\tlet raw: string;\n\t\ttry {\n\t\t\traw = await this.grep(command.pattern, real, include, signal);\n\t\t} catch (e) {\n\t\t\treturn `Error: ${e instanceof Error ? e.message : String(e)}`;\n\t\t}\n\t\t// Pi's grep prints paths relative to the searched directory, or the bare file name for a file.\n\t\tconst isFile = statSync(real).isFile();\n\t\tconst base = this.toVirtual(real);\n\t\tconst remapped = raw\n\t\t\t.split(\"\\n\")\n\t\t\t.map((line) => {\n\t\t\t\tconst m = line.match(/^(.+?):(\\d+):(.*)$/);\n\t\t\t\tif (!m) return line;\n\t\t\t\treturn `${isFile ? base : `${base}/${m[1]!.replace(/\\\\/g, \"/\")}`}:${m[2]}:${m[3]}`;\n\t\t\t})\n\t\t\t.join(\"\\n\");\n\t\treturn capLines(remapped.trim() || \"(no matches)\");\n\t}\n\n\tprivate readfile(command: Command): string {\n\t\tconst real = this.toReal(command.file);\n\t\tif (!real) return `Error: path outside project root: ${String(command.file)}`;\n\t\tlet content: string;\n\t\ttry {\n\t\t\tif (!statSync(real).isFile()) return `Error: not a file: ${String(command.file)}`;\n\t\t\tcontent = readFileSync(real, \"utf-8\");\n\t\t} catch {\n\t\t\treturn `Error: file not found: ${String(command.file)}`;\n\t\t}\n\t\tconst lines = content.split(\"\\n\");\n\t\tconst start = Math.max(1, Number(command.start_line) || 1);\n\t\tconst end = Math.min(lines.length, Number(command.end_line) || lines.length);\n\t\treturn capLines(\n\t\t\tlines\n\t\t\t\t.slice(start - 1, end)\n\t\t\t\t.map((line, i) => `${start + i}:${line}`)\n\t\t\t\t.join(\"\\n\"),\n\t\t);\n\t}\n\n\tprivate tree(command: Command): string {\n\t\tconst real = this.toReal(command.path ?? VIRTUAL_ROOT);\n\t\tif (!real) return `Error: path outside project root: ${String(command.path)}`;\n\t\ttry {\n\t\t\tif (!statSync(real).isDirectory()) return `Error: not a directory: ${String(command.path)}`;\n\t\t} catch {\n\t\t\treturn `Error: directory not found: ${String(command.path)}`;\n\t\t}\n\t\tconst levels = Math.min(Math.max(Number(command.levels) || 2, 1), 4);\n\t\treturn capLines(this.renderTree(real, levels));\n\t}\n\n\tprivate renderTree(dir: string, maxDepth: number): string {\n\t\tconst lines = [this.toVirtual(dir)];\n\t\tconst walk = (current: string, depth: number, prefix: string): void => {\n\t\t\tif (depth >= maxDepth) return;\n\t\t\tlet entries: string[];\n\t\t\ttry {\n\t\t\t\tentries = readdirSync(current)\n\t\t\t\t\t.filter((name) => !this.noise.has(name))\n\t\t\t\t\t.sort();\n\t\t\t} catch {\n\t\t\t\treturn;\n\t\t\t}\n\t\t\tentries.forEach((name, i) => {\n\t\t\t\tconst last = i === entries.length - 1;\n\t\t\t\tlines.push(`${prefix}${last ? \"└── \" : \"├── \"}${name}`);\n\t\t\t\tconst path = join(current, name);\n\t\t\t\tlet isDir = false;\n\t\t\t\ttry {\n\t\t\t\t\tisDir = statSync(path).isDirectory() && this.contains(path);\n\t\t\t\t} catch {}\n\t\t\t\tif (isDir) walk(path, depth + 1, `${prefix}${last ? \"    \" : \"│   \"}`);\n\t\t\t});\n\t\t};\n\t\twalk(dir, 0, \"\");\n\t\treturn lines.join(\"\\n\");\n\t}\n}\n\n/** Plain directory names from `.gitignore`; patterns with slashes or globs are left to grep. */\nfunction gitignoredNames(root: string): string[] {\n\tlet text: string;\n\ttry {\n\t\ttext = readFileSync(join(root, \".gitignore\"), \"utf-8\");\n\t} catch {\n\t\treturn [];\n\t}\n\treturn text\n\t\t.split(\"\\n\")\n\t\t.map((line) => line.trim().replace(/^\\/+|\\/+$/g, \"\"))\n\t\t.filter((name) => name && !name.startsWith(\"#\") && !name.startsWith(\"!\") && !/[/*?[\\]]/.test(name));\n}\n"]}

package/dist/main.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"main.d.ts","sourceRoot":"","sources":["../src/main.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAMH,OAAO,EAAE,KAAK,IAAI,EAAyD,MAAM,eAAe,CAAC;AAkCjG,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,4BAA4B,CAAC;AAalE,OAAO,EAAwB,cAAc,EAAE,MAAM,2BAA2B,CAAC;AAEjF,OAAO,EAAE,eAAe,EAAE,MAAM,4BAA4B,CAAC;AAyS7D,wBAAsB,oBAAoB,CACzC,MAAM,EAAE,IAAI,EACZ,GAAG,EAAE,MAAM,EACX,UAAU,EAAE,MAAM,GAAG,SAAS,EAC9B,eAAe,EAAE,eAAe,GAC9B,OAAO,CAAC,cAAc,CAAC,CAsFzB;AAwHD,MAAM,WAAW,WAAW;IAC3B,kBAAkB,CAAC,EAAE,eAAe,EAAE,CAAC;CACvC;AAED,wBAAsB,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE,EAAE,OAAO,CAAC,EAAE,WAAW,iBAya/D"}
+{"version":3,"file":"main.d.ts","sourceRoot":"","sources":["../src/main.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAMH,OAAO,EAAE,KAAK,IAAI,EAAyD,MAAM,eAAe,CAAC;AAkCjG,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,4BAA4B,CAAC;AAalE,OAAO,EAAwB,cAAc,EAAE,MAAM,2BAA2B,CAAC;AAEjF,OAAO,EAAE,eAAe,EAAE,MAAM,4BAA4B,CAAC;AAyS7D,wBAAsB,oBAAoB,CACzC,MAAM,EAAE,IAAI,EACZ,GAAG,EAAE,MAAM,EACX,UAAU,EAAE,MAAM,GAAG,SAAS,EAC9B,eAAe,EAAE,eAAe,GAC9B,OAAO,CAAC,cAAc,CAAC,CAsFzB;AAwHD,MAAM,WAAW,WAAW;IAC3B,kBAAkB,CAAC,EAAE,eAAe,EAAE,CAAC;CACvC;AAED,wBAAsB,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE,EAAE,OAAO,CAAC,EAAE,WAAW,iBA0a/D"}

package/dist/main.js

@@ -631,6 +631,7 @@ export async function main(args, options) {
                 additionalPromptTemplatePaths: resolvedPromptTemplatePaths,
                 additionalThemePaths: resolvedThemePaths,
                 noExtensions: parsed.noExtensions,
+                disabledBuiltinExtensions: parsed.noMcp ? ["mcp"] : undefined,
                 noSkills: parsed.noSkills,
                 noPromptTemplates: parsed.noPromptTemplates,
                 noThemes: parsed.noThemes,