Merge pull request #163 from gtiosclub/feed-sameer-getUserNumResponses

Added await getUserNumResponses() function
gtiosclub · Mar 3, 2025 · d00bf43 · d00bf43
2 parents 5671ac0 + cdd137f
commit d00bf43
Show file tree

Hide file tree

Showing 2 changed files with 37 additions and 2 deletions.
diff --git a/Gauge/Feed/ViewModels/PostFirebase.swift b/Gauge/Feed/ViewModels/PostFirebase.swift
@@ -500,7 +500,7 @@ class PostFirebase: ObservableObject {
     }
 
 
-    func deleteComment(postId: String, commentId: String){
+    func deleteComment(postId: String, commentId: String) {
         Firebase.db.collection("POSTS").document(postId).collection("COMMENTS").document(commentId).delete(){ error in
             if let error = error{
                 print("Error deleting Comment: \(error)")
@@ -523,6 +523,26 @@ class PostFirebase: ObservableObject {
         }
     }
 
+    func getUserNumResponses(postIds: [String]) async -> Int? {
+        do {
+            var totalResponses = 0
+
+            for postId in postIds {
+                let documentRef = Firebase.db.collection("POSTS").document(postId).collection("RESPONSES")
+                let querySnapshot = try await documentRef.getDocuments()
+                let count = querySnapshot.documents.count
+                print("Number of responses under \(postId): \(count)")
+                totalResponses += count
+            }
+
+            return totalResponses
+        } catch {
+            print("Error getting responses: \(error)")
+            return nil
+        }
+    }
+
+
     func removeView(postId: String, userId: String) {
         let viewRef = Firebase.db.collection("POSTS")
             .document(postId)

diff --git a/Gauge/Feed/Views/FirebaseTesting.swift b/Gauge/Feed/Views/FirebaseTesting.swift
@@ -111,7 +111,7 @@ struct FirebaseTesting: View {
                         }
                     }
 
-                    Section("Read Data") {
+                    Section(header: Text("Read Data")) {
                         Button("Get posts by userId") {
                             userVM.getPosts(userId: "tfeGCRCgt8UbJhCmKgNmuIFVzD73") { postIds in
                                 self.postIds = postIds
@@ -139,6 +139,21 @@ struct FirebaseTesting: View {
                                 print(object)
                             }
                         }
+
+                        Button("Get number of responses for a list of posts") {
+                            Task {
+                                if let count = await postVM.getUserNumResponses(postIds: [
+                                    "B2A9F081-A10C-4957-A6B8-0295F0C700A2",
+                                    "examplePost"
+                                ]) {
+                                    print("Number of responses: \(count)")
+                                } else {
+                                    print("Failed to get number of responses")
+                                }
+                            }
+                        }
+
+
                     }
 
                     Section("Update Data") {