passepartout-apple/Passepartout/Library/Sources/CommonLibrary/Business/MigrationManager.swift
Davide 9e5beff23a
Perform migrate + import in one step ()
- Drop the .importing / .imported steps
- Animate rows re-sorting during process
- Rephrase some strings better
- Test fake migration with launch argument
2024-11-16 21:16:25 +01:00

186 lines
5.8 KiB
Swift

//
// MigrationManager.swift
// Passepartout
//
// Created by Davide De Rosa on 11/13/24.
// Copyright (c) 2024 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 PassepartoutKit
@MainActor
public final class MigrationManager: ObservableObject {
public struct Simulation {
public let fakeProfiles: Bool
public let maxMigrationTime: Double?
public let randomFailures: Bool
public init(fakeProfiles: Bool, maxMigrationTime: Double?, randomFailures: Bool) {
self.fakeProfiles = fakeProfiles
self.maxMigrationTime = maxMigrationTime
self.randomFailures = randomFailures
}
}
private let profileStrategy: ProfileMigrationStrategy
private nonisolated let simulation: Simulation?
public init(
profileStrategy: ProfileMigrationStrategy? = nil,
simulation: Simulation? = nil
) {
self.profileStrategy = profileStrategy ?? DummyProfileStrategy()
self.simulation = simulation
}
}
// MARK: - Public interface
extension MigrationManager {
public func fetchMigratableProfiles() async throws -> [MigratableProfile] {
try await profileStrategy.fetchMigratableProfiles()
}
public func migratedProfile(withId profileId: UUID) async throws -> Profile? {
try await profileStrategy.fetchProfile(withId: profileId)
}
public func migratedProfiles(
_ migratableProfiles: [MigratableProfile],
onUpdate: @escaping @MainActor (UUID, MigrationStatus) -> Void
) async throws -> [Profile] {
migratableProfiles.forEach {
onUpdate($0.id, .pending)
}
return try await withThrowingTaskGroup(of: Profile?.self, returning: [Profile].self) { group in
migratableProfiles.forEach { migratable in
group.addTask {
do {
try await self.simulateBehavior()
guard let profile = try await self.simulateMigrateProfile(withId: migratable.id) else {
await onUpdate(migratable.id, .failed)
return nil
}
await onUpdate(migratable.id, .done)
return profile
} catch {
await onUpdate(migratable.id, .failed)
return nil
}
}
}
var profiles: [Profile] = []
for try await profile in group {
guard let profile else {
continue
}
profiles.append(profile)
}
return profiles
}
}
public func importProfiles(
_ profiles: [Profile],
into manager: ProfileManager,
onUpdate: @escaping @MainActor (UUID, MigrationStatus) -> Void
) async {
profiles.forEach {
onUpdate($0.id, .pending)
}
await withTaskGroup(of: Void.self) { group in
profiles.forEach { profile in
group.addTask {
do {
try await self.simulateBehavior()
try await self.simulateSaveProfile(profile, manager: manager)
await onUpdate(profile.id, .done)
} catch {
await onUpdate(profile.id, .failed)
}
}
}
}
}
public func deleteMigratableProfiles(withIds profileIds: Set<UUID>) async throws {
try await simulateDeleteProfiles(withIds: profileIds)
}
}
// MARK: - Simulation
private extension MigrationManager {
func simulateBehavior() async throws {
guard let simulation else {
return
}
if let maxMigrationTime = simulation.maxMigrationTime {
try await Task.sleep(for: .seconds(.random(in: 1.0..<maxMigrationTime)))
}
if simulation.randomFailures, Bool.random() {
throw PassepartoutError(.unhandled)
}
}
func simulateMigrateProfile(withId profileId: UUID) async throws -> Profile? {
if simulation?.fakeProfiles ?? false {
return try? Profile.Builder(id: profileId).tryBuild()
}
return try await profileStrategy.fetchProfile(withId: profileId)
}
func simulateSaveProfile(_ profile: Profile, manager: ProfileManager) async throws {
if simulation?.fakeProfiles ?? false {
return
}
try await manager.save(profile, force: true)
}
func simulateDeleteProfiles(withIds profileIds: Set<UUID>) async throws {
if simulation?.fakeProfiles ?? false {
return
}
try await profileStrategy.deleteProfiles(withIds: profileIds)
}
}
// MARK: - Dummy
private final class DummyProfileStrategy: ProfileMigrationStrategy {
public init() {
}
public func fetchMigratableProfiles() async throws -> [MigratableProfile] {
[]
}
func fetchProfile(withId profileId: UUID) async throws -> Profile? {
nil
}
func deleteProfiles(withIds profileIds: Set<UUID>) async throws {
}
}