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();
let metadata = fs::metadata(path.clone()).unwrap();
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);
}
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
}