]> Untitled Git - lemmy.git/blobdiff - crates/db_views/src/local_user_view.rs
Make functions work with both connection and pool (#3420)
[lemmy.git] / crates / db_views / src / local_user_view.rs
index 384bf900702555f0ff7f03fa7bebb7463be303fb..567ca3feb336e267bd675c6d179ab3480db6c2d6 100644 (file)
-use diesel::{result::Error, *};
-use lemmy_db_queries::{
-  aggregates::person_aggregates::PersonAggregates,
-  ToSafe,
-  ToSafeSettings,
-};
+use crate::structs::LocalUserView;
+use diesel::{result::Error, BoolExpressionMethods, ExpressionMethods, JoinOnDsl, QueryDsl};
+use diesel_async::RunQueryDsl;
 use lemmy_db_schema::{
-  schema::{person, person_aggregates, local_user},
-  source::person::{PersonSafe, Person},
-  source::local_user::{LocalUser, LocalUserSettings},
+  aggregates::structs::PersonAggregates,
+  newtypes::{LocalUserId, PersonId},
+  schema::{local_user, person, person_aggregates},
+  source::{local_user::LocalUser, person::Person},
+  traits::JoinView,
+  utils::{functions::lower, get_conn, DbPool},
 };
-use serde::Serialize;
-
-#[derive(Debug, Serialize, Clone)]
-pub struct LocalUserView {
-  pub person: Person,
-  pub counts: PersonAggregates,
-  pub local_user: LocalUser,
-}
 
-type LocalUserViewTuple = (Person, PersonAggregates, LocalUser);
+type LocalUserViewTuple = (LocalUser, Person, PersonAggregates);
 
 impl LocalUserView {
-  pub fn read(conn: &PgConnection, person_id: i32) -> Result<Self, Error> {
-    let (person, counts, local_user) = person::table
-      .find(person_id)
-      .inner_join(person_aggregates::table)
-      .inner_join(local_user::table)
-      .select((person::all_columns, person_aggregates::all_columns, local_user::all_columns))
-      .first::<LocalUserViewTuple>(conn)?;
-    Ok(Self { person, counts, local_user })
+  pub async fn read(pool: &mut DbPool<'_>, local_user_id: LocalUserId) -> Result<Self, Error> {
+    let conn = &mut get_conn(pool).await?;
+
+    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)))
+      .select((
+        local_user::all_columns,
+        person::all_columns,
+        person_aggregates::all_columns,
+      ))
+      .first::<LocalUserViewTuple>(conn)
+      .await?;
+    Ok(Self {
+      local_user,
+      person,
+      counts,
+    })
+  }
+
+  pub async fn read_person(pool: &mut DbPool<'_>, person_id: PersonId) -> Result<Self, Error> {
+    let conn = &mut get_conn(pool).await?;
+    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,
+      ))
+      .first::<LocalUserViewTuple>(conn)
+      .await?;
+    Ok(Self {
+      local_user,
+      person,
+      counts,
+    })
   }
 
-  // TODO check where this is used
-  pub fn read_from_name(conn: &PgConnection, name: &str) -> Result<Self, Error> {
-    let (person, counts, local_user) = person::table
-      .filter(person::name.eq(name))
-      .inner_join(person_aggregates::table)
-      .inner_join(local_user::table)
-      .select((person::all_columns, person_aggregates::all_columns, local_user::all_columns))
-      .first::<LocalUserViewTuple>(conn)?;
-    Ok(Self { person, counts, local_user })
+  pub async fn read_from_name(pool: &mut DbPool<'_>, name: &str) -> Result<Self, Error> {
+    let conn = &mut get_conn(pool).await?;
+    let (local_user, person, counts) = local_user::table
+      .filter(lower(person::name).eq(name.to_lowercase()))
+      .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,
+      ))
+      .first::<LocalUserViewTuple>(conn)
+      .await?;
+    Ok(Self {
+      local_user,
+      person,
+      counts,
+    })
   }
 
-  pub fn find_by_email_or_name(
-    conn: &PgConnection,
+  pub async fn find_by_email_or_name(
+    pool: &mut DbPool<'_>,
     name_or_email: &str,
   ) -> Result<Self, Error> {
-    let (person, counts, local_user) = person::table
-      .inner_join(person_aggregates::table)
-      .inner_join(local_user::table)
-      .filter(person::name.ilike(name_or_email).or(local_user::email.ilike(name_or_email)))
-      .select((person::all_columns, person_aggregates::all_columns, local_user::all_columns))
-      .first::<LocalUserViewTuple>(conn)?;
-    Ok(Self { person, counts, local_user })
+    let conn = &mut get_conn(pool).await?;
+    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)))
+      .filter(
+        lower(person::name)
+          .eq(lower(name_or_email))
+          .or(local_user::email.eq(name_or_email)),
+      )
+      .select((
+        local_user::all_columns,
+        person::all_columns,
+        person_aggregates::all_columns,
+      ))
+      .first::<LocalUserViewTuple>(conn)
+      .await?;
+    Ok(Self {
+      local_user,
+      person,
+      counts,
+    })
   }
-}
 
-#[derive(Debug, Serialize, Clone)]
-pub struct LocalUserSettingsView {
-  pub person: PersonSafe,
-  pub counts: PersonAggregates,
-  pub local_user: LocalUserSettings,
-}
+  pub async fn find_by_email(pool: &mut DbPool<'_>, from_email: &str) -> Result<Self, Error> {
+    let conn = &mut get_conn(pool).await?;
+    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)))
+      .filter(local_user::email.eq(from_email))
+      .select((
+        local_user::all_columns,
+        person::all_columns,
+        person_aggregates::all_columns,
+      ))
+      .first::<LocalUserViewTuple>(conn)
+      .await?;
+    Ok(Self {
+      local_user,
+      person,
+      counts,
+    })
+  }
 
-type LocalUserSettingsViewTuple = (PersonSafe, PersonAggregates, LocalUserSettings);
+  pub async fn list_admins_with_emails(pool: &mut 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((
+        local_user::all_columns,
+        person::all_columns,
+        person_aggregates::all_columns,
+      ))
+      .load::<LocalUserViewTuple>(conn)
+      .await?;
+
+    Ok(res.into_iter().map(LocalUserView::from_tuple).collect())
+  }
+}
 
-impl LocalUserSettingsView {
-  pub fn read(conn: &PgConnection, person_id: i32) -> Result<Self, Error> {
-    let (person, counts, local_user) = person::table
-      .find(person_id)
-      .inner_join(person_aggregates::table)
-      .inner_join(local_user::table)
-      .select((Person::safe_columns_tuple(), person_aggregates::all_columns, LocalUser::safe_settings_columns_tuple()))
-      .first::<LocalUserSettingsViewTuple>(conn)?;
-    Ok(Self { person, counts, local_user })
+impl JoinView for LocalUserView {
+  type JoinTuple = LocalUserViewTuple;
+  fn from_tuple(a: Self::JoinTuple) -> Self {
+    Self {
+      local_user: a.0,
+      person: a.1,
+      counts: a.2,
+    }
   }
 }