du 0.1.1

Implementing du -sb in order to learn Rust
Documentation 
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

//! `du -sb` implemented in Rust.
//!
//! This was solely written for the purpose of learning Rust. Do not use it in production code.
extern crate getopts;
extern crate du;

use std::env;
use std::io::Write;

use getopts::Options;


fn print_usage(program: &str, opts: Options) {
    let brief = format!("Usage: {} <path> [<path> ..]", program);
    println!("{}", opts.usage(&brief));
}


fn main() {
    let args: Vec<String> = env::args().collect();

    let mut stderr = std::io::stderr();

    let mut opts = Options::new();
    opts.optflag("", "help", "Print this help menu and exit.");

    match opts.parse(&args[1..]) {
        Ok(matches) => {
            if matches.opt_present("help") || matches.free.is_empty() {
                print_usage(&args[0], opts);
                return;
            }

            for path in &matches.free {
                match du::get_size(path) {
                    Ok(size) => {
                        println!("{}\t{}", size, path);
                    }
                    Err(err) => {
                        writeln!(&mut stderr, "Error: {}", err).unwrap();
                    }
                }
            }
        }
        Err(err) => {
            writeln!(&mut stderr, "{}", err).unwrap();
            print_usage(&args[0], opts);
        }
    }
}