feat(protolok): add and implement snowflakes

Author: TudbuT

Cargo.toml

@@ -1,2 +1,3 @@
 [workspace]
 members = ["loki-client", "loki-server", "loki-shared", "lokui", "protolok"]
+resolver = "2"

protolok/src/lib.rs

@@ -1,2 +1,3 @@
+pub mod ids;
 pub mod message;
 pub mod user;

protolok/src/message.rs

@@ -1,6 +1,9 @@
-use rsa::{PaddingScheme, Pkcs1v15Encrypt, PublicKey, RsaPrivateKey, RsaPublicKey};
+use rsa::{Pkcs1v15Encrypt, PublicKey, RsaPrivateKey, RsaPublicKey};
 
-use crate::user::UserHandle;
+use crate::{
+	ids::{make_id, LokiMeta, Object},
+	user::UserHandle,
+};
 
 #[derive(Clone)]
 pub struct UnencryptedContent {
@@ -19,11 +22,11 @@ pub enum MessageContent {
 	Unencrypted(UnencryptedContent),
 }
 impl MessageContent {
-	fn is_encrypted(&self) -> bool {
+	pub fn is_encrypted(&self) -> bool {
 		matches!(self, &MessageContent::Encrypted { .. })
 	}
 
-	fn decrypt(&self, key: &RsaPrivateKey) -> Result<MessageContent, rsa::errors::Error> {
+	pub fn decrypt(&self, key: &RsaPrivateKey) -> Result<MessageContent, rsa::errors::Error> {
 		match self {
 			MessageContent::Encrypted(EncryptedContent { hash, enc_text }) => {
 				key.decrypt(Pkcs1v15Encrypt, &enc_text).map(|x| {
@@ -39,9 +42,9 @@ impl MessageContent {
 }
 
 pub struct Message {
-	id: Option<String>,
-	from: Option<UserHandle>,
-	content: MessageContent,
+	pub id: u64,
+	pub from: Option<UserHandle>,
+	pub content: MessageContent,
 }
 
 pub fn encrypt(
@@ -57,3 +60,9 @@ pub fn encrypt(
 		)?,
 	}))
 }
+
+impl Object for Message {
+	fn initialize(&mut self, meta: &LokiMeta) {
+		self.id = make_id(meta);
+	}
+}

protolok/src/user.rs

@@ -1,23 +1,37 @@
-use std::time::SystemTime;
-
-use sha2::{Digest, Sha256};
+use crate::ids::{make_id, LokiMeta, Object};
 
 pub struct UserHandle {
-	name: String,
-	home: String,
+	pub name: String,
+	pub home: String,
+	pub id: u64,
+}
+
+impl Object for UserHandle {
+	fn initialize(&mut self, meta: &LokiMeta) {
+		self.id = make_id(meta);
+	}
 }
 
-impl UserHandle {
-	pub fn get_id(&self) -> String {
-		let mut buf = [0u8; 64];
-		format!(
-			"{:X}",
-			SystemTime::now()
-				.duration_since(SystemTime::UNIX_EPOCH)
-				.unwrap()
-				.as_millis()
-		) + &base16ct::upper::encode_str(&Sha256::digest(self.name.as_bytes()), &mut buf).unwrap()
-			[..6] + &base16ct::upper::encode_str(&Sha256::digest(self.home.as_bytes()), &mut buf)
-			.unwrap()[..6]
+#[cfg(test)]
+mod tests {
+	use std::sync::OnceLock;
+
+	use crate::ids::{id_to_parts, to_unix_time, LokiMeta, Object};
+
+	use super::UserHandle;
+
+	#[test]
+	fn init_test() {
+		static META: OnceLock<LokiMeta> = OnceLock::new();
+		META.get_or_init(|| LokiMeta::new("test.lokichat.xyz"));
+		let mut handle = UserHandle {
+			name: "TudbuT".to_owned(),
+			home: "test.lokichat.xyz".to_owned(),
+			id: 0,
+		};
+		handle.initialize(META.get().unwrap());
+		dbg!(handle.id);
+		dbg!(id_to_parts(handle.id));
+		dbg!(to_unix_time(id_to_parts(handle.id).0));
 	}
 }