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