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::protocol::activities::{
   9   create_or_update::private_message::CreateOrUpdatePrivateMessage,
  10   CreateOrUpdateType,
  11 };
  12 use lemmy_db_schema::{source::private_message::PrivateMessage, traits::Crud};
  13 use lemmy_utils::{utils::remove_slurs, ConnectionId, LemmyError};
  14 use lemmy_websocket::{send::send_pm_ws_message, LemmyContext, UserOperationCrud};
  15
  16 #[async_trait::async_trait(?Send)]
  17 impl PerformCrud for EditPrivateMessage {
  18   type Response = PrivateMessageResponse;
  19
  20   #[tracing::instrument(skip(self, context, websocket_id))]
  21   async fn perform(
  22     &self,
  23     context: &Data<LemmyContext>,
  24     websocket_id: Option<ConnectionId>,
  25   ) -> Result<PrivateMessageResponse, LemmyError> {
  26     let data: &EditPrivateMessage = self;
  27     let local_user_view =
  28       get_local_user_view_from_jwt(&data.auth, context.pool(), context.secret()).await?;
  29
  30     // Checking permissions
  31     let private_message_id = data.private_message_id;
  32     let orig_private_message = blocking(context.pool(), move |conn| {
  33       PrivateMessage::read(conn, private_message_id)
  34     })
  35     .await??;
  36     if local_user_view.person.id != orig_private_message.creator_id {
  37       return Err(LemmyError::from_message("no_private_message_edit_allowed"));
  38     }
  39
  40     // Doing the update
  41     let content_slurs_removed = remove_slurs(&data.content, &context.settings().slur_regex());
  42     let private_message_id = data.private_message_id;
  43     let updated_private_message = blocking(context.pool(), move |conn| {
  44       PrivateMessage::update_content(conn, private_message_id, &content_slurs_removed)
  45     })
  46     .await?
  47     .map_err(|e| LemmyError::from_error_message(e, "couldnt_update_private_message"))?;
  48
  49     // Send the apub update
  50     CreateOrUpdatePrivateMessage::send(
  51       updated_private_message.into(),
  52       &local_user_view.person.into(),
  53       CreateOrUpdateType::Update,
  54       context,
  55     )
  56     .await?;
  57
  58     let op = UserOperationCrud::EditPrivateMessage;
  59     send_pm_ws_message(data.private_message_id, op, websocket_id, context).await
  60   }
  61 }