RouterOS/RouterOSAssistant/Core/Networking/SSHTransport.swift

import Foundation
import Citadel
import NIOCore
import NIOSSH

/// SSH+CLI transport fallback for RouterOS devices/firmware without the REST API (pre-7.1).
final class SSHTransport: RouterOSTransport {
    let kind: RouterOSTransportKind = .ssh

    private let credentials: RouterOSCredentials
    private let hostKeyTrust: SSHHostKeyTrustStore
    private var client: SSHClient?

    init(credentials: RouterOSCredentials, hostKeyTrust: SSHHostKeyTrustStore = SSHHostKeyTrustStore()) {
        self.credentials = credentials
        self.hostKeyTrust = hostKeyTrust
    }

    func connect() async throws {
        do {
            client = try await SSHClient.connect(
                host: credentials.host,
                port: credentials.sshPort,
                authenticationMethod: .passwordBased(username: credentials.username, password: credentials.password),
                hostKeyValidator: .custom(self),
                reconnect: .never,
                // RouterOS' SSH server typically only offers legacy algorithms
                // (diffie-hellman-group14-sha1 key exchange, RSA host keys) that
                // Citadel's defaults don't include — `.all` adds them.
                algorithms: .all
            )
        } catch let error as RouterOSError {
            throw error
        } catch {
            // NIOSSHError's `.localizedDescription` is a useless generic NSError-bridged string
            // ("The operation couldn't be completed."); its real diagnostics only surface via
            // CustomStringConvertible, which `String(describing:)` picks up.
            throw RouterOSError.transportUnavailable("SSH-Verbindung fehlgeschlagen: \(String(describing: error))")
        }
    }

    func fetchDeviceInfo() async throws -> RouterDeviceInfo {
        let output = try await run("/system resource print without-paging")
        return RouterOSCliParser.parseDeviceInfo(output)
    }

    func fetchInterfaces() async throws -> [NetworkInterface] {
        let output = try await run("/interface print without-paging terse")
        return RouterOSCliParser.parseInterfaces(output)
    }

    func disconnect() async {
        try? await client?.close()
        client = nil
    }

    /// Full human-readable config export (`/export terse`), used for local backups.
    func exportConfiguration() async throws -> String {
        try await run("/export terse")
    }

    func fetchFirewallRuleCounts() async throws -> FirewallRuleCounts {
        let filterOutput = try await run("/ip firewall filter print count-only")
        let natOutput = try await run("/ip firewall nat print count-only")
        let filterCount = Int(filterOutput.trimmingCharacters(in: .whitespacesAndNewlines)) ?? 0
        let natCount = Int(natOutput.trimmingCharacters(in: .whitespacesAndNewlines)) ?? 0
        return FirewallRuleCounts(filterRuleCount: filterCount, natRuleCount: natCount)
    }

    /// RouterOS 7.24.2 (verified live against a hEX test device) does *not* include `.id` in
    /// `print terse` CLI output, unlike REST's JSON, which always does — every field parses
    /// fine, but every item's `id` would otherwise fall back to a synthetic, unusable "row-N".
    /// `:put [<menuPath> find]` returns the real internal IDs (e.g. "*C;*1;*2") in the same
    /// order as `print terse` (also verified live: position 0 → first ID, position 1 → second),
    /// so they're overlaid onto the parsed items by position here.
    func fetchMenuItems(menuPath: String, restPath: String) async throws -> [RouterOSMenuItem] {
        let output: String
        do {
            output = try await run("\(menuPath) print without-paging terse")
        } catch RouterOSError.invalidResponse(let detail) where detail.contains("bad parameter terse") {
            // Singleton menu (e.g. "/ip dns", "/system identity") — no list, no "terse" support.
            let plain = try await run("\(menuPath) print without-paging")
            return [RouterOSCliParser.parseSingletonItem(plain)]
        }
        var items = RouterOSCliParser.parseGenericItems(output)

        let idOutput = try await run(":put [\(menuPath) find]")
        let ids = idOutput
            .trimmingCharacters(in: .whitespacesAndNewlines)
            .split(separator: ";")
            .map(String.init)
        for index in items.indices where index < ids.count {
            items[index] = RouterOSMenuItem(id: ids[index], fields: items[index].fields)
        }
        return items
    }

    /// `:foreach i in=[<menuPath> find whereField=whereValue] do={:put [<menuPath> get $i
    /// returnField]}` — built from two independently confirmed-live primitives only: a bare
    /// `find` with one condition (verified repeatedly this session, e.g.
    /// `find dynamic=no` reliably returning exactly the right id), and `get <id> field` on a
    /// single, already-known id (standard, unambiguous RouterOS syntax). An earlier version tried
    /// `get [find ...] returnField` as one combined call to do this in a single round trip —
    /// that specific combined form was never actually verified and was confirmed live to be
    /// wrong (a real make-static conversion — confirmed via Winbox — wasn't found by it). `:put`
    /// inside `:foreach` prints one value per line, so this splits on newlines, not ";" (the
    /// semicolon-joined shape only applies to a single `:put [<menuPath> find ...]` list).
    func fetchFieldValues(menuPath: String, restPath: String, whereField: String, whereValue: String, returnField: String) async throws -> Set<String> {
        let script = ":foreach i in=[\(menuPath) find \(whereField)=\(whereValue)] do={:put [\(menuPath) get $i \(returnField)]}"
        let output = try await run(script)
        let values = output
            .split(whereSeparator: \.isNewline)
            .map { $0.trimmingCharacters(in: .whitespaces) }
            .filter { !$0.isEmpty }
        return Set(values)
    }

    /// RouterOS' SSH CLI exits 0 even when a command fails — confirmed live: both
    /// `/ip dhcp-server add ...` on an interface that already has one ("failure: server or
    /// relay with such interface already exists") and an invalid action ("syntax error (line 1
    /// column 46)") returned exit status 0, meaning `run()`'s exit-code check alone silently
    /// treats every such failure as success. A mutating command (add/set/remove) is always
    /// silent on success in every case observed live this session (15+ menu families) — so any
    /// non-empty output here is treated as the error text, since there is no more reliable
    /// signal available over this transport.
    func apply(_ command: RouterOSCommand) async throws {
        let output = try await run(command.cliLine)
        let trimmed = output.trimmingCharacters(in: .whitespacesAndNewlines)
        guard trimmed.isEmpty else {
            throw RouterOSError.invalidResponse(trimmed)
        }
    }

    /// Restores RouterOS' own vendor-default configuration and reboots the device. See
    /// FactoryResetService for why this bypasses the RouterOSCommand add/set model entirely.
    func resetToFactoryDefaults() async throws {
        _ = try await run("/system reset-configuration no-defaults=no skip-backup=no")
    }

    /// Uploads a `.rsc` script's text content to the router's own file storage via SFTP — the
    /// only documented way found to get a file from this Mac onto the router (RouterOS' SSH
    /// server has no documented SCP support, but DOES accept SFTP: confirmed live against a real
    /// hEX with a plain `sftp` CLI session; Citadel, already a dependency, ships an SFTP client).
    /// `remoteName` must be the full `flash/...`-prefixed form — confirmed live that RouterOS'
    /// `/file` and `/import` reject a bare filename ("file does not exist") even though `/file
    /// print` lists the same file that way too.
    func uploadScript(remoteName: String, contents: String) async throws {
        guard let client else { throw RouterOSError.notConnected }
        let sftp = try await client.openSFTP()
        do {
            try await sftp.withFile(filePath: remoteName, flags: [.write, .create, .truncate]) { file in
                try await file.write(ByteBuffer(string: contents))
            }
        } catch {
            try? await sftp.close()
            throw error
        }
        try await sftp.close()
    }

    /// The RouterOS-documented restore workflow in one command: wipe the entire configuration
    /// (`no-defaults=yes`, not even the vendor defaults — a genuinely blank slate) and
    /// immediately re-apply the given already-uploaded script. Safer than `/import`-ing straight
    /// over a live, different configuration, which MikroTik's own docs describe as needing a
    /// reset first (every `add`-type line in the script would otherwise risk colliding with
    /// whatever's already there). Reboots the device; like `resetToFactoryDefaults()`, the
    /// connection dying mid-command is the expected outcome, not a failure — ignored here the
    /// same way.
    func applyRestoreScript(remoteName: String) async throws {
        _ = try await run("/system reset-configuration no-defaults=yes run-after-reset=\(remoteName)")
    }

    /// Runs a command via `executeCommandStream` (not the simpler `executeCommand`), because
    /// `executeCommand` discards whatever output it already collected the moment the command
    /// exits non-zero — exactly the RouterOS error text we need. Collecting the stream ourselves
    /// keeps that text available even when the command fails.
    private func run(_ command: String) async throws -> String {
        guard let client else { throw RouterOSError.notConnected }

        var output = ""
        do {
            let stream = try await client.executeCommandStream(command)
            for try await chunk in stream {
                switch chunk {
                case .stdout(let buffer), .stderr(let buffer):
                    output += String(buffer: buffer)
                }
            }
            return output
        } catch let failure as SSHClient.CommandFailed {
            let detail = output.trimmingCharacters(in: .whitespacesAndNewlines)
            throw RouterOSError.invalidResponse(
                "RouterOS meldete Fehler (Exit-Code \(failure.exitCode)) für \"\(command)\""
                    + (detail.isEmpty ? "" : ": \(detail)")
            )
        }
    }
}

extension SSHTransport: NIOSSHClientServerAuthenticationDelegate {
    func validateHostKey(hostKey: NIOSSHPublicKey, validationCompletePromise: EventLoopPromise<Void>) {
        let fingerprint = SSHHostKeyFingerprint.sha256(of: hostKey)
        if hostKeyTrust.isTrusted(host: credentials.host, fingerprint: fingerprint) {
            validationCompletePromise.succeed(())
        } else {
            validationCompletePromise.fail(RouterOSError.untrustedSSHHostKey(fingerprint: fingerprint))
        }
    }
}