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 indexmap::IndexMap;
/// Parse the htpasswd string and create a IndexMap of user/password pairs.
///
/// # Examples
///
/// ```
/// use indexmap::IndexMap;
/// use htpasswd::parse;
/// let input = "
/// johndoe:$apr1$hdqQY4oe$6PtEz0XH6ORg.GPKCTpG31
/// janedoe
/// # comment
/// janedoe:$apr1$D7qCR.yD$vfKO/2urv89Okpxl8VGpb/
/// ";
/// let mut output = IndexMap::new();
/// output.insert("johndoe", "$apr1$hdqQY4oe$6PtEz0XH6ORg.GPKCTpG31");
/// output.insert("janedoe", "$apr1$D7qCR.yD$vfKO/2urv89Okpxl8VGpb/");
/// assert_eq!(output, parse(input));
/// ```
pub fn parse(text: &str) -> IndexMap<&str, &str> {
    let mut map = IndexMap::new();
    for line in text.trim().lines() {
        let parts: Vec<&str> = line.trim().split(":").collect();
        if parts.len() != 2 {
            continue;
        }
        if parts[0].starts_with("#") {
            continue;
        }
        map.insert(parts[0].trim(), parts[1].trim());
    }
    return map;
}

/// Transform the IndexMap of user/password pairs as create a htpasswd string.
///
/// # Examples
///
/// ```
/// use indexmap::IndexMap;
/// use htpasswd::stringify;
/// let output = "johndoe:$apr1$hdqQY4oe$6PtEz0XH6ORg.GPKCTpG31
/// janedoe:$apr1$D7qCR.yD$vfKO/2urv89Okpxl8VGpb/
/// ";
/// let mut input = IndexMap::new();
/// input.insert("johndoe", "$apr1$hdqQY4oe$6PtEz0XH6ORg.GPKCTpG31");
/// input.insert("janedoe", "$apr1$D7qCR.yD$vfKO/2urv89Okpxl8VGpb/");
/// assert_eq!(output, stringify(input));
/// ```
pub fn stringify<'a>(map: IndexMap<&str, &str>) -> String {
    let mut string: String = "".to_string();

    for (&username, &pwd_hash) in map.iter() {
        let line = format!("{}:{}\n", username, pwd_hash);
        string.push_str(line.as_str());
    }
    string
}

#[cfg(test)]
mod tests {
    use super::parse;
    use super::stringify;
    use indexmap::IndexMap;

    #[test]
    fn it_parse() {
        let input = "johndoe:$apr1$hdqQY4oe$6PtEz0XH6ORg.GPKCTpG31
janedoe
# comment
janedoe:$apr1$D7qCR.yD$vfKO/2urv89Okpxl8VGpb/
        ";
        let mut output = IndexMap::new();
        output.insert("johndoe", "$apr1$hdqQY4oe$6PtEz0XH6ORg.GPKCTpG31");
        output.insert("janedoe", "$apr1$D7qCR.yD$vfKO/2urv89Okpxl8VGpb/");
        assert_eq!(output, parse(input));
    }

    #[test]
    fn it_stringify() {
        let output = "johndoe:$apr1$hdqQY4oe$6PtEz0XH6ORg.GPKCTpG31
janedoe:$apr1$D7qCR.yD$vfKO/2urv89Okpxl8VGpb/
";
        let mut input = IndexMap::new();
        input.insert("johndoe", "$apr1$hdqQY4oe$6PtEz0XH6ORg.GPKCTpG31");
        input.insert("janedoe", "$apr1$D7qCR.yD$vfKO/2urv89Okpxl8VGpb/");
        assert_eq!(output, stringify(input));
    }
}