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