Add Macros

Author: saghm

src/lib.rs

@@ -54,8 +54,9 @@ pub use self::bson::{Bson, Document, Array};
 pub use self::encoder::{encode_document, EncoderResult, EncoderError};
 pub use self::decoder::{decode_document, DecoderResult, DecoderError};
 
-pub mod spec;
+pub mod macros;
 pub mod oid;
+pub mod spec;
 mod bson;
 mod encoder;
 mod decoder;

src/macros.rs

@@ -0,0 +1,32 @@
+#[macro_export]
+macro_rules! add_to_doc {
+    ($doc:expr, $key:expr => ($val:expr)) => {{
+        $doc.insert($key.to_owned(), ::std::convert::From::from($val));
+    }};
+
+    ($doc:expr, $key:expr => [$($val:expr),*]) => {{
+        let vec = vec![$(::std::convert::From::from($val)),*];
+        $doc.insert($key.to_owned(), $crate::Bson::Array(vec));
+    }};
+
+    ($doc:expr, $key:expr => { $($k:expr => $v:tt),* }) => {{
+        $doc.insert($key.to_owned(), $crate::Bson::Document(doc! {
+            $(
+                $k => $v
+            ),*
+        }));
+    }};
+}
+
+#[macro_export]
+macro_rules! doc {
+    ( $($key:expr => $val:tt),* ) => {{
+        let mut document = $crate::Document::new();
+
+        $(
+            add_to_doc!(document, $key => $val);
+        )*
+
+        document
+    }};
+}

tests/lib.rs

@@ -1,3 +1,4 @@
+#[macro_use(add_to_doc, doc)]
 extern crate bson;
 extern crate rustc_serialize;
 

tests/modules/macros.rs

@@ -0,0 +1,68 @@
+use bson::{self, Bson, Document};
+
+fn print_doc_with_indent_level(doc: &Document, n: i32) {
+    macro_rules! print_spaces {
+        ( $n:expr ) => {
+            for _ in 0..$n {
+                print!(" ");
+            }
+        };
+    }
+
+    println!("{{");
+
+    for (key, value) in doc.iter() {
+        print_spaces!(n + 4);
+        print!("{}: ", key);
+
+        print_bson_with_indent_level(value, n);
+        println!("");
+    }
+
+    print_spaces!(n);
+    println!("}}");
+}
+
+#[macro_export]
+macro_rules! print_doc {
+    ( $doc:expr ) => {
+        print_doc_with_indent_level($doc, 0)
+    };
+}
+
+fn print_bson_with_indent_level(bson: &Bson, n: i32) {
+    match bson {
+        &Bson::Document(ref d) => print_doc_with_indent_level(&d, n + 4),
+        &Bson::String(ref s) => print!("\"{}\",", s),
+        &Bson::Boolean(b) => print!("{},", b),
+        &Bson::Array(ref v) => {
+            print!("[ ");
+
+            for b in v {
+                print_bson_with_indent_level(b, n);
+                print!(" ");
+            }
+
+            print!("]");
+        },
+        ref bson => print!("{:?},", bson)
+    };
+}
+
+
+#[test]
+fn recursive_macro() {
+    let doc = doc! {
+        "a" => ("foo"),
+        "b" => {
+            "bar" => {
+                "harbor" => ["seal", false],
+                "jelly" => (42.0)
+            },
+            "grape" => (27)
+        },
+        "c" => [-7]
+    };
+
+    print_doc!(&doc);
+}

tests/modules/mod.rs

@@ -1,3 +1,4 @@
 mod encoder;
+mod macros;
 mod oid;
 mod ordered;