lemmy/crates/db_queries/src/source/local_user.rs

110 lines
2.8 KiB
Rust
Raw Normal View History

2021-03-11 22:47:44 +00:00
use crate::Crud;
2021-03-10 22:33:55 +00:00
use bcrypt::{hash, DEFAULT_COST};
2021-03-11 04:43:11 +00:00
use diesel::{dsl::*, result::Error, *};
use lemmy_db_schema::{
naive_now,
2021-03-11 04:43:11 +00:00
schema::local_user::dsl::*,
2021-03-11 22:47:44 +00:00
source::local_user::{LocalUser, LocalUserForm},
LocalUserId,
2021-03-11 04:43:11 +00:00
};
2021-03-10 22:33:55 +00:00
2021-02-26 13:49:58 +00:00
mod safe_settings_type {
use crate::ToSafeSettings;
2021-03-10 22:33:55 +00:00
use lemmy_db_schema::{schema::local_user::columns::*, source::local_user::LocalUser};
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,
lang,
show_avatars,
send_notifications_to_email,
validator_time,
show_scores,
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,
lang,
show_avatars,
send_notifications_to_email,
validator_time,
show_scores,
2021-02-26 13:49:58 +00:00
)
}
}
}
2021-03-10 22:33:55 +00:00
pub trait LocalUser_ {
fn register(conn: &PgConnection, form: &LocalUserForm) -> Result<LocalUser, Error>;
2021-03-11 04:43:11 +00:00
fn update_password(
conn: &PgConnection,
local_user_id: LocalUserId,
2021-03-11 04:43:11 +00:00
new_password: &str,
) -> Result<LocalUser, Error>;
2021-03-10 22:33:55 +00:00
}
impl LocalUser_ for LocalUser {
fn register(conn: &PgConnection, form: &LocalUserForm) -> Result<Self, Error> {
let mut edited_user = form.clone();
let password_hash =
hash(&form.password_encrypted, DEFAULT_COST).expect("Couldn't hash password");
edited_user.password_encrypted = password_hash;
Self::create(&conn, &edited_user)
}
2021-03-11 04:43:11 +00:00
fn update_password(
conn: &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)
}
}
impl Crud<LocalUserForm, LocalUserId> for LocalUser {
fn read(conn: &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: &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: &PgConnection, form: &LocalUserForm) -> Result<Self, Error> {
2021-03-11 04:43:11 +00:00
insert_into(local_user)
.values(form)
.get_result::<Self>(conn)
2021-03-10 22:33:55 +00:00
}
fn update(
conn: &PgConnection,
local_user_id: LocalUserId,
form: &LocalUserForm,
) -> 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)
}
}