crates/api/src/private_message/mark_read.rs

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