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

   1 use crate::{
   2   activities::{
   3     deletion::{receive_delete_action, verify_delete_activity, DeletableObjects},
   4     generate_activity_id,
   5   },
   6   local_instance,
   7   objects::{community::ApubCommunity, person::ApubPerson},
   8   protocol::activities::deletion::{delete::Delete, undo_delete::UndoDelete},
   9 };
  10 use activitypub_federation::{core::object_id::ObjectId, data::Data, traits::ActivityHandler};
  11 use activitystreams_kinds::activity::UndoType;
  12 use lemmy_db_schema::{
  13   source::{
  14     comment::{Comment, CommentUpdateForm},
  15     community::{Community, CommunityUpdateForm},
  16     moderator::{
  17       ModRemoveComment,
  18       ModRemoveCommentForm,
  19       ModRemoveCommunity,
  20       ModRemoveCommunityForm,
  21       ModRemovePost,
  22       ModRemovePostForm,
  23     },
  24     post::{Post, PostUpdateForm},
  25   },
  26   traits::Crud,
  27 };
  28 use lemmy_utils::error::LemmyError;
  29 use lemmy_websocket::{
  30   send::{send_comment_ws_message_simple, send_community_ws_message, send_post_ws_message},
  31   LemmyContext,
  32   UserOperationCrud,
  33 };
  34 use url::Url;
  35
  36 #[async_trait::async_trait(?Send)]
  37 impl ActivityHandler for UndoDelete {
  38   type DataType = LemmyContext;
  39   type Error = LemmyError;
  40
  41   fn id(&self) -> &Url {
  42     &self.id
  43   }
  44
  45   fn actor(&self) -> &Url {
  46     self.actor.inner()
  47   }
  48
  49   #[tracing::instrument(skip_all)]
  50   async fn verify(
  51     &self,
  52     context: &Data<LemmyContext>,
  53     request_counter: &mut i32,
  54   ) -> Result<(), LemmyError> {
  55     self.object.verify(context, request_counter).await?;
  56     verify_delete_activity(
  57       &self.object,
  58       self.object.summary.is_some(),
  59       context,
  60       request_counter,
  61     )
  62     .await?;
  63     Ok(())
  64   }
  65
  66   #[tracing::instrument(skip_all)]
  67   async fn receive(
  68     self,
  69     context: &Data<LemmyContext>,
  70     request_counter: &mut i32,
  71   ) -> Result<(), LemmyError> {
  72     if self.object.summary.is_some() {
  73       UndoDelete::receive_undo_remove_action(
  74         &self
  75           .actor
  76           .dereference(context, local_instance(context).await, request_counter)
  77           .await?,
  78         self.object.object.id(),
  79         context,
  80       )
  81       .await
  82     } else {
  83       receive_delete_action(
  84         self.object.object.id(),
  85         &self.actor,
  86         false,
  87         context,
  88         request_counter,
  89       )
  90       .await
  91     }
  92   }
  93 }
  94
  95 impl UndoDelete {
  96   #[tracing::instrument(skip_all)]
  97   pub(in crate::activities::deletion) fn new(
  98     actor: &ApubPerson,
  99     object: DeletableObjects,
 100     to: Url,
 101     community: Option<&Community>,
 102     summary: Option<String>,
 103     context: &LemmyContext,
 104   ) -> Result<UndoDelete, LemmyError> {
 105     let object = Delete::new(actor, object, to.clone(), community, summary, context)?;
 106
 107     let id = generate_activity_id(
 108       UndoType::Undo,
 109       &context.settings().get_protocol_and_hostname(),
 110     )?;
 111     let cc: Option<Url> = community.map(|c| c.actor_id.clone().into());
 112     Ok(UndoDelete {
 113       actor: ObjectId::new(actor.actor_id.clone()),
 114       to: vec![to],
 115       object,
 116       cc: cc.into_iter().collect(),
 117       kind: UndoType::Undo,
 118       id,
 119       audience: community.map(|c| ObjectId::<ApubCommunity>::new(c.actor_id.clone())),
 120     })
 121   }
 122
 123   #[tracing::instrument(skip_all)]
 124   pub(in crate::activities) async fn receive_undo_remove_action(
 125     actor: &ApubPerson,
 126     object: &Url,
 127     context: &LemmyContext,
 128   ) -> Result<(), LemmyError> {
 129     use UserOperationCrud::*;
 130     match DeletableObjects::read_from_db(object, context).await? {
 131       DeletableObjects::Community(community) => {
 132         if community.local {
 133           return Err(LemmyError::from_message(
 134             "Only local admin can restore community",
 135           ));
 136         }
 137         let form = ModRemoveCommunityForm {
 138           mod_person_id: actor.id,
 139           community_id: community.id,
 140           removed: Some(false),
 141           reason: None,
 142           expires: None,
 143         };
 144         ModRemoveCommunity::create(context.pool(), &form).await?;
 145         let deleted_community = Community::update(
 146           context.pool(),
 147           community.id,
 148           &CommunityUpdateForm::builder().removed(Some(false)).build(),
 149         )
 150         .await?;
 151         send_community_ws_message(deleted_community.id, EditCommunity, None, None, context).await?;
 152       }
 153       DeletableObjects::Post(post) => {
 154         let form = ModRemovePostForm {
 155           mod_person_id: actor.id,
 156           post_id: post.id,
 157           removed: Some(false),
 158           reason: None,
 159         };
 160         ModRemovePost::create(context.pool(), &form).await?;
 161         let removed_post = Post::update(
 162           context.pool(),
 163           post.id,
 164           &PostUpdateForm::builder().removed(Some(false)).build(),
 165         )
 166         .await?;
 167         send_post_ws_message(removed_post.id, EditPost, None, None, context).await?;
 168       }
 169       DeletableObjects::Comment(comment) => {
 170         let form = ModRemoveCommentForm {
 171           mod_person_id: actor.id,
 172           comment_id: comment.id,
 173           removed: Some(false),
 174           reason: None,
 175         };
 176         ModRemoveComment::create(context.pool(), &form).await?;
 177         let removed_comment = Comment::update(
 178           context.pool(),
 179           comment.id,
 180           &CommentUpdateForm::builder().removed(Some(false)).build(),
 181         )
 182         .await?;
 183         send_comment_ws_message_simple(removed_comment.id, EditComment, context).await?;
 184       }
 185       DeletableObjects::PrivateMessage(_) => unimplemented!(),
 186     }
 187     Ok(())
 188   }
 189 }