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::activities::{
   9   private_message::create_or_update::CreateOrUpdatePrivateMessage,
  10   CreateOrUpdateType,
  11 };
  12 use lemmy_db_queries::{source::private_message::PrivateMessage_, Crud};
  13 use lemmy_db_schema::source::private_message::PrivateMessage;
  14 use lemmy_utils::{utils::remove_slurs, ApiError, ConnectionId, LemmyError};
  15 use lemmy_websocket::{send::send_pm_ws_message, LemmyContext, UserOperationCrud};
  16
  17 #[async_trait::async_trait(?Send)]
  18 impl PerformCrud for EditPrivateMessage {
  19   type Response = PrivateMessageResponse;
  20
  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(ApiError::err("no_private_message_edit_allowed").into());
  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(|_| ApiError::err("couldnt_update_private_message"))?;
  48
  49     // Send the apub update
  50     CreateOrUpdatePrivateMessage::send(
  51       &updated_private_message,
  52       &local_user_view.person,
  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 }