mirror of
https://github.com/LemmyNet/lemmy.git
synced 2024-12-24 11:51:32 +00:00
fa4825b524
* Removing a few expects from production and test code. - Fixes #5192 * Using if let filter for admin emails. * Fixing unused error. * Adding expect_used = deny to clippy lints. * Update src/lib.rs Co-authored-by: Nutomic <me@nutomic.com> * Update crates/utils/src/settings/structs.rs Co-authored-by: Nutomic <me@nutomic.com> * Update crates/utils/src/settings/mod.rs Co-authored-by: Nutomic <me@nutomic.com> * Some more cleanup. * Fix clippy --------- Co-authored-by: Nutomic <me@nutomic.com>
182 lines
5.6 KiB
Rust
182 lines
5.6 KiB
Rust
use crate::structs::LocalUserView;
|
|
use actix_web::{dev::Payload, FromRequest, HttpMessage, HttpRequest};
|
|
use diesel::{result::Error, BoolExpressionMethods, ExpressionMethods, JoinOnDsl, QueryDsl};
|
|
use diesel_async::RunQueryDsl;
|
|
use lemmy_db_schema::{
|
|
newtypes::{LocalUserId, OAuthProviderId, PersonId},
|
|
schema::{local_user, local_user_vote_display_mode, oauth_account, person, person_aggregates},
|
|
source::{
|
|
instance::Instance,
|
|
local_user::{LocalUser, LocalUserInsertForm},
|
|
person::{Person, PersonInsertForm},
|
|
},
|
|
traits::Crud,
|
|
utils::{
|
|
functions::{coalesce, lower},
|
|
DbConn,
|
|
DbPool,
|
|
ListFn,
|
|
Queries,
|
|
ReadFn,
|
|
},
|
|
};
|
|
use lemmy_utils::error::{LemmyError, LemmyErrorExt, LemmyErrorType, LemmyResult};
|
|
use std::future::{ready, Ready};
|
|
|
|
enum ReadBy<'a> {
|
|
Id(LocalUserId),
|
|
Person(PersonId),
|
|
Name(&'a str),
|
|
NameOrEmail(&'a str),
|
|
Email(&'a str),
|
|
OAuthID(OAuthProviderId, &'a str),
|
|
}
|
|
|
|
enum ListMode {
|
|
AdminsWithEmails,
|
|
}
|
|
|
|
fn queries<'a>(
|
|
) -> Queries<impl ReadFn<'a, LocalUserView, ReadBy<'a>>, impl ListFn<'a, LocalUserView, ListMode>> {
|
|
let selection = (
|
|
local_user::all_columns,
|
|
local_user_vote_display_mode::all_columns,
|
|
person::all_columns,
|
|
person_aggregates::all_columns,
|
|
);
|
|
|
|
let read = move |mut conn: DbConn<'a>, search: ReadBy<'a>| async move {
|
|
let mut query = local_user::table.into_boxed();
|
|
query = match search {
|
|
ReadBy::Id(local_user_id) => query.filter(local_user::id.eq(local_user_id)),
|
|
ReadBy::Email(from_email) => {
|
|
query.filter(lower(coalesce(local_user::email, "")).eq(from_email.to_lowercase()))
|
|
}
|
|
_ => query,
|
|
};
|
|
let mut query = query.inner_join(person::table);
|
|
query = match search {
|
|
ReadBy::Person(person_id) => query.filter(person::id.eq(person_id)),
|
|
ReadBy::Name(name) => query.filter(lower(person::name).eq(name.to_lowercase())),
|
|
ReadBy::NameOrEmail(name_or_email) => query.filter(
|
|
lower(person::name)
|
|
.eq(lower(name_or_email.to_lowercase()))
|
|
.or(lower(coalesce(local_user::email, "")).eq(name_or_email.to_lowercase())),
|
|
),
|
|
_ => query,
|
|
};
|
|
let query = query
|
|
.inner_join(local_user_vote_display_mode::table)
|
|
.inner_join(person_aggregates::table.on(person::id.eq(person_aggregates::person_id)));
|
|
|
|
if let ReadBy::OAuthID(oauth_provider_id, oauth_user_id) = search {
|
|
query
|
|
.inner_join(oauth_account::table)
|
|
.filter(oauth_account::oauth_provider_id.eq(oauth_provider_id))
|
|
.filter(oauth_account::oauth_user_id.eq(oauth_user_id))
|
|
.select(selection)
|
|
.first(&mut conn)
|
|
.await
|
|
} else {
|
|
query.select(selection).first(&mut conn).await
|
|
}
|
|
};
|
|
|
|
let list = move |mut conn: DbConn<'a>, mode: ListMode| async move {
|
|
match mode {
|
|
ListMode::AdminsWithEmails => {
|
|
local_user::table
|
|
.inner_join(local_user_vote_display_mode::table)
|
|
.inner_join(person::table)
|
|
.inner_join(person_aggregates::table.on(person::id.eq(person_aggregates::person_id)))
|
|
.filter(local_user::email.is_not_null())
|
|
.filter(local_user::admin.eq(true))
|
|
.select(selection)
|
|
.load::<LocalUserView>(&mut conn)
|
|
.await
|
|
}
|
|
}
|
|
};
|
|
|
|
Queries::new(read, list)
|
|
}
|
|
|
|
impl LocalUserView {
|
|
pub async fn read(pool: &mut DbPool<'_>, local_user_id: LocalUserId) -> Result<Self, Error> {
|
|
queries().read(pool, ReadBy::Id(local_user_id)).await
|
|
}
|
|
|
|
pub async fn read_person(pool: &mut DbPool<'_>, person_id: PersonId) -> Result<Self, Error> {
|
|
queries().read(pool, ReadBy::Person(person_id)).await
|
|
}
|
|
|
|
pub async fn read_from_name(pool: &mut DbPool<'_>, name: &str) -> Result<Self, Error> {
|
|
queries().read(pool, ReadBy::Name(name)).await
|
|
}
|
|
|
|
pub async fn find_by_email_or_name(
|
|
pool: &mut DbPool<'_>,
|
|
name_or_email: &str,
|
|
) -> Result<Self, Error> {
|
|
queries()
|
|
.read(pool, ReadBy::NameOrEmail(name_or_email))
|
|
.await
|
|
}
|
|
|
|
pub async fn find_by_email(pool: &mut DbPool<'_>, from_email: &str) -> Result<Self, Error> {
|
|
queries().read(pool, ReadBy::Email(from_email)).await
|
|
}
|
|
|
|
pub async fn find_by_oauth_id(
|
|
pool: &mut DbPool<'_>,
|
|
oauth_provider_id: OAuthProviderId,
|
|
oauth_user_id: &str,
|
|
) -> Result<Self, Error> {
|
|
queries()
|
|
.read(pool, ReadBy::OAuthID(oauth_provider_id, oauth_user_id))
|
|
.await
|
|
}
|
|
|
|
pub async fn list_admins_with_emails(pool: &mut DbPool<'_>) -> Result<Vec<Self>, Error> {
|
|
queries().list(pool, ListMode::AdminsWithEmails).await
|
|
}
|
|
|
|
pub async fn create_test_user(
|
|
pool: &mut DbPool<'_>,
|
|
name: &str,
|
|
bio: &str,
|
|
admin: bool,
|
|
) -> LemmyResult<Self> {
|
|
let instance_id = Instance::read_or_create(pool, "example.com".to_string())
|
|
.await?
|
|
.id;
|
|
let person_form = PersonInsertForm {
|
|
display_name: Some(name.to_owned()),
|
|
bio: Some(bio.to_owned()),
|
|
..PersonInsertForm::test_form(instance_id, name)
|
|
};
|
|
let person = Person::create(pool, &person_form).await?;
|
|
|
|
let user_form = match admin {
|
|
true => LocalUserInsertForm::test_form_admin(person.id),
|
|
false => LocalUserInsertForm::test_form(person.id),
|
|
};
|
|
let local_user = LocalUser::create(pool, &user_form, vec![]).await?;
|
|
|
|
LocalUserView::read(pool, local_user.id)
|
|
.await
|
|
.with_lemmy_type(LemmyErrorType::NotFound)
|
|
}
|
|
}
|
|
|
|
impl FromRequest for LocalUserView {
|
|
type Error = LemmyError;
|
|
type Future = Ready<Result<Self, Self::Error>>;
|
|
|
|
fn from_request(req: &HttpRequest, _payload: &mut Payload) -> Self::Future {
|
|
ready(match req.extensions().get::<LocalUserView>() {
|
|
Some(c) => Ok(c.clone()),
|
|
None => Err(LemmyErrorType::IncorrectLogin.into()),
|
|
})
|
|
}
|
|
}
|