Fixed BSONDocument.init(predicate:)

PureSwift · Aug 21, 2023 · 1deaf94 · 1deaf94
1 parent ad8b287
commit 1deaf94
Showing 1 changed file with 9 additions and 7 deletions.
diff --git a/Sources/MongoDBModel/Predicate.swift b/Sources/MongoDBModel/Predicate.swift
@@ -46,15 +46,17 @@ public extension BSONDocument {
 
 public extension BSONDocument {
 
-    init?(predicate: Comparison) {
+    init?(predicate: FetchRequest.Predicate.Comparison) {
         // { <field>: { $eq: <value> } }
+        guard case let .keyPath(keyPath) = predicate.left,
+              let comparisonOperator = ComparisonQueryOperator(predicate: predicate.type),
+              case let .value(value) = predicate.right,
+              predicate.options.isEmpty,
+              predicate.modifier == nil,
+              let valueBSON = try? BSON(attributeValue: value) else {
             return nil
         }
-}
-
-public extension BSON {
-
-    init?(predicate: Expression) {
-        return nil
+        self = [
+            keyPath.rawValue: .document([comparisonOperator.rawValue: valueBSON])]
     }
 }