add account related functions

Cargo.toml

@@ -21,3 +21,5 @@ sqlx = { version = "0.7.4", features = [
 tracing = { version = "0.1", features = ["log"] }
 tracing-subscriber = { version = "0.3", features = ["env-filter"] }
 reqwest = "0.12.3"
+rand = "0.8"
+rust-argon2 = "2.1"

migrations/20240417084119_create_accounts_table.down.sql

@@ -0,0 +1,2 @@
+-- Add down migration script here
+DROP TABLE IF EXISTS accounts;

migrations/20240417084119_create_accounts_table.up.sql

@@ -0,0 +1,6 @@
+-- Add up migration script here
+CREATE TABLE IF NOT EXISTS accounts (
+    id serial NOT NULL,
+    email VARCHAR(255) NOT NULL PRIMARY KEY,
+    password VARCHAR(255) NOT NULL
+);

src/handlers/account.rs

@@ -0,0 +1,33 @@
+use argon2::Config;
+use axum::{extract::State, Json};
+use rand::Rng;
+use tracing::{event, Level};
+
+use crate::{common::error::Error, models::account::Account, repositories::store::Store};
+
+pub async fn register(
+    State(store): State<Store>,
+    Json(account): Json<Account>,
+) -> Result<String, Error> {
+    event!(target:"axum-web-dev", Level::INFO, "register new user");
+    let hashed_pwd = hash_passowrd(account.password.as_bytes());
+
+    let account = Account {
+        id: account.id,
+        email: account.email,
+        password: hashed_pwd,
+    };
+
+    let _res: Result<bool, Error> = match store.add_account(account).await {
+        Err(e) => return Err(e),
+        Ok(res) => Ok(res),
+    };
+
+    Ok(String::from("Success"))
+}
+
+pub fn hash_passowrd(password: &[u8]) -> String {
+    let salt = rand::thread_rng().gen::<[u8; 32]>();
+    let config = Config::default();
+    argon2::hash_encoded(password, &salt, &config).unwrap()
+}

src/handlers/mod.rs

@@ -1,5 +1,6 @@
 use axum::{response::IntoResponse, Json};
 
+pub mod account;
 pub mod answer;
 pub mod question;
 

src/models/account.rs

@@ -0,0 +1,11 @@
+use serde::{Deserialize, Serialize};
+
+#[derive(Serialize, Deserialize, Debug, Clone)]
+pub struct Account {
+    pub id: Option<AccountId>,
+    pub email: String,
+    pub password: String,
+}
+
+#[derive(Serialize, Deserialize, Debug, Clone, PartialEq, Eq, Hash)]
+pub struct AccountId(pub i32);

src/models/mod.rs

@@ -1,5 +1,6 @@
 use serde::Deserialize;
 
+pub mod account;
 pub mod answer;
 pub mod question;
 

src/repositories/store.rs

@@ -8,6 +8,7 @@ use tracing::event;
 use crate::{
     common::error::Error,
     models::{
+        account::Account,
         answer::{Answer, AnswerId, NewAnswer},
         question::{NewQuestion, Question, QuestionId},
     },
@@ -167,4 +168,33 @@ impl Store {
             }
         }
     }
+
+    pub async fn add_account(self, account: Account) -> Result<bool, Error> {
+        match sqlx::query("INSERT INTO accounts (email, password) VALUES($1, $2)")
+            .bind(account.email)
+            .bind(account.password)
+            .execute(&self.connection)
+            .await
+        {
+            Ok(_) => Ok(true),
+            Err(error) => {
+                let err_code = error
+                    .as_database_error()
+                    .unwrap()
+                    .code()
+                    .unwrap()
+                    .parse::<i32>()
+                    .unwrap();
+                let err_message = error.as_database_error().unwrap().message();
+                let err_constraint = error.as_database_error().unwrap().constraint().unwrap();
+                event!(
+                    tracing::Level::ERROR,
+                    code = err_code,
+                    message = err_message,
+                    constraint = err_constraint
+                );
+                Err(Error::DatabaseQueryError)
+            }
+        }
+    }
 }

src/routers/account.rs

@@ -0,0 +1,9 @@
+use axum::{routing::post, Router};
+
+use crate::{handlers::account::register, repositories::store::Store};
+
+pub fn create_router(store: Store) -> Router {
+    Router::new()
+        .route("/api/registration", post(register))
+        .with_state(store)
+}

src/routers/mod.rs

@@ -6,6 +6,7 @@ use tower_http::trace::TraceLayer;
 
 use crate::{handlers::health_check_handler, repositories::store::Store};
 
+pub mod account;
 pub mod answer;
 pub mod question;
 
@@ -14,5 +15,6 @@ pub fn create_router(store: Store) -> Router {
         .route("/api/healthcheck", get(health_check_handler))
         .merge(question::create_router(store.clone()))
         .merge(answer::create_router(store.clone()))
+        .merge(account::create_router(store.clone()))
         .layer(TraceLayer::new_for_http())
 }