@djolex999/vir-cli 0.12.0 → 0.14.0

package/dist/pipeline/projects.js

@@ -0,0 +1,189 @@
+import { closeSync, openSync, readSync, readdirSync } from "node:fs";
+import { basename, dirname, join, relative, sep } from "node:path";
+import { computeCost } from "../cost/pricing.js";
+function defaultReadDir(path) {
+    try {
+        return readdirSync(path, { withFileTypes: true })
+            .filter((e) => e.isDirectory())
+            .map((e) => e.name);
+    }
+    catch {
+        return [];
+    }
+}
+// Claude Code encodes a session's cwd into the transcript dir name by
+// replacing every non-alphanumeric character with "-". Apply the same
+// encoding to real dir names so on-disk candidates compare against the
+// encoded segments byte-for-byte.
+function encodeSegment(name) {
+    return name.replace(/[^A-Za-z0-9]/g, "-");
+}
+// Walk the real filesystem from `cur`, consuming `remaining` (the encoded
+// path with its leading dash stripped). Dashes are ambiguous — they encode
+// both path separators and literal dashes/dots/spaces — so try candidates
+// longest-encoding-first and backtrack: only a FULL resolution counts.
+// Returns the on-disk basename of the resolved leaf, or null.
+function resolveEncoded(cur, remaining, readDir) {
+    const candidates = readDir(cur)
+        .map((name) => ({ name, enc: encodeSegment(name) }))
+        .filter(({ enc }) => enc.length > 0 &&
+        (enc === remaining || remaining.startsWith(enc + "-")))
+        .sort((a, b) => b.enc.length - a.enc.length);
+    for (const { name, enc } of candidates) {
+        if (enc === remaining)
+            return name;
+        const leaf = resolveEncoded(join(cur, name), remaining.slice(enc.length + 1), readDir);
+        if (leaf !== null)
+            return leaf;
+    }
+    return null;
+}
+// Decode a transcript directory name ("-Users-me-projects-my-app") to its
+// project name ("my-app") by resolving the encoded absolute path against
+// directories that actually exist on disk, longest match wins. Falls back to
+// the raw dir name when nothing fully resolves — never throws, never guesses.
+export function decodeProjectName(dirName, projectsDir, deps) {
+    const readDir = deps?.readDir ?? defaultReadDir;
+    const name = dirName.includes("/") && dirName.startsWith(projectsDir)
+        ? basename(dirName)
+        : dirName;
+    if (!name.startsWith("-"))
+        return name;
+    const leaf = resolveEncoded("/", name.slice(1), readDir);
+    return leaf ?? name;
+}
+export function classifyTranscript(path, projectsDir) {
+    const rel = relative(projectsDir, path);
+    if (rel.startsWith(".."))
+        return "session";
+    const segments = rel.split(sep);
+    // segments[0] is the encoded project dir; look for `subagents` below it.
+    const subIdx = segments.indexOf("subagents", 1);
+    if (subIdx === -1)
+        return "session";
+    const below = segments.slice(subIdx + 1);
+    const isWorkflow = below.some((s) => s === "workflows" || s.startsWith("wf_"));
+    return isWorkflow ? "workflow" : "sidechain";
+}
+// SDK launch signature: the first `type:"user"` line's `entrypoint` starting
+// with "sdk" marks an agent-internal transcript (review/verify harness agents
+// stored top-level). Keyed on entrypoint ONLY — promptSource reads "sdk" even
+// on desktop-launched human sessions (the C23 serbeval trap), and turn count
+// would kill single-prompt autonomous runs. Returns the entrypoint, or null
+// for human sessions, truncated heads, or anything unparseable — a miss here
+// falls through to the parser backstop, never to a paid call.
+export function sniffAgentEntrypoint(head) {
+    for (const line of head.split("\n")) {
+        if (line.trim().length === 0)
+            continue;
+        let evt;
+        try {
+            evt = JSON.parse(line);
+        }
+        catch {
+            // truncated tail line or garbage — never a user line we can trust
+            continue;
+        }
+        if (evt.type !== "user")
+            continue;
+        return typeof evt.entrypoint === "string" &&
+            evt.entrypoint.startsWith("sdk")
+            ? evt.entrypoint
+            : null;
+    }
+    return null;
+}
+// How much of a transcript the scan-time sniff reads. Review-agent first
+// prompts embed whole diffs, so the first user line can be tens of KB; 256KB
+// covers every observed case while staying trivial next to the full-file
+// read the scanner already does for hashing.
+export const SNIFF_HEAD_BYTES = 262_144;
+export function readTranscriptHead(path, bytes = SNIFF_HEAD_BYTES) {
+    try {
+        const fd = openSync(path, "r");
+        try {
+            const buf = Buffer.alloc(bytes);
+            const n = readSync(fd, buf, 0, bytes, 0);
+            return buf.toString("utf8", 0, n);
+        }
+        finally {
+            closeSync(fd);
+        }
+    }
+    catch {
+        return "";
+    }
+}
+// Rough triage category for the init wizard's counts line
+// ("21 interactive · 67 SDK-launched · 9 stubs"). Display-only — the run
+// gate uses sniffAgentEntrypoint directly. A stub is a tiny COMPLETE
+// transcript (head covers the whole file) with ≤10 non-blank lines: /clear
+// markers and aborted launches. A truncated head is always interactive —
+// never downgrade what we can't see.
+export function categorizeTranscriptHead(head, sizeBytes) {
+    if (sniffAgentEntrypoint(head) !== null)
+        return "agent";
+    const complete = head.length >= sizeBytes;
+    const lines = head.split("\n").filter((l) => l.trim().length > 0).length;
+    return complete && lines <= 10 ? "stub" : "interactive";
+}
+export function decideProject(name, configured, flags) {
+    const configuredDecision = configured[name];
+    if (configuredDecision === "exclude")
+        return "exclude";
+    if (flags?.excludeProject?.includes(name) === true)
+        return "flag-skip";
+    if (flags?.only !== undefined &&
+        flags.only.length > 0 &&
+        !flags.only.includes(name)) {
+        return "flag-skip";
+    }
+    return configuredDecision === "include" ? "include" : "pending";
+}
+// Group discovered sessions by decoded project name. Two transcript dirs can
+// decode to the same project (rare, but a renamed cwd does it) — they merge.
+// The decode cache matters: decodeProjectName walks the filesystem, and a
+// scan can hold hundreds of sessions across a handful of dirs.
+export function groupByProject(sessions, projectsDir, deps) {
+    const cache = new Map();
+    const groups = new Map();
+    for (const s of sessions) {
+        // The project dir is the FIRST segment under projectsDir — workflow and
+        // subagent transcripts nest one level deeper (<encoded>/<session-id>/
+        // wf_*.jsonl) and must group under their project, not the inner dir.
+        const rel = relative(projectsDir, s.path);
+        const dirName = !rel.startsWith("..") && rel.includes(sep)
+            ? (rel.split(sep)[0] ?? basename(dirname(s.path)))
+            : basename(dirname(s.path));
+        let name = cache.get(dirName);
+        if (name === undefined) {
+            name = decodeProjectName(dirName, projectsDir, deps);
+            cache.set(dirName, name);
+        }
+        let g = groups.get(name);
+        if (!g) {
+            g = { name, sessions: [], totalBytes: 0 };
+            groups.set(name, g);
+        }
+        g.sessions.push(s);
+        g.totalBytes += s.size;
+    }
+    return groups;
+}
+// Rough per-session cost estimate from raw transcript size — for the triage
+// UIs (init multi-select, undecided prompt, `vir projects`), NOT billing.
+// Derivation from real dry-run calibration: transcript prose+tools ≈ half the
+// JSONL bytes, the moderate tool filter keeps ~45% of that, chars/3 density
+// → distill input ≈ bytes/13; classify sees prose only ≈ bytes/60. Upper
+// bound: heuristic-filtered sessions actually cost nothing.
+const DISTILL_BYTES_PER_TOKEN = 13;
+const CLASSIFY_BYTES_PER_TOKEN = 60;
+const MAX_DISTILL_INPUT_TOKENS = 150_000;
+const MAX_CLASSIFY_INPUT_TOKENS = 30_000;
+export function estimateSessionCost(provider, classifyModel, distillModel, sizeBytes, pricing, kieTopUpTier) {
+    const classifyIn = Math.min(Math.ceil(sizeBytes / CLASSIFY_BYTES_PER_TOKEN), MAX_CLASSIFY_INPUT_TOKENS);
+    const distillIn = Math.min(Math.ceil(sizeBytes / DISTILL_BYTES_PER_TOKEN), MAX_DISTILL_INPUT_TOKENS);
+    return (computeCost(provider, classifyModel, classifyIn, 350, pricing, kieTopUpTier) +
+        computeCost(provider, distillModel, distillIn, 4500, pricing, kieTopUpTier));
+}
+//# sourceMappingURL=projects.js.map

package/dist/pipeline/projects.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"projects.js","sourceRoot":"","sources":["../../src/pipeline/projects.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,QAAQ,EAAE,WAAW,EAAE,MAAM,SAAS,CAAC;AACrE,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,GAAG,EAAE,MAAM,WAAW,CAAC;AACnE,OAAO,EAAE,WAAW,EAAyB,MAAM,oBAAoB,CAAC;AAQxE,SAAS,cAAc,CAAC,IAAY;IAClC,IAAI,CAAC;QACH,OAAO,WAAW,CAAC,IAAI,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC;aAC9C,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;aAC9B,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IACxB,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,CAAC;IACZ,CAAC;AACH,CAAC;AAED,sEAAsE;AACtE,sEAAsE;AACtE,uEAAuE;AACvE,kCAAkC;AAClC,SAAS,aAAa,CAAC,IAAY;IACjC,OAAO,IAAI,CAAC,OAAO,CAAC,eAAe,EAAE,GAAG,CAAC,CAAC;AAC5C,CAAC;AAED,0EAA0E;AAC1E,2EAA2E;AAC3E,0EAA0E;AAC1E,uEAAuE;AACvE,8DAA8D;AAC9D,SAAS,cAAc,CACrB,GAAW,EACX,SAAiB,EACjB,OAAmC;IAEnC,MAAM,UAAU,GAAG,OAAO,CAAC,GAAG,CAAC;SAC5B,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,aAAa,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;SACnD,MAAM,CACL,CAAC,EAAE,GAAG,EAAE,EAAE,EAAE,CACV,GAAG,CAAC,MAAM,GAAG,CAAC;QACd,CAAC,GAAG,KAAK,SAAS,IAAI,SAAS,CAAC,UAAU,CAAC,GAAG,GAAG,GAAG,CAAC,CAAC,CACzD;SACA,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,MAAM,GAAG,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IAC/C,KAAK,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,UAAU,EAAE,CAAC;QACvC,IAAI,GAAG,KAAK,SAAS;YAAE,OAAO,IAAI,CAAC;QACnC,MAAM,IAAI,GAAG,cAAc,CACzB,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,EACf,SAAS,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,GAAG,CAAC,CAAC,EAC/B,OAAO,CACR,CAAC;QACF,IAAI,IAAI,KAAK,IAAI;YAAE,OAAO,IAAI,CAAC;IACjC,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,0EAA0E;AAC1E,yEAAyE;AACzE,6EAA6E;AAC7E,8EAA8E;AAC9E,MAAM,UAAU,iBAAiB,CAC/B,OAAe,EACf,WAAmB,EACnB,IAA0B;IAE1B,MAAM,OAAO,GAAG,IAAI,EAAE,OAAO,IAAI,cAAc,CAAC;IAChD,MAAM,IAAI,GACR,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,OAAO,CAAC,UAAU,CAAC,WAAW,CAAC;QACtD,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC;QACnB,CAAC,CAAC,OAAO,CAAC;IACd,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC;QAAE,OAAO,IAAI,CAAC;IACvC,MAAM,IAAI,GAAG,cAAc,CAAC,GAAG,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;IACzD,OAAO,IAAI,IAAI,IAAI,CAAC;AACtB,CAAC;AAYD,MAAM,UAAU,kBAAkB,CAChC,IAAY,EACZ,WAAmB;IAEnB,MAAM,GAAG,GAAG,QAAQ,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC;IACxC,IAAI,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC;QAAE,OAAO,SAAS,CAAC;IAC3C,MAAM,QAAQ,GAAG,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IAChC,yEAAyE;IACzE,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC;IAChD,IAAI,MAAM,KAAK,CAAC,CAAC;QAAE,OAAO,SAAS,CAAC;IACpC,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IACzC,MAAM,UAAU,GAAG,KAAK,CAAC,IAAI,CAC3B,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,WAAW,IAAI,CAAC,CAAC,UAAU,CAAC,KAAK,CAAC,CAChD,CAAC;IACF,OAAO,UAAU,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,WAAW,CAAC;AAC/C,CAAC;AAED,6EAA6E;AAC7E,8EAA8E;AAC9E,8EAA8E;AAC9E,6EAA6E;AAC7E,4EAA4E;AAC5E,6EAA6E;AAC7E,8DAA8D;AAC9D,MAAM,UAAU,oBAAoB,CAAC,IAAY;IAC/C,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;QACpC,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC,MAAM,KAAK,CAAC;YAAE,SAAS;QACvC,IAAI,GAA6C,CAAC;QAClD,IAAI,CAAC;YACH,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAA6C,CAAC;QACrE,CAAC;QAAC,MAAM,CAAC;YACP,kEAAkE;YAClE,SAAS;QACX,CAAC;QACD,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM;YAAE,SAAS;QAClC,OAAO,OAAO,GAAG,CAAC,UAAU,KAAK,QAAQ;YACvC,GAAG,CAAC,UAAU,CAAC,UAAU,CAAC,KAAK,CAAC;YAChC,CAAC,CAAC,GAAG,CAAC,UAAU;YAChB,CAAC,CAAC,IAAI,CAAC;IACX,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,yEAAyE;AACzE,6EAA6E;AAC7E,yEAAyE;AACzE,6CAA6C;AAC7C,MAAM,CAAC,MAAM,gBAAgB,GAAG,OAAO,CAAC;AAExC,MAAM,UAAU,kBAAkB,CAChC,IAAY,EACZ,QAAgB,gBAAgB;IAEhC,IAAI,CAAC;QACH,MAAM,EAAE,GAAG,QAAQ,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;QAC/B,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YAChC,MAAM,CAAC,GAAG,QAAQ,CAAC,EAAE,EAAE,GAAG,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC;YACzC,OAAO,GAAG,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QACpC,CAAC;gBAAS,CAAC;YACT,SAAS,CAAC,EAAE,CAAC,CAAC;QAChB,CAAC;IACH,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,CAAC;IACZ,CAAC;AACH,CAAC;AAED,0DAA0D;AAC1D,yEAAyE;AACzE,qEAAqE;AACrE,2EAA2E;AAC3E,yEAAyE;AACzE,qCAAqC;AACrC,MAAM,UAAU,wBAAwB,CACtC,IAAY,EACZ,SAAiB;IAEjB,IAAI,oBAAoB,CAAC,IAAI,CAAC,KAAK,IAAI;QAAE,OAAO,OAAO,CAAC;IACxD,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,IAAI,SAAS,CAAC;IAC1C,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,MAAM,CAAC;IACzE,OAAO,QAAQ,IAAI,KAAK,IAAI,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,aAAa,CAAC;AAC1D,CAAC;AAcD,MAAM,UAAU,aAAa,CAC3B,IAAY,EACZ,UAAiD,EACjD,KAAuB;IAEvB,MAAM,kBAAkB,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC;IAC5C,IAAI,kBAAkB,KAAK,SAAS;QAAE,OAAO,SAAS,CAAC;IACvD,IAAI,KAAK,EAAE,cAAc,EAAE,QAAQ,CAAC,IAAI,CAAC,KAAK,IAAI;QAAE,OAAO,WAAW,CAAC;IACvE,IACE,KAAK,EAAE,IAAI,KAAK,SAAS;QACzB,KAAK,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC;QACrB,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,EAC1B,CAAC;QACD,OAAO,WAAW,CAAC;IACrB,CAAC;IACD,OAAO,kBAAkB,KAAK,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC;AAClE,CAAC;AAQD,6EAA6E;AAC7E,6EAA6E;AAC7E,0EAA0E;AAC1E,+DAA+D;AAC/D,MAAM,UAAU,cAAc,CAC5B,QAA6D,EAC7D,WAAmB,EACnB,IAA0B;IAE1B,MAAM,KAAK,GAAG,IAAI,GAAG,EAAkB,CAAC;IACxC,MAAM,MAAM,GAAG,IAAI,GAAG,EAAwB,CAAC;IAC/C,KAAK,MAAM,CAAC,IAAI,QAAQ,EAAE,CAAC;QACzB,wEAAwE;QACxE,sEAAsE;QACtE,qEAAqE;QACrE,MAAM,GAAG,GAAG,QAAQ,CAAC,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC;QAC1C,MAAM,OAAO,GACX,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC;YACxC,CAAC,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;YAClD,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;QAChC,IAAI,IAAI,GAAG,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QAC9B,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;YACvB,IAAI,GAAG,iBAAiB,CAAC,OAAO,EAAE,WAAW,EAAE,IAAI,CAAC,CAAC;YACrD,KAAK,CAAC,GAAG,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;QAC3B,CAAC;QACD,IAAI,CAAC,GAAG,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACzB,IAAI,CAAC,CAAC,EAAE,CAAC;YACP,CAAC,GAAG,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,EAAE,UAAU,EAAE,CAAC,EAAE,CAAC;YAC1C,MAAM,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;QACtB,CAAC;QACD,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACnB,CAAC,CAAC,UAAU,IAAI,CAAC,CAAC,IAAI,CAAC;IACzB,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,4EAA4E;AAC5E,0EAA0E;AAC1E,8EAA8E;AAC9E,4EAA4E;AAC5E,yEAAyE;AACzE,4DAA4D;AAC5D,MAAM,uBAAuB,GAAG,EAAE,CAAC;AACnC,MAAM,wBAAwB,GAAG,EAAE,CAAC;AACpC,MAAM,wBAAwB,GAAG,OAAO,CAAC;AACzC,MAAM,yBAAyB,GAAG,MAAM,CAAC;AAEzC,MAAM,UAAU,mBAAmB,CACjC,QAA6B,EAC7B,aAAqB,EACrB,YAAoB,EACpB,SAAiB,EACjB,OAA0B,EAC1B,YAAkC;IAElC,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CACzB,IAAI,CAAC,IAAI,CAAC,SAAS,GAAG,wBAAwB,CAAC,EAC/C,yBAAyB,CAC1B,CAAC;IACF,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CACxB,IAAI,CAAC,IAAI,CAAC,SAAS,GAAG,uBAAuB,CAAC,EAC9C,wBAAwB,CACzB,CAAC;IACF,OAAO,CACL,WAAW,CAAC,QAAQ,EAAE,aAAa,EAAE,UAAU,EAAE,GAAG,EAAE,OAAO,EAAE,YAAY,CAAC;QAC5E,WAAW,CAAC,QAAQ,EAAE,YAAY,EAAE,SAAS,EAAE,IAAI,EAAE,OAAO,EAAE,YAAY,CAAC,CAC5E,CAAC;AACJ,CAAC"}