chore: import upstream snapshot with attribution

2026-07-13 12:22:33 +08:00
commit fc4fcbab58
1848 changed files with 472303 additions and 0 deletions
@@ -0,0 +1,63 @@
+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
+    }
+}