From 5b70d8df4f08b336674e32b401a3c76585a54f0f Mon Sep 17 00:00:00 2001 From: Sebastian Pravda Date: Fri, 11 Nov 2022 18:53:01 +0100 Subject: [PATCH] refactor: candidate variables --- api/src/routes/candidate.rs | 2 +- core/src/services/admin_service.rs | 2 +- core/src/services/candidate_service.rs | 18 +++++++++--------- core/src/services/session_service.rs | 4 ++-- 4 files changed, 13 insertions(+), 13 deletions(-) diff --git a/api/src/routes/candidate.rs b/api/src/routes/candidate.rs index 60fee41..671c1a7 100644 --- a/api/src/routes/candidate.rs +++ b/api/src/routes/candidate.rs @@ -67,7 +67,7 @@ pub async fn fill_details( let form = details.into_inner(); let candidate: entity::candidate::Model = session.into(); - let candidate = CandidateService::add_user_details(db, candidate, form).await; + let candidate = CandidateService::add_candidate_details(db, candidate, form).await; if candidate.is_err() { // TODO cleanup diff --git a/core/src/services/admin_service.rs b/core/src/services/admin_service.rs index f6c27a8..d0d9cdd 100644 --- a/core/src/services/admin_service.rs +++ b/core/src/services/admin_service.rs @@ -55,7 +55,7 @@ impl AdminService { match SessionService::auth_user_session(db, session_uuid).await { Ok(user) => match user { AdminUser::Admin(admin) => Ok(admin), - AdminUser::User(_) => Err(ServiceError::DbError), + AdminUser::Candidate(_) => Err(ServiceError::DbError), }, Err(e) => Err(e), } diff --git a/core/src/services/candidate_service.rs b/core/src/services/candidate_service.rs index 1e8cec8..41c25c1 100644 --- a/core/src/services/candidate_service.rs +++ b/core/src/services/candidate_service.rs @@ -68,9 +68,9 @@ impl CandidateService { .map_err(|_| ServiceError::DbError) } - pub async fn add_user_details( + pub async fn add_candidate_details( db: &DbConn, - user: candidate::Model, + candidate: candidate::Model, form: CandidateDetails, ) -> Result { let Ok(admin_public_keys) = Query::get_all_admin_public_keys(db).await else { @@ -80,13 +80,13 @@ impl CandidateService { let mut admin_public_keys_refrence: Vec<&str> = admin_public_keys.iter().map(|s| &**s).collect(); - let mut recipients = vec![&*user.public_key]; + let mut recipients = vec![&*candidate.public_key]; recipients.append(&mut admin_public_keys_refrence); let enc_details = EncryptedCandidateDetails::new(form, recipients).await; - Mutation::add_candidate_details(db, user, enc_details) + Mutation::add_candidate_details(db, candidate, enc_details) .await .map_err(|_| ServiceError::DbError) } @@ -178,15 +178,15 @@ impl CandidateService { pub async fn login( db: &DbConn, - user_id: i32, + candidate_id: i32, password: String, ip_addr: String, ) -> Result<(String, String), ServiceError> { let session_id = - SessionService::new_session(db, Some(user_id), None, password.clone(), ip_addr).await; + SessionService::new_session(db, Some(candidate_id), None, password.clone(), ip_addr).await; match session_id { Ok(session_id) => { - let private_key = Self::decrypt_private_key(db, user_id, password).await?; + let private_key = Self::decrypt_private_key(db, candidate_id, password).await?; Ok((session_id, private_key)) } Err(e) => Err(e), @@ -196,7 +196,7 @@ impl CandidateService { pub async fn auth(db: &DbConn, session_uuid: Uuid) -> Result { match SessionService::auth_user_session(db, session_uuid).await { Ok(user) => match user { - AdminUser::User(candidate) => Ok(candidate), + AdminUser::Candidate(candidate) => Ok(candidate), AdminUser::Admin(_) => Err(ServiceError::DbError), }, Err(e) => Err(e), @@ -311,7 +311,7 @@ mod tests { sex: "test".to_string(), study: "test".to_string(), }; - CandidateService::add_user_details(&db, candidate, form) + CandidateService::add_candidate_details(&db, candidate, form) .await .ok() .unwrap() diff --git a/core/src/services/session_service.rs b/core/src/services/session_service.rs index f827679..2474b29 100644 --- a/core/src/services/session_service.rs +++ b/core/src/services/session_service.rs @@ -11,7 +11,7 @@ use crate::{ pub enum AdminUser { Admin(entity::admin::Model), - User(entity::candidate::Model), + Candidate(entity::candidate::Model), } pub(in crate::services) struct SessionService; @@ -147,7 +147,7 @@ impl SessionService { if candidate.is_ok() { if let Some(candidate) = candidate.unwrap() { - return Ok(AdminUser::User(candidate)); + return Ok(AdminUser::Candidate(candidate)); } }