Implement password authentication

geneit_backend/src/services/login_token_service.rs

@@ -0,0 +1,58 @@
+//! # User tokens management
+
+use crate::connections::redis_connection;
+use crate::models::{User, UserID};
+use crate::utils::string_utils;
+use crate::utils::time_utils::time;
+use std::time::Duration;
+
+#[derive(serde::Serialize, serde::Deserialize, Debug, Clone)]
+struct LoginToken {
+    expire: u64,
+    hb: u64,
+    user_id: UserID,
+}
+
+impl LoginToken {
+    pub fn new(user: &User) -> (String, Self) {
+        let key = format!("tok-{}-{}", user.id().0, string_utils::rand_str(40));
+
+        (
+            key,
+            Self {
+                expire: time() + 3600 * 24,
+                hb: time(),
+                user_id: user.id(),
+            },
+        )
+    }
+
+    pub fn is_valid(&self) -> bool {
+        self.expire > time() && self.hb + 3600 > time()
+    }
+
+    pub fn refresh_hb(&self) -> Option<Self> {
+        if self.hb + 60 * 5 < time() {
+            let mut new = self.clone();
+            new.hb = time();
+            Some(new)
+        } else {
+            None
+        }
+    }
+
+    pub fn lifetime(&self) -> Duration {
+        Duration::from_secs(if self.expire <= time() {
+            0
+        } else {
+            self.expire - time()
+        })
+    }
+}
+
+/// Generate a new login token
+pub async fn gen_new_token(user: &User) -> anyhow::Result<String> {
+    let (key, token) = LoginToken::new(user);
+    redis_connection::set_value(&key, &token, token.lifetime()).await?;
+    Ok(key)
+}