pi-powerline-footer 0.2.12 → 0.2.13

package/CHANGELOG.md

@@ -2,6 +2,21 @@
 
 ## [Unreleased]
 
+## [0.2.13] - 2026-01-27
+
+### Added
+- **Theme system** — Colors now integrate with pi's theme system instead of hardcoded values
+- Each preset defines its own color scheme with semantic color names
+- Optional `theme.json` file for user customization (power user feature)
+- Colors can be theme names (`accent`, `primary`, `muted`) or hex values (`#ff5500`)
+- Added `theme.example.json` documenting all available color options
+
+### Changed
+- Segments now use pi's `Theme` object for color rendering
+- Removed hardcoded ANSI color codes in favor of theme-based colors
+- Presets include both layout AND color scheme for cohesive looks
+- Simplified thinking level colors to use semantic `thinking` color (rainbow preserved for high/xhigh)
+
 ## [0.2.12] - 2026-01-27
 
 ### Added

package/README.md

@@ -75,3 +75,44 @@ Configure via preset options: `path: { mode: "full" }`
 ## Separators
 
 `powerline` · `powerline-thin` · `slash` · `pipe` · `dot` · `chevron` · `star` · `block` · `none` · `ascii`
+
+## Theming
+
+Colors are configurable via pi's theme system. Each preset defines its own color scheme, and you can override individual colors with a `theme.json` file in the extension directory.
+
+### Default Colors
+
+| Semantic | Theme Color | Description |
+|----------|-------------|-------------|
+| `pi` | `accent` | Pi icon |
+| `model` | `primary` | Model name |
+| `path` | `muted` | Directory path |
+| `gitClean` | `success` | Git branch (clean) |
+| `gitDirty` | `warning` | Git branch (dirty) |
+| `thinking` | `muted` | Thinking level |
+| `context` | `dim` | Context usage |
+| `contextWarn` | `warning` | Context usage >70% |
+| `contextError` | `error` | Context usage >90% |
+| `cost` | `primary` | Cost display |
+| `tokens` | `muted` | Token counts |
+
+### Custom Theme Override
+
+Create `~/.pi/agent/extensions/powerline-footer/theme.json`:
+
+```json
+{
+  "colors": {
+    "pi": "#ff5500",
+    "model": "accent",
+    "path": "#00afaf",
+    "gitClean": "success"
+  }
+}
+```
+
+Colors can be:
+- **Theme color names**: `accent`, `primary`, `muted`, `dim`, `text`, `success`, `warning`, `error`, `borderMuted`
+- **Hex colors**: `#ff5500`, `#d787af`
+
+See `theme.example.json` for all available options.

package/index.ts

@@ -1,16 +1,17 @@
-import type { ExtensionAPI, ReadonlyFooterDataProvider } from "@mariozechner/pi-coding-agent";
+import type { ExtensionAPI, ReadonlyFooterDataProvider, Theme } from "@mariozechner/pi-coding-agent";
 import type { AssistantMessage } from "@mariozechner/pi-ai";
 import { visibleWidth } from "@mariozechner/pi-tui";
 import { readFileSync, existsSync } from "node:fs";
 import { join } from "node:path";
 
-import type { SegmentContext, StatusLinePreset, StatusLineSegmentId } from "./types.js";
+import type { ColorScheme, SegmentContext, StatusLinePreset, StatusLineSegmentId } from "./types.js";
 import { getPreset, PRESETS } from "./presets.js";
 import { getSeparator } from "./separators.js";
 import { renderSegment } from "./segments.js";
 import { getGitStatus, invalidateGitStatus, invalidateGitBranch } from "./git-status.js";
 import { ansi, getFgAnsiCode } from "./colors.js";
 import { WelcomeComponent, WelcomeHeader, discoverLoadedCounts, getRecentSessions } from "./welcome.js";
+import { getDefaultColors } from "./theme.js";
 
 // ═══════════════════════════════════════════════════════════════════════════
 // Configuration
@@ -328,8 +329,9 @@ export default function powerlineFooter(pi: ExtensionAPI) {
     },
   });
 
-  function buildSegmentContext(ctx: any, width: number): SegmentContext {
+  function buildSegmentContext(ctx: any, width: number, theme: Theme): SegmentContext {
     const presetDef = getPreset(config.preset);
+    const colors: ColorScheme = presetDef.colors ?? getDefaultColors();
 
     // Build usage stats and get thinking level from session
     let input = 0, output = 0, cacheRead = 0, cacheWrite = 0, cost = 0;
@@ -387,6 +389,8 @@ export default function powerlineFooter(pi: ExtensionAPI) {
       extensionStatuses: footerDataRef?.getExtensionStatuses() ?? new Map(),
       options: presetDef.segmentOptions ?? {},
       width,
+      theme,
+      colors,
     };
   }
 
@@ -394,7 +398,7 @@ export default function powerlineFooter(pi: ExtensionAPI) {
    * Get cached responsive layout or compute fresh one.
    * Layout is cached per render cycle (same width = same layout).
    */
-  function getResponsiveLayout(width: number): { topContent: string; secondaryContent: string } {
+  function getResponsiveLayout(width: number, theme: Theme): { topContent: string; secondaryContent: string } {
     const now = Date.now();
     // Cache is valid if same width and within 50ms (same render cycle)
     if (lastLayoutResult && lastLayoutWidth === width && now - lastLayoutTimestamp < 50) {
@@ -402,7 +406,7 @@ export default function powerlineFooter(pi: ExtensionAPI) {
     }
     
     const presetDef = getPreset(config.preset);
-    const segmentCtx = buildSegmentContext(currentCtx, width);
+    const segmentCtx = buildSegmentContext(currentCtx, width, theme);
     // Available width for top bar content (minus box corners: ╭─ and ─╮ = 4 chars)
     const topBarAvailable = width - 4;
     
@@ -473,7 +477,7 @@ export default function powerlineFooter(pi: ExtensionAPI) {
           
           // Top border: ╭─ status ────────────╮
           // Use responsive layout - overflow goes to secondary row
-          const layout = getResponsiveLayout(width);
+          const layout = getResponsiveLayout(width, theme);
           const statusContent = layout.topContent;
           const statusWidth = visibleWidth(statusContent);
           const topFillWidth = width - 4; // Reserve 4 for corners (╭─ and ─╮)
@@ -515,7 +519,7 @@ export default function powerlineFooter(pi: ExtensionAPI) {
       });
 
       // Set up footer data provider access via a minimal footer
-      ctx.ui.setFooter((tui: any, _theme: any, footerData: ReadonlyFooterDataProvider) => {
+      ctx.ui.setFooter((tui: any, theme: Theme, footerData: ReadonlyFooterDataProvider) => {
         footerDataRef = footerData;
         tuiRef = tui; // Store TUI reference for re-renders on git branch changes
         const unsub = footerData.onBranchChange(() => tui.requestRender());
@@ -529,7 +533,7 @@ export default function powerlineFooter(pi: ExtensionAPI) {
             if (!currentCtx) return [];
             
             const presetDef = getPreset(config.preset);
-            const segmentCtx = buildSegmentContext(currentCtx, width);
+            const segmentCtx = buildSegmentContext(currentCtx, width, theme);
             const lines: string[] = [];
             
             // During streaming, show primary status in footer (editor hidden)
@@ -573,7 +577,7 @@ export default function powerlineFooter(pi: ExtensionAPI) {
 
       // Set up secondary row as a widget below editor (above sub bar)
       // Shows overflow segments when top bar is too narrow
-      ctx.ui.setWidget("powerline-secondary", (tui: any, _theme: any) => {
+      ctx.ui.setWidget("powerline-secondary", (tui: any, theme: Theme) => {
         return {
           dispose() {},
           invalidate() {},
@@ -581,7 +585,7 @@ export default function powerlineFooter(pi: ExtensionAPI) {
             if (!currentCtx) return [];
             
             // Use responsive layout - secondary row shows overflow from top bar
-            const layout = getResponsiveLayout(width);
+            const layout = getResponsiveLayout(width, theme);
             
             // Only show secondary row if there's overflow content that fits
             if (layout.secondaryContent) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-powerline-footer",
-  "version": "0.2.12",
+  "version": "0.2.13",
   "description": "Powerline-style status bar extension for pi coding agent",
   "type": "module",
   "bin": {
@@ -9,6 +9,7 @@
   "files": [
     "*.ts",
     "*.md",
+    "*.json",
     "cli.js"
   ],
   "keywords": [

package/presets.ts

@@ -1,4 +1,28 @@
-import type { PresetDef, StatusLinePreset } from "./types.js";
+import type { ColorScheme, PresetDef, StatusLinePreset } from "./types.js";
+import { getDefaultColors } from "./theme.js";
+
+// Get base colors from theme.ts (single source of truth)
+const DEFAULT_COLORS: ColorScheme = getDefaultColors();
+
+// Minimal - more muted, less colorful
+const MINIMAL_COLORS: ColorScheme = {
+  ...DEFAULT_COLORS,
+  pi: "dim",
+  model: "text",
+  path: "text",
+  git: "dim",
+  gitClean: "dim",
+};
+
+// Nerd - vibrant colors
+const NERD_COLORS: ColorScheme = {
+  ...DEFAULT_COLORS,
+  pi: "accent",
+  model: "accent",
+  path: "success",
+  tokens: "primary",
+  cost: "warning",
+};
 
 export const PRESETS: Record<StatusLinePreset, PresetDef> = {
   default: {
@@ -6,6 +30,7 @@ export const PRESETS: Record<StatusLinePreset, PresetDef> = {
     rightSegments: [],
     secondarySegments: ["extension_statuses"],
     separator: "powerline-thin",
+    colors: DEFAULT_COLORS,
     segmentOptions: {
       model: { showThinkingLevel: false },
       path: { mode: "basename" },
@@ -17,6 +42,7 @@ export const PRESETS: Record<StatusLinePreset, PresetDef> = {
     leftSegments: ["path", "git"],
     rightSegments: ["context_pct"],
     separator: "slash",
+    colors: MINIMAL_COLORS,
     segmentOptions: {
       path: { mode: "basename" },
       git: { showBranch: true, showStaged: false, showUnstaged: false, showUntracked: false },
@@ -27,6 +53,7 @@ export const PRESETS: Record<StatusLinePreset, PresetDef> = {
     leftSegments: ["model", "git"],
     rightSegments: ["cost", "context_pct"],
     separator: "powerline-thin",
+    colors: DEFAULT_COLORS,
     segmentOptions: {
       model: { showThinkingLevel: false },
       git: { showBranch: true, showStaged: true, showUnstaged: true, showUntracked: false },
@@ -37,6 +64,7 @@ export const PRESETS: Record<StatusLinePreset, PresetDef> = {
     leftSegments: ["pi", "hostname", "model", "thinking", "path", "git", "subagents"],
     rightSegments: ["token_in", "token_out", "cache_read", "cost", "context_pct", "time_spent", "time", "extension_statuses"],
     separator: "powerline",
+    colors: DEFAULT_COLORS,
     segmentOptions: {
       model: { showThinkingLevel: false },
       path: { mode: "abbreviated", maxLength: 50 },
@@ -49,6 +77,7 @@ export const PRESETS: Record<StatusLinePreset, PresetDef> = {
     leftSegments: ["pi", "hostname", "model", "thinking", "path", "git", "session", "subagents"],
     rightSegments: ["token_in", "token_out", "cache_read", "cache_write", "cost", "context_pct", "context_total", "time_spent", "time", "extension_statuses"],
     separator: "powerline",
+    colors: NERD_COLORS,
     segmentOptions: {
       model: { showThinkingLevel: false },
       path: { mode: "abbreviated", maxLength: 60 },
@@ -61,6 +90,7 @@ export const PRESETS: Record<StatusLinePreset, PresetDef> = {
     leftSegments: ["model", "path", "git"],
     rightSegments: ["token_total", "cost", "context_pct"],
     separator: "ascii",
+    colors: MINIMAL_COLORS,
     segmentOptions: {
       model: { showThinkingLevel: true },
       path: { mode: "abbreviated", maxLength: 40 },
@@ -72,6 +102,7 @@ export const PRESETS: Record<StatusLinePreset, PresetDef> = {
     leftSegments: ["model", "path", "git"],
     rightSegments: ["token_total", "cost", "context_pct"],
     separator: "powerline-thin",
+    colors: DEFAULT_COLORS,
     segmentOptions: {},
   },
 };

package/segments.ts

@@ -1,9 +1,14 @@
 import { hostname as osHostname } from "node:os";
 import { basename } from "node:path";
-import type { RenderedSegment, SegmentContext, StatusLineSegment, StatusLineSegmentId } from "./types.js";
-import { fgOnly, rainbow } from "./colors.js";
+import type { RenderedSegment, SegmentContext, SemanticColor, StatusLineSegment, StatusLineSegmentId } from "./types.js";
+import { fg, rainbow, applyColor } from "./theme.js";
 import { getIcons, SEP_DOT, getThinkingText } from "./icons.js";
 
+// Helper to apply semantic color from context
+function color(ctx: SegmentContext, semantic: SemanticColor, text: string): string {
+  return fg(ctx.theme, semantic, text, ctx.colors);
+}
+
 // ═══════════════════════════════════════════════════════════════════════════
 // Helpers
 // ═══════════════════════════════════════════════════════════════════════════
@@ -36,11 +41,11 @@ function formatDuration(ms: number): string {
 
 const piSegment: StatusLineSegment = {
   id: "pi",
-  render(_ctx) {
+  render(ctx) {
     const icons = getIcons();
     if (!icons.pi) return { content: "", visible: false };
     const content = `${icons.pi} `;
-    return { content: fgOnly("accent", content), visible: true };
+    return { content: color(ctx, "pi", content), visible: true };
   },
 };
 
@@ -69,7 +74,7 @@ const modelSegment: StatusLineSegment = {
       }
     }
 
-    return { content: fgOnly("model", content), visible: true };
+    return { content: color(ctx, "model", content), visible: true };
   },
 };
 
@@ -107,7 +112,7 @@ const pathSegment: StatusLineSegment = {
     }
 
     const content = withIcon(icons.folder, pwd);
-    return { content: fgOnly("path", content), visible: true };
+    return { content: color(ctx, "path", content), visible: true };
   },
 };
 
@@ -125,29 +130,32 @@ const gitSegment: StatusLineSegment = {
 
     const isDirty = gitStatus && (gitStatus.staged > 0 || gitStatus.unstaged > 0 || gitStatus.untracked > 0);
     const showBranch = opts.showBranch !== false;
+    const branchColor: SemanticColor = isDirty ? "gitDirty" : "gitClean";
 
-    // Build content
+    // Build content - color branch separately from indicators
     let content = "";
     if (showBranch && branch) {
-      content = withIcon(icons.branch, branch);
+      // Color just the branch name (icon + branch text)
+      content = color(ctx, branchColor, withIcon(icons.branch, branch));
     }
 
-    // Add status indicators
+    // Add status indicators (each with their own color, not wrapped)
     if (gitStatus) {
       const indicators: string[] = [];
       if (opts.showUnstaged !== false && gitStatus.unstaged > 0) {
-        indicators.push(fgOnly("unstaged", `*${gitStatus.unstaged}`));
+        indicators.push(applyColor(ctx.theme, "warning", `*${gitStatus.unstaged}`));
       }
       if (opts.showStaged !== false && gitStatus.staged > 0) {
-        indicators.push(fgOnly("staged", `+${gitStatus.staged}`));
+        indicators.push(applyColor(ctx.theme, "success", `+${gitStatus.staged}`));
       }
       if (opts.showUntracked !== false && gitStatus.untracked > 0) {
-        indicators.push(fgOnly("untracked", `?${gitStatus.untracked}`));
+        indicators.push(applyColor(ctx.theme, "muted", `?${gitStatus.untracked}`));
       }
       if (indicators.length > 0) {
         const indicatorText = indicators.join(" ");
         if (!content && showBranch === false) {
-          content = withIcon(icons.git, indicatorText);
+          // No branch shown, color the git icon with branch color
+          content = color(ctx, branchColor, icons.git ? `${icons.git} ` : "") + indicatorText;
         } else {
           content += content ? ` ${indicatorText}` : indicatorText;
         }
@@ -156,9 +164,7 @@ const gitSegment: StatusLineSegment = {
 
     if (!content) return { content: "", visible: false };
 
-    // Wrap entire content in branch color
-    const colorName = isDirty ? "gitDirty" : "gitClean";
-    return { content: fgOnly(colorName, content), visible: true };
+    return { content, visible: true };
   },
 };
 
@@ -184,16 +190,8 @@ const thinkingSegment: StatusLineSegment = {
       return { content: rainbow(content), visible: true };
     }
 
-    // Use dedicated thinking colors (gradient: gray → purple → blue → teal)
-    const colorMap: Record<string, "thinkingOff" | "thinkingMinimal" | "thinkingLow" | "thinkingMedium"> = {
-      off: "thinkingOff",
-      minimal: "thinkingMinimal",
-      low: "thinkingLow",
-      medium: "thinkingMedium",
-    };
-    const color = colorMap[level] || "thinkingOff";
-    
-    return { content: fgOnly(color, content), visible: true };
+    // Use thinking color for lower levels
+    return { content: color(ctx, "thinking", content), visible: true };
   },
 };
 
@@ -215,7 +213,7 @@ const tokenInSegment: StatusLineSegment = {
     if (!input) return { content: "", visible: false };
 
     const content = withIcon(icons.input, formatTokens(input));
-    return { content: fgOnly("spend", content), visible: true };
+    return { content: color(ctx, "tokens", content), visible: true };
   },
 };
 
@@ -227,7 +225,7 @@ const tokenOutSegment: StatusLineSegment = {
     if (!output) return { content: "", visible: false };
 
     const content = withIcon(icons.output, formatTokens(output));
-    return { content: fgOnly("output", content), visible: true };
+    return { content: color(ctx, "tokens", content), visible: true };
   },
 };
 
@@ -240,7 +238,7 @@ const tokenTotalSegment: StatusLineSegment = {
     if (!total) return { content: "", visible: false };
 
     const content = withIcon(icons.tokens, formatTokens(total));
-    return { content: fgOnly("spend", content), visible: true };
+    return { content: color(ctx, "tokens", content), visible: true };
   },
 };
 
@@ -255,7 +253,7 @@ const costSegment: StatusLineSegment = {
     }
 
     const costDisplay = usingSubscription ? "(sub)" : `$${cost.toFixed(2)}`;
-    return { content: fgOnly("cost", costDisplay), visible: true };
+    return { content: color(ctx, "cost", costDisplay), visible: true };
   },
 };
 
@@ -269,14 +267,14 @@ const contextPctSegment: StatusLineSegment = {
     const autoIcon = ctx.autoCompactEnabled && icons.auto ? ` ${icons.auto}` : "";
     const text = `${pct.toFixed(1)}%/${formatTokens(window)}${autoIcon}`;
 
-    // Icon outside color, text inside
+    // Icon outside color, text inside - use semantic colors for thresholds
     let content: string;
     if (pct > 90) {
-      content = withIcon(icons.context, fgOnly("error", text));
+      content = withIcon(icons.context, color(ctx, "contextError", text));
     } else if (pct > 70) {
-      content = withIcon(icons.context, fgOnly("warning", text));
+      content = withIcon(icons.context, color(ctx, "contextWarn", text));
     } else {
-      content = withIcon(icons.context, fgOnly("context", text));
+      content = withIcon(icons.context, color(ctx, "context", text));
     }
 
     return { content, visible: true };
@@ -291,7 +289,7 @@ const contextTotalSegment: StatusLineSegment = {
     if (!window) return { content: "", visible: false };
 
     return {
-      content: fgOnly("context", withIcon(icons.context, formatTokens(window))),
+      content: color(ctx, "context", withIcon(icons.context, formatTokens(window))),
       visible: true,
     };
   },
@@ -367,7 +365,7 @@ const cacheReadSegment: StatusLineSegment = {
     // Space-separated parts
     const parts = [icons.cache, icons.input, formatTokens(cacheRead)].filter(Boolean);
     const content = parts.join(" ");
-    return { content: fgOnly("spend", content), visible: true };
+    return { content: color(ctx, "tokens", content), visible: true };
   },
 };
 
@@ -381,7 +379,7 @@ const cacheWriteSegment: StatusLineSegment = {
     // Space-separated parts
     const parts = [icons.cache, icons.output, formatTokens(cacheWrite)].filter(Boolean);
     const content = parts.join(" ");
-    return { content: fgOnly("output", content), visible: true };
+    return { content: color(ctx, "tokens", content), visible: true };
   },
 };
 

package/theme.example.json

@@ -0,0 +1,19 @@
+{
+  "colors": {
+    "pi": "accent",
+    "model": "primary",
+    "path": "muted",
+    "git": "success",
+    "gitDirty": "warning",
+    "gitClean": "success",
+    "thinking": "muted",
+    "thinkingHigh": "accent",
+    "context": "dim",
+    "contextWarn": "warning",
+    "contextError": "error",
+    "cost": "primary",
+    "tokens": "muted",
+    "separator": "dim",
+    "border": "borderMuted"
+  }
+}

package/theme.ts

@@ -0,0 +1,171 @@
+/**
+ * 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 "@mariozechner/pi-coding-agent";
+import { existsSync, readFileSync } from "node:fs";
+import { join, dirname } from "node:path";
+import { fileURLToPath } from "node:url";
+import type { ColorScheme, ColorValue, SemanticColor } from "./types.js";
+
+// Default color scheme (uses pi theme colors)
+const DEFAULT_COLORS: Required<ColorScheme> = {
+  pi: "accent",
+  model: "primary",
+  path: "muted",
+  git: "success",
+  gitDirty: "warning",
+  gitClean: "success",
+  thinking: "muted",
+  thinkingHigh: "accent",
+  context: "dim",
+  contextWarn: "warning",
+  contextError: "error",
+  cost: "primary",
+  tokens: "muted",
+  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;
+const CACHE_TTL = 5000; // 5 seconds
+
+/**
+ * Get the path to the theme.json file
+ */
+function getThemePath(): string {
+  const extDir = dirname(fileURLToPath(import.meta.url));
+  return join(extDir, "theme.json");
+}
+
+/**
+ * Load user theme overrides from theme.json
+ */
+function loadUserTheme(): ColorScheme {
+  const now = Date.now();
+  if (userThemeCache && now - userThemeCacheTime < CACHE_TTL) {
+    return userThemeCache;
+  }
+
+  const themePath = getThemePath();
+  try {
+    if (existsSync(themePath)) {
+      const content = readFileSync(themePath, "utf-8");
+      const parsed = JSON.parse(content);
+      userThemeCache = parsed.colors ?? {};
+      userThemeCacheTime = now;
+      return userThemeCache;
+    }
+  } catch {
+    // Ignore errors, use defaults
+  }
+
+  userThemeCache = {};
+  userThemeCacheTime = now;
+  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" && color.startsWith("#");
+}
+
+/**
+ * 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: Theme,
+  color: ColorValue,
+  text: string
+): string {
+  if (isHexColor(color)) {
+    return `${hexToAnsi(color)}${text}\x1b[0m`;
+  }
+  return theme.fg(color as ThemeColor, text);
+}
+
+/**
+ * Apply a semantic color to text
+ */
+export function fg(
+  theme: Theme,
+  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 };
+}
+
+/**
+ * Clear the user theme cache (for reloading)
+ */
+export function clearThemeCache(): void {
+  userThemeCache = null;
+  userThemeCacheTime = 0;
+}

package/types.ts

@@ -1,3 +1,29 @@
+import type { Theme, ThemeColor } from "@mariozechner/pi-coding-agent";
+
+// Theme color - either a pi theme color name or a custom hex color
+export type ColorValue = ThemeColor | `#${string}`;
+
+// Semantic color names for segments
+export type SemanticColor =
+  | "pi"
+  | "model"
+  | "path"
+  | "git"
+  | "gitDirty"
+  | "gitClean"
+  | "thinking"
+  | "thinkingHigh"
+  | "context"
+  | "contextWarn"
+  | "contextError"
+  | "cost"
+  | "tokens"
+  | "separator"
+  | "border";
+
+// Color scheme mapping semantic names to actual colors
+export type ColorScheme = Partial<Record<SemanticColor, ColorValue>>;
+
 // Segment identifiers
 export type StatusLineSegmentId =
   | "pi"
@@ -62,6 +88,8 @@ export interface PresetDef {
   secondarySegments?: StatusLineSegmentId[];
   separator: StatusLineSeparatorStyle;
   segmentOptions?: StatusLineSegmentOptions;
+  /** Color scheme for this preset */
+  colors?: ColorScheme;
 }
 
 // Separator definition
@@ -116,6 +144,10 @@ export interface SegmentContext {
   // Options
   options: StatusLineSegmentOptions;
   width: number;
+  
+  // Theming
+  theme: Theme;
+  colors: ColorScheme;
 }
 
 // Rendered segment output