use super::CredentialsStoreOperations;
use crate::biome::credentials::store::diesel::{schema::user_credentials, CredentialsStoreError};
use crate::biome::credentials::store::CredentialsModel;
use diesel::{dsl::delete, prelude::*, result::Error::NotFound};
pub(in crate::biome::credentials) trait CredentialsStoreRemoveCredentialsOperation {
fn remove_credentials(&self, user_id: &str) -> Result<(), CredentialsStoreError>;
}
impl<'a, C> CredentialsStoreRemoveCredentialsOperation for CredentialsStoreOperations<'a, C>
where
C: diesel::Connection,
i64: diesel::deserialize::FromSql<diesel::sql_types::BigInt, C::Backend>,
String: diesel::deserialize::FromSql<diesel::sql_types::Text, C::Backend>,
{
fn remove_credentials(&self, user_id: &str) -> Result<(), CredentialsStoreError> {
let credentials = user_credentials::table
.filter(user_credentials::user_id.eq(user_id))
.first::<CredentialsModel>(self.conn)
.map(Some)
.or_else(|err| if err == NotFound { Ok(None) } else { Err(err) })
.map_err(|err| CredentialsStoreError::QueryError {
context: "Failed check for existing user id".to_string(),
source: Box::new(err),
})?;
if credentials.is_none() {
return Err(CredentialsStoreError::NotFoundError(format!(
"Credentials not found for user id: {}",
user_id
)));
}
delete(user_credentials::table.filter(user_credentials::user_id.eq(user_id)))
.execute(self.conn)
.map(|_| ())
.map_err(|err| CredentialsStoreError::OperationError {
context: "Failed to delete credentials".to_string(),
source: Box::new(err),
})?;
Ok(())
}
}