bisonn 1.0.1

Bisonn is a simple and fast BSON parser written in Rust
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66

#![allow(unused_imports)]
use std::io::prelude::*;
use std::io::BufReader;
use std::fs::File;
use std::path::Path;
use std::env;
use std::collections::HashMap;

use serde_json::Value;
use serde_json::json;

use bson::Document;

fn main() {
    let args: Vec<String> = env::args().collect();
    if args.len() < 2 {
        println!("Usage: bisonn <file.bson>");
        std::process::exit(1);
    }

    let path = Path::new(&args[1]);
    let file = File::open(&path).unwrap();
    let mut reader = BufReader::new(file);

    let mut bytes = Vec::new();
    reader.read_to_end(&mut bytes).unwrap();

    let doc = Document::from_reader(&bytes[..]).unwrap();
    let doc = replace_bin(doc);
    let jdoc = serde_json::to_string_pretty(&doc).unwrap();
    println!("{}", jdoc);
}

fn replace_bin(data: Document) -> Document {
    let mut new_doc = Document::new();

    for (key, value) in data {
        let value = match value {
            bson::Bson::Binary(_) => bson::Bson::String("BINARY".to_owned()),
            bson::Bson::Document(doc) => bson::Bson::Document(replace_bin(doc)),
            bson::Bson::Array(arr) => bson::Bson::Array(replace_bin_array(arr)),
            _ => value,
        };

        new_doc.insert(key, value);
    }

    new_doc
}

fn replace_bin_array(data: bson::Array) -> bson::Array {
    let mut new_arr = bson::Array::new();

    for value in data {
        let value = match value {
            bson::Bson::Binary(_) => bson::Bson::String("BINARY".to_owned()),
            bson::Bson::Document(doc) => bson::Bson::Document(replace_bin(doc)),
            bson::Bson::Array(arr) => bson::Bson::Array(replace_bin_array(arr)),
            _ => value,
        };

        new_arr.push(value);
    }

    new_arr
}