@heyhuynhgiabuu/pi-pretty 0.6.24 → 0.6.26

package/dist/working-indicator.d.ts

@@ -0,0 +1,241 @@
+/**
+ * omp-style shimmer working indicator (widget takeover).
+ *
+ * Ports the shimmer sweep from oh-my-pi's `modes/theme/shimmer.ts`: the sweep
+ * is discretized into one pre-rendered frame per band position, cycled at
+ * 1000/30 ms so the band travels 30 cells/second — omp's exact speed.
+ *
+ * Why a widget: the host `Loader` extends `Text` with a hardcoded paddingX=1,
+ * and `setWorkingIndicator({ frames })` cannot change host component layout.
+ * To render the row flush-left we hide the host loader
+ * (`setWorkingVisible(false)`) and install our own zero-padding component via
+ * `setWidget(..., "aboveEditor")`, animating it with a 33ms interval.
+ *
+ * Frame layout: `<dim spinner> <shimmer text> <dim interrupt hint>`, one full
+ * sweep per phrase — `texts` rotates through phrases chapter by chapter with
+ * a continuous spinner phase.
+ */
+import { type ThinkingIndicatorConfig, type WorkingIndicatorConfig } from "./config.js";
+/** One cell per frame at 33ms = omp's 30 cells/second band speed. */
+export declare const WORKING_INTERVAL_MS: number;
+export type WorkingIndicatorMode = "shimmer" | "kitt" | "static";
+/** Tier colors as config values: a pi theme color name or a `#rrggbb` hex. */
+export interface WorkingIndicatorPalette {
+    low: string;
+    mid: string;
+    high: string;
+}
+export interface WorkingIndicatorSettings {
+    enabled: boolean;
+    /** Phrases rotated chapter-by-chapter across the sweep. */
+    texts: string[];
+    mode: WorkingIndicatorMode;
+    palette: WorkingIndicatorPalette;
+    bold: boolean;
+    hint: boolean;
+    /** Tint mid/high tiers (and dim the spinner) with a per-session accent color. */
+    sessionAccent: boolean;
+    /** True when the user explicitly configured `mid`/`high` — accent then stays off. */
+    tiersCustomized?: boolean;
+}
+export declare const WORKING_INDICATOR_DEFAULTS: WorkingIndicatorSettings;
+/**
+ * Resolve final settings from `pi-pretty.json`'s flat `workingIndicator` object
+ * and env overrides. Precedence: env var > config > default; empty or invalid
+ * env values mean unset (matching pi-pretty's config conventions).
+ */
+export declare function resolveWorkingIndicatorSettings(config: WorkingIndicatorConfig | undefined, env?: NodeJS.ProcessEnv): WorkingIndicatorSettings;
+/** Tier colors as raw ANSI open sequences. */
+export interface ResolvedPalette {
+    low: string;
+    mid: string;
+    high: string;
+}
+interface ThemeAnsiSource {
+    /** Narrowed tier names; pi's Theme (ThemeColor) satisfies this structurally. */
+    getFgAnsi?: (name: "dim" | "muted" | "accent" | "thinkingText") => string;
+}
+/**
+ * Resolve tier colors to ANSI open sequences. `#rrggbb` wins, then the active
+ * pi theme (`getFgAnsi`), then pi-pretty's built-in constants.
+ */
+export declare function resolvePaletteAnsi(palette: WorkingIndicatorPalette, theme?: ThemeAnsiSource): ResolvedPalette;
+export interface WorkingFrameOptions {
+    mode: WorkingIndicatorMode;
+    /** Resolved ANSI open sequences per tier. */
+    ansi: ResolvedPalette;
+    bold?: boolean;
+    spinner?: boolean;
+    /** Spinner glyph color; defaults to `ansi.low` (omp tints it with the dim accent). */
+    spinnerColor?: string;
+    /** Wrap every frame in italic (pi's hidden-thinking label is italic). */
+    italic?: boolean;
+    /** Pre-colored suffix (e.g. the dim interrupt hint) appended to every frame. */
+    hint?: string;
+    intervalMs?: number;
+}
+/**
+ * Build the frame list for the working row: one full sweep per phrase, in
+ * order, with the spinner phase running continuously across chapter
+ * boundaries. Same-tier runs share one ANSI pair (omp's run coalescing).
+ */
+export declare function buildWorkingFrames(texts: string[], options: WorkingFrameOptions): {
+    frames: string[];
+    intervalMs: number;
+};
+/** Estimate streamed output tokens: provider `usage.output` when exposed, else visible chars ÷ 4. */
+export declare function workingTokens(message: unknown): number;
+/** Slice of the pi-tui TUI instance the widget needs. */
+export interface WidgetTuiLike {
+    requestRender(): void;
+}
+/**
+ * Zero-padding working-row component. The host renders whatever `render()`
+ * returns with no margins, so the row sits flush-left — the thing the frames
+ * API could not do.
+ */
+export declare class WorkingWidget {
+    #private;
+    setFrames(frames: string[], intervalMs: number): void;
+    attach(tui: WidgetTuiLike): void;
+    /** Ask the host for a render — lets non-widget consumers (the thinking label
+     * ticker) reuse this widget's TUI handle instead of relying on natural
+     * streaming renders. */
+    requestRender(): void;
+    /** Dynamic right-side status segment (e.g. live token count), styled dim at
+     * render time so theme changes don't bake stale ANSI into it. */
+    setStats(text: string | undefined): void;
+    start(): void;
+    stop(): void;
+    /** Flush-left single line; invisible (no lines) while stopped. */
+    render(width: number): string[];
+    invalidate(): void;
+    dispose(): void;
+}
+export interface ThinkingIndicatorSettings {
+    enabled: boolean;
+}
+export declare const THINKING_INDICATOR_DEFAULTS: ThinkingIndicatorSettings;
+/**
+ * Resolve thinking-label shimmer settings from `pi-pretty.json`'s
+ * `thinkingIndicator` object and env overrides (env > config > default).
+ */
+export declare function resolveThinkingIndicatorSettings(config: ThinkingIndicatorConfig | undefined, env?: NodeJS.ProcessEnv): ThinkingIndicatorSettings;
+export interface ThinkingUiLike {
+    theme?: WorkingThemeLike;
+    setHiddenThinkingLabel(label?: string): void;
+}
+/** Controller for the host-component label interception. */
+export interface PerRowThinkingLabels {
+    /** Mark the message currently streaming a thinking block (its row animates). */
+    setActive(timestamp: number): void;
+    /** Freeze a message's completed thinking duration in ms (its row shows
+     * `Thought for Xs` derived from this). */
+    complete(timestamp: number, durationMs: number): void;
+    /** Drop the active mark (run ended); completed rows keep their durations. */
+    clearActive(): void;
+    /** Restore the original prototype setter. */
+    uninstall(): void;
+}
+/**
+ * Intercept the host's per-row label fan-out method
+ * (`AssistantMessageComponent.prototype.setHiddenThinkingLabel`): each row
+ * substitutes the incoming global label with its own — the streaming row
+ * animates, completed rows stay frozen at their own `Thought for Xs`, unknown
+ * rows fall back to pi's default. Any per-call failure passes the incoming
+ * label through unchanged, so a host internals change degrades to pi-pretty's
+ * global-label behavior instead of breaking. Durations live per session.
+ */
+export declare function installPerRowThinkingLabels(componentClass: unknown): PerRowThinkingLabels | undefined;
+export interface ThinkingLabelAnimator {
+    /** Apply the next shimmer frame, rebuilding frames when the label changes. */
+    tick(label?: string): void;
+    /** Apply a non-animated label (the host supplies its normal thinking style). */
+    show(label: string): void;
+    /** Restore pi's default static label. */
+    restore(): void;
+    /** Current pre-rendered label frames (exposed for diagnostics). */
+    readonly frames: readonly string[];
+}
+export interface ThinkingTimer {
+    /** Render the active label with the current elapsed duration. */
+    tick(): void;
+    /** Freeze the elapsed duration; returns the total in milliseconds, or
+     * undefined when the timer was already terminal. */
+    complete(): number | undefined;
+    /** Restore pi's default label and make the timer terminal. */
+    restore(): void;
+}
+/** Format elapsed milliseconds as whole seconds with compact padded units. */
+export declare function formatThinkingDuration(elapsedMs: number): string;
+/**
+ * Couple elapsed-time wording to the label animator through an injected clock.
+ * The timer starts on the first thinking delta, freezes once, and is terminal
+ * after completion or restore. `startElapsedMs` resumes a later run in the
+ * same message from its accumulated thinking time (per-message semantics:
+ * the host renders one label line per thinking run, all sharing the message
+ * label, so a fresh-zero restart would visibly rewind earlier lines).
+ */
+export declare function createThinkingTimer(animator: ThinkingLabelAnimator, now?: () => number, startElapsedMs?: number): ThinkingTimer;
+/**
+ * True while a streaming assistant message is currently emitting a thinking
+ * block (its last content block). Used to gate label ticks to the thinking
+ * phase — once text or tool calls stream, the label is frozen back to pi's
+ * default instead of rebuilding the transcript for a row nobody watches.
+ */
+export declare function thinkingBlockActive(message: unknown): boolean;
+/**
+ * Animate pi's hidden-thinking label ("Thinking...") with the same shimmer as
+ * the working row. Pi renders the label as static italic `thinkingText` text;
+ * the only extension lever is `setHiddenThinkingLabel(label)`, which rebuilds
+ * chat children, so index.ts drives it at 30fps only while the streaming
+ * message's last block is thinking. The elapsed label is rebuilt only when its
+ * whole-second text changes; the shimmer frame still advances every tick.
+ *
+ * Tiers: low = theme `thinkingText` (pi's own label look), mid/high = session
+ * accent when enabled; every frame is italic like pi's label; no spinner.
+ */
+export declare function createThinkingLabelAnimator(ui: ThinkingUiLike, workSettings: WorkingIndicatorSettings, sessionName?: string, thinking?: ThinkingIndicatorSettings): ThinkingLabelAnimator;
+/** Structural slice of pi's Theme needed for the hint, palette, and thinking label. */
+export interface WorkingThemeLike {
+    /** Narrowed to the only color the hint needs; pi's ThemeColor satisfies it. */
+    fg?: (name: "dim", text: string) => string;
+    getFgAnsi?: (name: "dim" | "muted" | "accent" | "thinkingText") => string;
+}
+export interface WorkingUiLike {
+    theme?: WorkingThemeLike;
+    setWorkingVisible(visible: boolean): void;
+    setWidget(key: string, factory: ((tui: WidgetTuiLike, theme: unknown) => WorkingWidget) | undefined, options?: {
+        placement?: "aboveEditor" | "belowEditor";
+    }): void;
+}
+export interface KeybindingsLike {
+    getKeys(name: string): string[];
+}
+export interface WorkingIndicatorController {
+    /** Streaming began — show and animate the row. */
+    start(): void;
+    /** Streaming ended — hide the row and pause the animation. */
+    stop(): void;
+    /** Remove the widget and stop the animation. */
+    dispose(): void;
+    /** Update the dim right-side status segment (live token count). */
+    setStats(text: string | undefined): void;
+    /** Request a host render via the widget's TUI handle. */
+    requestRender(): void;
+    /** Pre-rendered frames (exposed for diagnostics). */
+    readonly frames: readonly string[];
+}
+/**
+ * Take over pi's working row with our own flush-left shimmer widget.
+ *
+ * Installs a zero-padding component above the editor and hides the host
+ * loader; returns a controller the host lifecycle drives (`start` on
+ * agent_start, `stop` on agent_end). No-op (host defaults untouched) when
+ * disabled or the phrase list is empty.
+ */
+export declare function installWorkingIndicator(ui: WorkingUiLike, settings: WorkingIndicatorSettings, deps?: {
+    getKeybindings?: () => KeybindingsLike | undefined | Promise<KeybindingsLike | undefined>;
+}, sessionName?: string): Promise<WorkingIndicatorController>;
+export {};
+//# sourceMappingURL=working-indicator.d.ts.map

package/dist/working-indicator.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"working-indicator.d.ts","sourceRoot":"","sources":["../src/working-indicator.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;GAgBG;AAGH,OAAO,EAA6B,KAAK,uBAAuB,EAAE,KAAK,sBAAsB,EAAE,MAAM,aAAa,CAAC;AAWnH,qEAAqE;AACrE,eAAO,MAAM,mBAAmB,QAAwB,CAAC;AAgBzD,MAAM,MAAM,oBAAoB,GAAG,SAAS,GAAG,MAAM,GAAG,QAAQ,CAAC;AAEjE,8EAA8E;AAC9E,MAAM,WAAW,uBAAuB;IACvC,GAAG,EAAE,MAAM,CAAC;IACZ,GAAG,EAAE,MAAM,CAAC;IACZ,IAAI,EAAE,MAAM,CAAC;CACb;AAED,MAAM,WAAW,wBAAwB;IACxC,OAAO,EAAE,OAAO,CAAC;IACjB,2DAA2D;IAC3D,KAAK,EAAE,MAAM,EAAE,CAAC;IAChB,IAAI,EAAE,oBAAoB,CAAC;IAC3B,OAAO,EAAE,uBAAuB,CAAC;IACjC,IAAI,EAAE,OAAO,CAAC;IACd,IAAI,EAAE,OAAO,CAAC;IACd,iFAAiF;IACjF,aAAa,EAAE,OAAO,CAAC;IACvB,qFAAqF;IACrF,eAAe,CAAC,EAAE,OAAO,CAAC;CAC1B;AAED,eAAO,MAAM,0BAA0B,EAAE,wBASxC,CAAC;AA8CF;;;;GAIG;AACH,wBAAgB,+BAA+B,CAC9C,MAAM,EAAE,sBAAsB,GAAG,SAAS,EAC1C,GAAG,GAAE,MAAM,CAAC,UAAwB,GAClC,wBAAwB,CAwB1B;AAID,8CAA8C;AAC9C,MAAM,WAAW,eAAe;IAC/B,GAAG,EAAE,MAAM,CAAC;IACZ,GAAG,EAAE,MAAM,CAAC;IACZ,IAAI,EAAE,MAAM,CAAC;CACb;AAED,UAAU,eAAe;IACxB,gFAAgF;IAChF,SAAS,CAAC,EAAE,CAAC,IAAI,EAAE,KAAK,GAAG,OAAO,GAAG,QAAQ,GAAG,cAAc,KAAK,MAAM,CAAC;CAC1E;AAkBD;;;GAGG;AACH,wBAAgB,kBAAkB,CAAC,OAAO,EAAE,uBAAuB,EAAE,KAAK,CAAC,EAAE,eAAe,GAAG,eAAe,CAM7G;AAsDD,MAAM,WAAW,mBAAmB;IACnC,IAAI,EAAE,oBAAoB,CAAC;IAC3B,6CAA6C;IAC7C,IAAI,EAAE,eAAe,CAAC;IACtB,IAAI,CAAC,EAAE,OAAO,CAAC;IACf,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,sFAAsF;IACtF,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,yEAAyE;IACzE,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,gFAAgF;IAChF,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,UAAU,CAAC,EAAE,MAAM,CAAC;CACpB;AAED;;;;GAIG;AACH,wBAAgB,kBAAkB,CACjC,KAAK,EAAE,MAAM,EAAE,EACf,OAAO,EAAE,mBAAmB,GAC1B;IAAE,MAAM,EAAE,MAAM,EAAE,CAAC;IAAC,UAAU,EAAE,MAAM,CAAA;CAAE,CA0F1C;AAED,qGAAqG;AACrG,wBAAgB,aAAa,CAAC,OAAO,EAAE,OAAO,GAAG,MAAM,CAatD;AAID,yDAAyD;AACzD,MAAM,WAAW,aAAa;IAC7B,aAAa,IAAI,IAAI,CAAC;CACtB;AAED;;;;GAIG;AACH,qBAAa,aAAa;;IAUzB,SAAS,CAAC,MAAM,EAAE,MAAM,EAAE,EAAE,UAAU,EAAE,MAAM,GAAG,IAAI;IAMrD,MAAM,CAAC,GAAG,EAAE,aAAa,GAAG,IAAI;IAIhC;;4BAEwB;IACxB,aAAa,IAAI,IAAI;IAIrB;qEACiE;IACjE,QAAQ,CAAC,IAAI,EAAE,MAAM,GAAG,SAAS,GAAG,IAAI;IAIxC,KAAK,IAAI,IAAI;IAYb,IAAI,IAAI,IAAI;IAQZ,kEAAkE;IAClE,MAAM,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,EAAE;IAO/B,UAAU,IAAI,IAAI;IAIlB,OAAO,IAAI,IAAI;CAIf;AAID,MAAM,WAAW,yBAAyB;IACzC,OAAO,EAAE,OAAO,CAAC;CACjB;AAED,eAAO,MAAM,2BAA2B,EAAE,yBAA6C,CAAC;AAExF;;;GAGG;AACH,wBAAgB,gCAAgC,CAC/C,MAAM,EAAE,uBAAuB,GAAG,SAAS,EAC3C,GAAG,GAAE,MAAM,CAAC,UAAwB,GAClC,yBAAyB,CAM3B;AAED,MAAM,WAAW,cAAc;IAC9B,KAAK,CAAC,EAAE,gBAAgB,CAAC;IACzB,sBAAsB,CAAC,KAAK,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;CAC7C;AAID,4DAA4D;AAC5D,MAAM,WAAW,oBAAoB;IACpC,gFAAgF;IAChF,SAAS,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI,CAAC;IACnC;8CAC0C;IAC1C,QAAQ,CAAC,SAAS,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,GAAG,IAAI,CAAC;IACtD,6EAA6E;IAC7E,WAAW,IAAI,IAAI,CAAC;IACpB,6CAA6C;IAC7C,SAAS,IAAI,IAAI,CAAC;CAClB;AAQD;;;;;;;;GAQG;AACH,wBAAgB,2BAA2B,CAAC,cAAc,EAAE,OAAO,GAAG,oBAAoB,GAAG,SAAS,CAiDrG;AAED,MAAM,WAAW,qBAAqB;IACrC,8EAA8E;IAC9E,IAAI,CAAC,KAAK,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,gFAAgF;IAChF,IAAI,CAAC,KAAK,EAAE,MAAM,GAAG,IAAI,CAAC;IAC1B,yCAAyC;IACzC,OAAO,IAAI,IAAI,CAAC;IAChB,mEAAmE;IACnE,QAAQ,CAAC,MAAM,EAAE,SAAS,MAAM,EAAE,CAAC;CACnC;AAED,MAAM,WAAW,aAAa;IAC7B,iEAAiE;IACjE,IAAI,IAAI,IAAI,CAAC;IACb;wDACoD;IACpD,QAAQ,IAAI,MAAM,GAAG,SAAS,CAAC;IAC/B,8DAA8D;IAC9D,OAAO,IAAI,IAAI,CAAC;CAChB;AAID,8EAA8E;AAC9E,wBAAgB,sBAAsB,CAAC,SAAS,EAAE,MAAM,GAAG,MAAM,CAShE;AAED;;;;;;;GAOG;AACH,wBAAgB,mBAAmB,CAClC,QAAQ,EAAE,qBAAqB,EAC/B,GAAG,GAAE,MAAM,MAAiB,EAC5B,cAAc,SAAI,GAChB,aAAa,CAsBf;AAED;;;;;GAKG;AACH,wBAAgB,mBAAmB,CAAC,OAAO,EAAE,OAAO,GAAG,OAAO,CAI7D;AAED;;;;;;;;;;GAUG;AACH,wBAAgB,2BAA2B,CAC1C,EAAE,EAAE,cAAc,EAClB,YAAY,EAAE,wBAAwB,EACtC,WAAW,CAAC,EAAE,MAAM,EACpB,QAAQ,CAAC,EAAE,yBAAyB,GAClC,qBAAqB,CAwDvB;AAID,uFAAuF;AACvF,MAAM,WAAW,gBAAgB;IAChC,+EAA+E;IAC/E,EAAE,CAAC,EAAE,CAAC,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,KAAK,MAAM,CAAC;IAC3C,SAAS,CAAC,EAAE,CAAC,IAAI,EAAE,KAAK,GAAG,OAAO,GAAG,QAAQ,GAAG,cAAc,KAAK,MAAM,CAAC;CAC1E;AAED,MAAM,WAAW,aAAa;IAC7B,KAAK,CAAC,EAAE,gBAAgB,CAAC;IACzB,iBAAiB,CAAC,OAAO,EAAE,OAAO,GAAG,IAAI,CAAC;IAC1C,SAAS,CACR,GAAG,EAAE,MAAM,EACX,OAAO,EAAE,CAAC,CAAC,GAAG,EAAE,aAAa,EAAE,KAAK,EAAE,OAAO,KAAK,aAAa,CAAC,GAAG,SAAS,EAC5E,OAAO,CAAC,EAAE;QAAE,SAAS,CAAC,EAAE,aAAa,GAAG,aAAa,CAAA;KAAE,GACrD,IAAI,CAAC;CACR;AAED,MAAM,WAAW,eAAe;IAC/B,OAAO,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,EAAE,CAAC;CAChC;AAED,MAAM,WAAW,0BAA0B;IAC1C,kDAAkD;IAClD,KAAK,IAAI,IAAI,CAAC;IACd,8DAA8D;IAC9D,IAAI,IAAI,IAAI,CAAC;IACb,gDAAgD;IAChD,OAAO,IAAI,IAAI,CAAC;IAChB,mEAAmE;IACnE,QAAQ,CAAC,IAAI,EAAE,MAAM,GAAG,SAAS,GAAG,IAAI,CAAC;IACzC,yDAAyD;IACzD,aAAa,IAAI,IAAI,CAAC;IACtB,qDAAqD;IACrD,QAAQ,CAAC,MAAM,EAAE,SAAS,MAAM,EAAE,CAAC;CACnC;AA2CD;;;;;;;GAOG;AACH,wBAAsB,uBAAuB,CAC5C,EAAE,EAAE,aAAa,EACjB,QAAQ,EAAE,wBAAwB,EAClC,IAAI,CAAC,EAAE;IAAE,cAAc,CAAC,EAAE,MAAM,eAAe,GAAG,SAAS,GAAG,OAAO,CAAC,eAAe,GAAG,SAAS,CAAC,CAAA;CAAE,EACpG,WAAW,CAAC,EAAE,MAAM,GAClB,OAAO,CAAC,0BAA0B,CAAC,CA0DrC"}