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, context)?;
  54     let id = generate_activity_id(
  55       UndoType::Undo,
  56       &context.settings().get_protocol_and_hostname(),
  57     )?;
  58     let undo = UndoDeletePrivateMessage {
  59       actor: ObjectId::new(actor.actor_id()),
  60       to: ObjectId::new(recipient.actor_id()),
  61       object,
  62       kind: UndoType::Undo,
  63       id: id.clone(),
  64       context: lemmy_context(),
  65       unparsed: Default::default(),
  66     };
  67     let inbox = vec![recipient.get_shared_inbox_or_inbox_url()];
  68     send_activity_new(context, &undo, &id, actor, inbox, true).await
  69   }
  70 }
  71
  72 #[async_trait::async_trait(?Send)]
  73 impl ActivityHandler for UndoDeletePrivateMessage {
  74   async fn verify(
  75     &self,
  76     context: &LemmyContext,
  77     request_counter: &mut i32,
  78   ) -> Result<(), LemmyError> {
  79     verify_activity(self, &context.settings())?;
  80     verify_person(&self.actor, context, request_counter).await?;
  81     verify_urls_match(self.actor(), self.object.actor())?;
  82     verify_domains_match(self.actor(), &self.object.object)?;
  83     self.object.verify(context, request_counter).await?;
  84     Ok(())
  85   }
  86
  87   async fn receive(
  88     self,
  89     context: &LemmyContext,
  90     _request_counter: &mut i32,
  91   ) -> Result<(), LemmyError> {
  92     let ap_id = self.object.object.clone();
  93     let private_message = blocking(context.pool(), move |conn| {
  94       PrivateMessage::read_from_apub_id(conn, &ap_id.into())
  95     })
  96     .await??;
  97
  98     let deleted_private_message = blocking(context.pool(), move |conn| {
  99       PrivateMessage::update_deleted(conn, private_message.id, false)
 100     })
 101     .await??;
 102
 103     send_pm_ws_message(
 104       deleted_private_message.id,
 105       UserOperationCrud::EditPrivateMessage,
 106       None,
 107       context,
 108     )
 109     .await?;
 110
 111     Ok(())
 112   }
 113 }