enigma_node_types/
user_id.rs

1use blake3::Hasher;
2use serde::{Deserialize, Deserializer, Serialize, Serializer};
3
4use crate::error::{EnigmaNodeTypesError, Result};
5
6#[derive(Clone, Copy, Eq, PartialEq, Hash, Debug)]
7pub struct UserId(pub [u8; 32]);
8
9impl UserId {
10    pub fn from_username(username: &str) -> Result<Self> {
11        let normalized = normalize_username(username)?;
12        let mut hasher = Hasher::new();
13        hasher.update(b"enigma:user_id:v1");
14        hasher.update(normalized.as_bytes());
15        let hash = hasher.finalize();
16        let mut bytes = [0u8; 32];
17        bytes.copy_from_slice(hash.as_bytes());
18        Ok(UserId(bytes))
19    }
20
21    pub fn as_bytes(&self) -> &[u8; 32] {
22        &self.0
23    }
24
25    pub fn to_hex(&self) -> String {
26        hex::encode(self.0)
27    }
28
29    pub fn from_hex(s: &str) -> Result<Self> {
30        let decoded = hex::decode(s)?;
31        if decoded.len() != 32 {
32            return Err(EnigmaNodeTypesError::InvalidHex);
33        }
34        let mut bytes = [0u8; 32];
35        for (i, b) in decoded.into_iter().enumerate() {
36            bytes[i] = b;
37        }
38        Ok(UserId(bytes))
39    }
40}
41
42pub fn normalize_username(username: &str) -> Result<String> {
43    if username.chars().any(|c| c.is_control()) {
44        return Err(EnigmaNodeTypesError::InvalidUsername);
45    }
46    let trimmed = username.trim();
47    if trimmed.is_empty() {
48        return Err(EnigmaNodeTypesError::InvalidUsername);
49    }
50    if trimmed.len() > 64 {
51        return Err(EnigmaNodeTypesError::InvalidUsername);
52    }
53    Ok(trimmed.to_string())
54}
55
56impl Serialize for UserId {
57    fn serialize<S>(&self, serializer: S) -> std::result::Result<S::Ok, S::Error>
58    where
59        S: Serializer,
60    {
61        serializer.serialize_str(&self.to_hex())
62    }
63}
64
65struct UserIdVisitor;
66
67impl<'de> serde::de::Visitor<'de> for UserIdVisitor {
68    type Value = UserId;
69
70    fn expecting(&self, formatter: &mut std::fmt::Formatter) -> std::fmt::Result {
71        formatter.write_str("32-byte hex string")
72    }
73
74    fn visit_str<E>(self, v: &str) -> std::result::Result<Self::Value, E>
75    where
76        E: serde::de::Error,
77    {
78        UserId::from_hex(v).map_err(|_| E::custom("invalid user id hex"))
79    }
80}
81
82impl<'de> Deserialize<'de> for UserId {
83    fn deserialize<D>(deserializer: D) -> std::result::Result<Self, D::Error>
84    where
85        D: Deserializer<'de>,
86    {
87        deserializer.deserialize_str(UserIdVisitor)
88    }
89}