]> Untitled Git - lemmy.git/blob - crates/apub/src/activities/community/block_user.rs
Rewrite fetcher (#1792)
[lemmy.git] / crates / apub / src / activities / community / block_user.rs
1 use crate::{
2   activities::{
3     community::announce::AnnouncableActivities,
4     generate_activity_id,
5     verify_activity,
6     verify_mod_action,
7     verify_person_in_community,
8   },
9   activity_queue::send_to_community_new,
10   extensions::context::lemmy_context,
11   fetcher::object_id::ObjectId,
12   ActorType,
13 };
14 use activitystreams::{
15   activity::kind::BlockType,
16   base::AnyBase,
17   primitives::OneOrMany,
18   unparsed::Unparsed,
19 };
20 use lemmy_api_common::blocking;
21 use lemmy_apub_lib::{values::PublicUrl, ActivityFields, ActivityHandler};
22 use lemmy_db_queries::{Bannable, Followable};
23 use lemmy_db_schema::source::{
24   community::{
25     Community,
26     CommunityFollower,
27     CommunityFollowerForm,
28     CommunityPersonBan,
29     CommunityPersonBanForm,
30   },
31   person::Person,
32 };
33 use lemmy_utils::LemmyError;
34 use lemmy_websocket::LemmyContext;
35 use serde::{Deserialize, Serialize};
36 use url::Url;
37
38 #[derive(Clone, Debug, Deserialize, Serialize, ActivityFields)]
39 #[serde(rename_all = "camelCase")]
40 pub struct BlockUserFromCommunity {
41   actor: ObjectId<Person>,
42   to: [PublicUrl; 1],
43   pub(in crate::activities::community) object: ObjectId<Person>,
44   cc: [ObjectId<Community>; 1],
45   #[serde(rename = "type")]
46   kind: BlockType,
47   id: Url,
48   #[serde(rename = "@context")]
49   context: OneOrMany<AnyBase>,
50   #[serde(flatten)]
51   unparsed: Unparsed,
52 }
53
54 impl BlockUserFromCommunity {
55   pub(in crate::activities::community) fn new(
56     community: &Community,
57     target: &Person,
58     actor: &Person,
59   ) -> Result<BlockUserFromCommunity, LemmyError> {
60     Ok(BlockUserFromCommunity {
61       actor: ObjectId::new(actor.actor_id()),
62       to: [PublicUrl::Public],
63       object: ObjectId::new(target.actor_id()),
64       cc: [ObjectId::new(community.actor_id())],
65       kind: BlockType::Block,
66       id: generate_activity_id(BlockType::Block)?,
67       context: lemmy_context(),
68       unparsed: Default::default(),
69     })
70   }
71
72   pub async fn send(
73     community: &Community,
74     target: &Person,
75     actor: &Person,
76     context: &LemmyContext,
77   ) -> Result<(), LemmyError> {
78     let block = BlockUserFromCommunity::new(community, target, actor)?;
79     let block_id = block.id.clone();
80
81     let activity = AnnouncableActivities::BlockUserFromCommunity(block);
82     let inboxes = vec![target.get_shared_inbox_or_inbox_url()];
83     send_to_community_new(activity, &block_id, actor, community, inboxes, context).await
84   }
85 }
86
87 #[async_trait::async_trait(?Send)]
88 impl ActivityHandler for BlockUserFromCommunity {
89   async fn verify(
90     &self,
91     context: &LemmyContext,
92     request_counter: &mut i32,
93   ) -> Result<(), LemmyError> {
94     verify_activity(self)?;
95     verify_person_in_community(&self.actor, &self.cc[0], context, request_counter).await?;
96     verify_mod_action(&self.actor, self.cc[0].clone(), context).await?;
97     Ok(())
98   }
99
100   async fn receive(
101     self,
102     context: &LemmyContext,
103     request_counter: &mut i32,
104   ) -> Result<(), LemmyError> {
105     let community = self.cc[0].dereference(context, request_counter).await?;
106     let blocked_user = self.object.dereference(context, request_counter).await?;
107
108     let community_user_ban_form = CommunityPersonBanForm {
109       community_id: community.id,
110       person_id: blocked_user.id,
111     };
112
113     blocking(context.pool(), move |conn: &'_ _| {
114       CommunityPersonBan::ban(conn, &community_user_ban_form)
115     })
116     .await??;
117
118     // Also unsubscribe them from the community, if they are subscribed
119     let community_follower_form = CommunityFollowerForm {
120       community_id: community.id,
121       person_id: blocked_user.id,
122       pending: false,
123     };
124     blocking(context.pool(), move |conn: &'_ _| {
125       CommunityFollower::unfollow(conn, &community_follower_form)
126     })
127     .await?
128     .ok();
129
130     Ok(())
131   }
132 }