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
88
89
use std::path::Path;
use std::path::PathBuf;
use anyhow::bail;
use anyhow::ensure;
use walkdir::WalkDir;
pub fn parse_dnas(mut dnas: Vec<PathBuf>) -> anyhow::Result<Vec<PathBuf>> {
if dnas.is_empty() {
dnas.push(std::env::current_dir()?);
}
for dna in dnas.iter_mut() {
if dna.is_dir() {
let file_path = search_for_dna(&dna)?;
*dna = file_path;
}
ensure!(
dna.file_name()
.map(|f| f.to_string_lossy().ends_with(".dna"))
.unwrap_or(false),
"File {} is not a valid dna file name: (e.g. my-dna.dna)",
dna.display()
);
}
Ok(dnas)
}
pub fn parse_happ(happ: Option<PathBuf>) -> anyhow::Result<PathBuf> {
let mut happ = happ.unwrap_or(std::env::current_dir()?);
if happ.is_dir() {
let file_path = search_for_happ(&happ)?;
happ = file_path;
}
ensure!(
happ.file_name()
.map(|f| f.to_string_lossy().ends_with(".happ"))
.unwrap_or(false),
"File {} is not a valid happ file name: (e.g. my-happ.happ)",
happ.display()
);
Ok(happ)
}
fn search_for_dna(dna: &Path) -> anyhow::Result<PathBuf> {
let dir: Vec<_> = WalkDir::new(dna)
.max_depth(1)
.into_iter()
.filter_map(|e| e.ok())
.filter(|d| d.file_type().is_file())
.filter(|f| f.file_name().to_string_lossy().ends_with(".dna"))
.map(|f| f.into_path())
.collect();
if dir.len() != 1 {
bail!(
"Could not find a DNA file (e.g. my-dna.dna) in directory {}",
dna.display()
)
}
Ok(dir.into_iter().next().expect("Safe due to check above"))
}
fn search_for_happ(happ: &Path) -> anyhow::Result<PathBuf> {
let dir = WalkDir::new(happ)
.max_depth(1)
.into_iter()
.filter_map(|e| e.ok())
.filter(|d| d.file_type().is_file())
.find(|f| f.file_name().to_string_lossy().ends_with(".happ"))
.map(|f| f.into_path());
match dir {
Some(dir) => Ok(dir),
None => {
bail!(
"Could not find a happ file (e.g. my-happ.happ) in directory {}",
happ.display()
)
}
}
}