pg-connection-string 0.0.2

PostgreSQL-compatibly idiosyncratic connection string parsing
Documentation 
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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118

use super::Res;
use nom::{
    branch::alt,
    bytes::complete::{is_not, tag},
    character::complete::{alphanumeric1, one_of},
    combinator::{map, opt, recognize},
    multi::many1,
    sequence::{preceded, tuple},
};

#[derive(Debug, Default, PartialEq, Eq)]
pub(crate) struct UserSpec {
    pub user: String,
    pub password: Option<String>,
}

/// Parse the `userinfo` component of a URI.
pub(crate) fn userinfo(i: &str) -> Res<&str, UserSpec> {
    map(tuple((user, opt(password))), |(user, password)| UserSpec {
        user: user.into(),
        password: password.map(std::convert::Into::into),
    })(i)
}

/// Parse the `user` component of the `userinfo component of a uri.
fn user(i: &str) -> Res<&str, &str> {
    recognize(many1(alt((alphanumeric1, recognize(one_of("-_"))))))(i)
}

/// Parse the `password` component of the `userinfo component of a uri.
fn password(i: &str) -> Res<&str, &str> {
    preceded(tag(":"), is_not("@"))(i)
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_userinfo() {
        // Good cases.
        assert_eq!(
            userinfo("jack:secret"),
            Ok((
                "",
                UserSpec {
                    user: "jack".to_string(),
                    password: Some("secret".to_string())
                }
            ))
        );
        assert_eq!(
            userinfo("jack:secret@"),
            Ok((
                "@",
                UserSpec {
                    user: "jack".to_string(),
                    password: Some("secret".to_string())
                }
            ))
        );
        assert_eq!(
            userinfo("jack"),
            Ok((
                "",
                UserSpec {
                    user: "jack".to_string(),
                    password: None
                }
            ))
        );
        assert_eq!(
            userinfo("jack@"),
            Ok((
                "@",
                UserSpec {
                    user: "jack".to_string(),
                    password: None
                }
            ))
        );
        assert_eq!(
            userinfo("jack:"),
            Ok((
                ":",
                UserSpec {
                    user: "jack".to_string(),
                    password: None
                }
            ))
        );

        for input in vec![",jack"] {
            assert!(userinfo(input).is_err(), "input: {:?}", input);
        }
    }

    #[test]
    fn test_password() {
        // Good cases.
        assert_eq!(password(":secret@"), Ok(("@", "secret")));

        for input in vec!["secret", "secret@", ":@", ":"] {
            assert!(password(input).is_err(), "input: {}", input);
        }
    }

    #[test]
    fn test_user() {
        // Good cases.
        assert_eq!(user("jack:"), Ok((":", "jack")));
        assert_eq!(user("jack"), Ok(("", "jack")));

        // Bad cases.
        assert!(user(":jack").is_err());
        assert!(user("@jack").is_err());
    }
}