@tokscale/cli 1.0.5 → 1.0.7

package/src/tui/types/index.ts

@@ -0,0 +1,165 @@
+import type { ColorPaletteName } from "../config/themes.js";
+
+export type TabType = "overview" | "model" | "daily" | "stats";
+export type SortType = "cost" | "tokens";
+export type SourceType = "opencode" | "claude" | "codex" | "cursor" | "gemini";
+
+export type { ColorPaletteName };
+
+export interface ModelEntry {
+  source: string;
+  model: string;
+  input: number;
+  output: number;
+  cacheWrite: number;
+  cacheRead: number;
+  reasoning: number;
+  total: number;
+  cost: number;
+}
+
+export interface DailyEntry {
+  date: string;
+  input: number;
+  output: number;
+  cache: number;
+  total: number;
+  cost: number;
+}
+
+export interface ContributionDay {
+  date: string;
+  cost: number;
+  level: number;
+}
+
+export interface GridCell {
+  date: string | null;
+  level: number;
+}
+
+export interface TotalBreakdown {
+  input: number;
+  output: number;
+  cacheWrite: number;
+  cacheRead: number;
+  reasoning: number;
+  total: number;
+  cost: number;
+}
+
+export interface Stats {
+  favoriteModel: string;
+  totalTokens: number;
+  sessions: number;
+  longestSession: string;
+  currentStreak: number;
+  longestStreak: number;
+  activeDays: number;
+  totalDays: number;
+  peakHour: string;
+}
+
+export interface ModelWithPercentage {
+  modelId: string;
+  percentage: number;
+  inputTokens: number;
+  outputTokens: number;
+  cacheReadTokens: number;
+  cacheWriteTokens: number;
+  totalTokens: number;
+  cost: number;
+}
+
+export interface ChartModelData {
+  modelId: string;
+  tokens: number;
+  color: string;
+}
+
+export interface ChartDataPoint {
+  date: string;
+  models: ChartModelData[];
+  total: number;
+}
+
+export interface TUIData {
+  modelEntries: ModelEntry[];
+  dailyEntries: DailyEntry[];
+  contributions: ContributionDay[];
+  contributionGrid: GridCell[][];
+  stats: Stats;
+  totalCost: number;
+  totals: TotalBreakdown;
+  modelCount: number;
+  chartData: ChartDataPoint[];
+  topModels: ModelWithPercentage[];
+  dailyBreakdowns: Map<string, DailyModelBreakdown>;
+}
+
+export interface TUISettings {
+  colorPalette: string;
+}
+
+export type LoadingPhase = 
+  | "idle"
+  | "loading-pricing"
+  | "syncing-cursor"
+  | "parsing-sources"
+  | "finalizing-report"
+  | "complete";
+
+export interface DailyModelBreakdown {
+  date: string;
+  cost: number;
+  totalTokens: number;
+  models: Array<{
+    modelId: string;
+    source: string;
+    tokens: {
+      input: number;
+      output: number;
+      cacheRead: number;
+      cacheWrite: number;
+      reasoning: number;
+    };
+    cost: number;
+    messages: number;
+  }>;
+}
+
+export interface TUIOptions {
+  initialTab?: TabType;
+  enabledSources?: SourceType[];
+  sortBy?: SortType;
+  sortDesc?: boolean;
+  since?: string;
+  until?: string;
+  year?: string;
+  colorPalette?: ColorPaletteName;
+}
+
+
+
+export const LAYOUT = {
+  HEADER_HEIGHT: 2,
+  FOOTER_HEIGHT: 4,
+  MIN_CONTENT_HEIGHT: 12,
+  CHART_HEIGHT_RATIO: 0.35,
+  MIN_CHART_HEIGHT: 5,
+  MIN_LIST_HEIGHT: 4,
+  CHART_AXIS_WIDTH: 8,
+  MIN_CHART_WIDTH: 20,
+  MAX_VISIBLE_BARS: 52,
+} as const;
+
+export const SOURCE_LABELS: Record<SourceType, string> = {
+  opencode: "OC",
+  claude: "CC",
+  codex: "CX",
+  cursor: "CR",
+  gemini: "GM",
+} as const;
+
+export const TABS: readonly TabType[] = ["overview", "model", "daily", "stats"] as const;
+export const ALL_SOURCES: readonly SourceType[] = ["opencode", "claude", "codex", "cursor", "gemini"] as const;

package/src/tui/utils/cleanup.ts

@@ -0,0 +1,65 @@
+/**
+ * Terminal cleanup utility for restoring terminal state when TUI exits.
+ * Provides fallback cleanup for crash scenarios where OpenTUI's destroy() may not run.
+ */
+
+/**
+ * Complete terminal state restoration sequences.
+ * Based on research from xterm, kitty keyboard protocol, and popular TUI libraries.
+ */
+export const TERMINAL_CLEANUP_SEQUENCES = [
+  // Disable all mouse tracking modes
+  '\x1b[?1016l',  // SGR Pixel Mode
+  '\x1b[?1015l',  // URXVT Mouse
+  '\x1b[?1006l',  // SGR Mouse (produces "51;77;17M" sequences)
+  '\x1b[?1005l',  // UTF-8 Mouse
+  '\x1b[?1004l',  // Focus Events
+  '\x1b[?1003l',  // Any Event Mouse (motion tracking)
+  '\x1b[?1002l',  // Button Event Mouse (drag tracking)
+  '\x1b[?1001l',  // Highlight Mouse
+  '\x1b[?1000l',  // VT200 Mouse
+  '\x1b[?9l',     // X10 Mouse
+  
+  // Disable kitty keyboard protocol (produces "9;5u" sequences)
+  '\x1b[<u',      // Disable kitty keyboard progressive enhancement
+  '\x1b[>4;0m',   // Disable modifyOtherKeys (xterm)
+  
+  // Disable synchronized updates
+  '\x1b[?2026l',
+  
+  // Restore cursor and attributes
+  '\x1b[?25h',    // Show cursor (DECTCEM)
+  '\x1b[0m',      // Reset all text attributes (SGR 0)
+  
+  // Exit alternate screen buffer
+  '\x1b[?1049l',  // Exit alt screen + restore cursor
+  '\x1b[?47l',    // Legacy: exit alt screen
+  '\x1b[?1047l',  // Legacy: another alt screen mode
+].join('');
+
+let hasCleanedUp = false;
+
+/**
+ * Write terminal restoration sequences to stdout.
+ * Idempotent - safe to call multiple times.
+ * 
+ * This is a FALLBACK for crash scenarios. Normal exit should use
+ * renderer.destroy() which handles cleanup properly.
+ */
+export function restoreTerminalState(): void {
+  if (hasCleanedUp) return;
+  hasCleanedUp = true;
+  
+  try {
+    process.stdout.write(TERMINAL_CLEANUP_SEQUENCES);
+  } catch {
+    // Ignore write errors (stdout may be closed)
+  }
+}
+
+/**
+ * Reset cleanup state (for testing purposes).
+ */
+export function resetCleanupState(): void {
+  hasCleanedUp = false;
+}

package/src/tui/utils/colors.ts

@@ -0,0 +1,65 @@
+import type { SourceType } from "../types/index.js";
+
+export const PROVIDER_COLORS = {
+  anthropic: "#FF6B35",
+  openai: "#10B981",
+  google: "#3B82F6",
+  cursor: "#8B5CF6",
+  opencode: "#6B7280",
+  deepseek: "#06B6D4",
+  xai: "#EAB308",
+  meta: "#6366F1",
+  unknown: "#FFFFFF",
+} as const;
+
+export type ProviderType = keyof typeof PROVIDER_COLORS;
+
+const PROVIDER_PATTERNS: readonly [RegExp, ProviderType][] = [
+  [/claude|sonnet|opus|haiku/i, "anthropic"],
+  [/gpt|^o1|^o3|codex|text-embedding|dall-e|whisper|tts/i, "openai"],
+  [/gemini/i, "google"],
+  [/deepseek/i, "deepseek"],
+  [/grok/i, "xai"],
+  [/llama|mixtral/i, "meta"],
+  [/^auto$|cursor/i, "cursor"],
+] as const;
+
+const providerCache = new Map<string, ProviderType>();
+const colorCache = new Map<string, string>();
+
+export function getProviderFromModel(modelId: string): ProviderType {
+  const cached = providerCache.get(modelId);
+  if (cached) return cached;
+
+  let provider: ProviderType = "unknown";
+  for (const [pattern, type] of PROVIDER_PATTERNS) {
+    if (pattern.test(modelId)) {
+      provider = type;
+      break;
+    }
+  }
+
+  providerCache.set(modelId, provider);
+  return provider;
+}
+
+export function getModelColor(modelId: string): string {
+  const cached = colorCache.get(modelId);
+  if (cached) return cached;
+
+  const color = PROVIDER_COLORS[getProviderFromModel(modelId)];
+  colorCache.set(modelId, color);
+  return color;
+}
+
+export const SOURCE_COLORS: Record<SourceType, string> = {
+  opencode: "#22c55e",
+  claude: "#f97316",
+  codex: "#3b82f6",
+  cursor: "#a855f7",
+  gemini: "#06b6d4",
+};
+
+export function getSourceColor(source: SourceType | string): string {
+  return SOURCE_COLORS[source as SourceType] || "#888888";
+}

package/src/tui/utils/format.ts

@@ -0,0 +1,36 @@
+export function formatTokens(n: number): string {
+  if (!Number.isFinite(n) || n < 0) return "0";
+  return n.toLocaleString();
+}
+
+export function formatTokensShort(n: number): string {
+  if (!Number.isFinite(n) || n < 0) return "0";
+  if (n >= 1_000_000_000) return `${(n / 1_000_000_000).toFixed(1)}B`;
+  if (n >= 1_000_000) return `${(n / 1_000_000).toFixed(1)}M`;
+  if (n >= 1_000) return `${(n / 1_000).toFixed(1)}K`;
+  return n.toLocaleString();
+}
+
+export function formatTokensCompact(n: number): string {
+  if (!Number.isFinite(n) || n < 0) return "0";
+  if (n >= 1_000_000_000) return `${(n / 1_000_000_000).toFixed(1)}B`;
+  if (n >= 1_000_000) return `${(n / 1_000_000).toFixed(1)}M`;
+  if (n >= 1_000) return `${(n / 1_000).toFixed(0)}K`;
+  return n.toLocaleString();
+}
+
+export function formatCost(cost: number): string {
+  if (!Number.isFinite(cost) || cost < 0) return "$0.00";
+  if (cost >= 1000) return `$${(cost / 1000).toFixed(1)}K`;
+  return `$${cost.toFixed(2)}`;
+}
+
+export function formatCostFull(cost: number): string {
+  if (!Number.isFinite(cost) || cost < 0) return "$0.00";
+  return `$${cost.toFixed(2)}`;
+}
+
+export function formatPercentage(value: number): string {
+  if (!Number.isFinite(value)) return "0.0%";
+  return `${value.toFixed(1)}%`;
+}

package/src/tui/utils/responsive.ts

@@ -0,0 +1,8 @@
+export const NARROW_TERMINAL_WIDTH = 80;
+export const VERY_NARROW_TERMINAL_WIDTH = 60;
+
+export const isNarrow = (width: number | undefined): boolean =>
+  (width ?? 100) < NARROW_TERMINAL_WIDTH;
+
+export const isVeryNarrow = (width: number | undefined): boolean =>
+  (width ?? 100) < VERY_NARROW_TERMINAL_WIDTH;

package/src/types.d.ts

@@ -0,0 +1,28 @@
+declare module "string-width" {
+  export default function stringWidth(str: string): number;
+}
+
+declare module "./tui-bundle.js" {
+  import type { TUIOptions } from "./tui/types/index.js";
+  export function launchTUI(options?: TUIOptions): Promise<void>;
+  export type { TUIOptions };
+}
+
+declare module "bun" {
+  export interface BunSubprocess {
+    stdin: any;
+    stdout: any;
+    stderr: any;
+    exited: Promise<number>;
+    kill(signal?: string): void;
+  }
+
+  export function spawn(options: {
+    cmd: string[];
+    stdin?: "pipe";
+    stdout?: "pipe";
+    stderr?: "pipe";
+    timeout?: number;
+    killSignal?: string;
+  }): BunSubprocess;
+}