extern crate clap;
use clap::{App, Arg};
mod ds1;
mod ds2;
mod ds3;
mod ds4;
mod ds5;
mod report;

#[cfg(target_os = "linux")]
fn main() {
    let matches = App::new("DiskSpace")
        .version("0.1.0")
        .author("Eric Jackson <swiftgist@gmail.com>")
        .about("Displays disk space usage")
        .arg(
            Arg::with_name("all")
                .short("a")
                .long("all")
                .help("display all entries"),
        )
        .arg(
            Arg::with_name("parent")
                .short("p")
                .long("parent")
                .help("include all parent directories"),
        )
        .arg(
            Arg::with_name("reverse")
                .short("r")
                .long("reverse")
                .help("display entries descending"),
        )
        .arg(
            Arg::with_name("first")
                .short("1")
                .help("original implementation"),
        )
        .arg(
            Arg::with_name("second")
                .short("2")
                .help("second implementation"),
        )
        .arg(
            Arg::with_name("third")
                .short("3")
                .help("third implementation"),
        )
        .arg(
            Arg::with_name("fourth")
                .short("4")
                .help("fourth implementation"),
        )
        .arg(
            Arg::with_name("fifth")
                .short("5")
                .help("fifth implementation"),
        )
        .arg(Arg::with_name("directory").index(1).help("start location"))
        .get_matches();

    let anchor = match matches.value_of("directory") {
        Some(start) => start.to_string(),
        None => "./".to_string(),
    };

    if matches.is_present("fifth") {
        let disk_space = ds5::traverse(&anchor, &matches);
        report::report(disk_space, &matches);
    } else if matches.is_present("fourth") {
        let disk_space = ds4::traverse(&anchor, &matches);
        report::report(disk_space, &matches);
    } else if matches.is_present("third") {
        let disk_space = ds3::traverse(&anchor, &matches);
        report::report(disk_space, &matches);
    } else if matches.is_present("second") {
        let disk_space = ds2::traverse(&anchor, &matches);
        report::report(disk_space, &matches);
    } else if matches.is_present("first") {
        let disk_space = ds1::traverse(&anchor, &matches);
        report::report(disk_space, &matches);
    } else {
        let disk_space = ds4::traverse(&anchor, &matches);
        report::report(disk_space, &matches);
    }
}

#[cfg(target_os = "windows")]
fn main() {
    let matches = App::new("DiskSpace")
        .version("0.1.0")
        .author("Eric Jackson <swiftgist@gmail.com>")
        .about("Displays disk space usage")
        .arg(
            Arg::with_name("all")
                .short("a")
                .long("all")
                .help("display all entries"),
        )
        .arg(
            Arg::with_name("parent")
                .short("p")
                .long("parent")
                .help("include all parent directories"),
        )
        .arg(
            Arg::with_name("reverse")
                .short("r")
                .long("reverse")
                .help("display entries descending"),
        )
        .arg(Arg::with_name("directory").index(1).help("start location"))
        .get_matches();

    let anchor = match matches.value_of("directory") {
        Some(start) => start.to_string(),
        None => "./".to_string(),
    };

    let disk_space = ds4::traverse(&anchor, &matches);
    report::report(disk_space, &matches);
}