pi-lens 4.2.1 → 4.4.0

package/dist/clients/cache-manager.js

@@ -1,465 +0,0 @@
-/**
- * CacheManager for pi-lens.
- *
- * Manages persistent cache for scanner results and turn state.
- * Provides read/write/freshness checks for:
- * - Scanner cache: <project-data-dir>/cache/{scanner}.json
- * - Turn state: <project-data-dir>/turn-state.json
- *
- * `<project-data-dir>` is `getProjectDataDir(cwd)` (clients/file-utils.ts) --
- * `<cwd>/.pi-lens` only when that legacy directory already exists, otherwise
- * `~/.pi-lens/projects/<slug>` or a `PILENS_DATA_DIR` location. Never assume
- * the legacy spelling, and never show it to an agent: build a displayed path
- * with `displayProjectDataPath` (#2521).
- */
-import { createSubsystemLogger } from "./extension-log.js";
-import * as fs from "node:fs";
-import * as path from "node:path";
-import { getProjectDataDir } from "./file-utils.js";
-import { writeFileAtomic } from "./atomic-write.js";
-import { readJsonCache } from "./json-cache-read.js";
-import { isUnderDir, normalizeMapKey } from "./path-utils.js";
-// --- Defaults ---
-const DEFAULT_MAX_AGE_MS = 30 * 60 * 1000; // 30 minutes
-const DEFAULT_TURN_STATE = {
-    files: {},
-    turnCycles: 0,
-    maxCycles: 3,
-    lastUpdated: "",
-};
-export const MCP_TURN_STATE_OWNER_ID = `mcp-${process.pid}`;
-const TURN_OWNER_STALE_MS = 30 * 60 * 1000;
-// --- Helpers ---
-function getLensDir(cwd) {
-    return getProjectDataDir(cwd);
-}
-function getCacheDir(cwd) {
-    return path.join(getLensDir(cwd), "cache");
-}
-function getTurnStatePath(cwd) {
-    return path.join(getLensDir(cwd), "turn-state.json");
-}
-// --- Cache Manager ---
-export class CacheManager {
-    log;
-    constructor(verbose = false) {
-        this.log = verbose ? createSubsystemLogger("cache") : () => { };
-    }
-    /**
-     * Convert a file path to a stable turn-state key.
-     * Uses normalized absolute paths first, then stores cwd-relative keys when possible.
-     */
-    toTurnStateKey(filePath, cwd) {
-        const cwdNorm = normalizeMapKey(path.resolve(cwd));
-        const fileNorm = normalizeMapKey(path.resolve(cwd, filePath));
-        const rel = path.relative(cwdNorm, fileNorm).replace(/\\/g, "/");
-        if (!rel || rel === ".")
-            return fileNorm;
-        if (rel === ".." || rel.startsWith("../"))
-            return fileNorm;
-        return rel;
-    }
-    /**
-     * Get turn-state entry for a file path using normalized lookup.
-     */
-    /**
-     * #2504: is this path inside the PROJECT the worklist belongs to?
-     *
-     * Routed through `isUnderDir` (which normalises via `normalizeFilePath`) so
-     * the answer is separator- and case-form independent, the same key rule
-     * every other turn-state map obeys.
-     *
-     * The root is the PROJECT root, not the caller's `cwd` (#2504 review round
-     * 2, F1). The two coincide for every producer whose `cwd` IS the session
-     * root, but not for `clients/lsp-mutation.ts`: `tools/lsp-navigation.ts`
-     * threads a `cwd`-scoped context for a call issued from a sub-package
-     * directory, so a monorepo-wide server's edit on a SIBLING package is
-     * inside the project and outside that `cwd`. Judging it against `cwd`
-     * dropped exactly the server-initiated edits #2450/#2479 exist to record.
-     * This is the same root `isRecordableProjectPath` (clients/file-utils.ts)
-     * is given at that call site — one predicate root, taken from
-     * `runtime.projectRoot`, not two that can disagree.
-     */
-    isTurnStatePathWithinRoot(filePath, projectRoot) {
-        const root = path.resolve(projectRoot);
-        const abs = path.resolve(root, filePath);
-        if (normalizeMapKey(abs) === normalizeMapKey(root))
-            return false;
-        return isUnderDir(abs, root);
-    }
-    getTurnFileState(filePath, cwd) {
-        const state = this.readTurnState(cwd);
-        const key = this.toTurnStateKey(filePath, cwd);
-        return state.files[key];
-    }
-    // ---- Scanner Cache ----
-    /**
-     * Read a scanner cache entry. Returns null if not found or stale.
-     */
-    readCache(scanner, cwd, maxAgeMs = DEFAULT_MAX_AGE_MS) {
-        const cachePath = path.join(getCacheDir(cwd), `${scanner}.json`);
-        const metaPath = path.join(getCacheDir(cwd), `${scanner}.meta.json`);
-        if (!fs.existsSync(cachePath) || !fs.existsSync(metaPath)) {
-            this.log(`Cache miss: ${scanner} (files don't exist)`);
-            return null;
-        }
-        try {
-            const onReadError = (err) => {
-                this.log(`Cache read error: ${scanner} — ${err}`);
-            };
-            const meta = readJsonCache(metaPath, (parsed) => parsed, onReadError);
-            if (meta === undefined)
-                return null;
-            const age = Date.now() - new Date(meta.timestamp).getTime();
-            if (age > maxAgeMs) {
-                this.log(`Cache stale: ${scanner} (age: ${Math.round(age / 1000)}s, max: ${maxAgeMs / 1000}s)`);
-                return null;
-            }
-            const data = readJsonCache(cachePath, (parsed) => parsed, onReadError);
-            if (data === undefined)
-                return null;
-            this.log(`Cache hit: ${scanner} (age: ${Math.round(age / 1000)}s)`);
-            return { data, meta };
-        }
-        catch (err) {
-            this.log(`Cache read error: ${scanner} — ${err}`);
-            return null;
-        }
-    }
-    /**
-     * Write a scanner cache entry.
-     */
-    writeCache(scanner, data, cwd, extraMeta) {
-        const cacheDir = getCacheDir(cwd);
-        fs.mkdirSync(cacheDir, { recursive: true });
-        const cachePath = path.join(cacheDir, `${scanner}.json`);
-        const metaPath = path.join(cacheDir, `${scanner}.meta.json`);
-        const meta = {
-            timestamp: new Date().toISOString(),
-            ...extraMeta,
-        };
-        writeFileAtomic(cachePath, JSON.stringify(data, null, 2), {
-            bestEffort: false,
-        });
-        writeFileAtomic(metaPath, JSON.stringify(meta, null, 2), {
-            bestEffort: false,
-        });
-        this.log(`Cache written: ${scanner}`);
-    }
-    /** Inspect a cache without changing the behavior of readCache consumers. */
-    inspectCache(scanner, cwd, maxAgeMs = DEFAULT_MAX_AGE_MS) {
-        const cachePath = path.join(getCacheDir(cwd), `${scanner}.json`);
-        const metaPath = path.join(getCacheDir(cwd), `${scanner}.meta.json`);
-        for (const cachePathname of [cachePath, metaPath]) {
-            try {
-                fs.statSync(cachePathname);
-            }
-            catch (err) {
-                if (err.code === "ENOENT")
-                    return "missing";
-                return "unreadable";
-            }
-        }
-        try {
-            const meta = readJsonCache(metaPath, (parsed) => parsed);
-            if (!meta || typeof meta.timestamp !== "string")
-                return "malformed";
-            const timestamp = new Date(meta.timestamp).getTime();
-            if (!Number.isFinite(timestamp))
-                return "malformed";
-            const age = Date.now() - timestamp;
-            if (age < 0 || age > maxAgeMs)
-                return age < 0 ? "malformed" : "stale";
-            const data = readJsonCache(cachePath, (parsed) => parsed);
-            return data === undefined ? "malformed" : "fresh";
-        }
-        catch {
-            return "unreadable";
-        }
-    }
-    /**
-     * Check if a cache entry is fresh (exists and not expired).
-     */
-    isCacheFresh(scanner, cwd, maxAgeMs = DEFAULT_MAX_AGE_MS) {
-        const metaPath = path.join(getCacheDir(cwd), `${scanner}.meta.json`);
-        if (!fs.existsSync(metaPath))
-            return false;
-        try {
-            const meta = readJsonCache(metaPath, (parsed) => parsed);
-            if (meta === undefined)
-                return false;
-            const age = Date.now() - new Date(meta.timestamp).getTime();
-            return age <= maxAgeMs;
-        }
-        catch {
-            return false;
-        }
-    }
-    /**
-     * Clear a specific cache entry.
-     */
-    clearCache(scanner, cwd) {
-        const cachePath = path.join(getCacheDir(cwd), `${scanner}.json`);
-        const metaPath = path.join(getCacheDir(cwd), `${scanner}.meta.json`);
-        for (const p of [cachePath, metaPath]) {
-            try {
-                fs.unlinkSync(p);
-            }
-            catch (err) {
-                // ENOENT: file doesn't exist, other errors logged
-                if (err.code !== "ENOENT") {
-                    this.log(`Failed to delete ${p}: ${err}`);
-                }
-            }
-        }
-    }
-    // ---- Turn State ----
-    /**
-     * Read turn state. Returns default if not found.
-     */
-    readTurnState(cwd) {
-        const statePath = getTurnStatePath(cwd);
-        if (!fs.existsSync(statePath)) {
-            return {
-                ...DEFAULT_TURN_STATE,
-                files: {},
-                lastUpdated: new Date().toISOString(),
-            };
-        }
-        const state = readJsonCache(statePath, (parsed) => parsed);
-        return (state ?? {
-            ...DEFAULT_TURN_STATE,
-            files: {},
-            lastUpdated: new Date().toISOString(),
-        });
-    }
-    /**
-     * Write turn state.
-     */
-    writeTurnState(state, cwd) {
-        const lensDir = getLensDir(cwd);
-        fs.mkdirSync(lensDir, { recursive: true });
-        const statePath = getTurnStatePath(cwd);
-        state.lastUpdated = new Date().toISOString();
-        writeFileAtomic(statePath, JSON.stringify(state, null, 2));
-    }
-    /** Return whether a writer may read/write this workspace worklist. */
-    getTurnStateAccess(cwd, owner) {
-        const state = this.readTurnState(cwd);
-        const hasFiles = Object.keys(state.files ?? {}).length > 0;
-        if (!state.owner) {
-            if (!state.sessionId || state.sessionId === owner.id) {
-                // #2504: an OWNERLESS state was unconditionally "owned" here, and
-                // ownerless was the resting shape (pre-#2504 `clearTurnState`
-                // dropped the owner and `addModifiedRange` only stamps one when a
-                // sessionId is supplied). A worklist persisted by a session that
-                // has since exited was therefore adopted wholesale by the next
-                // session — 154 historical paths reported as "modified this turn"
-                // on a turn that made no tool calls at all. A worklist whose last
-                // write predates this session's start is not this session's work.
-                if (hasFiles && this.turnStatePredatesSession(state, owner)) {
-                    return "available";
-                }
-                return "owned";
-            }
-            // Pre-owner files have no liveness information. Preserve the existing
-            // stale-session eviction behavior for this legacy shape.
-            return "available";
-        }
-        if (state.owner.kind === owner.kind && state.owner.id === owner.id) {
-            return "owned";
-        }
-        // #2504: an EMPTY worklist carries nothing to protect, so it never
-        // latches a foreign writer out. This keeps the new `clearTurnState`
-        // owner stamp from turning a cleared worklist into a cross-process lock:
-        // an MCP writer in another process must still be able to register work
-        // against a worklist a pi session just cleared, exactly as it could when
-        // clearing left the state ownerless.
-        if (!hasFiles)
-            return "owned";
-        return this.isTurnStateOwnerStale(state.owner)
-            ? "available"
-            : "foreign-live";
-    }
-    /**
-     * #2504: true when this worklist was last written before the asking
-     * session began. `lastUpdated` is restamped by `writeTurnState` on EVERY
-     * write (add/clear/cycle), so a live session's own worklist is always
-     * newer than its start; only a carried-over file can be older. An
-     * unparseable stamp is treated as stale — a worklist we cannot date is
-     * exactly the legacy shape this gate exists to evict.
-     */
-    turnStatePredatesSession(state, owner) {
-        const startedAt = owner.sessionStartedAt;
-        if (startedAt === undefined || !Number.isFinite(startedAt))
-            return false;
-        const lastUpdated = Date.parse(state.lastUpdated ?? "");
-        if (!Number.isFinite(lastUpdated))
-            return true;
-        return lastUpdated < startedAt;
-    }
-    isTurnStateOwnerStale(owner) {
-        if (owner.pid > 0 && owner.pid !== process.pid) {
-            try {
-                process.kill(owner.pid, 0);
-                return false;
-            }
-            catch {
-                return true;
-            }
-        }
-        const lastSeen = Date.parse(owner.lastSeen);
-        return (!Number.isFinite(lastSeen) || Date.now() - lastSeen > TURN_OWNER_STALE_MS);
-    }
-    /**
-     * Add or update a file's modified ranges in turn state.
-     * Merges overlapping ranges. `sessionId:null` deliberately preserves the
-     * current owner; callers must provide an explicit owner id to claim a stale
-     * worklist.
-     */
-    addModifiedRange(filePath, range, importsChanged, cwd, sessionId, ownerKind = "pi", projectRoot) {
-        // #2504: the worklist is a PROJECT worklist. A path outside the project
-        // was accepted and keyed by its absolute path, so a prior session's
-        // scratchpad, `~/.claude/plans/*.md` and `~/.plegma/work/.../TASK.md`
-        // all became "modified files" that turn_end then fed to the test runner
-        // and opened in an LSP client. Reject before the owner stamp below, so
-        // an out-of-project write cannot claim the worklist either.
-        //
-        // `projectRoot` defaults to `cwd` — true for every producer whose cwd
-        // IS the session root. A caller whose `cwd` is NARROWER than the
-        // project (an LSP call issued from a sub-package directory) passes the
-        // real root explicitly; `cwd` still selects which `turn-state.json`
-        // the entry lands in, exactly as before (#2504 review round 2, F1).
-        const containmentRoot = projectRoot ?? cwd;
-        if (!this.isTurnStatePathWithinRoot(filePath, containmentRoot)) {
-            this.log(`turn-state: rejected out-of-project path ${filePath} (project root ${containmentRoot})`);
-            return this.readTurnState(cwd);
-        }
-        const state = this.readTurnState(cwd);
-        if (sessionId) {
-            const owner = {
-                kind: ownerKind,
-                id: sessionId,
-                pid: process.pid,
-                lastSeen: new Date().toISOString(),
-            };
-            if (this.getTurnStateAccess(cwd, owner) === "foreign-live")
-                return state;
-            state.sessionId = sessionId;
-            state.owner = owner;
-        }
-        const normalizedPath = this.toTurnStateKey(filePath, cwd);
-        const existing = state.files[normalizedPath];
-        if (existing) {
-            // Merge ranges
-            existing.modifiedRanges = this.mergeRanges([
-                ...existing.modifiedRanges,
-                range,
-            ]);
-            existing.importsChanged = existing.importsChanged || importsChanged;
-            existing.lastEdit = new Date().toISOString();
-        }
-        else {
-            state.files[normalizedPath] = {
-                modifiedRanges: [range],
-                importsChanged,
-                lastEdit: new Date().toISOString(),
-            };
-        }
-        this.writeTurnState(state, cwd);
-        return state;
-    }
-    /**
-     * Clear turn state (after turn_end processes it).
-     */
-    clearTurnState(cwd, owner) {
-        const currentState = this.readTurnState(cwd);
-        const isCurrentOwner = this.getTurnStateAccess(cwd, owner) !== "foreign-live";
-        if (!isCurrentOwner && process.pid !== currentState.owner?.pid)
-            return false;
-        // #2504: stamp the CLEARING owner. Dropping `owner`/`sessionId` here made
-        // ownerless the resting shape of turn-state.json, which is what let the
-        // next session read a carried-over worklist as its own. A cleared state
-        // now says who cleared it and when, so `getTurnStateAccess` can judge it
-        // by liveness like any other owned state — and, since the worklist is
-        // empty, the stamp never latches another writer out (see the
-        // empty-worklist branch there).
-        const state = {
-            ...DEFAULT_TURN_STATE,
-            files: {}, // fresh object — DEFAULT_TURN_STATE.files can be polluted by addModifiedRange
-            lastUpdated: new Date().toISOString(),
-            sessionId: owner.id,
-            owner: {
-                kind: owner.kind,
-                id: owner.id,
-                pid: process.pid,
-                lastSeen: new Date().toISOString(),
-                sessionStartedAt: owner.sessionStartedAt,
-            },
-        };
-        this.writeTurnState(state, cwd);
-        return true;
-    }
-    /**
-     * Increment turn cycle counter.
-     */
-    incrementTurnCycle(cwd, owner) {
-        const state = this.readTurnState(cwd);
-        const isCurrentOwner = this.getTurnStateAccess(cwd, owner) !== "foreign-live";
-        if (!isCurrentOwner && process.pid !== state.owner?.pid)
-            return state;
-        state.turnCycles++;
-        this.writeTurnState(state, cwd);
-        return state;
-    }
-    /**
-     * Check if max cycles exceeded.
-     */
-    isMaxCyclesExceeded(cwd) {
-        const state = this.readTurnState(cwd);
-        return state.turnCycles >= state.maxCycles;
-    }
-    /**
-     * Get files that need jscpd re-scan (any edit).
-     * Only returns source code files jscpd can meaningfully analyse.
-     */
-    getFilesForJscpd(cwd) {
-        const state = this.readTurnState(cwd);
-        return Object.keys(state.files).filter((f) => /\.(ts|tsx|js|jsx|mjs|cjs|py|go|rs|rb|java|cs|php|cpp|c|h|hpp|swift|kt)$/.test(f));
-    }
-    /**
-     * Get files that need madge re-scan (imports changed).
-     */
-    getFilesForMadge(cwd) {
-        const state = this.readTurnState(cwd);
-        return Object.entries(state.files)
-            .filter(([, f]) => f.importsChanged)
-            .map(([p]) => p);
-    }
-    // ---- Utilities ----
-    /**
-     * Merge overlapping or adjacent ranges.
-     */
-    mergeRanges(ranges) {
-        if (ranges.length === 0)
-            return [];
-        const sorted = [...ranges].sort((a, b) => a.start - b.start);
-        const merged = [sorted[0]];
-        for (const current of sorted.slice(1)) {
-            const last = merged[merged.length - 1];
-            if (current.start <= last.end + 1) {
-                last.end = Math.max(last.end, current.end);
-            }
-            else {
-                merged.push({ ...current });
-            }
-        }
-        return merged;
-    }
-    /**
-     * Check if a line falls within any modified range.
-     */
-    isLineInModifiedRange(line, ranges) {
-        return ranges.some((r) => r.start <= line && line <= r.end);
-    }
-}