feat: basic auth

This commit is contained in:
Maulana Sodiqin
2025-03-14 09:53:20 +07:00
parent acfb8259bc
commit 9ffe808797
7 changed files with 166 additions and 80 deletions
+3 -5
View File
@@ -1,6 +1,4 @@
use super::{ use super::{AuthLoginRequestDto, AuthRegisterRequestDto, AuthService};
mutation_login, mutation_register, AuthLoginRequestDto, AuthRegisterRequestDto,
};
use crate::AppState; use crate::AppState;
use axum::{response::IntoResponse, Extension, Json}; use axum::{response::IntoResponse, Extension, Json};
@@ -8,12 +6,12 @@ pub async fn post_login(
Extension(state): Extension<AppState>, Extension(state): Extension<AppState>,
Json(payload): Json<AuthLoginRequestDto>, Json(payload): Json<AuthLoginRequestDto>,
) -> impl IntoResponse { ) -> impl IntoResponse {
mutation_login(payload, &state).await AuthService::mutation_login(payload, &state).await
} }
pub async fn post_register( pub async fn post_register(
Extension(state): Extension<AppState>, Extension(state): Extension<AppState>,
Json(payload): Json<AuthRegisterRequestDto>, Json(payload): Json<AuthRegisterRequestDto>,
) -> impl IntoResponse { ) -> impl IntoResponse {
mutation_register(payload, &state).await AuthService::mutation_register(payload, &state).await
} }
+3
View File
@@ -1,6 +1,8 @@
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
use utoipa::ToSchema; use utoipa::ToSchema;
use crate::v1::UsersItemDto;
#[derive(Clone, Debug, Serialize, Deserialize, ToSchema)] #[derive(Clone, Debug, Serialize, Deserialize, ToSchema)]
pub struct AuthLoginRequestDto { pub struct AuthLoginRequestDto {
pub email: String, pub email: String,
@@ -10,6 +12,7 @@ pub struct AuthLoginRequestDto {
#[derive(Clone, Debug, Serialize, Deserialize, ToSchema)] #[derive(Clone, Debug, Serialize, Deserialize, ToSchema)]
pub struct AuthLoginResponsetDto { pub struct AuthLoginResponsetDto {
pub token: TokenDto, pub token: TokenDto,
pub user: UsersItemDto,
} }
#[derive(Clone, Debug, Serialize, Deserialize, ToSchema)] #[derive(Clone, Debug, Serialize, Deserialize, ToSchema)]
+37 -24
View File
@@ -1,32 +1,45 @@
use crate::{v1::UsersItemDto, AppState}; use crate::{v1::UsersItemDto, AppState, ResourceEnum};
use std::error::Error; use std::error::Error;
use super::AuthRegisterRequestDto; use super::AuthRegisterRequestDto;
pub async fn query_user_by_email( pub struct AuthRepository<'a> {
email: String, state: &'a AppState,
state: &AppState, }
) -> Result<UsersItemDto, Box<dyn Error>> {
let db = &state.surrealdb;
let mut result = db impl<'a> AuthRepository<'a> {
.query("SELECT * FROM app_users WHERE email = $email LIMIT 1;") pub fn new(state: &'a AppState) -> Self {
.bind(("email", email.clone())) Self { state }
}
pub async fn query_user_by_email(
&self,
email: String,
) -> Result<AuthRegisterRequestDto, Box<dyn Error>> {
let db = &self.state.surrealdb;
let result = db.select((ResourceEnum::Users.to_string(), email)).await?;
match result {
Some(user) => Ok(user),
None => Err("User not found for email".into()),
}
}
pub async fn query_create_user(
&self,
data: AuthRegisterRequestDto,
) -> Result<String, Box<dyn Error>> {
let db = &self.state.surrealdb;
let record: Option<UsersItemDto> = db
.create((ResourceEnum::Users.to_string(), &data.email))
.content(data)
.await?; .await?;
let user: Option<UsersItemDto> = result.take(0)?; match record {
Some(_) => Ok("Success create user".into()),
user.ok_or_else(|| format!("User not found for email: {}", email).into()) None => Err("Failed to create user".into()),
} }
}
pub async fn query_create_user(
data: AuthRegisterRequestDto,
state: &AppState,
) -> Result<String, Box<dyn Error>> {
let db = &state.surrealdb;
let _record: Option<UsersItemDto> =
db.create(("app_users", &data.email)).content(data).await?;
Ok("Success create user".into())
} }
+64 -13
View File
@@ -1,19 +1,29 @@
use axum::{http::StatusCode, response::Response}; use axum::{http::StatusCode, response::Response};
use redis::Commands;
use super::{ use super::{
query_create_user, query_user_by_email, AuthLoginRequestDto, AuthLoginRequestDto, AuthLoginResponsetDto, AuthRegisterRequestDto,
AuthRegisterRequestDto, AuthRepository, TokenDto,
};
use crate::{
common_response, encode_access_token, encode_refresh_token, hash_password,
success_response, v1::UsersItemDto, verify_password, AppState,
ResponseSuccessDto, TokenSub,
}; };
use crate::{common_response, hash_password, verify_password, AppState};
pub async fn mutation_login( pub struct AuthService;
impl AuthService {
pub async fn mutation_login(
payload: AuthLoginRequestDto, payload: AuthLoginRequestDto,
state: &AppState, state: &AppState,
) -> Response { ) -> Response {
match query_user_by_email(payload.email, state).await { let repository = AuthRepository::new(state);
match repository.query_user_by_email(payload.email.clone()).await {
Ok(user) => { Ok(user) => {
let is_password_correct = let is_password_correct =
verify_password(&payload.password, &user.password).unwrap_or(false); verify_password(&payload.password, &user.password)
.unwrap_or(false);
if is_password_correct { if is_password_correct {
common_response( common_response(
@@ -22,17 +32,57 @@ pub async fn mutation_login(
); );
} }
common_response(StatusCode::OK, "Success Login") let access_token = encode_access_token(TokenSub {
email: payload.email.clone(),
role_name: "Admin".to_string(),
});
let refresh_token = encode_refresh_token(TokenSub {
email: payload.email.clone(),
role_name: "Admin".to_string(),
});
let response = ResponseSuccessDto {
data: AuthLoginResponsetDto {
user: UsersItemDto {
fullname: user.fullname.clone(),
email: user.email.clone(),
},
token: TokenDto {
access_token: access_token.unwrap(),
refresh_token: refresh_token.unwrap(),
},
},
};
let redis_key =
format!("authenticated_users_data:{}", payload.email.clone());
match state.redisdb.get_connection().and_then(|mut conn| {
conn.set_ex::<_, String, ()>(
&redis_key,
serde_json::to_string(&user).unwrap_or_default(),
86400,
)
}) {
Ok(_) => success_response(response),
Err(err) => common_response(
StatusCode::INTERNAL_SERVER_ERROR,
&format!("Redis storage failed: {}", err),
),
}
} }
Err(err) => common_response(StatusCode::UNAUTHORIZED, &err.to_string()), Err(err) => common_response(StatusCode::UNAUTHORIZED, &err.to_string()),
} }
} }
pub async fn mutation_register( pub async fn mutation_register(
payload: AuthRegisterRequestDto, payload: AuthRegisterRequestDto,
state: &AppState, state: &AppState,
) -> Response { ) -> Response {
if query_user_by_email(payload.email.clone(), state) let repository = AuthRepository::new(state);
if repository
.query_user_by_email(payload.email.clone())
.await .await
.is_ok() .is_ok()
{ {
@@ -55,10 +105,11 @@ pub async fn mutation_register(
fullname: payload.fullname, fullname: payload.fullname,
}; };
match query_create_user(new_user, state).await { match repository.query_create_user(new_user).await {
Ok(_) => common_response(StatusCode::CREATED, "Registration successful"), Ok(_) => common_response(StatusCode::CREATED, "Registration successful"),
Err(err) => { Err(err) => {
common_response(StatusCode::INTERNAL_SERVER_ERROR, &err.to_string()) common_response(StatusCode::INTERNAL_SERVER_ERROR, &err.to_string())
} }
} }
}
} }
-1
View File
@@ -5,5 +5,4 @@ use utoipa::ToSchema;
pub struct UsersItemDto { pub struct UsersItemDto {
pub email: String, pub email: String,
pub fullname: String, pub fullname: String,
pub password: String,
} }
+3
View File
@@ -2,6 +2,9 @@ use crate::SurrealClient;
use surrealdb::engine::remote::http::{Client, Http}; use surrealdb::engine::remote::http::{Client, Http};
use surrealdb::{Result, Surreal}; use surrealdb::{Result, Surreal};
pub mod resource;
pub use resource::*;
pub async fn surrealdb_init() -> Result<SurrealClient> { pub async fn surrealdb_init() -> Result<SurrealClient> {
let db = Surreal::<Client>::init(); let db = Surreal::<Client>::init();
db.connect::<Http>("localhost:8000").await?; db.connect::<Http>("localhost:8000").await?;
+19
View File
@@ -0,0 +1,19 @@
use std::fmt;
#[derive(Debug, Clone, PartialEq, Eq)]
pub enum ResourceEnum {
Users,
Roles,
Permissions,
}
impl fmt::Display for ResourceEnum {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
let str = match self {
ResourceEnum::Users => "app_users",
ResourceEnum::Roles => "app_roles",
ResourceEnum::Permissions => "app_permissions",
};
write!(f, "{}", str)
}
}