crates/api/src/private_message/mark_read.rs

   1 use crate::Perform;
   2 use actix_web::web::Data;
   3 use lemmy_api_common::{
   4   context::LemmyContext,
   5   private_message::{MarkPrivateMessageAsRead, PrivateMessageResponse},
   6   utils::get_local_user_view_from_jwt,
   7   websocket::UserOperation,
   8 };
   9 use lemmy_db_schema::{
  10   source::private_message::{PrivateMessage, PrivateMessageUpdateForm},
  11   traits::Crud,
  12 };
  13 use lemmy_utils::{error::LemmyError, ConnectionId};
  14
  15 #[async_trait::async_trait(?Send)]
  16 impl Perform for MarkPrivateMessageAsRead {
  17   type Response = PrivateMessageResponse;
  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<PrivateMessageResponse, LemmyError> {
  25     let data: &MarkPrivateMessageAsRead = self;
  26     let local_user_view =
  27       get_local_user_view_from_jwt(&data.auth, context.pool(), context.secret()).await?;
  28
  29     // Checking permissions
  30     let private_message_id = data.private_message_id;
  31     let orig_private_message = PrivateMessage::read(context.pool(), private_message_id).await?;
  32     if local_user_view.person.id != orig_private_message.recipient_id {
  33       return Err(LemmyError::from_message("couldnt_update_private_message"));
  34     }
  35
  36     // Doing the update
  37     let private_message_id = data.private_message_id;
  38     let read = data.read;
  39     PrivateMessage::update(
  40       context.pool(),
  41       private_message_id,
  42       &PrivateMessageUpdateForm::builder().read(Some(read)).build(),
  43     )
  44     .await
  45     .map_err(|e| LemmyError::from_error_message(e, "couldnt_update_private_message"))?;
  46
  47     // No need to send an apub update
  48     context
  49       .send_pm_ws_message(
  50         &UserOperation::MarkPrivateMessageAsRead,
  51         data.private_message_id,
  52         websocket_id,
  53       )
  54       .await
  55   }
  56 }