feat: add ctx.ui.setWorkingMessage() extension API

Allows extensions to customize the streaming loader message.
Pass undefined to restore default.
This commit is contained in:
Nico Bailon 2026-01-10 21:02:41 -08:00
parent 016a24e9a1
commit 271b49da3c
6 changed files with 24 additions and 1 deletions

View file

@ -79,6 +79,7 @@ const noOpUIContext: ExtensionUIContext = {
input: async () => undefined,
notify: () => {},
setStatus: () => {},
setWorkingMessage: () => {},
setWidget: () => {},
setFooter: () => {},
setHeader: () => {},

View file

@ -79,6 +79,9 @@ export interface ExtensionUIContext {
/** Set status text in the footer/status bar. Pass undefined to clear. */
setStatus(key: string, text: string | undefined): void;
/** Set the working/loading message shown during streaming. Call with no argument to restore default. */
setWorkingMessage(message?: string): void;
/** Set a widget to display above the editor. Accepts string array or component factory. */
setWidget(key: string, content: string[] | undefined): void;
setWidget(key: string, content: ((tui: TUI, theme: Theme) => Component & { dispose?(): void }) | undefined): void;

View file

@ -135,6 +135,7 @@ export class InteractiveMode {
private isInitialized = false;
private onInputCallback?: (text: string) => void;
private loadingAnimation: Loader | undefined = undefined;
private readonly defaultWorkingMessage = "Working... (esc to interrupt)";
private lastSigintTime = 0;
private lastEscapeTime = 0;
@ -948,6 +949,11 @@ export class InteractiveMode {
input: (title, placeholder, opts) => this.showExtensionInput(title, placeholder, opts),
notify: (message, type) => this.showExtensionNotify(message, type),
setStatus: (key, text) => this.setExtensionStatus(key, text),
setWorkingMessage: (message) => {
if (this.loadingAnimation) {
this.loadingAnimation.setMessage(message ?? this.defaultWorkingMessage);
}
},
setWidget: (key, content) => this.setExtensionWidget(key, content),
setFooter: (factory) => this.setExtensionFooter(factory),
setHeader: (factory) => this.setExtensionHeader(factory),
@ -1559,7 +1565,7 @@ export class InteractiveMode {
this.ui,
(spinner) => theme.fg("accent", spinner),
(text) => theme.fg("muted", text),
"Working... (esc to interrupt)",
this.defaultWorkingMessage,
);
this.statusContainer.addChild(this.loadingAnimation);
this.ui.requestRender();

View file

@ -150,6 +150,10 @@ export async function runRpcMode(session: AgentSession): Promise<never> {
} as RpcExtensionUIRequest);
},
setWorkingMessage(_message?: string): void {
// Working message not supported in RPC mode - requires TUI loader access
},
setWidget(key: string, content: unknown): void {
// Only support string arrays in RPC mode - factory functions are ignored
if (content === undefined || Array.isArray(content)) {