RouterOS/RouterOSAssistant/Features/Backup/BackupViewModel.swift

import Foundation

@MainActor
final class BackupViewModel: ObservableObject {
    @Published private(set) var backups: [BackupRecord] = []
    @Published private(set) var isCreatingBackup = false
    @Published var errorMessage: String?
    @Published private(set) var backupDirectoryPath: String = BackupService.customDirectoryURL?.path
        ?? BackupService.defaultDirectoryURL.path

    @Published private(set) var isResettingToFactoryDefaults = false
    @Published var resetError: String?
    @Published private(set) var didSendFactoryReset = false

    @Published private(set) var isRestoring = false
    @Published var restoreError: String?
    @Published private(set) var didSendRestore = false

    private let backupService = BackupService()
    private let factoryResetService = FactoryResetService()

    func load() {
        backups = backupService.listBackups()
    }

    func setBackupDirectory(_ url: URL) {
        BackupService.customDirectoryURL = url
        backupDirectoryPath = url.path
        load()
    }

    func resetBackupDirectoryToDefault() {
        BackupService.customDirectoryURL = nil
        backupDirectoryPath = BackupService.defaultDirectoryURL.path
        load()
    }

    func createBackup(for credentials: RouterOSCredentials) {
        isCreatingBackup = true
        errorMessage = nil
        Task {
            do {
                _ = try await backupService.createBackup(for: credentials)
                load()
            } catch {
                errorMessage = error.localizedDescription
            }
            isCreatingBackup = false
        }
    }

    /// Backs up first (best-effort — a failed backup shouldn't block the reset itself, since
    /// RouterOS makes its own backup as part of the reset anyway), then restores the router's
    /// vendor-default configuration and reboots it.
    func resetToFactoryDefaults(for credentials: RouterOSCredentials) {
        isResettingToFactoryDefaults = true
        resetError = nil
        didSendFactoryReset = false
        Task {
            do {
                _ = try? await backupService.createBackup(for: credentials)
                try await factoryResetService.resetToVendorDefaults(for: credentials)
                didSendFactoryReset = true
            } catch {
                resetError = error.localizedDescription
            }
            isResettingToFactoryDefaults = false
        }
    }

    /// Wipes the router's config and replaces it with `record`'s. Model-compatibility checking
    /// happens in the View before this is ever called (needs `ConnectionService.deviceInfo`,
    /// which this view model doesn't hold) — this only performs the already-confirmed restore.
    func restore(_ record: BackupRecord, for credentials: RouterOSCredentials) {
        isRestoring = true
        restoreError = nil
        didSendRestore = false
        Task {
            do {
                try await backupService.restoreBackup(record, for: credentials)
                didSendRestore = true
            } catch {
                restoreError = error.localizedDescription
            }
            isRestoring = false
        }
    }
}