rrecutils/src/tools/select.rs

extern crate clap;
extern crate failure;
extern crate rrecutils;

mod common;

use failure::Error;

fn rr_select_args() -> clap::ArgMatches<'static> {
    clap::App::new("rr-sel")
        .version(common::VERSION)
        .author(common::AUTHOR)
        .about("Print records from a recfile")
        .arg(
            clap::Arg::with_name("input")
                .short("i")
                .long("input")
                .value_name("FILE")
                .help("The input recfile (or - for stdin)"),
        )
        .arg(
            clap::Arg::with_name("output")
                .short("o")
                .long("output")
                .value_name("FILE")
                .help("The desired output location (or - for stdout)"),
        )
        .arg(
            clap::Arg::with_name("type")
                .long("type")
                .short("t")
                .required(false)
                .takes_value(true),
        )
        .arg(
            clap::Arg::with_name("include-descriptors")
                .long("include-descriptors")
                .short("d")
                .required(false)
                .takes_value(false),
        )
        .arg(
            clap::Arg::with_name("collapse")
                .long("collapse")
                .short("C")
                .required(false)
                .takes_value(false),
        )
        .arg(
            clap::Arg::with_name("sort")
                .long("sort")
                .short("S")
                .required(false)
                .takes_value(true),
        )
        .arg(
            clap::Arg::with_name("group-by")
                .long("group-by")
                .short("G")
                .required(false)
                .takes_value(true),
        )
        .get_matches()
}

fn run() -> Result<(), Error> {
    let matches = rr_select_args();

    let input = common::input_from_spec(matches.value_of("input"))?;
    let mut output = common::output_from_spec(matches.value_of("output"))?;

    let mut records = rrecutils::Recfile::parse(input)?;

    if let Some(typ) = matches.value_of("type") {
        records.filter_by_type(typ);
    }

    records.write(&mut output)?;

    Ok(())
}

fn main() {
    match run() {
        Ok(()) => (),
        Err(e) => println!("{}", e),
    }
}