import CodexBarCore import Foundation struct Sub2APIProviderImplementation: ProviderImplementation { let id: UsageProvider = .sub2api @MainActor func presentation(context _: ProviderPresentationContext) -> ProviderPresentation { ProviderPresentation { _ in "api" } } @MainActor func observeSettings(_ settings: SettingsStore) { _ = settings.sub2APIAPIKey _ = settings.sub2APIBaseURL _ = settings.tokenAccountsData(for: .sub2api) } @MainActor func isAvailable(context: ProviderAvailabilityContext) -> Bool { Sub2APISettingsReader.apiKey(environment: context.environment) != nil && Sub2APISettingsReader.baseURL(environment: context.environment) != nil } @MainActor func settingsFields(context: ProviderSettingsContext) -> [ProviderSettingsFieldDescriptor] { [ ProviderSettingsFieldDescriptor( id: "sub2api-api-key", title: "Fallback API key", subtitle: "Used when no group API key account is selected.", kind: .secure, placeholder: "sk-…", binding: context.stringBinding(\.sub2APIAPIKey), actions: [], isVisible: nil, onActivate: nil), ProviderSettingsFieldDescriptor( id: "sub2api-base-url", title: "Base URL", subtitle: "Base URL of your sub2api instance. HTTPS is required except for local loopback testing.", kind: .plain, placeholder: "https://sub2api.example.com", binding: context.stringBinding(\.sub2APIBaseURL), actions: [], isVisible: nil, onActivate: nil), ] } @MainActor func appendUsageMenuEntries(context: ProviderMenuUsageContext, entries: inout [ProviderMenuEntry]) { guard let usage = context.snapshot?.sub2APIUsage else { return } if let balance = usage.balance { entries.append(.text( "\(L("Balance")): \(UsageFormatter.currencyString(balance, currencyCode: usage.unit))", .primary)) } if let today = usage.today { entries.append(.text("\(L("Today")): \(self.totalsText(today, unit: usage.unit))", .secondary)) } if let total = usage.total { entries.append(.text("\(L("Total")): \(self.totalsText(total, unit: usage.unit))", .secondary)) } } private func totalsText(_ totals: Sub2APIUsageDetails.Totals, unit: String) -> String { "\(UsageFormatter.tokenCountString(totals.requests)) \(L("requests")) · " + "\(UsageFormatter.tokenCountString(totals.totalTokens)) \(L("tokens")) · " + UsageFormatter.currencyString(totals.actualCostUSD, currencyCode: unit) } }