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, DeleteableOrRemoveable};
  13 use lemmy_db_schema::source::private_message::PrivateMessage;
  14 use lemmy_db_views::{local_user_view::LocalUserView, private_message_view::PrivateMessageView};
  15 use lemmy_utils::{utils::remove_slurs, ApiError, ConnectionId, LemmyError};
  16 use lemmy_websocket::{messages::SendUserRoomMessage, LemmyContext, UserOperationCrud};
  17
  18 #[async_trait::async_trait(?Send)]
  19 impl PerformCrud for EditPrivateMessage {
  20   type Response = PrivateMessageResponse;
  21
  22   async fn perform(
  23     &self,
  24     context: &Data<LemmyContext>,
  25     websocket_id: Option<ConnectionId>,
  26   ) -> Result<PrivateMessageResponse, LemmyError> {
  27     let data: &EditPrivateMessage = self;
  28     let local_user_view = get_local_user_view_from_jwt(&data.auth, context.pool()).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);
  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 private_message_id = data.private_message_id;
  59     let mut private_message_view = blocking(context.pool(), move |conn| {
  60       PrivateMessageView::read(conn, private_message_id)
  61     })
  62     .await??;
  63
  64     // Blank out deleted or removed info
  65     if private_message_view.private_message.deleted {
  66       private_message_view.private_message = private_message_view
  67         .private_message
  68         .blank_out_deleted_or_removed_info();
  69     }
  70
  71     let res = PrivateMessageResponse {
  72       private_message_view,
  73     };
  74
  75     // Send notifications to the local recipient, if one exists
  76     let recipient_id = orig_private_message.recipient_id;
  77     if let Ok(local_recipient) = blocking(context.pool(), move |conn| {
  78       LocalUserView::read_person(conn, recipient_id)
  79     })
  80     .await?
  81     {
  82       let local_recipient_id = local_recipient.local_user.id;
  83       context.chat_server().do_send(SendUserRoomMessage {
  84         op: UserOperationCrud::EditPrivateMessage,
  85         response: res.clone(),
  86         local_recipient_id,
  87         websocket_id,
  88       });
  89     }
  90
  91     Ok(res)
  92   }
  93 }