import type { Api, Model, Provider, Usage } from "@earendil-works/pi-ai";
import type { SessionHeader } from "@earendil-works/pi-coding-agent";

export type TimeRange = "1d" | "7d" | "30d" | "All";

export interface DaySpend {
  date: string; // "YYYY-MM-DD"
  cost: number;
}

export interface HourSpend {
  hour: number; // 0-23
  cost: number;
}

export interface LangStat {
  language: string;
  lines: number;
  edits: number;
}

export interface ModelStat {
  provider: string;
  model: string;
  cost: number;
  calls: number;
}

export interface ProjectStat {
  project: string;
  cost: number;
  sessions: number;
}

export interface ToolStat {
  name: string;
  count: number;
}

export interface ProviderStat {
  provider: Provider;
  cost: number;
  calls: number;
}

export interface StatsSummary {
  totalCost: number;
  sessionCount: number;
  totalMessages: number;
  totalTokens: number;
  totalInputTokens: number;
  totalOutputTokens: number;
  totalCacheReadTokens: number;
  totalCacheWriteTokens: number;
  daysActive: number;
  avgCostPerDay: number;
  todayCost: number;
  languages: LangStat[];
  models: ModelStat[];
  projects: ProjectStat[];
  tools: ToolStat[];
  providers: ProviderStat[];
  skills: SkillStat[];
  compactionCount: number;
  compactedTokens: number;
  modelChanges: number;
  thinkingLevelCount: Record<string, number>;
  dailySpend: DaySpend[];
  hourlySpend: HourSpend[];
}

export interface SkillUsage {
  cost: number;
  tokens: { input: number; output: number; total: number };
  calls: number;
}

export interface SkillStat {
  name: string;
  calls: number;
  sessions: number;
  cost: number;
  tokens: number;
}

export interface SessionAgg {
  timestamp: SessionHeader["timestamp"];
  sessionId: SessionHeader["id"];
  project: string;
  cwd: SessionHeader["cwd"];
  models: Record<Provider, Record<Model<Api>["name"], SessionModelUsage>>;
  skills: Record<string, SkillUsage>;
  userMsgs: number;
  toolResults: number;
  compactionCount: number;
  compactedTokens: number;
  modelChanges: number;
  thinkingLevelCount: Record<string, number>;
}

export interface SessionModelUsage {
  provider: Provider;
  api: Api;
  usage: Usage;
  calls: number;
  asstMsgs: number;
  tools: Record<string, number>; // tool name → call count
  languages: Record<string, LangUsage>;
}

export interface LangUsage {
  lines: number;
  edits: number;
}

export interface Filters {
  project?: string;
  model?: string;
  provider?: string;
}

export interface CachePayload {
  version: string;
  signature: string;
  generatedAt: string;
  sessions: SessionAgg[];
}