@ohgodtamit/pi-usage 0.1.0-alpha.0

package/src/mascot.ts

@@ -0,0 +1,227 @@
+/**
+ * Pi-chan — lightweight anime mascot for the usage panel TUI.
+ * ASCII poses + per-view icons keep the menu and Wrapped view playful
+ * without breaking narrow terminals.
+ */
+import type { Theme, ThemeColor } from "@earendil-works/pi-coding-agent";
+import type { WrappedStats } from "./aggregate.ts";
+import { formatHour } from "./format.ts";
+
+/** Selectable top-level views (Tokscale-style menu + Wrapped AI). */
+export type ViewKey =
+  | "overview"
+  | "models"
+  | "delegation"
+  | "daily"
+  | "stats"
+  | "hourly"
+  | "providers"
+  | "wrapped";
+
+/** Ordered list of views for tab/arrow navigation. */
+export const VIEW_ORDER: ViewKey[] = [
+  "overview",
+  "models",
+  "delegation",
+  "daily",
+  "stats",
+  "hourly",
+  "providers",
+  "wrapped",
+];
+
+export interface ViewTabMeta {
+  icon: string;
+  short: string;
+  label: string;
+  hint: string;
+  /** Accent when this tab is selected. */
+  color: ThemeColor;
+}
+
+export const VIEW_TABS: Record<ViewKey, ViewTabMeta> = {
+  overview: {
+    icon: "◈",
+    short: "Over",
+    label: "Overview",
+    hint: "Quota bars & headline stats at a glance",
+    color: "accent",
+  },
+  models: {
+    icon: "◎",
+    short: "Mod",
+    label: "Models",
+    hint: "Models, skills, plugins, tools & projects",
+    color: "success",
+  },
+  delegation: {
+    icon: "⑂",
+    short: "Del",
+    label: "Delegation",
+    hint: "Direct and delegated usage & concurrency",
+    color: "accent",
+  },
+  daily: {
+    icon: "☀",
+    short: "Day",
+    label: "Daily",
+    hint: "Per-day activity, uptime & top model",
+    color: "warning",
+  },
+  stats: {
+    icon: "▦",
+    short: "Stat",
+    label: "Stats",
+    hint: "Contribution graph & streaks",
+    color: "accent",
+  },
+  hourly: {
+    icon: "⏱",
+    short: "Hr",
+    label: "Hourly",
+    hint: "Peak coding hours through the day",
+    color: "success",
+  },
+  providers: {
+    icon: "⚡",
+    short: "Prov",
+    label: "Providers",
+    hint: "Usage split by provider / backend",
+    color: "warning",
+  },
+  wrapped: {
+    icon: "✦",
+    short: "Wrap",
+    label: "Wrapped",
+    hint: "Year in review — [ ] or y to change year",
+    color: "accent",
+  },
+};
+
+/** Cycle palette for rainbow bars (monthly chart, tool rows, etc.). */
+export const RAINBOW: ThemeColor[] = [
+  "accent",
+  "success",
+  "warning",
+  "error",
+  "accent",
+  "success",
+  "warning",
+  "accent",
+  "success",
+  "warning",
+  "error",
+  "accent",
+];
+
+export type MascotPose = "wave" | "celebrate" | "night" | "curious" | "sleepy" | "tools";
+
+/** Raw ASCII lines per pose (no theme — color applied at render). */
+const POSES: Record<MascotPose, string[]> = {
+  wave: ["  ∧＿∧", " (◕ ω ◕)", " /  つ  Year", " (  ～  )", "  ～ ～  "],
+  celebrate: [" \\(^o^)/", "  ∧＿∧ ", " (≧▽≦)", " /  つ  !", "  ★ ★ ★"],
+  night: ["  ∧＿∧ ", " (◕ ‿ ◕)", " /  zzz", " (  つ  )", "  ～ ☾  "],
+  curious: ["  ∧＿∧", " (◕ ◡ ◕)?", " /  つ  hm", " (  ～  )", "  ？ ？  "],
+  sleepy: ["  ∧＿∧", " ( - ω -)", " /  つ  zZ", " (  ～  )", "  ～ ～  "],
+  tools: ["  ∧＿∧", " (◕ ▽ ◕)", " /  つ ⚙", " (  ～  )", "  ♪ ♪ ♪"],
+};
+
+const VIEW_POSE: Partial<Record<ViewKey, MascotPose>> = {
+  overview: "wave",
+  models: "tools",
+  delegation: "curious",
+  daily: "curious",
+  stats: "celebrate",
+  hourly: "night",
+  providers: "wave",
+  wrapped: "celebrate",
+};
+
+/** Pick a mascot pose for the active view or Wrapped stats mood. */
+export function mascotPose(view: ViewKey, stats?: WrappedStats | null): MascotPose {
+  if (view === "wrapped") {
+    if (!stats) return "sleepy";
+    if (stats.longestStreak >= 7) return "celebrate";
+    if (stats.peakHour != null && stats.peakHour >= 22) return "night";
+    if (stats.modelCount >= 4) return "curious";
+    return "celebrate";
+  }
+  return VIEW_POSE[view] ?? "wave";
+}
+
+/** Compact mascot for the Wrapped report sidebar (4 lines, aligned). */
+const WRAPPED_POSES: Record<MascotPose, string[]> = {
+  wave: ["  ∧＿∧", " (◕‿◕)", " /つ wrap", "  ～～"],
+  celebrate: ["  ∧＿∧", " (≧◡≦)", " /つ ✦", "  ★ ★"],
+  night: ["  ∧＿∧", " (◕‿◕)", " /つ ☾", "  zzz"],
+  curious: ["  ∧＿∧", " (◕◡◕)", " /つ ?", "  ～～"],
+  sleepy: ["  ∧＿∧", " (－ω－)", " /つ …", "  ～～"],
+  tools: ["  ∧＿∧", " (◕‿◕)", " /つ ⚙", "  ～～"],
+};
+
+/** Theme-colored mascot lines. */
+export function renderMascot(pose: MascotPose, theme: Theme): string[] {
+  const lines = POSES[pose];
+  const faceColor: ThemeColor =
+    pose === "celebrate" ? "warning" : pose === "night" ? "accent" : "success";
+  return lines.map((line, i) => {
+    if (i === 1) return theme.fg(faceColor, line);
+    if (i === 4 && (pose === "celebrate" || pose === "tools")) return theme.fg("warning", line);
+    return theme.fg("muted", line);
+  });
+}
+
+/** Wrapped sidebar mascot — compact, report-card styling. */
+export function renderWrappedMascot(pose: MascotPose, theme: Theme): string[] {
+  const lines = WRAPPED_POSES[pose];
+  return lines.map((line, i) => {
+    if (i === 1) return theme.fg("accent", line);
+    if (i === 3 && pose === "celebrate") return theme.fg("warning", line);
+    return theme.fg("borderMuted", line);
+  });
+}
+
+/** One-line insight caption for the Wrapped footer. */
+export function wrappedMascotCaption(stats: WrappedStats | null, year: number): string {
+  if (!stats) return `No sessions recorded in ${year}. Pick another year with [ ].`;
+  if (stats.longestStreak >= 14) return `${stats.longestStreak}-day streak — standout consistency.`;
+  if (stats.peakHour != null && stats.peakHour >= 22)
+    return `Peak hour ${formatHour(stats.peakHour)} — late-night sessions dominated.`;
+  if (stats.peakHour != null && stats.peakHour < 7)
+    return `Peak hour ${formatHour(stats.peakHour)} — early starts shaped the year.`;
+  if (stats.topModels.length >= 1 && stats.topModels[0].pct >= 60)
+    return `${Math.round(stats.topModels[0].pct)}% of usage on ${stats.topModels[0].name}.`;
+  return `${stats.activeDays} active days across ${stats.modelCount} models.`;
+}
+
+/** Short speech line for menu hint row. */
+export function mascotQuip(view: ViewKey, stats?: WrappedStats | null): string {
+  if (view === "wrapped") {
+    if (!stats) return "Year in review — [ ] or y to change year";
+    if (stats.longestStreak >= 14) return `${stats.longestStreak}-day streak — your highlight stat`;
+    if (stats.peakHour != null && stats.peakHour >= 22)
+      return "Night sessions defined your peak hours";
+    return `${stats.year} report ready — scroll for breakdowns`;
+  }
+  const tab = VIEW_TABS[view];
+  return tab.hint;
+}
+
+/** Glyphs for common pi tool names (Models → Tools section). */
+const TOOL_GLYPHS: Record<string, string> = {
+  read: "↳",
+  write: "✎",
+  edit: "✎",
+  bash: "$",
+  grep: "⌕",
+  glob: "✶",
+  web_fetch: "@",
+  web_search: "◎",
+  task: "▶",
+  ask: "?",
+};
+
+export function toolGlyph(name: string): string {
+  const base = name.split(":").pop()?.split("/").pop() ?? name;
+  return TOOL_GLYPHS[base] ?? TOOL_GLYPHS[base.toLowerCase()] ?? "•";
+}

package/src/prices.ts

@@ -0,0 +1,63 @@
+/**
+ * Bundled default model prices ($/million tokens).
+ *
+ * pi records `cost = 0` for token-priced / proxied providers it has no pricing
+ * for (zai/GLM, 9Router `kr/…` `cx/…`, MiniMax, etc.). These defaults let the
+ * panel show an approximate cost out of the box. They are merged under the
+ * user's `~/.pi/agent/usage.json` `modelPrices`, so any user entry overrides
+ * the matching default. Set your own with `/usage-pricing`.
+ *
+ * Figures are taken from each provider's official pricing page (June 2026):
+ *   - Anthropic   platform.claude.com/docs/en/about-claude/pricing
+ *   - OpenAI      developers.openai.com/api/docs/pricing
+ *   - Google      ai.google.dev/gemini-api/docs/pricing
+ *   - Z.ai (GLM)  docs.z.ai/guides/overview/pricing
+ *   - MiniMax     platform.minimax.io/docs/guides/pricing-paygo
+ *
+ * Keys match a model ID exactly, or by base name (after the last `/`) so an
+ * entry like `claude-opus-4.7` also covers `kr/claude-opus-4.7`. Prices change
+ * often and subscription/proxy costs differ from list rates — treat these as
+ * estimates and override as needed.
+ */
+import type { ModelPrice } from "./config.ts";
+
+export const DEFAULT_MODEL_PRICES: Record<string, ModelPrice> = {
+  // Anthropic Claude — Opus tier ($5 / $25, cache hit $0.50, 5m write $6.25).
+  "claude-opus-4.8": { input: 5, output: 25, cacheRead: 0.5, cacheWrite: 6.25 },
+  "claude-opus-4.7": { input: 5, output: 25, cacheRead: 0.5, cacheWrite: 6.25 },
+  "claude-opus-4.6": { input: 5, output: 25, cacheRead: 0.5, cacheWrite: 6.25 },
+  "claude-opus-4-8": { input: 5, output: 25, cacheRead: 0.5, cacheWrite: 6.25 },
+  "claude-opus-4-8-high": { input: 5, output: 25, cacheRead: 0.5, cacheWrite: 6.25 },
+  "claude-opus-4-8-max": { input: 5, output: 25, cacheRead: 0.5, cacheWrite: 6.25 },
+  "claude-opus-4-8-thinking-max": { input: 5, output: 25, cacheRead: 0.5, cacheWrite: 6.25 },
+  "claude-opus-4-7": { input: 5, output: 25, cacheRead: 0.5, cacheWrite: 6.25 },
+  "claude-opus-4-5-thinking": { input: 5, output: 25, cacheRead: 0.5, cacheWrite: 6.25 },
+  // Anthropic Claude — Sonnet tier ($3 / $15, cache hit $0.30, 5m write $3.75).
+  "claude-sonnet-4-6": { input: 3, output: 15, cacheRead: 0.3, cacheWrite: 3.75 },
+  "claude-4.5-sonnet": { input: 3, output: 15, cacheRead: 0.3, cacheWrite: 3.75 },
+  // OpenAI GPT-5 family (cached input only; no cache-write charge).
+  "gpt-5.5": { input: 5, output: 30, cacheRead: 0.5, cacheWrite: 0 },
+  "gpt-5.5-review": { input: 5, output: 30, cacheRead: 0.5, cacheWrite: 0 },
+  "gpt-5.4": { input: 2.5, output: 15, cacheRead: 0.25, cacheWrite: 0 },
+  "gpt-5.3-codex": { input: 1.25, output: 10, cacheRead: 0.125, cacheWrite: 0 },
+  "gpt-5.1": { input: 1.25, output: 10, cacheRead: 0.125, cacheWrite: 0 },
+  // Google Gemini 3.
+  "gemini-3.1-pro-preview": { input: 2, output: 12, cacheRead: 0.2, cacheWrite: 0 },
+  "gemini-3.1-pro-high": { input: 2, output: 12, cacheRead: 0.2, cacheWrite: 0 },
+  "gemini-3.1-pro-low": { input: 2, output: 12, cacheRead: 0.2, cacheWrite: 0 },
+  "gemini-3-flash": { input: 0.5, output: 3, cacheRead: 0.05, cacheWrite: 0 },
+  // Z.ai GLM (per-token API list rates; subscription cost differs).
+  "glm-5.2": { input: 1.4, output: 4.4, cacheRead: 0.26, cacheWrite: 0 },
+  "glm-5.1": { input: 1.4, output: 4.4, cacheRead: 0.26, cacheWrite: 0 },
+  "glm-5": { input: 1.0, output: 3.2, cacheRead: 0.2, cacheWrite: 0 },
+  "glm-5-turbo": { input: 1.2, output: 4.0, cacheRead: 0.24, cacheWrite: 0 },
+  "glm-5v-turbo": { input: 1.2, output: 4.0, cacheRead: 0.24, cacheWrite: 0 },
+  "glm-4.7": { input: 0.6, output: 2.2, cacheRead: 0.11, cacheWrite: 0 },
+  "glm-4.6": { input: 0.6, output: 2.2, cacheRead: 0.11, cacheWrite: 0 },
+  "glm-4.6v": { input: 0.6, output: 2.2, cacheRead: 0.11, cacheWrite: 0 },
+  "glm-4.5v": { input: 0.6, output: 2.2, cacheRead: 0.11, cacheWrite: 0 },
+  // MiniMax M-series.
+  "MiniMax-M3": { input: 0.6, output: 2.4, cacheRead: 0.06, cacheWrite: 0.375 },
+  "MiniMax-M2.7": { input: 0.3, output: 1.2, cacheRead: 0.06, cacheWrite: 0.375 },
+  "MiniMax-M2.7-highspeed": { input: 0.6, output: 2.4, cacheRead: 0.06, cacheWrite: 0.375 },
+};