crates/api_crud/src/private_message/update.rs

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