Update new event with document ID returned from Firebase (#58)

mikaelacaron · Oct 4, 2023 · 06a450a · 06a450a
1 parent 783de05
commit 06a450a
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/Basic-Car-Maintenance/Shared/Dashboard/ViewModels/DashboardViewModel.swift b/Basic-Car-Maintenance/Shared/Dashboard/ViewModels/DashboardViewModel.swift
@@ -34,16 +34,20 @@ class DashboardViewModel: ObservableObject {
         if let uid = authenticationViewModel.user?.uid {
             var eventToAdd = maintenanceEvent
             eventToAdd.userID = uid
-            
-            try? Firestore
+
+            let eventAdded = try? Firestore
                 .firestore()
                 .collection("maintenance_events")
                 .addDocument(from: eventToAdd)
+
+            var event = maintenanceEvent
+            if let documentId = eventAdded?.documentID {
+                event.id = documentId
+            }
+            events.append(event)
         }
-
-        events.append(maintenanceEvent)
     }
-    
+
     func getMaintenanceEvents() async {
         if let uid = authenticationViewModel.user?.uid {
             let db = Firestore.firestore()