pi-mega-compact 0.11.4 → 0.11.6

package/dist/extensions/mega-runtime/widget.test.js

@@ -1,343 +0,0 @@
-/**
- * widget.test.ts — S31 buildWidgetLines render snapshot matrix.
- *
- * buildWidgetLines is a free function (no pi runtime needed), so we drive it
- * directly with a fake WidgetData across the full matrix:
- *   6 themes × {full,minimal} × {gameMode on/off} × {cachePct<100, >=100}
- * and assert the S31 invariants:
- *  (a) transparent theme → no '\x1b[48;' bg fill on any line
- *  (b) non-transparent themes → at least one line carries '\x1b[48;'
- *  (c) minimal mode → exactly one content line (between the two panel bars)
- *  (d) gameMode off → no 'LVL' + no 'MEGA CACHE'
- *  (e) megaCacheFlare + gameMode on + cachePct>=100 → 'MEGA CACHE' text present
- *  (f) every line visibleWidth <= the terminal width passed in
- *
- * Uses MEGACOMPACT_STATE_DIR + mkdtemp (G7). No pi runtime.
- */
-import { describe, it, before, after } from "node:test";
-import assert from "node:assert/strict";
-import { tmpdir } from "node:os";
-import { join } from "node:path";
-import { mkdtempSync, rmSync } from "node:fs";
-import { visibleWidth } from "@earendil-works/pi-tui";
-import { buildWidgetLines } from "./widget.js";
-import { THEME_IDS, DEFAULT_THEME, getTheme } from "../../src/config/themes.js";
-/** A minimal but complete WidgetData (all required fields populated). The S31
- *  game-mode fields are overridden per-case. */
-function baseWd(overrides = {}) {
-    return {
-        version: "0.0.0-test",
-        tierLabel: "low",
-        triggerLabel: "idle",
-        pctStr: "42%",
-        tokStr: "10k",
-        maxStr: "200k",
-        ctxPct: 0.42,
-        chk: 3,
-        agentStr: "",
-        turnStr: "",
-        dedupStr: "12%",
-        sessIn: 1000,
-        sessKept: 800,
-        sTxt: "20",
-        repoIn: 5000,
-        repoKept: 4000,
-        rTxt: "20",
-        repoChk: 9,
-        repoSess: 2,
-        modelStr: "test-model",
-        sinceCompact: null,
-        embedderName: "Trigram",
-        compStr: "1.2x",
-        driftStatus: "ok",
-        agentsActive: false,
-        fresh: false,
-        ticker: [],
-        lastWhy: undefined,
-        tierTrace: undefined,
-        pulsing: false,
-        ...overrides,
-    };
-}
-const WIDTH = 120;
-function contentLines(lines) {
-    // Strip the top + bottom panel bars; what's left is the content body.
-    return lines.slice(1, lines.length - 1);
-}
-describe("buildWidgetLines (S31)", () => {
-    let dir;
-    before(() => {
-        dir = mkdtempSync(join(tmpdir(), "mc-widget-"));
-        process.env.MEGACOMPACT_STATE_DIR = dir;
-    });
-    after(() => {
-        delete process.env.MEGACOMPACT_STATE_DIR;
-        rmSync(dir, { recursive: true, force: true });
-    });
-    it("null wd → warm-up panel (3 lines, all width-safe)", () => {
-        const lines = buildWidgetLines(null, WIDTH, 0);
-        assert.equal(lines.length, 3);
-        for (const l of lines)
-            assert.ok(visibleWidth(l) <= WIDTH, "width safe");
-    });
-    for (const theme of THEME_IDS) {
-        for (const tuiMode of ["full", "minimal"]) {
-            for (const gameMode of [false, true]) {
-                for (const cachePct of [42, 150]) {
-                    const flare = cachePct >= 100;
-                    const label = `theme=${theme} tui=${tuiMode} game=${gameMode} cache=${cachePct}`;
-                    it(label, () => {
-                        const wd = baseWd({
-                            theme,
-                            tuiMode,
-                            gameMode,
-                            level: 1,
-                            cachePct,
-                            megaCacheFlare: flare,
-                            // C.3: flare gate reads megaCacheFlarePct, not cachePct
-                            ...(flare ? { megaCacheFlarePct: cachePct } : {}),
-                        });
-                        const lines = buildWidgetLines(wd, WIDTH, 0);
-                        const body = contentLines(lines);
-                        const joined = lines.join("\n");
-                        // (f) every line width <= terminal (truncateToWidth respected)
-                        for (const l of lines) {
-                            assert.ok(visibleWidth(l) <= WIDTH, `width safe: ${visibleWidth(l)} > ${WIDTH}`);
-                        }
-                        // (a) transparent theme → no '\x1b[48;' bg fill anywhere
-                        if (theme === DEFAULT_THEME) {
-                            assert.ok(!joined.includes("\x1b[48;"), `transparent has no 48; bg fill: ${label}`);
-                        }
-                        else {
-                            // (b) non-transparent themes → their bg SGR appears on some line.
-                            // (Themes may use 3-bit bg like \x1b[40m, not just 48;5; form.)
-                            const bgParams = getTheme(theme).ansi.bg;
-                            const bgEsc = `\x1b[${bgParams}m`;
-                            assert.ok(joined.includes(bgEsc), `themed has bg fill ${bgEsc}: ${label}`);
-                        }
-                        // (c) minimal mode → exactly one content line
-                        if (tuiMode === "minimal") {
-                            assert.equal(body.length, 1, `minimal one line: ${label}`);
-                        }
-                        // (d) gameMode off → no LVL + no MEGA CACHE
-                        if (!gameMode) {
-                            assert.ok(!joined.includes("LVL"), `no LVL when off: ${label}`);
-                            assert.ok(!joined.includes("MEGA CACHE"), `no MEGA CACHE when off: ${label}`);
-                        }
-                        else {
-                            // gameMode on → LVL shown (in full mode header, or minimal line)
-                            assert.ok(joined.includes("LVL"), `LVL shown when on: ${label}`);
-                        }
-                        // (e) MEGA CACHE flare text appears only when flare + gameMode on
-                        const expectMega = flare && gameMode;
-                        const hasMega = joined.includes("MEGA CACHE");
-                        assert.equal(hasMega, expectMega, `MEGA CACHE flare: ${label} (got ${hasMega}, want ${expectMega})`);
-                    });
-                }
-            }
-        }
-    }
-});
-describe("buildWidgetLines ambient border effect (v0.8.3)", () => {
-    const effBase = (overrides = {}) => baseWd({
-        theme: DEFAULT_THEME,
-        tuiMode: "full",
-        gameMode: true,
-        level: 1,
-        cachePct: 42,
-        ...overrides,
-    });
-    const isBorder = (l) => l.includes("─");
-    it("activeEffect (pulse, mid-window) -> border lines carry a 256-color fg SGR", () => {
-        const ae = {
-            type: "pulse",
-            role: "accent",
-            startedAt: Date.now() - 250,
-            durationMs: 2000,
-        };
-        const lines = buildWidgetLines(effBase({ activeEffect: ae }), WIDTH, 0);
-        const borders = lines.filter(isBorder);
-        assert.ok(borders.length >= 2, "has top + bottom borders");
-        for (const b of borders) {
-            assert.ok(b.includes("\x1b[38;5;"), `border carries 256-color fg: ${JSON.stringify(b)}`);
-        }
-    });
-    it("activeEffect null -> plain borders, no 38;5 fg SGR on border lines", () => {
-        const lines = buildWidgetLines(effBase({ activeEffect: null }), WIDTH, 0);
-        const borders = lines.filter(isBorder);
-        for (const b of borders) {
-            assert.ok(!b.includes("\x1b[38;5;"), `no effect SGR on plain border: ${JSON.stringify(b)}`);
-        }
-    });
-    it("expired activeEffect -> plain borders (per-frame expiry enforced)", () => {
-        const ae = {
-            type: "pulse",
-            role: "accent",
-            startedAt: Date.now() - 5000,
-            durationMs: 1000,
-        };
-        const lines = buildWidgetLines(effBase({ activeEffect: ae }), WIDTH, 0);
-        const borders = lines.filter(isBorder);
-        for (const b of borders) {
-            assert.ok(!b.includes("\x1b[38;5;"), `expired effect -> plain border: ${JSON.stringify(b)}`);
-        }
-    });
-    it("activeEffect border lines are width-safe (pulse, minimal + full)", () => {
-        for (const tuiMode of ["minimal", "full"]) {
-            const ae = {
-                type: "pulse",
-                role: "mega",
-                startedAt: Date.now() - 100,
-                durationMs: 2000,
-            };
-            const lines = buildWidgetLines(effBase({ activeEffect: ae, tuiMode }), 60, 0);
-            for (const l of lines)
-                assert.ok(visibleWidth(l) <= 60, `width safe (${tuiMode}): ${visibleWidth(l)}`);
-        }
-    });
-    it("flash effect mid-window border carries the full base index SGR", () => {
-        // Force an 'on' phase of the 120ms alternate by starting just now.
-        const ae = {
-            type: "flash",
-            role: "red",
-            startedAt: Date.now(),
-            durationMs: 1200,
-        };
-        const lines = buildWidgetLines(effBase({ activeEffect: ae }), WIDTH, 0);
-        const borders = lines.filter(isBorder);
-        assert.ok(borders.some((b) => b.includes("\x1b[38;5;203m")), `flash-on phase uses red base 203`);
-    });
-});
-describe("buildWidgetLines C.3 flare gate regression", () => {
-    const flareBase = (overrides = {}) => baseWd({
-        theme: DEFAULT_THEME,
-        tuiMode: "full",
-        gameMode: true,
-        level: 1,
-        cachePct: 42,
-        ...overrides,
-    });
-    it("flare fires when megaCacheFlarePct >= 100 even when cachePct < 100", () => {
-        // Regression: the old code read (wd.cachePct ?? 0) >= 100, so when
-        // cachePct was 56 (provider hit rate) the flare silently broke.
-        const lines = buildWidgetLines(flareBase({
-            cachePct: 56,
-            megaCacheFlare: true,
-            megaCacheFlarePct: 287,
-        }), WIDTH, 0);
-        assert.ok(lines.join("\n").includes("MEGA CACHE"), "flare fires from megaCacheFlarePct");
-    });
-    it("flare suppressed when megaCacheFlarePct < 100 even when cachePct >= 100", () => {
-        // Symmetric regression: high cachePct should NOT alone trigger flare.
-        const lines = buildWidgetLines(flareBase({
-            cachePct: 150,
-            megaCacheFlare: true,
-            megaCacheFlarePct: 42,
-        }), WIDTH, 0);
-        assert.ok(!lines.join("\n").includes("MEGA CACHE"), "no flare when megaCacheFlarePct below threshold");
-    });
-    it("flare suppressed when megaCacheFlarePct is missing (undefined)", () => {
-        const lines = buildWidgetLines(flareBase({
-            cachePct: 150,
-            megaCacheFlare: true,
-        }), WIDTH, 0);
-        assert.ok(!lines.join("\n").includes("MEGA CACHE"), "no flare without megaCacheFlarePct");
-    });
-});
-describe("buildWidgetLines footer stability (P1 — no 250ms rotation)", () => {
-    // P1: the L5 ticker branch used to re-pick the head text every 250ms via
-    // `step = floor(Date.now()/250)`, flipping the footer line on a 250·N ms
-    // cycle. After the fix it pins to the most-recent entry, so two renders of
-    // the SAME WidgetData at t and t+500ms (well past one rotation slot) must
-    // produce byte-identical footer lines. Proves the metronome is gone.
-    const stabBase = (overrides = {}) => baseWd({
-        theme: DEFAULT_THEME,
-        tuiMode: "full",
-        gameMode: false,
-        level: 1,
-        cachePct: 42,
-        ...overrides,
-    });
-    it("ticker footer is byte-identical across simulated frames 500ms apart", () => {
-        const wd = stabBase({
-            ticker: [
-                { text: "first", at: 1000 },
-                { text: "second", at: 2000 },
-                { text: "third", at: 3000 },
-            ],
-            lastWhy: "because",
-        });
-        const realNow = Date.now;
-        try {
-            Date.now = () => 1_000_000; // t
-            const a = buildWidgetLines(wd, WIDTH, 0);
-            Date.now = () => 1_000_500; // t+500ms (would have rotated twice under the old code)
-            const b = buildWidgetLines(wd, WIDTH, 0);
-            assert.deepEqual(a, b, "footer byte-identical across 500ms with no state change");
-            // And the pinned head is the most-recent entry (third), not a rotation:
-            assert.ok(a.some((l) => l.includes("third")), "shows most-recent ticker entry");
-            assert.ok(a.some((l) => l.includes("(+2 more)")), "(+N more) suffix preserved");
-            assert.ok(a.some((l) => l.includes("because")), "lastWhy preserved");
-        }
-        finally {
-            Date.now = realNow;
-        }
-    });
-    it("footer still flips when the ticker actually changes (new pushTicker)", () => {
-        const realNow = Date.now;
-        try {
-            Date.now = () => 1_000_000;
-            const before = buildWidgetLines(stabBase({ ticker: [{ text: "a", at: 1 }] }), WIDTH, 0);
-            const after = buildWidgetLines(stabBase({
-                ticker: [
-                    { text: "a", at: 1 },
-                    { text: "b", at: 2 },
-                ],
-            }), WIDTH, 0);
-            assert.notDeepEqual(before, after, "footer changes when ticker grows");
-        }
-        finally {
-            Date.now = realNow;
-        }
-    });
-});
-describe("buildWidgetLines achievement flare (S35)", () => {
-    const achBase = (overrides = {}) => baseWd({
-        theme: DEFAULT_THEME,
-        tuiMode: "full",
-        gameMode: true,
-        level: 1,
-        cachePct: 42,
-        ...overrides,
-    });
-    it("achievementFlare + titles -> renders the unlock toast line", () => {
-        const lines = buildWidgetLines(achBase({
-            achievementFlare: true,
-            achievementFlareTitles: ["First Compact"],
-        }), WIDTH, 0);
-        assert.ok(lines.some((l) => l.includes("Achievement unlocked: First Compact")), "toast line present");
-    });
-    it("achievementFlare off -> no toast line", () => {
-        const lines = buildWidgetLines(achBase({
-            achievementFlare: false,
-            achievementFlareTitles: ["First Compact"],
-        }), WIDTH, 0);
-        assert.ok(!lines.some((l) => l.includes("Achievement unlocked")), "no toast when flare off");
-    });
-    it("gameMode off -> no toast even if flare set", () => {
-        const lines = buildWidgetLines(achBase({
-            gameMode: false,
-            achievementFlare: true,
-            achievementFlareTitles: ["X"],
-        }), WIDTH, 0);
-        assert.ok(!lines.some((l) => l.includes("Achievement unlocked")), "no toast when game off");
-    });
-    it("achievement toast is width-safe", () => {
-        const lines = buildWidgetLines(achBase({
-            achievementFlare: true,
-            achievementFlareTitles: ["First Compact", "Turn Veteran"],
-        }), 60, 0);
-        for (const l of lines)
-            assert.ok(visibleWidth(l) <= 60, "width safe");
-    });
-});

package/dist/extensions/mega-shutdown-widget.test.js

@@ -1,121 +0,0 @@
-/**
- * mega-shutdown-widget.test.ts — regression cover for two host-integration
- * fixes that the rest of the suite cannot see, because both are about what
- * happens at the boundary with pi rather than inside the compaction engine:
- *
- *  1. session_shutdown must close the PGlite indexes. They are lazily-opened
- *     module singletons whose handles keep node's event loop alive, so leaving
- *     them open made `pi -p` produce its answer and then hang instead of
- *     exiting. closeVectorIndex()/closeMemoryIndex() existed but had no
- *     non-test callers.
- *
- *  2. The above-editor widget must be suppressible. It is a persistent,
- *     animated, full-width panel that repaints on its own cadence, which fights
- *     terminals where the user drives scrollback (pi inside a Neovim
- *     `:terminal`). MEGACOMPACT_TUI_WIDGET=0 turns it off without touching
- *     compaction.
- */
-import { test } from "node:test";
-import assert from "node:assert/strict";
-import { mkdtempSync, rmSync } from "node:fs";
-import { tmpdir } from "node:os";
-import { join } from "node:path";
-import { closeVectorIndex, initVectorIndex, isVectorIndexDisabled, } from "../src/store/vectorIndex.js";
-import { closeMemoryIndex } from "../src/store/memoryIndex.js";
-import { loadConfig } from "./mega-config.js";
-import { MegaRuntime } from "./mega-runtime.js";
-const baseTmp = mkdtempSync(join(tmpdir(), "mc-shutdown-"));
-process.env.MEGACOMPACT_INDEX_DIR = join(baseTmp, "index");
-let counter = 0;
-/** Fresh per-test state dir so concurrent runs never collide on disk. */
-function isolate() {
-    process.env.MEGACOMPACT_STATE_DIR = join(baseTmp, `run-${counter++}`);
-}
-/** Minimal ExtensionContext slice: renderWidget only reaches ctx.ui.setWidget. */
-function widgetCtx(calls) {
-    return {
-        ui: {
-            setWidget: (key) => calls.push(key),
-        },
-    };
-}
-test.after(async () => {
-    // This file is itself a demonstration of the bug under test: without these
-    // closes the PGlite handles opened above keep the event loop alive and the
-    // test process never exits.
-    await Promise.all([closeVectorIndex(), closeMemoryIndex()]);
-    rmSync(baseTmp, { recursive: true, force: true });
-});
-test("MEGACOMPACT_TUI_WIDGET defaults on and is disabled by 0", () => {
-    isolate();
-    delete process.env.MEGACOMPACT_TUI_WIDGET;
-    assert.equal(loadConfig().tuiWidget, true, "widget should default to on");
-    process.env.MEGACOMPACT_TUI_WIDGET = "0";
-    assert.equal(loadConfig().tuiWidget, false);
-    process.env.MEGACOMPACT_TUI_WIDGET = "1";
-    assert.equal(loadConfig().tuiWidget, true);
-    delete process.env.MEGACOMPACT_TUI_WIDGET;
-});
-test("renderWidget registers the panel when tuiWidget is on", () => {
-    isolate();
-    delete process.env.MEGACOMPACT_TUI_WIDGET;
-    const runtime = new MegaRuntime(loadConfig());
-    try {
-        const calls = [];
-        runtime.renderWidget(widgetCtx(calls));
-        assert.equal(calls.length, 1, "expected one setWidget registration");
-    }
-    finally {
-        runtime.dispose();
-    }
-});
-test("renderWidget registers nothing when tuiWidget is off", () => {
-    isolate();
-    process.env.MEGACOMPACT_TUI_WIDGET = "0";
-    const runtime = new MegaRuntime(loadConfig());
-    try {
-        const calls = [];
-        // Repeated calls, because the panel is re-registered on every snapshot
-        // and every game-state change — one guarded path is not enough.
-        runtime.renderWidget(widgetCtx(calls));
-        runtime.renderWidget(widgetCtx(calls));
-        assert.deepEqual(calls, [], "widget must never be registered when disabled");
-    }
-    finally {
-        runtime.dispose();
-        delete process.env.MEGACOMPACT_TUI_WIDGET;
-    }
-});
-test("the extension's session_shutdown handler awaits index teardown", async () => {
-    isolate();
-    const handlers = {};
-    const pi = {
-        on(event, handler) {
-            (handlers[event] ??= []).push(handler);
-        },
-        registerCommand() { },
-        registerProvider() { },
-    };
-    const { default: extension } = await import("./mega-compact.js");
-    extension(pi);
-    const shutdown = handlers["session_shutdown"] ?? [];
-    assert.ok(shutdown.length, "extension must register a session_shutdown handler");
-    const event = { type: "session_shutdown" };
-    const ctx = {
-        ui: { setStatus: () => { }, notify: () => { }, setWidget: () => { } },
-        cwd: process.env.MEGACOMPACT_STATE_DIR,
-    };
-    // Assert the close by identity rather than by "some handler returned a
-    // promise": several modules register on session_shutdown and at least one
-    // other is already async, so a promise proves nothing about teardown. The
-    // index is a module singleton, so if shutdown really closed it, re-init
-    // hands back a *different* instance.
-    const before = await initVectorIndex();
-    if (isVectorIndexDisabled() || !before)
-        return; // PGlite unavailable — nothing to close
-    await Promise.all(shutdown.map((handler) => handler(event, ctx)));
-    const after = await initVectorIndex();
-    assert.notEqual(after, before, "session_shutdown must close the PGlite vector index");
-    // Idempotent: a second shutdown (reload, double-fire) must not throw.
-    await Promise.all(shutdown.map((handler) => handler(event, ctx)));
-});

package/dist/extensions/mega-teamrun.test.js

@@ -1,156 +0,0 @@
-/**
- * mega-teamrun.test.ts — regression test for the "auto-compact runs but context
- * never relieves during a team run (sub-agents)" bug.
- *
- * Loads the REAL compiled extension (extensions/mega-compact.js) through a
- * faithful mock pi (mirrors mega-compact.test.ts's harness) and drives the
- * exact event sequence a long team run produces:
- *
- *   agent_start -> context (over threshold) xN -> agent_end   (repeat x3)
- *
- * Asserts the TWO fixes:
- *   1. live trim FIRES per-call (computeLiveTrimCut no longer returns null on
- *      the anchor floor — was `cutNull`, liveTrimFires===0 before the fix).
- *   2. the DURABLE trim fires at agent_end while idle + over threshold
- *      (mid-run durable trigger), not only at parent settle.
- *
- * The mock ctx.compact() drives session_before_compact so we observe the
- * durable truncation. Counters come from MegaRuntime.diag* (set behind the
- * real handler code, inert in production).
- *
- * MEGACOMPACT_PGLITE_DISABLED keeps the run fast (no WASM index init).
- */
-import { test } from "node:test";
-import assert from "node:assert/strict";
-import { mkdtempSync, rmSync } from "node:fs";
-import { tmpdir } from "node:os";
-import { join } from "node:path";
-import { createRequire } from "node:module";
-import { closeVectorIndex } from "../src/store/vectorIndex.js";
-const require = createRequire(import.meta.url);
-const baseTmp = mkdtempSync(join(tmpdir(), "mc-team-"));
-process.env.MEGACOMPACT_INDEX_DIR = join(baseTmp, "index");
-process.env.MEGACOMPACT_PGLITE_DISABLED = "true"; // fast: skip WASM index
-let counter = 0;
-function harness() {
-    const stateDir = join(baseTmp, `run-${counter++}`);
-    process.env.MEGACOMPACT_STATE_DIR = stateDir;
-    process.env.MEGACOMPACT_DEBUG = "true";
-    process.env.MEGACOMPACT_THRESHOLD_TOKENS = "50";
-    process.env.MEGACOMPACT_FAST_GATE_PCT = "1";
-    // Strict race-guard mode double-counts diagAgentEndDurable (sync++ at branch
-    // entry + deferred++ in the setTimeout(500) callback = 6, not 3), lands
-    // compactCalls after the synchronous assertions, and leaks timers that hang
-    // `node --test`. The strict deferred path is covered by the two S38.5 tests
-    // in mega-compact.test.ts. Use the synchronous v0.7.4 path here.
-    process.env.MEGACOMPACT_RACE_GUARD_STRICT = "false";
-    process.env.MEGACOMPACT_ANCHOR_USER_MESSAGES = "1";
-    process.env.MEGACOMPACT_DURABLE_TRIM_FLOOR = "0"; // piCompactWouldNoop must not skip
-    process.env.MEGACOMPACT_MEMORY_AUTO_REVIEW = "false";
-    process.env.MEGACOMPACT_RAPTOR_ENABLED = "false";
-    delete process.env.MEGACOMPACT_LEGACY_DURABLE_TRIM;
-    const handlers = {};
-    const compactCalls = [];
-    function msg(role, text, toolName) {
-        if (role === "assistant" && toolName) {
-            return { role: "assistant", content: [{ type: "toolCall", name: toolName, id: "c1", arguments: {} }], api: "anthropic-messages", provider: "anthropic", model: "m", usage: { inputTokens: 1, outputTokens: 1, cacheReadTokens: 0, cacheWriteTokens: 0 }, stopReason: "tool_use", timestamp: 0 };
-        }
-        if (role === "toolResult" && toolName) {
-            return { role: "toolResult", content: [{ type: "text", text }], toolCallId: "c1", toolName, isError: false, timestamp: 0 };
-        }
-        return { role: "user", content: text, timestamp: 0 };
-    }
-    const session = [];
-    for (let i = 0; i < 14; i++) {
-        session.push(msg("user", `actually we decided to use approach ${i} for module ${i}`));
-        session.push(msg("assistant", `edited module ${i}`, "Edit"));
-        session.push(msg("toolResult", `edited module ${i}`, "Edit"));
-    }
-    const toEntry = (m, i) => ({ type: "message", id: `e${i}`, parentId: null, timestamp: String(i), message: m });
-    const sessionManager = {
-        getSessionId: () => "sess_team_001",
-        getEntries: () => session.map(toEntry),
-        getBranch: () => session.map(toEntry),
-    };
-    function makeCtx(over = {}) {
-        return {
-            ui: { setStatus: () => { }, notify: () => { }, select: () => { }, confirm: async () => true, input: async () => "", setWidget: () => { } },
-            mode: "tui", hasUI: true, cwd: stateDir, sessionManager,
-            modelRegistry: {}, model: undefined, isIdle: () => true, isProjectTrusted: () => true,
-            signal: undefined, abort: () => { }, hasPendingMessages: () => false, shutdown: () => { },
-            getContextUsage: () => ({ tokens: 200000, contextWindow: 200000, percent: 100 }),
-            // Mock ctx.compact() runs pi's flow and fires session_before_compact.
-            compact: (opts) => {
-                compactCalls.push(opts);
-                if (handlers["session_before_compact"]) {
-                    return handlers["session_before_compact"]({ type: "session_before_compact", reason: "threshold", willRetry: false, signal: undefined, preparation: { firstKeptEntryId: "e2", messagesToSummarize: session.slice(0, 2), tokensBefore: 500 } }, makeCtx());
-                }
-                return undefined;
-            },
-            getSystemPrompt: () => "system base",
-            ...over,
-        };
-    }
-    const pi = {
-        on: (ev, h) => { handlers[ev] = h; },
-        registerCommand: () => { }, registerTool: () => { }, registerShortcut: () => { },
-        registerFlag: () => { }, getFlag: () => undefined, registerMessageRenderer: () => { },
-        registerEntryRenderer: () => { }, sendMessage: () => { }, sendUserMessage: () => { },
-        appendEntry: () => { }, setSessionName: () => { }, getSessionName: () => undefined,
-        setLabel: () => { }, exec: async () => ({ stdout: "", stderr: "", code: 0 }),
-        getActiveTools: () => [], getAllTools: () => [], setActiveTools: () => { },
-        getCommands: () => [], setModel: async () => false, getThinkingLevel: () => "off",
-        setThinkingLevel: () => { },
-    };
-    const mod = require("./mega-compact.js");
-    mod.default(pi);
-    const { lastRuntime } = require("./mega-events.js");
-    const fire = (ev, event, ctx) => handlers[ev](event, ctx);
-    return {
-        stateDir, handlers, compactCalls, fire, ctx: makeCtx, session,
-        runtime: lastRuntime, // MegaRuntime with diag* counters
-        // Advance the debounce so agent_end (same instant) can trigger durable trim.
-        clearDebounce: () => { if (lastRuntime)
-            lastRuntime.debounceUntil = 0; },
-    };
-}
-test("team run: live trim fires AND durable trim fires per sub-agent (relieves context)", async () => {
-    const h = harness();
-    const ctx = h.ctx();
-    for (let a = 0; a < 3; a++) {
-        await h.fire("agent_start", { type: "agent_start", messages: [] }, ctx);
-        for (let i = 0; i < 4; i++) {
-            await h.fire("context", { type: "context", messages: h.session }, ctx);
-        }
-        // Real team runs settle seconds after the last context event; mimic that
-        // so the 2s debounce has elapsed and the durable trigger can fire.
-        await new Promise((r) => setTimeout(r, 2100));
-        h.clearDebounce();
-        await h.fire("agent_end", { type: "agent_end", messages: [] }, ctx);
-    }
-    const rt = h.runtime;
-    // FIX 1: live trim must actually fire (was 0 — computeLiveTrimCut returned null).
-    assert.ok(rt.diagLiveTrimFires > 0, "live trim fires during the team run (anchor-floor fix)");
-    assert.equal(rt.diagCtxCutNull, 0, "no live-trim cut skipped on anchor floor");
-    // FIX 2: durable trim must fire at each agent_end (was 0 — only at parent settle).
-    assert.equal(rt.diagAgentEndDurable, 3, "mid-run durable trigger fired at each agent_end");
-    assert.equal(rt.diagBeforeCompactSupplied, 3, "our durable trim supplied 3x (context relieved)");
-    assert.ok(h.compactCalls.length >= 3, "ctx.compact() invoked for durable trim between sub-agents");
-});
-test("control: session_before_compact supplies a durable compaction (parent settles)", async () => {
-    const h = harness();
-    const res = await h.fire("session_before_compact", { type: "session_before_compact", reason: "threshold", willRetry: false, signal: undefined, preparation: { firstKeptEntryId: "e2", messagesToSummarize: h.session.slice(0, 4), tokensBefore: 500 } }, h.ctx());
-    assert.ok(res?.compaction, "compaction result returned to pi");
-    assert.equal(res.compaction.firstKeptEntryId, "e2", "reuses pi's boundary (PREVENT-PI-002)");
-});
-test("cleanup", async () => {
-    // Race closeVectorIndex with a timeout to prevent 40-min hangs.
-    try {
-        await Promise.race([
-            closeVectorIndex(),
-            new Promise((r) => setTimeout(r, 3000)),
-        ]);
-    }
-    catch { /* ignore */ }
-    rmSync(baseTmp, { recursive: true, force: true });
-});

package/dist/extensions/mega-trim.test.js

@@ -1,58 +0,0 @@
-/**
- * mega-trim.test.ts — tests for the live compaction view builder (S16).
- */
-import { test } from "node:test";
-import assert from "node:assert/strict";
-import { buildLiveTrimmedView } from "./mega-trim.js";
-function m(role, text, extra = {}) {
-    return { role, text, toolName: undefined, input: undefined, output: undefined, ...extra };
-}
-test("buildLiveTrimmedView: prepends a compacted summary and keeps the recent anchor", () => {
-    const view = [
-        m("user", "old request one"), m("assistant", "old answer one"),
-        m("user", "old request two"), m("assistant", "old answer two"),
-        m("user", "recent keep me"), m("assistant", "recent keep me too"),
-    ];
-    // Compacted region = first 4; recent anchor = last 2.
-    const result = buildLiveTrimmedView(view, {
-        compactedFrom: 4, // index where the compacted region ends
-        summary: "<summary>earlier work on old requests</summary>",
-        anchorUserMessages: 1,
-    });
-    // First element is the injected compacted summary as a user-role message.
-    assert.equal(result[0].role, "user");
-    assert.ok(String(result[0].text).includes("earlier work on old requests"));
-    // Recent anchor preserved in order, no older messages leak through.
-    assert.equal(result.length, 1 + 2, "summary + 2 recent");
-    assert.ok(result.slice(1).some((x) => String(x.text).includes("recent keep me")));
-});
-test("buildLiveTrimmedView: empty summary returns the original view unchanged", () => {
-    const view = [m("user", "x"), m("assistant", "y")];
-    const result = buildLiveTrimmedView(view, { compactedFrom: 0, summary: "", anchorUserMessages: 1 });
-    assert.deepEqual(result, view);
-});
-test("buildLiveTrimmedView: never splits a toolCall/toolResult pair (PREVENT-PI-002)", () => {
-    const view = [
-        m("user", "q"), m("assistant", "calls tool", { toolName: "read" }), m("tool", "result"),
-        m("user", "keep"), m("assistant", "ok"),
-    ];
-    // cut=3 would start the preserved run on the orphaned tool result at index 2 —
-    // the builder must snap back so the toolCall/toolResult pair is not split.
-    const result = buildLiveTrimmedView(view, { compactedFrom: 3, summary: "<summary>s</summary>", anchorUserMessages: 1 });
-    // The tool result must never appear preserved WITHOUT its preceding toolCall.
-    const preserved = result.slice(1);
-    const hasToolResult = preserved.some((x) => x.role === "tool");
-    const hasToolCall = preserved.some((x) => x.role === "assistant" && x.toolName);
-    // Either the tool pair is kept together, or the tool result is dropped into
-    // the compacted region — it is never left orphaned.
-    assert.ok(!(hasToolResult && !hasToolCall), "no orphaned tool result in the preserved run");
-});
-test("buildLiveTrimmedView: honors the anchor floor (PREVENT-PI-001)", () => {
-    // cut would leave zero user messages in the anchor — must skip the trim.
-    const view = [
-        m("user", "old q"), m("assistant", "old a"),
-        m("assistant", "only assistant kept"),
-    ];
-    const result = buildLiveTrimmedView(view, { compactedFrom: 2, summary: "<summary>s</summary>", anchorUserMessages: 1 });
-    assert.deepEqual(result, view, "below anchor floor → no trim this call");
-});