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, UserOperation};
  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 = match blocking(context.pool(), move |conn| {
  41       PrivateMessage::update_content(conn, private_message_id, &content_slurs_removed)
  42     })
  43     .await?
  44     {
  45       Ok(private_message) => private_message,
  46       Err(_e) => return Err(ApiError::err("couldnt_update_private_message").into()),
  47     };
  48
  49     // Send the apub update
  50     updated_private_message
  51       .send_update(&local_user_view.person, context)
  52       .await?;
  53
  54     let private_message_id = data.private_message_id;
  55     let private_message_view = blocking(context.pool(), move |conn| {
  56       PrivateMessageView::read(conn, private_message_id)
  57     })
  58     .await??;
  59
  60     let res = PrivateMessageResponse {
  61       private_message_view,
  62     };
  63
  64     // Send notifications to the local recipient, if one exists
  65     let recipient_id = orig_private_message.recipient_id;
  66     if let Ok(local_recipient) = blocking(context.pool(), move |conn| {
  67       LocalUserView::read_person(conn, recipient_id)
  68     })
  69     .await?
  70     {
  71       let local_recipient_id = local_recipient.local_user.id;
  72       context.chat_server().do_send(SendUserRoomMessage {
  73         op: UserOperation::EditPrivateMessage,
  74         response: res.clone(),
  75         local_recipient_id,
  76         websocket_id,
  77       });
  78     }
  79
  80     Ok(res)
  81   }
  82 }