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_db_views::{local_user_view::LocalUserView, private_message_view::PrivateMessageView};
  11 use lemmy_utils::{ApiError, ConnectionId, LemmyError};
  12 use lemmy_websocket::{messages::SendUserRoomMessage, LemmyContext, UserOperation};
  13
  14 #[async_trait::async_trait(?Send)]
  15 impl Perform for MarkPrivateMessageAsRead {
  16   type Response = PrivateMessageResponse;
  17
  18   async fn perform(
  19     &self,
  20     context: &Data<LemmyContext>,
  21     websocket_id: Option<ConnectionId>,
  22   ) -> Result<PrivateMessageResponse, LemmyError> {
  23     let data: &MarkPrivateMessageAsRead = &self;
  24     let local_user_view = get_local_user_view_from_jwt(&data.auth, context.pool()).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 private_message_id = data.private_message_id;
  47     let private_message_view = blocking(context.pool(), move |conn| {
  48       PrivateMessageView::read(conn, private_message_id)
  49     })
  50     .await??;
  51
  52     let res = PrivateMessageResponse {
  53       private_message_view,
  54     };
  55
  56     // Send notifications to the local recipient, if one exists
  57     let recipient_id = orig_private_message.recipient_id;
  58     if let Ok(local_recipient) = blocking(context.pool(), move |conn| {
  59       LocalUserView::read_person(conn, recipient_id)
  60     })
  61     .await?
  62     {
  63       let local_recipient_id = local_recipient.local_user.id;
  64       context.chat_server().do_send(SendUserRoomMessage {
  65         op: UserOperation::MarkPrivateMessageAsRead,
  66         response: res.clone(),
  67         local_recipient_id,
  68         websocket_id,
  69       });
  70     }
  71
  72     Ok(res)
  73   }
  74 }