KimiCodeSwitch/src/hooks/useDashboard.ts

import { useCallback, useEffect, useState } from "react";
import { invoke } from "@tauri-apps/api/core";
import type { SummaryResult } from "../types/dashboard";

export type DashboardRange = "today" | "yesterday" | "7d" | "30d" | "all";

const RANGE_STORAGE_KEY = "kimi-switch-dashboard-range";

export function useDashboard() {
  const [range, setRange] = useState<DashboardRange>(() => {
    try {
      const stored = localStorage.getItem(RANGE_STORAGE_KEY) as DashboardRange | null;
      if (stored === "today" || stored === "yesterday" || stored === "7d" || stored === "30d" || stored === "all") {
        return stored;
      }
    } catch {
      // ignore
    }
    return "30d";
  });

  const [data, setData] = useState<SummaryResult | null>(null);
  const [loading, setLoading] = useState(false);
  const [error, setError] = useState<string | null>(null);
  /** Round-trip timing for the last get_summary call (user-perceived lag). */
  const [loadStats, setLoadStats] = useState<{ ms: number; kb: number } | null>(null);

  const refresh = useCallback(async (force = false) => {
    setLoading(true);
    setError(null);
    const start = performance.now();
    try {
      const result = await invoke<SummaryResult>("get_summary", {
        range,
        refresh: force,
      });
      setData(result);
      setLoadStats({
        ms: Math.round(performance.now() - start),
        kb: Math.round(JSON.stringify(result).length / 1024),
      });
    } catch (err) {
      const msg = err instanceof Error ? err.message : String(err);
      setError(msg);
    } finally {
      setLoading(false);
    }
  }, [range]);

  useEffect(() => {
    refresh();
  }, [refresh]);

  const changeRange = useCallback((next: DashboardRange) => {
    setRange(next);
    try {
      localStorage.setItem(RANGE_STORAGE_KEY, next);
    } catch {
      // ignore
    }
  }, []);

  return { range, changeRange, data, loading, error, refresh, loadStats };
}