Refactor users management (#2)
All checks were successful
continuous-integration/drone/push Build is passing

* Create UserBackend trait
This commit is contained in:
2022-11-19 18:18:46 +01:00
parent 65d334b947
commit fcf3ec7036
3 changed files with 96 additions and 61 deletions

View File

@ -1,8 +1,23 @@
use actix::{Actor, Context, Handler, Message, MessageResult};
use std::net::IpAddr;
use crate::data::entity_manager::EntityManager;
use crate::data::user::{User, UserID};
use crate::utils::err::Res;
/// User storage interface
pub trait UsersBackend {
fn find_by_username_or_email(&self, u: &str) -> Option<User>;
fn find_by_user_id(&self, id: &UserID) -> Option<User>;
fn get_entire_users_list(&self) -> Vec<User>;
fn change_user_password(&mut self, id: &UserID, password: &str, temporary: bool) -> bool;
fn verify_user_password(&self, user: &UserID, password: &str) -> bool;
fn save_new_successful_2fa_authentication(&mut self, id: &UserID, ip: IpAddr) -> bool;
fn clear_2fa_login_history(&mut self, id: &UserID) -> bool;
fn delete_account(&mut self, id: &UserID) -> bool;
// FIXME : remove this
fn update_or_insert_user(&mut self, user: User) -> Res;
}
#[derive(Debug)]
pub enum LoginResult {
@ -69,12 +84,17 @@ pub struct UpdateUserRequest(pub User);
pub struct DeleteUserRequest(pub UserID);
pub struct UsersActor {
manager: EntityManager<User>,
manager: Box<dyn UsersBackend>,
}
impl UsersActor {
pub fn new(manager: EntityManager<User>) -> Self {
Self { manager }
pub fn new<E>(manager: E) -> Self
where
E: UsersBackend + 'static,
{
Self {
manager: Box::new(manager),
}
}
}
@ -158,7 +178,7 @@ impl Handler<GetAllUsersRequest> for UsersActor {
type Result = MessageResult<GetAllUsersRequest>;
fn handle(&mut self, _msg: GetAllUsersRequest, _ctx: &mut Self::Context) -> Self::Result {
MessageResult(GetAllUsersResult(self.manager.cloned()))
MessageResult(GetAllUsersResult(self.manager.get_entire_users_list()))
}
}
@ -166,7 +186,7 @@ impl Handler<UpdateUserRequest> for UsersActor {
type Result = <UpdateUserRequest as actix::Message>::Result;
fn handle(&mut self, msg: UpdateUserRequest, _ctx: &mut Self::Context) -> Self::Result {
match self.manager.update_or_push(msg.0) {
match self.manager.update_or_insert_user(msg.0) {
Ok(_) => true,
Err(e) => {
log::error!("Failed to update user information! {:?}", e);
@ -180,23 +200,6 @@ impl Handler<DeleteUserRequest> for UsersActor {
type Result = <DeleteUserRequest as actix::Message>::Result;
fn handle(&mut self, msg: DeleteUserRequest, _ctx: &mut Self::Context) -> Self::Result {
let user = match self.manager.find_by_user_id(&msg.0) {
None => {
log::warn!(
"Could not delete account {:?} because it was not found!",
msg.0
);
return false;
}
Some(s) => s,
};
match self.manager.remove(&user) {
Ok(_) => true,
Err(e) => {
log::error!("Failed to update delete account! {:?}", e);
false
}
}
self.manager.delete_account(&msg.0)
}
}