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