]> Untitled Git - lemmy.git/blob - crates/api/src/local_user/report_count.rs
Make functions work with both connection and pool (#3420)
[lemmy.git] / crates / api / src / local_user / report_count.rs
1 use crate::Perform;
2 use actix_web::web::Data;
3 use lemmy_api_common::{
4   context::LemmyContext,
5   person::{GetReportCount, GetReportCountResponse},
6   utils::local_user_view_from_jwt,
7 };
8 use lemmy_db_views::structs::{CommentReportView, PostReportView, PrivateMessageReportView};
9 use lemmy_utils::error::LemmyError;
10
11 #[async_trait::async_trait(?Send)]
12 impl Perform for GetReportCount {
13   type Response = GetReportCountResponse;
14
15   #[tracing::instrument(skip(context))]
16   async fn perform(
17     &self,
18     context: &Data<LemmyContext>,
19   ) -> Result<GetReportCountResponse, LemmyError> {
20     let data: &GetReportCount = self;
21     let local_user_view = local_user_view_from_jwt(&data.auth, context).await?;
22
23     let person_id = local_user_view.person.id;
24     let admin = local_user_view.person.admin;
25     let community_id = data.community_id;
26
27     let comment_reports =
28       CommentReportView::get_report_count(&mut context.pool(), person_id, admin, community_id)
29         .await?;
30
31     let post_reports =
32       PostReportView::get_report_count(&mut context.pool(), person_id, admin, community_id).await?;
33
34     let private_message_reports = if admin && community_id.is_none() {
35       Some(PrivateMessageReportView::get_report_count(&mut context.pool()).await?)
36     } else {
37       None
38     };
39
40     Ok(GetReportCountResponse {
41       community_id,
42       comment_reports,
43       post_reports,
44       private_message_reports,
45     })
46   }
47 }