chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,72 @@
|
||||
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)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user