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

use crate::core::Parser;
use crate::utils::Utils;
use regex::Regex;

impl Parser {
    /// Extract the domain fields from the url.
    ///
    /// # Example
    /// ```rust
    /// use url_parse::core::Parser;
    /// let input = "https://user:pass@www.example.co.uk";
    /// let expected = (Some("user"), Some("pass"));
    /// let result = Parser::new(None).login(input);
    /// assert_eq!(result, expected);
    /// ```
    pub fn login<'a>(&self, input: &'a str) -> (Option<&'a str>, Option<&'a str>) {
        let input = Utils::substring_after_scheme(self, input);
        let input = match input.find("/") {
            Some(pos) => &input[..pos],
            None => input,
        };

        let re = Regex::new(r"(.*)@(.*).*").unwrap();
        let caps = re.captures(input);
        if caps.is_none() {
            return (None, None);
        }

        let caps = caps.unwrap();
        let user_with_pass = caps.get(1).unwrap().as_str();
        let (user, pass) = match user_with_pass.find(":") {
            Some(v) => (Some(&user_with_pass[..v]), Some(&user_with_pass[v + 1..])),
            None => (Some(user_with_pass), None),
        };
        (user, pass)
    }
}

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

    #[test]
    fn test_login_works_when_full_url_with_login() {
        let input =
            "https://user:pass@www.example.co.uk:443/blog/article/search?docid=720&hl=en#dayone";
        let expected = (Some("user"), Some("pass"));
        let result = Parser::new(None).login(input);
        assert_eq!(result, expected);
    }

    #[test]
    fn test_login_works_when_full_url_no_port() {
        let input =
            "https://user:pass@www.example.co.uk/blog/article/search?docid=720&hl=en#dayone";
        let expected = (Some("user"), Some("pass"));
        let result = Parser::new(None).login(input);
        assert_eq!(result, expected);
    }

    #[test]
    fn test_login_works_when_full_url_no_port_no_path() {
        let input = "https://user:pass@www.example.co.uk";
        let expected = (Some("user"), Some("pass"));
        let result = Parser::new(None).login(input);
        assert_eq!(result, expected);
    }

    #[test]
    fn test_login_works_when_user_only() {
        let input = "https://user@www.example.co.uk:443/blog/article/search?docid=720&hl=en#dayone";
        let expected = (Some("user"), None);
        let result = Parser::new(None).login(input);
        assert_eq!(result, expected);
    }

    #[test]
    fn test_login_works_when_user_only_no_port_no_path() {
        let input = "https://user@www.example.co.uk";
        let expected = (Some("user"), None);
        let result = Parser::new(None).login(input);
        assert_eq!(result, expected);
    }

    #[test]
    fn test_login_works_when_no_login() {
        let input = "https://www.example.co.uk:443/blog/article/search?docid=720&hl=en#dayone";
        let expected = (None, None);
        let result = Parser::new(None).login(input);
        assert_eq!(result, expected);
    }
}