@kage-core/kage-graph-mcp 1.1.35 → 1.1.37

package/dist/daemon.js

@@ -1,5 +1,9 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
+exports.viewerStaticHeaders = viewerStaticHeaders;
+exports.viewerRedirectLocation = viewerRedirectLocation;
+exports.viewerBenchmarkReport = viewerBenchmarkReport;
+exports.daemonContextReport = daemonContextReport;
 exports.readDaemonStatus = readDaemonStatus;
 exports.daemonDoctor = daemonDoctor;
 exports.stopDaemon = stopDaemon;
@@ -34,6 +38,107 @@ function contentType(filePath) {
         return "text/markdown; charset=utf-8";
     return "application/octet-stream";
 }
+function viewerStaticHeaders(filePath) {
+    return {
+        "content-type": contentType(filePath),
+        "x-content-type-options": "nosniff",
+        "referrer-policy": "no-referrer",
+        "cross-origin-opener-policy": "same-origin",
+        "content-security-policy": [
+            "default-src 'self'",
+            "script-src 'self'",
+            "script-src-attr 'none'",
+            "style-src 'self' 'unsafe-inline'",
+            "img-src 'self' data:",
+            "connect-src 'self'",
+            "font-src 'none'",
+            "object-src 'none'",
+            "base-uri 'none'",
+            "frame-ancestors 'none'",
+        ].join("; "),
+    };
+}
+function viewerRedirectLocation(pathname, search, fallbackSearch) {
+    if (pathname !== "/" && pathname !== "/viewer" && pathname !== "/viewer/")
+        return null;
+    return `/viewer/index.html${search || fallbackSearch}`;
+}
+function viewerBenchmarkReport(projectDir) {
+    const gates = (0, kernel_js_1.benchmarkProject)(projectDir);
+    const memoryQuality = (0, kernel_js_1.benchmarkCodingMemoryQuality)();
+    const memoryScale = (0, kernel_js_1.benchmarkMemoryScale)({ sizes: [240], topK: 10 });
+    const requiredGates = gates.gates.filter((gate) => gate.required);
+    const requiredGatePasses = requiredGates.filter((gate) => gate.pass).length;
+    const requiredGatePercent = requiredGates.length ? Math.round((requiredGatePasses / requiredGates.length) * 100) : (gates.ok ? 100 : 0);
+    const recallAt10 = Number(memoryQuality.summary.recall_at_10_percent ?? memoryQuality.summary.recall_at_k_percent ?? 0);
+    const sourceDiversity = memoryQuality.source_diversity;
+    const scaleHitRate = Number(memoryScale.summary.largest_hit_rate_percent ?? 0);
+    const scaleContextCut = Number(memoryScale.summary.largest_context_reduction_percent ?? 0);
+    return {
+        ...gates,
+        summary: memoryQuality.summary,
+        memory_quality: {
+            dataset: memoryQuality.dataset,
+            summary: memoryQuality.summary,
+            source_diversity: memoryQuality.source_diversity,
+            by_category: memoryQuality.by_category,
+            baselines: memoryQuality.baselines,
+            caveats: memoryQuality.caveats,
+        },
+        memory_scale: {
+            sizes: memoryScale.sizes,
+            summary: memoryScale.summary,
+            results: memoryScale.results,
+            caveats: memoryScale.caveats,
+        },
+        proof_ledger: [
+            {
+                id: "memory-quality",
+                label: "Coding-memory retrieval",
+                metric: `${recallAt10.toFixed(2)}% R@10, ${memoryQuality.summary.ndcg_at_10.toFixed(4)} NDCG@10`,
+                target: ">=95% R@10 and >=0.85 NDCG@10",
+                actual: recallAt10,
+                unit: "percent",
+                pass: recallAt10 >= 95 && memoryQuality.summary.ndcg_at_10 >= 0.85,
+                command: "kage benchmark --memory-quality --json",
+                next_action: "Use this as the fast coding-memory retrieval proof before publishing benchmark claims.",
+            },
+            {
+                id: "source-diversity",
+                label: "Source diversity",
+                metric: `${sourceDiversity.unique_sources} sources in top ${sourceDiversity.top_k}; max ${sourceDiversity.max_results_from_one_source} from one source`,
+                target: ">=2 sources and <=3 results from one observed session",
+                actual: sourceDiversity.unique_sources,
+                unit: "score",
+                pass: sourceDiversity.pass,
+                command: "kage benchmark --memory-quality --json",
+                next_action: sourceDiversity.pass ? "Use this to prove noisy sessions cannot crowd out independent repo memory." : "Inspect recall source diversity before publishing memory-quality claims.",
+            },
+            {
+                id: "memory-scale",
+                label: "Memory scale sanity",
+                metric: `${memoryScale.summary.largest_packets} packets, ${scaleHitRate.toFixed(2)}% hit rate, ${scaleContextCut.toFixed(2)}% context cut`,
+                target: ">=95% hit rate and >=80% context reduction",
+                actual: scaleHitRate,
+                unit: "percent",
+                pass: scaleHitRate >= 95 && scaleContextCut >= 80,
+                command: "kage benchmark --scale --sizes 240 --json",
+                next_action: "Run kage benchmark --scale --sizes 240,1000,5000 --json for a larger release proof.",
+            },
+            {
+                id: "local-gates",
+                label: "Repo trust gates",
+                metric: `${requiredGatePasses}/${requiredGates.length || gates.gates.length} required gates passing`,
+                target: "100% required gates passing",
+                actual: requiredGatePercent,
+                unit: "percent",
+                pass: gates.ok,
+                command: "kage benchmark --project . --json",
+                next_action: gates.ok ? "Keep this green before handoff, README updates, or release." : "Fix failing repo benchmark gates before sharing this memory state.",
+            },
+        ],
+    };
+}
 function statusPath(projectDir) {
     return (0, node_path_1.join)(daemonDir(projectDir), "status.json");
 }
@@ -64,6 +169,68 @@ function readBody(req) {
         req.on("error", reject);
     });
 }
+function stringArray(value) {
+    if (Array.isArray(value))
+        return value.map(String).filter(Boolean);
+    if (typeof value === "string")
+        return value.split(",").map((item) => item.trim()).filter(Boolean);
+    return [];
+}
+function filePathHints(query) {
+    const matches = query.match(/[A-Za-z0-9_./@-]+\.(?:ts|tsx|js|jsx|mjs|cjs|py|go|rs|java|kt|kts|rb|php|cs|c|h|cc|cpp|hpp|swift|json|md)\b/g) ?? [];
+    return [...new Set(matches.map((match) => match.replace(/^\.\//, "")).filter((match) => !/^https?:\/\//.test(match)))];
+}
+function wantsDependencyPath(query) {
+    return /\b(connect|connected|dependency|depend|depends|path|impact|flow|trace)\b/i.test(query);
+}
+function riskContextBlock(result) {
+    const targets = Object.values(result.targets);
+    if (!targets.length)
+        return "";
+    const lines = targets.slice(0, 5).map((item) => {
+        const coChange = item.git.co_change_partners.length
+            ? ` Co-change: ${item.git.co_change_partners.slice(0, 3).map((partner) => `${partner.file_path} (${partner.count})`).join(", ")}.`
+            : "";
+        return `- ${item.risk_summary}${coChange}`;
+    });
+    return `\n## Risk Signals\n${lines.join("\n")}`;
+}
+function daemonContextReport(projectDir, body) {
+    const query = String(body.query ?? "");
+    const limit = Number(body.limit ?? 5);
+    const validation = (0, kernel_js_1.validateProject)(projectDir);
+    const recallResult = (0, kernel_js_1.recall)(projectDir, query, limit, Boolean(body.explain));
+    const graphResult = (0, kernel_js_1.queryGraph)(projectDir, query, 5);
+    const explicitTargets = [...stringArray(body.targets), ...filePathHints(query)];
+    const changedFiles = stringArray(body.changed_files);
+    const riskResult = explicitTargets.length || changedFiles.length ? (0, kernel_js_1.kageRisk)(projectDir, explicitTargets, changedFiles) : null;
+    const pathHints = filePathHints(query);
+    const dependencyResult = wantsDependencyPath(query) && pathHints.length >= 2
+        ? (0, kernel_js_1.kageDependencyPath)(projectDir, pathHints[0], pathHints[1])
+        : null;
+    const reconciliation = (0, kernel_js_1.kageMemoryReconciliation)(projectDir, {
+        sessionId: typeof body.session_id === "string" ? body.session_id : undefined,
+        limit: 5,
+    });
+    const validationText = validation.ok ? "Memory healthy." : `Warnings: ${validation.warnings.join("; ")}`;
+    const contextBlock = [
+        recallResult.context_block,
+        graphResult.context_block ? `\n## Graph Facts\n${graphResult.context_block}` : "",
+        riskResult ? riskContextBlock(riskResult) : "",
+        dependencyResult ? `\n## Dependency Path\n${dependencyResult.summary}${dependencyResult.path.length ? `\nPath: ${dependencyResult.path.join(" -> ")}` : ""}` : "",
+        reconciliation.unresolved_count ? `\n## Memory Reconciliation\n${reconciliation.agent_instruction}` : "",
+        `\n_${validationText}_`,
+    ].filter(Boolean).join("");
+    return {
+        context_block: contextBlock,
+        recall: recallResult,
+        graph: graphResult,
+        risk: riskResult,
+        dependency_path: dependencyResult,
+        validation,
+        memory_reconciliation: reconciliation,
+    };
+}
 function readDaemonStatus(projectDir) {
     const path = statusPath(projectDir);
     if (!(0, node_fs_1.existsSync)(path))
@@ -93,9 +260,20 @@ function daemonDoctor(projectDir) {
         status,
         endpoints: [
             `GET http://${DEFAULT_HOST}:${restPort}/health`,
+            `POST http://${DEFAULT_HOST}:${restPort}/kage/context`,
             `POST http://${DEFAULT_HOST}:${restPort}/kage/recall`,
+            `POST http://${DEFAULT_HOST}:${restPort}/kage/capture`,
+            `POST http://${DEFAULT_HOST}:${restPort}/kage/learn`,
+            `POST http://${DEFAULT_HOST}:${restPort}/kage/feedback`,
             `POST http://${DEFAULT_HOST}:${restPort}/kage/observe`,
             `POST http://${DEFAULT_HOST}:${restPort}/kage/distill`,
+            `GET http://${DEFAULT_HOST}:${restPort}/kage/replay`,
+            `GET http://${DEFAULT_HOST}:${restPort}/kage/setup-doctor`,
+            `GET http://${DEFAULT_HOST}:${restPort}/kage/profile`,
+            `GET http://${DEFAULT_HOST}:${restPort}/kage/capabilities`,
+            `GET http://${DEFAULT_HOST}:${restPort}/kage/context-slots`,
+            `POST http://${DEFAULT_HOST}:${restPort}/kage/context-slots`,
+            `DELETE http://${DEFAULT_HOST}:${restPort}/kage/context-slots/:label`,
             `GET http://${DEFAULT_HOST}:${restPort}/kage/metrics`,
             `GET http://${DEFAULT_HOST}:${restPort}/kage/quality`,
             `GET http://${DEFAULT_HOST}:${restPort}/kage/inbox`,
@@ -187,12 +365,60 @@ async function startDaemon(projectDir, options = {}) {
                 json(res, 200, (0, kernel_js_1.qualityReport)(projectDir));
                 return;
             }
+            if (req.method === "GET" && url.pathname === "/kage/profile") {
+                json(res, 200, (0, kernel_js_1.kageProjectProfile)(projectDir));
+                return;
+            }
+            if (req.method === "GET" && url.pathname === "/kage/capabilities") {
+                json(res, 200, (0, kernel_js_1.kageCapabilityAudit)(projectDir));
+                return;
+            }
+            if (req.method === "GET" && url.pathname === "/kage/replay") {
+                json(res, 200, (0, kernel_js_1.kageSessionReplay)(projectDir, {
+                    sessionId: url.searchParams.get("session") ?? undefined,
+                    limit: Number(url.searchParams.get("limit") ?? 200),
+                }));
+                return;
+            }
+            if (req.method === "GET" && url.pathname === "/kage/context-slots") {
+                json(res, 200, (0, kernel_js_1.kageContextSlots)(projectDir));
+                return;
+            }
+            if (req.method === "GET" && url.pathname === "/kage/lifecycle") {
+                json(res, 200, (0, kernel_js_1.kageMemoryLifecycle)(projectDir));
+                return;
+            }
+            if (req.method === "GET" && url.pathname === "/kage/memory-audit") {
+                json(res, 200, (0, kernel_js_1.kageMemoryAudit)(projectDir, Number(url.searchParams.get("limit") ?? 100)));
+                return;
+            }
+            if (req.method === "GET" && url.pathname === "/kage/handoff") {
+                json(res, 200, (0, kernel_js_1.kageMemoryHandoff)(projectDir));
+                return;
+            }
+            if (req.method === "GET" && url.pathname === "/kage/timeline") {
+                json(res, 200, (0, kernel_js_1.kageMemoryTimeline)(projectDir, Number(url.searchParams.get("days") ?? 14)));
+                return;
+            }
+            if (req.method === "GET" && url.pathname === "/kage/lineage") {
+                json(res, 200, (0, kernel_js_1.kageMemoryLineage)(projectDir));
+                return;
+            }
             if (req.method === "GET" && url.pathname === "/kage/inbox") {
                 json(res, 200, (0, kernel_js_1.memoryInbox)(projectDir));
                 return;
             }
             if (req.method === "GET" && url.pathname === "/kage/benchmark") {
-                json(res, 200, (0, kernel_js_1.benchmarkProject)(projectDir));
+                json(res, 200, url.searchParams.get("mode") === "memory_quality" ? (0, kernel_js_1.benchmarkCodingMemoryQuality)() : (0, kernel_js_1.benchmarkProject)(projectDir));
+                return;
+            }
+            if (req.method === "GET" && url.pathname === "/kage/setup-doctor") {
+                json(res, 200, (0, kernel_js_1.setupDoctor)(projectDir));
+                return;
+            }
+            if (req.method === "POST" && url.pathname === "/kage/context") {
+                const body = await readBody(req);
+                json(res, 200, daemonContextReport(projectDir, body));
                 return;
             }
             if (req.method === "POST" && url.pathname === "/kage/recall") {
@@ -200,6 +426,63 @@ async function startDaemon(projectDir, options = {}) {
                 json(res, 200, (0, kernel_js_1.recall)(projectDir, String(body.query ?? ""), Number(body.limit ?? 5), Boolean(body.explain)));
                 return;
             }
+            if (req.method === "POST" && url.pathname === "/kage/capture") {
+                const body = await readBody(req);
+                const result = (0, kernel_js_1.capture)({
+                    projectDir,
+                    title: String(body.title ?? ""),
+                    summary: body.summary == null ? undefined : String(body.summary),
+                    body: String(body.body ?? body.learning ?? ""),
+                    type: body.type == null ? undefined : String(body.type),
+                    tags: stringArray(body.tags),
+                    paths: stringArray(body.paths),
+                    stack: stringArray(body.stack),
+                });
+                json(res, result.ok ? 200 : 400, result);
+                return;
+            }
+            if (req.method === "POST" && url.pathname === "/kage/learn") {
+                const body = await readBody(req);
+                const result = (0, kernel_js_1.learn)({
+                    projectDir,
+                    learning: String(body.learning ?? body.body ?? ""),
+                    title: body.title == null ? undefined : String(body.title),
+                    type: body.type == null ? undefined : String(body.type),
+                    evidence: body.evidence == null ? undefined : String(body.evidence),
+                    verifiedBy: body.verified_by == null ? body.verifiedBy == null ? undefined : String(body.verifiedBy) : String(body.verified_by),
+                    tags: stringArray(body.tags),
+                    paths: stringArray(body.paths),
+                    stack: stringArray(body.stack),
+                });
+                json(res, result.ok ? 200 : 400, result);
+                return;
+            }
+            if (req.method === "POST" && url.pathname === "/kage/feedback") {
+                const body = await readBody(req);
+                const result = (0, kernel_js_1.recordFeedback)(projectDir, String(body.packet_id ?? body.packet ?? ""), String(body.kind ?? ""));
+                json(res, result.ok ? 200 : 400, result);
+                return;
+            }
+            if (req.method === "POST" && url.pathname === "/kage/context-slots") {
+                const body = await readBody(req);
+                const result = (0, kernel_js_1.setContextSlot)(projectDir, {
+                    label: String(body.label ?? ""),
+                    content: String(body.content ?? ""),
+                    description: body.description == null ? undefined : String(body.description),
+                    pinned: body.pinned == null ? undefined : Boolean(body.pinned),
+                    size_limit: body.size_limit == null ? undefined : Number(body.size_limit),
+                    paths: stringArray(body.paths),
+                    tags: stringArray(body.tags),
+                });
+                json(res, result.ok ? 200 : 400, result);
+                return;
+            }
+            if (req.method === "DELETE" && url.pathname.startsWith("/kage/context-slots/")) {
+                const label = decodeURIComponent(url.pathname.slice("/kage/context-slots/".length));
+                const result = (0, kernel_js_1.deleteContextSlot)(projectDir, label);
+                json(res, result.ok ? 200 : 404, result);
+                return;
+            }
             if (req.method === "POST" && url.pathname === "/kage/observe") {
                 const body = await readBody(req);
                 json(res, 200, (0, kernel_js_1.observe)(projectDir, body));
@@ -244,11 +527,23 @@ async function startViewer(projectDir, options = {}) {
     const qualityPath = (0, node_path_1.join)(reportsDir, "quality.json");
     const benchmarkPath = (0, node_path_1.join)(reportsDir, "benchmark.json");
     const contributorsPath = (0, node_path_1.join)(reportsDir, "contributors.json");
+    const profilePath = (0, node_path_1.join)(reportsDir, "profile.json");
+    const capabilitiesPath = (0, node_path_1.join)(reportsDir, "capabilities.json");
+    const slotsPath = (0, node_path_1.join)(reportsDir, "context-slots.json");
     const decisionsPath = (0, node_path_1.join)(reportsDir, "decisions.json");
     const riskPath = (0, node_path_1.join)(reportsDir, "risk.json");
     const moduleHealthPath = (0, node_path_1.join)(reportsDir, "module-health.json");
     const graphInsightsPath = (0, node_path_1.join)(reportsDir, "graph-insights.json");
     const workspacePath = (0, node_path_1.join)(reportsDir, "workspace.json");
+    const sessionsPath = (0, node_path_1.join)(reportsDir, "sessions.json");
+    const replayPath = (0, node_path_1.join)(reportsDir, "replay.json");
+    const memoryAccessPath = (0, node_path_1.join)(reportsDir, "memory-access.json");
+    const memoryAuditPath = (0, node_path_1.join)(reportsDir, "memory-audit.json");
+    const handoffPath = (0, node_path_1.join)(reportsDir, "handoff.json");
+    const lifecyclePath = (0, node_path_1.join)(reportsDir, "lifecycle.json");
+    const timelinePath = (0, node_path_1.join)(reportsDir, "timeline.json");
+    const lineagePath = (0, node_path_1.join)(reportsDir, "lineage.json");
+    const setupPath = (0, node_path_1.join)(reportsDir, "setup.json");
     // Pre-generate lightweight JSON reports so the viewer can load them directly.
     try {
         (0, node_fs_1.mkdirSync)(reportsDir, { recursive: true });
@@ -257,24 +552,36 @@ async function startViewer(projectDir, options = {}) {
         const inbox = (0, kernel_js_1.memoryInbox)(projectDir);
         (0, node_fs_1.writeFileSync)(inboxPath, JSON.stringify(inbox, null, 2));
         (0, node_fs_1.writeFileSync)(qualityPath, JSON.stringify((0, kernel_js_1.qualityReport)(projectDir), null, 2));
-        (0, node_fs_1.writeFileSync)(benchmarkPath, JSON.stringify((0, kernel_js_1.benchmarkProject)(projectDir), null, 2));
+        (0, node_fs_1.writeFileSync)(benchmarkPath, JSON.stringify(viewerBenchmarkReport(projectDir), null, 2));
         (0, node_fs_1.writeFileSync)(contributorsPath, JSON.stringify((0, kernel_js_1.kageContributors)(projectDir), null, 2));
+        (0, node_fs_1.writeFileSync)(profilePath, JSON.stringify((0, kernel_js_1.kageProjectProfile)(projectDir), null, 2));
+        (0, node_fs_1.writeFileSync)(capabilitiesPath, JSON.stringify((0, kernel_js_1.kageCapabilityAudit)(projectDir), null, 2));
+        (0, node_fs_1.writeFileSync)(slotsPath, JSON.stringify((0, kernel_js_1.kageContextSlots)(projectDir), null, 2));
         (0, node_fs_1.writeFileSync)(decisionsPath, JSON.stringify((0, kernel_js_1.kageDecisionIntelligence)(projectDir), null, 2));
         (0, node_fs_1.writeFileSync)(riskPath, JSON.stringify((0, kernel_js_1.kageRisk)(projectDir), null, 2));
         (0, node_fs_1.writeFileSync)(moduleHealthPath, JSON.stringify((0, kernel_js_1.kageModuleHealth)(projectDir), null, 2));
         (0, node_fs_1.writeFileSync)(graphInsightsPath, JSON.stringify((0, kernel_js_1.kageGraphInsights)(projectDir), null, 2));
         (0, node_fs_1.writeFileSync)(workspacePath, JSON.stringify((0, kernel_js_1.kageWorkspace)(projectDir), null, 2));
+        (0, node_fs_1.writeFileSync)(sessionsPath, JSON.stringify((0, kernel_js_1.kageSessionCaptureReport)(projectDir), null, 2));
+        (0, node_fs_1.writeFileSync)(replayPath, JSON.stringify((0, kernel_js_1.kageSessionReplay)(projectDir), null, 2));
+        (0, node_fs_1.writeFileSync)(memoryAccessPath, JSON.stringify((0, kernel_js_1.kageMemoryAccess)(projectDir), null, 2));
+        (0, node_fs_1.writeFileSync)(memoryAuditPath, JSON.stringify((0, kernel_js_1.kageMemoryAudit)(projectDir), null, 2));
+        (0, node_fs_1.writeFileSync)(handoffPath, JSON.stringify((0, kernel_js_1.kageMemoryHandoff)(projectDir), null, 2));
+        (0, node_fs_1.writeFileSync)(lifecyclePath, JSON.stringify((0, kernel_js_1.kageMemoryLifecycle)(projectDir), null, 2));
+        (0, node_fs_1.writeFileSync)(timelinePath, JSON.stringify((0, kernel_js_1.kageMemoryTimeline)(projectDir), null, 2));
+        (0, node_fs_1.writeFileSync)(lineagePath, JSON.stringify((0, kernel_js_1.kageMemoryLineage)(projectDir), null, 2));
+        (0, node_fs_1.writeFileSync)(setupPath, JSON.stringify((0, kernel_js_1.setupDoctor)(projectDir), null, 2));
     }
     catch {
         // non-fatal: viewer will show 404 for reports if generation fails
     }
-    const url = `http://${host}:${port}/viewer/index.html?graph=${encodeURIComponent(graphPath)}&code=${encodeURIComponent(codePath)}&metrics=${encodeURIComponent(metricsPath)}&inbox=${encodeURIComponent(inboxPath)}&review=${encodeURIComponent(reviewPath)}&pending=${encodeURIComponent(pendingDir)}&quality=${encodeURIComponent(qualityPath)}&benchmark=${encodeURIComponent(benchmarkPath)}&contributors=${encodeURIComponent(contributorsPath)}&decisions=${encodeURIComponent(decisionsPath)}&risk=${encodeURIComponent(riskPath)}&moduleHealth=${encodeURIComponent(moduleHealthPath)}&graphInsights=${encodeURIComponent(graphInsightsPath)}&workspace=${encodeURIComponent(workspacePath)}&view=code`;
+    const url = `http://${host}:${port}/viewer/index.html?graph=${encodeURIComponent(graphPath)}&code=${encodeURIComponent(codePath)}&metrics=${encodeURIComponent(metricsPath)}&inbox=${encodeURIComponent(inboxPath)}&review=${encodeURIComponent(reviewPath)}&pending=${encodeURIComponent(pendingDir)}&quality=${encodeURIComponent(qualityPath)}&benchmark=${encodeURIComponent(benchmarkPath)}&contributors=${encodeURIComponent(contributorsPath)}&profile=${encodeURIComponent(profilePath)}&capabilities=${encodeURIComponent(capabilitiesPath)}&slots=${encodeURIComponent(slotsPath)}&decisions=${encodeURIComponent(decisionsPath)}&risk=${encodeURIComponent(riskPath)}&moduleHealth=${encodeURIComponent(moduleHealthPath)}&graphInsights=${encodeURIComponent(graphInsightsPath)}&workspace=${encodeURIComponent(workspacePath)}&sessions=${encodeURIComponent(sessionsPath)}&replay=${encodeURIComponent(replayPath)}&memoryAccess=${encodeURIComponent(memoryAccessPath)}&memoryAudit=${encodeURIComponent(memoryAuditPath)}&handoff=${encodeURIComponent(handoffPath)}&lifecycle=${encodeURIComponent(lifecyclePath)}&timeline=${encodeURIComponent(timelinePath)}&lineage=${encodeURIComponent(lineagePath)}&setup=${encodeURIComponent(setupPath)}&view=code`;
     const server = (0, node_http_1.createServer)((req, res) => {
         const requestUrl = new URL(req.url ?? "/", `http://${host}:${port}`);
         let filePath = null;
-        if (requestUrl.pathname === "/" || requestUrl.pathname === "/viewer") {
-            const viewerSearch = requestUrl.search || new URL(url).search;
-            res.writeHead(302, { location: `/viewer/index.html${viewerSearch}` });
+        const redirectLocation = viewerRedirectLocation(requestUrl.pathname, requestUrl.search, new URL(url).search);
+        if (redirectLocation) {
+            res.writeHead(302, { location: redirectLocation });
             res.end();
             return;
         }
@@ -303,7 +610,7 @@ async function startViewer(projectDir, options = {}) {
             json(res, 200, { ok: true, files });
             return;
         }
-        res.writeHead(200, { "content-type": contentType(filePath) });
+        res.writeHead(200, viewerStaticHeaders(filePath));
         res.end((0, node_fs_1.readFileSync)(filePath));
     });
     await new Promise((resolveListen) => server.listen(port, host, resolveListen));