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
82
83
84
85
86
87
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: String) -> Vec<String> {
let mut files_vec: Vec<String> = vec![];
for entry in WalkDir::new(Path::new(&path))
.into_iter()
.filter_map(|e| e.ok())
{
match entry.file_name().to_str() {
Some(path_str) => files_vec.push(path_str.to_string()),
None => println!("Could not read optional type"),
}
}
return files_vec;
}
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 {
return Err(io::Error::new(
io::ErrorKind::Other,
format!("Could not copy: {}", src.display()),
));
}
}
pub fn get_first_line(file_path: Option<String>) -> Option<String> {
match file_path {
Some(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,
},
None => None,
}
}