lemmy/crates/db_views/src/local_user_view.rs

181 lines
5.4 KiB
Rust
Raw Normal View History

use crate::structs::{LocalUserSettingsView, LocalUserView};
2022-11-09 10:05:00 +00:00
use diesel::{result::Error, BoolExpressionMethods, ExpressionMethods, JoinOnDsl, QueryDsl};
use diesel_async::RunQueryDsl;
2021-03-10 22:33:55 +00:00
use lemmy_db_schema::{
aggregates::structs::PersonAggregates,
2021-10-16 13:33:38 +00:00
newtypes::{LocalUserId, PersonId},
2021-03-11 04:43:11 +00:00
schema::{local_user, person, person_aggregates},
source::{
local_user::{LocalUser, LocalUserSettings},
person::{Person, PersonSafe},
},
traits::{ToSafe, ToSafeSettings, ViewToVec},
2022-11-09 10:05:00 +00:00
utils::{functions::lower, get_conn, DbPool},
2021-03-10 22:33:55 +00:00
};
2021-03-11 22:47:44 +00:00
type LocalUserViewTuple = (LocalUser, Person, PersonAggregates);
2021-03-10 22:33:55 +00:00
impl LocalUserView {
2022-11-09 10:05:00 +00:00
pub async fn read(pool: &DbPool, local_user_id: LocalUserId) -> Result<Self, Error> {
let conn = &mut get_conn(pool).await?;
2021-03-11 22:47:44 +00:00
let (local_user, person, counts) = local_user::table
.find(local_user_id)
.inner_join(person::table)
.inner_join(person_aggregates::table.on(person::id.eq(person_aggregates::person_id)))
2021-03-11 04:43:11 +00:00
.select((
2021-03-11 22:47:44 +00:00
local_user::all_columns,
2021-03-11 04:43:11 +00:00
person::all_columns,
person_aggregates::all_columns,
))
2022-11-09 10:05:00 +00:00
.first::<LocalUserViewTuple>(conn)
.await?;
2021-03-11 04:43:11 +00:00
Ok(Self {
2021-03-11 22:47:44 +00:00
local_user,
2021-03-11 04:43:11 +00:00
person,
counts,
2021-03-11 22:47:44 +00:00
})
}
2022-11-09 10:05:00 +00:00
pub async fn read_person(pool: &DbPool, person_id: PersonId) -> Result<Self, Error> {
let conn = &mut get_conn(pool).await?;
2021-03-11 22:47:44 +00:00
let (local_user, person, counts) = local_user::table
.filter(person::id.eq(person_id))
.inner_join(person::table)
.inner_join(person_aggregates::table.on(person::id.eq(person_aggregates::person_id)))
.select((
local_user::all_columns,
person::all_columns,
person_aggregates::all_columns,
))
2022-11-09 10:05:00 +00:00
.first::<LocalUserViewTuple>(conn)
.await?;
2021-03-11 22:47:44 +00:00
Ok(Self {
2021-03-11 04:43:11 +00:00
local_user,
2021-03-11 22:47:44 +00:00
person,
counts,
2021-03-11 04:43:11 +00:00
})
2021-03-10 22:33:55 +00:00
}
// TODO check where this is used
2022-11-09 10:05:00 +00:00
pub async fn read_from_name(pool: &DbPool, name: &str) -> Result<Self, Error> {
let conn = &mut get_conn(pool).await?;
2021-03-11 22:47:44 +00:00
let (local_user, person, counts) = local_user::table
2021-03-10 22:33:55 +00:00
.filter(person::name.eq(name))
2021-03-11 22:47:44 +00:00
.inner_join(person::table)
.inner_join(person_aggregates::table.on(person::id.eq(person_aggregates::person_id)))
2021-03-11 04:43:11 +00:00
.select((
2021-03-11 22:47:44 +00:00
local_user::all_columns,
2021-03-11 04:43:11 +00:00
person::all_columns,
person_aggregates::all_columns,
))
2022-11-09 10:05:00 +00:00
.first::<LocalUserViewTuple>(conn)
.await?;
2021-03-11 04:43:11 +00:00
Ok(Self {
2021-03-25 19:30:15 +00:00
local_user,
2021-03-11 04:43:11 +00:00
person,
counts,
})
}
2022-11-09 10:05:00 +00:00
pub async fn find_by_email_or_name(pool: &DbPool, name_or_email: &str) -> Result<Self, Error> {
let conn = &mut get_conn(pool).await?;
2021-03-11 22:47:44 +00:00
let (local_user, person, counts) = local_user::table
.inner_join(person::table)
.inner_join(person_aggregates::table.on(person::id.eq(person_aggregates::person_id)))
2021-03-11 04:43:11 +00:00
.filter(
lower(person::name)
.eq(lower(name_or_email))
2021-11-19 01:58:11 +00:00
.or(local_user::email.eq(name_or_email)),
2021-03-11 04:43:11 +00:00
)
.select((
2021-03-11 22:47:44 +00:00
local_user::all_columns,
2021-03-11 04:43:11 +00:00
person::all_columns,
person_aggregates::all_columns,
))
2022-11-09 10:05:00 +00:00
.first::<LocalUserViewTuple>(conn)
.await?;
2021-03-11 04:43:11 +00:00
Ok(Self {
2021-03-25 19:30:15 +00:00
local_user,
2021-03-11 04:43:11 +00:00
person,
counts,
})
2021-03-10 22:33:55 +00:00
}
2022-11-09 10:05:00 +00:00
pub async fn find_by_email(pool: &DbPool, from_email: &str) -> Result<Self, Error> {
let conn = &mut get_conn(pool).await?;
2021-03-11 22:47:44 +00:00
let (local_user, person, counts) = local_user::table
.inner_join(person::table)
.inner_join(person_aggregates::table.on(person::id.eq(person_aggregates::person_id)))
2021-03-11 04:43:11 +00:00
.filter(local_user::email.eq(from_email))
.select((
2021-03-11 22:47:44 +00:00
local_user::all_columns,
2021-03-11 04:43:11 +00:00
person::all_columns,
person_aggregates::all_columns,
))
2022-11-09 10:05:00 +00:00
.first::<LocalUserViewTuple>(conn)
.await?;
2021-03-11 04:43:11 +00:00
Ok(Self {
2021-03-25 19:30:15 +00:00
local_user,
2021-03-11 04:43:11 +00:00
person,
counts,
})
2021-03-10 22:33:55 +00:00
}
}
2021-03-11 22:47:44 +00:00
type LocalUserSettingsViewTuple = (LocalUserSettings, PersonSafe, PersonAggregates);
2021-03-10 22:33:55 +00:00
impl LocalUserSettingsView {
2022-11-09 10:05:00 +00:00
pub async fn read(pool: &DbPool, local_user_id: LocalUserId) -> Result<Self, Error> {
let conn = &mut get_conn(pool).await?;
2021-03-11 22:47:44 +00:00
let (local_user, person, counts) = local_user::table
.find(local_user_id)
.inner_join(person::table)
.inner_join(person_aggregates::table.on(person::id.eq(person_aggregates::person_id)))
2021-03-11 04:43:11 +00:00
.select((
2021-03-11 22:47:44 +00:00
LocalUser::safe_settings_columns_tuple(),
2021-03-11 04:43:11 +00:00
Person::safe_columns_tuple(),
person_aggregates::all_columns,
))
2022-11-09 10:05:00 +00:00
.first::<LocalUserSettingsViewTuple>(conn)
.await?;
2021-03-11 04:43:11 +00:00
Ok(Self {
2021-03-25 19:30:15 +00:00
local_user,
2021-03-11 04:43:11 +00:00
person,
counts,
})
2021-03-10 22:33:55 +00:00
}
2022-11-09 10:05:00 +00:00
pub async fn list_admins_with_emails(pool: &DbPool) -> Result<Vec<Self>, Error> {
let conn = &mut get_conn(pool).await?;
let res = local_user::table
.filter(person::admin.eq(true))
.filter(local_user::email.is_not_null())
.inner_join(person::table)
.inner_join(person_aggregates::table.on(person::id.eq(person_aggregates::person_id)))
.select((
LocalUser::safe_settings_columns_tuple(),
Person::safe_columns_tuple(),
person_aggregates::all_columns,
))
2022-11-09 10:05:00 +00:00
.load::<LocalUserSettingsViewTuple>(conn)
.await?;
Ok(LocalUserSettingsView::from_tuple_to_vec(res))
}
}
impl ViewToVec for LocalUserSettingsView {
type DbTuple = LocalUserSettingsViewTuple;
fn from_tuple_to_vec(items: Vec<Self::DbTuple>) -> Vec<Self> {
items
.into_iter()
.map(|a| Self {
local_user: a.0,
person: a.1,
counts: a.2,
})
.collect::<Vec<Self>>()
}
2021-03-10 22:33:55 +00:00
}