Initialize Document with a Dictionary/Array primitive

orlandos-nl · May 18, 2017 · 979f7cd · 979f7cd
1 parent 40657e6
commit 979f7cd
Showing 1 changed file with 14 additions and 3 deletions.
diff --git a/Sources/BSON/Interpreting.swift b/Sources/BSON/Interpreting.swift
@@ -10,11 +10,22 @@ import Foundation
 
 extension Document {
     public init?(_ value: Primitive?) {
-        guard let value = value as? Document else {
-            return nil
+        if let value = value as? Document {
+            self = value
+            return
         }
 
-        self = value
+        if let dict = value as? [String: Primitive] {
+            self = Document(data: dict.makeBinary())
+            return
+        }
+
+        if let array = value as? [Primitive] {
+            self = Document(data: array.makeBinary())
+            return
+        }
+
+        return nil
     }
 }