import type { TranslationKey } from "@/lib/i18n";
import { quotaToUsd } from "@/lib/metrics";

export type DetailStatKey =
  | "calls"
  | "total_tokens"
  | "quota"
  | "prompt_tokens"
  | "completion_tokens"
  | "cache_creation_tokens"
  | "cache_read_tokens";

export type DetailStatFormat = "number" | "tokens" | "usd";

export interface DetailStatsData {
  calls: number;
  total_tokens: number;
  quota: number;
  prompt_tokens: number;
  completion_tokens: number;
  cache_creation_tokens: number;
  cache_read_tokens: number;
}

export interface DetailStatItem {
  key: DetailStatKey;
  labelKey: TranslationKey;
  value: number;
  format?: DetailStatFormat;
}

export function getDetailStats(data: DetailStatsData): DetailStatItem[] {
  return [
    { key: "calls", labelKey: "dash.totalCalls", value: data.calls },
    { key: "total_tokens", labelKey: "th.totalToken", value: data.total_tokens, format: "tokens" },
    { key: "quota", labelKey: "th.cost", value: quotaToUsd(data.quota), format: "usd" },
    { key: "prompt_tokens", labelKey: "th.input", value: data.prompt_tokens, format: "tokens" },
    { key: "completion_tokens", labelKey: "th.output", value: data.completion_tokens, format: "tokens" },
    { key: "cache_creation_tokens", labelKey: "th.cacheCreation", value: data.cache_creation_tokens, format: "tokens" },
    { key: "cache_read_tokens", labelKey: "th.cacheRead", value: data.cache_read_tokens, format: "tokens" },
  ];
}