@unerr-ai/unerr 0.2.1 → 0.2.3

package/dist/tracking/auto-snapshot-triggers.js

@@ -1,246 +0,0 @@
-/**
- * Auto-Snapshot Triggers — detects when to auto-create working snapshots.
- *
- * Three trigger types:
- *   1. test_pass: tool call result contains a test command + exit code 0
- *   2. session_end: graceful shutdown detected
- *   3. pre_critical_change: before modifying entity with fan_in above threshold
- *
- * Stateless per-call: each function evaluates the trigger condition independently
- * and returns a trigger descriptor or null.
- */
-import { createModuleLogger } from "../utils/logger.js";
-const log = createModuleLogger("auto-snapshot");
-const TEST_COMMANDS = [
-    "test",
-    "vitest",
-    "jest",
-    "mocha",
-    "pytest",
-    "cargo test",
-    "go test",
-    "npm test",
-    "pnpm test",
-    "yarn test",
-    "npx vitest",
-    "npx jest",
-    "pnpm run test",
-    "npm run test",
-    "pnpm exec vitest",
-    "pnpm exec jest",
-    "make test",
-    "rake test",
-    "rspec",
-    "phpunit",
-    "dotnet test",
-    "mvn test",
-    "gradle test",
-];
-const TEST_PATTERNS = [
-    /\bvitest\b/,
-    /\bjest\b/,
-    /\bmocha\b/,
-    /\bpytest\b/,
-    /\bcargo\s+test\b/,
-    /\bgo\s+test\b/,
-    /\bnpm\s+(?:run\s+)?test\b/,
-    /\bpnpm\s+(?:run\s+|exec\s+)?(?:vitest|jest|test)\b/,
-    /\byarn\s+(?:run\s+)?test\b/,
-    /\bnpx\s+(?:vitest|jest)\b/,
-    /\brspec\b/,
-    /\bphpunit\b/,
-    /\bdotnet\s+test\b/,
-    /\bmvn\s+test\b/,
-    /\bgradle\s+test\b/,
-    /\bmake\s+test\b/,
-    /\brake\s+test\b/,
-];
-const DEFAULT_FAN_IN_THRESHOLD = 50;
-/**
- * Determines if a string is a test command based on known patterns.
- */
-export function isTestCommand(command) {
-    const trimmed = command.trim().toLowerCase();
-    for (const known of TEST_COMMANDS) {
-        if (trimmed === known || trimmed.startsWith(`${known} `)) {
-            return true;
-        }
-    }
-    for (const pattern of TEST_PATTERNS) {
-        if (pattern.test(trimmed)) {
-            return true;
-        }
-    }
-    if (/\btest[_\-.]?(?:suite|run|all|unit|int|e2e|integration)\b/i.test(trimmed)) {
-        return true;
-    }
-    return false;
-}
-/**
- * Evaluates whether the given tool call + result should trigger an auto-snapshot.
- *
- * Returns a trigger descriptor if a snapshot should be taken, null otherwise.
- */
-export function shouldAutoSnapshot(toolName, args, result, fanInThreshold) {
-    const testTrigger = checkTestPass(toolName, args, result);
-    if (testTrigger)
-        return testTrigger;
-    const criticalTrigger = checkPreCriticalChange(toolName, args, result, fanInThreshold ?? DEFAULT_FAN_IN_THRESHOLD);
-    if (criticalTrigger)
-        return criticalTrigger;
-    return null;
-}
-/**
- * Creates a session-end trigger. Called during graceful shutdown.
- */
-export function createSessionEndTrigger() {
-    return { type: "session_end" };
-}
-function checkTestPass(toolName, args, result) {
-    const command = extractCommand(toolName, args);
-    if (!command)
-        return null;
-    if (!isTestCommand(command))
-        return null;
-    const exitCode = extractExitCode(result);
-    if (exitCode !== 0)
-        return null;
-    log.info(`Test pass detected: ${command}`);
-    return {
-        type: "test_pass",
-        triggerTool: toolName,
-    };
-}
-function checkPreCriticalChange(toolName, args, result, threshold) {
-    const modifyTools = [
-        "sync_local_diff",
-        "unerr_revert_entity",
-        "unerr_revert_to_working_state",
-    ];
-    if (!modifyTools.includes(toolName))
-        return null;
-    const entityKey = extractTargetEntity(args);
-    if (!entityKey)
-        return null;
-    const fanIn = extractFanIn(args, result);
-    if (fanIn === null || fanIn < threshold)
-        return null;
-    log.info(`Pre-critical-change trigger: ${entityKey} (fan_in=${fanIn}, threshold=${threshold})`);
-    return {
-        type: "pre_critical_change",
-        triggerTool: toolName,
-        entityKey,
-    };
-}
-function extractCommand(toolName, args) {
-    if (typeof args.command === "string")
-        return args.command;
-    if (toolName === "run_terminal_command" || toolName === "execute_command") {
-        if (typeof args.cmd === "string")
-            return args.cmd;
-        if (typeof args.shell_command === "string")
-            return args.shell_command;
-    }
-    if (typeof args.script === "string")
-        return args.script;
-    return null;
-}
-function extractExitCode(result) {
-    if (result === null || result === undefined)
-        return null;
-    if (typeof result === "object") {
-        const obj = result;
-        if (typeof obj.exitCode === "number")
-            return obj.exitCode;
-        if (typeof obj.exit_code === "number")
-            return obj.exit_code;
-        if (typeof obj.code === "number")
-            return obj.code;
-        if (typeof obj.content === "string") {
-            return parseExitCodeFromText(obj.content);
-        }
-        if (Array.isArray(obj.content)) {
-            for (const item of obj.content) {
-                if (typeof item === "object" && item !== null) {
-                    const textItem = item;
-                    if (typeof textItem.text === "string") {
-                        const code = parseExitCodeFromText(textItem.text);
-                        if (code !== null)
-                            return code;
-                    }
-                }
-            }
-        }
-    }
-    if (typeof result === "string") {
-        return parseExitCodeFromText(result);
-    }
-    return null;
-}
-function parseExitCodeFromText(text) {
-    const passPatterns = [
-        /all\s+(?:\d+\s+)?tests?\s+passed/i,
-        /\d+\s+passing/i,
-        /tests?\s+passed/i,
-        /exit\s+code[:\s]+0/i,
-        /exited\s+with\s+0/i,
-        /✓\s+\d+\s+tests?/i,
-        /Tests:\s+\d+\s+passed,\s+\d+\s+total/i,
-    ];
-    for (const pattern of passPatterns) {
-        if (pattern.test(text))
-            return 0;
-    }
-    const failPatterns = [
-        /\d+\s+failing/i,
-        /tests?\s+failed/i,
-        /exit\s+code[:\s]+([1-9]\d*)/i,
-        /FAIL/,
-    ];
-    for (const pattern of failPatterns) {
-        const match = pattern.exec(text);
-        if (match) {
-            if (match[1])
-                return Number.parseInt(match[1], 10);
-            return 1;
-        }
-    }
-    return null;
-}
-function extractTargetEntity(args) {
-    if (typeof args.entity_key === "string")
-        return args.entity_key;
-    if (typeof args.entityKey === "string")
-        return args.entityKey;
-    if (typeof args.file === "string")
-        return args.file;
-    if (typeof args.path === "string")
-        return args.path;
-    return null;
-}
-function extractFanIn(args, result) {
-    if (typeof args.fan_in === "number")
-        return args.fan_in;
-    if (typeof args.fanIn === "number")
-        return args.fanIn;
-    if (result !== null && typeof result === "object") {
-        const obj = result;
-        if (typeof obj.fan_in === "number")
-            return obj.fan_in;
-        if (typeof obj.fanIn === "number")
-            return obj.fanIn;
-        if (typeof obj.entity === "object" && obj.entity !== null) {
-            const entity = obj.entity;
-            if (typeof entity.fan_in === "number")
-                return entity.fan_in;
-            if (typeof entity.fanIn === "number")
-                return entity.fanIn;
-        }
-        if (typeof obj.blast_radius === "object" && obj.blast_radius !== null) {
-            const br = obj.blast_radius;
-            if (typeof br.direct_callers === "number")
-                return br.direct_callers;
-        }
-    }
-    return null;
-}

package/dist/tracking/branch-context.js

@@ -1,115 +0,0 @@
-/**
- * Branch Context Tracker — computes and maintains branch state.
- *
- * Tracks: current branch, base branch, merge base commit, commits ahead/behind.
- * Updated on proxy startup and on branch switch (detected via .git/HEAD polling).
- *
- * All git operations use the centralized git utility — no stdout pollution.
- */
-import { existsSync, readFileSync } from "node:fs";
-import { join } from "node:path";
-import { gitQuery } from "../utils/exec.js";
-/**
- * Compute branch context from git state.
- * Never throws — returns safe defaults on any git failure.
- */
-export async function computeBranchContext(cwd) {
-    const dir = cwd ?? process.cwd();
-    const currentBranch = (await gitQuery(["branch", "--show-current"], dir)) || "HEAD";
-    const headSha = (await gitQuery(["rev-parse", "HEAD"], dir)) || "";
-    let baseBranch = "main";
-    if (currentBranch && currentBranch !== "HEAD") {
-        const configured = await gitQuery(["config", "--get", `branch.${currentBranch}.merge`], dir);
-        if (configured) {
-            baseBranch = configured.replace(/^refs\/heads\//, "");
-        }
-        else {
-            const mainExists = await gitQuery(["rev-parse", "--verify", "origin/main"], dir);
-            if (!mainExists) {
-                const masterExists = await gitQuery(["rev-parse", "--verify", "origin/master"], dir);
-                if (masterExists)
-                    baseBranch = "master";
-            }
-        }
-    }
-    let baseCommit = null;
-    let commitsAhead = 0;
-    let commitsBehind = null;
-    const remoteBranch = `origin/${baseBranch}`;
-    baseCommit = await gitQuery(["merge-base", "HEAD", remoteBranch], dir);
-    if (baseCommit) {
-        const ahead = await gitQuery(["rev-list", "--count", `${baseCommit}..HEAD`], dir);
-        commitsAhead = ahead ? Number.parseInt(ahead, 10) : 0;
-        const behind = await gitQuery(["rev-list", "--count", `HEAD..${remoteBranch}`], dir);
-        commitsBehind = behind ? Number.parseInt(behind, 10) : null;
-    }
-    return {
-        currentBranch,
-        baseBranch,
-        baseCommit,
-        commitsAhead,
-        commitsBehind,
-        headSha,
-        computedAt: new Date().toISOString(),
-    };
-}
-/**
- * Alias retained for call-site compatibility — both versions are now async.
- */
-export const computeBranchContextAsync = computeBranchContext;
-/**
- * Detect if a branch switch occurred by comparing .git/HEAD content.
- * Returns the new branch name if switched, null otherwise.
- */
-export function detectBranchSwitch(previousBranch, cwd) {
-    const current = getCurrentBranch(cwd);
-    if (current && current !== previousBranch) {
-        return current;
-    }
-    return null;
-}
-/**
- * Get the current branch name by reading .git/HEAD directly (fast, no process spawn).
- */
-export function getCurrentBranch(cwd) {
-    const gitDir = cwd ? join(cwd, ".git") : join(process.cwd(), ".git");
-    const headPath = join(gitDir, "HEAD");
-    if (!existsSync(headPath))
-        return null;
-    try {
-        const content = readFileSync(headPath, "utf-8").trim();
-        // ref: refs/heads/feature/auth → feature/auth
-        if (content.startsWith("ref: refs/heads/")) {
-            return content.slice("ref: refs/heads/".length);
-        }
-        // Detached HEAD — return short SHA
-        return content.slice(0, 8);
-    }
-    catch {
-        return null;
-    }
-}
-/**
- * Get the current HEAD SHA via git rev-parse.
- */
-export async function getHeadSha(cwd) {
-    return (await gitQuery(["rev-parse", "HEAD"], cwd ?? process.cwd())) || "";
-}
-/**
- * Start a branch switch poller. Calls onSwitch when branch changes.
- * Returns a dispose function to stop polling.
- */
-export function startBranchPoller(onSwitch, intervalMs = 5000, cwd) {
-    let currentBranch = getCurrentBranch(cwd) ?? "unknown";
-    const timer = setInterval(async () => {
-        const switched = detectBranchSwitch(currentBranch, cwd);
-        if (switched) {
-            currentBranch = switched;
-            const context = await computeBranchContext(cwd);
-            onSwitch(switched, context);
-        }
-    }, intervalMs);
-    // Don't prevent process exit
-    timer.unref();
-    return () => clearInterval(timer);
-}

package/dist/tracking/branch-snapshot.js

@@ -1,217 +0,0 @@
-/**
- * Sprint 6.2: Per-Branch Overlay Snapshots — save/restore drift state on branch switch.
- *
- * On branch switch from A → B:
- *   1. SAVE: Serialize drift_overlay + file hashes → .unerr/drift/branches/{A}/
- *   2. CLEAR CozoDB drift_overlay + file hashes + mtime cache
- *   3. RESTORE: If branches/{B}/ snapshot exists, bulk insert back (<10ms)
- *              Else compute from scratch (first visit)
- *
- * Max 20 branch snapshots stored (LRU eviction).
- * Branches deleted from git are garbage-collected.
- */
-import { existsSync, mkdirSync, readFileSync, readdirSync, rmSync, statSync, writeFileSync, } from "node:fs";
-import { join } from "node:path";
-import { listBranches } from "../utils/git.js";
-import { createModuleLogger } from "../utils/logger.js";
-/** Maximum number of branch snapshots to retain (LRU). */
-const MAX_BRANCH_SNAPSHOTS = 20;
-const OVERLAY_FILE = "overlay_snapshot.json";
-const HASHES_FILE = "file_hashes.json";
-const log = createModuleLogger("branch");
-/**
- * Sanitize branch name for use as a directory name.
- * Replaces `/` with `__` and strips unsafe chars.
- */
-function sanitizeBranchName(branch) {
-    return branch.replace(/\//g, "__").replace(/[^a-zA-Z0-9_.\-]/g, "_");
-}
-export class BranchSnapshotManager {
-    branchDir;
-    projectRoot;
-    constructor(unerrDir, projectRoot) {
-        this.branchDir = join(unerrDir, "drift", "branches");
-        this.projectRoot = projectRoot;
-    }
-    /**
-     * Save the current drift overlay + file hashes for a branch.
-     * Called before clearing overlay on branch switch (save outgoing branch).
-     */
-    async saveSnapshot(branch, localGraph, fileHashState) {
-        const entities = await localGraph.getAllDriftEntities();
-        const edges = await localGraph.getAllDriftEdges();
-        if (entities.length === 0 && edges.length === 0) {
-            log.info(`No drift entities/edges to snapshot for branch ${branch}`);
-            // Still save empty snapshot so we know we visited this branch
-            // (avoids recompute on return if there truly was no drift)
-        }
-        const dirName = sanitizeBranchName(branch);
-        const snapshotDir = join(this.branchDir, dirName);
-        if (!existsSync(snapshotDir)) {
-            mkdirSync(snapshotDir, { recursive: true });
-        }
-        const snapshot = {
-            branch,
-            entities,
-            edges,
-            fileHashes: fileHashState,
-            savedAt: new Date().toISOString(),
-        };
-        writeFileSync(join(snapshotDir, OVERLAY_FILE), JSON.stringify(snapshot, null, 2), "utf-8");
-        writeFileSync(join(snapshotDir, HASHES_FILE), JSON.stringify(fileHashState, null, 2), "utf-8");
-        // Enforce LRU cap
-        this.enforceLruCap();
-        log.info(`Saved branch snapshot: ${branch} (${entities.length} entities, ${edges.length} edges)`);
-        return true;
-    }
-    /**
-     * Restore drift overlay from a branch snapshot.
-     * Returns the snapshot if found, null if this is a first visit.
-     */
-    async restoreSnapshot(branch, localGraph) {
-        const dirName = sanitizeBranchName(branch);
-        const snapshotDir = join(this.branchDir, dirName);
-        const overlayPath = join(snapshotDir, OVERLAY_FILE);
-        if (!existsSync(overlayPath)) {
-            log.info(`No snapshot for branch ${branch} — first visit`);
-            return null;
-        }
-        try {
-            const raw = readFileSync(overlayPath, "utf-8");
-            const snapshot = JSON.parse(raw);
-            // Bulk insert entities back into drift overlay
-            for (const entity of snapshot.entities) {
-                await localGraph.upsertDriftEntity(entity);
-            }
-            // Bulk insert edges back into drift_edges (Task 6.4)
-            if (snapshot.edges) {
-                for (const edge of snapshot.edges) {
-                    await localGraph.upsertDriftEdge(edge);
-                }
-            }
-            // Touch the snapshot dir to update LRU ordering
-            const now = new Date();
-            writeFileSync(join(snapshotDir, ".last_access"), now.toISOString(), "utf-8");
-            log.info(`Restored branch snapshot: ${branch} (${snapshot.entities.length} entities, ${snapshot.edges?.length ?? 0} edges)`);
-            return snapshot;
-        }
-        catch (err) {
-            log.warn(`Failed to restore snapshot for ${branch}: ${err instanceof Error ? err.message : String(err)}`);
-            return null;
-        }
-    }
-    /**
-     * Check if a snapshot exists for a given branch.
-     */
-    hasSnapshot(branch) {
-        const dirName = sanitizeBranchName(branch);
-        return existsSync(join(this.branchDir, dirName, OVERLAY_FILE));
-    }
-    /**
-     * Get the file hash state from a branch snapshot.
-     */
-    getSnapshotFileHashes(branch) {
-        const dirName = sanitizeBranchName(branch);
-        const hashesPath = join(this.branchDir, dirName, HASHES_FILE);
-        if (!existsSync(hashesPath))
-            return null;
-        try {
-            const raw = readFileSync(hashesPath, "utf-8");
-            return JSON.parse(raw);
-        }
-        catch {
-            return null;
-        }
-    }
-    /**
-     * Delete snapshot for a specific branch.
-     */
-    deleteSnapshot(branch) {
-        const dirName = sanitizeBranchName(branch);
-        const snapshotDir = join(this.branchDir, dirName);
-        if (!existsSync(snapshotDir))
-            return false;
-        rmSync(snapshotDir, { recursive: true, force: true });
-        log.info(`Deleted branch snapshot: ${branch}`);
-        return true;
-    }
-    /**
-     * Garbage-collect snapshots for branches that no longer exist in git.
-     */
-    async garbageCollect() {
-        if (!existsSync(this.branchDir))
-            return 0;
-        const gitBranches = new Set(await listBranches(this.projectRoot));
-        if (gitBranches.size === 0)
-            return 0;
-        const snapshots = this.listSnapshots();
-        let removed = 0;
-        for (const snapshot of snapshots) {
-            if (!gitBranches.has(snapshot.branch)) {
-                const snapshotDir = join(this.branchDir, snapshot.id);
-                rmSync(snapshotDir, { recursive: true, force: true });
-                log.info(`GC removed snapshot for deleted branch: ${snapshot.branch}`);
-                removed++;
-            }
-        }
-        return removed;
-    }
-    /**
-     * List all branch snapshots, sorted by most recently accessed first.
-     */
-    listSnapshots() {
-        if (!existsSync(this.branchDir))
-            return [];
-        try {
-            const entries = readdirSync(this.branchDir, { withFileTypes: true });
-            const snapshots = [];
-            for (const entry of entries) {
-                if (!entry.isDirectory())
-                    continue;
-                const overlayPath = join(this.branchDir, entry.name, OVERLAY_FILE);
-                if (!existsSync(overlayPath))
-                    continue;
-                try {
-                    const raw = readFileSync(overlayPath, "utf-8");
-                    const snapshot = JSON.parse(raw);
-                    // Use .last_access if available, else overlay file mtime
-                    const accessPath = join(this.branchDir, entry.name, ".last_access");
-                    let accessedAt;
-                    if (existsSync(accessPath)) {
-                        accessedAt = statSync(accessPath).mtime;
-                    }
-                    else {
-                        accessedAt = statSync(overlayPath).mtime;
-                    }
-                    snapshots.push({
-                        id: entry.name,
-                        branch: snapshot.branch,
-                        accessedAt,
-                    });
-                }
-                catch {
-                    // Skip corrupt snapshots
-                }
-            }
-            snapshots.sort((a, b) => b.accessedAt.getTime() - a.accessedAt.getTime());
-            return snapshots;
-        }
-        catch {
-            return [];
-        }
-    }
-    /**
-     * Enforce LRU cap — remove oldest snapshots beyond MAX_BRANCH_SNAPSHOTS.
-     */
-    enforceLruCap() {
-        const snapshots = this.listSnapshots();
-        if (snapshots.length <= MAX_BRANCH_SNAPSHOTS)
-            return;
-        const toRemove = snapshots.slice(MAX_BRANCH_SNAPSHOTS);
-        for (const snapshot of toRemove) {
-            const dir = join(this.branchDir, snapshot.id);
-            rmSync(dir, { recursive: true, force: true });
-            log.info(`LRU evicted branch snapshot: ${snapshot.branch}`);
-        }
-    }
-}