lemmy/crates/db_schema/src/impls/local_user.rs

142 lines
3.8 KiB
Rust
Raw Normal View History

2021-10-16 13:33:38 +00:00
use crate::{
newtypes::LocalUserId,
2021-03-11 04:43:11 +00:00
schema::local_user::dsl::*,
2022-08-23 19:38:51 +00:00
source::{
actor_language::{LocalUserLanguage, SiteLanguage},
local_user::{LocalUser, LocalUserInsertForm, LocalUserUpdateForm},
2022-08-23 19:38:51 +00:00
},
2021-10-16 13:33:38 +00:00
traits::Crud,
utils::naive_now,
2021-03-11 04:43:11 +00:00
};
2021-10-16 13:33:38 +00:00
use bcrypt::{hash, DEFAULT_COST};
use diesel::{dsl::*, result::Error, *};
2021-03-10 22:33:55 +00:00
2021-02-26 13:49:58 +00:00
mod safe_settings_type {
2021-10-16 13:33:38 +00:00
use crate::{
schema::local_user::columns::*,
source::local_user::LocalUser,
traits::ToSafeSettings,
};
2021-02-26 13:49:58 +00:00
type Columns = (
id,
2021-03-10 22:33:55 +00:00
person_id,
2021-02-26 13:49:58 +00:00
email,
show_nsfw,
theme,
default_sort_type,
default_listing_type,
interface_language,
2021-02-26 13:49:58 +00:00
show_avatars,
send_notifications_to_email,
validator_time,
show_bot_accounts,
show_scores,
show_read_posts,
show_new_post_notifs,
email_verified,
accepted_application,
2021-02-26 13:49:58 +00:00
);
2021-03-10 22:33:55 +00:00
impl ToSafeSettings for LocalUser {
2021-02-26 13:49:58 +00:00
type SafeSettingsColumns = Columns;
2021-03-11 04:43:11 +00:00
2021-03-10 22:33:55 +00:00
/// Includes everything but the hashed password
2021-02-26 13:49:58 +00:00
fn safe_settings_columns_tuple() -> Self::SafeSettingsColumns {
(
id,
2021-03-10 22:33:55 +00:00
person_id,
2021-02-26 13:49:58 +00:00
email,
show_nsfw,
theme,
default_sort_type,
default_listing_type,
interface_language,
2021-02-26 13:49:58 +00:00
show_avatars,
send_notifications_to_email,
validator_time,
show_bot_accounts,
show_scores,
show_read_posts,
show_new_post_notifs,
email_verified,
accepted_application,
2021-02-26 13:49:58 +00:00
)
}
}
}
2021-10-16 13:33:38 +00:00
impl LocalUser {
pub fn update_password(
conn: &mut PgConnection,
local_user_id: LocalUserId,
2021-03-11 04:43:11 +00:00
new_password: &str,
) -> Result<Self, Error> {
2021-03-10 22:33:55 +00:00
let password_hash = hash(new_password, DEFAULT_COST).expect("Couldn't hash password");
diesel::update(local_user.find(local_user_id))
.set((
password_encrypted.eq(password_hash),
validator_time.eq(naive_now()),
))
2021-03-10 22:33:55 +00:00
.get_result::<Self>(conn)
}
pub fn set_all_users_email_verified(conn: &mut PgConnection) -> Result<Vec<Self>, Error> {
diesel::update(local_user)
.set(email_verified.eq(true))
.get_results::<Self>(conn)
}
pub fn set_all_users_registration_applications_accepted(
conn: &mut PgConnection,
) -> Result<Vec<Self>, Error> {
diesel::update(local_user)
.set(accepted_application.eq(true))
.get_results::<Self>(conn)
}
2021-03-10 22:33:55 +00:00
}
impl Crud for LocalUser {
type InsertForm = LocalUserInsertForm;
type UpdateForm = LocalUserUpdateForm;
type IdType = LocalUserId;
fn read(conn: &mut PgConnection, local_user_id: LocalUserId) -> Result<Self, Error> {
2021-03-11 04:43:11 +00:00
local_user.find(local_user_id).first::<Self>(conn)
2021-03-10 22:33:55 +00:00
}
fn delete(conn: &mut PgConnection, local_user_id: LocalUserId) -> Result<usize, Error> {
2021-03-10 22:33:55 +00:00
diesel::delete(local_user.find(local_user_id)).execute(conn)
}
fn create(conn: &mut PgConnection, form: &Self::InsertForm) -> Result<Self, Error> {
let mut form_with_encrypted_password = form.clone();
let password_hash =
hash(&form.password_encrypted, DEFAULT_COST).expect("Couldn't hash password");
form_with_encrypted_password.password_encrypted = password_hash;
2022-08-23 19:38:51 +00:00
let local_user_ = insert_into(local_user)
.values(form_with_encrypted_password)
2022-08-23 19:38:51 +00:00
.get_result::<Self>(conn)?;
let site_languages = SiteLanguage::read_local(conn);
if let Ok(langs) = site_languages {
// if site exists, init user with site languages
LocalUserLanguage::update(conn, langs, local_user_.id)?;
} else {
// otherwise, init with all languages (this only happens during tests and
// for first admin user, which is created before site)
LocalUserLanguage::update(conn, vec![], local_user_.id)?;
}
2022-08-23 19:38:51 +00:00
Ok(local_user_)
2021-03-10 22:33:55 +00:00
}
fn update(
conn: &mut PgConnection,
local_user_id: LocalUserId,
form: &Self::UpdateForm,
) -> Result<Self, Error> {
2021-03-10 22:33:55 +00:00
diesel::update(local_user.find(local_user_id))
.set(form)
.get_result::<Self>(conn)
}
}