aboutsummaryrefslogtreecommitdiffstats
path: root/src/bin/geniza-register.rs
blob: a328bbcc2c755fb583c6694acd8c10c7c2c6deff (plain)
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

#[macro_use]
extern crate error_chain;
extern crate clap;
extern crate env_logger;
extern crate geniza;

// TODO: more careful import
use geniza::*;
use std::path::Path;
use clap::{App, SubCommand};

fn run() -> Result<()> {

    env_logger::init().unwrap();
 
    let matches = App::new("geniza-register")
        .version(env!("CARGO_PKG_VERSION"))
        .subcommand(SubCommand::with_name("info")
            .about("Reads a SLEEP dir register and shows some basic metadata")
            .arg_from_usage("<DIR> 'directory containing files'")
            .arg_from_usage("<prefix> 'prefix for each data file'"))
        .subcommand(SubCommand::with_name("create")
            .about("Creates an SLEEP directory register (with header)")
            .arg_from_usage("<DIR> 'directory containing files'")
            .arg_from_usage("<prefix> 'prefix for each data file'"))
        .get_matches();


    match matches.subcommand() {
        ("info", Some(subm)) => {
            let dir = Path::new(subm.value_of("DIR").unwrap());
            let prefix = subm.value_of("prefix").unwrap();
            let mut sdr = SleepDirRegister::open(dir, prefix, false)?;
            //debug!(println!("{:?}", sdr));
            println!("Entry count: {}", sdr.len()?);
            println!("Total size (bytes): {}", sdr.len_bytes()?);
        },
        ("create", Some(subm)) => {
            let dir = Path::new(subm.value_of("DIR").unwrap());
            let prefix = subm.value_of("prefix").unwrap();
            SleepDirRegister::create(dir, prefix)?;
            println!("Done!");
        },
        _ => {
            println!("Missing or unimplemented command!");
            println!("{}", matches.usage());
            ::std::process::exit(-1);
        },
    }
    Ok(())
}

quick_main!(run);