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