crates/api/src/local_user/notifications/unread_count.rs

   1 use crate::Perform;
   2 use actix_web::web::Data;
   3 use lemmy_api_common::{
   4   context::LemmyContext,
   5   person::{GetUnreadCount, GetUnreadCountResponse},
   6   utils::local_user_view_from_jwt,
   7 };
   8 use lemmy_db_views::structs::PrivateMessageView;
   9 use lemmy_db_views_actor::structs::{CommentReplyView, PersonMentionView};
  10 use lemmy_utils::error::LemmyError;
  11
  12 #[async_trait::async_trait(?Send)]
  13 impl Perform for GetUnreadCount {
  14   type Response = GetUnreadCountResponse;
  15
  16   #[tracing::instrument(skip(context))]
  17   async fn perform(&self, context: &Data<LemmyContext>) -> Result<Self::Response, LemmyError> {
  18     let data = self;
  19     let local_user_view = local_user_view_from_jwt(&data.auth, context).await?;
  20
  21     let person_id = local_user_view.person.id;
  22
  23     let replies = CommentReplyView::get_unread_replies(context.pool(), person_id).await?;
  24
  25     let mentions = PersonMentionView::get_unread_mentions(context.pool(), person_id).await?;
  26
  27     let private_messages =
  28       PrivateMessageView::get_unread_messages(context.pool(), person_id).await?;
  29
  30     Ok(Self::Response {
  31       replies,
  32       mentions,
  33       private_messages,
  34     })
  35   }
  36 }