1 files changed, 79 insertions, 0 deletions
diff --git a/src/types/user.rs b/src/types/user.rs
new file mode 100644
index 0000000..1213a75
--- /dev/null
+++ b/src/types/user.rs
@@ -0,0 +1,79 @@
+use serde::{Serialize, Deserialize};
+
+use crate::database;
+use crate::types::response::{Result, ResponseCode};
+
+
+#[derive(Serialize, Deserialize, Debug)]
+pub struct User {
+    pub user_id: u64,
+    pub firstname: String,
+    pub lastname: String,
+    pub email: String,
+    pub password: String,
+    pub gender: String,
+    pub date: u64,
+    pub day: u8,
+    pub month: u8,
+    pub year: u32,
+}
+
+impl User {
+
+    pub fn from_user_id(user_id: u64, hide_password: bool) -> Result<Self> {
+        let Ok(Some(user)) = database::users::get_user_by_id(user_id, hide_password) else {
+            return Err(ResponseCode::BadRequest.msg("User does not exist"))
+        };
+
+        Ok(user)
+    }
+
+    pub fn from_user_ids(user_ids: Vec<u64>) -> Vec<Self> {
+        user_ids.iter().map(|user_id| {
+            let Ok(Some(user)) = database::users::get_user_by_id(*user_id, true) else {
+                return None;
+            };
+            Some(user)
+        }).flatten().collect()
+    }
+
+    pub fn from_user_page(page: u64) -> Result<Vec<Self>> {
+        let Ok(users) = database::users::get_user_page(page, true) else {
+            return Err(ResponseCode::BadRequest.msg("Failed to fetch users"))
+        };
+        Ok(users)
+    }
+
+    pub fn from_email(email: &str) -> Result<Self> {
+        let Ok(Some(user)) = database::users::get_user_by_email(email, false) else {
+            return Err(ResponseCode::BadRequest.msg("User does not exist"))
+        };
+
+        Ok(user)
+    }
+
+    pub fn from_password(password: &str) -> Result<Self> {
+        let Ok(Some(user)) = database::users::get_user_by_password(password, true) else {
+            return Err(ResponseCode::BadRequest.msg("User does not exist"))
+        };
+
+        Ok(user)
+    }
+
+    pub fn new(firstname: String, lastname: String, email: String, password: String, gender: String, day: u8, month: u8, year: u32) -> Result<Self> {
+        if let Ok(_) = User::from_email(&email) {
+            return Err(ResponseCode::BadRequest.msg(&format!("Email is already in use by {}", &email)))
+        }
+
+        if let Ok(user) = User::from_password(&password) {
+            return Err(ResponseCode::BadRequest.msg(&format!("Password is already in use by {}", user.email)))
+        }
+        
+        let Ok(user) = database::users::add_user(&firstname, &lastname, &email, &password, &gender, day, month, year) else {
+            return Err(ResponseCode::InternalServerError.msg("Failed to create new uesr"))
+        };
+
+        Ok(user)
+    }
+    
+}
+\ No newline at end of file