79 lines · 2.8 KB · Swift
RawDownload
import AppKitimport CodexBarCoreimport Foundationimport 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))        }    }}