]> Untitled Git - lemmy.git/blob - crates/apub/src/activities/comment/undo_remove.rs
Merge pull request #1678 from LemmyNet/rewrite-post
[lemmy.git] / crates / apub / src / activities / comment / undo_remove.rs
1 use crate::{
2   activities::{
3     comment::{remove::RemoveComment, send_websocket_message},
4     verify_mod_action,
5   },
6   check_is_apub_id_valid,
7   fetcher::objects::get_or_fetch_and_insert_comment,
8 };
9 use activitystreams::activity::kind::UndoType;
10 use lemmy_api_common::blocking;
11 use lemmy_apub_lib::{
12   values::PublicUrl,
13   verify_domains_match,
14   ActivityCommonFields,
15   ActivityHandlerNew,
16 };
17 use lemmy_db_queries::source::comment::Comment_;
18 use lemmy_db_schema::source::comment::Comment;
19 use lemmy_utils::LemmyError;
20 use lemmy_websocket::{LemmyContext, UserOperationCrud};
21 use url::Url;
22
23 #[derive(Clone, Debug, serde::Deserialize, serde::Serialize)]
24 #[serde(rename_all = "camelCase")]
25 pub struct UndoRemoveComment {
26   to: PublicUrl,
27   object: RemoveComment,
28   cc: [Url; 1],
29   #[serde(rename = "type")]
30   kind: UndoType,
31   #[serde(flatten)]
32   common: ActivityCommonFields,
33 }
34
35 #[async_trait::async_trait(?Send)]
36 impl ActivityHandlerNew for UndoRemoveComment {
37   async fn verify(
38     &self,
39     context: &LemmyContext,
40     request_counter: &mut i32,
41   ) -> Result<(), LemmyError> {
42     verify_domains_match(&self.common.actor, self.common.id_unchecked())?;
43     check_is_apub_id_valid(&self.common.actor, false)?;
44     verify_mod_action(&self.common.actor, self.cc[0].clone(), context).await?;
45     self.object.verify(context, request_counter).await
46   }
47
48   async fn receive(
49     &self,
50     context: &LemmyContext,
51     request_counter: &mut i32,
52   ) -> Result<(), LemmyError> {
53     let comment =
54       get_or_fetch_and_insert_comment(&self.object.object, context, request_counter).await?;
55
56     let removed_comment = blocking(context.pool(), move |conn| {
57       Comment::update_removed(conn, comment.id, false)
58     })
59     .await??;
60
61     send_websocket_message(
62       removed_comment.id,
63       vec![],
64       UserOperationCrud::EditComment,
65       context,
66     )
67     .await
68   }
69
70   fn common(&self) -> &ActivityCommonFields {
71     &self.common
72   }
73 }