wehub-resource-sync
64 lines
2.5 KiB
Swift
64 lines
2.5 KiB
Swift
import Foundation
|
|
|
|
public enum Sub2APIProviderDescriptor {
|
|
public static func primaryLabel(details: Sub2APIUsageDetails?) -> String? {
|
|
details?.kind == .subscription ? "Daily quota" : nil
|
|
}
|
|
|
|
public static let descriptor = ProviderDescriptor(
|
|
id: .sub2api,
|
|
metadata: ProviderMetadata(
|
|
id: .sub2api,
|
|
displayName: "sub2api",
|
|
sessionLabel: "Quota",
|
|
weeklyLabel: "Weekly quota",
|
|
opusLabel: "Monthly quota",
|
|
supportsOpus: true,
|
|
supportsCredits: false,
|
|
creditsHint: "Reads key quota, subscription limits, usage, and wallet balance from /v1/usage.",
|
|
toggleTitle: "Show sub2api usage",
|
|
cliName: "sub2api",
|
|
defaultEnabled: false,
|
|
dashboardURL: nil,
|
|
statusPageURL: nil),
|
|
branding: ProviderBranding(
|
|
iconStyle: .sub2api,
|
|
iconResourceName: "ProviderIcon-sub2api",
|
|
color: ProviderColor(red: 45 / 255, green: 198 / 255, blue: 216 / 255)),
|
|
tokenCost: ProviderTokenCostConfig(
|
|
supportsTokenCost: false,
|
|
noDataMessage: { "sub2api spend is reported by its usage API." }),
|
|
fetchPlan: ProviderFetchPlan(
|
|
sourceModes: [.auto, .api],
|
|
pipeline: ProviderFetchPipeline(resolveStrategies: { _ in [Sub2APIAPIFetchStrategy()] })),
|
|
cli: ProviderCLIConfig(
|
|
name: "sub2api",
|
|
aliases: ["sub-2-api"],
|
|
versionDetector: nil))
|
|
}
|
|
|
|
struct Sub2APIAPIFetchStrategy: ProviderFetchStrategy {
|
|
let id = "sub2api.api"
|
|
let kind: ProviderFetchKind = .apiToken
|
|
|
|
func isAvailable(_ context: ProviderFetchContext) async -> Bool {
|
|
Sub2APISettingsReader.apiKey(environment: context.env) != nil &&
|
|
Sub2APISettingsReader.baseURL(environment: context.env) != nil
|
|
}
|
|
|
|
func fetch(_ context: ProviderFetchContext) async throws -> ProviderFetchResult {
|
|
guard let apiKey = Sub2APISettingsReader.apiKey(environment: context.env) else {
|
|
throw Sub2APIUsageError.missingCredentials
|
|
}
|
|
guard let baseURL = Sub2APISettingsReader.baseURL(environment: context.env) else {
|
|
throw Sub2APIUsageError.missingBaseURL
|
|
}
|
|
let usage = try await Sub2APIUsageFetcher.fetchUsage(apiKey: apiKey, baseURL: baseURL)
|
|
return self.makeResult(usage: usage.toUsageSnapshot(), sourceLabel: "api")
|
|
}
|
|
|
|
func shouldFallback(on _: Error, context _: ProviderFetchContext) -> Bool {
|
|
false
|
|
}
|
|
}
|