import CodexBarCoreimport Foundation enum ProviderImplementationRegistry { private final class Store: @unchecked Sendable { var ordered: [any ProviderImplementation] = [] var byID: [UsageProvider: any ProviderImplementation] = [:] } private static let lock = NSLock() private static let store = Store() private static func makeImplementation(for provider: UsageProvider) -> (any ProviderImplementation) { switch provider { case .codex: CodexProviderImplementation() case .claude: ClaudeProviderImplementation() case .cursor: CursorProviderImplementation() case .opencode: OpenCodeProviderImplementation() case .factory: FactoryProviderImplementation() case .gemini: GeminiProviderImplementation() case .antigravity: AntigravityProviderImplementation() case .copilot: CopilotProviderImplementation() case .zai: ZaiProviderImplementation() case .minimax: MiniMaxProviderImplementation() case .kimi: KimiProviderImplementation() case .kiro: KiroProviderImplementation() case .vertexai: VertexAIProviderImplementation() case .augment: AugmentProviderImplementation() case .jetbrains: JetBrainsProviderImplementation() case .kimik2: KimiK2ProviderImplementation() case .amp: AmpProviderImplementation() case .synthetic: SyntheticProviderImplementation() } } private static let bootstrap: Void = { for provider in UsageProvider.allCases { _ = ProviderImplementationRegistry.register(makeImplementation(for: provider)) } }() private static func ensureBootstrapped() { _ = self.bootstrap } @discardableResult static func register(_ implementation: any ProviderImplementation) -> any ProviderImplementation { self.lock.lock() defer { self.lock.unlock() } if self.store.byID[implementation.id] == nil { self.store.ordered.append(implementation) } self.store.byID[implementation.id] = implementation return implementation } static var all: [any ProviderImplementation] { self.ensureBootstrapped() self.lock.lock() defer { self.lock.unlock() } return self.store.ordered } static func implementation(for id: UsageProvider) -> (any ProviderImplementation)? { self.ensureBootstrapped() if let found = self.store.byID[id] { return found } return self.all.first(where: { $0.id == id }) }}
import CodexBarCore
import Foundation
enum ProviderImplementationRegistry {
private final class Store: @unchecked Sendable {
var ordered: [any ProviderImplementation] = []
var byID: [UsageProvider: any ProviderImplementation] = [:]
}
private static let lock = NSLock()
private static let store = Store()
private static func makeImplementation(for provider: UsageProvider) -> (any ProviderImplementation) {
switch provider {
case .codex: CodexProviderImplementation()
case .claude: ClaudeProviderImplementation()
case .cursor: CursorProviderImplementation()
case .opencode: OpenCodeProviderImplementation()
case .factory: FactoryProviderImplementation()
case .gemini: GeminiProviderImplementation()
case .antigravity: AntigravityProviderImplementation()
case .copilot: CopilotProviderImplementation()
case .zai: ZaiProviderImplementation()
case .minimax: MiniMaxProviderImplementation()
case .kimi: KimiProviderImplementation()
case .kiro: KiroProviderImplementation()
case .vertexai: VertexAIProviderImplementation()
case .augment: AugmentProviderImplementation()
case .jetbrains: JetBrainsProviderImplementation()
case .kimik2: KimiK2ProviderImplementation()
case .amp: AmpProviderImplementation()
case .synthetic: SyntheticProviderImplementation()
}
}
private static let bootstrap: Void = {
for provider in UsageProvider.allCases {
_ = ProviderImplementationRegistry.register(makeImplementation(for: provider))
}
}()
private static func ensureBootstrapped() {
_ = self.bootstrap
}
@discardableResult
static func register(_ implementation: any ProviderImplementation) -> any ProviderImplementation {
self.lock.lock()
defer { self.lock.unlock() }
if self.store.byID[implementation.id] == nil {
self.store.ordered.append(implementation)
}
self.store.byID[implementation.id] = implementation
return implementation
}
static var all: [any ProviderImplementation] {
self.ensureBootstrapped()
self.lock.lock()
defer { self.lock.unlock() }
return self.store.ordered
}
static func implementation(for id: UsageProvider) -> (any ProviderImplementation)? {
self.ensureBootstrapped()
if let found = self.store.byID[id] { return found }
return self.all.first(where: { $0.id == id })
}
}