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
extern crate pad;
extern crate regex;

pub mod types;

use std::fs;
use std::path::PathBuf;
use regex::Regex;
use types::*;

fn is_artifact(p: PathBuf, re: Option<Regex>) -> bool {
    let regex = if let Some(r) = re { r } 
        else { Regex::new(r"^\S+.a").unwrap() };
    let path_str = &p.into_os_string().into_string().expect("OS String invalid.");
    regex.is_match(path_str)
}

pub fn read_files(in_paths: &PathBuf, depth: u8) -> FileTree {
    let paths = fs::read_dir(in_paths.clone()).unwrap();
    let mut tree = FileTree::new();
    let mut total_size = FileSize::new(0);

    for p in paths {
        let path = p.unwrap().path(); // TODO no unwraps
        let metadata = fs::metadata(path.clone()).unwrap();

        // append file size/name for a file
        if metadata.is_file() {
            let file_size = FileSize::new(metadata.len());
            tree.push(path.clone(), file_size, None, depth + 1);
            total_size.add(file_size);
        }
        // otherwise, go deeper
        else if metadata.is_dir() {
            let mut subtree = read_files(&path, depth + 1);
            let dir_size = subtree.file_size;
            tree.push(path, dir_size, Some(&mut subtree), depth + 1);
            total_size.add(dir_size);
        }
    }
    tree
}