use crate::Perform;
use actix_web::web::Data;
use lemmy_api_common::{
  context::LemmyContext,
  person::{GetReportCount, GetReportCountResponse},
  utils::local_user_view_from_jwt,
};
use lemmy_db_views::structs::{CommentReportView, PostReportView, PrivateMessageReportView};
use lemmy_utils::error::LemmyError;

#[async_trait::async_trait(?Send)]
impl Perform for GetReportCount {
  type Response = GetReportCountResponse;

  #[tracing::instrument(skip(context))]
  async fn perform(
    &self,
    context: &Data<LemmyContext>,
  ) -> Result<GetReportCountResponse, LemmyError> {
    let data: &GetReportCount = self;
    let local_user_view = local_user_view_from_jwt(&data.auth, context).await?;

    let person_id = local_user_view.person.id;
    let admin = local_user_view.local_user.admin;
    let community_id = data.community_id;

    let comment_reports =
      CommentReportView::get_report_count(&mut context.pool(), person_id, admin, community_id)
        .await?;

    let post_reports =
      PostReportView::get_report_count(&mut context.pool(), person_id, admin, community_id).await?;

    let private_message_reports = if admin && community_id.is_none() {
      Some(PrivateMessageReportView::get_report_count(&mut context.pool()).await?)
    } else {
      None
    };

    Ok(GetReportCountResponse {
      community_id,
      comment_reports,
      post_reports,
      private_message_reports,
    })
  }
}