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
extern crate rand;
#[cfg(test)]
mod tests;
use rand::{sample, Rng};
use std::collections::HashMap;
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct GenStringError {
empty: Vec<String>,
failed: Vec<String>,
attempt: String
}
impl GenStringError {
pub fn new() -> Self {
GenStringError {
empty: Vec::new(),
failed: Vec::new(),
attempt: String::new()
}
}
}
pub fn gen_string<'a, P, R>(cat: &HashMap<String, &'a [u8]>,
pat: P,
rng: &mut R) -> Result<String, GenStringError>
where P: IntoIterator<Item=String>,
R: Rng
{
let pat = pat.into_iter();
let mut failed = false;
let mut error = GenStringError::new();
let output: String = pat.map(|key| {
if let Some(catagory) = cat.get(&key) {
if let Some(u) = sample(rng, catagory.iter(), 1).iter().nth(0) {
**u as char
} else {
failed = true;
error.empty.push(key);
0u8 as char
}
} else {
failed = true;
error.failed.push(key);
0u8 as char
}
}).collect();
if failed {
error.attempt = output;
Err( error )
} else {
Ok( output )
}
}