protobuf 0.0.10

Rust implementation of Google protocol buffers 
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

#![crate_type = "bin"]
#![feature(globs)]

extern crate protobuf;
extern crate getopts;

use std::io::fs::*;
use std::io::Reader;
use std::io::Writer;
use std::path::Path;
use std::os;

use protobuf::parse_from_reader;
use protobuf::descriptor::*;
use protobuf::codegen::*;


fn write_file(bin: &str, gen_options: &GenOptions) {
    let mut is = File::open(&Path::new(bin)).unwrap();
    let fds = parse_from_reader::<FileDescriptorSet>(&mut is as &mut Reader).unwrap();

    let file_names: Vec<String> = fds.get_file().iter()
        .map(|f| f.get_name().to_string())
        .collect();
    let results = gen(fds.get_file(), file_names.as_slice(), gen_options);

    for r in results.iter() {
        let mut file_writer = File::create(&Path::new(r.name.as_slice())).unwrap();
        file_writer.write(r.content.as_slice()).unwrap();
    }
}

fn main() {
    let args = os::args();
    let opts = vec!();
    let matches = getopts::getopts(args.tail(), opts.as_slice()).unwrap();
    let pb_bin = match matches.free.as_slice() {
        [ref pb_bin] => pb_bin.to_string(),
        _ => panic!("must have exactly one argument")
    };
    let gen_options = GenOptions {
        dummy: false,
    };
    write_file(pb_bin.as_slice(), &gen_options);
}