passepartout-apple/Passepartout/App/Views/AddHostViewModel.swift

116 lines
3.5 KiB
Swift

//
// AddHostViewModel.swift
// Passepartout
//
// Created by Davide De Rosa on 2/14/22.
// Copyright (c) 2023 Davide De Rosa. All rights reserved.
//
// https://github.com/passepartoutvpn
//
// This file is part of Passepartout.
//
// Passepartout is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// Passepartout is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with Passepartout. If not, see <http://www.gnu.org/licenses/>.
//
import Foundation
import PassepartoutLibrary
import TunnelKitOpenVPN
import TunnelKitWireGuard
extension AddHostView {
struct ViewModel: Equatable {
private var isNamePreset = false
var profileName = ""
private(set) var requiresPassphrase = false
var encryptionPassphrase = ""
var processedProfile: Profile = .placeholder
private(set) var errorMessage: String?
var isAskingOverwrite = false
mutating func presetName(withURL url: URL) {
guard !isNamePreset else {
return
}
isNamePreset = true
profileName = url.filename
}
@MainActor
mutating func processURL(
_ url: URL,
with profileManager: ProfileManager,
replacingExisting: Bool,
deletingURLOnSuccess: Bool
) {
profileName = profileName.stripped
guard !profileName.isEmpty else {
return
}
if !replacingExisting {
guard !profileManager.isExistingProfile(withName: profileName) else {
isAskingOverwrite = true
return
}
}
errorMessage = nil
do {
let profile = try profileManager.profile(
withHeader: .init(name: profileName),
fromURL: url,
passphrase: encryptionPassphrase
)
processedProfile = profile
if deletingURLOnSuccess {
try? FileManager.default.removeItem(at: url)
}
} catch {
if case Passepartout.ProfileError.decryptionFailure = error {
requiresPassphrase = true
} else {
requiresPassphrase = false
}
setErrorMessage(for: error)
}
}
@MainActor
mutating func addProcessedProfile(to profileManager: ProfileManager) -> Bool {
guard !processedProfile.isPlaceholder else {
assertionFailure("Saving profile without processing first?")
return false
}
errorMessage = nil
profileManager.saveProfile(processedProfile, isActive: nil)
return true
}
private mutating func setErrorMessage(for error: Error) {
setErrorMessage(AppError(error).localizedDescription)
}
private mutating func setErrorMessage(_ message: String) {
errorMessage = message.withTrailingDot
}
}
}