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, .claude: ClaudeProviderDescriptor.descriptor, .cursor: CursorProviderDescriptor.descriptor, .opencode: OpenCodeProviderDescriptor.descriptor, .factory: FactoryProviderDescriptor.descriptor, .gemini: GeminiProviderDescriptor.descriptor, .antigravity: AntigravityProviderDescriptor.descriptor, .copilot: CopilotProviderDescriptor.descriptor, .zai: ZaiProviderDescriptor.descriptor, .minimax: MiniMaxProviderDescriptor.descriptor, .kimi: KimiProviderDescriptor.descriptor, .kiro: KiroProviderDescriptor.descriptor, .vertexai: VertexAIProviderDescriptor.descriptor, .augment: AugmentProviderDescriptor.descriptor, .jetbrains: JetBrainsProviderDescriptor.descriptor, .kimik2: KimiK2ProviderDescriptor.descriptor, .amp: AmpProviderDescriptor.descriptor, .ollama: OllamaProviderDescriptor.descriptor, .synthetic: SyntheticProviderDescriptor.descriptor, .openrouter: OpenRouterProviderDescriptor.descriptor, .warp: WarpProviderDescriptor.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 }}
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,
.claude: ClaudeProviderDescriptor.descriptor,
.cursor: CursorProviderDescriptor.descriptor,
.opencode: OpenCodeProviderDescriptor.descriptor,
.factory: FactoryProviderDescriptor.descriptor,
.gemini: GeminiProviderDescriptor.descriptor,
.antigravity: AntigravityProviderDescriptor.descriptor,
.copilot: CopilotProviderDescriptor.descriptor,
.zai: ZaiProviderDescriptor.descriptor,
.minimax: MiniMaxProviderDescriptor.descriptor,
.kimi: KimiProviderDescriptor.descriptor,
.kiro: KiroProviderDescriptor.descriptor,
.vertexai: VertexAIProviderDescriptor.descriptor,
.augment: AugmentProviderDescriptor.descriptor,
.jetbrains: JetBrainsProviderDescriptor.descriptor,
.kimik2: KimiK2ProviderDescriptor.descriptor,
.amp: AmpProviderDescriptor.descriptor,
.ollama: OllamaProviderDescriptor.descriptor,
.synthetic: SyntheticProviderDescriptor.descriptor,
.openrouter: OpenRouterProviderDescriptor.descriptor,
.warp: WarpProviderDescriptor.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
}
}