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,169 @@
+import Foundation
+
+public struct ProviderTokenCostConfig: Sendable {
+    public let supportsTokenCost: Bool
+    public let noDataMessage: @Sendable () -> String
+
+    public init(supportsTokenCost: Bool, noDataMessage: @escaping @Sendable () -> String) {
+        self.supportsTokenCost = supportsTokenCost
+        self.noDataMessage = noDataMessage
+    }
+}
+
+public struct ProviderDescriptor: Sendable {
+    public let id: UsageProvider
+    public let metadata: ProviderMetadata
+    public let branding: ProviderBranding
+    public let tokenCost: ProviderTokenCostConfig
+    public let fetchPlan: ProviderFetchPlan
+    public let cli: ProviderCLIConfig
+
+    public init(
+        id: UsageProvider,
+        metadata: ProviderMetadata,
+        branding: ProviderBranding,
+        tokenCost: ProviderTokenCostConfig,
+        fetchPlan: ProviderFetchPlan,
+        cli: ProviderCLIConfig)
+    {
+        self.id = id
+        self.metadata = metadata
+        self.branding = branding
+        self.tokenCost = tokenCost
+        self.fetchPlan = fetchPlan
+        self.cli = cli
+    }
+
+    public func fetchOutcome(context: ProviderFetchContext) async -> ProviderFetchOutcome {
+        await self.fetchPlan.fetchOutcome(context: context, provider: self.id)
+    }
+
+    public func fetch(context: ProviderFetchContext) async throws -> ProviderFetchResult {
+        let outcome = await self.fetchOutcome(context: context)
+        return try outcome.result.get()
+    }
+}
+
+public enum ProviderDescriptorRegistry {
+    private final class Store: @unchecked Sendable {
+        var ordered: [ProviderDescriptor] = []
+        var byID: [UsageProvider: ProviderDescriptor] = [:]
+    }
+
+    private static let lock = NSLock()
+    private static let store = Store()
+    private static let descriptorsByID: [UsageProvider: ProviderDescriptor] = [
+        .codex: CodexProviderDescriptor.descriptor,
+        .openai: OpenAIAPIProviderDescriptor.descriptor,
+        .azureopenai: AzureOpenAIProviderDescriptor.descriptor,
+        .claude: ClaudeProviderDescriptor.descriptor,
+        .cursor: CursorProviderDescriptor.descriptor,
+        .opencode: OpenCodeProviderDescriptor.descriptor,
+        .opencodego: OpenCodeGoProviderDescriptor.descriptor,
+        .alibaba: AlibabaCodingPlanProviderDescriptor.descriptor,
+        .alibabatokenplan: AlibabaTokenPlanProviderDescriptor.descriptor,
+        .factory: FactoryProviderDescriptor.descriptor,
+        .gemini: GeminiProviderDescriptor.descriptor,
+        .antigravity: AntigravityProviderDescriptor.descriptor,
+        .copilot: CopilotProviderDescriptor.descriptor,
+        .devin: DevinProviderDescriptor.descriptor,
+        .zai: ZaiProviderDescriptor.descriptor,
+        .minimax: MiniMaxProviderDescriptor.descriptor,
+        .manus: ManusProviderDescriptor.descriptor,
+        .kimi: KimiProviderDescriptor.descriptor,
+        .kilo: KiloProviderDescriptor.descriptor,
+        .kiro: KiroProviderDescriptor.descriptor,
+        .vertexai: VertexAIProviderDescriptor.descriptor,
+        .augment: AugmentProviderDescriptor.descriptor,
+        .jetbrains: JetBrainsProviderDescriptor.descriptor,
+        .kimik2: KimiK2ProviderDescriptor.descriptor,
+        .moonshot: MoonshotProviderDescriptor.descriptor,
+        .amp: AmpProviderDescriptor.descriptor,
+        .t3chat: T3ChatProviderDescriptor.descriptor,
+        .ollama: OllamaProviderDescriptor.descriptor,
+        .synthetic: SyntheticProviderDescriptor.descriptor,
+        .openrouter: OpenRouterProviderDescriptor.descriptor,
+        .elevenlabs: ElevenLabsProviderDescriptor.descriptor,
+        .warp: WarpProviderDescriptor.descriptor,
+        .windsurf: WindsurfProviderDescriptor.descriptor,
+        .zed: ZedProviderDescriptor.descriptor,
+        .perplexity: PerplexityProviderDescriptor.descriptor,
+        .mimo: MiMoProviderDescriptor.descriptor,
+        .doubao: DoubaoProviderDescriptor.descriptor,
+        .sakana: SakanaProviderDescriptor.descriptor,
+        .abacus: AbacusProviderDescriptor.descriptor,
+        .mistral: MistralProviderDescriptor.descriptor,
+        .deepseek: DeepSeekProviderDescriptor.descriptor,
+        .codebuff: CodebuffProviderDescriptor.descriptor,
+        .crof: CrofProviderDescriptor.descriptor,
+        .venice: VeniceProviderDescriptor.descriptor,
+        .commandcode: CommandCodeProviderDescriptor.descriptor,
+        .qoder: QoderProviderDescriptor.descriptor,
+        .stepfun: StepFunProviderDescriptor.descriptor,
+        .bedrock: BedrockProviderDescriptor.descriptor,
+        .grok: GrokProviderDescriptor.descriptor,
+        .groq: GroqProviderDescriptor.descriptor,
+        .llmproxy: LLMProxyProviderDescriptor.descriptor,
+        .litellm: LiteLLMProviderDescriptor.descriptor,
+        .deepgram: DeepgramProviderDescriptor.descriptor,
+        .poe: PoeProviderDescriptor.descriptor,
+        .chutes: ChutesProviderDescriptor.descriptor,
+        .crossmodel: CrossModelProviderDescriptor.descriptor,
+        .clawrouter: ClawRouterProviderDescriptor.descriptor,
+        .sub2api: Sub2APIProviderDescriptor.descriptor,
+        .wayfinder: WayfinderProviderDescriptor.descriptor,
+    ]
+    private static let bootstrap: Void = {
+        for provider in UsageProvider.allCases {
+            guard let descriptor = descriptorsByID[provider] else {
+                preconditionFailure("Missing ProviderDescriptor for \(provider.rawValue)")
+            }
+            _ = ProviderDescriptorRegistry.register(descriptor)
+        }
+    }()
+
+    private static func ensureBootstrapped() {
+        _ = self.bootstrap
+    }
+
+    @discardableResult
+    public static func register(_ descriptor: ProviderDescriptor) -> ProviderDescriptor {
+        self.lock.lock()
+        defer { self.lock.unlock() }
+        if self.store.byID[descriptor.id] == nil {
+            self.store.ordered.append(descriptor)
+        }
+        self.store.byID[descriptor.id] = descriptor
+        return descriptor
+    }
+
+    public static var all: [ProviderDescriptor] {
+        self.ensureBootstrapped()
+        self.lock.lock()
+        defer { self.lock.unlock() }
+        return self.store.ordered
+    }
+
+    public static var metadata: [UsageProvider: ProviderMetadata] {
+        Dictionary(uniqueKeysWithValues: self.all.map { ($0.id, $0.metadata) })
+    }
+
+    public static func descriptor(for id: UsageProvider) -> ProviderDescriptor {
+        self.ensureBootstrapped()
+        if let found = self.store.byID[id] { return found }
+        if let found = self.all.first(where: { $0.id == id }) { return found }
+        fatalError("Missing ProviderDescriptor for \(id.rawValue)")
+    }
+
+    public static var cliNameMap: [String: UsageProvider] {
+        self.ensureBootstrapped()
+        var map: [String: UsageProvider] = [:]
+        for descriptor in self.all {
+            map[descriptor.cli.name] = descriptor.id
+            for alias in descriptor.cli.aliases {
+                map[alias] = descriptor.id
+            }
+        }
+        return map
+    }
+}