import CodexBarCore
import Foundation
import Testing

struct CostUsageWindowSummaryTests {
    @Test
    func `summaries use calendar windows instead of the last nonempty rows`() {
        let snapshot = Self.snapshot(historyDays: 90)
        let summary = snapshot.summary(forLastDays: 7, calendar: Self.utcCalendar)

        #expect(summary.days == 7)
        #expect(summary.entryCount == 2)
        #expect(summary.totalCostUSD == 9)
        #expect(summary.totalTokens == 900)
        #expect(summary.totalRequests == 9)
    }

    @Test
    func `comparison periods are unique sorted and bounded by scanned history`() {
        let snapshot = Self.snapshot(historyDays: 90)

        #expect(snapshot.comparisonSummaries(periods: [30, 7, 90, 7], calendar: Self.utcCalendar).map(\.days) == [
            7,
            30,
        ])
    }

    @Test
    func `summary preserves unavailable totals as nil`() {
        let snapshot = CostUsageTokenSnapshot(
            sessionTokens: nil,
            sessionCostUSD: nil,
            last30DaysTokens: nil,
            last30DaysCostUSD: nil,
            historyDays: 30,
            daily: [Self.entry(day: "2026-07-01", cost: nil, tokens: nil, requests: nil)],
            updatedAt: Self.now)

        let summary = snapshot.summary(forLastDays: 7, calendar: Self.utcCalendar)
        #expect(summary.totalCostUSD == nil)
        #expect(summary.totalTokens == nil)
        #expect(summary.totalRequests == nil)
    }

    private static func snapshot(historyDays: Int) -> CostUsageTokenSnapshot {
        CostUsageTokenSnapshot(
            sessionTokens: 500,
            sessionCostUSD: 5,
            last30DaysTokens: 1000,
            last30DaysCostUSD: 10,
            historyDays: historyDays,
            daily: [
                self.entry(day: "2026-06-01", cost: 1, tokens: 100, requests: 1),
                self.entry(day: "2026-06-25", cost: 4, tokens: 400, requests: 4),
                self.entry(day: "2026-07-01", cost: 5, tokens: 500, requests: 5),
            ],
            updatedAt: self.now)
    }

    private static func entry(day: String, cost: Double?, tokens: Int?, requests: Int?)
        -> CostUsageDailyReport.Entry
    {
        CostUsageDailyReport.Entry(
            date: day,
            inputTokens: nil,
            outputTokens: nil,
            totalTokens: tokens,
            requestCount: requests,
            costUSD: cost,
            modelsUsed: nil,
            modelBreakdowns: nil)
    }

    private static let now = Date(timeIntervalSince1970: 1_782_864_000) // 2026-07-01 00:00:00 UTC
    private static var utcCalendar: Calendar {
        var calendar = Calendar(identifier: .gregorian)
        calendar.timeZone = TimeZone(secondsFromGMT: 0)!
        return calendar
    }
}