crates/api/src/private_message.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::{MarkPrivateMessageAsRead, PrivateMessageResponse},
   7 };
   8 use lemmy_db_schema::{source::private_message::PrivateMessage, traits::Crud};
   9 use lemmy_utils::{ConnectionId, LemmyError};
  10 use lemmy_websocket::{send::send_pm_ws_message, LemmyContext, UserOperation};
  11
  12 #[async_trait::async_trait(?Send)]
  13 impl Perform for MarkPrivateMessageAsRead {
  14   type Response = PrivateMessageResponse;
  15
  16   #[tracing::instrument(skip(context, websocket_id))]
  17   async fn perform(
  18     &self,
  19     context: &Data<LemmyContext>,
  20     websocket_id: Option<ConnectionId>,
  21   ) -> Result<PrivateMessageResponse, LemmyError> {
  22     let data: &MarkPrivateMessageAsRead = self;
  23     let local_user_view =
  24       get_local_user_view_from_jwt(&data.auth, context.pool(), context.secret()).await?;
  25
  26     // Checking permissions
  27     let private_message_id = data.private_message_id;
  28     let orig_private_message = blocking(context.pool(), move |conn| {
  29       PrivateMessage::read(conn, private_message_id)
  30     })
  31     .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     blocking(context.pool(), move |conn| {
  40       PrivateMessage::update_read(conn, private_message_id, read)
  41     })
  42     .await?
  43     .map_err(LemmyError::from)
  44     .map_err(|e| e.with_message("couldnt_update_private_message"))?;
  45
  46     // No need to send an apub update
  47     let op = UserOperation::MarkPrivateMessageAsRead;
  48     send_pm_ws_message(data.private_message_id, op, websocket_id, context).await
  49   }
  50 }