eklavya 1.91.1 → 1.92.0

package/dist/economy/shell-edits.js

@@ -0,0 +1,238 @@
+// Translated and adapted from microsoft/TokenEconomizer,
+// tokeneconomizer/loopguard/shell_edits.py at d3e193b8734ec76230da1e08d7929ba9d14640d8.
+// Licensed under the Apache License, Version 2.0 (see ./LICENSE in this folder).
+// Changes: ported from Python to TypeScript; the Codex `apply_patch` form is kept,
+// `shlex` is replaced by a small tokenizer with the same punctuation rules.
+/**
+ * Edits made through the shell, which an edit-tool counter does not see.
+ *
+ * An agent that edits with `sed -i`, `perl -pi`, a redirect or `tee` into a
+ * source file, an inline Python script that writes the file back, `patch` or
+ * `git apply` looks like one that edited nothing. `shellEditTargets` reads those
+ * writes off the command so the loop guard's turn state counts them.
+ */
+const HEREDOC_START = /<<-?\s*(['"]?)([A-Za-z_]\w*)\1/;
+const OPS = new Set(['&&', '||', ';', ';;', '|', '|&', '&', '\n', '(', ')']);
+const OUT = new Set(['>', '>>', '>|', '&>', '&>>']);
+const SKIP_OPERAND = new Set(['<', '<<', '<<<', '<&', '>&', '<>']);
+const PUNCT = '();<>|&\n';
+/** Redirect and `tee` targets count only when they look like source, so `> out.log` is not an edit. */
+export const SOURCE_FILE = /\.(?:py|pyi|pyx|pxd|js|jsx|mjs|cjs|ts|tsx|rs|go|java|kt|kts|scala|c|cc|cpp|cxx|h|hh|hpp|rb|php|cs|swift|m|mm|sh|bash|zsh|lua|jl|ex|exs|erl|hs|ml|clj|sql|toml|ya?ml|json|cfg|ini|conf|md|rst|html?|css|scss|vue|svelte|xml|gradle|properties|in)$/i;
+const PYTHON = /(?:^|[\s/;&|(])python\d?(?:\.\d+)?\b/;
+const PY_OPEN_LIT = /open\(\s*(['"])([^'"\n]+)\1\s*,\s*(['"])([^'"]*)\3/g;
+const PY_OPEN_VAR = /open\(\s*([A-Za-z_]\w*)\s*,\s*(['"])([^'"]*)\2/g;
+const PY_WRITE_PATH_LIT = /(?:pathlib\.)?Path\(\s*(['"])([^'"\n]+)\1\s*\)\.write_(?:text|bytes)\(/g;
+const PY_WRITE_PATH_VAR = /\b([A-Za-z_]\w*)\.write_(?:text|bytes)\(/g;
+const PATCH_FILE = /^\*\*\* (?:Update|Add|Delete) File:\s*(.+?)\s*$/gm;
+export const basename = (p) => p.replace(/\/+$/, '').split('/').pop();
+const notScratch = (p) => Boolean(p) && !/^\/(dev|proc|tmp)\//.test(p);
+const unquote = (t) => (t.length >= 2 && t[0] === t[t.length - 1] && (t[0] === "'" || t[0] === '"') ? t.slice(1, -1) : t);
+const escapeRe = (s) => s.replace(/[.*+?^${}()|[\]\\]/g, '\\$&');
+/** The command without heredoc bodies, and the bodies. */
+function splitHeredocs(cmd) {
+    const shell = [];
+    const bodies = [];
+    let tag = null;
+    let body = [];
+    for (const line of cmd.split('\n')) {
+        if (tag !== null) {
+            if (line.trim() === tag) {
+                bodies.push(body.join('\n'));
+                tag = null;
+                body = [];
+            }
+            else
+                body.push(line);
+            continue;
+        }
+        shell.push(line);
+        const m = HEREDOC_START.exec(line);
+        if (m)
+            tag = m[2];
+    }
+    if (tag !== null)
+        bodies.push(body.join('\n'));
+    return [shell.join('\n'), bodies];
+}
+/**
+ * Shell words and operators. Quotes stay on quoted words, so a quoted `'>'` is
+ * never an operator; a run of punctuation is one token. Throws on an
+ * unbalanced quote, as `shlex` does.
+ */
+export function shellTokens(text) {
+    const out = [];
+    let word = '';
+    let i = 0;
+    const flush = () => {
+        if (word)
+            out.push(word);
+        word = '';
+    };
+    while (i < text.length) {
+        const c = text[i];
+        if (c === "'" || c === '"') {
+            const end = text.indexOf(c, i + 1);
+            if (end === -1)
+                throw new Error('unbalanced quote');
+            word += text.slice(i, end + 1);
+            i = end + 1;
+        }
+        else if (c === ' ' || c === '\t' || c === '\r') {
+            flush();
+            i++;
+        }
+        else if (PUNCT.includes(c)) {
+            flush();
+            let j = i;
+            while (j < text.length && PUNCT.includes(text[j]))
+                j++;
+            out.push(text.slice(i, j));
+            i = j;
+        }
+        else {
+            word += c;
+            i++;
+        }
+    }
+    flush();
+    return out;
+}
+/** Simple commands as unquoted word lists, and the files their redirects write. */
+function commands(shell) {
+    let streams;
+    try {
+        streams = [shellTokens(shell)];
+    }
+    catch {
+        // One bad line must not hide the others.
+        streams = [];
+        for (const line of shell.split('\n')) {
+            try {
+                streams.push([...shellTokens(line), '\n']);
+            }
+            catch {
+                /* skip the line */
+            }
+        }
+    }
+    const cmds = [];
+    const writes = [];
+    for (const toks of streams) {
+        let cur = [];
+        for (let i = 0; i < toks.length; i++) {
+            const t = toks[i];
+            if (OPS.has(t)) {
+                if (cur.length)
+                    cmds.push(cur);
+                cur = [];
+            }
+            else if (OUT.has(t) || SKIP_OPERAND.has(t)) {
+                // `2>`: the fd number is not an operand.
+                if (cur.length && /^\d+$/.test(cur[cur.length - 1]))
+                    cur.pop();
+                if (OUT.has(t) && i + 1 < toks.length && !OPS.has(toks[i + 1]))
+                    writes.push(unquote(toks[i + 1]));
+                i++;
+            }
+            else
+                cur.push(unquote(t));
+        }
+        if (cur.length)
+            cmds.push(cur);
+    }
+    return [cmds, writes];
+}
+/** File operands of an in-place editor (`sed -i`, `perl -pi`), or [] if not in place. */
+function inplaceFiles(tokens, inplace, scriptFlag) {
+    if (!tokens.slice(1).some(inplace))
+        return [];
+    const files = [];
+    let haveScript = false;
+    let skip = false;
+    for (const t of tokens.slice(1)) {
+        if (skip) {
+            skip = false;
+            haveScript = true;
+            continue;
+        }
+        if (!t || t.startsWith('-')) {
+            if (t && scriptFlag(t))
+                skip = true;
+            continue;
+        }
+        if (!haveScript) {
+            haveScript = true;
+            continue;
+        }
+        files.push(t);
+    }
+    return files;
+}
+/** Paths an inline Python script writes; "(script)" for a write whose path this cannot resolve. */
+function pyWrites(text) {
+    const out = [];
+    for (const m of text.matchAll(PY_OPEN_LIT))
+        if (/[wax+]/.test(m[4]))
+            out.push(m[2]);
+    const variables = [];
+    for (const m of text.matchAll(PY_OPEN_VAR))
+        if (/[wax+]/.test(m[3]))
+            variables.push(m[1]);
+    for (const m of text.matchAll(PY_WRITE_PATH_LIT))
+        out.push(m[2]);
+    for (const m of text.matchAll(PY_WRITE_PATH_VAR))
+        variables.push(m[1]);
+    for (const v of new Set(variables)) {
+        const name = escapeRe(v);
+        const found = [...text.matchAll(new RegExp(`\\b${name}\\s*=\\s*(?:(?:pathlib\\.)?Path\\(\\s*)?(['"])([^'"\\n]+)\\1`, 'g'))].map((m) => m[2]);
+        const loop = new RegExp(`\\bfor\\s+${name}\\s+in\\s+[\\[(]([^\\])]*)[\\])]`).exec(text);
+        const paths = [...found, ...(loop ? [...loop[1].matchAll(/['"]([^'"\n]+)['"]/g)].map((m) => m[1]) : [])];
+        out.push(...(paths.length ? paths : ['(script)']));
+    }
+    return out;
+}
+/** Basenames of the files a shell command writes, in order, without repeats. */
+export function shellEditTargets(cmd) {
+    if (typeof cmd !== 'string' || !cmd.trim())
+        return [];
+    const [shell, bodies] = splitHeredocs(cmd);
+    const [cmds, writes] = commands(shell);
+    const found = writes.filter((w) => SOURCE_FILE.test(w));
+    for (let tokens of cmds) {
+        while (tokens.length && (['sudo', 'command', 'env', 'time', 'nohup'].includes(tokens[0]) || /^\w+=/.test(tokens[0]))) {
+            tokens = tokens.slice(1);
+        }
+        if (!tokens.length)
+            continue;
+        const head = tokens[0].split('/').pop();
+        if (head === 'sed') {
+            found.push(...inplaceFiles(tokens, (t) => /^(?:-[a-zA-Z]*i|--in-place)/.test(t), (t) => ['-e', '-f', '--expression', '--file'].includes(t)));
+        }
+        else if (head === 'perl') {
+            found.push(...inplaceFiles(tokens, (t) => /^-[a-zA-Z]*i/.test(t), (t) => /^-[a-zA-Z]*e$/.test(t)));
+        }
+        else if (head === 'tee') {
+            found.push(...tokens.slice(1).filter((t) => !t.startsWith('-') && SOURCE_FILE.test(t)));
+        }
+        else if (head === 'apply_patch') {
+            const named = [...[...bodies, tokens.slice(1).join(' ')].join('\n').matchAll(PATCH_FILE)].map((m) => m[1]);
+            found.push(...(named.length ? named : ['(patch)']));
+        }
+        else if (head === 'patch' || (head === 'git' && tokens[1] === 'apply')) {
+            found.push('(patch)');
+        }
+        else if (head === 'git' && tokens.length > 2 && (tokens[1] === 'checkout' || tokens[1] === 'restore')) {
+            const dash = tokens.indexOf('--');
+            const rest = dash !== -1 ? tokens.slice(dash + 1) : tokens[1] === 'restore' ? tokens.slice(2) : [];
+            found.push(...rest.filter((t) => !t.startsWith('-')));
+        }
+    }
+    if (PYTHON.test(shell))
+        found.push(...pyWrites([shell, ...bodies].join('\n')));
+    const names = found
+        .filter((p) => p.startsWith('(') || notScratch(p))
+        .map((p) => (p.startsWith('(') ? p : basename(p)))
+        .filter(Boolean);
+    return [...new Set(names)];
+}
+//# sourceMappingURL=shell-edits.js.map

package/dist/economy/shell-edits.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"shell-edits.js","sourceRoot":"","sources":["../../src/economy/shell-edits.ts"],"names":[],"mappings":"AAAA,yDAAyD;AACzD,wFAAwF;AACxF,iFAAiF;AACjF,mFAAmF;AACnF,4EAA4E;AAE5E;;;;;;;GAOG;AAEH,MAAM,aAAa,GAAG,gCAAgC,CAAC;AACvD,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;AAC7E,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC;AACpD,MAAM,YAAY,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC;AACnE,MAAM,KAAK,GAAG,WAAW,CAAC;AAC1B,uGAAuG;AACvG,MAAM,CAAC,MAAM,WAAW,GACtB,oPAAoP,CAAC;AACvP,MAAM,MAAM,GAAG,sCAAsC,CAAC;AACtD,MAAM,WAAW,GAAG,qDAAqD,CAAC;AAC1E,MAAM,WAAW,GAAG,iDAAiD,CAAC;AACtE,MAAM,iBAAiB,GAAG,yEAAyE,CAAC;AACpG,MAAM,iBAAiB,GAAG,2CAA2C,CAAC;AACtE,MAAM,UAAU,GAAG,mDAAmD,CAAC;AAEvE,MAAM,CAAC,MAAM,QAAQ,GAAG,CAAC,CAAS,EAAU,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAG,CAAC;AACvF,MAAM,UAAU,GAAG,CAAC,CAAS,EAAE,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAC/E,MAAM,OAAO,GAAG,CAAC,CAAS,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;AAClI,MAAM,QAAQ,GAAG,CAAC,CAAS,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,qBAAqB,EAAE,MAAM,CAAC,CAAC;AAEzE,0DAA0D;AAC1D,SAAS,aAAa,CAAC,GAAW;IAChC,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,MAAM,MAAM,GAAa,EAAE,CAAC;IAC5B,IAAI,GAAG,GAAkB,IAAI,CAAC;IAC9B,IAAI,IAAI,GAAa,EAAE,CAAC;IACxB,KAAK,MAAM,IAAI,IAAI,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;QACnC,IAAI,GAAG,KAAK,IAAI,EAAE,CAAC;YACjB,IAAI,IAAI,CAAC,IAAI,EAAE,KAAK,GAAG,EAAE,CAAC;gBACxB,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;gBAC7B,GAAG,GAAG,IAAI,CAAC;gBACX,IAAI,GAAG,EAAE,CAAC;YACZ,CAAC;;gBAAM,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACvB,SAAS;QACX,CAAC;QACD,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACjB,MAAM,CAAC,GAAG,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACnC,IAAI,CAAC;YAAE,GAAG,GAAG,CAAC,CAAC,CAAC,CAAE,CAAC;IACrB,CAAC;IACD,IAAI,GAAG,KAAK,IAAI;QAAE,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;IAC/C,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,MAAM,CAAC,CAAC;AACpC,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,WAAW,CAAC,IAAY;IACtC,MAAM,GAAG,GAAa,EAAE,CAAC;IACzB,IAAI,IAAI,GAAG,EAAE,CAAC;IACd,IAAI,CAAC,GAAG,CAAC,CAAC;IACV,MAAM,KAAK,GAAG,GAAG,EAAE;QACjB,IAAI,IAAI;YAAE,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzB,IAAI,GAAG,EAAE,CAAC;IACZ,CAAC,CAAC;IACF,OAAO,CAAC,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;QACvB,MAAM,CAAC,GAAG,IAAI,CAAC,CAAC,CAAE,CAAC;QACnB,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,GAAG,EAAE,CAAC;YAC3B,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;YACnC,IAAI,GAAG,KAAK,CAAC,CAAC;gBAAE,MAAM,IAAI,KAAK,CAAC,kBAAkB,CAAC,CAAC;YACpD,IAAI,IAAI,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,GAAG,CAAC,CAAC,CAAC;YAC/B,CAAC,GAAG,GAAG,GAAG,CAAC,CAAC;QACd,CAAC;aAAM,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC;YACjD,KAAK,EAAE,CAAC;YACR,CAAC,EAAE,CAAC;QACN,CAAC;aAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC;YAC7B,KAAK,EAAE,CAAC;YACR,IAAI,CAAC,GAAG,CAAC,CAAC;YACV,OAAO,CAAC,GAAG,IAAI,CAAC,MAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAE,CAAC;gBAAE,CAAC,EAAE,CAAC;YACxD,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;YAC3B,CAAC,GAAG,CAAC,CAAC;QACR,CAAC;aAAM,CAAC;YACN,IAAI,IAAI,CAAC,CAAC;YACV,CAAC,EAAE,CAAC;QACN,CAAC;IACH,CAAC;IACD,KAAK,EAAE,CAAC;IACR,OAAO,GAAG,CAAC;AACb,CAAC;AAED,mFAAmF;AACnF,SAAS,QAAQ,CAAC,KAAa;IAC7B,IAAI,OAAmB,CAAC;IACxB,IAAI,CAAC;QACH,OAAO,GAAG,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC;IACjC,CAAC;IAAC,MAAM,CAAC;QACP,yCAAyC;QACzC,OAAO,GAAG,EAAE,CAAC;QACb,KAAK,MAAM,IAAI,IAAI,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;YACrC,IAAI,CAAC;gBACH,OAAO,CAAC,IAAI,CAAC,CAAC,GAAG,WAAW,CAAC,IAAI,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC;YAC7C,CAAC;YAAC,MAAM,CAAC;gBACP,mBAAmB;YACrB,CAAC;QACH,CAAC;IACH,CAAC;IACD,MAAM,IAAI,GAAe,EAAE,CAAC;IAC5B,MAAM,MAAM,GAAa,EAAE,CAAC;IAC5B,KAAK,MAAM,IAAI,IAAI,OAAO,EAAE,CAAC;QAC3B,IAAI,GAAG,GAAa,EAAE,CAAC;QACvB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACrC,MAAM,CAAC,GAAG,IAAI,CAAC,CAAC,CAAE,CAAC;YACnB,IAAI,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;gBACf,IAAI,GAAG,CAAC,MAAM;oBAAE,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBAC/B,GAAG,GAAG,EAAE,CAAC;YACX,CAAC;iBAAM,IAAI,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC7C,yCAAyC;gBACzC,IAAI,GAAG,CAAC,MAAM,IAAI,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,GAAG,CAAC,CAAE,CAAC;oBAAE,GAAG,CAAC,GAAG,EAAE,CAAC;gBAChE,IAAI,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,MAAM,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAE,CAAC;oBAAE,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAE,CAAC,CAAC,CAAC;gBACpG,CAAC,EAAE,CAAC;YACN,CAAC;;gBAAM,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;QAC9B,CAAC;QACD,IAAI,GAAG,CAAC,MAAM;YAAE,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IACjC,CAAC;IACD,OAAO,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;AACxB,CAAC;AAED,yFAAyF;AACzF,SAAS,YAAY,CAAC,MAAgB,EAAE,OAA+B,EAAE,UAAkC;IACzG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC;QAAE,OAAO,EAAE,CAAC;IAC9C,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,IAAI,UAAU,GAAG,KAAK,CAAC;IACvB,IAAI,IAAI,GAAG,KAAK,CAAC;IACjB,KAAK,MAAM,CAAC,IAAI,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;QAChC,IAAI,IAAI,EAAE,CAAC;YACT,IAAI,GAAG,KAAK,CAAC;YACb,UAAU,GAAG,IAAI,CAAC;YAClB,SAAS;QACX,CAAC;QACD,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;YAC5B,IAAI,CAAC,IAAI,UAAU,CAAC,CAAC,CAAC;gBAAE,IAAI,GAAG,IAAI,CAAC;YACpC,SAAS;QACX,CAAC;QACD,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,UAAU,GAAG,IAAI,CAAC;YAClB,SAAS;QACX,CAAC;QACD,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAChB,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,mGAAmG;AACnG,SAAS,QAAQ,CAAC,IAAY;IAC5B,MAAM,GAAG,GAAa,EAAE,CAAC;IACzB,KAAK,MAAM,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC;QAAE,IAAI,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAE,CAAC;YAAE,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAE,CAAC,CAAC;IACtF,MAAM,SAAS,GAAa,EAAE,CAAC;IAC/B,KAAK,MAAM,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC;QAAE,IAAI,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAE,CAAC;YAAE,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAE,CAAC,CAAC;IAC5F,KAAK,MAAM,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,iBAAiB,CAAC;QAAE,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAE,CAAC,CAAC;IAClE,KAAK,MAAM,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,iBAAiB,CAAC;QAAE,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAE,CAAC,CAAC;IACxE,KAAK,MAAM,CAAC,IAAI,IAAI,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;QACnC,MAAM,IAAI,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;QACzB,MAAM,KAAK,GAAG,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,MAAM,CAAC,MAAM,IAAI,8DAA8D,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAE,CAAC,CAAC;QAC9I,MAAM,IAAI,GAAG,IAAI,MAAM,CAAC,aAAa,IAAI,kCAAkC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACxF,MAAM,KAAK,GAAG,CAAC,GAAG,KAAK,EAAE,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,CAAE,CAAC,QAAQ,CAAC,qBAAqB,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QAC3G,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;IACrD,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC;AAED,gFAAgF;AAChF,MAAM,UAAU,gBAAgB,CAAC,GAAY;IAC3C,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE;QAAE,OAAO,EAAE,CAAC;IACtD,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC;IAC3C,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;IACvC,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;IACxD,KAAK,IAAI,MAAM,IAAI,IAAI,EAAE,CAAC;QACxB,OAAO,MAAM,CAAC,MAAM,IAAI,CAAC,CAAC,MAAM,EAAE,SAAS,EAAE,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAE,CAAC,IAAI,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAE,CAAC,CAAC,EAAE,CAAC;YACvH,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC3B,CAAC;QACD,IAAI,CAAC,MAAM,CAAC,MAAM;YAAE,SAAS;QAC7B,MAAM,IAAI,GAAG,MAAM,CAAC,CAAC,CAAE,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAG,CAAC;QAC1C,IAAI,IAAI,KAAK,KAAK,EAAE,CAAC;YACnB,KAAK,CAAC,IAAI,CAAC,GAAG,YAAY,CAAC,MAAM,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,6BAA6B,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,cAAc,EAAE,QAAQ,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC/I,CAAC;aAAM,IAAI,IAAI,KAAK,MAAM,EAAE,CAAC;YAC3B,KAAK,CAAC,IAAI,CAAC,GAAG,YAAY,CAAC,MAAM,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACrG,CAAC;aAAM,IAAI,IAAI,KAAK,KAAK,EAAE,CAAC;YAC1B,KAAK,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC1F,CAAC;aAAM,IAAI,IAAI,KAAK,aAAa,EAAE,CAAC;YAClC,MAAM,KAAK,GAAG,CAAC,GAAG,CAAC,GAAG,MAAM,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAE,CAAC,CAAC;YAC5G,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;QACtD,CAAC;aAAM,IAAI,IAAI,KAAK,OAAO,IAAI,CAAC,IAAI,KAAK,KAAK,IAAI,MAAM,CAAC,CAAC,CAAC,KAAK,OAAO,CAAC,EAAE,CAAC;YACzE,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACxB,CAAC;aAAM,IAAI,IAAI,KAAK,KAAK,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,UAAU,IAAI,MAAM,CAAC,CAAC,CAAC,KAAK,SAAS,CAAC,EAAE,CAAC;YACxG,MAAM,IAAI,GAAG,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;YAClC,MAAM,IAAI,GAAG,IAAI,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;YACnG,KAAK,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;QACxD,CAAC;IACH,CAAC;IACD,IAAI,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC;QAAE,KAAK,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC,CAAC,KAAK,EAAE,GAAG,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAC/E,MAAM,KAAK,GAAG,KAAK;SAChB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,UAAU,CAAC,CAAC,CAAC,CAAC;SACjD,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;SACjD,MAAM,CAAC,OAAO,CAAC,CAAC;IACnB,OAAO,CAAC,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC;AAC7B,CAAC"}

package/dist/economy/signals.js

@@ -0,0 +1,164 @@
+// Translated and adapted from microsoft/TokenEconomizer,
+// tokeneconomizer/loopguard/state.py at d3e193b8734ec76230da1e08d7929ba9d14640d8.
+// Licensed under the Apache License, Version 2.0 (see ./LICENSE in this folder).
+// Changes: ported from Python to TypeScript; Claude Code tool names only; the
+// Go `FAIL` lines use a second, case-sensitive pattern (JavaScript has no
+// inline flag groups); a host-reported failure counts as an error output even
+// without an error marker; more test runners are recognised, and `python3 -m unittest` as well as
+// `python -m unittest` (found by a live run of the pinned-facts gate); a verification verdict is kept so a turn that edited
+// after its last check reads as unknown, not green; shell edits are folded in
+// `update` rather than by the caller.
+import crypto from 'node:crypto';
+import { basename, shellEditTargets } from './shell-edits.js';
+/** An output that ran code to verify behaviour: the test step of the edit→test loop. */
+export const TEST_RUN_RE = /(runtests(\.py)?|\bpytest\b|py\.test|manage\.py\s+test|python\d?(?:\.\d+)?\s+-m\s+unittest|\btox\b|\bnox\b|\btrial\b|python\d?\s+-c\b|\bnpm (?:run )?test\b|\bnpx (?:vitest|jest|mocha)\b|\bvitest\b|\bjest\b|\bgo test\b|\bcargo test\b|\bmvn (?:-\S+ )*test\b|\bgradlew? test\b|\brspec\b|node --test\b)/i;
+const ERR_RE = /(Traceback \(most recent call last\)|\b\w*Error\b|\bFAILED\b|\bassert\b|Exception|errors during collection|^# fail [1-9]|^not ok \d|\b[1-9]\d* (?:failing|failures?)\b|Failures: [1-9])/im;
+/** Go's `--- FAIL: T` and `FAIL<tab>pkg`, case-sensitive so unittest's `FAIL: test_x` keeps its own line. */
+const GO_FAIL_RE = /^(?:--- FAIL: |FAIL\t|FAIL$)/m;
+export const isErrorText = (text) => ERR_RE.test(text) || GO_FAIL_RE.test(text);
+const EDIT_TOOLS = new Set(['Edit', 'Write', 'MultiEdit', 'NotebookEdit']);
+/** Tools whose output is file content, not the outcome of running anything. */
+const CONTENT_TOOLS = new Set(['Read', 'Grep', 'Glob', 'NotebookRead']);
+/**
+ * A normalised signature of a failing output: numbers, addresses and paths
+ * dropped, so two runs that fail the same way hash the same. A hash, never
+ * the content.
+ */
+export function failureSignature(text) {
+    if (!text)
+        return null;
+    const first = text.split('\n').find((ln) => isErrorText(ln));
+    if (first === undefined)
+        return null;
+    const norm = first.replace(/0x[0-9a-f]+|\d+|\/\S+/g, '').trim();
+    return crypto.createHash('md5').update(norm).digest('hex').slice(0, 10);
+}
+export const emptyTurn = () => ({
+    calls: 0,
+    editCounts: {},
+    errorOutputs: 0,
+    testRuns: 0,
+    failSigs: {},
+    repeatedFailSigs: 0,
+    lastCheck: null,
+    edited: false,
+    recent: [],
+});
+/** Numbers and quoted strings replaced, so a digest carries a command's shape, not its data. */
+export const commandTemplate = (cmd) => cmd
+    .replace(/\s+/g, ' ')
+    .trim()
+    .replace(/(['"])(?:(?!\1).)*\1/g, '$1…$1')
+    .replace(/\b\d+\b/g, 'N')
+    .slice(0, 90);
+/**
+ * One turn's failure-loop signals, folded one completed tool call at a time:
+ * edits per file, files edited, error outputs, verification runs and how often
+ * a failure repeats one seen before (`stagnation`).
+ */
+export class TurnState {
+    d;
+    constructor(d = emptyTurn()) {
+        this.d = d;
+    }
+    get editChurn() {
+        const v = Object.values(this.d.editCounts);
+        return v.length ? Math.max(...v) : 0;
+    }
+    get distinctEditFiles() {
+        return Object.keys(this.d.editCounts).length;
+    }
+    /** Share of failing outputs that repeat an earlier identical failure. */
+    get stagnation() {
+        const total = Object.values(this.d.failSigs).reduce((a, b) => a + b, 0);
+        return total ? this.d.repeatedFailSigs / total : 0;
+    }
+    /**
+     * Green only when a verification ran after the last edit and passed; a turn
+     * that edited and never checked is unknown, and so is one with no edits.
+     */
+    get verdict() {
+        if (this.d.lastCheck === 'red')
+            return 'red';
+        if (this.d.edited && this.d.lastCheck === 'green')
+            return 'green';
+        return 'unknown';
+    }
+    edit(name) {
+        this.d.editCounts[name] = (this.d.editCounts[name] ?? 0) + 1;
+        this.d.edited = true;
+        this.d.lastCheck = null;
+    }
+    /**
+     * Fold one completed call. `failed` is what the host said: true for a tool
+     * error, false for success, null when it did not say. A call known to have
+     * succeeded is never a failing output; a content tool only when the tool
+     * itself failed; otherwise the output's own markers decide.
+     */
+    update(tool, args, result, failed) {
+        this.d.calls++;
+        let action = tool || '?';
+        if (EDIT_TOOLS.has(tool)) {
+            const b = basename(String(args.file_path ?? args.notebook_path ?? args.path ?? ''));
+            if (b) {
+                this.edit(b);
+                action = `edit ${b}`;
+            }
+        }
+        const cmd = tool === 'Bash' && typeof args.command === 'string' ? args.command : '';
+        let isTest = false;
+        if (cmd) {
+            action = `$ ${commandTemplate(cmd)}`;
+            for (const b of shellEditTargets(cmd))
+                this.edit(b);
+            if (TEST_RUN_RE.test(cmd)) {
+                isTest = true;
+                this.d.testRuns++;
+            }
+        }
+        const canFail = CONTENT_TOOLS.has(tool) ? failed === true : failed !== false;
+        let sig = null;
+        // A failure the host reported counts whatever its text says (`ls: cannot
+        // access` names no Error); otherwise the output's markers decide.
+        const errored = canFail && (failed === true || (Boolean(result) && isErrorText(result)));
+        if (errored) {
+            this.d.errorOutputs++;
+            sig = failureSignature(result);
+            if (sig) {
+                if (this.d.failSigs[sig])
+                    this.d.repeatedFailSigs++;
+                this.d.failSigs[sig] = (this.d.failSigs[sig] ?? 0) + 1;
+            }
+        }
+        if (isTest)
+            this.d.lastCheck = errored ? 'red' : 'green';
+        this.d.recent.push(`${this.d.calls}: ${action}${sig ? `  [err#${sig}]` : ''}`);
+        if (this.d.recent.length > 40)
+            this.d.recent = this.d.recent.slice(-40);
+    }
+    snapshot() {
+        return {
+            calls: this.d.calls,
+            edit_churn: this.editChurn,
+            distinct_edit_files: this.distinctEditFiles,
+            error_outputs: this.d.errorOutputs,
+            test_runs: this.d.testRuns,
+            stagnation: Math.round(this.stagnation * 1000) / 1000,
+            repeated_fail_sigs: this.d.repeatedFailSigs,
+        };
+    }
+    /** The judge's digest: counts and the recent action log, never file content or output. */
+    digest(envelope, n = 24) {
+        const s = this.snapshot();
+        let head = `calls=${s.calls} edit_churn=${s.edit_churn} distinct_edit_files=${s.distinct_edit_files} ` +
+            `error_outputs=${s.error_outputs} test_runs=${s.test_runs} repeated_identical_failures=${s.repeated_fail_sigs} ` +
+            `stagnation=${this.stagnation.toFixed(2)}`;
+        if (envelope) {
+            head +=
+                `\nrepo_context: turns here that end green usually finish within ~${Math.round(envelope)} calls; ` +
+                    `this turn is at ${s.calls} calls = ${(s.calls / envelope).toFixed(1)}x that. Checks are not green.`;
+        }
+        return `${head}\nRecent actions (oldest->newest):\n${this.d.recent.slice(-n).join('\n')}`;
+    }
+}
+//# sourceMappingURL=signals.js.map

package/dist/economy/signals.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"signals.js","sourceRoot":"","sources":["../../src/economy/signals.ts"],"names":[],"mappings":"AAAA,yDAAyD;AACzD,kFAAkF;AAClF,iFAAiF;AACjF,8EAA8E;AAC9E,0EAA0E;AAC1E,8EAA8E;AAC9E,kGAAkG;AAClG,4HAA4H;AAC5H,8EAA8E;AAC9E,sCAAsC;AAEtC,OAAO,MAAM,MAAM,aAAa,CAAC;AACjC,OAAO,EAAE,QAAQ,EAAE,gBAAgB,EAAE,MAAM,kBAAkB,CAAC;AAE9D,wFAAwF;AACxF,MAAM,CAAC,MAAM,WAAW,GACtB,6SAA6S,CAAC;AAChT,MAAM,MAAM,GACV,2LAA2L,CAAC;AAC9L,6GAA6G;AAC7G,MAAM,UAAU,GAAG,+BAA+B,CAAC;AACnD,MAAM,CAAC,MAAM,WAAW,GAAG,CAAC,IAAY,EAAW,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAEjG,MAAM,UAAU,GAAG,IAAI,GAAG,CAAC,CAAC,MAAM,EAAE,OAAO,EAAE,WAAW,EAAE,cAAc,CAAC,CAAC,CAAC;AAC3E,+EAA+E;AAC/E,MAAM,aAAa,GAAG,IAAI,GAAG,CAAC,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,cAAc,CAAC,CAAC,CAAC;AAExE;;;;GAIG;AACH,MAAM,UAAU,gBAAgB,CAAC,IAAY;IAC3C,IAAI,CAAC,IAAI;QAAE,OAAO,IAAI,CAAC;IACvB,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC,CAAC;IAC7D,IAAI,KAAK,KAAK,SAAS;QAAE,OAAO,IAAI,CAAC;IACrC,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,CAAC,wBAAwB,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;IAChE,OAAO,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;AAC1E,CAAC;AA8BD,MAAM,CAAC,MAAM,SAAS,GAAG,GAAa,EAAE,CAAC,CAAC;IACxC,KAAK,EAAE,CAAC;IACR,UAAU,EAAE,EAAE;IACd,YAAY,EAAE,CAAC;IACf,QAAQ,EAAE,CAAC;IACX,QAAQ,EAAE,EAAE;IACZ,gBAAgB,EAAE,CAAC;IACnB,SAAS,EAAE,IAAI;IACf,MAAM,EAAE,KAAK;IACb,MAAM,EAAE,EAAE;CACX,CAAC,CAAC;AAEH,gGAAgG;AAChG,MAAM,CAAC,MAAM,eAAe,GAAG,CAAC,GAAW,EAAU,EAAE,CACrD,GAAG;KACA,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC;KACpB,IAAI,EAAE;KACN,OAAO,CAAC,uBAAuB,EAAE,OAAO,CAAC;KACzC,OAAO,CAAC,UAAU,EAAE,GAAG,CAAC;KACxB,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;AAElB;;;;GAIG;AACH,MAAM,OAAO,SAAS;IACC;IAArB,YAAqB,IAAc,SAAS,EAAE;QAAzB,MAAC,GAAD,CAAC,CAAwB;IAAG,CAAC;IAElD,IAAI,SAAS;QACX,MAAM,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;QAC3C,OAAO,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACvC,CAAC;IAED,IAAI,iBAAiB;QACnB,OAAO,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,MAAM,CAAC;IAC/C,CAAC;IAED,yEAAyE;IACzE,IAAI,UAAU;QACZ,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;QACxE,OAAO,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,gBAAgB,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;IACrD,CAAC;IAED;;;OAGG;IACH,IAAI,OAAO;QACT,IAAI,IAAI,CAAC,CAAC,CAAC,SAAS,KAAK,KAAK;YAAE,OAAO,KAAK,CAAC;QAC7C,IAAI,IAAI,CAAC,CAAC,CAAC,MAAM,IAAI,IAAI,CAAC,CAAC,CAAC,SAAS,KAAK,OAAO;YAAE,OAAO,OAAO,CAAC;QAClE,OAAO,SAAS,CAAC;IACnB,CAAC;IAEO,IAAI,CAAC,IAAY;QACvB,IAAI,CAAC,CAAC,CAAC,UAAU,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;QAC7D,IAAI,CAAC,CAAC,CAAC,MAAM,GAAG,IAAI,CAAC;QACrB,IAAI,CAAC,CAAC,CAAC,SAAS,GAAG,IAAI,CAAC;IAC1B,CAAC;IAED;;;;;OAKG;IACH,MAAM,CAAC,IAAY,EAAE,IAA6B,EAAE,MAAc,EAAE,MAAsB;QACxF,IAAI,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC;QACf,IAAI,MAAM,GAAG,IAAI,IAAI,GAAG,CAAC;QACzB,IAAI,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;YACzB,MAAM,CAAC,GAAG,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,aAAa,IAAI,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,CAAC;YACpF,IAAI,CAAC,EAAE,CAAC;gBACN,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;gBACb,MAAM,GAAG,QAAQ,CAAC,EAAE,CAAC;YACvB,CAAC;QACH,CAAC;QACD,MAAM,GAAG,GAAG,IAAI,KAAK,MAAM,IAAI,OAAO,IAAI,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;QACpF,IAAI,MAAM,GAAG,KAAK,CAAC;QACnB,IAAI,GAAG,EAAE,CAAC;YACR,MAAM,GAAG,KAAK,eAAe,CAAC,GAAG,CAAC,EAAE,CAAC;YACrC,KAAK,MAAM,CAAC,IAAI,gBAAgB,CAAC,GAAG,CAAC;gBAAE,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YACpD,IAAI,WAAW,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;gBAC1B,MAAM,GAAG,IAAI,CAAC;gBACd,IAAI,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC;YACpB,CAAC;QACH,CAAC;QACD,MAAM,OAAO,GAAG,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,KAAK,IAAI,CAAC,CAAC,CAAC,MAAM,KAAK,KAAK,CAAC;QAC7E,IAAI,GAAG,GAAkB,IAAI,CAAC;QAC9B,yEAAyE;QACzE,kEAAkE;QAClE,MAAM,OAAO,GAAG,OAAO,IAAI,CAAC,MAAM,KAAK,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;QACzF,IAAI,OAAO,EAAE,CAAC;YACZ,IAAI,CAAC,CAAC,CAAC,YAAY,EAAE,CAAC;YACtB,GAAG,GAAG,gBAAgB,CAAC,MAAM,CAAC,CAAC;YAC/B,IAAI,GAAG,EAAE,CAAC;gBACR,IAAI,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC;oBAAE,IAAI,CAAC,CAAC,CAAC,gBAAgB,EAAE,CAAC;gBACpD,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;YACzD,CAAC;QACH,CAAC;QACD,IAAI,MAAM;YAAE,IAAI,CAAC,CAAC,CAAC,SAAS,GAAG,OAAO,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC;QACzD,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,KAAK,KAAK,MAAM,GAAG,GAAG,CAAC,CAAC,CAAC,UAAU,GAAG,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QAC/E,IAAI,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,GAAG,EAAE;YAAE,IAAI,CAAC,CAAC,CAAC,MAAM,GAAG,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC;IAC1E,CAAC;IAED,QAAQ;QACN,OAAO;YACL,KAAK,EAAE,IAAI,CAAC,CAAC,CAAC,KAAK;YACnB,UAAU,EAAE,IAAI,CAAC,SAAS;YAC1B,mBAAmB,EAAE,IAAI,CAAC,iBAAiB;YAC3C,aAAa,EAAE,IAAI,CAAC,CAAC,CAAC,YAAY;YAClC,SAAS,EAAE,IAAI,CAAC,CAAC,CAAC,QAAQ;YAC1B,UAAU,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,GAAG,IAAI;YACrD,kBAAkB,EAAE,IAAI,CAAC,CAAC,CAAC,gBAAgB;SAC5C,CAAC;IACJ,CAAC;IAED,0FAA0F;IAC1F,MAAM,CAAC,QAAuB,EAAE,CAAC,GAAG,EAAE;QACpC,MAAM,CAAC,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;QAC1B,IAAI,IAAI,GACN,SAAS,CAAC,CAAC,KAAK,eAAe,CAAC,CAAC,UAAU,wBAAwB,CAAC,CAAC,mBAAmB,GAAG;YAC3F,iBAAiB,CAAC,CAAC,aAAa,cAAc,CAAC,CAAC,SAAS,gCAAgC,CAAC,CAAC,kBAAkB,GAAG;YAChH,cAAc,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;QAC7C,IAAI,QAAQ,EAAE,CAAC;YACb,IAAI;gBACF,oEAAoE,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,UAAU;oBAClG,mBAAmB,CAAC,CAAC,KAAK,YAAY,CAAC,CAAC,CAAC,KAAK,GAAG,QAAQ,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,+BAA+B,CAAC;QACzG,CAAC;QACD,OAAO,GAAG,IAAI,uCAAuC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;IAC5F,CAAC;CACF"}

package/dist/economy/toolprune.js

@@ -0,0 +1,209 @@
+// Translated and adapted from microsoft/TokenEconomizer,
+// tokeneconomizer/plugin/denylist.py and toolprune/tools.py at
+// d3e193b8734ec76230da1e08d7929ba9d14640d8.
+// Licensed under the Apache License, Version 2.0 (see ./LICENSE in this folder).
+// Changes: ported from Python to TypeScript; propose mode first (nothing is
+// written until the developer applies); the protected list includes
+// Eklavya's own tools and the tools its quiz and delegation depend on; the
+// file's original bytes are kept so a restore that undoes every change puts
+// them back exactly; `.claude` reached through a symlink out of the checkout
+// is refused.
+import { execFileSync } from 'node:child_process';
+import fs from 'node:fs';
+import path from 'node:path';
+import { isEklavyaTool } from './transcript.js';
+/**
+ * Tool pruning: tools a repository's sessions load and never use cost context
+ * on every request. Eklavya proposes denying them in this checkout's
+ * `.claude/settings.local.json`, and writes only when the developer runs
+ * `eklavya toolprune apply`.
+ */
+/** Never proposed, whatever the evidence. */
+export const PROTECTED = new Set([
+    'Bash', 'Read', 'Edit', 'Write', 'MultiEdit', 'NotebookEdit', 'Grep', 'Glob',
+    'AskUserQuestion', 'EnterPlanMode', 'ExitPlanMode', 'TodoWrite',
+    'WebFetch', 'WebSearch',
+    'Agent', 'Task', 'Skill', 'ToolSearch',
+]);
+export function isProtected(tool, extra = []) {
+    return PROTECTED.has(tool) || isEklavyaTool(tool) || extra.includes(tool);
+}
+/** Unused for `evidence` prompts across `min_sessions` sessions, or for `sessions` sessions. */
+export function propose(rows, rule) {
+    return rows
+        .filter((r) => !isProtected(r.tool, rule.protect))
+        .filter((r) => (r.prompts_since >= rule.evidence && r.sessions_since >= rule.min_sessions) || r.sessions_since >= rule.sessions)
+        .sort((a, b) => b.sessions_since - a.sessions_since || b.prompts_since - a.prompts_since || a.tool.localeCompare(b.tool));
+}
+/** Is `target` (followed through symlinks) inside `root`? */
+function inside(root, target) {
+    const rel = path.relative(fs.realpathSync(root), target);
+    return rel === '' || (!rel.startsWith('..') && !path.isAbsolute(rel));
+}
+/** Two spaces unless the file's first indented line says otherwise. */
+const indentOf = (text) => /\n([ \t]+)\S/.exec(text)?.[1] ?? '  ';
+/** The settings file for a checkout, refusing a `.claude` that leads out of it. */
+export function settingsPath(repoRoot) {
+    const dir = path.join(repoRoot, '.claude');
+    if (fs.existsSync(dir) && !inside(repoRoot, fs.realpathSync(dir))) {
+        return { reason: '.claude is a symlink that leads out of the repository' };
+    }
+    const file = path.join(dir, 'settings.local.json');
+    if (fs.existsSync(file) && !inside(repoRoot, fs.realpathSync(file))) {
+        return { reason: 'settings.local.json is a symlink that leads out of the repository' };
+    }
+    return { file };
+}
+function readSettings(file) {
+    if (!fs.existsSync(file))
+        return { text: null, json: {} };
+    const text = fs.readFileSync(file, 'utf8');
+    try {
+        const json = JSON.parse(text);
+        if (!json || typeof json !== 'object' || Array.isArray(json))
+            return { reason: 'settings.local.json is not a JSON object' };
+        const perms = json.permissions;
+        if (perms !== undefined && (!perms || typeof perms !== 'object' || Array.isArray(perms))) {
+            return { reason: 'settings.local.json has a permissions value that is not an object' };
+        }
+        const deny = perms?.deny;
+        if (deny !== undefined && !Array.isArray(deny))
+            return { reason: 'settings.local.json has a permissions.deny that is not a list' };
+        return { text, json: json };
+    }
+    catch {
+        return { reason: 'settings.local.json is not valid JSON' };
+    }
+}
+function writeJson(file, json, indent, trailingNewline) {
+    fs.mkdirSync(path.dirname(file), { recursive: true });
+    const tmp = `${file}.eklavya-tmp`;
+    fs.writeFileSync(tmp, JSON.stringify(json, null, indent) + (trailingNewline ? '\n' : ''));
+    fs.renameSync(tmp, file);
+}
+/**
+ * Adds `tools` to `permissions.deny` as bare names, leaving every existing
+ * rule alone. `previous` is the record of an earlier apply to this file, so a
+ * second apply keeps the first one's original bytes.
+ */
+export function applyDeny(repoRoot, tools, previous = null) {
+    const where = settingsPath(repoRoot);
+    if ('reason' in where)
+        return { ok: false, reason: where.reason };
+    const read = readSettings(where.file);
+    if ('reason' in read)
+        return { ok: false, reason: read.reason };
+    const json = read.json;
+    const perms = (json.permissions ??= {});
+    const deny = (perms.deny ??= []);
+    const added = tools.filter((t) => !deny.includes(t));
+    const record = previous
+        ? { ...previous, added: [...new Set([...previous.added, ...added])] }
+        : { file: where.file, original: read.text, added, excludeLine: null };
+    if (!added.length)
+        return { ok: true, record };
+    deny.push(...added);
+    writeJson(where.file, json, read.text ? indentOf(read.text) : '  ', read.text ? read.text.endsWith('\n') : true);
+    if (read.text === null && !previous)
+        record.excludeLine = excludeFile(repoRoot, where.file);
+    return { ok: true, record };
+}
+/** Adds the file to `.git/info/exclude` when Eklavya created it; returns the line, or null. */
+function excludeFile(repoRoot, file) {
+    const exclude = excludePath(repoRoot);
+    if (!exclude)
+        return null;
+    const line = `/${path.relative(repoRoot, file).split(path.sep).join('/')}`;
+    const current = fs.existsSync(exclude) ? fs.readFileSync(exclude, 'utf8') : '';
+    if (current.split('\n').includes(line))
+        return null;
+    fs.mkdirSync(path.dirname(exclude), { recursive: true });
+    fs.writeFileSync(exclude, `${current}${current && !current.endsWith('\n') ? '\n' : ''}${line}\n`);
+    return line;
+}
+/**
+ * Removes exactly the entries Eklavya added. When nothing else in the file
+ * changed since, the original bytes come back (or the file goes, if Eklavya
+ * created it); otherwise only those entries leave and the rest is kept.
+ */
+export function restoreDeny(repoRoot, record, only) {
+    const read = readSettings(record.file);
+    if ('reason' in read)
+        return { ok: false, reason: read.reason };
+    const removing = only ? record.added.filter((t) => only.includes(t)) : record.added;
+    const json = read.json;
+    const perms = json.permissions;
+    const deny = perms?.deny ?? [];
+    const kept = deny.filter((t) => !(typeof t === 'string' && removing.includes(t)));
+    const removed = removing.filter((t) => deny.includes(t));
+    const remaining = record.added.filter((t) => !removing.includes(t));
+    // Untouched: the file is still exactly what the apply made of the original,
+    // so the original bytes (or no file) can come back.
+    const original = record.original === null ? {} : JSON.parse(record.original);
+    const expected = structuredClone(original);
+    const ep = (expected.permissions ??= {});
+    ep.deny = [...(ep.deny ?? []), ...record.added.filter((t) => !(original.permissions?.deny ?? []).includes(t))];
+    const untouched = !remaining.length && JSON.stringify(read.json) === JSON.stringify(expected);
+    if (perms)
+        perms.deny = kept;
+    if (untouched) {
+        if (record.original === null) {
+            fs.rmSync(record.file, { force: true });
+            if (record.excludeLine)
+                unexclude(repoRoot, record.excludeLine);
+        }
+        else
+            fs.writeFileSync(record.file, record.original);
+    }
+    else if (removed.length) {
+        // Something was removed, so the file exists and has text.
+        writeJson(record.file, json, indentOf(read.text), read.text.endsWith('\n'));
+    }
+    return { ok: true, removed };
+}
+/**
+ * The checkout's `info/exclude`, asked of git: in a linked worktree `.git` is a
+ * file and the exclude file lives in the shared git dir. Null outside a
+ * checkout, or when git cannot say.
+ */
+export function excludePath(repoRoot) {
+    if (!fs.existsSync(path.join(repoRoot, '.git')))
+        return null;
+    try {
+        const out = execFileSync('git', ['rev-parse', '--git-path', 'info/exclude'], {
+            cwd: repoRoot,
+            encoding: 'utf8',
+            timeout: 2_000,
+            stdio: ['ignore', 'pipe', 'ignore'],
+        }).trim();
+        return path.resolve(repoRoot, out);
+    }
+    catch {
+        return null;
+    }
+}
+function unexclude(repoRoot, line) {
+    const exclude = excludePath(repoRoot);
+    if (!exclude || !fs.existsSync(exclude))
+        return;
+    const lines = fs.readFileSync(exclude, 'utf8').split('\n');
+    const at = lines.lastIndexOf(line);
+    if (at === -1)
+        return;
+    lines.splice(at, 1);
+    fs.writeFileSync(exclude, lines.join('\n'));
+}
+/**
+ * The size of tool definitions before and after denying `tools`, from a
+ * catalog of definition sizes in estimated tokens. A deferred MCP tool costs
+ * one listing line, not its schema, so it is counted as such.
+ */
+export function measure(catalog, deferred, denied) {
+    const LISTING = 12;
+    const cost = (t) => (deferred.includes(t) ? LISTING : catalog[t]);
+    const all = new Set([...Object.keys(catalog), ...deferred]);
+    const before = [...all].reduce((s, t) => s + cost(t), 0);
+    const after = [...all].filter((t) => !denied.includes(t)).reduce((s, t) => s + cost(t), 0);
+    return { before, after, saved: before - after };
+}
+//# sourceMappingURL=toolprune.js.map