@ian-pascoe/pi-advisor 0.5.0 → 0.7.0

package/src/advisor-menu.ts

@@ -1,23 +1,17 @@
+import { getSettingsListTheme } from "@earendil-works/pi-coding-agent";
+import { SettingsList, type Component, type SettingItem } from "@earendil-works/pi-tui";
 import {
-  DynamicBorder,
-  ExtensionEditorComponent,
-  getSelectListTheme,
-  getSettingsListTheme,
-  type KeybindingsManager,
-} from "@earendil-works/pi-coding-agent";
-import {
-  fuzzyFilter,
-  getKeybindings,
-  Input,
-  SelectList,
-  SettingsList,
-  type Component,
-  type SelectItem,
-  type SettingItem,
-  type TUI,
-  type TuiMouseEvent,
-  type TuiMouseEventResult,
-} from "@earendil-works/pi-tui";
+  cycleDisplay,
+  EditorChooser,
+  effectiveWithSource,
+  errorText,
+  ModelPicker,
+  nextCycleValue,
+  scopeRow,
+  SettingsMenu,
+  ValueInput,
+  type SettingsMenuUi,
+} from "@ian-pascoe/pi-utils/settings-menu";
 import { Value } from "typebox/value";
 import { formatAdvisorOption, type AdvisorRenderTheme } from "./advisor-rendering.js";
 import {
@@ -61,11 +55,8 @@ export interface AdvisorMenuHost {
 }
 
 /** Native UI collaborators supplied by `ctx.ui.custom`. */
-export interface AdvisorMenuUi {
-  tui: TUI;
-  keybindings: KeybindingsManager;
+export interface AdvisorMenuUi extends SettingsMenuUi {
   theme: AdvisorRenderTheme;
-  externalEditorCommand?: string;
 }
 
 const thinkingLevels = ["off", "minimal", "low", "medium", "high", "xhigh", "max"];
@@ -99,6 +90,13 @@ const descriptions = {
   maxToolCalls: "Investigative tool calls per Review",
   maxCorrectiveTurns: "Automatic Corrective Turns per request",
   maxFindingsPerReview: "Findings accepted from one Review (1–32)",
+  maxNitsPerRequest: "Nits delivered per request; further Nits are dropped (0 delivers none)",
+  seedBudgetTokens:
+    "Token budget for the Context Seed; auto is a quarter of the Advisor model's context window, at most 100k",
+  reviewEvery:
+    "When Reviews run: every turn, every N turns, or once per request; a tool error reviews at once",
+  maxSessionTokens:
+    "Advisor Session size that triggers native compaction; auto is half the Advisor model's context window, at most 200k",
 } satisfies Record<keyof AdvisorOptions, string>;
 const inheritRow = "\u0000inherit";
 const inputHints = {
@@ -107,13 +105,11 @@ const inputHints = {
   maxToolCalls: "a number, or inherit",
   maxCorrectiveTurns: "a number, or inherit",
   maxFindingsPerReview: "a number from 1 to 32, or inherit",
+  maxNitsPerRequest: "a number, or inherit",
+  seedBudgetTokens: "a token count, auto, or inherit",
+  reviewEvery: "turn, request, a number of turns, or inherit",
+  maxSessionTokens: "a token count, auto, or inherit",
 } as const;
-const listActions = [
-  "tui.select.up",
-  "tui.select.down",
-  "tui.select.confirm",
-  "tui.select.cancel",
-] as const;
 
 /** Convert one typed or selected menu value into a validated change. */
 function parseAdvisorMenuValue(
@@ -133,9 +129,18 @@ function parseAdvisorMenuValue(
         return { catchUpThreshold: value === "off" ? value : Number(value) };
       case "reviewTimeoutMs":
         return { reviewTimeoutMs: Math.round(Number(value) * 1_000) };
+      case "seedBudgetTokens":
+        return { seedBudgetTokens: value === "auto" ? value : Number(value) };
+      case "maxSessionTokens":
+        return { maxSessionTokens: value === "auto" ? value : Number(value) };
+      case "reviewEvery":
+        return {
+          reviewEvery: value === "turn" || value === "request" ? value : Number(value),
+        };
       case "maxToolCalls":
       case "maxCorrectiveTurns":
       case "maxFindingsPerReview":
+      case "maxNitsPerRequest":
         return { [key]: Number(value) };
       default:
         return { [key]: value };
@@ -144,188 +149,34 @@ function parseAdvisorMenuValue(
   return { action: "set", key, patch: parseAdvisorOptions(patch, scope) };
 }
 
-function errorText(cause: unknown): string {
-  return cause instanceof Error ? cause.message : String(cause);
-}
-
-/** Single-line value entry with inline validation. */
-class ValueInput implements Component {
-  private readonly input: Input;
-  private error: string | undefined;
-
-  constructor(
-    private readonly title: string,
-    private readonly hint: string,
-    private readonly theme: AdvisorRenderTheme,
-    submit: (text: string) => void,
-    cancel: () => void,
-  ) {
-    this.input = new Input({ placeholder: hint });
-    this.input.focused = true;
-    this.input.onSubmit = (text) => {
-      try {
-        submit(text);
-      } catch (cause) {
-        this.error = errorText(cause);
-      }
-    };
-    this.input.onEscape = cancel;
-  }
-  handleInput(data: string): void {
-    this.error = undefined;
-    this.input.handleInput(data);
-  }
-  render(width: number): string[] {
-    return [
-      this.theme.bold(this.title),
-      ...this.input.render(width),
-      this.error ? this.theme.fg("error", `✖ ${this.error}`) : this.theme.fg("dim", this.hint),
-    ];
-  }
-  invalidate(): void {
-    this.input.invalidate();
-  }
-}
-
-/** Fuzzy-searchable model list with an inherit choice. */
-class ModelPicker implements Component {
-  private readonly input = new Input({ placeholder: "type to search" });
-  private list: SelectList;
-
-  constructor(
-    private readonly models: readonly string[],
-    private readonly choose: (value: string) => void,
-    private readonly cancel: () => void,
-  ) {
-    this.input.focused = true;
-    this.list = this.createList("");
-  }
-  private createList(query: string): SelectList {
-    const items: SelectItem[] = ["inherit", ...this.models].map((value) => ({
-      value,
-      label: value,
-    }));
-    const list = new SelectList(
-      query ? fuzzyFilter(items, query, (item) => item.value) : items,
-      10,
-      getSelectListTheme(),
-    );
-    list.onSelect = (item) => this.choose(item.value);
-    list.onCancel = this.cancel;
-    return list;
-  }
-  handleInput(data: string): void {
-    const bindings = getKeybindings();
-    if (listActions.some((action) => bindings.matches(data, action))) {
-      this.list.handleInput(data);
-      return;
-    }
-    this.input.handleInput(data);
-    this.list = this.createList(this.input.getValue());
-  }
-  render(width: number): string[] {
-    return [...this.input.render(width), ...this.list.render(width)];
-  }
-  invalidate(): void {
-    this.input.invalidate();
-    this.list.invalidate();
-  }
-}
-
-/** Choose between editing the prompt in Pi's editor component and inheriting it. */
-class PromptChooser implements Component {
-  private readonly list: SelectList;
-  private editor: ExtensionEditorComponent | undefined;
-  private error: string | undefined;
-
-  /** `submit` throws a user-facing message for an invalid prompt; the editor stays open. */
-  constructor(
-    ui: AdvisorMenuUi,
-    prompt: string,
-    submit: (text: string) => void,
-    inherit: () => void,
-    cancel: () => void,
-  ) {
-    this.list = new SelectList(
-      [
-        { value: "edit", label: "Edit…" },
-        { value: "inherit", label: "inherit" },
-      ],
-      2,
-      getSelectListTheme(),
-    );
-    this.list.onCancel = cancel;
-    this.list.onSelect = (item) => {
-      if (item.value === "inherit") {
-        inherit();
-        return;
-      }
-      this.editor = new ExtensionEditorComponent(
-        ui.tui,
-        ui.keybindings,
-        "Advisor Prompt",
-        prompt,
-        (text) => {
-          try {
-            submit(text);
-          } catch (cause) {
-            this.error = ui.theme.fg("error", `✖ ${errorText(cause)}`);
-          }
-        },
-        cancel,
-        undefined,
-        ui.externalEditorCommand,
-      );
-      this.editor.focused = true;
-    };
-  }
-  handleInput(data: string): void {
-    this.error = undefined;
-    (this.editor ?? this.list).handleInput(data);
-  }
-  render(width: number): string[] {
-    const lines = (this.editor ?? this.list).render(width);
-    return this.error ? [...lines, this.error] : lines;
-  }
-  invalidate(): void {
-    (this.editor ?? this.list).invalidate();
-  }
-}
-
 /** Menu row ids that are not Advisor options. */
 const actionRows = { resume: "resume", scope: "scope" } as const;
 
 /** `/advisor` settings menu built from Pi's native settings list. */
-export class AdvisorSettingsMenu implements Component {
+export class AdvisorSettingsMenu extends SettingsMenu {
   private scope: AdvisorSettingScope = "session";
   private view: AdvisorMenuView;
   private rows: SettingItem[] = [];
-  private list: SettingsList;
-  private error: string | undefined;
   private lastRow: string = actionRows.scope;
-  /** Edits run one at a time so each reads the result of the previous one. */
-  private pending: Promise<void> = Promise.resolve();
   private submenuOpen = false;
   private rebuildPending = false;
-  private readonly border: DynamicBorder;
 
   constructor(
     private readonly host: AdvisorMenuHost,
-    private readonly ui: AdvisorMenuUi,
-    private readonly done: () => void,
+    protected override readonly ui: AdvisorMenuUi,
+    done: () => void,
   ) {
+    super("Advisor settings", ui, done);
     this.view = host.view();
-    this.border = new DynamicBorder((text) => ui.theme.fg("border", text));
-    this.list = this.createList();
+    this.createList();
   }
 
-  /** Resolves once every edit started so far has been applied or rejected. */
-  settled(): Promise<void> {
-    return this.pending;
+  protected headline(): readonly string[] {
+    return this.view.headline;
   }
 
   /** Re-read the host after external state changes, such as a Review starting. */
-  refresh(): void {
+  override refresh(): void {
     const paused = this.view.paused;
     try {
       this.view = this.host.view();
@@ -345,8 +196,7 @@ export class AdvisorSettingsMenu implements Component {
 
   private rebuild(): void {
     this.rebuildPending = false;
-    this.list = this.createList();
-    this.list.selectItem(this.view.paused ? actionRows.resume : this.lastRow);
+    this.createList().selectItem(this.view.paused ? actionRows.resume : this.lastRow);
   }
 
   private createList(): SettingsList {
@@ -356,13 +206,7 @@ export class AdvisorSettingsMenu implements Component {
       ...advisorOptionKeys,
     ];
     this.rows = ids.map((id) => this.createRow(id));
-    return new SettingsList(
-      this.rows,
-      this.rows.length,
-      getSettingsListTheme(),
-      (id, value) => this.change(id, value),
-      this.done,
-    );
+    return this.setList(this.rows, (id, value) => this.change(id, value));
   }
 
   private createRow(id: string): SettingItem {
@@ -374,14 +218,7 @@ export class AdvisorSettingsMenu implements Component {
         values: ["retry"],
         description: "Retry the Paused Advisor with its current settings",
       };
-    if (id === actionRows.scope)
-      return {
-        id,
-        label: "Scope",
-        currentValue: this.scope,
-        values: [...this.view.scopes],
-        description: "Where edits are written",
-      };
+    if (id === actionRows.scope) return scopeRow(this.scope, this.view.scopes);
     return this.optionRow(advisorOptionKey(id));
   }
 
@@ -450,8 +287,9 @@ export class AdvisorSettingsMenu implements Component {
           submenu: this.submenu(
             key,
             (close) =>
-              new PromptChooser(
+              new EditorChooser(
                 this.ui,
+                "Advisor Prompt",
                 settings.prompt ?? "",
                 (text) => {
                   // Throws for an invalid prompt, keeping the editor open.
@@ -540,12 +378,14 @@ export class AdvisorSettingsMenu implements Component {
 
   /** This scope's own value, or what it inherits; notes when another scope overrides it. */
   private cycleDisplay(key: CycleKey): string {
-    const own = cycleValue(key, this.view.authored[this.scope] ?? {});
     const effective = cycleValue(key, this.view.settings);
     const source = this.view.sources[key] ?? "default";
-    const inEffect = effective === undefined ? "observed agent" : `${effective} · ${source}`;
-    if (own === undefined) return `inherit (${inEffect})`;
-    return source === this.scope ? own : `${own} (overridden: ${inEffect})`;
+    return cycleDisplay({
+      own: cycleValue(key, this.view.authored[this.scope] ?? {}),
+      inEffect: effective === undefined ? "observed agent" : effectiveWithSource(effective, source),
+      source,
+      scope: this.scope,
+    });
   }
 
   private change(id: string, value: string): void {
@@ -566,8 +406,7 @@ export class AdvisorSettingsMenu implements Component {
     // text is not one of the values, so ignore its suggestion.
     const values = cycleValues[key];
     const own = cycleValue(key, this.view.authored[this.scope] ?? {}) ?? "inherit";
-    const next =
-      values[(values.findIndex((option) => option === own) + 1) % values.length] ?? "inherit";
+    const next = nextCycleValue(values, own) ?? "inherit";
     let change: AdvisorChange;
     try {
       change = parseAdvisorMenuValue(key, next, this.scope);
@@ -588,42 +427,4 @@ export class AdvisorSettingsMenu implements Component {
     const scope = this.scope;
     this.run(() => this.host.apply(scope, change));
   }
-
-  private run(task: () => Promise<void>): void {
-    this.error = undefined;
-    this.pending = this.pending
-      .then(task)
-      .catch((cause: unknown) => {
-        this.error = errorText(cause);
-      })
-      .finally(() => {
-        this.refresh();
-        this.ui.tui.requestRender();
-      });
-  }
-
-  handleInput(data: string): void {
-    this.list.handleInput(data);
-  }
-
-  handleMouse(event: TuiMouseEvent): TuiMouseEventResult | undefined {
-    return this.list.handleMouse(event);
-  }
-
-  render(width: number): string[] {
-    const { theme } = this.ui;
-    return [
-      ...this.border.render(width),
-      ` ${theme.bold("Advisor settings")}`,
-      ...this.view.headline.map((line) => ` ${line}`),
-      "",
-      ...this.list.render(width),
-      ...(this.error ? [theme.fg("error", ` ✖ ${this.error}`)] : []),
-      ...this.border.render(width),
-    ];
-  }
-
-  invalidate(): void {
-    this.list.invalidate();
-  }
 }