import AppKit
import CodexBarCore
import Foundation
import SwiftUI

struct CrossModelProviderImplementation: ProviderImplementation {
    let id: UsageProvider = .crossmodel

    @MainActor
    func presentation(context _: ProviderPresentationContext) -> ProviderPresentation {
        ProviderPresentation { _ in "api" }
    }

    @MainActor
    func observeSettings(_ settings: SettingsStore) {
        _ = settings.crossModelAPIToken
    }

    @MainActor
    func settingsSnapshot(context: ProviderSettingsSnapshotContext) -> ProviderSettingsSnapshotContribution? {
        _ = context
        return nil
    }

    @MainActor
    func isAvailable(context: ProviderAvailabilityContext) -> Bool {
        if CrossModelSettingsReader.apiToken(environment: context.environment) != nil {
            return true
        }
        return !context.settings.crossModelAPIToken.trimmingCharacters(in: .whitespacesAndNewlines).isEmpty
    }

    @MainActor
    func settingsPickers(context _: ProviderSettingsContext) -> [ProviderSettingsPickerDescriptor] {
        []
    }

    @MainActor
    func settingsFields(context: ProviderSettingsContext) -> [ProviderSettingsFieldDescriptor] {
        [
            ProviderSettingsFieldDescriptor(
                id: "crossmodel-api-key",
                title: "API key",
                subtitle: "Stored in ~/.codexbar/config.json. "
                    + "Create a key in the CrossModel console at crossmodel.ai/console/api-keys.",
                kind: .secure,
                placeholder: "cm-...",
                binding: context.stringBinding(\.crossModelAPIToken),
                actions: [],
                isVisible: nil,
                onActivate: nil),
        ]
    }

    @MainActor
    func appendUsageMenuEntries(context: ProviderMenuUsageContext, entries: inout [ProviderMenuEntry]) {
        guard let usage = context.snapshot?.crossModelUsage else { return }

        entries.append(.text("\(L("Balance")): \(usage.balanceDisplay)", .primary))
        if let daily = usage.daily {
            entries.append(.text(
                "\(L("Today")): \(usage.currencyString(daily.cost)) · " +
                    "\(UsageFormatter.tokenCountString(daily.totalTokens)) \(L("tokens"))",
                .secondary))
        }
        if let weekly = usage.weekly {
            entries.append(.text(
                "\(L("Week")): \(usage.currencyString(weekly.cost)) · " +
                    "\(UsageFormatter.tokenCountString(weekly.requestCount)) \(L("requests"))",
                .secondary))
        }
        if let monthly = usage.monthly {
            entries.append(.text(
                "\(L("Month")): \(usage.currencyString(monthly.cost)) · " +
                    "\(UsageFormatter.tokenCountString(monthly.requestCount)) \(L("requests"))",
                .secondary))
        }
    }
}