@tokscale/cli 1.0.4 → 1.0.6

package/src/tui/config/settings.ts

@@ -0,0 +1,130 @@
+import { homedir } from "os";
+import { join } from "path";
+import { readFileSync, writeFileSync, existsSync, mkdirSync } from "fs";
+import type { TUIData, DailyModelBreakdown } from "../types/index.js";
+
+const CONFIG_DIR = join(homedir(), ".config", "tokscale");
+const CACHE_DIR = join(homedir(), ".cache", "tokscale");
+const CONFIG_FILE = join(CONFIG_DIR, "tui-settings.json");
+const CACHE_FILE = join(CACHE_DIR, "tui-data-cache.json");
+
+const CACHE_STALE_THRESHOLD_MS = 60 * 1000;
+
+interface TUISettings {
+  colorPalette: string;
+}
+
+interface CachedTUIData {
+  timestamp: number;
+  enabledSources: string[];
+  data: Omit<TUIData, 'dailyBreakdowns'> & {
+    dailyBreakdowns: Array<[string, DailyModelBreakdown]>;
+  };
+}
+
+export function loadSettings(): TUISettings {
+  try {
+    if (existsSync(CONFIG_FILE)) {
+      return JSON.parse(readFileSync(CONFIG_FILE, "utf-8"));
+    }
+  } catch {
+  }
+  return { colorPalette: "green" };
+}
+
+export function saveSettings(settings: TUISettings): void {
+  if (!existsSync(CONFIG_DIR)) {
+    mkdirSync(CONFIG_DIR, { recursive: true });
+  }
+  writeFileSync(CONFIG_FILE, JSON.stringify(settings, null, 2));
+}
+
+function sourcesMatch(enabledSources: Set<string>, cachedSources: string[]): boolean {
+  const cachedSet = new Set(cachedSources);
+  if (enabledSources.size !== cachedSet.size) {
+    return false;
+  }
+  for (const source of enabledSources) {
+    if (!cachedSet.has(source)) {
+      return false;
+    }
+  }
+  return true;
+}
+
+export function loadCachedData(enabledSources: Set<string>): TUIData | null {
+  try {
+    if (!existsSync(CACHE_FILE)) {
+      return null;
+    }
+    
+    const cached: CachedTUIData = JSON.parse(readFileSync(CACHE_FILE, "utf-8"));
+    
+    if (!sourcesMatch(enabledSources, cached.enabledSources)) {
+      return null;
+    }
+    
+    if (!cached.data.dailyBreakdowns || !Array.isArray(cached.data.dailyBreakdowns)) {
+      return null;
+    }
+    
+    return {
+      ...cached.data,
+      dailyBreakdowns: new Map(cached.data.dailyBreakdowns),
+    };
+  } catch {
+    return null;
+  }
+}
+
+export function saveCachedData(data: TUIData, enabledSources: Set<string>): void {
+  try {
+    if (!existsSync(CACHE_DIR)) {
+      mkdirSync(CACHE_DIR, { recursive: true });
+    }
+    
+    const serializableBreakdowns = Array.from(data.dailyBreakdowns.entries());
+    const cached: CachedTUIData = {
+      timestamp: Date.now(),
+      enabledSources: Array.from(enabledSources),
+      data: {
+        ...data,
+        dailyBreakdowns: serializableBreakdowns,
+      },
+    };
+    
+    writeFileSync(CACHE_FILE, JSON.stringify(cached));
+  } catch {
+  }
+}
+
+export function isCacheStale(enabledSources: Set<string>): boolean {
+  try {
+    if (!existsSync(CACHE_FILE)) {
+      return true;
+    }
+    
+    const cached: CachedTUIData = JSON.parse(readFileSync(CACHE_FILE, "utf-8"));
+    const cacheAge = Date.now() - cached.timestamp;
+    
+    if (!sourcesMatch(enabledSources, cached.enabledSources)) {
+      return true;
+    }
+    
+    return cacheAge > CACHE_STALE_THRESHOLD_MS;
+  } catch {
+    return true;
+  }
+}
+
+export function getCacheTimestamp(): number | null {
+  try {
+    if (!existsSync(CACHE_FILE)) {
+      return null;
+    }
+    const cached: CachedTUIData = JSON.parse(readFileSync(CACHE_FILE, "utf-8"));
+    return cached.timestamp;
+  } catch {
+    return null;
+  }
+}

package/src/tui/config/themes.ts

@@ -0,0 +1,115 @@
+export interface GraphColorPalette {
+  name: string;
+  grade0: string;
+  grade1: string;
+  grade2: string;
+  grade3: string;
+  grade4: string;
+}
+
+export type ColorPaletteName =
+  | "green"
+  | "halloween"
+  | "teal"
+  | "blue"
+  | "pink"
+  | "purple"
+  | "orange"
+  | "monochrome"
+  | "YlGnBu";
+
+const GRAPH_EMPTY = "#1F1F20";
+
+export const colorPalettes: Record<ColorPaletteName, GraphColorPalette> = {
+  green: {
+    name: "Green",
+    grade0: GRAPH_EMPTY,
+    grade1: "#9be9a8",
+    grade2: "#40c463",
+    grade3: "#30a14e",
+    grade4: "#216e39",
+  },
+  halloween: {
+    name: "Halloween",
+    grade0: GRAPH_EMPTY,
+    grade1: "#FFEE4A",
+    grade2: "#FFC501",
+    grade3: "#FE9600",
+    grade4: "#03001C",
+  },
+  teal: {
+    name: "Teal",
+    grade0: GRAPH_EMPTY,
+    grade1: "#7ee5e5",
+    grade2: "#2dc5c5",
+    grade3: "#0d9e9e",
+    grade4: "#0e6d6d",
+  },
+  blue: {
+    name: "Blue",
+    grade0: GRAPH_EMPTY,
+    grade1: "#79b8ff",
+    grade2: "#388bfd",
+    grade3: "#1f6feb",
+    grade4: "#0d419d",
+  },
+  pink: {
+    name: "Pink",
+    grade0: GRAPH_EMPTY,
+    grade1: "#f0b5d2",
+    grade2: "#d961a0",
+    grade3: "#bf4b8a",
+    grade4: "#99286e",
+  },
+  purple: {
+    name: "Purple",
+    grade0: GRAPH_EMPTY,
+    grade1: "#cdb4ff",
+    grade2: "#a371f7",
+    grade3: "#8957e5",
+    grade4: "#6e40c9",
+  },
+  orange: {
+    name: "Orange",
+    grade0: GRAPH_EMPTY,
+    grade1: "#ffd699",
+    grade2: "#ffb347",
+    grade3: "#ff8c00",
+    grade4: "#cc5500",
+  },
+  monochrome: {
+    name: "Monochrome",
+    grade0: GRAPH_EMPTY,
+    grade1: "#9e9e9e",
+    grade2: "#757575",
+    grade3: "#424242",
+    grade4: "#212121",
+  },
+  YlGnBu: {
+    name: "YlGnBu",
+    grade0: GRAPH_EMPTY,
+    grade1: "#a1dab4",
+    grade2: "#41b6c4",
+    grade3: "#2c7fb8",
+    grade4: "#253494",
+  },
+};
+
+export const DEFAULT_PALETTE: ColorPaletteName = "green";
+
+export const getPaletteNames = (): ColorPaletteName[] =>
+  Object.keys(colorPalettes) as ColorPaletteName[];
+
+export const getPalette = (name: ColorPaletteName): GraphColorPalette =>
+  colorPalettes[name] || colorPalettes[DEFAULT_PALETTE];
+
+export const getGradeColor = (
+  palette: GraphColorPalette,
+  intensity: 0 | 1 | 2 | 3 | 4
+): string => {
+  return (
+    [palette.grade0, palette.grade1, palette.grade2, palette.grade3, palette.grade4][
+      intensity
+    ] || palette.grade0
+  );
+};