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