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
|
///! Formats the message to be sent to the user
use time::{at_utc, Timespec, strftime};
use url::percent_encoding::{USERNAME_ENCODE_SET, PASSWORD_ENCODE_SET, percent_encode};
pub fn format_message(user_jid: &str, token: String, valid_from: i64, valid_until: i64, original_url: Option<String>) -> String {
return format!("Token: {}. Valid from {} until {}. \n{}",
token,
strftime("%F %X", &at_utc(Timespec::new(valid_from, 0))).unwrap(),
strftime("%F %X", &at_utc(Timespec::new(valid_until, 0))).unwrap(),
match original_url {
Some(url) => insert_token_password(url, user_jid, &token),
None => "".to_string()
});
}
fn insert_token_password(url: String, user_jid: &str, token: &str) -> String {
match url.find("://") {
Some(pos) => {
let (scheme, rest) = url.split_at(pos + 3);
return scheme.to_string() +
percent_encode(user_jid.as_bytes(), USERNAME_ENCODE_SET).as_str() + ":" +
percent_encode(token.as_bytes(), PASSWORD_ENCODE_SET).as_str() + "@" + rest;
},
None => return url
}
}
#[test]
fn test() {
assert_eq!(format_message("foo@bar.com", "7A-74-F4".to_string(), 0, 1481831953, Some("".to_string())),
"Token: 7A-74-F4. Valid from 1970-01-01 00:00:00 until 2016-12-15 19:59:13. \n");
assert_eq!(format_message("foo@bar.com", "7A-74-F4".to_string(), 0, 1481831953, Some("http".to_string())),
"Token: 7A-74-F4. Valid from 1970-01-01 00:00:00 until 2016-12-15 19:59:13. \nhttp");
assert_eq!(
insert_token_password(String::from("http://foo.bar/ads?123"), "user_jid", "token"),
"http://user_jid:token@foo.bar/ads?123");
assert_eq!(
insert_token_password(String::from("invalid"), "user_jid", "token"),
"invalid");
assert_eq!(
insert_token_password(String::from("http://host/path"), "user@host", "@@##"),
"http://user%40host:%40%40%23%23@host/path");
}
|
