crates/api_crud/src/private_message/update.rs

   1 use crate::PerformCrud;
   2 use actix_web::web::Data;
   3 use lemmy_api_common::{
   4   blocking,
   5   get_local_user_view_from_jwt,
   6   person::{EditPrivateMessage, PrivateMessageResponse},
   7 };
   8 use lemmy_apub::ApubObjectType;
   9 use lemmy_db_queries::{source::private_message::PrivateMessage_, Crud};
  10 use lemmy_db_schema::source::private_message::PrivateMessage;
  11 use lemmy_db_views::{local_user_view::LocalUserView, private_message_view::PrivateMessageView};
  12 use lemmy_utils::{utils::remove_slurs, ApiError, ConnectionId, LemmyError};
  13 use lemmy_websocket::{messages::SendUserRoomMessage, LemmyContext, UserOperationCrud};
  14
  15 #[async_trait::async_trait(?Send)]
  16 impl PerformCrud for EditPrivateMessage {
  17   type Response = PrivateMessageResponse;
  18
  19   async fn perform(
  20     &self,
  21     context: &Data<LemmyContext>,
  22     websocket_id: Option<ConnectionId>,
  23   ) -> Result<PrivateMessageResponse, LemmyError> {
  24     let data: &EditPrivateMessage = &self;
  25     let local_user_view = get_local_user_view_from_jwt(&data.auth, context.pool()).await?;
  26
  27     // Checking permissions
  28     let private_message_id = data.private_message_id;
  29     let orig_private_message = blocking(context.pool(), move |conn| {
  30       PrivateMessage::read(conn, private_message_id)
  31     })
  32     .await??;
  33     if local_user_view.person.id != orig_private_message.creator_id {
  34       return Err(ApiError::err("no_private_message_edit_allowed").into());
  35     }
  36
  37     // Doing the update
  38     let content_slurs_removed = remove_slurs(&data.content);
  39     let private_message_id = data.private_message_id;
  40     let updated_private_message = blocking(context.pool(), move |conn| {
  41       PrivateMessage::update_content(conn, private_message_id, &content_slurs_removed)
  42     })
  43     .await?
  44     .map_err(|_| ApiError::err("couldnt_update_private_message"))?;
  45
  46     // Send the apub update
  47     updated_private_message
  48       .send_update(&local_user_view.person, context)
  49       .await?;
  50
  51     let private_message_id = data.private_message_id;
  52     let private_message_view = blocking(context.pool(), move |conn| {
  53       PrivateMessageView::read(conn, private_message_id)
  54     })
  55     .await??;
  56
  57     let res = PrivateMessageResponse {
  58       private_message_view,
  59     };
  60
  61     // Send notifications to the local recipient, if one exists
  62     let recipient_id = orig_private_message.recipient_id;
  63     if let Ok(local_recipient) = blocking(context.pool(), move |conn| {
  64       LocalUserView::read_person(conn, recipient_id)
  65     })
  66     .await?
  67     {
  68       let local_recipient_id = local_recipient.local_user.id;
  69       context.chat_server().do_send(SendUserRoomMessage {
  70         op: UserOperationCrud::EditPrivateMessage,
  71         response: res.clone(),
  72         local_recipient_id,
  73         websocket_id,
  74       });
  75     }
  76
  77     Ok(res)
  78   }
  79 }