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

use crate::Perform;
use actix_web::web::Data;
use lemmy_api_common::{
  context::LemmyContext,
  person::{GetUnreadCount, GetUnreadCountResponse},
  utils::local_user_view_from_jwt,
};
use lemmy_db_views::structs::PrivateMessageView;
use lemmy_db_views_actor::structs::{CommentReplyView, PersonMentionView};
use lemmy_utils::error::LemmyError;

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

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

    let person_id = local_user_view.person.id;

    let replies = CommentReplyView::get_unread_replies(&mut conn, person_id).await?;

    let mentions = PersonMentionView::get_unread_mentions(&mut conn, person_id).await?;

    let private_messages = PrivateMessageView::get_unread_messages(&mut conn, person_id).await?;

    Ok(Self::Response {
      replies,
      mentions,
      private_messages,
    })
  }
}