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
pub use self::{extractor::*, files::*, utils::*};
mod extractor;
mod files;
pub trait ExtractorSpec<T> {
fn extract(&self) -> Vec<T>;
}
pub trait FileExtSpec: ExtractorSpec<String> {
fn path(&self) -> std::path::Path;
}
pub(crate) mod utils {
use crate::DEFAULT_IGNORE_CHARS;
use std::str::FromStr;
pub fn base_extractor<S: ToString, T: FromStr + ToString>(
bp: char,
data: &S,
exclude: Option<&[char]>,
) -> Vec<T>
where
<T as FromStr>::Err: std::fmt::Debug,
{
let data = data.to_string();
let skip = exclude.unwrap_or(DEFAULT_IGNORE_CHARS);
let trimmed: &str = data.trim_matches(skip);
trimmed
.split(bp)
.map(|i| i.trim_matches(skip).parse::<T>().unwrap())
.collect()
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_file_extractor() {
let fp = "../README.md";
let mut a = FileExtractor::new(fp.to_string());
let mut b = a.clone();
assert_eq!(a.extract(), b.extract())
}
#[test]
fn test_extractor_comma() {
let a = Extractor::new(',', "[0, 0, 0, 0]".to_string(), None);
assert_eq!(a.extract::<u8>(), vec![0, 0, 0, 0])
}
#[test]
fn test_extractor_period() {
let a = Extractor::new('.', "0.0.0.0".to_string(), None);
assert_eq!(a.extract::<u8>(), vec![0, 0, 0, 0])
}
}