RouterOS/RouterOSAssistant/Core/Models/RouterOSCommand.swift

import Foundation

/// A single RouterOS configuration change, expressed once and executed over either transport
/// (rendered as a CLI line for SSH, as a JSON body — plus a lookup for `.set` — for REST).
struct RouterOSCommand: Equatable, Identifiable {
    enum Operation: Equatable {
        /// Creates a new item under `restPath` (SSH: `<menuPath> add ...`, REST: `POST`).
        case add
        /// Modifies an existing item matched by one field's value (SSH: `<menuPath> set [find
        /// field=value] ...`; REST has no such inline lookup, so it needs a GET first to find
        /// the item's `.id`, then `PATCH restPath/<id>`).
        case set(matchField: String, matchValue: String)
        /// Deletes an existing item matched by one field's value (SSH: `<menuPath> remove [find
        /// field=value]`; REST again needs a GET-for-id first, then `DELETE restPath/<id>`).
        case remove(matchField: String, matchValue: String)
    }

    var id: String {
        switch operation {
        case .add:
            return "add:\(menuPath):\(summary)"
        case .set(let field, let value):
            return "set:\(menuPath):\(field)=\(value):\(summary)"
        case .remove(let field, let value):
            return "remove:\(menuPath):\(field)=\(value):\(summary)"
        }
    }

    /// RouterOS menu path without the action word, e.g. "/ip address" or "/interface wireless".
    let menuPath: String
    /// REST resource path under `/rest/`, e.g. "ip/address" or "interface/wireless".
    let restPath: String
    let operation: Operation
    /// RouterOS "words" (key=value fields) to add or change.
    let arguments: [String: String]
    /// Human-readable description shown on the review screen before applying.
    let summary: String

    static func add(menuPath: String, restPath: String, arguments: [String: String], summary: String) -> RouterOSCommand {
        RouterOSCommand(menuPath: menuPath, restPath: restPath, operation: .add, arguments: arguments, summary: summary)
    }

    static func set(
        menuPath: String,
        restPath: String,
        matchField: String,
        matchValue: String,
        arguments: [String: String],
        summary: String
    ) -> RouterOSCommand {
        RouterOSCommand(
            menuPath: menuPath,
            restPath: restPath,
            operation: .set(matchField: matchField, matchValue: matchValue),
            arguments: arguments,
            summary: summary
        )
    }

    static func remove(
        menuPath: String,
        restPath: String,
        matchField: String,
        matchValue: String,
        summary: String
    ) -> RouterOSCommand {
        RouterOSCommand(
            menuPath: menuPath,
            restPath: restPath,
            operation: .remove(matchField: matchField, matchValue: matchValue),
            arguments: [:],
            summary: summary
        )
    }

    /// Renders as a RouterOS CLI line, e.g. `/ip address add address=192.168.88.1/24 interface=bridge`
    /// or `/interface wireless set [find name=wlan1] ssid=Home`.
    var cliLine: String {
        let args = renderedArguments
        switch operation {
        case .add:
            return args.isEmpty ? "\(menuPath) add" : "\(menuPath) add \(args)"
        case .set(let field, let value):
            // Empty matchField = singleton menu (e.g. "/ip dns", "/system identity"): these hold
            // exactly one item and are "set" directly, with no "[find ...]" — there's nothing to
            // find. Confirmed live against RouterOS 7.24.2.
            guard !field.isEmpty else {
                return args.isEmpty ? "\(menuPath) set" : "\(menuPath) set \(args)"
            }
            let finder = "[find \(field)=\(Self.quoteIfNeeded(value))]"
            return args.isEmpty ? "\(menuPath) set \(finder)" : "\(menuPath) set \(finder) \(args)"
        case .remove(let field, let value):
            return "\(menuPath) remove [find \(field)=\(Self.quoteIfNeeded(value))]"
        }
    }

    private var renderedArguments: String {
        arguments
            .sorted { $0.key < $1.key }
            .map { "\($0.key)=\(Self.quoteIfNeeded($0.value))" }
            .joined(separator: " ")
    }

    private static func quoteIfNeeded(_ value: String) -> String {
        value.contains(" ") ? "\"\(value)\"" : value
    }
}