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