crates/api_crud/src/private_message/update.rs

   1 use activitypub_federation::config::Data;
   2 use actix_web::web::Json;
   3 use lemmy_api_common::{
   4   context::LemmyContext,
   5   private_message::{EditPrivateMessage, PrivateMessageResponse},
   6   send_activity::{ActivityChannel, SendActivityData},
   7   utils::{local_site_to_slur_regex, local_user_view_from_jwt, sanitize_html},
   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_db_views::structs::PrivateMessageView;
  18 use lemmy_utils::{
  19   error::{LemmyError, LemmyErrorExt, LemmyErrorType},
  20   utils::{slurs::remove_slurs, validation::is_valid_body_field},
  21 };
  22
  23 #[tracing::instrument(skip(context))]
  24 pub async fn update_private_message(
  25   data: Json<EditPrivateMessage>,
  26   context: Data<LemmyContext>,
  27 ) -> Result<Json<PrivateMessageResponse>, LemmyError> {
  28   let local_user_view = local_user_view_from_jwt(&data.auth, &context).await?;
  29   let local_site = LocalSite::read(&mut context.pool()).await?;
  30
  31   // Checking permissions
  32   let private_message_id = data.private_message_id;
  33   let orig_private_message = PrivateMessage::read(&mut context.pool(), private_message_id).await?;
  34   if local_user_view.person.id != orig_private_message.creator_id {
  35     return Err(LemmyErrorType::EditPrivateMessageNotAllowed)?;
  36   }
  37
  38   // Doing the update
  39   let content = sanitize_html(&data.content);
  40   let content = remove_slurs(&content, &local_site_to_slur_regex(&local_site));
  41   is_valid_body_field(&Some(content.clone()), false)?;
  42
  43   let private_message_id = data.private_message_id;
  44   PrivateMessage::update(
  45     &mut context.pool(),
  46     private_message_id,
  47     &PrivateMessageUpdateForm {
  48       content: Some(content),
  49       updated: Some(Some(naive_now())),
  50       ..Default::default()
  51     },
  52   )
  53   .await
  54   .with_lemmy_type(LemmyErrorType::CouldntUpdatePrivateMessage)?;
  55
  56   let view = PrivateMessageView::read(&mut context.pool(), private_message_id).await?;
  57
  58   ActivityChannel::submit_activity(
  59     SendActivityData::UpdatePrivateMessage(view.clone()),
  60     &context,
  61   )
  62   .await?;
  63
  64   Ok(Json(PrivateMessageResponse {
  65     private_message_view: view,
  66   }))
  67 }