@timurproko/a1 0.1.8-dev.226 → 0.1.8-dev.259

package/dist/integrations/pi/engine/index.d.ts

@@ -1,6 +1,7 @@
 export * from "./adapter.js";
 export * from "./conformance.js";
 export * from "./runtime-integration.js";
+export * from "./session-selection.js";
 export * from "./session-integration.js";
 export * from "./model-auth-integration.js";
 export * from "./settings-integration.js";

package/dist/integrations/pi/engine/index.js

@@ -1,6 +1,7 @@
 export * from "./adapter.js";
 export * from "./conformance.js";
 export * from "./runtime-integration.js";
+export * from "./session-selection.js";
 export * from "./session-integration.js";
 export * from "./model-auth-integration.js";
 export * from "./settings-integration.js";

package/dist/integrations/pi/engine/package-integration.js

@@ -15,21 +15,33 @@ const MODEL_REFRESH_TIMEOUT_MS = 15_000;
  */
 export function createPiPackagesPort(input) {
     const { profileRoot, cwd } = input;
-    const settingsManager = SettingsManager.create(cwd, profileRoot, { projectTrusted: false });
-    const packageManager = new DefaultPackageManager({ cwd, agentDir: profileRoot, settingsManager });
-    packageManager.setProgressCallback(event => {
-        if (event.type !== "start" || !event.message)
-            return;
-        input.onProgress?.({ operation: operationFor(event.action), message: event.message });
-    });
-    const configured = () => packageManager.listConfiguredPackages()
+    // Compatibility: model-only refresh bypasses package settings and their diagnostics, as Pi does.
+    let packageManager;
+    const manager = () => {
+        if (packageManager)
+            return packageManager;
+        const settingsManager = SettingsManager.create(cwd, profileRoot, { projectTrusted: false });
+        for (const { scope, error } of settingsManager.drainErrors()) {
+            input.onDiagnostic?.({
+                message: `Warning (package command, ${scope} settings): ${error.message}`,
+                ...(error.stack ? { detail: error.stack } : {}),
+            });
+        }
+        packageManager = new DefaultPackageManager({ cwd, agentDir: profileRoot, settingsManager });
+        packageManager.setProgressCallback(event => {
+            if (event.type !== "start")
+                return;
+            input.onProgress?.({ operation: operationFor(event.action), message: String(event.message) });
+        });
+        return packageManager;
+    };
+    const configured = () => manager().listConfiguredPackages()
         .filter(entry => entry.scope === "user")
         .map(entry => Object.freeze({
         source: entry.source,
         installedPath: entry.installedPath ?? null,
         filtered: entry.filtered,
     }));
-    const isConfigured = (source) => configured().some(entry => entry.source === source || entry.source.startsWith(`${source}@`));
     return Object.freeze({
         capabilities: Object.freeze({ install: true, remove: true, update: true, refreshModels: true }),
         profileRoot,
@@ -38,13 +50,13 @@ export function createPiPackagesPort(input) {
         },
         async install(source) {
             return await attempt("install", source, async () => {
-                await packageManager.installAndPersist(source);
+                await manager().installAndPersist(source);
                 return agentPackageOutcome("install", "completed", null, source, configured());
             });
         },
         async remove(source) {
             return await attempt("remove", source, async () => {
-                const removed = await packageManager.removeAndPersist(source);
+                const removed = await manager().removeAndPersist(source);
                 if (!removed)
                     return agentPackageOutcome("remove", "not-found", null, source);
                 return agentPackageOutcome("remove", "completed", null, source, configured());
@@ -52,11 +64,8 @@ export function createPiPackagesPort(input) {
         },
         async update(source) {
             return await attempt("update", source ?? null, async () => {
-                // Rationale: asking first keeps "that package is not installed here" a structural
-                // answer rather than a string a caller would have to recognize.
-                if (source !== undefined && !isConfigured(source))
-                    return agentPackageOutcome("update", "not-found", null, source);
-                await packageManager.update(source);
+                // Compatibility: Pi owns source identity and its contextual no-match suggestions.
+                await manager().update(source);
                 return agentPackageOutcome("update", "completed", null, source ?? null, configured());
             });
         },
@@ -73,7 +82,9 @@ async function attempt(operation, source, run) {
         return await run();
     }
     catch (error) {
-        return agentPackageOutcome(operation, "failed", describe(error), source);
+        const detail = error instanceof Error ? error.message : operation === "refresh-models"
+            ? "Unknown model catalog refresh error" : "Unknown package command error";
+        return agentPackageOutcome(operation, "failed", detail, source);
     }
 }
 async function refreshModelCatalogs(profileRoot) {
@@ -88,9 +99,10 @@ async function refreshModelCatalogs(profileRoot) {
         });
         const result = await modelRuntime.refresh({ allowNetwork: true, force: true, signal: controller.signal });
         if (result.aborted)
-            throw new Error(`model catalog refresh timed out after ${MODEL_REFRESH_TIMEOUT_MS}ms`);
+            throw new Error("Model catalog refresh timed out.");
         if (result.errors.size > 0) {
-            throw new Error(Array.from(result.errors, ([provider, error]) => `${provider}: ${error.message}`).join("; "));
+            const details = Array.from(result.errors, ([provider, error]) => `${provider}: ${error.message}`).join("; ");
+            throw new Error(`Could not refresh model catalogs: ${details}`);
         }
     }
     finally {
@@ -104,9 +116,3 @@ function operationFor(action) {
         return "remove";
     return "update";
 }
-function describe(error) {
-    const message = (error instanceof Error ? error.message : String(error)).replace(/\s+/g, " ").trim();
-    if (message.length === 0)
-        return "unknown package failure";
-    return message.length > 600 ? `${message.slice(0, 597)}...` : message;
-}

package/dist/integrations/pi/engine/runtime-integration.d.ts

@@ -1,10 +1,13 @@
 import { createAgentSessionServices, SettingsManager, type AgentSession, type AgentSessionRuntime, type AgentSessionServices, type ScopedModel } from "@earendil-works/pi-coding-agent";
 import { resolvePiProjectTrustPreflight, type PiProjectTrustPreflightPrompt } from "./project-trust-preflight.js";
+import { type PiSessionForkPrompt, type PiSessionSelection } from "./session-selection.js";
 export interface PiRuntimeIntegrationOptions {
     readonly cwd: string;
     readonly agentDir: string;
     readonly sessionDir?: string;
     readonly sessionPath?: string;
+    readonly sessionSelection?: PiSessionSelection;
+    readonly sessionForkPrompt?: PiSessionForkPrompt;
     readonly projectTrustPrompt?: PiProjectTrustPreflightPrompt;
     readonly preflightDependencies?: PiRuntimePreflightDependencies;
 }

package/dist/integrations/pi/engine/runtime-integration.js

@@ -1,5 +1,6 @@
 import { createAgentSessionFromServices, createAgentSessionRuntime, createAgentSessionServices, resolveModelScopeWithDiagnostics, SessionManager, SettingsManager, } from "@earendil-works/pi-coding-agent";
 import { resolvePiProjectTrustPreflight, } from "./project-trust-preflight.js";
+import { openSelectedPiSession, resolveSessionArgumentPath } from "./session-selection.js";
 import { markStartupPhase } from "../../../foundation/startup/index.js";
 /**
  * Mirrors pinned Pi's CLI startup: resolve the `models` patterns from settings
@@ -49,10 +50,16 @@ export async function createPiRuntimeServicesAfterTrust(options) {
     return { services, trust };
 }
 export async function createPiRuntimeIntegration(options) {
-    const sessionDir = options.sessionDir ?? process.env.PI_CODING_AGENT_SESSION_DIR;
-    const sessionManager = options.sessionPath === undefined
+    const directory = options.sessionDir ?? process.env.PI_CODING_AGENT_SESSION_DIR;
+    const sessionDir = directory === undefined ? undefined : resolveSessionArgumentPath(directory, options.cwd);
+    const selection = options.sessionSelection ?? (options.sessionPath === undefined ? undefined : { target: resolveSessionArgumentPath(options.sessionPath, options.cwd) });
+    const sessionManager = selection === undefined
         ? SessionManager.create(options.cwd, sessionDir)
-        : SessionManager.open(options.sessionPath, sessionDir, options.cwd);
+        : await openSelectedPiSession({
+            cwd: options.cwd, selection,
+            ...(sessionDir === undefined ? {} : { sessionDir }),
+            ...(options.sessionForkPrompt === undefined ? {} : { forkPrompt: options.sessionForkPrompt }),
+        });
     const createRuntime = async ({ cwd, sessionManager: targetSessionManager, sessionStartEvent, }) => {
         const { services, trust } = await createPiRuntimeServicesAfterTrust({
             cwd,
@@ -81,7 +88,7 @@ export async function createPiRuntimeIntegration(options) {
         };
     };
     return createAgentSessionRuntime(createRuntime, {
-        cwd: options.cwd,
+        cwd: sessionManager.getCwd(),
         agentDir: options.agentDir,
         sessionManager,
     });

package/dist/integrations/pi/engine/session-selection.d.ts

@@ -0,0 +1,28 @@
+import { SessionManager, type SessionInfo } from "@earendil-works/pi-coding-agent";
+export interface PiSessionSelection {
+    readonly target: string;
+    readonly sessionDir?: string;
+}
+export type PiSessionForkPrompt = (request: {
+    readonly sourceCwd: string;
+}) => Promise<boolean>;
+/** Expected resume outcomes are concise CLI diagnostics, not engine crashes. */
+export declare class PiSessionSelectionError extends Error {
+    readonly exitCode: number;
+    constructor(message: string, exitCode?: number);
+}
+interface SessionSelectionOptions {
+    readonly cwd: string;
+    readonly selection: PiSessionSelection;
+    readonly sessionDir?: string;
+    readonly forkPrompt?: PiSessionForkPrompt;
+    /** Public API seam for deterministic lookup/open race and listing-order evidence. */
+    readonly sessions?: Pick<typeof SessionManager, "list" | "open" | "forkFrom"> & {
+        readonly listAll: (sessionDir?: string) => Promise<SessionInfo[]>;
+    };
+}
+/** Pinned Pi 0.84.2 main.ts policy, using SDK APIs rather than importing the CLI. */
+export declare function openSelectedPiSession(options: SessionSelectionOptions): Promise<SessionManager>;
+/** Resolve before switching effective cwd, including quoted home-relative paths. */
+export declare function resolveSessionArgumentPath(path: string, cwd: string): string;
+export {};

package/dist/integrations/pi/engine/session-selection.js

@@ -0,0 +1,90 @@
+import { accessSync, constants, existsSync, realpathSync, statSync } from "node:fs";
+import { homedir } from "node:os";
+import { resolve } from "node:path";
+import { SessionManager } from "@earendil-works/pi-coding-agent";
+/** Expected resume outcomes are concise CLI diagnostics, not engine crashes. */
+export class PiSessionSelectionError extends Error {
+    exitCode;
+    constructor(message, exitCode = 1) {
+        super(message);
+        this.exitCode = exitCode;
+        this.name = "PiSessionSelectionError";
+    }
+}
+/** Pinned Pi 0.84.2 main.ts policy, using SDK APIs rather than importing the CLI. */
+export async function openSelectedPiSession(options) {
+    try {
+        const sessions = options.sessions ?? SessionManager;
+        const cwd = resolve(options.cwd);
+        const directory = options.selection.sessionDir ?? options.sessionDir ?? process.env.PI_CODING_AGENT_SESSION_DIR;
+        const sessionDir = directory === undefined ? undefined : resolveSessionArgumentPath(directory, cwd);
+        const target = options.selection.target;
+        let path;
+        let global;
+        if (target.includes("/") || target.includes("\\") || target.endsWith(".jsonl")) {
+            path = resolveSessionArgumentPath(target, cwd);
+        }
+        else {
+            const local = matchSession(await sessions.list(cwd, sessionDir), target);
+            const globalMatch = local ? undefined : matchSession(await sessions.listAll(sessionDir), target);
+            // Compatibility: macOS may expose one temporary directory as /var to one process
+            // and /private/var to another. Physical path identity must not turn that alias into
+            // a cross-project fork prompt.
+            global = globalMatch && !sameProjectDirectory(globalMatch.cwd, cwd) ? globalMatch : undefined;
+            const matched = local ?? globalMatch;
+            if (!matched)
+                throw new PiSessionSelectionError(`No session found matching '${target}'`);
+            path = matched.path;
+        }
+        assertResumeFile(path);
+        if (global) {
+            if (!options.forkPrompt || !await options.forkPrompt({ sourceCwd: global.cwd })) {
+                throw new PiSessionSelectionError("Session resume cancelled.", 0);
+            }
+            assertResumeFile(path);
+            assertCwd(cwd);
+            return sessions.forkFrom(path, cwd, sessionDir);
+        }
+        // Invariant: no await between the final guard and SDK open. A disappeared target
+        // can otherwise produce a new in-memory header instead of an existing session.
+        const manager = sessions.open(path, sessionDir);
+        assertResumeFile(path);
+        assertCwd(manager.getCwd());
+        return manager;
+    }
+    catch (error) {
+        if (error instanceof PiSessionSelectionError)
+            throw error;
+        throw new PiSessionSelectionError(`Could not resume session: ${error instanceof Error ? error.message : String(error)}`);
+    }
+}
+function sameProjectDirectory(left, right) {
+    try {
+        const first = realpathSync.native(resolve(left));
+        const second = realpathSync.native(resolve(right));
+        return process.platform === "win32" ? first.toLowerCase() === second.toLowerCase() : first === second;
+    }
+    catch {
+        return resolve(left) === resolve(right);
+    }
+}
+function matchSession(sessions, target) {
+    return sessions.find(session => session.id === target) ?? sessions.find(session => session.id.startsWith(target));
+}
+/** Resolve before switching effective cwd, including quoted home-relative paths. */
+export function resolveSessionArgumentPath(path, cwd) {
+    const expanded = path === "~" ? homedir() : /^~[/\\]/.test(path) ? resolve(homedir(), path.slice(2)) : path;
+    return resolve(cwd, expanded);
+}
+function assertResumeFile(path) {
+    if (!existsSync(path))
+        throw new PiSessionSelectionError(`Session file does not exist: ${path}`);
+    const metadata = statSync(path);
+    if (!metadata.isFile() || metadata.size === 0)
+        throw new PiSessionSelectionError(`Session file is empty or not a regular file: ${path}`);
+    accessSync(path, constants.R_OK);
+}
+function assertCwd(cwd) {
+    if (!existsSync(cwd) || !statSync(cwd).isDirectory())
+        throw new PiSessionSelectionError(`Session working directory does not exist: ${cwd}`);
+}

package/dist/integrations/pi/engine/workflows.d.ts

@@ -62,6 +62,10 @@ export interface PiWorkflowRequest {
     readonly argument: string;
     readonly selection?: string;
     readonly confirmed?: boolean;
+    /** Recovery cwd selected after an import/resume source cwd is unavailable. */
+    readonly cwdOverride?: string;
+    /** Cancellation for an owned operation surface such as `/share`. */
+    readonly signal?: AbortSignal;
     readonly treeSummary?: {
         readonly summarize: boolean;
         readonly customInstructions?: string;
@@ -100,10 +104,19 @@ export interface PiSessionInfoPresentation {
     }[];
 }
 export type PiWorkflowPresentation = PiSessionInfoPresentation;
+export type PiWorkflowMessageKind = "status" | "warning" | "error" | "accent" | "silent";
+export interface PiWorkflowMessage {
+    readonly kind: Exclude<PiWorkflowMessageKind, "silent">;
+    readonly message: string;
+}
 export interface PiWorkflowResult {
     readonly command: PiWorkflowRoute;
     readonly outcome: PiWorkflowOutcome;
     readonly message: string;
+    /** Explicit presentation semantics; never inferred from message prefixes. */
+    readonly messageKind?: PiWorkflowMessageKind;
+    /** Ordered messages for truthful partial-success outcomes. */
+    readonly messages?: readonly PiWorkflowMessage[];
     readonly detail?: string;
     readonly selectorTitle?: string;
     readonly options?: readonly PiWorkflowOption[];
@@ -153,11 +166,15 @@ export interface PiWorkflowInteractionHost {
     startLogin?(request: PiWorkflowLoginStart): void;
     prompt(request: PiWorkflowInteractionRequest): Promise<string | null>;
     notify(event: PiWorkflowLoginNotification): void;
+    /** Deliver a post-login status or warning after the authentication dialog closes. */
+    publish?(message: PiWorkflowMessage): void;
     finishLogin?(): void;
 }
 export interface PiWorkflowHost {
     copyText(text: string): Promise<void>;
-    runCommand(command: string, arguments_: readonly string[]): Promise<{
+    runCommand(command: string, arguments_: readonly string[], options?: {
+        readonly signal?: AbortSignal;
+    }): Promise<{
         readonly stdout: string;
         readonly stderr: string;
     }>;

package/dist/integrations/pi/session-ui/session-shell-root.d.ts

@@ -1,7 +1,7 @@
 import type { OwnedUiSessionViewModel, OwnedUiViewportSettings, OwnedUiViewportSettingsPort } from "../../../contracts/owned-ui/index.js";
 import type { UiRouteHost } from "../../../ui/apps/index.js";
 import { type TranscriptViewportFrame, type TranscriptViewportFrameDescriptor } from "../../../ui/components/index.js";
-import { type PiEngineAdapter, type PiWorkflowResult } from "../engine/index.js";
+import { type PiEngineAdapter, type PiWorkflowMessage, type PiWorkflowResult } from "../engine/index.js";
 import { type PiShellClipboardContent, type PiShellComponentPort, type PiShellEditorPort, type PiShellExtensionRendererResolver, type PiShellHeaderOptions, type PiShellHeaderPort, type PiShellImageAssetResolver, type PiShellLoadedResourcesPort, type PiShellResourceEntry, type PiShellTranscriptComponentPort } from "../components/index.js";
 import { type PiTuiComponentPort, type PiTuiInputSurfaceKind, type PiTuiInputCoordinationScheduler, type PiTuiInputDiagnosticsEvent, type PiTuiLayoutNode, type PiTuiTerminalPort } from "../tui-runtime/index.js";
 import { type PreparedPrompt } from "./prompt-chips.js";
@@ -60,6 +60,7 @@ export declare class OwnedUiSessionShellRoot implements PiTuiComponentPort {
         readonly getRows: () => number;
         readonly requestRender: (force?: boolean) => void;
         readonly onViewportFrame?: (frame: TranscriptViewportFrame) => void;
+        readonly requestHyperlinkCleanup?: () => void;
         readonly enableDockInputReuse?: boolean;
         readonly onSubmit: (text: string) => void;
         readonly onInterrupt: () => void;
@@ -91,6 +92,8 @@ export declare class OwnedUiSessionShellRoot implements PiTuiComponentPort {
     applyTranscriptBlock(block: OwnedUiSessionViewModel["transcript"][number]): void;
     render(width: number): readonly string[];
     viewportFrameDescriptor(): TranscriptViewportFrameDescriptor | null;
+    /** Visible rows of the transient working-status tail, for rendering evidence. */
+    viewportTransientTailRowCount(): number;
     hasActiveSelection(): boolean;
     setViewportConfig(config: OwnedUiViewportSettings): void;
     resumeViewportFollowing(): void;
@@ -106,6 +109,7 @@ export declare class OwnedUiSessionShellRoot implements PiTuiComponentPort {
     transcriptComponent(id: string): PiShellTranscriptComponentPort | undefined;
     exitTranscript(width: number): string;
     appendWorkflowStatus(message: string): void;
+    appendWorkflowMessage(message: PiWorkflowMessage): void;
     appendWorkflowResult(result: PiWorkflowResult): void;
     appendDaxnuts(): void;
     toggleThinkingVisibility(): void;

package/dist/integrations/pi/session-ui/session-shell-root.js

@@ -44,6 +44,8 @@ export class OwnedUiSessionShellRoot {
     #inputSurfaceCoordination = "editor";
     #dockInputReuseEnabled;
     #dockInputCandidate = false;
+    #dockInputSnapshot;
+    #dockInputStatusSignature;
     #visibleViewportSnapshot;
     #fullViewportCompositions = 0;
     #dockOnlyViewportCompositions = 0;
@@ -111,6 +113,7 @@ export class OwnedUiSessionShellRoot {
             enabled: this.#customViewport,
             editor: this.editor,
             requestRender: force => this.#componentRuntime.requestRender(force),
+            requestHyperlinkCleanup: () => handlers.requestHyperlinkCleanup?.(),
             hasEditorLinks: () => this.#promptChips.hyperlinkRanges(this.editor.getText()).length > 0,
         });
         // Performance: stable painter identities let the neutral viewport retain row-level
@@ -229,9 +232,14 @@ export class OwnedUiSessionShellRoot {
         // intentional final gutter after their right-aligned timestamp.
         const documentWidth = width;
         const document = this.#renderDocumentLayout(documentWidth);
-        // Invariant: prompt-adjacent transient surfaces always belong to the dock.
-        // Changing from fitting to overflowing content reallocates transcript rows
-        // only; it never moves queued or Working rows into selectable history.
+        const statusRows = this.#renderStatus(width);
+        const statusSignature = statusRows.length === 0 ? undefined : `${statusRows.length}\u0000${statusRows.join("\u0000")}`;
+        const snapshot = this.#visibleViewportSnapshot;
+        const dockInputCandidate = this.#dockInputCandidate;
+        const dockInputSnapshot = dockInputCandidate ? this.#dockInputSnapshot : undefined;
+        const dockInputStatusSignature = dockInputCandidate ? this.#dockInputStatusSignature : statusSignature;
+        // Invariant: queued rows stay docked while live Working rows form one scrollable tail.
+        const scrollRows = [...document.rows, ...statusRows];
         const dockRows = dock.rows;
         const selectableDocumentRowCount = document.rows.length;
         const dockStartRow = height - dockRows.length + 1;
@@ -242,29 +250,32 @@ export class OwnedUiSessionShellRoot {
                 rowEnd: dockStartRow + editorOffset + dock.inputRows - 1,
             }
             : undefined);
-        const snapshot = this.#visibleViewportSnapshot;
-        const dockInputCandidate = this.#dockInputCandidate;
         this.#dockInputCandidate = false;
         let frame = dockInputCandidate
             && snapshot !== undefined
             && snapshot.width === width
             && snapshot.height === height
-            && snapshot.documentRows === document.rows
-            && snapshot.promptAnchors === document.promptAnchors
+            && snapshot.documentRows === (dockInputSnapshot?.documentRows ?? scrollRows)
+            && snapshot.statusSignature === dockInputStatusSignature
+            && snapshot.promptAnchors === (dockInputSnapshot?.promptAnchors ?? document.promptAnchors)
             && snapshot.dockLength === dockRows.length
-            && snapshot.inputSurface === this.#inputSurface
-            && snapshot.viewportRevision === this.#viewportController.presentationRevision
-            && snapshot.selectionRevision === this.#viewportController.selectionRevision
+            && snapshot.inputSurface === (dockInputSnapshot?.inputSurface ?? this.#inputSurface)
+            && snapshot.viewportRevision === (dockInputSnapshot?.viewportRevision ?? this.#viewportController.presentationRevision)
+            && snapshot.selectionRevision === (dockInputSnapshot?.selectionRevision ?? this.#viewportController.selectionRevision)
             ? this.#viewportController.composeDockOnly(dockRows, width, height)
             : null;
+        // Performance: a still-active spinner can tick before its first delayed presentation.
+        // Its newest tail signature is recomputed below and therefore cannot hide that change.
+        if (frame !== null && dockInputStatusSignature !== statusSignature)
+            frame = null;
         if (frame === null) {
             frame = this.#viewportController.compose({
-                documentRows: document.rows,
+                documentRows: scrollRows,
                 ...(this.#viewportController.transcriptPointerSelecting
                     ? { paintDocumentRow: heldNativeHyperlinkStyle }
                     : {}),
-                // Invariant: selection and copying stop at the real document tail; queued
-                // and Working rows remain dock-owned status chrome at every fit boundary.
+                // Invariant: selection and copying stop at the real document tail; live Working
+                // rows remain transient presentation chrome at every fit boundary.
                 selectableDocumentRowCount,
                 dockRows,
                 promptAnchors: document.promptAnchors,
@@ -282,7 +293,9 @@ export class OwnedUiSessionShellRoot {
         this.#visibleViewportSnapshot = {
             width,
             height,
-            documentRows: document.rows,
+            documentRows: scrollRows,
+            statusSignature,
+            dockInputStatusSignature,
             promptAnchors: document.promptAnchors,
             dockLength: dockRows.length,
             inputSurface: this.#inputSurface,
@@ -295,6 +308,10 @@ export class OwnedUiSessionShellRoot {
     viewportFrameDescriptor() {
         return this.#viewportController.frame?.descriptor ?? null;
     }
+    /** Visible rows of the transient working-status tail, for rendering evidence. */
+    viewportTransientTailRowCount() {
+        return this.#viewportController.frame?.hits.transientTail.length ?? 0;
+    }
     hasActiveSelection() {
         return this.#viewportController.hasSelection || this.editor.hasSelection();
     }
@@ -322,7 +339,7 @@ export class OwnedUiSessionShellRoot {
     }
     #renderDockLayout(width) {
         const queued = this.#view.editor.queuedSubmissions.length === 0 ? [] : this.#queued.render(width);
-        const statusRows = this.#renderStatus(width);
+        const statusRows = this.#customViewport ? this.#status.renderDock(width) : this.#renderStatus(width);
         const transientRows = [...queued, ...statusRows];
         const aboveWidgets = this.#renderWidgets("aboveEditor", width);
         const input = this.#inputSurface.render(width);
@@ -535,7 +552,25 @@ export class OwnedUiSessionShellRoot {
         this.#workflowStatusMessages.set(id, message);
         this.#lastWorkflowStatusId = id;
     }
+    appendWorkflowMessage(message) {
+        if (message.kind === "status") {
+            this.appendWorkflowStatus(message.message);
+            return;
+        }
+        this.#lastWorkflowStatusId = undefined;
+        const prefix = message.kind === "warning" ? "Warning" : message.kind === "error" ? "Error" : undefined;
+        const color = message.kind === "accent" ? "accent" : message.kind;
+        const text = prefix === undefined ? message.message : `${prefix}: ${message.message}`;
+        this.#appendAnchoredWorkflowComponent(() => ["", ` ${piTheme().fg(color, text)}`]);
+    }
     appendWorkflowResult(result) {
+        if (result.messages !== undefined) {
+            for (const message of result.messages)
+                this.appendWorkflowMessage(message);
+            return;
+        }
+        if (result.messageKind === "silent" || (result.outcome === "cancelled" && result.messageKind === undefined))
+            return;
         if (result.command === "reload" && result.outcome === "completed") {
             this.appendWorkflowStatus(result.message);
             return;
@@ -559,19 +594,16 @@ export class OwnedUiSessionShellRoot {
             return;
         }
         if (result.outcome === "failed") {
-            this.#lastWorkflowStatusId = undefined;
-            const warning = (result.command === "name" && result.message.startsWith("Usage:"))
-                || (result.command === "reload" && result.message.startsWith("Wait for "));
-            const prefix = warning ? "Warning" : "Error";
-            const color = warning ? "warning" : "error";
-            this.#appendAnchoredWorkflowComponent(() => ["", ` ${piTheme().fg(color, `${prefix}: ${result.message}`)}`]);
+            this.appendWorkflowMessage({
+                kind: result.messageKind === "warning" ? "warning" : "error",
+                message: result.message,
+            });
             return;
         }
         if (result.outcome === "completed" && (result.command === "quit" || result.command === "compact"))
             return;
         if (result.command === "new" && result.outcome === "completed") {
-            this.#lastWorkflowStatusId = undefined;
-            this.#appendAnchoredWorkflowComponent(() => ["", ` ${piTheme().fg("accent", result.message)}`]);
+            this.appendWorkflowMessage({ kind: "accent", message: result.message });
             return;
         }
         if (result.command === "name" && result.outcome === "completed") {
@@ -737,6 +769,8 @@ export class OwnedUiSessionShellRoot {
     handleInput(data) {
         this.#dockInputCandidate = this.#customViewport && this.#dockInputReuseEnabled
             && classifyPiTuiInput(data, this.#inputSurfaceCoordination) === "safe";
+        if (this.#dockInputCandidate)
+            this.#captureDockInputSnapshot();
         try {
             this.#inputSurface.handleInput?.(data);
         }
@@ -745,6 +779,22 @@ export class OwnedUiSessionShellRoot {
             throw error;
         }
     }
+    #captureDockInputSnapshot() {
+        const width = Math.max(1, this.#componentRuntime.getColumns());
+        const statusRows = this.#renderStatus(width);
+        this.#dockInputStatusSignature = statusRows.length === 0 ? undefined : `${statusRows.length}\u0000${statusRows.join("\u0000")}`;
+        const snapshot = this.#visibleViewportSnapshot;
+        this.#dockInputSnapshot = snapshot === undefined
+            ? undefined
+            : {
+                documentRows: snapshot.documentRows,
+                statusSignature: this.#dockInputStatusSignature,
+                promptAnchors: snapshot.promptAnchors,
+                inputSurface: this.#inputSurface,
+                viewportRevision: this.#viewportController.presentationRevision,
+                selectionRevision: this.#viewportController.selectionRevision,
+            };
+    }
     invalidate() {
         this.header.invalidate();
         this.resources.invalidate();
@@ -773,6 +823,8 @@ export class OwnedUiSessionShellRoot {
         this.#documentLayouts.clear();
         this.#visibleViewportSnapshot = undefined;
         this.#dockInputCandidate = false;
+        this.#dockInputSnapshot = undefined;
+        this.#dockInputStatusSignature = undefined;
         if (this.#inputSurface !== this.editor)
             this.#inputSurface.dispose?.();
         this.#extensionHeader?.dispose?.();
@@ -869,7 +921,7 @@ export class OwnedUiSessionShellRoot {
         return placement === "aboveEditor" ? ["", ...rows] : rows;
     }
     #renderStatus(width) {
-        return this.#status.render(width);
+        return this.#customViewport ? this.#status.renderLive(width) : this.#status.render(width);
     }
     #renderFooter(width) {
         return (this.#extensionFooter ?? this.#footer).render(width);

package/dist/integrations/pi/session-ui/session-shell.d.ts

@@ -31,7 +31,7 @@ export declare class OwnedUiSessionShell {
     queueFollowUp(): Promise<AdapterCommandResult>;
     restoreQueuedInput(): void;
     showSelector(title: string, options: readonly PiShellSelectorOption[], onSelect: (id: string) => void, onCancel?: () => void): PiTuiOverlayHandle;
-    showModelSelector(): void;
+    showModelSelector(initialSearchInput?: string): void;
     showForkSelector(): void;
     showLogoutSelector(): Promise<void>;
     showLoginMethodSelector(providerReference: string): void;