Fix filter bug

fummicc1 · Nov 8, 2024 · b19c4d3 · b19c4d3
1 parent 6b8f2e3
commit b19c4d3
Showing 1 changed file with 5 additions and 8 deletions.
diff --git a/Sources/Firestore/FirestoreClient.swift b/Sources/Firestore/FirestoreClient.swift
@@ -118,26 +118,23 @@ public struct FirestoreEqualFilter: FirestoreQueryFilter {
 public struct FirestoreContainFilter: FirestoreQueryFilter {
 
     public var fieldPath: String?
-    public var value: [Any]
+    public var value: Any
 
-    public init(fieldPath: String?, value: [Any]) {
+    public init(fieldPath: String?, value: Any) {
         self.fieldPath = fieldPath
         self.value = value
     }
 
     public func build(from: Query) -> Query {
-        guard let fieldPath = fieldPath, !value.isEmpty else {
+        guard let fieldPath = fieldPath else {
             return from
         }
-        return from.whereField(fieldPath, in: value)
+        return from.whereField(fieldPath, arrayContains: value)
     }
 
     public func build<Model>(type: Model.Type) -> Query where Model: FirestoreModel {
         let from = Firestore.firestore().collection(type.collectionName)
-        guard let fieldPath = fieldPath, !value.isEmpty else {
-            return from
-        }
-        return from.whereField(fieldPath, in: value)
+        return build(from: from)
     }
 }