crates/apub/src/activities/private_message/undo_delete.rs

   1 use crate::{
   2   activities::{
   3     generate_activity_id,
   4     private_message::delete::DeletePrivateMessage,
   5     verify_activity,
   6     verify_person,
   7   },
   8   activity_queue::send_activity_new,
   9   extensions::context::lemmy_context,
  10   fetcher::object_id::ObjectId,
  11   ActorType,
  12 };
  13 use activitystreams::{
  14   activity::kind::UndoType,
  15   base::AnyBase,
  16   primitives::OneOrMany,
  17   unparsed::Unparsed,
  18 };
  19 use lemmy_api_common::blocking;
  20 use lemmy_apub_lib::{verify_domains_match, verify_urls_match, ActivityFields, ActivityHandler};
  21 use lemmy_db_queries::{source::private_message::PrivateMessage_, ApubObject, Crud};
  22 use lemmy_db_schema::source::{person::Person, private_message::PrivateMessage};
  23 use lemmy_utils::LemmyError;
  24 use lemmy_websocket::{send::send_pm_ws_message, LemmyContext, UserOperationCrud};
  25 use serde::{Deserialize, Serialize};
  26 use url::Url;
  27
  28 #[derive(Clone, Debug, Deserialize, Serialize, ActivityFields)]
  29 #[serde(rename_all = "camelCase")]
  30 pub struct UndoDeletePrivateMessage {
  31   actor: ObjectId<Person>,
  32   to: ObjectId<Person>,
  33   object: DeletePrivateMessage,
  34   #[serde(rename = "type")]
  35   kind: UndoType,
  36   id: Url,
  37   #[serde(rename = "@context")]
  38   context: OneOrMany<AnyBase>,
  39   #[serde(flatten)]
  40   unparsed: Unparsed,
  41 }
  42
  43 impl UndoDeletePrivateMessage {
  44   pub async fn send(
  45     actor: &Person,
  46     pm: &PrivateMessage,
  47     context: &LemmyContext,
  48   ) -> Result<(), LemmyError> {
  49     let recipient_id = pm.recipient_id;
  50     let recipient =
  51       blocking(context.pool(), move |conn| Person::read(conn, recipient_id)).await??;
  52
  53     let object = DeletePrivateMessage::new(actor, pm)?;
  54     let id = generate_activity_id(UndoType::Undo)?;
  55     let undo = UndoDeletePrivateMessage {
  56       actor: ObjectId::new(actor.actor_id()),
  57       to: ObjectId::new(recipient.actor_id()),
  58       object,
  59       kind: UndoType::Undo,
  60       id: id.clone(),
  61       context: lemmy_context(),
  62       unparsed: Default::default(),
  63     };
  64     let inbox = vec![recipient.get_shared_inbox_or_inbox_url()];
  65     send_activity_new(context, &undo, &id, actor, inbox, true).await
  66   }
  67 }
  68
  69 #[async_trait::async_trait(?Send)]
  70 impl ActivityHandler for UndoDeletePrivateMessage {
  71   async fn verify(
  72     &self,
  73     context: &LemmyContext,
  74     request_counter: &mut i32,
  75   ) -> Result<(), LemmyError> {
  76     verify_activity(self)?;
  77     verify_person(&self.actor, context, request_counter).await?;
  78     verify_urls_match(self.actor(), self.object.actor())?;
  79     verify_domains_match(self.actor(), &self.object.object)?;
  80     self.object.verify(context, request_counter).await?;
  81     Ok(())
  82   }
  83
  84   async fn receive(
  85     self,
  86     context: &LemmyContext,
  87     _request_counter: &mut i32,
  88   ) -> Result<(), LemmyError> {
  89     let ap_id = self.object.object.clone();
  90     let private_message = blocking(context.pool(), move |conn| {
  91       PrivateMessage::read_from_apub_id(conn, &ap_id.into())
  92     })
  93     .await??;
  94
  95     let deleted_private_message = blocking(context.pool(), move |conn| {
  96       PrivateMessage::update_deleted(conn, private_message.id, false)
  97     })
  98     .await??;
  99
 100     send_pm_ws_message(
 101       deleted_private_message.id,
 102       UserOperationCrud::EditPrivateMessage,
 103       None,
 104       context,
 105     )
 106     .await?;
 107
 108     Ok(())
 109   }
 110 }