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 = get_local_user_view_from_jwt(&data.auth, context.pool()).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("no_private_message_edit_allowed").into());
  37     }
  38
  39     // Doing the update
  40     let content_slurs_removed = remove_slurs(&data.content);
  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(|_| ApiError::err("couldnt_update_private_message"))?;
  47
  48     // Send the apub update
  49     CreateOrUpdatePrivateMessage::send(
  50       &updated_private_message,
  51       &local_user_view.person,
  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 }