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_queries::{source::private_message::PrivateMessage_, Crud};
   9 use lemmy_db_schema::source::private_message::PrivateMessage;
  10 use lemmy_utils::{ApiError, ConnectionId, LemmyError};
  11 use lemmy_websocket::{send::send_pm_ws_message, LemmyContext, UserOperation};
  12
  13 #[async_trait::async_trait(?Send)]
  14 impl Perform for MarkPrivateMessageAsRead {
  15   type Response = PrivateMessageResponse;
  16
  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(ApiError::err("couldnt_update_private_message").into());
  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(|_| ApiError::err("couldnt_update_private_message"))?;
  44
  45     // No need to send an apub update
  46     let op = UserOperation::MarkPrivateMessageAsRead;
  47     send_pm_ws_message(data.private_message_id, op, websocket_id, context).await
  48   }
  49 }