@thoth-agents/pi-thoth-theme 0.1.0 → 0.3.0

package/src/status-line/index.ts

@@ -1,242 +0,0 @@
-import type {
-  ExtensionAPI,
-  ExtensionContext,
-} from '@earendil-works/pi-coding-agent';
-import {
-  decorateEditor,
-  type EditorDecoration,
-} from '../input-box/decorate.ts';
-import { createWorkingState } from '../input-box/state.ts';
-import type { ThemeConfig } from '../shared/config.ts';
-import { calculateSessionCost } from './cost.ts';
-import { type ActiveThemeLike, formatCwd, renderStatusLine } from './layout.ts';
-import type { StatusSnapshot, StatusSnapshotProvider } from './snapshot.ts';
-import { createThroughputTracker } from './throughput.ts';
-import { calculateSessionTokens } from './tokens.ts';
-
-export { calculateSessionCost, formatCost } from './cost.ts';
-export {
-  type ActiveThemeLike,
-  type ContextUsageInfo,
-  formatCwd,
-  formatTokens,
-  type RenderStatusLineOptions,
-  renderStatusLine,
-  type StatusData,
-} from './layout.ts';
-export type { StatusSnapshot, StatusSnapshotProvider } from './snapshot.ts';
-export { calculateSessionTokens, type SessionTokenTotals } from './tokens.ts';
-
-/**
- * Renders the cached status row below the input box and discovers editors in place.
- * Session aggregation is cached until an event or session leaf change; this
- * extension never registers an editor factory with setEditorComponent.
- */
-export function registerStatusLine(
-  pi: ExtensionAPI,
-  ctx: ExtensionContext,
-  config: ThemeConfig,
-): void {
-  const subscriptionProviders = config.statusLine.subscriptionProviders ?? [
-    'claude-bridge',
-  ];
-  const inputBoxEnabled = config.inputBox?.enabled !== false;
-  ctx.ui.setFooter((tui, theme, footerData) => {
-    const unsubs: Array<() => void> = [];
-    let disposed = false;
-    const throughput = createThroughputTracker();
-    let sessionLeafId = ctx.sessionManager.getLeafId();
-    let session = readSession();
-    let sessionDirty = false;
-    let subagentCost = 0;
-    let cachedKey = '';
-    let cachedLines: string[] = [];
-    let snapshotKey = '';
-    let snapshot: Readonly<StatusSnapshot>;
-
-    // Cheap live fields are sampled by either consumer, never by footer order.
-    // Checking the leaf is O(1); unchanged frames do not copy/aggregate entries.
-    const getStatusSnapshot: StatusSnapshotProvider = () => {
-      if (!disposed) {
-        const leafId = ctx.sessionManager.getLeafId();
-        if (sessionDirty || leafId !== sessionLeafId) {
-          session = readSession();
-          sessionLeafId = leafId;
-          sessionDirty = false;
-        }
-      }
-      const data: StatusSnapshot = {
-        modelName: ctx.model?.name,
-        modelId: ctx.model?.id,
-        thinkingLevel: ctx.thinkingLevel,
-        gitBranch: footerData?.getGitBranch?.() ?? null,
-        cwd: formatCwd(ctx.cwd, process.env.HOME || process.env.USERPROFILE),
-        ...session,
-        subagentCost,
-        tokensPerSecond: throughput.tokensPerSecond,
-      };
-      const key = JSON.stringify(data);
-      if (key !== snapshotKey) {
-        snapshotKey = key;
-        snapshot = data;
-      }
-      return snapshot;
-    };
-
-    function readSession() {
-      const usage = ctx.getContextUsage?.();
-      const provider = ctx.model?.provider;
-      const entries = ctx.sessionManager.getEntries();
-      return {
-        cost: calculateSessionCost(entries),
-        tokenTotals: calculateSessionTokens(entries),
-        isSubscription:
-          provider !== undefined && subscriptionProviders.includes(provider),
-        contextTokens: usage?.tokens ?? null,
-        contextPercent: usage?.percent ?? null,
-        contextWindow: usage?.contextWindow,
-      };
-    }
-
-    const requestRender = () => {
-      if (!disposed) tui.requestRender?.();
-    };
-    const invalidateRender = () => {
-      if (disposed) return;
-      cachedKey = '';
-      requestRender();
-    };
-    const working = createWorkingState(invalidateRender);
-    const editorDecorations = new Set<EditorDecoration>();
-    const refreshSession = () => {
-      if (disposed) return;
-      // message_end reaches extensions before Pi persists the entry. An early
-      // consumer may read now; the leaf key still detects the later append.
-      sessionDirty = true;
-      invalidateRender();
-    };
-
-    unsubs.push(
-      pi.events.on('thoth:subagent-usage', (data) => {
-        if (disposed || !data || typeof data !== 'object') return;
-        const { parentSessionId, totalCost } = data as {
-          parentSessionId?: unknown;
-          totalCost?: unknown;
-        };
-        if (
-          parentSessionId !== ctx.sessionManager.getSessionId() ||
-          typeof totalCost !== 'number' ||
-          !Number.isFinite(totalCost) ||
-          totalCost < 0
-        )
-          return;
-        // The publisher sends a cumulative snapshot, not a delta.
-        subagentCost = totalCost;
-        requestRender();
-      }),
-    );
-
-    if (footerData?.onBranchChange) {
-      unsubs.push(footerData.onBranchChange(invalidateRender));
-    }
-
-    const requestSubagentUsage = () => {
-      pi.events.emit('thoth:subagent-usage:request', {
-        parentSessionId: ctx.sessionManager.getSessionId(),
-      });
-    };
-
-    if (pi?.on) {
-      unsubs.push(
-        pi.on('session_start', () => {
-          if (disposed) return;
-          subagentCost = 0;
-          throughput.reset();
-          if (inputBoxEnabled) working.end();
-          refreshSession();
-          requestSubagentUsage();
-        }),
-      );
-      if (inputBoxEnabled) {
-        unsubs.push(pi.on('agent_start', working.start));
-        unsubs.push(pi.on('agent_end', working.end));
-      }
-      unsubs.push(
-        pi.on('message_start', (event) => {
-          if (!disposed && event) throughput.observe(event);
-        }),
-      );
-      unsubs.push(
-        pi.on('message_end', (event) => {
-          if (disposed) return;
-          if (event) throughput.observe(event);
-          refreshSession();
-        }),
-      );
-      unsubs.push(pi.on('turn_end', refreshSession));
-      unsubs.push(pi.on('agent_end', refreshSession));
-      unsubs.push(pi.on('session_compact', refreshSession));
-      unsubs.push(pi.on('model_select', refreshSession));
-      unsubs.push(pi.on('thinking_level_select', invalidateRender));
-    }
-
-    // Subscribe first: a request responder can publish synchronously.
-    requestSubagentUsage();
-
-    return {
-      getStatusSnapshot,
-      render(width: number): string[] {
-        if (!disposed && inputBoxEnabled) {
-          // Discovery must precede cached-row returns: focus and editor identity
-          // can change without any status data changing.
-          const focusedTui = tui as typeof tui & {
-            getFocusedComponent?: () => unknown;
-          };
-          const editor = focusedTui.getFocusedComponent?.();
-          const decoration = decorateEditor(editor, {
-            theme: theme as ActiveThemeLike,
-            working,
-            getStatusSnapshot,
-            get iconMode() {
-              return config.icons;
-            },
-          });
-          if (decoration && !editorDecorations.has(decoration)) {
-            editorDecorations.add(decoration);
-            requestRender();
-          }
-        }
-        if (width <= 0) return [];
-        const data = getStatusSnapshot();
-        const key = `${width}\u0000${JSON.stringify(data)}`;
-        if (key === cachedKey) return cachedLines;
-        const line = renderStatusLine(data, {
-          width,
-          mode: config.icons,
-          theme: theme as ActiveThemeLike,
-        });
-        cachedKey = key;
-        // Working ticks invalidate styling, but unchanged footer output keeps
-        // its identity and avoids needless downstream updates.
-        if (line !== cachedLines[0]) cachedLines = line ? [line] : [];
-        return cachedLines;
-      },
-      invalidate: invalidateRender,
-      dispose() {
-        // Pi may have snapshotted lifecycle handlers before replacing the footer.
-        disposed = true;
-        working.dispose();
-        for (const decoration of editorDecorations) decoration.dispose();
-        editorDecorations.clear();
-        for (const unsub of unsubs) {
-          try {
-            unsub();
-          } catch {
-            // ignore disposal errors during cleanup
-          }
-        }
-        unsubs.length = 0;
-      },
-    };
-  });
-}

package/src/status-line/layout.ts

@@ -1,272 +0,0 @@
-import { isAbsolute, relative, resolve, sep } from 'node:path';
-import { truncateToWidth, visibleWidth } from '@earendil-works/pi-tui';
-import type { IconMode } from '../shared/config.ts';
-import { icon } from '../shared/icons.ts';
-import type { SessionTokenTotals } from './tokens.ts';
-
-export interface ContextUsageInfo {
-  percent?: number | null;
-  tokens?: number | null;
-  contextWindow?: number;
-}
-
-export interface StatusData {
-  modelName?: string;
-  modelId?: string;
-  thinkingLevel?: string;
-  gitBranch?: string | null;
-  /** Working directory formatted for display, including home abbreviation. */
-  cwd?: string;
-  contextTokens?: number | null;
-  contextWindow?: number;
-  contextPercent?: number | null;
-  /** Parent session cost; subagent cost is accounted for separately. */
-  cost?: number;
-  subagentCost?: number;
-  isSubscription?: boolean;
-}
-
-export interface ActiveThemeLike {
-  fg?: (token: string, text: string) => string;
-}
-
-export interface RenderStatusLineOptions {
-  width: number;
-  mode: IconMode;
-  theme?: ActiveThemeLike;
-}
-
-function themeFg(
-  theme: ActiveThemeLike | undefined,
-  token: string,
-  text: string,
-): string {
-  return theme?.fg ? theme.fg(token, text) : text;
-}
-
-export function formatCwd(cwd: string, home?: string): string {
-  if (!home) return cwd;
-  const relativeToHome = relative(resolve(home), resolve(cwd));
-  if (
-    relativeToHome === '..' ||
-    relativeToHome.startsWith(`..${sep}`) ||
-    isAbsolute(relativeToHome)
-  ) {
-    return cwd;
-  }
-  return relativeToHome === '' ? '~' : `~${sep}${relativeToHome}`;
-}
-
-export function formatTokens(count: number): string {
-  if (count < 1000) return count.toString();
-  if (count < 1000000) {
-    const k = (count / 1000).toFixed(1).replace(/\.0$/, '');
-    return `${k}K`;
-  }
-  const m = (count / 1000000).toFixed(1).replace(/\.0$/, '');
-  return `${m}M`;
-}
-
-function thinkingToken(level: string): string {
-  const cap = level.charAt(0).toUpperCase() + level.slice(1);
-  return `thinking${cap}`;
-}
-
-export interface SegmentOptions {
-  mode?: IconMode;
-  theme?: ActiveThemeLike;
-}
-
-/** Git branch piece for the input-box border; empty when no branch. */
-export function formatBranchSegment(
-  data: StatusData,
-  { mode = 'nerd', theme }: SegmentOptions = {},
-): string {
-  if (!data.gitBranch) return '';
-  return `${themeFg(theme, 'dim', icon('separator', mode))} ${themeFg(
-    theme,
-    'success',
-    `${icon('branch', mode)} ${data.gitBranch}`,
-  )}`;
-}
-
-/** Model then optional effort pieces; each piece starts a droppable region. */
-export function formatModelSegments(
-  data: StatusData,
-  { mode = 'nerd', theme }: SegmentOptions = {},
-): string[] {
-  const pieces: string[] = [];
-  const model = data.modelName || data.modelId;
-  if (model) {
-    pieces.push(themeFg(theme, 'mdLink', `${icon('model', mode)} ${model}`));
-  }
-  const level = data.thinkingLevel;
-  if (level && level !== 'off') {
-    const label =
-      level === 'minimal' ? 'min' : level === 'medium' ? 'med' : level;
-    const effort = themeFg(
-      theme,
-      thinkingToken(level),
-      `${icon('effort', mode)} ${label}`,
-    );
-    pieces.push(
-      model
-        ? `${themeFg(theme, 'dim', icon('separator', mode))} ${effort}`
-        : effort,
-    );
-  }
-  return pieces;
-}
-
-/** Working directory with folder icon, richest first: full path then `…/leaf`. */
-export function formatCwdSegments(
-  data: StatusData,
-  { mode = 'nerd', theme }: SegmentOptions = {},
-): string[] {
-  const cwd = data.cwd;
-  if (!cwd) return [];
-  const folder = icon('folder', mode);
-  const variants = [`${folder} ${cwd}`];
-  const leaf = cwd.split(/[/\\]/).filter(Boolean).pop();
-  const separator = cwd.includes('\\') && !cwd.includes('/') ? '\\' : '/';
-  const isDriveRoot = leaf !== undefined && /^[A-Za-z]:$/.test(leaf);
-  if (leaf && leaf !== cwd && leaf !== '~' && !isDriveRoot) {
-    variants.push(`${folder} ${icon('ellipsis', mode)}${separator}${leaf}`);
-  }
-  return variants.map((text) => themeFg(theme, 'muted', text));
-}
-
-function contextBar(
-  percent: number,
-  isAscii: boolean,
-  theme: ActiveThemeLike | undefined,
-): { bar: string; pct: string } {
-  let token = 'success';
-  if (percent >= 90) token = 'error';
-  else if (percent >= 70) token = 'warning';
-  const filled = Math.min(10, Math.max(0, Math.round((percent / 100) * 10)));
-  const fillChar = isAscii ? '#' : '█';
-  const emptyChar = isAscii ? '-' : '░';
-  const dim = (text: string) => themeFg(theme, 'dim', text);
-  return {
-    bar: `${dim('[')}${themeFg(theme, token, fillChar.repeat(filled))}${dim(
-      emptyChar.repeat(10 - filled),
-    )}${dim(']')}`,
-    pct: themeFg(theme, token, `${Math.round(percent)}%`),
-  };
-}
-
-/**
- * Context usage, richest first: bar + percent + tokens, percent + tokens,
- * tokens. Unavailable usage collapses to a dash.
- */
-export function formatContextSegments(
-  data: StatusData,
-  { mode = 'nerd', theme }: SegmentOptions = {},
-): string[] {
-  const contextIcon = themeFg(theme, 'muted', icon('context', mode));
-  const windowText = data.contextWindow
-    ? `/${formatTokens(data.contextWindow)}`
-    : '';
-  const hasTokens =
-    data.contextTokens !== null && data.contextTokens !== undefined;
-  const hasPercent =
-    data.contextPercent !== null && data.contextPercent !== undefined;
-  const tokens = themeFg(
-    theme,
-    'muted',
-    `${hasTokens ? formatTokens(data.contextTokens as number) : '—'}${windowText}`,
-  );
-  if (!hasPercent) {
-    const hasAny =
-      hasTokens ||
-      windowText !== '' ||
-      data.contextPercent !== undefined ||
-      data.contextTokens !== undefined;
-    return hasAny
-      ? [`${contextIcon} ${tokens}`]
-      : [themeFg(theme, 'muted', '—')];
-  }
-  const { bar, pct } = contextBar(
-    data.contextPercent as number,
-    mode === 'ascii',
-    theme,
-  );
-  return [
-    `${contextIcon} ${bar} ${pct} ${tokens}`,
-    `${contextIcon} ${pct} ${tokens}`,
-    `${contextIcon} ${tokens}`,
-  ];
-}
-
-export interface FooterData extends StatusData {
-  tokenTotals?: SessionTokenTotals;
-  /** `null` means not yet measured; `undefined` hides the segment. */
-  tokensPerSecond?: number | null;
-}
-
-function formatRate(rate: number | null): string {
-  if (rate === null || !Number.isFinite(rate)) return '—';
-  return rate < 10 ? rate.toFixed(1) : Math.round(rate).toString();
-}
-
-/**
- * One-row footer below the editor: cost, total input/output, cache hit %, tok/s.
- * Lowest priority drops first: tok/s, cache, tokens; cost truncates last.
- */
-export function renderStatusLine(
-  data: FooterData,
-  options: RenderStatusLineOptions,
-): string {
-  const { width, mode, theme } = options;
-  if (width <= 0) return '';
-
-  const sep = ` ${themeFg(theme, 'border', icon('separator', mode))} `;
-  const segments: string[] = [];
-
-  if (data.cost !== undefined) {
-    const total = data.cost + (data.subagentCost ?? 0);
-    segments.push(
-      themeFg(
-        theme,
-        'accent',
-        `${icon('cost', mode)}${mode === 'ascii' ? '' : ' '}${total.toFixed(3)}${data.isSubscription ? ' (sub)' : ''}`,
-      ),
-    );
-  }
-  const totals = data.tokenTotals;
-  if (totals) {
-    const totalInput = totals.input + totals.cacheRead + totals.cacheWrite;
-    const cacheHitRatio =
-      totalInput === 0
-        ? '—'
-        : `${Math.round((totals.cacheRead / totalInput) * 100)}%`;
-    const up = icon('tokensIn', mode);
-    const down = icon('tokensOut', mode);
-    segments.push(
-      themeFg(
-        theme,
-        'muted',
-        `${up}${formatTokens(totalInput)} ${down}${formatTokens(totals.output)}`,
-      ),
-      themeFg(theme, 'muted', `${icon('cache', mode)} ${cacheHitRatio}`),
-    );
-  }
-  if (data.tokensPerSecond !== undefined) {
-    segments.push(
-      themeFg(
-        theme,
-        'muted',
-        mode === 'ascii'
-          ? `${formatRate(data.tokensPerSecond)} ${icon('throughput', mode)}`
-          : `${icon('throughput', mode)} ${formatRate(data.tokensPerSecond)} tok/s`,
-      ),
-    );
-  }
-
-  for (let count = segments.length; count >= 1; count--) {
-    const line = segments.slice(0, count).join(sep);
-    if (visibleWidth(line) <= width) return line;
-  }
-  return segments.length ? truncateToWidth(segments[0], width, '') : '';
-}

package/src/status-line/snapshot.ts

@@ -1,11 +0,0 @@
-import type { StatusData } from './layout.ts';
-import type { SessionTokenTotals } from './tokens.ts';
-
-/** Shared nonvisual data for the footer and in-place editor decoration. */
-export interface StatusSnapshot extends StatusData {
-  readonly tokenTotals: SessionTokenTotals;
-  /** Runtime-measured main-session output tokens per generation second. */
-  readonly tokensPerSecond: number | null;
-}
-
-export type StatusSnapshotProvider = () => Readonly<StatusSnapshot>;

package/src/status-line/throughput.ts

@@ -1,61 +0,0 @@
-export interface GenerationEvent {
-  readonly type: 'message_start' | 'message_end';
-  readonly message: {
-    readonly role: string;
-    readonly id?: string;
-    readonly usage?: { readonly output: number };
-  };
-}
-
-/** Runtime-only main-session assistant generation timing. */
-export function createThroughputTracker() {
-  let pending: { at: number; id?: string } | undefined;
-  let outputTokens = 0;
-  let generationMs = 0;
-  const startedIds = new Set<string>();
-  const endedIds = new Set<string>();
-  let startedMessages = new WeakSet<object>();
-  let endedMessages = new WeakSet<object>();
-
-  return {
-    observe(event: GenerationEvent, observedAt = Date.now()): void {
-      if (event.message.role !== 'assistant') return;
-      const id = event.message.id;
-      if (event.type === 'message_start') {
-        if (startedMessages.has(event.message) || (id && startedIds.has(id)))
-          return;
-        startedMessages.add(event.message);
-        if (id) startedIds.add(id);
-        pending = { at: observedAt, id };
-        return;
-      }
-      if (endedMessages.has(event.message) || (id && endedIds.has(id))) return;
-      const start = pending;
-      if (!start || (start.id && id && start.id !== id)) return;
-      pending = undefined;
-      endedMessages.add(event.message);
-      if (id) endedIds.add(id);
-      const elapsed = observedAt - start.at;
-      if (!Number.isFinite(elapsed) || elapsed <= 0) return;
-      const output = event.message.usage?.output;
-      if (typeof output !== 'number' || !Number.isFinite(output) || output < 0)
-        return;
-      outputTokens += output;
-      generationMs += elapsed;
-    },
-    reset(): void {
-      pending = undefined;
-      outputTokens = 0;
-      generationMs = 0;
-      startedIds.clear();
-      endedIds.clear();
-      startedMessages = new WeakSet<object>();
-      endedMessages = new WeakSet<object>();
-    },
-    get tokensPerSecond(): number | null {
-      if (!Number.isFinite(generationMs) || generationMs <= 0) return null;
-      const speed = (outputTokens * 1000) / generationMs;
-      return Number.isFinite(speed) ? speed : null;
-    },
-  };
-}

package/src/status-line/tokens.ts

@@ -1,51 +0,0 @@
-import type { SessionEntry } from '@earendil-works/pi-coding-agent';
-import type { SessionEntriesSource } from './cost.ts';
-
-export interface SessionTokenTotals {
-  readonly input: number;
-  readonly output: number;
-  readonly cacheRead: number;
-  readonly cacheWrite: number;
-}
-
-/** Main-session totals across the same entry types and history scope as cost. */
-export function calculateSessionTokens(
-  source: readonly SessionEntry[] | SessionEntriesSource | undefined | null,
-): SessionTokenTotals {
-  const entries: readonly SessionEntry[] = Array.isArray(source)
-    ? source
-    : source &&
-        typeof (source as SessionEntriesSource).getEntries === 'function'
-      ? (source as SessionEntriesSource).getEntries()
-      : [];
-  const totals = { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 };
-  for (const entry of entries) {
-    if (!entry) continue;
-    const usage =
-      entry.type === 'message'
-        ? (entry.message?.role === 'assistant' ||
-            entry.message?.role === 'toolResult') &&
-          'usage' in entry.message
-          ? entry.message.usage
-          : undefined
-        : (entry.type === 'branch_summary' ||
-              entry.type === 'compaction' ||
-              entry.type === 'usage') &&
-            'usage' in entry
-          ? entry.usage
-          : undefined;
-    if (!usage) continue;
-    for (const field of [
-      'input',
-      'output',
-      'cacheRead',
-      'cacheWrite',
-    ] as const) {
-      const value = usage[field];
-      if (typeof value === 'number' && Number.isFinite(value) && value >= 0) {
-        totals[field] += value;
-      }
-    }
-  }
-  return totals;
-}