//
// Copyright 2025 Element Creations Ltd.
// Copyright 2024-2025 New Vector Ltd.
//
// SPDX-License-Identifier: AGPL-3.0-only OR LicenseRef-Element-Commercial.
// Please see LICENSE files in the repository root for full details.
//

import Foundation
import MatrixRustSDK
import MatrixRustSDKMocks

extension AuthenticationClientFactoryMock {
    struct Configuration {
        var homeserverClients = [
            "matrix.org": ClientSDKMock(configuration: .init()),
            "https://matrix-client.matrix.org": ClientSDKMock(configuration: .init()),
            "example.com": ClientSDKMock(configuration: .init(serverAddress: "example.com",
                                                              homeserverURL: "https://matrix.example.com",
                                                              slidingSyncVersion: .native,
                                                              oAuthLoginURL: nil,
                                                              supportsOAuthCreatePrompt: false,
                                                              supportsPasswordLogin: true)),
            "company.com": ClientSDKMock(configuration: .init(serverAddress: "company.com",
                                                              homeserverURL: "https://matrix.company.com",
                                                              slidingSyncVersion: .native,
                                                              oAuthLoginURL: "https://auth.company.com/login",
                                                              supportsOAuthCreatePrompt: false,
                                                              supportsPasswordLogin: false)),
            "server.net": ClientSDKMock(configuration: .init(serverAddress: "server.net",
                                                             homeserverURL: "https://matrix.server.net",
                                                             slidingSyncVersion: .native,
                                                             oAuthLoginURL: nil,
                                                             supportsOAuthCreatePrompt: false,
                                                             supportsPasswordLogin: false)),
            "secure.gov": ClientSDKMock(configuration: .init(serverAddress: "secure.gov",
                                                             homeserverURL: "https://ess.secure.gov",
                                                             slidingSyncVersion: .native,
                                                             oAuthLoginURL: "https://auth.secure.gov/login",
                                                             supportsOAuthCreatePrompt: false,
                                                             supportsPasswordLogin: false,
                                                             elementWellKnown: "{\"version\":1,\"enforce_element_pro\":true}"))
        ]
    }
    
    convenience init(configuration: Configuration) {
        self.init()
        
        makeClientHomeserverAddressSessionDirectoriesPassphraseClientSessionDelegateAppSettingsAppHooksClosure = { address, _, _, _, _, _ in
            guard let client = configuration.homeserverClients[address] else {
                throw ClientBuildError.ServerUnreachable(message: "Not a known homeserver.")
            }
            return client
        }
        
        makeInMemoryClientHomeserverAddressClientSessionDelegateAppSettingsAppHooksClosure = { address, _, _, _ in
            guard let client = configuration.homeserverClients[address] else {
                throw ClientBuildError.ServerUnreachable(message: "Not a known homeserver.")
            }
            return client
        }
    }
}