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