crates/api_crud/src/private_message/delete.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::{DeletePrivateMessage, PrivateMessageResponse},
   7 };
   8 use lemmy_apub::ApubObjectType;
   9 use lemmy_db_queries::{source::private_message::PrivateMessage_, Crud};
  10 use lemmy_db_schema::source::private_message::PrivateMessage;
  11 use lemmy_db_views::{local_user_view::LocalUserView, private_message_view::PrivateMessageView};
  12 use lemmy_utils::{ApiError, ConnectionId, LemmyError};
  13 use lemmy_websocket::{messages::SendUserRoomMessage, LemmyContext, UserOperationCrud};
  14
  15 #[async_trait::async_trait(?Send)]
  16 impl PerformCrud for DeletePrivateMessage {
  17   type Response = PrivateMessageResponse;
  18
  19   async fn perform(
  20     &self,
  21     context: &Data<LemmyContext>,
  22     websocket_id: Option<ConnectionId>,
  23   ) -> Result<PrivateMessageResponse, LemmyError> {
  24     let data: &DeletePrivateMessage = &self;
  25     let local_user_view = get_local_user_view_from_jwt(&data.auth, context.pool()).await?;
  26
  27     // Checking permissions
  28     let private_message_id = data.private_message_id;
  29     let orig_private_message = blocking(context.pool(), move |conn| {
  30       PrivateMessage::read(conn, private_message_id)
  31     })
  32     .await??;
  33     if local_user_view.person.id != orig_private_message.creator_id {
  34       return Err(ApiError::err("no_private_message_edit_allowed").into());
  35     }
  36
  37     // Doing the update
  38     let private_message_id = data.private_message_id;
  39     let deleted = data.deleted;
  40     let updated_private_message = blocking(context.pool(), move |conn| {
  41       PrivateMessage::update_deleted(conn, private_message_id, deleted)
  42     })
  43     .await?
  44     .map_err(|_| ApiError::err("couldnt_update_private_message"))?;
  45
  46     // Send the apub update
  47     if data.deleted {
  48       updated_private_message
  49         .send_delete(&local_user_view.person, context)
  50         .await?;
  51     } else {
  52       updated_private_message
  53         .send_undo_delete(&local_user_view.person, context)
  54         .await?;
  55     }
  56
  57     let private_message_id = data.private_message_id;
  58     let private_message_view = blocking(context.pool(), move |conn| {
  59       PrivateMessageView::read(conn, private_message_id)
  60     })
  61     .await??;
  62
  63     let res = PrivateMessageResponse {
  64       private_message_view,
  65     };
  66
  67     // Send notifications to the local recipient, if one exists
  68     let recipient_id = orig_private_message.recipient_id;
  69     if let Ok(local_recipient) = blocking(context.pool(), move |conn| {
  70       LocalUserView::read_person(conn, recipient_id)
  71     })
  72     .await?
  73     {
  74       let local_recipient_id = local_recipient.local_user.id;
  75       context.chat_server().do_send(SendUserRoomMessage {
  76         op: UserOperationCrud::DeletePrivateMessage,
  77         response: res.clone(),
  78         local_recipient_id,
  79         websocket_id,
  80       });
  81     }
  82
  83     Ok(res)
  84   }
  85 }