RouterOS/RouterOSAssistant/Features/Wizard/Steps/Connect/ConnectViewModel.swift

import Foundation

@MainActor
final class ConnectViewModel: ObservableObject {
    @Published var host: String = "192.168.88.1"
    @Published var username: String = "admin"
    @Published var password: String = ""
    @Published var rememberPassword: Bool = true

    let connectionService: ConnectionService
    private let keychain = KeychainService()

    init(connectionService: ConnectionService) {
        self.connectionService = connectionService
    }

    func onAppear() {
        if let saved = keychain.loadPassword(forHost: host, username: username) {
            password = saved
        }
    }

    func connect() {
        let credentials = RouterOSCredentials(host: host, username: username, password: password)
        if rememberPassword {
            keychain.save(password: password, forHost: host, username: username)
        }
        Task {
            await connectionService.connect(with: credentials)
        }
    }

    func trustAndRetry(fingerprint: String) {
        Task {
            await connectionService.trustCurrentCertificateAndRetry(fingerprint: fingerprint)
        }
    }

    func trustSSHHostKeyAndRetry(fingerprint: String) {
        Task {
            await connectionService.trustCurrentSSHHostKeyAndRetry(fingerprint: fingerprint)
        }
    }
}