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

   1 use crate::Perform;
   2 use actix_web::web::Data;
   3 use lemmy_api_common::{
   4   blocking,
   5   get_local_user_view_from_jwt,
   6   person::{MarkPersonMentionAsRead, PersonMentionResponse},
   7 };
   8 use lemmy_db_schema::{source::person_mention::PersonMention, traits::Crud};
   9 use lemmy_db_views_actor::person_mention_view::PersonMentionView;
  10 use lemmy_utils::{ConnectionId, LemmyError};
  11 use lemmy_websocket::LemmyContext;
  12
  13 #[async_trait::async_trait(?Send)]
  14 impl Perform for MarkPersonMentionAsRead {
  15   type Response = PersonMentionResponse;
  16
  17   #[tracing::instrument(skip(context, _websocket_id))]
  18   async fn perform(
  19     &self,
  20     context: &Data<LemmyContext>,
  21     _websocket_id: Option<ConnectionId>,
  22   ) -> Result<PersonMentionResponse, LemmyError> {
  23     let data: &MarkPersonMentionAsRead = self;
  24     let local_user_view =
  25       get_local_user_view_from_jwt(&data.auth, context.pool(), context.secret()).await?;
  26
  27     let person_mention_id = data.person_mention_id;
  28     let read_person_mention = blocking(context.pool(), move |conn| {
  29       PersonMention::read(conn, person_mention_id)
  30     })
  31     .await??;
  32
  33     if local_user_view.person.id != read_person_mention.recipient_id {
  34       return Err(LemmyError::from_message("couldnt_update_comment"));
  35     }
  36
  37     let person_mention_id = read_person_mention.id;
  38     let read = data.read;
  39     let update_mention =
  40       move |conn: &'_ _| PersonMention::update_read(conn, person_mention_id, read);
  41     blocking(context.pool(), update_mention)
  42       .await?
  43       .map_err(|e| LemmyError::from_error_message(e, "couldnt_update_comment"))?;
  44
  45     let person_mention_id = read_person_mention.id;
  46     let person_id = local_user_view.person.id;
  47     let person_mention_view = blocking(context.pool(), move |conn| {
  48       PersonMentionView::read(conn, person_mention_id, Some(person_id))
  49     })
  50     .await??;
  51
  52     Ok(PersonMentionResponse {
  53       person_mention_view,
  54     })
  55   }
  56 }