]> Untitled Git - lemmy.git/blob - crates/api/src/site/registration_applications/unread_count.rs
Make functions work with both connection and pool (#3420)
[lemmy.git] / crates / api / src / site / registration_applications / unread_count.rs
1 use crate::Perform;
2 use actix_web::web::Data;
3 use lemmy_api_common::{
4   context::LemmyContext,
5   site::{GetUnreadRegistrationApplicationCount, GetUnreadRegistrationApplicationCountResponse},
6   utils::{is_admin, local_user_view_from_jwt},
7 };
8 use lemmy_db_schema::source::local_site::LocalSite;
9 use lemmy_db_views::structs::RegistrationApplicationView;
10 use lemmy_utils::error::LemmyError;
11
12 #[async_trait::async_trait(?Send)]
13 impl Perform for GetUnreadRegistrationApplicationCount {
14   type Response = GetUnreadRegistrationApplicationCountResponse;
15
16   async fn perform(&self, context: &Data<LemmyContext>) -> Result<Self::Response, LemmyError> {
17     let data = self;
18     let local_user_view = local_user_view_from_jwt(&data.auth, context).await?;
19     let local_site = LocalSite::read(&mut context.pool()).await?;
20
21     // Only let admins do this
22     is_admin(&local_user_view)?;
23
24     let verified_email_only = local_site.require_email_verification;
25
26     let registration_applications =
27       RegistrationApplicationView::get_unread_count(&mut context.pool(), verified_email_only)
28         .await?;
29
30     Ok(Self::Response {
31       registration_applications,
32     })
33   }
34 }