1use nutype::nutype;
2use uuid::Uuid;
3
4use crate::error::OAuthError;
5
6#[derive(Debug, Clone)]
7pub struct AuthorizeDeviceRequest {
8    pub user_code: UserCode,
9    pub user_id: Uuid,
10}
11
12impl AuthorizeDeviceRequest {
13    pub fn new(user_code: &str, user_id: Uuid) -> Result<Self, OAuthError> {
14        Ok(Self {
15            user_code: UserCode::try_new(user_code)
16                .map_err(|e| OAuthError::InvalidUserCode(e.to_string()))?,
17            user_id,
18        })
19    }
20}
21
22fn is_valid_user_code(s: &str) -> bool {
23    s.len() == 6
24        && s.chars()
25            .all(|c| c.is_ascii_uppercase() || ('2'..='9').contains(&c))
26}
27
28#[nutype(
29    validate(predicate = is_valid_user_code),
30    derive(Debug, Clone, PartialEq, Eq, AsRef, Deref)
31)]
32pub struct UserCode(String);
33
34#[cfg(test)]
35mod tests {
36    use super::*;
37
38    mod user_code {
39        use super::*;
40
41        #[test]
42        fn valid_code() {
43            let code = UserCode::try_new("ABC234").unwrap();
44            assert_eq!(code.as_ref(), "ABC234");
45        }
46
47        #[test]
48        fn rejects_too_short() {
49            assert!(UserCode::try_new("ABC23").is_err());
50        }
51
52        #[test]
53        fn rejects_too_long() {
54            assert!(UserCode::try_new("ABC2345").is_err());
55        }
56
57        #[test]
58        fn rejects_invalid_characters() {
59            assert!(UserCode::try_new("ABC230").is_err());
60            assert!(UserCode::try_new("ABC231").is_err());
61        }
62
63        #[test]
64        fn accepts_all_uppercase_letters() {
65            assert!(UserCode::try_new("ABCDIO").is_ok());
66        }
67
68        #[test]
69        fn rejects_special_characters() {
70            assert!(UserCode::try_new("ABC-23").is_err());
71            assert!(UserCode::try_new("ABC@23").is_err());
72        }
73    }
74
75    mod authorize_device_request {
76        use super::*;
77
78        #[test]
79        fn valid_request() {
80            let user_id = Uuid::new_v4();
81            let request = AuthorizeDeviceRequest::new("ABC234", user_id).unwrap();
82
83            assert_eq!(request.user_code.as_ref(), "ABC234");
84            assert_eq!(request.user_id, user_id);
85        }
86
87        #[test]
88        fn rejects_invalid_code() {
89            let user_id = Uuid::new_v4();
90            let result = AuthorizeDeviceRequest::new("invalid", user_id);
91
92            assert!(matches!(result, Err(OAuthError::InvalidUserCode(_))));
93        }
94    }
95}
96