feat: determines if migration needs to occur

brain-marks/ContentView.swift

@@ -8,24 +8,22 @@
 import SwiftUI
 
 struct ContentView: View {
-    let migrationService: MigrationService
+    let migrationService: MigrationService = MigrationService()
 
     @State private var showAddSheet = false
 
-    init(storageProvider: StorageProvider) {
-        self.migrationService = MigrationService(managedObjectContext: storageProvider.context)
-    }
-
     var body: some View {
         CategoryList()
             .onAppear {
-                migrationService.performMigration()
+                if migrationService.checkIfMigrationShouldRun() {
+                    migrationService.performMigration()
+                }
             }
     }
 }
 
 struct ContentView_Previews: PreviewProvider {
     static var previews: some View {
-        ContentView(storageProvider: StorageProvider())
+        ContentView()
     }
 }

brain-marks/Services/MigrationService.swift

@@ -10,18 +10,15 @@ import Foundation
 
 /// Controls the migration from Amplify to CoreData
 class MigrationService {
-    /// Need to get category from Amplify
-    /// Store Category inside CD
-    /// Loop through category and get tweets
-    /// Store tweet in CD and associate to proper category
+    private let storageProvider = StorageProvider.shared
     private var awsCategories: [AWSCategory] = []
 
     private let amplifyDataStore = DataStoreManger.shared
 
     private var managedObjectContext: NSManagedObjectContext
 
-    init(managedObjectContext: NSManagedObjectContext) {
-        self.managedObjectContext = managedObjectContext
+    init() {
+        self.managedObjectContext = storageProvider.context
     }
 
     func performMigration() {
@@ -38,12 +35,21 @@ class MigrationService {
                 }
             }
             do {
-            try managedObjectContext.save()
+                try managedObjectContext.save()
             } catch {
                 print("❌ MigrationService.performMigration() Error: \(error)")
             }
         }
+        UserDefaults.standard.set(true, forKey: "migrationToCoreDataRan")
+    }
+
+    public func checkIfMigrationShouldRun() -> Bool {
+        let migrationToCoreDataRan = UserDefaults.standard.bool(forKey: "migrationToCoreDataRan")
 
+        if storageProvider.getAllCategories().isEmpty && !migrationToCoreDataRan {
+            return true
+        }
+        return false
     }
 
     private func getCategories() {
@@ -52,7 +58,7 @@ class MigrationService {
             case .success(let categories):
                 self.awsCategories = categories
             case .failure(let error):
-                print("MigrationService.getCategories(): Error: \(error)")
+                print("❌ MigrationService.getCategories(): Error: \(error)")
             }
         })
     }

brain-marks/brain_marksApp.swift

@@ -12,7 +12,9 @@ import TelemetryClient
 
 @main
 struct brain_marksApp: App {
-
+    @AppStorage("migrationToCoreDataRan")
+    private var migrationToCoreDataRan: Bool = false
+
     init() {
         configureAmplify()
 
@@ -21,11 +23,9 @@ struct brain_marksApp: App {
         TelemetryManager.send(TelemetrySignals.appLaunchedRegularly)
     }
 
-    let storageProvider = StorageProvider.shared
-
     var body: some Scene {
         WindowGroup {
-            ContentView(storageProvider: storageProvider)
+            ContentView()
         }
     }