init_data_rust/
parse.rs

1use std::collections::HashSet;
2use url::Url;
3
4use crate::{errors::ParseDataError, models::InitData};
5
6/// Converts passed init data presented as query string to InitData object.
7pub fn parse(init_data: &str) -> Result<InitData, ParseDataError> {
8    // Parse passed init data as query string
9    let url = Url::parse(&format!("http://dummy.com?{}", init_data))?;
10
11    // Properties that should always be interpreted as strings
12    let string_props: HashSet<&str> = ["start_param"].iter().cloned().collect();
13
14    // Build JSON map
15    let mut map = serde_json::Map::new();
16    for (key, value) in url.query_pairs() {
17        let val_str = value.into_owned();
18        let key_str = key.into_owned();
19        if string_props.contains(key_str.as_str()) {
20            map.insert(key_str, serde_json::Value::String(val_str));
21        } else {
22            // Try to parse the value as JSON
23            match serde_json::from_str(&val_str) {
24                Ok(json_value) => {
25                    map.insert(key_str, json_value);
26                }
27                Err(_) => {
28                    // Use as string
29                    map.insert(key_str, serde_json::Value::String(val_str));
30                }
31            }
32        }
33    }
34
35    // Create final JSON value
36    let json_value = serde_json::Value::Object(map);
37
38    // Deserialize JSON into InitData struct
39    Ok(serde_json::from_value(json_value)?)
40}
41
42#[cfg(test)]
43mod tests {
44
45    use crate::{
46        models::{InitData, User},
47        parse,
48    };
49
50    #[test]
51    fn test_parse_valid_data() {
52        let init_data = "query_id=AAHdF6IQAAAAAN0XohDhrOrc&user=%7B%22id%22%3A279058397%2C%22first_name%22%3A%22Vladislav%22%2C%22last_name%22%3A%22Kibenko%22%2C%22username%22%3A%22vdkfrost%22%2C%22language_code%22%3A%22ru%22%2C%22is_premium%22%3Atrue%7D&auth_date=1662771648&hash=c501b71e775f74ce10e377dea85a7ea24ecd640b223ea86dfe453e0eaed2e2b2&start_param=abc";
53        let result = parse(init_data);
54        assert!(result.is_ok());
55        let data = result.unwrap();
56        assert_eq!(
57            data,
58            InitData {
59                auth_date: 1662771648,
60                can_send_after: None,
61                chat: None,
62                chat_type: None,
63                chat_instance: None,
64                hash: "c501b71e775f74ce10e377dea85a7ea24ecd640b223ea86dfe453e0eaed2e2b2"
65                    .to_string(),
66                query_id: Some("AAHdF6IQAAAAAN0XohDhrOrc".to_string()),
67                receiver: None,
68                start_param: Some("abc".to_string()),
69                user: Some(User {
70                    added_to_attachment_menu: None,
71                    allows_write_to_pm: None,
72                    is_premium: Some(true),
73                    first_name: "Vladislav".to_string(),
74                    id: 279058397,
75                    is_bot: None,
76                    last_name: Some("Kibenko".to_string()),
77                    language_code: Some("ru".to_string()),
78                    photo_url: None,
79                    username: Some("vdkfrost".to_string())
80                })
81            }
82        );
83    }
84
85    #[test]
86    fn test_parse_invalid_data() {
87        let init_data = "invalid data";
88        let result = parse(init_data);
89        assert!(result.is_err());
90    }
91}
init_data_rust/parse.rs

init_data_rust/
parse.rs
