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   person::{GetUnreadCount, GetUnreadCountResponse},
   5   utils::{blocking, get_local_user_view_from_jwt},
   6 };
   7 use lemmy_db_views::structs::{CommentView, PrivateMessageView};
   8 use lemmy_db_views_actor::structs::PersonMentionView;
   9 use lemmy_utils::{error::LemmyError, ConnectionId};
  10 use lemmy_websocket::LemmyContext;
  11
  12 #[async_trait::async_trait(?Send)]
  13 impl Perform for GetUnreadCount {
  14   type Response = GetUnreadCountResponse;
  15
  16   #[tracing::instrument(skip(context, _websocket_id))]
  17   async fn perform(
  18     &self,
  19     context: &Data<LemmyContext>,
  20     _websocket_id: Option<ConnectionId>,
  21   ) -> Result<Self::Response, LemmyError> {
  22     let data = self;
  23     let local_user_view =
  24       get_local_user_view_from_jwt(&data.auth, context.pool(), context.secret()).await?;
  25
  26     let person_id = local_user_view.person.id;
  27
  28     let replies = blocking(context.pool(), move |conn| {
  29       CommentView::get_unread_replies(conn, person_id)
  30     })
  31     .await??;
  32
  33     let mentions = blocking(context.pool(), move |conn| {
  34       PersonMentionView::get_unread_mentions(conn, person_id)
  35     })
  36     .await??;
  37
  38     let private_messages = blocking(context.pool(), move |conn| {
  39       PrivateMessageView::get_unread_messages(conn, person_id)
  40     })
  41     .await??;
  42
  43     let res = Self::Response {
  44       replies,
  45       mentions,
  46       private_messages,
  47     };
  48
  49     Ok(res)
  50   }
  51 }