crates/api_crud/src/private_message/update.rs

   1 use crate::PerformCrud;
   2 use actix_web::web::Data;
   3 use lemmy_api_common::{
   4   private_message::{EditPrivateMessage, PrivateMessageResponse},
   5   utils::{get_local_user_view_from_jwt, local_site_to_slur_regex},
   6   websocket::{send::send_pm_ws_message, UserOperationCrud},
   7   LemmyContext,
   8 };
   9 use lemmy_apub::protocol::activities::{
  10   create_or_update::chat_message::CreateOrUpdateChatMessage,
  11   CreateOrUpdateType,
  12 };
  13 use lemmy_db_schema::{
  14   source::{
  15     local_site::LocalSite,
  16     private_message::{PrivateMessage, PrivateMessageUpdateForm},
  17   },
  18   traits::Crud,
  19   utils::naive_now,
  20 };
  21 use lemmy_utils::{error::LemmyError, utils::remove_slurs, ConnectionId};
  22
  23 #[async_trait::async_trait(?Send)]
  24 impl PerformCrud for EditPrivateMessage {
  25   type Response = PrivateMessageResponse;
  26
  27   #[tracing::instrument(skip(self, context, websocket_id))]
  28   async fn perform(
  29     &self,
  30     context: &Data<LemmyContext>,
  31     websocket_id: Option<ConnectionId>,
  32   ) -> Result<PrivateMessageResponse, LemmyError> {
  33     let data: &EditPrivateMessage = self;
  34     let local_user_view =
  35       get_local_user_view_from_jwt(&data.auth, context.pool(), context.secret()).await?;
  36     let local_site = LocalSite::read(context.pool()).await?;
  37
  38     // Checking permissions
  39     let private_message_id = data.private_message_id;
  40     let orig_private_message = PrivateMessage::read(context.pool(), private_message_id).await?;
  41     if local_user_view.person.id != orig_private_message.creator_id {
  42       return Err(LemmyError::from_message("no_private_message_edit_allowed"));
  43     }
  44
  45     // Doing the update
  46     let content_slurs_removed = remove_slurs(&data.content, &local_site_to_slur_regex(&local_site));
  47     let private_message_id = data.private_message_id;
  48     let updated_private_message = PrivateMessage::update(
  49       context.pool(),
  50       private_message_id,
  51       &PrivateMessageUpdateForm::builder()
  52         .content(Some(content_slurs_removed))
  53         .updated(Some(Some(naive_now())))
  54         .build(),
  55     )
  56     .await
  57     .map_err(|e| LemmyError::from_error_message(e, "couldnt_update_private_message"))?;
  58
  59     // Send the apub update
  60     CreateOrUpdateChatMessage::send(
  61       updated_private_message.into(),
  62       &local_user_view.person.into(),
  63       CreateOrUpdateType::Update,
  64       context,
  65     )
  66     .await?;
  67
  68     let op = UserOperationCrud::EditPrivateMessage;
  69     send_pm_ws_message(data.private_message_id, op, websocket_id, context).await
  70   }
  71 }