HostApp.swift

import Client
import ComposableArchitecture
import Foundation
import KeyboardShortcuts

#if canImport(LicenseManagement)
import ProHostApp
#endif

extension KeyboardShortcuts.Name {
    static let showHideWidget = Self("ShowHideWidget")
}

@Reducer
struct HostApp {
    @ObservableState
    struct State: Equatable {
        var general = General.State()
        var chatModelManagement = ChatModelManagement.State()
        var embeddingModelManagement = EmbeddingModelManagement.State()
    }

    enum Action {
        case appear
        case general(General.Action)
        case chatModelManagement(ChatModelManagement.Action)
        case embeddingModelManagement(EmbeddingModelManagement.Action)
    }

    @Dependency(\.toast) var toast
    
    init() {
        KeyboardShortcuts.userDefaults = .shared
    }

    var body: some ReducerOf<Self> {
        Scope(state: \.general, action: \.general) {
            General()
        }

        Scope(state: \.chatModelManagement, action: \.chatModelManagement) {
            ChatModelManagement()
        }

        Scope(state: \.embeddingModelManagement, action: \.embeddingModelManagement) {
            EmbeddingModelManagement()
        }

        Reduce { _, action in
            switch action {
            case .appear:
                #if canImport(ProHostApp)
                ProHostApp.start()
                #endif
                return .none

            case .general:
                return .none

            case .chatModelManagement:
                return .none

            case .embeddingModelManagement:
                return .none
            }
        }
    }
}

import Dependencies
import Keychain
import Preferences

struct UserDefaultsDependencyKey: DependencyKey {
    static var liveValue: UserDefaultsType = UserDefaults.shared
    static var previewValue: UserDefaultsType = {
        let it = UserDefaults(suiteName: "HostAppPreview")!
        it.removePersistentDomain(forName: "HostAppPreview")
        return it
    }()

    static var testValue: UserDefaultsType = {
        let it = UserDefaults(suiteName: "HostAppTest")!
        it.removePersistentDomain(forName: "HostAppTest")
        return it
    }()
}

extension DependencyValues {
    var userDefaults: UserDefaultsType {
        get { self[UserDefaultsDependencyKey.self] }
        set { self[UserDefaultsDependencyKey.self] = newValue }
    }
}

struct APIKeyKeychainDependencyKey: DependencyKey {
    static var liveValue: KeychainType = Keychain.apiKey
    static var previewValue: KeychainType = FakeKeyChain()
    static var testValue: KeychainType = FakeKeyChain()
}

extension DependencyValues {
    var apiKeyKeychain: KeychainType {
        get { self[APIKeyKeychainDependencyKey.self] }
        set { self[APIKeyKeychainDependencyKey.self] = newValue }
    }
}