mirror of
https://github.com/LemmyNet/lemmy.git
synced 2024-05-20 10:48:05 +00:00
91834d0d21
* h * Start doing stuff * Default impl for Crud::read * Simplify Crud::read lifetimes * fmt * Stuff * Stuff * Successfully make default read implementation work * Restore Person::read * Clean up default Crud::read and rename 'query2 * Replace filter with find * Attempt default Crud::create * Change Crud to Crud<'a> (won't compile) * Revert "Change Crud to Crud<'a> (won't compile)" This reverts commit7ed20f5f71
. * Default Crud::delete * Remove Crud::delete definitions that match default * Remove commented Site::read * Insert trait * Revert "Insert trait" This reverts commit9d780c2403
. * Use non-borrowed forms * Revert "Use non-borrowed forms" This reverts commitd2dd442563
. * Revert "Revert "Change Crud to Crud<'a> (won't compile)"" This reverts commit25a27165a8
. * Fix lifetime for everything except Crud::delete * Fix Crud::delete * Add comment about futures * Attempt Crud::create * Attempt separate CrudBounds * Revert "Attempt separate CrudBounds" This reverts commit1b4ca321c3
. * Try to fix Crud::create * Move lifetime parameters to associated types * Revert "Move lifetime parameters to associated types" This reverts commitaf1bc858ce
. * Revert "Try to fix Crud::create" This reverts commiteec238496c
. * Revert "Revert "Attempt separate CrudBounds"" This reverts commit1ec33ce502
. * Revert "Attempt separate CrudBounds" This reverts commit1b4ca321c3
. * Revert "Attempt Crud::create" This reverts commit47e8071b68
. * Revert "Add comment about futures" This reverts commitb266b14653
. * Revert "Fix Crud::delete" This reverts commit3abcce2eec
. * Revert "Fix lifetime for everything except Crud::delete" This reverts commitc1ad7a161b
. * Revert "Revert "Revert "Change Crud to Crud<'a> (won't compile)""" This reverts commit3129cd0fc3
. * Clean up * Update site.rs
109 lines
3.2 KiB
Rust
109 lines
3.2 KiB
Rust
use crate::{
|
|
newtypes::LocalUserId,
|
|
schema::local_user::dsl::{
|
|
accepted_application,
|
|
email,
|
|
email_verified,
|
|
local_user,
|
|
password_encrypted,
|
|
validator_time,
|
|
},
|
|
source::{
|
|
actor_language::{LocalUserLanguage, SiteLanguage},
|
|
local_user::{LocalUser, LocalUserInsertForm, LocalUserUpdateForm},
|
|
},
|
|
traits::Crud,
|
|
utils::{get_conn, naive_now, DbPool},
|
|
};
|
|
use bcrypt::{hash, DEFAULT_COST};
|
|
use diesel::{dsl::insert_into, result::Error, ExpressionMethods, QueryDsl};
|
|
use diesel_async::RunQueryDsl;
|
|
|
|
impl LocalUser {
|
|
pub async fn update_password(
|
|
pool: &mut DbPool<'_>,
|
|
local_user_id: LocalUserId,
|
|
new_password: &str,
|
|
) -> Result<Self, Error> {
|
|
let conn = &mut get_conn(pool).await?;
|
|
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()),
|
|
))
|
|
.get_result::<Self>(conn)
|
|
.await
|
|
}
|
|
|
|
pub async fn set_all_users_email_verified(pool: &mut DbPool<'_>) -> Result<Vec<Self>, Error> {
|
|
let conn = &mut get_conn(pool).await?;
|
|
diesel::update(local_user)
|
|
.set(email_verified.eq(true))
|
|
.get_results::<Self>(conn)
|
|
.await
|
|
}
|
|
|
|
pub async fn set_all_users_registration_applications_accepted(
|
|
pool: &mut DbPool<'_>,
|
|
) -> Result<Vec<Self>, Error> {
|
|
let conn = &mut get_conn(pool).await?;
|
|
diesel::update(local_user)
|
|
.set(accepted_application.eq(true))
|
|
.get_results::<Self>(conn)
|
|
.await
|
|
}
|
|
|
|
pub async fn is_email_taken(pool: &mut DbPool<'_>, email_: &str) -> Result<bool, Error> {
|
|
use diesel::dsl::{exists, select};
|
|
let conn = &mut get_conn(pool).await?;
|
|
select(exists(local_user.filter(email.eq(email_))))
|
|
.get_result(conn)
|
|
.await
|
|
}
|
|
}
|
|
|
|
#[async_trait]
|
|
impl Crud for LocalUser {
|
|
type InsertForm = LocalUserInsertForm;
|
|
type UpdateForm = LocalUserUpdateForm;
|
|
type IdType = LocalUserId;
|
|
|
|
async fn create(pool: &mut DbPool<'_>, form: &Self::InsertForm) -> Result<Self, Error> {
|
|
let conn = &mut get_conn(pool).await?;
|
|
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;
|
|
|
|
let local_user_ = insert_into(local_user)
|
|
.values(form_with_encrypted_password)
|
|
.get_result::<Self>(conn)
|
|
.await?;
|
|
|
|
let site_languages = SiteLanguage::read_local_raw(pool).await;
|
|
if let Ok(langs) = site_languages {
|
|
// if site exists, init user with site languages
|
|
LocalUserLanguage::update(pool, langs, local_user_.id).await?;
|
|
} else {
|
|
// otherwise, init with all languages (this only happens during tests and
|
|
// for first admin user, which is created before site)
|
|
LocalUserLanguage::update(pool, vec![], local_user_.id).await?;
|
|
}
|
|
|
|
Ok(local_user_)
|
|
}
|
|
async fn update(
|
|
pool: &mut DbPool<'_>,
|
|
local_user_id: LocalUserId,
|
|
form: &Self::UpdateForm,
|
|
) -> Result<Self, Error> {
|
|
let conn = &mut get_conn(pool).await?;
|
|
diesel::update(local_user.find(local_user_id))
|
|
.set(form)
|
|
.get_result::<Self>(conn)
|
|
.await
|
|
}
|
|
}
|