@selesai/code 0.9.14 → 0.10.0

package/dist/extensions/pi-powerline-footer/theme.ts

@@ -1,241 +0,0 @@
-/**
- * Theme system for powerline-footer
- *
- * Colors are resolved in order:
- * 1. User overrides from theme.json (if exists)
- * 2. Preset colors
- * 3. Default colors
- */
-
-import type { Theme, ThemeColor } from "@selesai/code";
-import { existsSync, readFileSync } from "node:fs";
-import { join, dirname } from "node:path";
-import { fileURLToPath } from "node:url";
-import { getAgentPath } from "./paths.ts";
-import type { ColorScheme, ColorValue, SemanticColor, ThemeLike } from "./types.ts";
-
-export interface PowerlineThemeConfig {
-  colors?: unknown;
-  icons?: unknown;
-}
-
-// Default color scheme (uses pi theme colors)
-const DEFAULT_COLORS: Required<ColorScheme> = {
-  model: "#d787af",  // Pink/mauve (matching original colors.ts)
-  shellMode: "accent",
-  path: "#00afaf",  // Teal/cyan (matching original colors.ts)
-  gitDirty: "warning",
-  gitClean: "success",
-  thinking: "thinkingOff",
-  thinkingMinimal: "thinkingMinimal",
-  thinkingLow: "thinkingLow",
-  thinkingMedium: "thinkingMedium",
-  context: "dim",
-  contextWarn: "warning",
-  contextError: "error",
-  cost: "text",
-  tokens: "muted",
-  queue: "accent",
-  separator: "dim",
-  border: "borderMuted",
-};
-
-// Rainbow colors for high thinking levels
-const RAINBOW_COLORS = [
-  "#b281d6", "#d787af", "#febc38", "#e4c00f",
-  "#89d281", "#00afaf", "#178fb9", "#b281d6",
-];
-
-// Cache for user theme overrides
-let userThemeCache: ColorScheme | null = null;
-let userThemeCacheTime = 0;
-let userThemeCacheKey = "";
-let themeConfigCache: PowerlineThemeConfig | null = null;
-let themeConfigCacheTime = 0;
-let themeConfigCacheKey = "";
-const CACHE_TTL = 5000; // 5 seconds
-const warnedInvalidThemeColors = new Set<string>();
-
-function isRecord(value: unknown): value is Record<string, unknown> {
-  return typeof value === "object" && value !== null && !Array.isArray(value);
-}
-
-function sanitizeUserThemeOverrides(value: unknown): ColorScheme {
-  if (!isRecord(value)) {
-    return {};
-  }
-
-  const sanitized: ColorScheme = {};
-  for (const [key, rawColor] of Object.entries(value)) {
-    if (!Object.prototype.hasOwnProperty.call(DEFAULT_COLORS, key)) {
-      continue;
-    }
-    if (typeof rawColor !== "string") {
-      continue;
-    }
-
-    const color = rawColor.trim();
-    if (!color) {
-      continue;
-    }
-
-    sanitized[key as SemanticColor] = color as ColorValue;
-  }
-
-  return sanitized;
-}
-
-/**
- * Get theme.json lookup paths.
- */
-function getThemePaths(): string[] {
-  const extDir = dirname(fileURLToPath(import.meta.url));
-  return Array.from(new Set([
-    getAgentPath("extensions", "powerline-footer", "theme.json"),
-    join(extDir, "theme.json"),
-  ]));
-}
-
-/**
- * Load user theme config from theme.json.
- */
-export function loadThemeConfig(): PowerlineThemeConfig {
-  const now = Date.now();
-  const themePaths = getThemePaths();
-  const cacheKey = themePaths.join("\0");
-  if (themeConfigCache && themeConfigCacheKey === cacheKey && now - themeConfigCacheTime < CACHE_TTL) {
-    return themeConfigCache;
-  }
-
-  for (const themePath of themePaths) {
-    try {
-      if (existsSync(themePath)) {
-        const content = readFileSync(themePath, "utf-8");
-        const parsed = JSON.parse(content);
-        themeConfigCache = isRecord(parsed) ? parsed : {};
-        themeConfigCacheTime = now;
-        themeConfigCacheKey = cacheKey;
-        return themeConfigCache;
-      }
-    } catch (error) {
-      // Theme overrides are optional. If the file is unreadable or malformed,
-      // keep rendering with built-in defaults instead of breaking the footer.
-      console.debug(`[powerline-theme] Failed to load ${themePath}:`, error);
-    }
-  }
-
-  themeConfigCache = {};
-  themeConfigCacheTime = now;
-  themeConfigCacheKey = cacheKey;
-  return themeConfigCache;
-}
-
-function loadUserTheme(): ColorScheme {
-  const now = Date.now();
-  const cacheKey = getThemePaths().join("\0");
-  if (userThemeCache && userThemeCacheKey === cacheKey && now - userThemeCacheTime < CACHE_TTL) {
-    return userThemeCache;
-  }
-
-  userThemeCache = sanitizeUserThemeOverrides(loadThemeConfig().colors);
-  userThemeCacheTime = now;
-  userThemeCacheKey = cacheKey;
-  return userThemeCache;
-}
-
-/**
- * Resolve a semantic color to an actual color value
- */
-export function resolveColor(
-  semantic: SemanticColor,
-  presetColors?: ColorScheme
-): ColorValue {
-  const userTheme = loadUserTheme();
-
-  // Priority: user overrides > preset colors > defaults
-  return userTheme[semantic]
-    ?? presetColors?.[semantic]
-    ?? DEFAULT_COLORS[semantic];
-}
-
-/**
- * Check if a color value is a hex color
- */
-function isHexColor(color: ColorValue): color is `#${string}` {
-  return typeof color === "string" && /^#[0-9a-fA-F]{6}$/.test(color);
-}
-
-/**
- * Convert hex color to ANSI escape code
- */
-function hexToAnsi(hex: string): string {
-  const h = hex.replace("#", "");
-  const r = parseInt(h.slice(0, 2), 16);
-  const g = parseInt(h.slice(2, 4), 16);
-  const b = parseInt(h.slice(4, 6), 16);
-  return `\x1b[38;2;${r};${g};${b}m`;
-}
-
-/**
- * Apply a color to text using the pi theme or custom hex
- */
-export function applyColor(
-  theme: ThemeLike,
-  color: ColorValue,
-  text: string
-): string {
-  if (isHexColor(color)) {
-    return `${hexToAnsi(color)}${text}\x1b[0m`;
-  }
-
-  try {
-    return theme.fg(color as ThemeColor, text);
-  } catch (error) {
-    const key = String(color);
-    if (!warnedInvalidThemeColors.has(key)) {
-      warnedInvalidThemeColors.add(key);
-      if (warnedInvalidThemeColors.size > 200) {
-        warnedInvalidThemeColors.clear();
-      }
-      console.debug(`[powerline-theme] Invalid theme color "${key}"; falling back to "text".`, error);
-    }
-    return theme.fg("text", text);
-  }
-}
-
-/**
- * Apply a semantic color to text
- */
-export function fg(
-  theme: ThemeLike,
-  semantic: SemanticColor,
-  text: string,
-  presetColors?: ColorScheme
-): string {
-  const color = resolveColor(semantic, presetColors);
-  return applyColor(theme, color, text);
-}
-
-/**
- * Apply rainbow gradient to text (for high thinking levels)
- */
-export function rainbow(text: string): string {
-  let result = "";
-  let colorIndex = 0;
-  for (const char of text) {
-    if (char === " " || char === ":") {
-      result += char;
-    } else {
-      result += hexToAnsi(RAINBOW_COLORS[colorIndex % RAINBOW_COLORS.length]) + char;
-      colorIndex++;
-    }
-  }
-  return result + "\x1b[0m";
-}
-
-/**
- * Get the default color scheme
- */
-export function getDefaultColors(): Required<ColorScheme> {
-  return { ...DEFAULT_COLORS };
-}

package/dist/extensions/pi-powerline-footer/token-stats.ts

@@ -1,304 +0,0 @@
-import type { AssistantMessage } from "@earendil-works/pi-ai";
-
-type SessionAssistantUsage = AssistantMessage["usage"];
-
-export interface SessionTokenStats {
-  input: number;
-  output: number;
-  cacheRead: number;
-  cacheWrite: number;
-  cost: number;
-  subagentCost: number;
-  lastAssistant: AssistantMessage | undefined;
-  thinkingLevelFromSession: string | null;
-}
-
-function isRecord(value: unknown): value is Record<string, unknown> {
-  return typeof value === "object" && value !== null && !Array.isArray(value);
-}
-
-function hasSessionAssistantUsage(value: unknown): value is SessionAssistantUsage {
-  if (!isRecord(value)) {
-    return false;
-  }
-
-  if (
-    typeof value.input !== "number" ||
-    typeof value.output !== "number" ||
-    typeof value.cacheRead !== "number" ||
-    typeof value.cacheWrite !== "number"
-  ) {
-    return false;
-  }
-
-  return isRecord(value.cost) && typeof value.cost.total === "number";
-}
-
-function isSessionAssistantMessage(value: unknown): value is AssistantMessage {
-  return isRecord(value)
-    && value.role === "assistant"
-    && hasSessionAssistantUsage(value.usage)
-    && (value.stopReason === undefined || typeof value.stopReason === "string");
-}
-
-function getUsageTokenTotal(usage: SessionAssistantUsage): number {
-  const totalTokens = "totalTokens" in usage && typeof usage.totalTokens === "number" ? usage.totalTokens : 0;
-  return totalTokens || usage.input + usage.output + usage.cacheRead + usage.cacheWrite;
-}
-
-// Matches pi-subagents' internal SLASH_RESULT_TYPE custom_message marker.
-const SUBAGENT_SLASH_RESULT_TYPE = "subagent-slash-result";
-
-function subagentDetailsFromSessionEntry(e: Record<string, unknown>): { results: unknown[] } | undefined {
-  if (e.type === "custom_message" && e.customType === SUBAGENT_SLASH_RESULT_TYPE) {
-    const details = isRecord(e.details) ? e.details : undefined;
-    const result = isRecord(details?.result) ? details.result : undefined;
-    const inner = isRecord(result?.details) ? result.details : undefined;
-    return Array.isArray(inner?.results) ? { results: inner.results } : undefined;
-  }
-  if (e.type === "message" && isRecord(e.message)) {
-    const message = e.message as { role?: unknown; toolName?: unknown; details?: unknown };
-    if (message.role === "toolResult" && message.toolName === "subagent" && isRecord(message.details) && Array.isArray(message.details.results)) {
-      return { results: message.details.results };
-    }
-  }
-  return undefined;
-}
-
-// Sums subagent child usage cost recorded on a single session entry (parallel/chain/single runs
-// launched via the subagent tool or /parallel, /worker etc. slash commands), so the footer can
-// show total spend rather than only the interactive parent session's cost.
-function extractSubagentResultCost(e: Record<string, unknown>): number {
-  const details = subagentDetailsFromSessionEntry(e);
-  if (!details) return 0;
-  let total = 0;
-  for (const result of details.results) {
-    if (!isRecord(result)) continue;
-    const usage = isRecord(result.usage) ? result.usage : undefined;
-    if (typeof usage?.cost === "number") total += usage.cost;
-  }
-  return total;
-}
-
-/**
- * Fingerprint of the fields on a session event that influence the aggregated
- * stats. Comparing only the event count is not enough: while streaming, pi
- * may update the trailing assistant message in place (usage grows, stopReason
- * flips), so the same event reference can carry fresh numbers.
- */
-function eventStatsSignature(event: unknown): string {
-  if (!isRecord(event)) {
-    return "?";
-  }
-
-  if (event.type === "thinking_level_change") {
-    return `t:${typeof event.thinkingLevel === "string" ? event.thinkingLevel : ""}`;
-  }
-
-  const subagentDetails = subagentDetailsFromSessionEntry(event);
-  if (subagentDetails) {
-    return `s:${subagentDetails.results.length}:${extractSubagentResultCost(event)}`;
-  }
-
-  if (event.type === "message" && isRecord(event.message)) {
-    const message = event.message;
-    const role = typeof message.role === "string" ? message.role : "";
-    const stopReason = typeof message.stopReason === "string" ? message.stopReason : "";
-    if (role === "assistant" && hasSessionAssistantUsage(message.usage)) {
-      const usage = message.usage;
-      return `a:${stopReason}:${usage.input}/${usage.output}/${usage.cacheRead}/${usage.cacheWrite}/${usage.cost.total}`;
-    }
-    return `m:${role}:${stopReason}`;
-  }
-
-  return `e:${typeof event.type === "string" ? event.type : "?"}`;
-}
-
-function emptySessionTokenStats(): SessionTokenStats {
-  return {
-    input: 0,
-    output: 0,
-    cacheRead: 0,
-    cacheWrite: 0,
-    cost: 0,
-    subagentCost: 0,
-    lastAssistant: undefined,
-    thinkingLevelFromSession: null,
-  };
-}
-
-function copySessionTokenStats(stats: SessionTokenStats): SessionTokenStats {
-  return { ...stats };
-}
-
-function accumulateSessionEvent(stats: SessionTokenStats, event: unknown): void {
-  if (!isRecord(event)) return;
-
-  if (event.type === "thinking_level_change" && typeof event.thinkingLevel === "string") {
-    stats.thinkingLevelFromSession = event.thinkingLevel;
-  }
-
-  stats.subagentCost += extractSubagentResultCost(event);
-
-  if (event.type !== "message" || !isSessionAssistantMessage(event.message)) return;
-
-  const message = event.message;
-  if (message.stopReason === "error" || message.stopReason === "aborted") return;
-
-  stats.input += message.usage.input;
-  stats.output += message.usage.output;
-  stats.cacheRead += message.usage.cacheRead;
-  stats.cacheWrite += message.usage.cacheWrite;
-  stats.cost += message.usage.cost.total;
-  if (getUsageTokenTotal(message.usage) > 0) {
-    stats.lastAssistant = message;
-  }
-}
-
-export interface SessionBranchProvider {
-  getLeafId(): string | null;
-  getBranch(): readonly unknown[];
-}
-
-function isSessionBranchProvider(value: unknown): value is SessionBranchProvider {
-  return isRecord(value) && typeof value.getLeafId === "function" && typeof value.getBranch === "function";
-}
-
-/**
- * Session branches are immutable for a given leaf. Keep the already-built path
- * while streaming updates mutate only its trailing message in place.
- */
-export class SessionBranchCache {
-  private provider: SessionBranchProvider | null = null;
-  private leafId: string | null = null;
-  private branch: readonly unknown[] = [];
-
-  get(source: unknown): readonly unknown[] {
-    if (!isSessionBranchProvider(source)) return [];
-
-    const provider = source;
-    const leafId = provider.getLeafId();
-    // A reset cache holds provider === null, which never matches a live provider.
-    if (this.provider !== provider || this.leafId !== leafId) {
-      this.provider = provider;
-      this.leafId = leafId;
-      this.branch = provider.getBranch();
-    }
-    return this.branch;
-  }
-
-  reset(): void {
-    this.provider = null;
-    this.leafId = null;
-    this.branch = [];
-  }
-}
-
-export function computeSessionTokenStats(sessionEvents: readonly unknown[]): SessionTokenStats {
-  let input = 0, output = 0, cacheRead = 0, cacheWrite = 0, cost = 0, subagentCost = 0;
-  let lastAssistant: AssistantMessage | undefined;
-  let thinkingLevelFromSession: string | null = null;
-
-  for (const e of sessionEvents) {
-    if (!isRecord(e)) continue;
-
-    if (e.type === "thinking_level_change" && typeof e.thinkingLevel === "string") {
-      thinkingLevelFromSession = e.thinkingLevel;
-    }
-
-    subagentCost += extractSubagentResultCost(e);
-
-    if (e.type !== "message" || !isSessionAssistantMessage(e.message)) continue;
-
-    const m = e.message;
-    if (m.stopReason === "error" || m.stopReason === "aborted") continue;
-
-    input += m.usage.input;
-    output += m.usage.output;
-    cacheRead += m.usage.cacheRead;
-    cacheWrite += m.usage.cacheWrite;
-    cost += m.usage.cost.total;
-    if (getUsageTokenTotal(m.usage) > 0) {
-      lastAssistant = m;
-    }
-  }
-
-  return { input, output, cacheRead, cacheWrite, cost, subagentCost, lastAssistant, thinkingLevelFromSession };
-}
-
-/**
- * Cache for token counting: avoid re-scanning the full session event list on
- * every render (250ms-1s cadence while streaming). Historical session entries
- * are append-only, so appended entries are accumulated and an in-place update
- * to the trailing streaming entry only recomputes that entry.
- */
-export class SessionTokenStatsCache {
-  private eventCount = -1;
-  private lastEvent: unknown;
-  private lastSignature = "";
-  private prefixStats: SessionTokenStats | null = null;
-  private stats: SessionTokenStats | null = null;
-
-  get(sessionEvents: readonly unknown[]): SessionTokenStats {
-    const eventCount = sessionEvents.length;
-    const lastEvent = eventCount > 0 ? sessionEvents[eventCount - 1] : undefined;
-    const lastSignature = eventStatsSignature(lastEvent);
-
-    if (
-      this.stats !== null
-      && this.eventCount === eventCount
-      && this.lastEvent === lastEvent
-      && this.lastSignature === lastSignature
-    ) {
-      return this.stats;
-    }
-
-    // getBranch() walks unique parent links, so an identical trailing event at
-    // the previous count proves the whole previous array is an unchanged prefix.
-    const previousStats = this.stats;
-    const canExtendPreviousStats = previousStats !== null
-      && eventCount > this.eventCount
-      && (this.eventCount === 0 || (
-        sessionEvents[this.eventCount - 1] === this.lastEvent
-        && eventStatsSignature(sessionEvents[this.eventCount - 1]) === this.lastSignature
-      ));
-
-    let prefixStats: SessionTokenStats;
-    if (canExtendPreviousStats && previousStats !== null) {
-      prefixStats = copySessionTokenStats(previousStats);
-      for (let index = this.eventCount; index < eventCount - 1; index += 1) {
-        accumulateSessionEvent(prefixStats, sessionEvents[index]);
-      }
-    } else if (
-      this.prefixStats !== null
-      && this.eventCount === eventCount
-      && this.lastEvent === lastEvent
-    ) {
-      // Same array with an in-place tail mutation: reuse the cached prefix.
-      prefixStats = this.prefixStats;
-    } else {
-      prefixStats = emptySessionTokenStats();
-      for (let index = 0; index < eventCount - 1; index += 1) {
-        accumulateSessionEvent(prefixStats, sessionEvents[index]);
-      }
-    }
-
-    const stats = copySessionTokenStats(prefixStats);
-    if (eventCount > 0) accumulateSessionEvent(stats, lastEvent);
-
-    this.eventCount = eventCount;
-    this.lastEvent = lastEvent;
-    this.lastSignature = lastSignature;
-    this.prefixStats = prefixStats;
-    this.stats = stats;
-    return stats;
-  }
-
-  reset(): void {
-    this.eventCount = -1;
-    this.lastEvent = undefined;
-    this.lastSignature = "";
-    this.prefixStats = null;
-    this.stats = null;
-  }
-}

package/dist/extensions/pi-powerline-footer/tsconfig.json

@@ -1,13 +0,0 @@
-{
-  "compilerOptions": {
-    "target": "ES2022",
-    "module": "NodeNext",
-    "moduleResolution": "NodeNext",
-    "strict": true,
-    "noEmit": true,
-    "skipLibCheck": true,
-    "types": ["node"],
-    "allowImportingTsExtensions": true
-  },
-  "include": ["*.ts", "bash-mode/**/*.ts", "queue/**/*.ts"]
-}