2022-04-13 18:12:25 +00:00
|
|
|
use crate::Perform;
|
|
|
|
use actix_web::web::Data;
|
|
|
|
use lemmy_api_common::{
|
|
|
|
person::{GetUnreadCount, GetUnreadCountResponse},
|
2022-05-03 17:44:13 +00:00
|
|
|
utils::{blocking, get_local_user_view_from_jwt},
|
2022-04-13 18:12:25 +00:00
|
|
|
};
|
2022-07-30 03:55:59 +00:00
|
|
|
use lemmy_db_views::structs::PrivateMessageView;
|
|
|
|
use lemmy_db_views_actor::structs::{CommentReplyView, PersonMentionView};
|
2022-06-02 14:33:41 +00:00
|
|
|
use lemmy_utils::{error::LemmyError, ConnectionId};
|
2022-04-13 18:12:25 +00:00
|
|
|
use lemmy_websocket::LemmyContext;
|
|
|
|
|
|
|
|
#[async_trait::async_trait(?Send)]
|
|
|
|
impl Perform for GetUnreadCount {
|
|
|
|
type Response = GetUnreadCountResponse;
|
|
|
|
|
|
|
|
#[tracing::instrument(skip(context, _websocket_id))]
|
|
|
|
async fn perform(
|
|
|
|
&self,
|
|
|
|
context: &Data<LemmyContext>,
|
|
|
|
_websocket_id: Option<ConnectionId>,
|
|
|
|
) -> Result<Self::Response, LemmyError> {
|
|
|
|
let data = self;
|
|
|
|
let local_user_view =
|
|
|
|
get_local_user_view_from_jwt(&data.auth, context.pool(), context.secret()).await?;
|
|
|
|
|
|
|
|
let person_id = local_user_view.person.id;
|
|
|
|
|
|
|
|
let replies = blocking(context.pool(), move |conn| {
|
2022-07-30 03:55:59 +00:00
|
|
|
CommentReplyView::get_unread_replies(conn, person_id)
|
2022-04-13 18:12:25 +00:00
|
|
|
})
|
|
|
|
.await??;
|
|
|
|
|
|
|
|
let mentions = blocking(context.pool(), move |conn| {
|
|
|
|
PersonMentionView::get_unread_mentions(conn, person_id)
|
|
|
|
})
|
|
|
|
.await??;
|
|
|
|
|
|
|
|
let private_messages = blocking(context.pool(), move |conn| {
|
|
|
|
PrivateMessageView::get_unread_messages(conn, person_id)
|
|
|
|
})
|
|
|
|
.await??;
|
|
|
|
|
|
|
|
let res = Self::Response {
|
|
|
|
replies,
|
|
|
|
mentions,
|
|
|
|
private_messages,
|
|
|
|
};
|
|
|
|
|
|
|
|
Ok(res)
|
|
|
|
}
|
|
|
|
}
|