dtz_identifier/
identity_id.rs

1use serde::{de::Visitor, Deserialize, Deserializer, Serialize};
2use std::fmt::Display;
3
4static PREFIX: &str = "identity-";
5
6#[derive(Debug, Clone, PartialEq)]
7pub struct IdentityId {
8    pub id: String,
9}
10
11impl Display for IdentityId {
12    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
13        f.write_str(&format!("{PREFIX}{}", self.id))
14    }
15}
16
17impl Default for IdentityId {
18    fn default() -> Self {
19        Self {
20            id: crate::generate_internal_id(8),
21        }
22    }
23}
24
25impl<'de> Deserialize<'de> for IdentityId {
26    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
27    where
28        D: Deserializer<'de>,
29    {
30        struct IdentityIdVisitor;
31
32        impl Visitor<'_> for IdentityIdVisitor {
33            type Value = IdentityId;
34
35            fn expecting(&self, formatter: &mut std::fmt::Formatter) -> std::fmt::Result {
36                formatter.write_str("a string starting with '")?;
37                formatter.write_str(PREFIX)?;
38                formatter.write_str("' followed by a 8-character alphanumeric string")
39            }
40
41            fn visit_str<E>(self, value: &str) -> Result<Self::Value, E>
42            where
43                E: serde::de::Error,
44            {
45                if let Some(id_str) = value.strip_prefix(PREFIX) {
46                    Ok(IdentityId {
47                        id: id_str.to_string(),
48                    })
49                } else {
50                    Err(E::custom("invalid format"))
51                }
52            }
53        }
54
55        deserializer.deserialize_str(IdentityIdVisitor)
56    }
57}
58
59impl Serialize for IdentityId {
60    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
61    where
62        S: serde::Serializer,
63    {
64        serializer.serialize_str(&self.to_string())
65    }
66}
67
68impl TryFrom<String> for IdentityId {
69    type Error = String;
70
71    fn try_from(value: String) -> Result<Self, Self::Error> {
72        if let Some(id_str) = value.strip_prefix(PREFIX) {
73            Ok(IdentityId {
74                id: id_str.to_string(),
75            })
76        } else {
77            Err("invalid format".to_string())
78        }
79    }
80}
81
82impl TryFrom<&str> for IdentityId {
83    type Error = String;
84
85    fn try_from(value: &str) -> Result<Self, Self::Error> {
86        if let Some(id_str) = value.strip_prefix(PREFIX) {
87            Ok(IdentityId {
88                id: id_str.to_string(),
89            })
90        } else {
91            Err("invalid format".to_string())
92        }
93    }
94}
95
96#[test]
97fn id_tostring() {
98    let id = IdentityId::default();
99    let full_id = format!("{id}");
100    println!("full-id: {full_id}");
101    assert!(full_id.starts_with(PREFIX));
102}
103
104#[test]
105fn key_invalid_1() {
106    let k = "abc-dsfdg";
107    let identity: Result<IdentityId, String> = IdentityId::try_from(k);
108    assert!(identity.is_err())
109}
110
111#[test]
112fn key_valid_1() {
113    let k = "identity-dsfdg";
114    let identity: Result<IdentityId, String> = IdentityId::try_from(k);
115    assert!(identity.is_ok())
116}
117
118#[test]
119fn key_valid_2() {
120    let k = "identity-0190c589-eb70-7980-97cf-af67b3a84116";
121    let identity: Result<IdentityId, String> = IdentityId::try_from(k);
122    assert!(identity.is_ok())
123}
124
125#[test]
126fn key_invalid_2() {
127    let k = "abc-0190c589-eb70-7980-97cf-af67b3a84116";
128    let identity: Result<IdentityId, String> = IdentityId::try_from(k);
129    assert!(identity.is_err())
130}