import Foundation
import TermindApplication
import TermindDomain

import struct TermindDomain.Host

/// Saved-vault access exposed to transfer presentation and session entry points.
@Observable
final class TransferLibraryModel {
    private let library: HostLibraryModel

    init(library: HostLibraryModel) {
        self.library = library
    }
}

extension TransferLibraryModel: RemoteSessionHostStore {
    func host(_ id: Host.ID) -> Host? { library.host(id) }
    func networkConfigurationStamp(for host: Host) -> Data { library.networkConfigurationStamp(for: host) }
    func tailscaleNetwork(for host: Host) throws -> (any SSHNetworkDialing)? {
        try library.tailscaleNetwork(for: host)
    }
    func savedTailscaleUsername(for host: Host) -> String? { library.savedTailscaleUsername(for: host) }
    func tailscaleUsername(for host: Host) -> String { library.tailscaleUsername(for: host) }
    @discardableResult func saveTailscaleUsername(_ username: String, hostID: Host.ID) -> Bool {
        library.saveTailscaleUsername(username, hostID: hostID)
    }
    func identity(_ id: Identity.ID) -> Identity? { library.identity(id) }
    func identity(for host: Host) -> Identity? { library.identity(for: host) }
    func customIdentity(for hostID: Host.ID, purpose: HostCustomIdentityPurpose) -> Identity? {
        library.customIdentity(for: hostID, purpose: purpose)
    }
    var customPasswordStorage: PasswordStorage { library.customPasswordStorage }
    func key(_ id: SSHKeyRef.ID) -> SSHKeyRef? { library.key(id) }
    func knownHosts(for host: Host) -> [KnownHostEntry] { library.knownHosts(for: host) }
    func route(for host: Host) -> [Host] { library.route(for: host) }
    func expectedHostKeys(
        for host: Host, temporaryHostKeys: [Host.ID: SSHHostKeyIdentity],
        trustPolicy: SSHHostTrustPolicy
    ) throws -> Set<SSHHostKeyIdentity> {
        try library.expectedHostKeys(
            for: host, temporaryHostKeys: temporaryHostKeys,
            trustPolicy: trustPolicy
        )
    }
    func prepareSSHConnection(
        for host: Host, submissions: [Host.ID: SSHAuthenticationSubmission],
        proxySubmissions: [Host.ID: SSHProxyAuthenticationSubmission]
    ) throws -> SSHConnectionPreparation {
        try library.prepareSSHConnection(
            for: host, submissions: submissions, proxySubmissions: proxySubmissions)
    }
    func sshConnectionConfiguration(
        for host: Host, submissions: [Host.ID: SSHAuthenticationSubmission],
        proxySubmissions: [Host.ID: SSHProxyAuthenticationSubmission],
        temporaryHostKeys: [Host.ID: SSHHostKeyIdentity],
        trustPolicy: SSHHostTrustPolicy
    ) throws -> SSHConnectionConfiguration {
        try library.sshConnectionConfiguration(
            for: host, submissions: submissions, proxySubmissions: proxySubmissions,
            temporaryHostKeys: temporaryHostKeys,
            trustPolicy: trustPolicy
        )
    }
    func sshJumpConfigurations(
        for host: Host, submissions: [Host.ID: SSHAuthenticationSubmission],
        proxySubmissions: [Host.ID: SSHProxyAuthenticationSubmission],
        temporaryHostKeys: [Host.ID: SSHHostKeyIdentity],
        trustPolicy: SSHHostTrustPolicy
    ) throws -> [SSHRouteHopConfiguration] {
        try library.sshJumpConfigurations(
            for: host, submissions: submissions, proxySubmissions: proxySubmissions,
            temporaryHostKeys: temporaryHostKeys,
            trustPolicy: trustPolicy
        )
    }
    func prepareSSHConnection(
        for route: SSHConnectionRoute, submissions: [Host.ID: SSHAuthenticationSubmission],
        proxySubmissions: [Host.ID: SSHProxyAuthenticationSubmission]
    ) throws -> SSHConnectionPreparation {
        try library.prepareSSHConnection(
            for: route, submissions: submissions, proxySubmissions: proxySubmissions)
    }
    func sshJumpConfigurations(
        for route: SSHConnectionRoute, submissions: [Host.ID: SSHAuthenticationSubmission],
        proxySubmissions: [Host.ID: SSHProxyAuthenticationSubmission],
        temporaryHostKeys: [Host.ID: SSHHostKeyIdentity], trustPolicy: SSHHostTrustPolicy
    ) throws -> [SSHRouteHopConfiguration] {
        try library.sshJumpConfigurations(
            for: route, submissions: submissions, proxySubmissions: proxySubmissions,
            temporaryHostKeys: temporaryHostKeys, trustPolicy: trustPolicy
        )
    }
    func saveConnectionRoute(
        _ route: SSHConnectionRoute, authentication: SSHAuthenticationSubmission?,
        authenticationHostID: Host.ID?, suppliedPassword: String?
    ) throws -> SSHConnectionRoute {
        try library.saveConnectionRoute(
            route, authentication: authentication, authenticationHostID: authenticationHostID,
            suppliedPassword: suppliedPassword
        )
    }
    func transportProxy(for host: Host) throws -> SSHTransportProxy? {
        try library.transportProxy(for: host)
    }
    func transportProxy(
        for host: Host, submission: SSHProxyAuthenticationSubmission?
    ) throws -> SSHTransportProxy? {
        try library.transportProxy(for: host, submission: submission)
    }
    func sshCredentials(for identity: Identity) -> SSHCredentialResolution {
        library.sshCredentials(for: identity)
    }
    func passwordSecret(for identity: Identity) throws -> SSHSecret? {
        try library.passwordSecret(for: identity)
    }
    func passwordAvailability(for identity: Identity) -> SecretAvailability {
        library.passwordAvailability(for: identity)
    }
    func keySecretCredential(for key: SSHKeyRef) throws -> SSHResolvedKeyCredential? {
        try library.keySecretCredential(for: key)
    }
    func setConnectionStatus(
        _ status: ConnectionStatus, connectionID: UUID, hostID: Host.ID
    ) {
        library.setConnectionStatus(status, connectionID: connectionID, hostID: hostID)
    }
    func recordRemoteSystemInfo(_ systemInfo: RemoteSystemInfo?, for hostID: Host.ID) {
        library.recordRemoteSystemInfo(systemInfo, for: hostID)
    }
    func trust(_ entry: KnownHostEntry) { library.trust(entry) }
    @discardableResult func upsert(_ identity: Identity) -> Bool { library.upsert(identity) }
    func upsert(_ identity: Identity, passwordUpdate: IdentityPasswordUpdate) throws {
        try library.upsert(identity, passwordUpdate: passwordUpdate)
    }
    func upsert(_ host: Host) { library.upsert(host) }
}