mirror of
https://github.com/LemmyNet/lemmy.git
synced 2024-05-17 01:12:39 +00:00
dc327652a5
* wip * stuff * fmt * fmt 2 * fmt 3 * fix default feature * use Authorization header * store ip and user agent for each login * add list_logins endpoint * serde(skip) for token * fix api tests * A few suggestions for login_token (#3991) * A few suggestions. * Fixing SQL format. * review * review * rename cookie --------- Co-authored-by: Dessalines <dessalines@users.noreply.github.com>
111 lines
3.2 KiB
Rust
111 lines
3.2 KiB
Rust
use crate::{
|
|
newtypes::LocalUserId,
|
|
schema::local_user::dsl::{
|
|
accepted_application,
|
|
email,
|
|
email_verified,
|
|
local_user,
|
|
password_encrypted,
|
|
},
|
|
source::{
|
|
actor_language::{LocalUserLanguage, SiteLanguage},
|
|
local_user::{LocalUser, LocalUserInsertForm, LocalUserUpdateForm},
|
|
},
|
|
traits::Crud,
|
|
utils::{
|
|
functions::{coalesce, lower},
|
|
get_conn,
|
|
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),))
|
|
.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(lower(coalesce(email, "")).eq(email_.to_lowercase())),
|
|
))
|
|
.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
|
|
}
|
|
}
|