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
mod error;

use error::ParseError;

#[derive(Debug, PartialEq)]
pub struct Credential {
    pub user_id: String,
    pub password: String
}

pub fn decode(s: &str) -> Result<Credential, ParseError> {
    match s {
	s if s.starts_with("Basic ") => {
	    let decoded = base64::decode(&s[6..])?;
	    let decoded = std::str::from_utf8(&decoded)?;
	    let parts: Vec<&str> = decoded.splitn(2, ":").collect();
	    match &parts.as_slice() {
		[user, pass] => Ok(Credential { user_id: user.to_string(), password: pass.to_string() }),
		_ => Err(ParseError::Format)
	    }
	}
	_ => Err(ParseError::Scheme)
    }
}

impl std::str::FromStr for Credential {
    type Err = ParseError;
    fn from_str(s: &str) -> Result<Self, Self::Err> {
	decode(s)
    }
}

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

    fn encode(c: &Credential) -> String {
	let unencoded = format!("{}:{}", c.user_id, c.password);
	let encoded = base64::encode(unencoded);
	format!("Basic {}", encoded)
    }

    fn encode_str(s: &str) -> String {
	let encoded = base64::encode(s);
	format!("Basic {}", encoded)
    }

    #[test]
    fn identify_scheme() {
	let c = Credential {
	    user_id: "Aladdin".to_string(),
	    password: "open sesame".to_string()
	};
	let s = encode(&c);
	let c2 = decode(&s);
	assert_eq!(Ok(c), c2);
    }

    #[test]
    fn no_colon() {
	let s = encode_str("john");
	let c = decode(&s);
	assert_eq!(Err(ParseError::Format), c);
    }

    #[test]
    fn allow_empty_password() {
	let s = encode_str("john:");
	let c = decode(&s);
	assert_ne!(Err(ParseError::Format), c);
    }

    #[test]
    fn bad_base64() {
	let s = "Basic abcdefg";
	let c = decode(&s);
	assert_eq!(Err(ParseError::Decode), c);
    }

    #[test]
    fn bad_scheme() {
	let c = Credential {
	    user_id: "john".to_string(),
	    password: "hunter2".to_string(),
	};
	let unencoded = format!("{}:{}", c.user_id, c.password);
	let encoded = base64::encode(unencoded);
	let formatted = format!("Bearer {}", encoded);
	let c = decode(&formatted);
	assert_eq!(Err(ParseError::Scheme), c);
    }
}