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::{blocking, get_local_user_view_from_jwt},
   6 };
   7 use lemmy_apub::protocol::activities::{
   8   create_or_update::private_message::CreateOrUpdatePrivateMessage,
   9   CreateOrUpdateType,
  10 };
  11 use lemmy_db_schema::{source::private_message::PrivateMessage, traits::Crud};
  12 use lemmy_utils::{error::LemmyError, utils::remove_slurs, ConnectionId};
  13 use lemmy_websocket::{send::send_pm_ws_message, LemmyContext, UserOperationCrud};
  14
  15 #[async_trait::async_trait(?Send)]
  16 impl PerformCrud for EditPrivateMessage {
  17   type Response = PrivateMessageResponse;
  18
  19   #[tracing::instrument(skip(self, context, websocket_id))]
  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(LemmyError::from_message("no_private_message_edit_allowed"));
  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| LemmyError::from_error_message(e, "couldnt_update_private_message"))?;
  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 }