Make functions work with both connection and pool (#3420)

[lemmy.git] / crates / apub / src / api / read_person.rs

diff --git a/crates/apub/src/api/read_person.rs b/crates/apub/src/api/read_person.rs
index 35fd59cfe83e0f1232300d7685d2aca88bedd84e..71f4abd519a90293ceea0b462f31f1bf6399a964 100644 (file)
--- a/crates/apub/src/api/read_person.rs
+++ b/crates/apub/src/api/read_person.rs
@@ -25,7 +25,7 @@ pub async fn read_person(
   }
 
   let local_user_view = local_user_view_from_jwt_opt(data.auth.as_ref(), &context).await;
-  let local_site = LocalSite::read(context.pool()).await?;
+  let local_site = LocalSite::read(&mut context.pool()).await?;
   let is_admin = local_user_view.as_ref().map(|luv| is_admin(luv).is_ok());
 
   check_private_instance(&local_user_view, &local_site)?;
@@ -46,7 +46,7 @@ pub async fn read_person(
 
   // You don't need to return settings for the user, since this comes back with GetSite
   // `my_user`
-  let person_view = PersonView::read(context.pool(), person_details_id).await?;
+  let person_view = PersonView::read(&mut context.pool(), person_details_id).await?;
 
   let sort = data.sort;
   let page = data.page;
@@ -56,51 +56,52 @@ pub async fn read_person(
   let local_user = local_user_view.map(|l| l.local_user);
   let local_user_clone = local_user.clone();
 
-  let posts_query = PostQuery::builder()
-    .pool(context.pool())
+  let posts = PostQuery::builder()
+    .pool(&mut context.pool())
     .sort(sort)
     .saved_only(saved_only)
     .local_user(local_user.as_ref())
     .community_id(community_id)
     .is_mod_or_admin(is_admin)
     .page(page)
-    .limit(limit);
-
-  // If its saved only, you don't care what creator it was
-  // Or, if its not saved, then you only want it for that specific creator
-  let posts = if !saved_only.unwrap_or(false) {
-    posts_query
-      .creator_id(Some(person_details_id))
-      .build()
-      .list()
-  } else {
-    posts_query.build().list()
-  }
-  .await?;
+    .limit(limit)
+    .creator_id(
+      // If its saved only, you don't care what creator it was
+      // Or, if its not saved, then you only want it for that specific creator
+      if !saved_only.unwrap_or(false) {
+        Some(person_details_id)
+      } else {
+        None
+      },
+    )
+    .build()
+    .list()
+    .await?;
 
-  let comments_query = CommentQuery::builder()
-    .pool(context.pool())
+  let comments = CommentQuery::builder()
+    .pool(&mut context.pool())
     .local_user(local_user_clone.as_ref())
     .sort(sort.map(post_to_comment_sort_type))
     .saved_only(saved_only)
     .show_deleted_and_removed(Some(false))
     .community_id(community_id)
     .page(page)
-    .limit(limit);
-
-  // If its saved only, you don't care what creator it was
-  // Or, if its not saved, then you only want it for that specific creator
-  let comments = if !saved_only.unwrap_or(false) {
-    comments_query
-      .creator_id(Some(person_details_id))
-      .build()
-      .list()
-  } else {
-    comments_query.build().list()
-  }
-  .await?;
+    .limit(limit)
+    .creator_id(
+      // If its saved only, you don't care what creator it was
+      // Or, if its not saved, then you only want it for that specific creator
+      if !saved_only.unwrap_or(false) {
+        Some(person_details_id)
+      } else {
+        None
+      },
+    )
+    .build()
+    .list()
+    .await?;
 
-  let moderates = CommunityModeratorView::for_person(context.pool(), person_details_id).await?;
+  let moderates =
+    CommunityModeratorView::for_person(&mut context.pool(), person_details_id).await?;
 
   // Return the jwt
   Ok(Json(GetPersonDetailsResponse {