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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
use std::fs;
use std::io::{self, BufRead};
use std::path::{Path, PathBuf};
use walkdir::WalkDir;
pub fn delete(path_buf: PathBuf) -> io::Result<()> {
if path_buf.is_dir() {
fs::remove_dir_all(path_buf)
} else if path_buf.is_file() {
fs::remove_file(path_buf)
} else {
Ok(())
}
}
pub fn copy_dir_to(src: &Path, dst: &Path) -> io::Result<u64> {
let mut counter = 0u64;
if !dst.is_dir() {
fs::create_dir(dst)?
}
for entry_result in src.read_dir()? {
let entry = entry_result?;
let file_type = entry.file_type()?;
let count = copy_to(&entry.path(), file_type, &dst.join(entry.file_name()))?;
counter += count;
}
Ok(counter)
}
pub fn list_files(path: &Path) -> Vec<PathBuf> {
WalkDir::new(path)
.sort_by(|a, b| a.path().cmp(b.path()))
.min_depth(1)
.into_iter()
.filter_map(|x| x.ok())
.filter(|x| x.file_type().is_file())
.filter_map(|x| x.path().strip_prefix(path).map(|x| x.to_path_buf()).ok())
.collect()
}
fn copy_to(src: &Path, src_type: fs::FileType, dst: &Path) -> io::Result<u64> {
if src_type.is_file() {
fs::copy(src, dst)
} else if src_type.is_dir() {
copy_dir_to(src, dst)
} else {
Err(io::Error::new(
io::ErrorKind::Other,
format!("Could not copy: {}", src.display()),
))
}
}
pub fn get_first_line(file_path: Option<String>) -> Option<String> {
file_path.and_then(|path| match fs::File::open(path) {
Ok(file) => {
let buf_reader = io::BufReader::new(file);
let mut lines_iter = buf_reader.lines().map(|l| l.unwrap());
lines_iter.next()
}
Err(_) => None,
})
}