From 0c656f0fe6b61c35bd40d43829a6a63fe353c589 Mon Sep 17 00:00:00 2001
From: Santo Cariotti <santo@dcariotti.me>
Date: Mon, 22 Aug 2022 22:32:54 +0200
Subject: Create user

---
 server/src/models/user.rs | 29 ++++++++++++++++++++++++-----
 1 file changed, 24 insertions(+), 5 deletions(-)

(limited to 'server/src/models')

diff --git a/server/src/models/user.rs b/server/src/models/user.rs
index 08514a8..9545fac 100644
--- a/server/src/models/user.rs
+++ b/server/src/models/user.rs
@@ -8,22 +8,41 @@ pub struct User {
     id: i32,
     email: String,
     password: String,
+    is_staff: Option<bool>,
 }
 
 #[derive(Deserialize, Serialize)]
 pub struct UserList {
     id: i32,
     email: String,
+    is_staff: Option<bool>,
+}
+
+#[derive(Deserialize)]
+pub struct UserCreate {
+    pub email: String,
+    pub password: String,
 }
 
 impl User {
-    pub async fn create(user: User) -> Result<i32, AppError> {
+    pub fn new(email: String, password: String) -> Self {
+        let crypted_password = sha256::digest(password);
+        Self {
+            id: 0,
+            email,
+            password: crypted_password,
+            is_staff: Some(false),
+        }
+    }
+
+    pub async fn create(user: User) -> Result<UserList, AppError> {
         let pool = unsafe { get_client() };
-        let rec = sqlx::query!(
+        let rec = sqlx::query_as!(
+            UserList,
             r#"
                 INSERT INTO users (email, password)
                 VALUES ( $1, $2 )
-                RETURNING id
+                RETURNING id, email, is_staff
             "#,
             user.email,
             user.password
@@ -31,12 +50,12 @@ impl User {
         .fetch_one(pool)
         .await?;
 
-        Ok(rec.id)
+        Ok(rec)
     }
 
     pub async fn list() -> Result<Vec<UserList>, AppError> {
         let pool = unsafe { get_client() };
-        let rows = sqlx::query_as!(UserList, r#"SELECT id, email FROM users"#)
+        let rows = sqlx::query_as!(UserList, r#"SELECT id, email, is_staff FROM users"#)
             .fetch_all(pool)
             .await?;
 
-- 
cgit v1.2.3-18-g5258