@femtomc/mu-agent 26.3.3 → 26.3.4

package/README.md

@@ -77,7 +77,7 @@ Current stack:
 
 - `brandingExtension` — mu compact header/footer branding + default theme
 - `eventLogExtension` — event tail + watch widget
-- `uiExtension` — programmable `UiDoc` surface (`/mu ui ...`, `mu_ui`)
+- `uiExtension` — programmable `UiDoc` surface (`/mu ui ...`, `mu_ui`) with session-resume persistence via `custom` entries (`mu-ui-state/v1`)
 
 Default operator UI theme is `mu-gruvbox-dark`.
 

package/dist/extensions/ui.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"ui.d.ts","sourceRoot":"","sources":["../../src/extensions/ui.ts"],"names":[],"mappings":"AAUA,OAAO,KAAK,EAAE,YAAY,EAAoB,MAAM,+BAA+B,CAAC;AAq/CpF,wBAAgB,WAAW,CAAC,EAAE,EAAE,YAAY,QAkO3C;AAED,eAAe,WAAW,CAAC"}
+{"version":3,"file":"ui.d.ts","sourceRoot":"","sources":["../../src/extensions/ui.ts"],"names":[],"mappings":"AAWA,OAAO,KAAK,EAAE,YAAY,EAAoB,MAAM,+BAA+B,CAAC;AA0qDpF,wBAAgB,WAAW,CAAC,EAAE,EAAE,YAAY,QA4P3C;AAED,eAAe,WAAW,CAAC"}

package/dist/extensions/ui.js

@@ -1,4 +1,4 @@
-import { normalizeUiDocs, parseUiDoc, resolveUiStatusProfileName, uiStatusProfileWarnings, } from "@femtomc/mu-core";
+import { normalizeUiDocs, parseUiDoc, resolveUiStatusProfileName, stableSerializeJson, uiStatusProfileWarnings, } from "@femtomc/mu-core";
 import { matchesKey, truncateToWidth, visibleWidth } from "@mariozechner/pi-tui";
 import { registerMuSubcommand } from "./mu-command-dispatcher.js";
 const UI_DISPLAY_DOCS_MAX = 16;
@@ -6,6 +6,8 @@ const UI_PICKER_COMPONENTS_MAX = 8;
 const UI_PICKER_LIST_ITEMS_MAX = 4;
 const UI_PICKER_KEYVALUE_ROWS_MAX = 4;
 const UI_SESSION_KEY_FALLBACK = "__mu_ui_active_session__";
+const UI_STATE_CUSTOM_TYPE = "mu-ui-state/v1";
+const UI_STATE_SCHEMA_VERSION = 1;
 const UI_PROMPT_PREVIEW_MAX = 160;
 const UI_INTERACT_SHORTCUT_PRIMARY = "ctrl+shift+u";
 const UI_INTERACT_SHORTCUT_FALLBACK = "alt+u";
@@ -46,6 +48,7 @@ function createState() {
         interactionDepth: 0,
         lastStatusText: null,
         lastWidgetSignature: null,
+        lastPersistSignature: null,
     };
 }
 function pruneStaleStates(nowMs) {
@@ -389,6 +392,164 @@ function parseDocListInput(value) {
     }
     return { ok: true, docs: normalizeUiDocs(docs, { maxDocs: UI_DISPLAY_DOCS_MAX }) };
 }
+function normalizeStringArray(value) {
+    if (!Array.isArray(value)) {
+        return [];
+    }
+    const out = [];
+    const seen = new Set();
+    for (const entry of value) {
+        if (typeof entry !== "string") {
+            continue;
+        }
+        const trimmed = entry.trim();
+        if (!trimmed || seen.has(trimmed)) {
+            continue;
+        }
+        seen.add(trimmed);
+        out.push(trimmed);
+    }
+    return out;
+}
+function parsePersistedPendingPrompt(value) {
+    if (!isPlainObject(value)) {
+        return null;
+    }
+    const kind = value.kind === "action" || value.kind === "review" ? value.kind : null;
+    if (!kind) {
+        return null;
+    }
+    const uiIdRaw = typeof value.ui_id === "string" ? value.ui_id.trim() : "";
+    if (!uiIdRaw) {
+        return null;
+    }
+    const actionIdRaw = typeof value.action_id === "string" ? value.action_id.trim() : "";
+    if (kind === "review") {
+        return { kind, ui_id: uiIdRaw };
+    }
+    if (actionIdRaw) {
+        return { kind, ui_id: uiIdRaw, action_id: actionIdRaw };
+    }
+    return { kind, ui_id: uiIdRaw };
+}
+function parsePersistedUiState(value) {
+    if (!isPlainObject(value)) {
+        return null;
+    }
+    if (value.v !== UI_STATE_SCHEMA_VERSION) {
+        return null;
+    }
+    const parsedDocs = parseDocListInput(value.docs);
+    if (!parsedDocs.ok) {
+        return null;
+    }
+    const pendingPrompts = [];
+    const pendingRaw = Array.isArray(value.pending_prompts) ? value.pending_prompts : [];
+    for (const pending of pendingRaw) {
+        const parsed = parsePersistedPendingPrompt(pending);
+        if (parsed) {
+            pendingPrompts.push(parsed);
+        }
+    }
+    return {
+        v: 1,
+        docs: parsedDocs.docs,
+        pending_prompts: pendingPrompts,
+        prompted_revision_keys: normalizeStringArray(value.prompted_revision_keys),
+        awaiting_ui_ids: normalizeStringArray(value.awaiting_ui_ids),
+    };
+}
+function serializeUiState(state) {
+    retainPromptedRevisionKeysForActiveDocs(state);
+    retainAwaitingUiIdsForActiveDocs(state);
+    retainPendingPromptsForActiveDocs(state);
+    const docs = activeDocs(state, UI_DISPLAY_DOCS_MAX);
+    return {
+        v: 1,
+        docs,
+        pending_prompts: state.pendingPrompts.map((pending) => ({
+            kind: pending.kind,
+            ui_id: pending.uiId,
+            ...(pending.actionId ? { action_id: pending.actionId } : {}),
+        })),
+        prompted_revision_keys: [...state.promptedRevisionKeys].sort((left, right) => left.localeCompare(right)),
+        awaiting_ui_ids: [...state.awaitingUiIds].sort((left, right) => left.localeCompare(right)),
+    };
+}
+function uiStatePersistSignature(value) {
+    return stableSerializeJson(value);
+}
+function persistUiStateSnapshot(pi, state) {
+    const snapshot = serializeUiState(state);
+    const signature = uiStatePersistSignature(snapshot);
+    if (state.lastPersistSignature === signature) {
+        return;
+    }
+    pi.appendEntry(UI_STATE_CUSTOM_TYPE, snapshot);
+    state.lastPersistSignature = signature;
+}
+function applyPersistedUiState(state, persisted) {
+    state.docsById.clear();
+    state.pendingPrompts = [];
+    state.promptedRevisionKeys.clear();
+    state.awaitingUiIds.clear();
+    state.interactionDepth = 0;
+    state.lastStatusText = null;
+    state.lastWidgetSignature = null;
+    for (const doc of persisted.docs) {
+        state.docsById.set(doc.ui_id, doc);
+    }
+    for (const revisionKey of persisted.prompted_revision_keys) {
+        state.promptedRevisionKeys.add(revisionKey);
+    }
+    for (const uiId of persisted.awaiting_ui_ids) {
+        state.awaitingUiIds.add(uiId);
+    }
+    for (const pending of persisted.pending_prompts) {
+        enqueuePendingPrompt(state, {
+            kind: pending.kind,
+            uiId: pending.ui_id,
+            actionId: pending.action_id,
+        });
+    }
+    retainPromptedRevisionKeysForActiveDocs(state);
+    retainAwaitingUiIdsForActiveDocs(state);
+    retainPendingPromptsForActiveDocs(state);
+    state.lastPersistSignature = uiStatePersistSignature(serializeUiState(state));
+}
+function latestPersistedUiStateFromBranch(entries) {
+    let latest = null;
+    for (const entry of entries) {
+        if (!isPlainObject(entry)) {
+            continue;
+        }
+        if (entry.type !== "custom" || entry.customType !== UI_STATE_CUSTOM_TYPE) {
+            continue;
+        }
+        const parsed = parsePersistedUiState(entry.data);
+        if (!parsed) {
+            continue;
+        }
+        latest = parsed;
+    }
+    return latest;
+}
+function restoreStateFromSessionBranch(ctx, key) {
+    const manager = ctx.sessionManager;
+    const branch = typeof manager.getBranch === "function" ? manager.getBranch() : [];
+    const restored = createState();
+    const persisted = latestPersistedUiStateFromBranch(branch);
+    if (persisted) {
+        applyPersistedUiState(restored, persisted);
+    }
+    const nowMs = Date.now();
+    pruneStaleStates(nowMs);
+    STATE_BY_SESSION.set(key, {
+        state: restored,
+        lastAccessMs: nowMs,
+    });
+    return restored;
+}
 function statusProfileWarningsExtraForDoc(doc) {
     const warnings = uiStatusProfileWarnings(doc);
     if (warnings.length === 0) {
@@ -1355,6 +1516,7 @@ export function uiExtension(pi) {
         removePendingPromptsForUiId(state, selectedDoc.ui_id);
         retainAwaitingUiIdsForActiveDocs(state);
         retainPendingPromptsForActiveDocs(state);
+        persistUiStateSnapshot(pi, state);
         ctx.ui.notify(`Submitted prompt from ${selectedDoc.ui_id}/${selectedAction.id}.`, "info");
     });
     registerMuSubcommand(pi, {
@@ -1432,15 +1594,35 @@ export function uiExtension(pi) {
             if (ctx.hasUI && result.ok && result.changedUiIds && result.changedUiIds.length > 0) {
                 armAutoPromptForUiDocs(state, result.changedUiIds);
             }
+            const shouldPersistMutation = result.ok &&
+                (result.action === "set" ||
+                    result.action === "update" ||
+                    result.action === "replace" ||
+                    result.action === "remove" ||
+                    result.action === "clear");
+            if (shouldPersistMutation) {
+                persistUiStateSnapshot(pi, state);
+            }
             refreshUi(ctx);
             return buildToolResult({ state, ...result });
         },
     });
-    pi.on("session_start", (_event, ctx) => {
+    const restoreAndRefresh = (ctx) => {
+        const key = sessionKey(ctx);
+        restoreStateFromSessionBranch(ctx, key);
         refreshUi(ctx);
+    };
+    pi.on("session_start", (_event, ctx) => {
+        restoreAndRefresh(ctx);
     });
     pi.on("session_switch", (_event, ctx) => {
-        refreshUi(ctx);
+        restoreAndRefresh(ctx);
+    });
+    pi.on("session_fork", (_event, ctx) => {
+        restoreAndRefresh(ctx);
+    });
+    pi.on("session_tree", (_event, ctx) => {
+        restoreAndRefresh(ctx);
     });
     pi.on("agent_end", async (_event, ctx) => {
         if (!ctx.hasUI) {
@@ -1453,6 +1635,7 @@ export function uiExtension(pi) {
         if (!pending) {
             return;
         }
+        persistUiStateSnapshot(pi, state);
         if (pending.kind === "action") {
             ctx.ui.notify(`Agent requested input via ${pending.uiId}. Submit now or press ${UI_INTERACT_SHORTCUT_HINT} later.`, "info");
         }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@femtomc/mu-agent",
-  "version": "26.3.3",
+  "version": "26.3.4",
   "description": "Shared operator runtime for mu assistant sessions and serve extensions.",
   "keywords": [
     "mu",
@@ -24,7 +24,7 @@
     "themes/**"
   ],
   "dependencies": {
-    "@femtomc/mu-core": "26.3.3",
+    "@femtomc/mu-core": "26.3.4",
     "@mariozechner/pi-agent-core": "^0.54.2",
     "@mariozechner/pi-ai": "^0.54.2",
     "@mariozechner/pi-coding-agent": "^0.54.2",