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

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