Initial commit

2026-02-11 22:22:55 +01:00
commit 42f54954c1
58 changed files with 10639 additions and 0 deletions
--- a/oAI/Providers/ProviderRegistry.swift
+++ b/oAI/Providers/ProviderRegistry.swift
@@ -0,0 +1,102 @@
+//
+//  ProviderRegistry.swift
+//  oAI
+//
+//  Registry for managing multiple AI providers
+//
+
+import Foundation
+import os
+
+class ProviderRegistry {
+    static let shared = ProviderRegistry()
+    
+    private var providers: [Settings.Provider: AIProvider] = [:]
+    private let settings = SettingsService.shared
+    
+    private init() {}
+    
+    // MARK: - Get Provider
+    
+    func getProvider(for providerType: Settings.Provider) -> AIProvider? {
+        // Return cached provider if exists
+        if let provider = providers[providerType] {
+            return provider
+        }
+        
+        // Create new provider based on type
+        let provider: AIProvider?
+
+        switch providerType {
+        case .openrouter:
+            guard let apiKey = settings.openrouterAPIKey, !apiKey.isEmpty else {
+                Log.api.warning("No API key configured for OpenRouter")
+                return nil
+            }
+            provider = OpenRouterProvider(apiKey: apiKey)
+
+        case .anthropic:
+            if AnthropicOAuthService.shared.isAuthenticated {
+                // OAuth (Pro/Max subscription) takes precedence
+                provider = AnthropicProvider(oauth: true)
+            } else if let apiKey = settings.anthropicAPIKey, !apiKey.isEmpty {
+                provider = AnthropicProvider(apiKey: apiKey)
+            } else {
+                Log.api.warning("No API key or OAuth configured for Anthropic")
+                return nil
+            }
+
+        case .openai:
+            guard let apiKey = settings.openaiAPIKey, !apiKey.isEmpty else {
+                Log.api.warning("No API key configured for OpenAI")
+                return nil
+            }
+            provider = OpenAIProvider(apiKey: apiKey)
+
+        case .ollama:
+            provider = OllamaProvider(baseURL: settings.ollamaEffectiveURL)
+        }
+        
+        // Cache and return
+        if let provider = provider {
+            Log.api.info("Created \(providerType.rawValue) provider")
+            providers[providerType] = provider
+        }
+        
+        return provider
+    }
+    
+    // MARK: - Current Provider
+    
+    func getCurrentProvider() -> AIProvider? {
+        let currentProviderType = settings.defaultProvider
+        return getProvider(for: currentProviderType)
+    }
+    
+    // MARK: - Clear Cache
+    
+    func clearCache() {
+        providers.removeAll()
+    }
+    
+    // MARK: - Validate API Key
+    
+    func hasValidAPIKey(for providerType: Settings.Provider) -> Bool {
+        switch providerType {
+        case .openrouter:
+            return settings.openrouterAPIKey != nil && !settings.openrouterAPIKey!.isEmpty
+        case .anthropic:
+            return AnthropicOAuthService.shared.isAuthenticated
+                || (settings.anthropicAPIKey != nil && !settings.anthropicAPIKey!.isEmpty)
+        case .openai:
+            return settings.openaiAPIKey != nil && !settings.openaiAPIKey!.isEmpty
+        case .ollama:
+            return settings.ollamaConfigured
+        }
+    }
+
+    /// Providers that have credentials configured (API key or, for Ollama, a saved URL)
+    var configuredProviders: [Settings.Provider] {
+        Settings.Provider.allCases.filter { hasValidAPIKey(for: $0) }
+    }
+}