use std::fmt::Display; use schemars::JsonSchema; use schemars::r#gen::SchemaGenerator; use schemars::schema::Schema; use serde::Deserialize; use serde::Serialize; use ts_rs::TS; use uuid::Uuid; use crate::ThreadId; #[derive(Debug, Clone, Copy, PartialEq, Eq, TS, Hash)] #[ts(type = "string")] pub struct SessionId { pub(crate) uuid: Uuid, } impl SessionId { pub fn new() -> Self { Self { uuid: Uuid::now_v7(), } } pub fn from_string(s: &str) -> Result { Ok(Self { uuid: Uuid::parse_str(s)?, }) } } impl TryFrom<&str> for SessionId { type Error = uuid::Error; fn try_from(value: &str) -> Result { Self::from_string(value) } } impl TryFrom for SessionId { type Error = uuid::Error; fn try_from(value: String) -> Result { Self::from_string(value.as_str()) } } impl From for String { fn from(value: SessionId) -> Self { value.to_string() } } impl From for SessionId { fn from(value: ThreadId) -> Self { Self { uuid: value.uuid } } } impl From for ThreadId { fn from(value: SessionId) -> Self { ThreadId { uuid: value.uuid } } } impl Default for SessionId { fn default() -> Self { Self::new() } } impl Display for SessionId { fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result { Display::fmt(&self.uuid, f) } } impl Serialize for SessionId { fn serialize(&self, serializer: S) -> Result where S: serde::Serializer, { serializer.collect_str(&self.uuid) } } impl<'de> Deserialize<'de> for SessionId { fn deserialize(deserializer: D) -> Result where D: serde::Deserializer<'de>, { let value = String::deserialize(deserializer)?; let uuid = Uuid::parse_str(&value).map_err(serde::de::Error::custom)?; Ok(Self { uuid }) } } impl JsonSchema for SessionId { fn schema_name() -> String { "SessionId".to_string() } fn json_schema(generator: &mut SchemaGenerator) -> Schema { ::json_schema(generator) } } #[cfg(test)] mod tests { use super::*; #[test] fn test_session_id_default_is_not_zeroes() { let id = SessionId::default(); assert_ne!(id.uuid, Uuid::nil()); } #[test] fn converts_to_and_from_thread_id() { let thread_id = ThreadId::new(); let session_id = SessionId::from(thread_id); assert_eq!(ThreadId::from(session_id), thread_id); } }