project-librarian 0.4.0 → 0.4.2

package/dist/init-project-wiki.js

@@ -2,6 +2,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 const args_1 = require("./args");
+const agent_surfaces_1 = require("./agent-surfaces");
 const hooks_1 = require("./hooks");
 const install_skill_1 = require("./install-skill");
 const modes_1 = require("./modes");
@@ -39,10 +40,12 @@ Options:
   --refresh-index                  Update the managed auto-discovered wiki index block.
   --capture-inbox                  Append a candidate note with --title, --content, and optional --category.
   --glossary-init                  Create and route the optional glossary page.
+  --agents <list>                  With init/update, write only selected agent surfaces: codex, claude, cursor, gemini, or all. Existing project skill/setup surfaces are preserved by default.
   --prune-check                    Report active pages with stale or unresolved signals.
   --review-migration               Sync unit coverage and compatible inbox statuses into migration review files.
   --no-git-config                  Install hook files without changing git core.hooksPath.
   --code-index                     Build the disposable .project-wiki code evidence index.
+  --code-index-health              Inspect code evidence cache compatibility and print rebuild guidance without writing.
   --acknowledge-small-repo         With --code-index, proceed below the small-repo scale gate after its cost warning.
   --incremental                    With --code-index, require an existing compatible index and update only changes.
   --code-index-full                With --code-index, force a full rebuild even when incremental update is possible.
@@ -68,6 +71,20 @@ Commands:
 function exitAfterStdoutDrain(code) {
     process.stdout.write("", () => process.exit(code));
 }
+function activeCodeEvidenceCliModes() {
+    const modes = [
+        { active: args_1.codeQueryMode, drainStdout: true, run: (codeIndexModule) => codeIndexModule.runCodeQueryMode() },
+        { active: args_1.codeReportMode, drainStdout: true, run: (codeIndexModule) => codeIndexModule.runCodeReportMode() },
+        { active: args_1.codeStatusMode, drainStdout: true, run: (codeIndexModule) => codeIndexModule.runCodeStatusMode() },
+        { active: args_1.codeFilesMode, drainStdout: true, run: (codeIndexModule) => codeIndexModule.runCodeFilesMode() },
+        { active: args_1.codeImpactMode, drainStdout: true, run: (codeIndexModule) => codeIndexModule.runCodeImpactMode() },
+        { active: args_1.codeContextPackMode, drainStdout: true, run: (codeIndexModule) => codeIndexModule.runCodeContextPackMode() },
+        { active: args_1.codeSearchSymbolMode, drainStdout: true, run: (codeIndexModule) => codeIndexModule.runCodeSearchSymbolMode() },
+        { active: args_1.codeIndexHealthMode, drainStdout: true, run: (codeIndexModule) => codeIndexModule.runCodeIndexHealthMode() },
+        { active: args_1.codeIndexMode, drainStdout: false, run: (codeIndexModule) => codeIndexModule.runCodeIndexMode() },
+    ];
+    return modes.filter((mode) => mode.active);
+}
 if (args_1.helpMode) {
     printUsage();
     process.exit(0);
@@ -92,6 +109,11 @@ if (args_1.missingValueOptions.length > 0) {
     printUsage();
     process.exit(1);
 }
+if (args_1.invalidAgentTargets.length > 0) {
+    console.error(`invalid --agents entr${args_1.invalidAgentTargets.length === 1 ? "y" : "ies"}: ${args_1.invalidAgentTargets.join(", ")}; expected codex, claude, cursor, gemini, or all`);
+    printUsage();
+    process.exit(1);
+}
 if (args_1.command === "update" && args_1.migrateMode) {
     console.error("update cannot be combined with --migrate or --adopt-existing; use project-librarian --migrate for migration.");
     process.exit(1);
@@ -147,50 +169,20 @@ else {
     runInitCommand();
 }
 function runInitCommand() {
-    const activeCodeModes = [args_1.codeQueryMode, args_1.codeReportMode, args_1.codeStatusMode, args_1.codeFilesMode, args_1.codeImpactMode, args_1.codeContextPackMode, args_1.codeSearchSymbolMode, args_1.codeIndexMode].filter(Boolean).length;
-    if (activeCodeModes > 1) {
-        console.error("Use one code evidence mode at a time: --code-index, --code-query, --code-report, --code-status, --code-files, --code-impact, --code-context-pack, or --code-search-symbol.");
+    const activeCodeModes = activeCodeEvidenceCliModes();
+    if (activeCodeModes.length > 1) {
+        console.error("Use one code evidence mode at a time: --code-index, --code-index-health, --code-query, --code-report, --code-status, --code-files, --code-impact, --code-context-pack, or --code-search-symbol.");
         process.exit(1);
     }
-    if (args_1.codeQueryMode) {
-        codeIndex().runCodeQueryMode();
-        exitAfterStdoutDrain(0);
-        return;
-    }
-    if (args_1.codeReportMode) {
-        codeIndex().runCodeReportMode();
-        exitAfterStdoutDrain(0);
-        return;
-    }
-    if (args_1.codeStatusMode) {
-        codeIndex().runCodeStatusMode();
-        exitAfterStdoutDrain(0);
-        return;
-    }
-    if (args_1.codeFilesMode) {
-        codeIndex().runCodeFilesMode();
-        exitAfterStdoutDrain(0);
-        return;
-    }
-    if (args_1.codeImpactMode) {
-        codeIndex().runCodeImpactMode();
-        exitAfterStdoutDrain(0);
-        return;
-    }
-    if (args_1.codeContextPackMode) {
-        codeIndex().runCodeContextPackMode();
-        exitAfterStdoutDrain(0);
-        return;
-    }
-    if (args_1.codeSearchSymbolMode) {
-        codeIndex().runCodeSearchSymbolMode();
-        exitAfterStdoutDrain(0);
+    const activeCodeMode = activeCodeModes[0];
+    if (activeCodeMode) {
+        activeCodeMode.run(codeIndex());
+        if (activeCodeMode.drainStdout)
+            exitAfterStdoutDrain(0);
+        else
+            process.exit(0);
         return;
     }
-    if (args_1.codeIndexMode) {
-        codeIndex().runCodeIndexMode();
-        process.exit(0);
-    }
     if (args_1.wikiImpactMode) {
         (0, modes_1.runWikiImpactMode)();
         exitAfterStdoutDrain(0);
@@ -258,6 +250,20 @@ function runInitCommand() {
         (0, modes_1.runLintMode)();
         process.exit(0);
     }
+    if (args_1.refreshIndexMode && !args_1.migrateMode && !args_1.glossaryMode && !args_1.captureInboxMode) {
+        runRefreshIndexOnlyMode();
+        process.exit(0);
+    }
+    const selectedAgentSurfaces = args_1.agentTargets.length > 0
+        ? args_1.agentTargets
+        : args_1.migrateMode
+            ? Array.from(agent_surfaces_1.allAgentSurfaces)
+            : (0, agent_surfaces_1.resolveBootstrapAgentSurfaces)(args_1.agentTargets, workspace_1.exists, workspace_1.read);
+    const shouldWriteSurface = (surface) => (0, agent_surfaces_1.includesAgentSurface)(selectedAgentSurfaces, surface);
+    const writeCodexSurface = shouldWriteSurface("codex");
+    const writeClaudeSurface = shouldWriteSurface("claude");
+    const writeCursorSurface = shouldWriteSurface("cursor");
+    const writeGeminiSurface = shouldWriteSurface("gemini");
     const migrationState = args_1.migrateMode ? (0, migration_1.prepareMigrationMode)() : null;
     const results = [];
     if (migrationState)
@@ -267,11 +273,16 @@ function runInitCommand() {
     (0, workspace_1.mkdirp)("wiki/inbox");
     (0, workspace_1.mkdirp)("wiki/meta");
     (0, workspace_1.mkdirp)("wiki/sources");
-    (0, workspace_1.mkdirp)(".codex/hooks");
-    (0, workspace_1.mkdirp)(".claude/hooks");
-    (0, workspace_1.mkdirp)(".cursor/hooks");
-    (0, workspace_1.mkdirp)(".cursor/rules");
-    (0, workspace_1.mkdirp)(".gemini/hooks");
+    if (writeCodexSurface)
+        (0, workspace_1.mkdirp)(".codex/hooks");
+    if (writeClaudeSurface)
+        (0, workspace_1.mkdirp)(".claude/hooks");
+    if (writeCursorSurface) {
+        (0, workspace_1.mkdirp)(".cursor/hooks");
+        (0, workspace_1.mkdirp)(".cursor/rules");
+    }
+    if (writeGeminiSurface)
+        (0, workspace_1.mkdirp)(".gemini/hooks");
     (0, workspace_1.mkdirp)(".githooks");
     // B1 fallback: sync the CURRENT startup.md TL;DR into the managed AGENTS.md block
     // so non-interactive `codex exec` (which does not run SessionStart hooks) still
@@ -282,23 +293,34 @@ function runInitCommand() {
     const startupForSync = (0, workspace_1.exists)("wiki/startup.md") ? (0, workspace_1.read)("wiki/startup.md") : templates_1.startup;
     const startupTldrForAgents = (0, templates_1.extractStartupTldr)(startupForSync);
     results.push(["AGENTS.md", (0, workspace_1.upsertMarkedSection)("AGENTS.md", "<!-- PROJECT-WIKI-FIRST:START -->", "<!-- PROJECT-WIKI-FIRST:END -->", (0, templates_1.agentsSection)(startupTldrForAgents))]);
-    results.push(["CLAUDE.md", (0, workspace_1.upsertMarkedSection)("CLAUDE.md", "<!-- PROJECT-WIKI-CLAUDE:START -->", "<!-- PROJECT-WIKI-CLAUDE:END -->", templates_1.claudeSection)]);
-    results.push(["GEMINI.md", (0, workspace_1.upsertMarkedSection)("GEMINI.md", "<!-- PROJECT-WIKI-GEMINI:START -->", "<!-- PROJECT-WIKI-GEMINI:END -->", templates_1.geminiSection)]);
-    results.push([".cursor/rules/project-librarian.mdc", (0, workspace_1.writeManaged)(".cursor/rules/project-librarian.mdc", templates_1.cursorRule)]);
+    if (writeClaudeSurface)
+        results.push(["CLAUDE.md", (0, workspace_1.upsertMarkedSection)("CLAUDE.md", "<!-- PROJECT-WIKI-CLAUDE:START -->", "<!-- PROJECT-WIKI-CLAUDE:END -->", templates_1.claudeSection)]);
+    if (writeGeminiSurface)
+        results.push(["GEMINI.md", (0, workspace_1.upsertMarkedSection)("GEMINI.md", "<!-- PROJECT-WIKI-GEMINI:START -->", "<!-- PROJECT-WIKI-GEMINI:END -->", templates_1.geminiSection)]);
+    if (writeCursorSurface)
+        results.push([".cursor/rules/project-librarian.mdc", (0, workspace_1.writeManaged)(".cursor/rules/project-librarian.mdc", templates_1.cursorRule)]);
     results.push(["wiki/AGENTS.md", (0, workspace_1.upsertMarkedSection)("wiki/AGENTS.md", "<!-- PROJECT-WIKI-INTERNAL:START -->", "<!-- PROJECT-WIKI-INTERNAL:END -->", templates_1.wikiAgentsSection)]);
     results.push([".githooks/prepare-commit-msg", (0, workspace_1.writeManaged)(".githooks/prepare-commit-msg", hooks_1.gitPrepareCommitMsgHook)]);
     (0, workspace_1.makeExecutable)(".githooks/prepare-commit-msg");
     results.push([".githooks/wiki-commit-trailers.js", (0, workspace_1.writeManaged)(".githooks/wiki-commit-trailers.js", hooks_1.gitWikiCommitTrailersScript)]);
     (0, workspace_1.makeExecutable)(".githooks/wiki-commit-trailers.js");
     results.push(["git core.hooksPath", (0, hooks_1.upsertGitHooksPath)()]);
-    results.push([".codex/hooks.json", (0, hooks_1.upsertHookConfig)()]);
-    results.push([".codex/hooks/wiki-session-start.js", (0, workspace_1.writeManaged)(".codex/hooks/wiki-session-start.js", hooks_1.hookScript)]);
-    results.push([".claude/settings.json", (0, hooks_1.upsertClaudeHookConfig)()]);
-    results.push([".claude/hooks/wiki-session-start.js", (0, workspace_1.writeManaged)(".claude/hooks/wiki-session-start.js", hooks_1.hookScript)]);
-    results.push([".cursor/hooks.json", (0, hooks_1.upsertCursorHookConfig)()]);
-    results.push([".cursor/hooks/wiki-session-start.js", (0, workspace_1.writeManaged)(".cursor/hooks/wiki-session-start.js", hooks_1.cursorHookScript)]);
-    results.push([".gemini/settings.json", (0, hooks_1.upsertGeminiHookConfig)()]);
-    results.push([".gemini/hooks/wiki-session-start.js", (0, workspace_1.writeManaged)(".gemini/hooks/wiki-session-start.js", hooks_1.hookScript)]);
+    if (writeCodexSurface) {
+        results.push([".codex/hooks.json", (0, hooks_1.upsertHookConfig)()]);
+        results.push([".codex/hooks/wiki-session-start.js", (0, workspace_1.writeManaged)(".codex/hooks/wiki-session-start.js", hooks_1.hookScript)]);
+    }
+    if (writeClaudeSurface) {
+        results.push([".claude/settings.json", (0, hooks_1.upsertClaudeHookConfig)()]);
+        results.push([".claude/hooks/wiki-session-start.js", (0, workspace_1.writeManaged)(".claude/hooks/wiki-session-start.js", hooks_1.hookScript)]);
+    }
+    if (writeCursorSurface) {
+        results.push([".cursor/hooks.json", (0, hooks_1.upsertCursorHookConfig)()]);
+        results.push([".cursor/hooks/wiki-session-start.js", (0, workspace_1.writeManaged)(".cursor/hooks/wiki-session-start.js", hooks_1.cursorHookScript)]);
+    }
+    if (writeGeminiSurface) {
+        results.push([".gemini/settings.json", (0, hooks_1.upsertGeminiHookConfig)()]);
+        results.push([".gemini/hooks/wiki-session-start.js", (0, workspace_1.writeManaged)(".gemini/hooks/wiki-session-start.js", hooks_1.hookScript)]);
+    }
     // Bootstrap-managed MCP registration (preservation-first, idempotent). Claude
     // Code reads `.mcp.json`, Cursor reads `.cursor/mcp.json`, and Gemini reads
     // `mcpServers` inside `.gemini/settings.json`. Codex only supports user-level MCP
@@ -307,16 +329,24 @@ function runInitCommand() {
     // Registration is scale-gated (2026-06-12 decision): below the measured
     // file-count threshold with no existing .project-wiki index, the rows report the
     // skip reason instead of writing config; an existing index registers regardless.
-    const mcpGate = (0, hooks_1.mcpRegistrationGate)();
-    if (mcpGate.register) {
-        results.push([".mcp.json", (0, hooks_1.upsertClaudeMcpConfig)()]);
-        results.push([".cursor/mcp.json", (0, hooks_1.upsertCursorMcpConfig)()]);
-        results.push([".gemini/settings.json mcpServers", (0, hooks_1.upsertGeminiMcpConfig)()]);
-    }
-    else {
-        results.push([".mcp.json", mcpGate.reason]);
-        results.push([".cursor/mcp.json", mcpGate.reason]);
-        results.push([".gemini/settings.json mcpServers", mcpGate.reason]);
+    if (writeClaudeSurface || writeCursorSurface || writeGeminiSurface) {
+        const mcpGate = (0, hooks_1.mcpRegistrationGate)();
+        if (mcpGate.register) {
+            if (writeClaudeSurface)
+                results.push([".mcp.json", (0, hooks_1.upsertClaudeMcpConfig)()]);
+            if (writeCursorSurface)
+                results.push([".cursor/mcp.json", (0, hooks_1.upsertCursorMcpConfig)()]);
+            if (writeGeminiSurface)
+                results.push([".gemini/settings.json mcpServers", (0, hooks_1.upsertGeminiMcpConfig)()]);
+        }
+        else {
+            if (writeClaudeSurface)
+                results.push([".mcp.json", mcpGate.reason]);
+            if (writeCursorSurface)
+                results.push([".cursor/mcp.json", mcpGate.reason]);
+            if (writeGeminiSurface)
+                results.push([".gemini/settings.json mcpServers", mcpGate.reason]);
+        }
     }
     // Routers accumulate user-maintained project state after bootstrap, so they are
     // starter files: templates are written only when the file is absent, never rebuilt.
@@ -366,3 +396,12 @@ function runInitCommand() {
         console.log(`${String(status).padEnd(7)} ${relativePath}`);
     }
 }
+function runRefreshIndexOnlyMode() {
+    const results = [];
+    results.push(["wiki/index.md", (0, workspace_1.writeStarter)("wiki/index.md", templates_1.index)]);
+    results.push(["wiki/index.md auto-discovered pages", (0, workspace_1.upsertMarkedSection)("wiki/index.md", "<!-- PROJECT-WIKI-AUTO-INDEX:START -->", "<!-- PROJECT-WIKI-AUTO-INDEX:END -->", (0, modes_1.buildRefreshIndexBlock)())]);
+    console.log("Project Librarian refresh-index complete.");
+    for (const [relativePath, status] of results) {
+        console.log(`${String(status).padEnd(7)} ${relativePath}`);
+    }
+}

package/dist/install-skill.js

@@ -37,9 +37,9 @@ exports.runInstallSkillMode = runInstallSkillMode;
 const fs = __importStar(require("node:fs"));
 const os = __importStar(require("node:os"));
 const path = __importStar(require("node:path"));
+const agent_surfaces_1 = require("./agent-surfaces");
 const args_1 = require("./args");
 const skillName = "project-librarian";
-const allAgentTargets = ["codex", "claude", "cursor", "gemini"];
 const packageFiles = [
     "SKILL.md",
     "dist",
@@ -65,10 +65,10 @@ function installAgents() {
     const agents = new Set();
     for (const part of parts) {
         if (part === "all") {
-            for (const agent of allAgentTargets)
+            for (const agent of agent_surfaces_1.allAgentSurfaces)
                 agents.add(agent);
         }
-        else if (allAgentTargets.includes(part)) {
+        else if (agent_surfaces_1.allAgentSurfaces.includes(part)) {
             agents.add(part);
         }
         else {

package/dist/mcp-server.js

@@ -58,6 +58,7 @@ const fs = __importStar(require("node:fs"));
 const path = __importStar(require("node:path"));
 const code_index_1 = require("./code-index");
 const code_index_file_policy_1 = require("./code-index-file-policy");
+const schema_1 = require("./code-index/schema");
 const workspace_1 = require("./workspace");
 // Pinned MCP protocol version. This is the spec revision this server is written
 // against; one constant so the supported version is auditable in a single place.
@@ -366,8 +367,8 @@ function ownershipAnswer(filePath) {
     lines.push(`Workspace owner: ${info.owner} (source: ${info.owner_source})${info.codeowners ? `; codeowners ${info.codeowners}` : ""}.`);
     return lines.join("\n");
 }
-function impactAnswer(database, term) {
-    const impact = (0, code_index_1.codeImpact)(database, term);
+function impactAnswer(database, term, context) {
+    const impact = (0, code_index_1.codeImpact)(database, term, { staleness: context.staleness });
     const matches = (impact.matches ?? {});
     const edges = (impact.edges ?? {});
     const owners = asRows(impact.impacted_owners);
@@ -483,7 +484,7 @@ const TOOLS = [
             properties: { term: { type: "string", description: "Path, symbol, route, module, or concept to build a compact code context pack for." } },
             required: ["term"],
         },
-        run: (database, args) => (0, code_index_1.codeContextPack)(database, requireStringArg(args, "term")),
+        run: (database, args, context) => (0, code_index_1.codeContextPack)(database, requireStringArg(args, "term"), { staleness: context.staleness }),
     },
     {
         name: "code_impact",
@@ -493,7 +494,7 @@ const TOOLS = [
             properties: { term: { type: "string", description: "File path, symbol name, route, or module to trace." } },
             required: ["term"],
         },
-        run: (database, args) => impactAnswer(database, requireStringArg(args, "term")),
+        run: (database, args, context) => impactAnswer(database, requireStringArg(args, "term"), context),
     },
     {
         name: "code_ownership",
@@ -537,6 +538,59 @@ const TOOLS = [
     },
 ];
 const TOOLS_BY_NAME = new Map(TOOLS.map((tool) => [tool.name, tool]));
+let cachedStaleness = null;
+function statFingerprint(relativePath) {
+    const absolutePath = path.join(workspace_1.root, relativePath);
+    try {
+        const stat = fs.statSync(absolutePath);
+        if (!stat.isFile())
+            return `${relativePath}:not-file`;
+        return [
+            relativePath,
+            stat.size,
+            stat.mtimeMs,
+            stat.ctimeMs,
+        ].join(":");
+    }
+    catch {
+        return `${relativePath}:missing`;
+    }
+}
+function codeEvidenceFreshnessCacheKey(opened) {
+    try {
+        const database = opened.database;
+        const scopes = (0, schema_1.indexedScopes)(database);
+        const parserMode = (0, schema_1.indexedParserMode)(database);
+        const schemaVersion = (0, schema_1.readMetaValue)(database, "schema_version");
+        const updatedAt = (0, schema_1.readMetaValue)(database, "updated_at");
+        const currentFiles = (0, code_index_file_policy_1.discoverCodeFiles)(scopes.length > 0 ? scopes : ["."]);
+        const currentFingerprint = currentFiles.map(statFingerprint).join("\n");
+        const indexedFingerprint = database.prepare("SELECT path, hash, mtime_ms, size FROM files ORDER BY path")
+            .all()
+            .map((row) => `${String(row.path)}:${String(row.hash)}:${Number(row.mtime_ms)}:${Number(row.size)}`)
+            .join("\n");
+        return [
+            opened.relativePath,
+            schemaVersion,
+            updatedAt,
+            parserMode,
+            scopes.join("\0"),
+            currentFingerprint,
+            indexedFingerprint,
+        ].join("\n---\n");
+    }
+    catch {
+        return null;
+    }
+}
+function codeIndexStalenessForToolCall(opened) {
+    const cacheKey = codeEvidenceFreshnessCacheKey(opened);
+    if (cacheKey && cachedStaleness?.cacheKey === cacheKey)
+        return cachedStaleness.staleness;
+    const staleness = (0, code_index_1.codeIndexStaleness)(opened.database);
+    cachedStaleness = cacheKey ? { cacheKey, staleness } : null;
+    return staleness;
+}
 // ---------------------------------------------------------------------------
 // tools/call dispatch
 // ---------------------------------------------------------------------------
@@ -559,10 +613,10 @@ function callTool(name, rawArgs) {
         throw error;
     }
     try {
-        const staleness = (0, code_index_1.codeIndexStaleness)(opened.database);
+        const staleness = codeIndexStalenessForToolCall(opened);
         const body = name === "code_status"
             ? statusAnswer(opened.database, opened.relativePath, staleness)
-            : TOOLS_BY_NAME.get(name).run(opened.database, args);
+            : TOOLS_BY_NAME.get(name).run(opened.database, args, { staleness });
         return toolResultContent(finalizeAnswer(body, staleness));
     }
     catch (error) {

package/dist/migration.js

@@ -37,6 +37,7 @@ exports.generatedMigrationInboxFiles = void 0;
 exports.classifyMarkdown = classifyMarkdown;
 exports.formOnlyMigrationDocumentReason = formOnlyMigrationDocumentReason;
 exports.extractMigrationUnits = extractMigrationUnits;
+exports.loadMigrationUnitContext = loadMigrationUnitContext;
 exports.collectMigrationCoverageDiagnostics = collectMigrationCoverageDiagnostics;
 exports.collectMigrationUnitMapDiagnostics = collectMigrationUnitMapDiagnostics;
 exports.collectMigrationSplitPlanDiagnostics = collectMigrationSplitPlanDiagnostics;
@@ -321,6 +322,9 @@ function expectedMigrationUnits() {
         .flatMap((legacyRoot) => (0, wiki_files_1.walkMarkdownFiles)((0, workspace_1.abs)(legacyRoot), [], (0, workspace_1.abs)(legacyRoot)))
         .flatMap((file) => extractMigrationUnits(file.basePath, (0, workspace_1.read)(file.path)));
 }
+function loadMigrationUnitContext() {
+    return { units: expectedMigrationUnits() };
+}
 function isMigrationConfidence(value) {
     return ["high", "medium", "low"].includes(value);
 }
@@ -353,8 +357,8 @@ function isReviewedCoverageRetarget(cells) {
     return note.includes("reviewed low-confidence content; retargeted")
         || reason.includes("reviewed source context; taxonomy target assigned");
 }
-function collectMigrationCoverageDiagnostics() {
-    const units = expectedMigrationUnits();
+function collectMigrationCoverageDiagnostics(context = loadMigrationUnitContext()) {
+    const units = context.units;
     if (units.length === 0)
         return [];
     if (!(0, workspace_1.exists)("wiki/migration/coverage.md")) {
@@ -417,8 +421,8 @@ function collectMigrationCoverageDiagnostics() {
     }
     return diagnostics.sort((a, b) => a.file.localeCompare(b.file) || a.code.localeCompare(b.code) || a.message.localeCompare(b.message));
 }
-function collectMigrationUnitMapDiagnostics() {
-    const units = expectedMigrationUnits();
+function collectMigrationUnitMapDiagnostics(context = loadMigrationUnitContext()) {
+    const units = context.units;
     if (units.length === 0)
         return [];
     const file = "wiki/migration/unit-map.md";
@@ -485,8 +489,8 @@ function collectMigrationUnitMapDiagnostics() {
     }
     return diagnostics.sort((a, b) => a.file.localeCompare(b.file) || a.code.localeCompare(b.code) || a.message.localeCompare(b.message));
 }
-function collectMigrationSplitPlanDiagnostics() {
-    const units = expectedMigrationUnits();
+function collectMigrationSplitPlanDiagnostics(context = loadMigrationUnitContext()) {
+    const units = context.units;
     if (units.length === 0)
         return [];
     const file = "wiki/migration/split-plan.md";