From 9a42ed5b80d15845ec0f65b62d5e3c1c18404c99 Mon Sep 17 00:00:00 2001
From: Matthew Kaminski <kaminskimateusz7@gmail.com>
Date: Wed, 28 Aug 2024 23:14:27 -0400
Subject: [PATCH] Fix forget password

---
 src/server/auth/forgot_password.rs | 13 +++++++++++--
 src/server/auth/register.rs        |  4 ++--
 2 files changed, 13 insertions(+), 4 deletions(-)

diff --git a/src/server/auth/forgot_password.rs b/src/server/auth/forgot_password.rs
index 1b7cb41..d1f5b8d 100644
--- a/src/server/auth/forgot_password.rs
+++ b/src/server/auth/forgot_password.rs
@@ -7,7 +7,7 @@ use axum::{
     extract::{Json, State},
     http::StatusCode,
 };
-use sea_orm::{ColumnTrait, EntityTrait, IntoActiveModel, QueryFilter, Set};
+use sea_orm::{ActiveModelTrait, ColumnTrait, EntityTrait, IntoActiveModel, QueryFilter, Set};
 
 pub async fn post_forgot_password(
     State(state): State<ServerState>,
@@ -23,7 +23,16 @@ pub async fn post_forgot_password(
         Some(user) => {
             let mut user = user.into_active_model();
             user.forgot_password_request = Set(Some(password_request.contact_info));
-            (StatusCode::OK, Json(GenericResponse::ok()))
+            let user = user.update(&state.db_conn).await;
+            match user {
+                Ok(_) => return (StatusCode::OK, Json(GenericResponse::ok())),
+                Err(_) => {
+                    return (
+                        StatusCode::BAD_REQUEST,
+                        Json(GenericResponse::err("Database error")),
+                    )
+                }
+            }
         }
         None => (
             StatusCode::BAD_REQUEST,
diff --git a/src/server/auth/register.rs b/src/server/auth/register.rs
index 6238874..5e808b0 100644
--- a/src/server/auth/register.rs
+++ b/src/server/auth/register.rs
@@ -74,10 +74,10 @@ pub async fn post_register_user(
     let db_resp = user::Entity::insert(new_user).exec(&state.db_conn).await;
     match db_resp {
         Ok(_) => {}
-        Err(err) => {
+        Err(_) => {
             return (
                 StatusCode::INTERNAL_SERVER_ERROR,
-                Json(GenericResponse::err(err.to_string().as_str())),
+                Json(GenericResponse::err("Database error")),
             );
         }
     };