@jmfederico/pi-web 1.202607.1 → 1.202607.3

package/dist/server/sessions/piSessionService.js

@@ -1,23 +1,53 @@
 import { statSync } from "node:fs";
 import { join } from "node:path";
-import { open, readFile, writeFile } from "node:fs/promises";
+import { readFile, writeFile } from "node:fs/promises";
 import { createAgentSessionFromServices, createAgentSessionRuntime, createAgentSessionServices, createEditToolDefinition, defineTool, readStoredCredential, SessionManager, } from "@earendil-works/pi-coding-agent";
 import { projectBrowserMessage } from "../browserMessageProjection.js";
 import { pageMessagesAtSafeBoundary } from "./messagePaging.js";
 import { BUILTIN_COMMANDS } from "./builtinCommands.js";
 import { SessionCommandService } from "./sessionCommandService.js";
+import { projectSessionTree } from "./sessionTreeProjection.js";
 import { SessionArchiveStore } from "./sessionArchiveStore.js";
 import { findArchiveCandidateByIdOrPrefix, planSessionArchiveTree } from "./sessionArchiveTree.js";
 import { deterministicSessionName, fallbackSessionName, generateShortSessionName } from "./sessionNameGenerator.js";
 import { computeEditPreview } from "./editPreview.js";
 import { attachmentsToInlineImages, saveAttachmentsToWorkspace } from "./attachmentService.js";
 import { parsePromptAttachments } from "../../shared/promptAttachments.js";
+import { ASK_USER_ANSWERS_CUSTOM_TYPE, SESSION_TREE_CUSTOM_INSTRUCTIONS_MAX_LENGTH, SESSION_UNREAD_LIMIT } from "../../shared/apiTypes.js";
 import { canonicalizeStoredCwd, cwdPathsEqual } from "../workingDirectory.js";
+import { readSessionHeaderSummary } from "./sessionFileHeader.js";
+import { countOutOfListingChildren, locateOutOfListingParents } from "./parentSessionLocator.js";
+import { siblingWorkspaceCwds } from "../workspaces/projectWorkspaceCwds.js";
+import { createAskUserToolDefinition } from "./askUserTool.js";
+import { PendingAskStore, renderAskUserAnswersText } from "./pendingAskStore.js";
+import { PendingExtensionDialogStore } from "./pendingExtensionDialogStore.js";
+import { ExtensionDialogWaiters, effectiveExtensionDialogTimeoutMs, extensionDialogCancelValue } from "./extensionDialogWaiters.js";
+import { DEFAULT_EXTENSION_DIALOGS_TIMEOUT_MS } from "../../config.js";
 import { createSpawnSessionToolDefinition } from "./spawnSessionTool.js";
 import { createSubsessionToolDefinitions } from "./spawnSubsessionTool.js";
 import { buildTranscriptView } from "./subsessionTranscript.js";
 import { planSessionCleanup, summarizeSessionCleanupExecution } from "./sessionCleanup.js";
+import { SessionNotificationStore, } from "./sessionNotificationStore.js";
+import { plainTextTheme } from "./plainTextTheme.js";
+import { SessionUnreadStore } from "./sessionUnreadStore.js";
 const noopLogger = { info() { } };
+const DEFAULT_UNREAD_PUBLICATION_RETRY_MS = 1_000;
+/**
+ * User-facing names for the two phases of session startup PI WEB can prove it
+ * is inside: it awaits exactly one call for each, so the phase is a fact rather
+ * than a guess. Deliberately free of internal symbol names and file paths.
+ */
+const STARTUP_PHASE_RUNTIME = "Starting the Pi session";
+const STARTUP_PHASE_EXTENSIONS = "Loading session extensions";
+/**
+ * Appended to whichever phase is running when a background provider catalog
+ * refresh happens to be in flight. It is stated as a concurrent fact, never as
+ * the cause: PI WEB can verify that a refresh is running, but not that this
+ * particular startup is waiting on it.
+ */
+const STARTUP_CONCURRENT_CATALOG_REFRESH = "provider model lists are refreshing";
+const MAX_UNREAD_PUBLICATION_RETRY_MS = 30_000;
+const MAX_PENDING_UNREAD_MUTATIONS = SESSION_UNREAD_LIMIT + 1;
 function noop() {
     // Intentionally empty default unsubscribe callback.
 }
@@ -38,6 +68,40 @@ function isPiSessionRef(ref) {
 function lookupMatchesActiveSession(ref, active) {
     return !isPiSessionRef(ref) || cwdPathsEqual(active.runtime.cwd, ref.cwd);
 }
+function lookupMatchesStartupSession(ref, session) {
+    return !isPiSessionRef(ref) || cwdPathsEqual(session.sessionManager.getCwd(), ref.cwd);
+}
+function sessionTreeNavigationOptions(request) {
+    switch (request.summary.mode) {
+        case "none":
+            return { summarize: false };
+        case "default":
+            return { summarize: true };
+        case "custom": {
+            const customInstructions = request.summary.instructions.trim();
+            if (customInstructions === "")
+                throw new Error("Custom branch-summary instructions are required");
+            if (customInstructions.length > SESSION_TREE_CUSTOM_INSTRUCTIONS_MAX_LENGTH) {
+                throw new Error(`Custom branch-summary instructions must be at most ${String(SESSION_TREE_CUSTOM_INSTRUCTIONS_MAX_LENGTH)} characters`);
+            }
+            return { summarize: true, customInstructions };
+        }
+    }
+}
+function decrementWeakCount(counts, key) {
+    const remaining = (counts.get(key) ?? 1) - 1;
+    if (remaining <= 0)
+        counts.delete(key);
+    else
+        counts.set(key, remaining);
+}
+function decrementMapCount(counts, key) {
+    const remaining = (counts.get(key) ?? 1) - 1;
+    if (remaining <= 0)
+        counts.delete(key);
+    else
+        counts.set(key, remaining);
+}
 function requirePromptText(value) {
     if (typeof value !== "string")
         throw new Error("Prompt text is required");
@@ -50,6 +114,8 @@ function parsePromptStreamingBehavior(value) {
         return value;
     throw new Error('Prompt streamingBehavior must be "steer" or "followUp"');
 }
+const CLEAR_RUNTIME_NOTIFICATIONS = { kind: "clear", reason: "runtime-close" };
+const DEFER_RUNTIME_NOTIFICATIONS = { kind: "defer" };
 function resourceDiagnosticToWarning(diagnostic, source) {
     return {
         severity: diagnostic.type === "error" ? "error" : "warning",
@@ -178,19 +244,22 @@ function createRuntimeWithOneShotSessionOptions(createRuntime, initialModel, del
         });
     };
 }
-export function createPiWebCustomToolDefinitions(cwd, delegationEnabled, spawn, subsessions) {
+export function createPiWebCustomToolDefinitions(cwd, delegationEnabled, spawn, subsessions, askUser) {
     return [
         createPiWebEditToolDefinition(cwd),
         ...(delegationEnabled && spawn !== undefined ? [createSpawnSessionToolDefinition(cwd, { spawn })] : []),
         ...(delegationEnabled && subsessions !== undefined ? createSubsessionToolDefinitions(cwd, subsessions) : []),
+        // Asking the user is not delegation: the questions land in the session the
+        // user is already watching, so tracked children may ask too.
+        ...(askUser === undefined ? [] : [createAskUserToolDefinition(askUser)]),
     ];
 }
-function createDefaultRuntimeFactory(modelRuntime, sessionManagers, spawn, subsessions) {
+function createDefaultRuntimeFactory(modelRuntime, sessionManagers, spawn, subsessions, askUser) {
     return async ({ cwd, agentDir, sessionManager, sessionStartEvent, initialModel, delegationToolsEnabled }) => {
         const services = await createAgentSessionServices({ cwd, agentDir, modelRuntime });
         const resolvedDelegationToolsEnabled = delegationToolsEnabled
             ?? await sessionAllowsDelegationTools(sessionManager, sessionManagers);
-        const customTools = createPiWebCustomToolDefinitions(cwd, resolvedDelegationToolsEnabled, spawn, subsessions);
+        const customTools = createPiWebCustomToolDefinitions(cwd, resolvedDelegationToolsEnabled, spawn, subsessions, askUser);
         const result = await createAgentSessionFromServices({
             services,
             sessionManager,
@@ -227,7 +296,24 @@ export class PiSessionService {
         this.events = events;
         this.active = new Map();
         this.pendingSessionOpens = new Map();
+        /**
+         * Sessions whose extension binding is still in flight. A `session_start`
+         * dialog parks that window before the session ever becomes active, so this
+         * is the only way the dialog answer/cancel and status paths can reach it;
+         * {@link getOrOpen} never consults it, keeping every other operation gated
+         * on full readiness.
+         */
+        this.startupSessions = new Map();
         this.activities = new Map();
+        /** Runtime-identity gate held while Pi may await abandoned-branch summarization. */
+        this.treeNavigations = new WeakSet();
+        /** Counts async operations that may append an entry before they settle. */
+        this.sessionEntryMutationCounts = new WeakMap();
+        /** Runtime/session-identity reservations for operations that must not overlap tree navigation. */
+        this.treeExclusiveRuntimeOperationCounts = new WeakMap();
+        this.treeExclusiveSessionOperationCounts = new Map();
+        this.deferredSubsessionNotifications = new WeakMap();
+        this.deferredGeneratedSessionNames = new WeakMap();
         this.compactionPromptQueues = new Map();
         this.compactionDrainTimers = new Map();
         this.authLossWarnings = new Set();
@@ -239,19 +325,50 @@ export class PiSessionService {
         this.subsessionLinks = new Map();
         /** Parent id/file identities whose persisted links have already been loaded. */
         this.subsessionHydratedParents = new Set();
+        /** Session file path -> its parsed header. Headers are written once, so successful reads are cached. */
+        this.sessionHeaderCache = new Map();
         /**
          * Tracked subsession id -> whether a completion notification is armed.
          * Armed when the child starts working; firing on completion disarms it so a
          * child that works again (and stops again) notifies the parent each time.
          */
         this.subsessionNotifyArmed = new Map();
+        this.notificationGenerationBySession = new WeakMap();
+        /** The parked extension Promise resolvers behind the store's open dialogs. */
+        this.dialogWaiters = new ExtensionDialogWaiters();
+        this.pendingUnreadMutations = [];
+        this.unreadPublicationFlushRequested = false;
+        this.unreadPublicationStopped = false;
+        /**
+         * Read a session file header, memoized per path. Pi writes the header once at
+         * session creation, so a successful read stays valid for the process lifetime;
+         * failures are not cached so a session file that appears later is picked up.
+         */
+        this.readCachedSessionHeader = async (sessionFile) => {
+            const cached = this.sessionHeaderCache.get(sessionFile);
+            if (cached !== undefined)
+                return cached;
+            const header = await readSessionHeaderSummary(sessionFile);
+            if (header !== undefined)
+                this.sessionHeaderCache.set(sessionFile, header);
+            return header;
+        };
         this.archiveStore = deps.archiveStore ?? new SessionArchiveStore();
         this.agentDir = deps.agentDir;
         this.sessionManager = deps.sessionManager;
         this.modelRuntime = deps.modelRuntime;
         this.spawnTargets = deps.spawnTargets;
+        this.projectWorkspaces = deps.projectWorkspaces;
         this.logger = deps.logger ?? noopLogger;
         this.now = deps.now ?? (() => new Date());
+        this.notificationStore = deps.notificationStore ?? new SessionNotificationStore();
+        this.unreadStore = deps.unreadStore ?? new SessionUnreadStore();
+        this.pendingAskStore = deps.pendingAskStore ?? new PendingAskStore();
+        this.pendingExtensionDialogStore = deps.pendingExtensionDialogStore ?? new PendingExtensionDialogStore();
+        this.extensionDialogsTimeoutMs = deps.extensionDialogsTimeoutMs ?? DEFAULT_EXTENSION_DIALOGS_TIMEOUT_MS;
+        this.catalogRefreshStatus = deps.catalogRefreshStatus;
+        this.unreadPublicationRetryInitialMs = Math.max(0, deps.unreadPublicationRetryDelayMs ?? DEFAULT_UNREAD_PUBLICATION_RETRY_MS);
+        this.unreadPublicationRetryDelayMs = this.unreadPublicationRetryInitialMs;
         // Subsessions are a beta capability gated behind their own flag, and they
         // also require the spawn capability (they share its project-scope resolver).
         const subsessionsActive = this.spawnTargets !== undefined && deps.subsessionsEnabled === true;
@@ -260,25 +377,63 @@ export class PiSessionService {
             list: (parentSessionId, parentSessionFile) => this.listSubsessions(parentSessionId, parentSessionFile),
             check: (parentSessionId, sessionId, parentSessionFile) => this.checkSubsession(parentSessionId, sessionId, parentSessionFile),
             read: (parentSessionId, sessionId, query, parentSessionFile) => this.readSubsession(parentSessionId, sessionId, query, parentSessionFile),
-        });
+        }, deps.askUserEnabled === true ? { open: (input) => this.openAsk(input) } : undefined);
         this.createAgentRuntime = deps.createAgentRuntime ?? defaultCreateAgentRuntime;
         this.workspaceActivity = deps.workspaceActivity;
         this.heartbeat = setInterval(() => { this.publishHeartbeats(); }, deps.heartbeatIntervalMs ?? 2000);
         this.commandService = new SessionCommandService((sessionId) => this.getActive(sessionId), (sessionId, text) => this.prompt(sessionId, text, undefined, undefined, { echoUserMessage: false }), events, {
             onCompactionStart: (session) => {
+                this.beginSessionEntryMutation(session, "compact the session");
                 this.publishActivity(session, "compacting", "active");
                 this.publishStatus(session);
             },
             onCompactionEnd: (session, result, detail) => {
+                this.endSessionEntryMutation(session);
                 this.publishActivity(session, result === "success" ? "compaction complete" : "compaction failed", result === "success" ? "idle" : "error", detail);
                 this.publishStatus(session);
             },
             reloadSession: (session) => this.reloadSessionRuntime(session),
+            getSessionTree: (session) => {
+                if (typeof session.sessionManager.getTree !== "function" || typeof session.navigateTree !== "function")
+                    return undefined;
+                return projectSessionTree(session.sessionManager.getTree(), session.sessionManager.getLeafId());
+            },
+            hasActiveWork: (session) => this.hasActiveWork(session),
+            isTreeNavigationActive: (session) => this.treeNavigations.has(session),
+            runSessionReplacement: (session, operation) => this.runTreeExclusiveOperation([{ sessionId: session.sessionId, session }], "Stop current session activity before replacing the session", operation),
         }, { listSessionNames: (cwd) => this.listSessionNames(cwd) });
     }
     activeCount() {
         return this.active.size;
     }
+    notificationCatalog() {
+        return this.notificationStore.catalogSnapshot();
+    }
+    async unreadCatalog() {
+        await this.publishUnreadMutations([]);
+        return this.unreadStore.durableCatalogSnapshot();
+    }
+    async acknowledgeUnread(sessionId, request) {
+        const result = this.unreadStore.acknowledge(sessionId, {
+            ...request,
+            cwd: canonicalizeStoredCwd(request.cwd),
+        });
+        await this.publishUnreadMutations(result.mutations);
+        return this.unreadStore.durableCatalogSnapshot();
+    }
+    notificationInbox(ref) {
+        return this.notificationStore.inboxSnapshot(ref.id, canonicalizeStoredCwd(ref.cwd));
+    }
+    dismissNotification(ref, request) {
+        const result = this.notificationStore.dismissNotification(ref.id, canonicalizeStoredCwd(ref.cwd), request.daemonInstanceId, request.notificationId);
+        this.publishNotificationMutations(result.mutations);
+        return result.snapshot;
+    }
+    dismissAllNotifications(ref, request) {
+        const result = this.notificationStore.dismissAll(ref.id, canonicalizeStoredCwd(ref.cwd), request.daemonInstanceId, request.throughOrder, request.throughOverflowWatermark);
+        this.publishNotificationMutations(result.mutations);
+        return result.snapshot;
+    }
     async cleanupPreview(request) {
         return previewResponseFromPlan(await this.cleanupPlan(request));
     }
@@ -296,22 +451,24 @@ export class PiSessionService {
                 skippedBusySessionIds.add(input.sessionId);
                 continue;
             }
-            await this.closeActive(input.sessionId);
+            await this.closeActive(input.sessionId, { kind: "clear", reason: "archive" });
             readyArchiveInputs.push(input);
         }
         await this.archiveStoreArchiveMany(readyArchiveInputs);
         archiveInputs.push(...readyArchiveInputs);
+        await this.forgetUnreadSessions(readyArchiveInputs);
         for (const record of plan.deleteRecords) {
             if (this.activeSessionHasWork(record.sessionId)) {
                 skippedBusySessionIds.add(record.sessionId);
                 continue;
             }
-            await this.closeActive(record.sessionId);
+            await this.closeActive(record.sessionId, { kind: "clear", reason: "delete" });
             readyDeleteRecords.push(record);
         }
         await this.ensureArchivedRecordsMoved(readyDeleteRecords);
         const deletedSessionIds = new Set(await this.archiveStoreDeleteArchivedMany(readyDeleteRecords.map((record) => record.sessionId)));
         deleteRecords.push(...readyDeleteRecords.filter((record) => deletedSessionIds.has(record.sessionId)));
+        await this.forgetUnreadSessions(deleteRecords);
         return summarizeSessionCleanupExecution({
             archiveInputs,
             deleteRecords,
@@ -321,14 +478,26 @@ export class PiSessionService {
         });
     }
     async dispose() {
+        this.unreadPublicationStopped = true;
+        this.clearUnreadPublicationRetry();
         clearInterval(this.heartbeat);
         this.clearCompactionDrainTimers();
+        // Same startup-park hazard as closeActive(): settle `session_start` dialogs
+        // of sessions still binding extensions before awaiting their pending opens.
+        for (const sessionId of this.startupSessions.keys())
+            this.endSessionExtensionDialogs(sessionId);
         const pendingOpens = this.pendingSessionOpenPromises();
         if (pendingOpens.length > 0)
             await Promise.allSettled(pendingOpens);
         const activeSessions = Array.from(new Set(this.active.values()));
+        for (const active of activeSessions) {
+            this.forgetUnreadActivity(active.runtime.session);
+            this.pendingAskStore.forgetSession(active.runtime.session.sessionId);
+            this.endSessionExtensionDialogs(active.runtime.session.sessionId);
+        }
         this.active.clear();
         this.pendingSessionOpens.clear();
+        this.startupSessions.clear();
         this.activities.clear();
         this.compactionPromptQueues.clear();
         this.authLossWarnings.clear();
@@ -337,16 +506,20 @@ export class PiSessionService {
         this.subsessionLinks.clear();
         this.subsessionHydratedParents.clear();
         this.subsessionNotifyArmed.clear();
+        this.sessionHeaderCache.clear();
+        this.notificationStore.clearAll("service-dispose");
         await Promise.all(activeSessions.map(async (active) => {
             active.unsubscribe();
+            active.runtime.setRebindSession(undefined);
             this.workspaceActivity?.removeSession(active.runtime.session.sessionId, active.runtime.session.sessionManager.getCwd());
             try {
-                await active.runtime.session.abort();
+                await this.abortSessionOperations(active.runtime.session);
             }
             finally {
                 await active.runtime.dispose();
             }
         }));
+        await this.publishUnreadMutations([]);
     }
     async list(cwd) {
         const [sessions, archivedRecords] = await Promise.all([this.sessionManager.list(cwd), this.archiveStore.list()]);
@@ -355,19 +528,82 @@ export class PiSessionService {
             .filter((record) => record.cwd === cwd)
             .map((record) => this.ensureArchivedSessionMoved(record, sessionsById.get(record.sessionId))));
         const archivedById = new Map(archivedForCwd.map((record) => [record.sessionId, record]));
+        for (const record of archivedForCwd) {
+            this.publishNotificationMutations(this.notificationStore.clearSession(record.sessionId, "archive-reconcile"));
+        }
         const unarchivedSessions = sessions.filter((session) => !archivedById.has(session.id)).map(clientSessionFromListEntry);
-        this.workspaceActivity?.reconcileSessionActivity(cwd, this.reconcilableSessionIds(cwd, unarchivedSessions.map((session) => session.id), archivedById));
+        const reconcilableSessionIds = this.reconcilableSessionIds(cwd, unarchivedSessions.map((session) => session.id), archivedById);
+        this.workspaceActivity?.reconcileSessionActivity(cwd, reconcilableSessionIds);
+        await this.publishUnreadMutations(this.unreadStore.reconcileCwd(canonicalizeStoredCwd(cwd), reconcilableSessionIds));
         const archivedSessions = archivedForCwd
             .sort(compareArchivedRecords)
             .map((record) => clientSessionFromArchivedRecord(record, sessionsById.get(record.sessionId)))
             .filter(isDefined);
-        return [...unarchivedSessions, ...archivedSessions];
+        return await this.withRelatedSessionsElsewhere([...unarchivedSessions, ...archivedSessions], cwd);
+    }
+    /**
+     * Annotate a listing with the session relationships that cross workspace
+     * boundaries: where an out-of-listing parent lives, and how many children a
+     * listed session has in sibling workspaces.
+     *
+     * Both directions are best-effort. An unreadable parent header or a sibling
+     * workspace that cannot be listed leaves the session unannotated rather than
+     * failing the listing, and the browser falls back to its generic states.
+     */
+    async withRelatedSessionsElsewhere(sessions, cwd) {
+        const [parentLocations, childCounts] = await Promise.all([
+            locateOutOfListingParents(sessions, cwd, this.readCachedSessionHeader),
+            this.countChildrenInSiblingWorkspaces(sessions, cwd),
+        ]);
+        return sessions.map((session) => {
+            const parent = session.parentSessionPath === undefined ? undefined : parentLocations.get(session.parentSessionPath);
+            const childrenElsewhere = childCounts.get(session.path);
+            if (parent === undefined && childrenElsewhere === undefined)
+                return session;
+            const annotated = { ...session };
+            if (parent !== undefined) {
+                annotated.parentSessionId = parent.parentSessionId;
+                annotated.parentSessionCwd = parent.parentSessionCwd;
+            }
+            if (childrenElsewhere !== undefined)
+                annotated.childSessionsElsewhere = childrenElsewhere;
+            return annotated;
+        });
+    }
+    /**
+     * Count children of the listed sessions that live in other workspaces of the
+     * same project.
+     *
+     * Only sibling workspaces are scanned: agents may only spawn into workspaces
+     * of the spawning session's own project, so that bounds where a child can be.
+     * Listing is skipped entirely when no project-workspace locator is configured
+     * or the cwd belongs to no registered project.
+     */
+    async countChildrenInSiblingWorkspaces(sessions, cwd) {
+        if (this.projectWorkspaces === undefined || sessions.length === 0)
+            return new Map();
+        try {
+            const siblingCwds = await siblingWorkspaceCwds(this.projectWorkspaces, cwd);
+            if (siblingCwds.length === 0)
+                return new Map();
+            const listings = await Promise.all(siblingCwds.map(async (siblingCwd) => {
+                const entries = await this.sessionManager.list(siblingCwd);
+                return entries.flatMap((entry) => entry.parentSessionPath === undefined ? [] : [entry.parentSessionPath]);
+            }));
+            return countOutOfListingChildren(sessions, listings.flat());
+        }
+        catch (error) {
+            this.logger.info({ cwd, error: error instanceof Error ? error.message : String(error) }, "failed to count child sessions in sibling workspaces");
+            return new Map();
+        }
     }
     async start(cwd, options = {}) {
         return this.startSession(cwd, options);
     }
     async startSession(cwd, options) {
         const active = await this.create(this.sessionManager.create(cwd, options.parentSession === undefined ? undefined : { parentSession: options.parentSession }), cwd, {
+            startupIntent: "create",
+            ...(options.startupToken === undefined ? {} : { startupToken: options.startupToken }),
             ...(options.initialModel === undefined ? {} : { initialModel: options.initialModel }),
             ...(options.creationProvenance === undefined ? {} : { creationProvenance: options.creationProvenance }),
         });
@@ -431,13 +667,224 @@ export class PiSessionService {
             ...(parentSessionFile === undefined ? {} : { parentSessionFile }),
             cwd: decision.cwd,
         };
-        this.registerVerifiedSubsession(link);
+        await this.registerVerifiedSubsession(link);
         this.persistSubsessionLink(link);
         this.persistSubsessionChildMarker(input.parentSessionId, created.id);
         await this.prompt(created.id, input.prompt);
         this.logger.info({ parentSessionId: input.parentSessionId, sessionId: created.id, cwd: decision.cwd, promptLength: input.prompt.length }, "spawn_subsession started a tracked child session");
         return { sessionId: created.id, cwd: decision.cwd };
     }
+    /**
+     * Register the question set an agent wants the user to answer as the session's
+     * open ask. Deliberately does not wait for the user: `ask_user` terminates the
+     * run and the submitted answers come back later as a follow-up message.
+     *
+     * Rejected question sets throw {@link PendingAskValidationError}, which the
+     * agent loop reports to the model as an error tool result.
+     */
+    // eslint-disable-next-line @typescript-eslint/require-await -- async so a rejected question set becomes a rejection rather than a synchronous throw from a promise-returning method.
+    async openAsk(input) {
+        const result = this.pendingAskStore.open(input);
+        // A supersede closes the earlier ask, so the browsers watching it must hear
+        // that before they hear about its replacement.
+        if (result.superseded !== undefined)
+            this.publishAskClosed(input.sessionId, result.superseded);
+        this.events.publish(input.sessionId, { type: "ask.opened", ask: result.ask });
+        this.publishStatusForSessionId(input.sessionId);
+        return result;
+    }
+    /**
+     * Record the user's answers to the session's open ask and hand them to the
+     * model. The answers travel as a system-authored custom message rather than a
+     * user message, so they are not attributed to the human in the transcript;
+     * they still wake an idle session (`triggerTurn`) and queue behind in-flight
+     * work (`deliverAs: "followUp"`), which is how the run that `ask_user`
+     * terminated continues.
+     */
+    async submitAsk(ref, askId, submission) {
+        await this.assertWritable(ref);
+        const session = await this.getOrOpen(ref);
+        // Checked before the store closes the ask so a refused delivery cannot
+        // discard answers the user already submitted.
+        this.assertTreeNavigationInactive(session, "answer questions");
+        return this.closeAsk(session, this.pendingAskStore.submit(session.sessionId, askId, submission));
+    }
+    /**
+     * Close the open ask without answers. The model is still told, naming every
+     * question as unanswered: it was promised a follow-up message and would
+     * otherwise wait for one that never comes.
+     */
+    async cancelAsk(ref, askId) {
+        await this.assertWritable(ref);
+        const session = await this.getOrOpen(ref);
+        this.assertTreeNavigationInactive(session, "dismiss questions");
+        return this.closeAsk(session, this.pendingAskStore.cancel(session.sessionId, askId));
+    }
+    /**
+     * Publish and deliver a closed ask. A stale close is reported rather than
+     * thrown: losing the race against a supersede, another browser, or a session
+     * that went away is ordinary, and the returned status tells the browser what
+     * the session's open ask is now.
+     */
+    async closeAsk(session, result) {
+        if (result.status === "stale")
+            return { result: "stale", sessionStatus: this.statusFromSession(session) };
+        const { outcome } = result;
+        this.publishAskClosed(session.sessionId, outcome);
+        await this.runSessionEntryMutation(session, "deliver answers to your questions", () => session.sendCustomMessage({ customType: ASK_USER_ANSWERS_CUSTOM_TYPE, content: renderAskUserAnswersText(outcome), display: true, details: outcome }, { triggerTurn: true, deliverAs: "followUp" }));
+        this.publishStatus(session);
+        return { result: "closed", outcome, sessionStatus: this.statusFromSession(session) };
+    }
+    publishAskClosed(sessionId, outcome) {
+        this.events.publish(sessionId, { type: "ask.closed", askId: outcome.askId, reason: outcome.reason });
+    }
+    /**
+     * Void the session's open ask because the user sent a chat message instead of
+     * answering it. Every browser closes the card as cancelled, and the model is
+     * told — without being woken — so the notice rides into the turn the message
+     * itself triggers rather than becoming a turn of its own.
+     */
+    async voidOpenAskForUserMessage(session) {
+        const outcome = this.pendingAskStore.cancelOpen(session.sessionId);
+        if (outcome === undefined)
+            return;
+        this.publishAskClosed(session.sessionId, outcome);
+        await this.runSessionEntryMutation(session, "void the open questions", () => session.sendCustomMessage({ customType: ASK_USER_ANSWERS_CUSTOM_TYPE, content: renderAskUserAnswersText(outcome), display: true, details: outcome }, { triggerTurn: false, deliverAs: "followUp" }));
+        this.publishStatus(session);
+    }
+    /**
+     * Record the user's answer to an open extension dialog and resolve the
+     * extension's parked Promise with it. Unlike an ask, nothing is delivered to
+     * the model: the waiter is extension code inside an already in-flight run
+     * (or an idle handler), so no custom message and no turn are triggered.
+     */
+    async answerDialog(ref, dialogId, value) {
+        await this.assertWritable(ref);
+        const session = await this.sessionForStatusOrDialogClose(ref);
+        const result = this.pendingExtensionDialogStore.answer(session.sessionId, dialogId, value);
+        if (result.status === "stale")
+            return { result: "stale", sessionStatus: this.statusFromSession(session) };
+        const { outcome } = result;
+        this.publishDialogClosed(session.sessionId, outcome);
+        // `value` is what the store validated and recorded as the outcome's answer.
+        this.dialogWaiters.settleWithAnswer(dialogId, value);
+        this.publishStatus(session);
+        return { result: "closed", outcome, sessionStatus: this.statusFromSession(session) };
+    }
+    /** Close an open extension dialog without an answer; the extension's wait settles with its kind's cancel value. */
+    async cancelDialog(ref, dialogId) {
+        await this.assertWritable(ref);
+        const session = await this.sessionForStatusOrDialogClose(ref);
+        const result = this.pendingExtensionDialogStore.cancel(session.sessionId, dialogId, "cancelled");
+        if (result.status === "stale")
+            return { result: "stale", sessionStatus: this.statusFromSession(session) };
+        const { outcome } = result;
+        this.publishDialogClosed(session.sessionId, outcome);
+        this.dialogWaiters.settleWithCancelValue(dialogId);
+        this.publishStatus(session);
+        return { result: "closed", outcome, sessionStatus: this.statusFromSession(session) };
+    }
+    /**
+     * Implement one `ctx.ui.select()`/`confirm()`/`input()` call from extension
+     * code: open the store record, tell the browsers, and park a Promise that
+     * settles when the browser answers or cancels, the extension's own
+     * `signal`/`timeout` dismisses the dialog, the daemon default timeout
+     * elapses, or the runtime goes away. `store.open` validates the dialog, so a
+     * malformed one rejects the extension's call rather than rendering garbage.
+     * `async` so a rejected dialog becomes a rejection rather than a synchronous
+     * throw from a promise-returning method.
+     */
+    async openExtensionDialog(session, request, opts) {
+        const signal = opts?.signal;
+        // A pre-aborted signal dismisses the dialog before it ever opens.
+        if (signal?.aborted === true)
+            return extensionDialogCancelValue(request.kind);
+        const timeoutMs = effectiveExtensionDialogTimeoutMs(opts?.timeout, this.extensionDialogsTimeoutMs);
+        const dialog = this.pendingExtensionDialogStore.open({
+            sessionId: session.sessionId,
+            kind: request.kind,
+            title: request.title,
+            ...(request.message === undefined ? {} : { message: request.message }),
+            ...(request.options === undefined ? {} : { options: request.options }),
+            ...(request.placeholder === undefined ? {} : { placeholder: request.placeholder }),
+            ...(timeoutMs === undefined ? {} : { timeoutMs }),
+            runScoped: session.isStreaming,
+        });
+        this.events.publish(session.sessionId, { type: "dialog.opened", dialog });
+        this.publishStatus(session);
+        return this.dialogWaiters.park(dialog, {
+            ...(timeoutMs === undefined ? {} : { timeoutMs }),
+            ...(signal === undefined ? {} : { signal }),
+            onTrigger: (reason) => {
+                if (this.closeExtensionDialogFromTrigger(session.sessionId, dialog.dialogId, reason))
+                    this.publishStatusForSessionId(session.sessionId);
+            },
+        });
+    }
+    /**
+     * Close a dialog whose wait ended without the browser (timeout, signal
+     * abort, run end, runtime teardown) and settle its parked Promise. Returns
+     * whether this call closed the dialog; a stale close means a browser answer
+     * or an earlier trigger already settled everything.
+     */
+    closeExtensionDialogFromTrigger(sessionId, dialogId, reason) {
+        const result = this.pendingExtensionDialogStore.cancel(sessionId, dialogId, reason);
+        if (result.status !== "closed")
+            return false;
+        this.publishDialogClosed(sessionId, result.outcome);
+        this.dialogWaiters.settleWithCancelValue(dialogId);
+        return true;
+    }
+    /**
+     * Settle the session's run-scoped dialogs as `"aborted"`. Runs at
+     * abort-request time (a user abort parks the agent loop behind the dialog
+     * handler, so `agent_end` would never arrive on its own) and again from
+     * the `agent_end` observer as the run-crash backstop — the store makes the
+     * second settlement a stale no-op. Idle-opened dialogs (a `session_start`
+     * probe, say) are not run-scoped and survive, because their waiter
+     * outlives the run.
+     */
+    abortRunScopedExtensionDialogs(sessionId) {
+        let closedAny = false;
+        for (const dialog of this.pendingExtensionDialogStore.pendingDialogs(sessionId)) {
+            if (dialog.runScoped)
+                closedAny = this.closeExtensionDialogFromTrigger(sessionId, dialog.dialogId, "aborted") || closedAny;
+        }
+        if (closedAny)
+            this.publishStatusForSessionId(sessionId);
+    }
+    /**
+     * Settle every dialog of the session as `"session-ended"`: the runtime
+     * whose extension code is parked on them is being closed, replaced, or
+     * disposed, so those Promises would otherwise never settle.
+     */
+    endSessionExtensionDialogs(sessionId) {
+        let closedAny = false;
+        for (const dialog of this.pendingExtensionDialogStore.pendingDialogs(sessionId)) {
+            closedAny = this.closeExtensionDialogFromTrigger(sessionId, dialog.dialogId, "session-ended") || closedAny;
+        }
+        // Publishes only while the session is still (or already re-)registered as
+        // active, so teardown paths stay silent and runtime replacement refreshes.
+        if (closedAny)
+            this.publishStatusForSessionId(sessionId);
+    }
+    publishDialogClosed(sessionId, outcome) {
+        this.events.publish(sessionId, {
+            type: "dialog.closed",
+            dialogId: outcome.dialogId,
+            reason: outcome.reason,
+            ...(outcome.answer === undefined ? {} : { answer: outcome.answer }),
+        });
+    }
+    /**
+     * Publish status for a session known only by id, as the ask tools are: they
+     * run inside the session's own runtime, so the active entry is the session.
+     */
+    publishStatusForSessionId(sessionId) {
+        const session = this.active.get(sessionId)?.runtime.session;
+        if (session !== undefined)
+            this.publishStatus(session);
+    }
     /** Summaries of the tracked subsessions spawned by `parentSessionId`. */
     async listSubsessions(parentSessionId, parentSessionFile) {
         const parentFile = nonEmptyString(parentSessionFile);
@@ -506,7 +953,7 @@ export class PiSessionService {
             return undefined;
         return sessionFileMatches(session, link.childSessionFile) ? link : undefined;
     }
-    registerVerifiedSubsession(link) {
+    async registerVerifiedSubsession(link) {
         const { childSessionId, parentSessionId } = link;
         const previousParentId = this.subsessionParents.get(childSessionId);
         if (previousParentId !== undefined && previousParentId !== parentSessionId) {
@@ -522,6 +969,24 @@ export class PiSessionService {
         this.subsessionLinks.set(childSessionId, link);
         if (!this.subsessionNotifyArmed.has(childSessionId))
             this.subsessionNotifyArmed.set(childSessionId, false);
+        const cwd = this.cwdForVerifiedSubsession(link);
+        await this.publishUnreadMutations(this.unreadStore.excludeSession(childSessionId, cwd));
+    }
+    cwdForVerifiedSubsession(link) {
+        const activeCwd = this.activeChildForSubsessionLink(link)?.runtime.session.sessionManager.getCwd();
+        const linkedCwd = nonEmptyString(activeCwd) ?? nonEmptyString(link.cwd);
+        if (linkedCwd !== undefined)
+            return canonicalizeStoredCwd(linkedCwd);
+        const childSessionFile = link.childSessionFile;
+        if (childSessionFile !== undefined) {
+            try {
+                return canonicalizeStoredCwd(this.sessionManager.open(childSessionFile).getCwd());
+            }
+            catch (error) {
+                throw new Error("Could not resolve cwd for verified tracked sub-session", { cause: error });
+            }
+        }
+        throw new Error("Could not resolve cwd for verified tracked sub-session");
     }
     unregisterSubsession(childSessionId) {
         const parentSessionId = this.subsessionParents.get(childSessionId);
@@ -568,40 +1033,44 @@ export class PiSessionService {
         const activeParent = this.active.get(parentSessionId);
         if (activeParent !== undefined && (parentSessionFile === undefined || activeSessionFileMatches(activeParent, parentSessionFile))) {
             const activeParentFile = nonEmptyString(activeParent.runtime.session.sessionFile);
-            await this.registerPersistedSubsessionLinks(parentSessionId, activeParent.runtime.session.sessionManager, activeParentFile);
-            this.subsessionHydratedParents.add(hydrationKey);
+            const complete = await this.registerPersistedSubsessionLinks(parentSessionId, activeParent.runtime.session.sessionManager, activeParentFile);
+            if (complete)
+                this.subsessionHydratedParents.add(hydrationKey);
             return;
         }
         if (parentSessionFile === undefined)
             return;
-        if ((await readSessionHeaderSummary(parentSessionFile))?.id !== parentSessionId) {
-            this.subsessionHydratedParents.add(hydrationKey);
+        if ((await readSessionHeaderSummary(parentSessionFile))?.id !== parentSessionId)
             return;
-        }
         let parentManager;
         try {
             parentManager = this.sessionManager.open(parentSessionFile);
         }
         catch {
-            this.subsessionHydratedParents.add(hydrationKey);
             return;
         }
-        await this.registerPersistedSubsessionLinks(parentSessionId, parentManager, parentSessionFile);
-        this.subsessionHydratedParents.add(hydrationKey);
+        const complete = await this.registerPersistedSubsessionLinks(parentSessionId, parentManager, parentSessionFile);
+        if (complete)
+            this.subsessionHydratedParents.add(hydrationKey);
     }
     async registerPersistedSubsessionLinks(parentSessionId, parentManager, parentSessionFile) {
         // Parent custom links are the authoritative recovery record: verify the
-        // exact live child file/header before tracking.
+        // exact live child file/header before tracking. Do not negatively cache a
+        // scan while a candidate child is temporarily unavailable.
         const entries = parentManager.getEntries?.() ?? parentManager.getBranch();
+        let complete = true;
         for (const entry of entries) {
             const link = parsePersistedParentSubsessionLink(entry);
-            if (link === undefined)
+            if (link?.spawnedBySessionId !== parentSessionId)
                 continue;
             const verified = await this.verifiedSubsessionLinkFromParentLink(parentSessionId, parentSessionFile, link);
-            if (verified === undefined)
+            if (verified === undefined) {
+                complete = false;
                 continue;
-            this.registerVerifiedSubsession(verified);
+            }
+            await this.registerVerifiedSubsession(verified);
         }
+        return complete;
     }
     async verifiedSubsessionLinkFromParentLink(parentSessionId, parentSessionFile, link) {
         if (parentSessionFile === undefined)
@@ -620,7 +1089,7 @@ export class PiSessionService {
         const link = await this.verifiedSubsessionLinkFromOpenedChild(session);
         if (link === undefined)
             return;
-        this.registerVerifiedSubsession(link);
+        await this.registerVerifiedSubsession(link);
     }
     verifiedSubsessionLinkFromOpenedChild(session) {
         return verifiedTrackedSubsessionLink(this.sessionManager, {
@@ -729,19 +1198,31 @@ export class PiSessionService {
     async notifyParentOfSubsession(parentId, childId, text) {
         try {
             const session = await this.getOrOpenParentForSubsession(parentId, childId);
-            await session.sendCustomMessage({ customType: SUBSESSION_NOTIFICATION_CUSTOM_TYPE, content: text, display: true, details: { sessionId: childId } }, { triggerTurn: true, deliverAs: "followUp" });
-            this.publishStatus(session);
+            if (this.treeNavigations.has(session)) {
+                const pending = this.deferredSubsessionNotifications.get(session) ?? [];
+                pending.push({ parentId, childId, text });
+                this.deferredSubsessionNotifications.set(session, pending);
+                return;
+            }
+            await this.deliverSubsessionNotification(session, { parentId, childId, text });
         }
         catch (error) {
-            this.logger.info({ parentSessionId: parentId, sessionId: childId, error: error instanceof Error ? error.message : String(error) }, "failed to notify parent of subsession completion");
+            this.logSubsessionNotificationFailure(parentId, childId, error);
         }
     }
+    async deliverSubsessionNotification(session, notification) {
+        await this.runSessionEntryMutation(session, "deliver a subsession notification", () => session.sendCustomMessage({ customType: SUBSESSION_NOTIFICATION_CUSTOM_TYPE, content: notification.text, display: true, details: { sessionId: notification.childId } }, { triggerTurn: true, deliverAs: "followUp" }));
+        this.publishStatus(session);
+    }
+    logSubsessionNotificationFailure(parentId, childId, error) {
+        this.logger.info({ parentSessionId: parentId, sessionId: childId, error: error instanceof Error ? error.message : String(error) }, "failed to notify parent of subsession completion");
+    }
     async messages(ref, page) {
         const session = await this.getOrOpen(ref);
         return pageMessagesAtSafeBoundary(historyMessages(session), page);
     }
     async status(ref) {
-        return this.statusFromSession(await this.getOrOpen(ref));
+        return this.statusFromSession(await this.sessionForStatusOrDialogClose(ref));
     }
     /**
      * Join-time snapshot of the in-flight assistant stream. The `seq` watermark and
@@ -764,7 +1245,7 @@ export class PiSessionService {
     }
     async availableModels(ref) {
         const session = await this.getOrOpen(ref);
-        await session.modelRuntime.reloadConfig();
+        await session.modelRuntime.refresh();
         const models = session.scopedModels.length > 0
             ? session.scopedModels.map((scoped) => scoped.model)
             : session.modelRuntime.getAvailableSnapshot();
@@ -773,7 +1254,9 @@ export class PiSessionService {
     async setModel(ref, provider, modelId) {
         await this.assertWritable(ref);
         const session = await this.getOrOpen(ref);
-        await session.modelRuntime.reloadConfig();
+        this.assertTreeNavigationInactive(session, "change models");
+        await session.modelRuntime.refresh();
+        this.assertTreeNavigationInactive(session, "change models");
         const candidates = session.scopedModels.length > 0
             ? session.scopedModels.map((scoped) => scoped.model)
             : session.modelRuntime.getAvailableSnapshot();
@@ -781,7 +1264,7 @@ export class PiSessionService {
             ?? session.modelRuntime.getModel(provider, modelId);
         if (model === undefined)
             throw new Error(`Model not found: ${provider}/${modelId}`);
-        await session.setModel(model);
+        await this.runSessionEntryMutation(session, "change models", () => session.setModel(model));
         this.publishActivity(session, `model: ${model.id}`, "idle", model.provider);
         this.publishStatus(session);
         return this.statusFromSession(session);
@@ -789,7 +1272,7 @@ export class PiSessionService {
     async cycleModel(ref, direction) {
         await this.assertWritable(ref);
         const session = await this.getOrOpen(ref);
-        const result = await session.cycleModel(direction);
+        const result = await this.runSessionEntryMutation(session, "change models", () => session.cycleModel(direction));
         if (result === undefined)
             throw new Error(session.scopedModels.length > 0 ? "Only one model in scope" : "Only one model available");
         this.publishActivity(session, `model: ${result.model.id}`, "idle", result.model.provider);
@@ -803,6 +1286,7 @@ export class PiSessionService {
     async setThinkingLevel(ref, level) {
         await this.assertWritable(ref);
         const session = await this.getOrOpen(ref);
+        this.assertTreeNavigationInactive(session, "change the thinking level");
         // pi owns the valid set; validate against the session's live levels rather
         // than a hardcoded union so this stays correct if pi changes the set.
         const available = session.getAvailableThinkingLevels();
@@ -817,6 +1301,7 @@ export class PiSessionService {
     async cycleThinkingLevel(ref) {
         await this.assertWritable(ref);
         const session = await this.getOrOpen(ref);
+        this.assertTreeNavigationInactive(session, "change the thinking level");
         const level = session.cycleThinkingLevel();
         if (level === undefined)
             throw new Error("Current model does not support thinking");
@@ -850,6 +1335,7 @@ export class PiSessionService {
         const images = (await attachmentsToInlineImages(parsedAttachments)).map((entry) => entry.image);
         await this.assertWritable(ref);
         const session = await this.getOrOpen(ref);
+        this.assertTreeNavigationInactive(session, "send a prompt");
         this.maybeGenerateSessionName(session, promptText);
         const isQueued = session.isStreaming || session.isCompacting;
         const behavior = isQueued ? requestedBehavior ?? "followUp" : undefined;
@@ -858,6 +1344,11 @@ export class PiSessionService {
             this.publishStatus(session);
             return;
         }
+        // A chat message answers the session's open ask in the user's own words, so
+        // the form is void: keeping it open would invite answers to questions the
+        // conversation has already moved past. Ignored duplicates skip this on
+        // purpose: they must not void an ask posted after the queued original.
+        await this.voidOpenAskForUserMessage(session);
         if (session.isCompacting) {
             this.enqueuePromptDuringCompaction(session, promptText, behavior ?? "followUp", images, echoUserMessage);
             return;
@@ -869,7 +1360,7 @@ export class PiSessionService {
         if (behavior === undefined && echoUserMessage)
             this.events.publish(session.sessionId, { type: "message.append", message: userMessage(text, images) });
         const promptOptions = buildPromptOptions(behavior, images);
-        const promptPromise = session.prompt(text, promptOptions).catch((error) => {
+        const promptPromise = this.runSessionEntryMutation(session, "send a prompt", () => session.prompt(text, promptOptions)).catch((error) => {
             const message = error instanceof Error ? error.message : String(error);
             this.publishActivity(session, "error", "error", message);
             this.events.publish(session.sessionId, { type: "session.error", message });
@@ -896,6 +1387,7 @@ export class PiSessionService {
         await this.assertWritable(ref);
         const active = await this.getActive(ref);
         const { session } = active.runtime;
+        this.assertTreeNavigationInactive(session, "run a shell command");
         const isExcluded = text.startsWith("!!");
         const command = (isExcluded ? text.slice(2) : text.slice(1)).trim();
         if (!command)
@@ -904,11 +1396,11 @@ export class PiSessionService {
             throw new Error("A bash command is already running");
         this.publishActivity(session, "running bash", "active", command);
         this.events.publish(session.sessionId, { type: "shell.start", command, excludeFromContext: isExcluded });
-        void session.executeBash(command, (chunk) => {
+        void this.runSessionEntryMutation(session, "run a shell command", () => session.executeBash(command, (chunk) => {
             this.events.publish(session.sessionId, { type: "shell.chunk", chunk });
             this.publishActivity(session, "running bash", "active", command);
             this.publishStatus(session);
-        }, { excludeFromContext: isExcluded }).then((result) => {
+        }, { excludeFromContext: isExcluded })).then((result) => {
             this.events.publish(session.sessionId, {
                 type: "shell.end",
                 output: result.output,
@@ -937,30 +1429,103 @@ export class PiSessionService {
         const active = await this.getActive(ref);
         return this.commandService.respond(active.runtime.session.sessionId, requestId, value);
     }
-    async reloadSessionRuntime(session) {
+    async navigateTree(ref, request) {
+        if (request.targetId.trim() === "")
+            throw new Error("Session tree target is required");
+        if (this.isTreeExclusiveSessionIdentityActive(sessionIdFromLookup(ref))) {
+            throw new Error("Stop current session activity before navigating the session tree");
+        }
+        await this.assertWritable(ref);
+        const options = sessionTreeNavigationOptions(request);
+        const session = await this.getOrOpen(ref);
+        if (typeof session.navigateTree !== "function")
+            throw new Error("Session tree navigation is not supported by this Pi runtime");
         if (this.hasActiveWork(session))
-            throw new Error("Stop current session activity before reloading");
-        this.publishActivity(session, "reloading resources", "active");
+            throw new Error("Stop current session activity before navigating the session tree");
+        // Acquire synchronously after the active-work check. No leaf-producing work
+        // may enter this runtime until Pi's potentially asynchronous summary settles.
+        this.treeNavigations.add(session);
         try {
-            await session.reload();
-            this.publishActivity(session, "resources reloaded", "idle");
+            if (session.sessionManager.getLeafId() !== request.expectedLeafId) {
+                throw new Error("The session changed since /tree was opened. Reopen /tree and try again.");
+            }
+            this.publishActivity(session, options.summarize ? "summarizing branch" : "navigating session tree", "active");
             this.publishStatus(session);
+            const result = await session.navigateTree(request.targetId, options);
+            if (result.cancelled) {
+                if (this.isCurrentActiveSession(session)) {
+                    this.publishActivity(session, result.aborted === true ? "branch summary aborted" : "tree navigation cancelled", "idle");
+                }
+                return { cancelled: true, ...(result.aborted === undefined ? {} : { aborted: result.aborted }) };
+            }
+            if (this.isCurrentActiveSession(session))
+                this.publishActivity(session, "session tree navigated", "idle");
+            return { cancelled: false, ...(result.editorText === undefined ? {} : { editorText: result.editorText }) };
         }
         catch (error) {
             const message = error instanceof Error ? error.message : String(error);
-            this.publishActivity(session, "reload failed", "error", message);
-            this.events.publish(session.sessionId, { type: "session.error", message });
-            this.publishStatus(session);
+            if (this.isCurrentActiveSession(session)) {
+                this.publishActivity(session, "tree navigation failed", "error", message);
+                this.events.publish(session.sessionId, { type: "session.error", message });
+            }
             throw error;
         }
+        finally {
+            this.treeNavigations.delete(session);
+            if (this.isCurrentActiveSession(session)) {
+                this.flushDeferredTreeNavigationWork(session);
+                this.publishStatus(session);
+            }
+            else {
+                this.deferredGeneratedSessionNames.delete(session);
+                this.deferredSubsessionNotifications.delete(session);
+            }
+        }
+    }
+    async reloadSessionRuntime(session) {
+        if (this.hasActiveWork(session))
+            throw new Error("Stop current session activity before reloading");
+        await this.runTreeExclusiveOperation([{ sessionId: session.sessionId, session }], "Stop current session activity before reloading", async () => {
+            this.publishActivity(session, "reloading resources", "active");
+            const priorGeneration = this.notificationGenerationBySession.get(session);
+            let candidateGeneration;
+            try {
+                await session.reload(priorGeneration === undefined ? undefined : {
+                    beforeSessionStart: () => {
+                        candidateGeneration = this.notificationStore.beginReplacement(priorGeneration, notificationIdentityForSession(session));
+                        this.notificationGenerationBySession.set(session, candidateGeneration);
+                        this.replaceSessionNotificationContext(session, candidateGeneration);
+                    },
+                });
+                if (candidateGeneration !== undefined) {
+                    this.publishNotificationMutations(this.notificationStore.commitReplacement(candidateGeneration));
+                }
+                this.publishActivity(session, "resources reloaded", "idle");
+                this.publishStatus(session);
+            }
+            catch (error) {
+                if (candidateGeneration !== undefined) {
+                    this.publishNotificationMutations(this.notificationStore.abortReplacement(candidateGeneration, "candidate"));
+                    this.notificationGenerationBySession.set(session, candidateGeneration);
+                }
+                const message = error instanceof Error ? error.message : String(error);
+                this.publishActivity(session, "reload failed", "error", message);
+                this.events.publish(session.sessionId, { type: "session.error", message });
+                this.publishStatus(session);
+                throw error;
+            }
+        });
     }
     async archive(ref) {
         const session = await this.getOrOpen(ref);
         if (this.hasActiveWork(session))
             throw new Error("Stop current session activity before archiving");
-        const archiveInput = await this.archiveInputForSession(session);
-        await this.closeActive(session.sessionId);
-        await this.archiveStore.archive(archiveInput);
+        await this.runTreeExclusiveOperation([{ sessionId: session.sessionId, session }], "Stop current session activity before archiving", async () => {
+            const archiveInput = await this.archiveInputForSession(session);
+            await this.closeActive(session.sessionId, { kind: "clear", reason: "archive" });
+            await this.archiveStore.archive(archiveInput);
+            await this.forgetUnreadSessions([archiveInput]);
+        });
     }
     async archiveMany(refs) {
         const uniqueRefs = uniqueBulkSessionRefs(refs);
@@ -970,11 +1535,14 @@ export class PiSessionService {
         ]);
         const failures = [];
         const alreadyArchivedSessionIds = [];
+        const unreadArchivedIdentities = [];
         const planItems = [];
         for (const ref of uniqueRefs) {
             const archived = findArchivedRecordForBulkRef(archivedRecords, ref);
             if (archived !== undefined) {
+                this.publishNotificationMutations(this.notificationStore.clearSession(archived.sessionId, "archive"));
                 alreadyArchivedSessionIds.push(archived.sessionId);
+                unreadArchivedIdentities.push(archived);
                 continue;
             }
             const active = this.activeForLookup(bulkRefToLookup(ref));
@@ -999,25 +1567,41 @@ export class PiSessionService {
                 failures.push({ sessionId: resolvedSessionId, error: errorMessage(error) });
             }
         }
-        const readyInputs = [];
+        const readyPlanItems = [];
         for (const item of planItems) {
+            const active = this.activeForLookup({ id: item.input.sessionId, cwd: item.input.cwd });
+            if (active !== undefined && this.hasActiveWork(active.runtime.session)) {
+                failures.push({ sessionId: item.input.sessionId, error: "Stop current session activity before archiving" });
+                continue;
+            }
+            readyPlanItems.push(active === undefined ? item : { ...item, active });
+        }
+        const readyInputs = [];
+        const archivedSessionIds = [...alreadyArchivedSessionIds];
+        await this.runTreeExclusiveOperation(readyPlanItems.map(({ input, active }) => ({
+            sessionId: input.sessionId,
+            ...(active === undefined ? {} : { session: active.runtime.session, runtime: active.runtime }),
+        })), "Stop current session activity before archiving", async () => {
+            for (const item of readyPlanItems) {
+                try {
+                    await this.closeActive(item.input.sessionId, { kind: "clear", reason: "archive" });
+                    readyInputs.push(item.input);
+                }
+                catch (error) {
+                    failures.push({ sessionId: item.input.sessionId, error: errorMessage(error) });
+                }
+            }
             try {
-                await this.closeActive(item.input.sessionId);
-                readyInputs.push(item.input);
+                const archived = await this.archiveStoreArchiveMany(readyInputs);
+                archivedSessionIds.push(...archived.map((record) => record.sessionId));
+                unreadArchivedIdentities.push(...archived);
             }
             catch (error) {
-                failures.push({ sessionId: item.input.sessionId, error: errorMessage(error) });
+                for (const input of readyInputs)
+                    failures.push({ sessionId: input.sessionId, error: errorMessage(error) });
             }
-        }
-        const archivedSessionIds = [...alreadyArchivedSessionIds];
-        try {
-            const archived = await this.archiveStoreArchiveMany(readyInputs);
-            archivedSessionIds.push(...archived.map((record) => record.sessionId));
-        }
-        catch (error) {
-            for (const input of readyInputs)
-                failures.push({ sessionId: input.sessionId, error: errorMessage(error) });
-        }
+        });
+        await this.forgetUnreadSessions(unreadArchivedIdentities);
         return {
             archived: true,
             archivedSessionIds: uniqueStrings(archivedSessionIds),
@@ -1034,9 +1618,19 @@ export class PiSessionService {
         if (busy !== undefined)
             throw new Error(`Stop current session activity before archiving ${sessionDisplayName(busy)}`);
         const archiveInputs = plan.unarchivedTargets.map((target) => archiveInputFromCandidate(target));
-        for (const input of archiveInputs)
-            await this.closeActive(input.sessionId);
-        await this.archiveStoreArchiveMany(archiveInputs);
+        await this.runTreeExclusiveOperation(plan.unarchivedTargets.map((target) => ({
+            sessionId: target.id,
+            ...(target.activeSession === undefined ? {} : { session: target.activeSession }),
+        })), `Stop current session activity before archiving ${sessionDisplayName(session)}`, async () => {
+            for (const target of plan.targets) {
+                if (target.archived)
+                    this.publishNotificationMutations(this.notificationStore.clearSession(target.id, "archive"));
+            }
+            for (const input of archiveInputs)
+                await this.closeActive(input.sessionId, { kind: "clear", reason: "archive" });
+            await this.archiveStoreArchiveMany(archiveInputs);
+        });
+        await this.forgetUnreadSessions(plan.targets.map((target) => ({ sessionId: target.id, cwd: target.cwd })));
         return {
             archived: true,
             sessionIds: archiveInputs.map((input) => input.sessionId),
@@ -1048,8 +1642,9 @@ export class PiSessionService {
         const archived = await this.getArchived(ref);
         if (archived === undefined)
             throw new Error("Session not found");
-        await this.closeActive(archived.sessionId);
+        await this.closeActive(archived.sessionId, { kind: "clear", reason: "restore" });
         await this.archiveStore.restore(archived.sessionId);
+        await this.forgetUnreadSessions([archived]);
     }
     async deleteArchived(ref) {
         const record = await this.getArchived(ref);
@@ -1057,10 +1652,11 @@ export class PiSessionService {
             throw new Error("Archived session not found");
         if (this.archiveStore.deleteArchived === undefined)
             throw new Error("Archive store does not support deletion");
-        await this.closeActive(record.sessionId);
+        await this.closeActive(record.sessionId, { kind: "clear", reason: "delete" });
         if (record.archivePath === undefined)
             await this.ensureArchivedRecordMoved(record);
         await this.archiveStore.deleteArchived(record.sessionId);
+        await this.forgetUnreadSessions([record]);
     }
     async deleteArchivedMany(refs) {
         if (this.archiveStore.deleteArchived === undefined && this.archiveStore.deleteArchivedMany === undefined)
@@ -1085,7 +1681,7 @@ export class PiSessionService {
         const readyRecords = [];
         for (const item of planItems) {
             try {
-                await this.closeActive(item.record.sessionId);
+                await this.closeActive(item.record.sessionId, { kind: "clear", reason: "delete" });
                 readyRecords.push(item.record);
             }
             catch (error) {
@@ -1106,6 +1702,8 @@ export class PiSessionService {
             for (const sessionId of deleteIds)
                 failures.push({ sessionId, error: errorMessage(error) });
         }
+        const deletedIdSet = new Set(deletedSessionIds);
+        await this.forgetUnreadSessions(readyRecords.filter((record) => deletedIdSet.has(record.sessionId)));
         return {
             deleted: true,
             deletedSessionIds,
@@ -1118,9 +1716,29 @@ export class PiSessionService {
         const session = await this.getOrOpen(ref);
         if (this.hasActiveWork(session))
             throw new Error("Stop current session activity before reloading");
-        await this.closeActive(session.sessionId);
-        const reopened = await this.getActive(ref);
-        this.publishStatus(reopened.runtime.session);
+        const reopenedSession = await this.runTreeExclusiveOperation([{ sessionId: session.sessionId, session }], "Stop current session activity before reloading", async () => {
+            const priorGeneration = this.notificationGenerationBySession.get(session);
+            const { sessionId, cwd } = notificationIdentityForSession(session);
+            let candidateGeneration;
+            try {
+                await this.closeActive(sessionId, priorGeneration === undefined ? CLEAR_RUNTIME_NOTIFICATIONS : DEFER_RUNTIME_NOTIFICATIONS);
+                candidateGeneration = priorGeneration === undefined
+                    ? undefined
+                    : this.notificationStore.beginReplacement(priorGeneration, { sessionId, cwd });
+                const reopened = await this.getActive(ref, candidateGeneration === undefined ? {} : { notificationGeneration: candidateGeneration });
+                if (candidateGeneration !== undefined) {
+                    this.publishNotificationMutations(this.notificationStore.commitReplacement(candidateGeneration));
+                }
+                return reopened.runtime.session;
+            }
+            catch (error) {
+                if (candidateGeneration !== undefined) {
+                    this.publishNotificationMutations(this.notificationStore.abortReplacement(candidateGeneration));
+                }
+                throw error;
+            }
+        });
+        this.publishStatus(reopenedSession);
     }
     async detachParent(ref) {
         const session = await this.getOrOpen(ref);
@@ -1130,6 +1748,7 @@ export class PiSessionService {
         await clearParentSession(sessionFile);
         clearParentSessionHeader(session.sessionManager);
         this.unregisterSubsession(session.sessionId);
+        await this.forgetUnreadSessions([{ sessionId: session.sessionId, cwd: session.sessionManager.getCwd() }]);
     }
     async clearQueue(ref) {
         await this.assertWritable(ref);
@@ -1152,17 +1771,36 @@ export class PiSessionService {
         const sessionId = active.runtime.session.sessionId;
         this.clearCompactionPromptQueue(sessionId);
         clearSessionQueue(active.runtime.session);
-        await active.runtime.session.abort();
-        this.publishActivity(active.runtime.session, "stopped", "idle");
-        this.publishStatus(active.runtime.session);
+        // Settle run-scoped dialogs now, at abort-request time: pi's agent loop
+        // waits for a parked `tool_call` dialog handler before it can emit
+        // `agent_end`, so leaving settlement to the `agent_end` observer would
+        // strand the dialog until its timeout. Settling before the runtime abort
+        // also means a failing or hung abort cannot strand the parked waiter.
+        this.abortRunScopedExtensionDialogs(sessionId);
+        try {
+            await this.abortSessionOperations(active.runtime.session);
+            this.publishActivity(active.runtime.session, "stopped", "idle");
+        }
+        catch (error) {
+            const message = error instanceof Error ? error.message : String(error);
+            this.publishActivity(active.runtime.session, "stop failed", "error", message);
+            throw error;
+        }
+        finally {
+            this.publishStatus(active.runtime.session);
+        }
     }
-    stop(ref) {
+    async stop(ref) {
         const active = this.activeForLookup(ref);
-        if (active === undefined)
+        if (active !== undefined) {
+            await this.closeActive(active.runtime.session.sessionId);
             return;
-        void this.closeActive(active.runtime.session.sessionId).catch(() => {
-            // Best-effort shutdown; callers that need errors await closeActive directly.
-        });
+        }
+        if (isPiSessionRef(ref)) {
+            this.publishNotificationMutations(this.notificationStore.clearSessionIdentity(ref.id, canonicalizeStoredCwd(ref.cwd), "runtime-close"));
+            return;
+        }
+        await this.closeActive(ref);
     }
     async bulkSessionLookupContext(refs) {
         const cwdSet = new Set();
@@ -1351,13 +1989,32 @@ export class PiSessionService {
         }
         return [...names];
     }
-    async closeActive(sessionId) {
+    async closeActive(sessionId, notificationPolicy = CLEAR_RUNTIME_NOTIFICATIONS) {
+        // A session whose open is parked on a `session_start` dialog holds its
+        // pending open until the dialog settles; settle it first so closing cannot
+        // block behind the dialog timeout (which `0` makes infinite).
+        if (this.startupSessions.has(sessionId))
+            this.endSessionExtensionDialogs(sessionId);
         const pendingOpens = this.pendingSessionOpenPromises(sessionId);
         if (pendingOpens.length > 0)
             await Promise.allSettled(pendingOpens);
         const active = this.active.get(sessionId);
+        if (notificationPolicy.kind === "clear") {
+            const generation = active === undefined ? undefined : this.notificationGenerationBySession.get(active.runtime.session);
+            const mutations = generation === undefined
+                ? this.notificationStore.clearSession(sessionId, notificationPolicy.reason)
+                : this.notificationStore.clearGeneration(generation, notificationPolicy.reason);
+            this.publishNotificationMutations(mutations);
+        }
         if (!active)
             return;
+        this.forgetUnreadActivity(active.runtime.session);
+        // An open ask is meaningful only while the runtime that posted it exists: no
+        // one is left to receive the answers, so it is dropped without an outcome.
+        this.pendingAskStore.forgetSession(sessionId);
+        // Open dialogs share that stance, but their extension waiters are parked
+        // Promises inside the dying runtime: settle them rather than dropping them.
+        this.endSessionExtensionDialogs(sessionId);
         this.active.delete(sessionId);
         this.activities.delete(sessionId);
         this.workspaceActivity?.removeSession(sessionId, active.runtime.session.sessionManager.getCwd());
@@ -1370,13 +2027,36 @@ export class PiSessionService {
             this.subsessionNotifyArmed.delete(sessionId);
         clearSessionQueue(active.runtime.session);
         active.unsubscribe();
+        active.runtime.setRebindSession(undefined);
         try {
-            await active.runtime.session.abort();
+            await this.abortSessionOperations(active.runtime.session);
         }
         finally {
             await active.runtime.dispose();
         }
     }
+    async abortSessionOperations(session) {
+        let branchSummaryAbortFailed = false;
+        let branchSummaryAbortError;
+        try {
+            session.abortBranchSummary?.();
+        }
+        catch (error) {
+            branchSummaryAbortFailed = true;
+            branchSummaryAbortError = error;
+        }
+        try {
+            await session.abort();
+        }
+        catch (abortError) {
+            if (branchSummaryAbortFailed) {
+                throw new AggregateError([branchSummaryAbortError, abortError], "Failed to abort session operations", { cause: abortError });
+            }
+            throw abortError;
+        }
+        if (branchSummaryAbortFailed)
+            throw branchSummaryAbortError;
+    }
     async assertWritable(ref) {
         if (await this.getArchived(ref) !== undefined)
             throw new Error("Archived sessions are read-only. Restore the session to continue.");
@@ -1384,23 +2064,23 @@ export class PiSessionService {
     async getOrOpen(ref) {
         return (await this.getActive(ref)).runtime.session;
     }
-    async getActive(ref) {
+    async getActive(ref, options = {}) {
         const active = this.activeForLookup(ref);
         if (active !== undefined)
             return active;
         const archived = await this.getArchived(ref);
         if (archived?.archivePath !== undefined) {
             const { archivePath } = archived;
-            return this.openExistingSession(archived.sessionId, archived.cwd, () => this.sessionManager.open(archivePath));
+            return this.openExistingSession(archived.sessionId, archived.cwd, () => this.sessionManager.open(archivePath), { notifications: "disabled" });
         }
         const match = isPiSessionRef(ref)
             ? (await this.sessionManager.list(ref.cwd)).find((s) => s.id === ref.id || s.id.startsWith(ref.id))
             : (await this.sessionManager.listAll?.() ?? []).find((s) => s.id === ref || s.id.startsWith(ref));
         if (!match)
             throw new Error("Session not found");
-        return this.openExistingSession(match.id, match.cwd, () => this.sessionManager.open(match.path));
+        return this.openExistingSession(match.id, match.cwd, () => this.sessionManager.open(match.path), options);
     }
-    openExistingSession(sessionId, cwd, openSessionManager) {
+    openExistingSession(sessionId, cwd, openSessionManager, options = {}) {
         const active = this.activeForLookup({ id: sessionId, cwd });
         if (active !== undefined)
             return Promise.resolve(active);
@@ -1410,7 +2090,7 @@ export class PiSessionService {
             return existing.promise;
         const pending = {
             sessionId,
-            promise: this.create(openSessionManager(), cwd),
+            promise: this.create(openSessionManager(), cwd, options),
         };
         pending.promise = pending.promise.finally(() => {
             if (this.pendingSessionOpens.get(key) === pending)
@@ -1432,6 +2112,9 @@ export class PiSessionService {
             return undefined;
         return archived;
     }
+    isCurrentActiveSession(session) {
+        return this.active.get(session.sessionId)?.runtime.session === session;
+    }
     activeForLookup(ref) {
         const sessionId = sessionIdFromLookup(ref);
         const exact = this.active.get(sessionId);
@@ -1443,7 +2126,47 @@ export class PiSessionService {
         }
         return undefined;
     }
+    startupSessionForLookup(ref) {
+        const sessionId = sessionIdFromLookup(ref);
+        const exact = this.startupSessions.get(sessionId);
+        if (exact !== undefined && lookupMatchesStartupSession(ref, exact))
+            return exact;
+        for (const [candidateId, session] of this.startupSessions.entries()) {
+            if (candidateId.startsWith(sessionId) && lookupMatchesStartupSession(ref, session))
+                return session;
+        }
+        return undefined;
+    }
+    /**
+     * The session to serve a read-only status or a dialog close for, while it
+     * can still be found: active first, then still starting up, and only then
+     * the on-demand open path (which a stale close on an idle session needs for
+     * its status projection).
+     */
+    async sessionForStatusOrDialogClose(ref) {
+        const reachable = this.activeForLookup(ref)?.runtime.session ?? this.startupSessionForLookup(ref);
+        if (reachable !== undefined)
+            return reachable;
+        return this.getOrOpen(ref);
+    }
+    /**
+     * Construct a session while telling waiting browsers which phase of startup
+     * they are waiting on. The reporting wraps the *whole* construction rather
+     * than the inner bookkeeping `try`, because the runtime construction that runs
+     * first is both the slowest phase and one that can fail on its own; a clear
+     * that only ran for the later phases would leave a stale label behind.
+     */
     async create(sessionManager, cwd, options = {}) {
+        const startup = this.startupProgress(sessionManager, options.startupIntent ?? "open", options.startupToken);
+        try {
+            return await this.createSessionRuntime(sessionManager, cwd, options, startup);
+        }
+        finally {
+            startup.end();
+        }
+    }
+    async createSessionRuntime(sessionManager, cwd, options, startup) {
+        startup.report(STARTUP_PHASE_RUNTIME);
         const delegationToolsEnabled = options.creationProvenance !== "tracked-subsession"
             && await sessionAllowsDelegationTools(sessionManager, this.sessionManager);
         const runtime = await this.createAgentRuntime(this.createRuntime, {
@@ -1454,21 +2177,91 @@ export class PiSessionService {
             ...(options.initialModel === undefined ? {} : { initialModel: options.initialModel }),
         });
         const active = { runtime, unsubscribe: noop };
+        let boundSession = runtime.session;
+        let notificationGeneration = options.notificationGeneration;
+        let notificationOwnership = options.notifications === "disabled"
+            ? "disabled"
+            : notificationGeneration === undefined
+                ? "registered"
+                : "external";
+        if (notificationOwnership === "registered") {
+            const notificationIdentity = notificationIdentityForSession(runtime.session);
+            const existingCandidate = this.notificationStore.beginReplacementForSession(notificationIdentity.sessionId, notificationIdentity.cwd);
+            if (existingCandidate !== undefined) {
+                notificationGeneration = existingCandidate;
+                notificationOwnership = "replacement";
+            }
+            else {
+                const registration = this.notificationStore.registerSession(notificationIdentity.sessionId, notificationIdentity.cwd);
+                notificationGeneration = registration.generation;
+                this.publishNotificationMutations(registration.mutations);
+            }
+        }
+        if (notificationGeneration !== undefined)
+            this.notificationGenerationBySession.set(runtime.session, notificationGeneration);
         try {
-            await this.bindSessionExtensions(runtime.session);
+            if (options.creationProvenance === "tracked-subsession") {
+                await this.publishUnreadMutations(this.unreadStore.excludeSession(runtime.session.sessionId, canonicalizeStoredCwd(runtime.session.sessionManager.getCwd())));
+            }
+            else {
+                await this.recoverSubsessionTrackingForOpenedSession(runtime.session);
+            }
+            startup.report(STARTUP_PHASE_EXTENSIONS);
+            await this.bindSessionExtensions(runtime.session, notificationGeneration);
             this.bindRuntime(active);
             runtime.setRebindSession(async (session) => {
-                await this.bindSessionExtensions(session);
-                this.bindRuntime(active);
-                await this.recoverSubsessionTrackingForOpenedSession(session);
+                const priorGeneration = notificationGeneration;
+                let candidateGeneration;
+                try {
+                    await this.prepareUnreadRuntimeRebind(boundSession, session);
+                    await this.recoverSubsessionTrackingForOpenedSession(session);
+                    if (priorGeneration !== undefined) {
+                        candidateGeneration = this.notificationStore.beginReplacement(priorGeneration, notificationIdentityForSession(session));
+                        this.notificationGenerationBySession.set(session, candidateGeneration);
+                    }
+                    this.bindRuntime(active, session);
+                    // The runtime being replaced parked every dialog the store still
+                    // holds for this session; settle those waits before the new
+                    // runtime's extensions can open fresh dialogs under the same id.
+                    this.endSessionExtensionDialogs(boundSession.sessionId);
+                    boundSession = session;
+                    await this.bindSessionExtensions(session, candidateGeneration);
+                    if (candidateGeneration !== undefined) {
+                        this.publishNotificationMutations(this.notificationStore.commitReplacement(candidateGeneration));
+                        notificationGeneration = candidateGeneration;
+                    }
+                }
+                catch (error) {
+                    if (candidateGeneration !== undefined) {
+                        this.publishNotificationMutations(this.notificationStore.abortReplacement(candidateGeneration, "candidate"));
+                        notificationGeneration = candidateGeneration;
+                        this.notificationGenerationBySession.set(session, candidateGeneration);
+                    }
+                    throw error;
+                }
             });
             this.active.set(runtime.session.sessionId, active);
-            await this.recoverSubsessionTrackingForOpenedSession(runtime.session);
+            if (notificationOwnership === "replacement" && notificationGeneration !== undefined) {
+                this.publishNotificationMutations(this.notificationStore.commitReplacement(notificationGeneration));
+                notificationOwnership = "external";
+            }
             this.publishStatus(runtime.session);
             return active;
         }
         catch (error) {
+            if (notificationGeneration !== undefined) {
+                if (notificationOwnership === "registered") {
+                    this.publishNotificationMutations(this.notificationStore.clearSession(runtime.session.sessionId, "initialization-failed"));
+                }
+                else if (notificationOwnership === "replacement") {
+                    this.publishNotificationMutations(this.notificationStore.abortReplacement(notificationGeneration));
+                }
+            }
             active.unsubscribe();
+            this.forgetUnreadActivity(boundSession);
+            // A session_start dialog may already be parked when a later startup
+            // step fails; its waiter dies with the runtime being torn down here.
+            this.endSessionExtensionDialogs(boundSession.sessionId);
             let removedActive = false;
             for (const [sessionId, candidate] of this.active.entries()) {
                 if (candidate !== active)
@@ -1491,39 +2284,190 @@ export class PiSessionService {
             throw error;
         }
     }
-    async bindSessionExtensions(session) {
+    async bindSessionExtensions(session, generation) {
+        const uiContext = this.sessionUiContext(session, generation);
+        // A `session_start` hook can park this bind on a dialog the browser has
+        // not answered yet. On the initial create/open path the session becomes
+        // active only after this returns, so register it for the duration: the
+        // answer that unblocks startup has to be reachable while it waits.
+        this.startupSessions.set(session.sessionId, session);
+        try {
+            await session.bindExtensions({
+                uiContext,
+                mode: "rpc",
+                onError: (error) => {
+                    const message = `${error.extensionPath}: ${error.error}`;
+                    this.publishActivity(session, "extension error", "error", message);
+                    this.events.publish(session.sessionId, { type: "session.error", message });
+                },
+            });
+        }
+        finally {
+            this.startupSessions.delete(session.sessionId);
+        }
+    }
+    replaceSessionNotificationContext(session, generation) {
+        session.extensionRunner.setUIContext(this.sessionUiContext(session, generation), "rpc");
+    }
+    sessionUiContext(session, generation) {
         const baseUiContext = session.extensionRunner.getUIContext();
         const notify = (message, type) => {
+            if (generation === undefined) {
+                this.events.publish(session.sessionId, {
+                    type: "command.output",
+                    level: type === "error" ? "error" : "info",
+                    message,
+                });
+                return;
+            }
+            const added = this.notificationStore.addNotification(generation, message, type);
+            this.publishNotificationMutations(added.mutations);
+            if (added.notification === undefined)
+                return;
             this.events.publish(session.sessionId, {
                 type: "command.output",
                 level: type === "error" ? "error" : "info",
                 message,
+                notificationId: added.notification.id,
             });
         };
-        // PI WEB is a remote UI host, but currently only extension notifications
-        // cross this boundary. Delegate every other UI method to Pi's headless
-        // defaults so unsupported dialogs cancel safely instead of hanging.
-        const uiContext = new Proxy(baseUiContext, {
-            get(target, property, receiver) {
+        // PI WEB owns the browser-facing dialog, notification, and text-formatting
+        // boundaries: the three dialog primitives park daemon-held Promises that
+        // the browser answers, while every other UI method delegates to Pi's
+        // headless defaults so unsupported surfaces cancel safely instead of
+        // hanging.
+        return new Proxy(baseUiContext, {
+            get: (target, property, receiver) => {
                 if (property === "notify")
                     return notify;
+                if (property === "theme")
+                    return plainTextTheme;
+                if (property === "confirm") {
+                    return (title, message, opts) => this.openExtensionDialog(session, { kind: "confirm", title, message }, opts);
+                }
+                if (property === "select") {
+                    return (title, options, opts) => this.openExtensionDialog(session, { kind: "select", title, options }, opts);
+                }
+                if (property === "input") {
+                    return (title, placeholder, opts) => this.openExtensionDialog(session, { kind: "input", title, placeholder }, opts);
+                }
                 const value = Reflect.get(target, property, receiver);
                 return value;
             },
         });
-        await session.bindExtensions({
-            uiContext,
-            mode: "rpc",
-            onError: (error) => {
-                const message = `${error.extensionPath}: ${error.error}`;
-                this.publishActivity(session, "extension error", "error", message);
-                this.events.publish(session.sessionId, { type: "session.error", message });
-            },
+    }
+    publishNotificationMutations(mutations) {
+        for (const mutation of mutations) {
+            this.events.publish(mutation.sessionId, mutation.inboxEvent);
+            this.events.publishNotificationSummary(mutation.summaryEvent);
+        }
+    }
+    async prepareUnreadRuntimeRebind(previous, next) {
+        const previousCwd = canonicalizeStoredCwd(previous.sessionManager.getCwd());
+        this.unreadStore.forgetActivity(previous.sessionId, previousCwd);
+        const nextCwd = canonicalizeStoredCwd(next.sessionManager.getCwd());
+        if (previous.sessionId === next.sessionId && cwdPathsEqual(previousCwd, nextCwd))
+            return;
+        await this.publishUnreadMutations(this.unreadStore.forgetSession(previous.sessionId, previousCwd));
+    }
+    forgetUnreadActivity(session) {
+        this.unreadStore.forgetActivity(session.sessionId, canonicalizeStoredCwd(session.sessionManager.getCwd()));
+    }
+    async forgetUnreadSessions(identities) {
+        const mutations = [];
+        for (const identity of identities) {
+            mutations.push(...this.unreadStore.forgetSession(identity.sessionId, canonicalizeStoredCwd(identity.cwd)));
+        }
+        await this.publishUnreadMutations(mutations);
+    }
+    observeUnreadActivityState(session) {
+        const mutations = this.unreadStore.observeActivityState(session.sessionId, canonicalizeStoredCwd(session.sessionManager.getCwd()), this.hasActiveWork(session));
+        if (mutations.length === 0)
+            return;
+        void this.publishUnreadMutations(mutations).catch(() => undefined);
+    }
+    publishUnreadMutations(mutations) {
+        this.enqueueUnreadMutations(mutations);
+        this.unreadPublicationFlushRequested = true;
+        if (this.unreadPublication === undefined && this.unreadPublicationRetryTimer !== undefined) {
+            const failure = this.unreadPublicationFailure;
+            return Promise.reject(failure instanceof Error
+                ? failure
+                : new Error("Session unread publication is awaiting retry", { cause: failure }));
+        }
+        return this.ensureUnreadPublication();
+    }
+    ensureUnreadPublication() {
+        const existing = this.unreadPublication;
+        if (existing !== undefined)
+            return existing;
+        const publication = this.drainUnreadPublication();
+        this.unreadPublication = publication;
+        void publication.then(() => {
+            if (this.unreadPublication === publication)
+                this.unreadPublication = undefined;
+        }, (error) => {
+            if (this.unreadPublication === publication)
+                this.unreadPublication = undefined;
+            this.unreadPublicationFailure = error;
+            this.logger.info({ error: error instanceof Error ? error.message : String(error) }, "failed to publish durable session unread mutations");
+            this.scheduleUnreadPublicationRetry();
         });
+        return publication;
+    }
+    async drainUnreadPublication() {
+        while (this.unreadPublicationFlushRequested || this.pendingUnreadMutations.length > 0) {
+            this.unreadPublicationFlushRequested = false;
+            const batch = this.pendingUnreadMutations.splice(0);
+            let publishedCount = 0;
+            try {
+                await this.unreadStore.flush();
+                for (const mutation of batch) {
+                    this.events.publishGlobal(mutation.event);
+                    publishedCount += 1;
+                }
+            }
+            catch (error) {
+                this.prependUnreadMutations(batch.slice(publishedCount));
+                this.unreadPublicationFlushRequested = true;
+                throw error;
+            }
+            this.unreadPublicationFailure = undefined;
+            this.clearUnreadPublicationRetry();
+        }
+    }
+    enqueueUnreadMutations(mutations) {
+        this.pendingUnreadMutations.push(...mutations);
+        this.trimPendingUnreadMutations();
     }
-    bindRuntime(active) {
+    prependUnreadMutations(mutations) {
+        this.pendingUnreadMutations.unshift(...mutations);
+        this.trimPendingUnreadMutations();
+    }
+    trimPendingUnreadMutations() {
+        const excess = this.pendingUnreadMutations.length - MAX_PENDING_UNREAD_MUTATIONS;
+        if (excess > 0)
+            this.pendingUnreadMutations.splice(0, excess);
+    }
+    scheduleUnreadPublicationRetry() {
+        if (this.unreadPublicationStopped || this.unreadPublicationRetryTimer !== undefined)
+            return;
+        const delay = this.unreadPublicationRetryDelayMs;
+        this.unreadPublicationRetryDelayMs = Math.min(Math.max(delay * 2, this.unreadPublicationRetryInitialMs), Math.max(MAX_UNREAD_PUBLICATION_RETRY_MS, this.unreadPublicationRetryInitialMs));
+        this.unreadPublicationRetryTimer = setTimeout(() => {
+            this.unreadPublicationRetryTimer = undefined;
+            void this.ensureUnreadPublication().catch(() => undefined);
+        }, delay);
+        this.unreadPublicationRetryTimer.unref();
+    }
+    clearUnreadPublicationRetry() {
+        if (this.unreadPublicationRetryTimer !== undefined)
+            clearTimeout(this.unreadPublicationRetryTimer);
+        this.unreadPublicationRetryTimer = undefined;
+        this.unreadPublicationRetryDelayMs = this.unreadPublicationRetryInitialMs;
+    }
+    bindRuntime(active, session = active.runtime.session) {
         active.unsubscribe();
-        const { session } = active.runtime;
         for (const [sessionId, candidate] of this.active.entries()) {
             if (candidate === active) {
                 this.active.delete(sessionId);
@@ -1535,6 +2479,8 @@ export class PiSessionService {
             this.events.publish(session.sessionId, toClientEvent(event));
             this.publishActivityForEvent(session, event);
             const eventType = getString(event, "type");
+            if (eventType === "agent_end")
+                this.abortRunScopedExtensionDialogs(session.sessionId);
             if (eventType === "compaction_end")
                 this.scheduleCompactionQueueDrain(session.sessionId);
             if (eventType === "agent_start" || eventType === "agent_end")
@@ -1614,7 +2560,7 @@ export class PiSessionService {
         const model = session.model;
         if (model === undefined)
             return;
-        void generateShortSessionName(session.agent.streamFn, model, firstMessage).then((name) => {
+        void generateShortSessionName(session.agent.streamFunction, model, firstMessage).then((name) => {
             this.applyGeneratedSessionName(session, name ?? fallbackSessionName(firstMessage));
         }).catch(() => {
             this.applyGeneratedSessionName(session, fallbackSessionName(firstMessage));
@@ -1623,9 +2569,32 @@ export class PiSessionService {
     applyGeneratedSessionName(session, name) {
         if (name === undefined || session.sessionName !== undefined)
             return;
+        if (this.treeNavigations.has(session)) {
+            this.deferredGeneratedSessionNames.set(session, name);
+            return;
+        }
         session.setSessionName(name);
         this.publishSessionName(session);
     }
+    flushDeferredTreeNavigationWork(session) {
+        const generatedName = this.deferredGeneratedSessionNames.get(session);
+        this.deferredGeneratedSessionNames.delete(session);
+        if (generatedName !== undefined) {
+            try {
+                this.applyGeneratedSessionName(session, generatedName);
+            }
+            catch (error) {
+                this.logger.info({ sessionId: session.sessionId, error: error instanceof Error ? error.message : String(error) }, "failed to apply deferred session name");
+            }
+        }
+        const notifications = this.deferredSubsessionNotifications.get(session) ?? [];
+        this.deferredSubsessionNotifications.delete(session);
+        for (const notification of notifications) {
+            void this.deliverSubsessionNotification(session, notification).catch((error) => {
+                this.logSubsessionNotificationFailure(notification.parentId, notification.childId, error);
+            });
+        }
+    }
     applyAuthChange(change = {}) {
         // ModelRuntime.login()/logout() refresh the shared runtime before AuthService
         // emits the change, so no refresh is needed here. Keeping this synchronous
@@ -1694,6 +2663,10 @@ export class PiSessionService {
         }
     }
     activityLabelFromStatus(session) {
+        if (this.treeNavigations.has(session))
+            return "navigating session tree";
+        if (this.isSessionEntryMutationActive(session))
+            return "updating session";
         if (session.isCompacting)
             return "compacting";
         if (session.isBashRunning)
@@ -1705,7 +2678,92 @@ export class PiSessionService {
         return "active";
     }
     hasActiveWork(session) {
-        return sessionHasActiveWork(session, this.compactionQueuedMessages(session.sessionId).length);
+        return this.treeNavigations.has(session)
+            || this.isSessionEntryMutationActive(session)
+            || this.isTreeExclusiveOperationActive(session)
+            || sessionHasActiveWork(session, this.compactionQueuedMessages(session.sessionId).length);
+    }
+    async runTreeExclusiveOperation(targets, activeError, operation) {
+        const sessionIds = new Set();
+        const runtimes = new Set();
+        const sessions = new Set();
+        for (const target of targets) {
+            const runtime = target.runtime ?? (target.session === undefined ? undefined : this.activeRuntimeForSession(target.session));
+            const session = target.session ?? runtime?.session;
+            if (session !== undefined && this.hasActiveWork(session))
+                throw new Error(activeError);
+            sessionIds.add(target.sessionId);
+            if (runtime !== undefined)
+                runtimes.add(runtime);
+            if (session !== undefined)
+                sessions.add(session);
+        }
+        for (const sessionId of sessionIds) {
+            this.treeExclusiveSessionOperationCounts.set(sessionId, (this.treeExclusiveSessionOperationCounts.get(sessionId) ?? 0) + 1);
+        }
+        for (const runtime of runtimes) {
+            this.treeExclusiveRuntimeOperationCounts.set(runtime, (this.treeExclusiveRuntimeOperationCounts.get(runtime) ?? 0) + 1);
+        }
+        for (const session of sessions)
+            this.observeUnreadActivityState(session);
+        try {
+            return await operation();
+        }
+        finally {
+            for (const runtime of runtimes)
+                decrementWeakCount(this.treeExclusiveRuntimeOperationCounts, runtime);
+            for (const sessionId of sessionIds)
+                decrementMapCount(this.treeExclusiveSessionOperationCounts, sessionId);
+            for (const session of sessions) {
+                if (this.isCurrentActiveSession(session))
+                    this.observeUnreadActivityState(session);
+            }
+        }
+    }
+    isTreeExclusiveSessionIdentityActive(sessionId) {
+        return (this.treeExclusiveSessionOperationCounts.get(sessionId) ?? 0) > 0;
+    }
+    isTreeExclusiveOperationActive(session) {
+        if (this.isTreeExclusiveSessionIdentityActive(session.sessionId))
+            return true;
+        const runtime = this.activeRuntimeForSession(session);
+        return runtime !== undefined && (this.treeExclusiveRuntimeOperationCounts.get(runtime) ?? 0) > 0;
+    }
+    activeRuntimeForSession(session) {
+        for (const active of new Set(this.active.values())) {
+            if (active.runtime.session === session)
+                return active.runtime;
+        }
+        return undefined;
+    }
+    assertTreeNavigationInactive(session, action) {
+        if (this.treeNavigations.has(session))
+            throw new Error(`Cannot ${action} while session tree navigation is active`);
+    }
+    async runSessionEntryMutation(session, action, operation) {
+        this.beginSessionEntryMutation(session, action);
+        try {
+            return await operation();
+        }
+        finally {
+            this.endSessionEntryMutation(session);
+        }
+    }
+    beginSessionEntryMutation(session, action) {
+        this.assertTreeNavigationInactive(session, action);
+        this.sessionEntryMutationCounts.set(session, (this.sessionEntryMutationCounts.get(session) ?? 0) + 1);
+        this.observeUnreadActivityState(session);
+    }
+    endSessionEntryMutation(session) {
+        const remaining = (this.sessionEntryMutationCounts.get(session) ?? 1) - 1;
+        if (remaining <= 0)
+            this.sessionEntryMutationCounts.delete(session);
+        else
+            this.sessionEntryMutationCounts.set(session, remaining);
+        this.observeUnreadActivityState(session);
+    }
+    isSessionEntryMutationActive(session) {
+        return (this.sessionEntryMutationCounts.get(session) ?? 0) > 0;
     }
     publishActivityForEvent(session, event) {
         const eventType = getString(event, "type");
@@ -1759,6 +2817,54 @@ export class PiSessionService {
         if (this.hasActiveWork(session))
             this.publishActivity(session, eventType.replaceAll("_", " "), "active");
     }
+    /**
+     * Build the reporter for one session construction.
+     *
+     * The session id is known before any await — a `SessionManager` has its id
+     * from construction — so the daemon can name what it is starting even though
+     * the `PiAgentSession` that {@link publishActivity} needs does not exist yet.
+     * Without an id there is nothing to report against, so the reporter stays
+     * silent and the browser keeps its own generic wording.
+     */
+    startupProgress(sessionManager, intent, startupToken) {
+        const sessionId = sessionManager.getSessionId();
+        if (sessionId === "")
+            return { report: noop, end: noop };
+        const label = intent === "create" ? "Creating session" : "Opening session";
+        return {
+            report: (phase) => { this.publishStartupProgress(sessionId, startupToken, label, "active", this.startupDetail(phase)); },
+            end: () => {
+                // A real activity published during the window (an extension error, say)
+                // is the truth about this session and must survive the clear.
+                if (this.activities.has(sessionId))
+                    return;
+                this.publishStartupProgress(sessionId, startupToken, "idle", "idle", undefined);
+            },
+        };
+    }
+    startupDetail(phase) {
+        return this.catalogRefreshStatus?.isRefreshInFlight() === true
+            ? `${phase} · ${STARTUP_CONCURRENT_CATALOG_REFRESH}`
+            : phase;
+    }
+    /**
+     * Report startup progress on the global channel only, echoing the caller's
+     * correlation token so a waiting browser row recognises its own construction.
+     *
+     * Unlike {@link publishActivity} this deliberately records nothing: no
+     * `activities` entry, no workspace activity, no unread observation. There is
+     * no session to own that state, and a failed creation would leave it stranded.
+     *
+     * Every report is marked `startup`, which is what keeps a session that is
+     * merely opening from counting as one doing work. This is the only publisher
+     * that sets the marker, and because it writes no `activities` entry no later
+     * heartbeat re-publication can carry it.
+     */
+    publishStartupProgress(sessionId, startupToken, label, phase, detail) {
+        const at = new Date().toISOString();
+        const activity = detail === undefined ? { sessionId, phase, label, at, startup: true } : { sessionId, phase, label, detail, at, startup: true };
+        this.events.publishGlobal(startupToken === undefined ? { type: "session.startup", activity } : { type: "session.startup", startupToken, activity });
+    }
     publishActivity(session, label, phase, detail) {
         const at = new Date().toISOString();
         const stored = detail === undefined ? { phase, label, at } : { phase, label, detail, at };
@@ -1767,6 +2873,7 @@ export class PiSessionService {
         this.workspaceActivity?.applySessionActivity(session.sessionManager.getCwd(), activity);
         this.events.publish(session.sessionId, { type: "activity.update", activity });
         this.events.publishGlobal({ type: "activity.update", activity });
+        this.observeUnreadActivityState(session);
     }
     publishStatus(session) {
         const status = this.statusFromSession(session);
@@ -1774,6 +2881,7 @@ export class PiSessionService {
         this.workspaceActivity?.applySessionStatus(session.sessionManager.getCwd(), status);
         this.events.publish(session.sessionId, { type: "status.update", status });
         this.events.publishGlobal({ type: "status.update", status });
+        this.observeUnreadActivityState(session);
     }
     clearStaleActiveActivity(session) {
         const current = this.activities.get(session.sessionId);
@@ -1791,6 +2899,8 @@ export class PiSessionService {
         const model = session.model === undefined ? undefined : modelToClientModel(session.model);
         const contextUsage = session.getContextUsage();
         const warnings = this.warningsForSession(session);
+        const pendingAsk = this.pendingAskStore.pendingAsk(session.sessionId);
+        const pendingDialogs = this.pendingExtensionDialogStore.pendingDialogs(session.sessionId);
         return {
             sessionId: session.sessionId,
             persisted: sessionFileExists(session.sessionFile),
@@ -1806,6 +2916,8 @@ export class PiSessionService {
             cost: stats.cost,
             ...(contextUsage === undefined ? {} : { contextUsage }),
             ...(warnings.length === 0 ? {} : { warnings }),
+            ...(pendingAsk === undefined ? {} : { pendingAsk }),
+            ...(pendingDialogs.length === 0 ? {} : { pendingDialogs }),
         };
     }
     /**
@@ -1886,6 +2998,12 @@ function modelToClientModel(model) {
         ...(reasoning === undefined ? {} : { reasoning }),
     };
 }
+function notificationIdentityForSession(session) {
+    return {
+        sessionId: session.sessionId,
+        cwd: canonicalizeStoredCwd(session.sessionManager.getCwd()),
+    };
+}
 function clientSessionFromListEntry(session) {
     return {
         id: session.id,
@@ -2175,28 +3293,6 @@ function activeSessionFileMatches(active, expectedSessionFile) {
 function trackedLinkParentFileMatches(link, parentSessionFile) {
     return link.parentSessionFile !== undefined && sessionPathsEqual(link.parentSessionFile, parentSessionFile);
 }
-async function readSessionHeaderSummary(sessionFile) {
-    let file;
-    try {
-        file = await open(sessionFile, "r");
-        const buffer = Buffer.alloc(4096);
-        const { bytesRead } = await file.read(buffer, 0, buffer.length, 0);
-        const firstLine = buffer.toString("utf8", 0, bytesRead).split("\n", 1)[0];
-        if (firstLine === undefined || firstLine === "")
-            return undefined;
-        const header = JSON.parse(firstLine);
-        if (!isRecord(header) || header["type"] !== "session" || typeof header["id"] !== "string")
-            return undefined;
-        const parentSession = getString(header, "parentSession");
-        return { id: header["id"], ...(parentSession === undefined ? {} : { parentSession }) };
-    }
-    catch {
-        return undefined;
-    }
-    finally {
-        await file?.close().catch(() => undefined);
-    }
-}
 async function sessionFileHeaderMatches(sessionFile, expected) {
     const header = await readSessionHeaderSummary(sessionFile);
     if (header?.id !== expected.sessionId)