]> Untitled Git - lemmy.git/blob - crates/apub/src/activities/community/block_user.rs
Rewrite remaining activities (#1712)
[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::{community::get_or_fetch_and_upsert_community, person::get_or_fetch_and_upsert_person},
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: Url,
42   to: PublicUrl,
43   pub(in crate::activities::community) object: Url,
44   cc: [Url; 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: actor.actor_id(),
62       to: PublicUrl::Public,
63       object: target.actor_id(),
64       cc: [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 =
106       get_or_fetch_and_upsert_community(&self.cc[0], context, request_counter).await?;
107     let blocked_user =
108       get_or_fetch_and_upsert_person(&self.object, context, request_counter).await?;
109
110     let community_user_ban_form = CommunityPersonBanForm {
111       community_id: community.id,
112       person_id: blocked_user.id,
113     };
114
115     blocking(context.pool(), move |conn: &'_ _| {
116       CommunityPersonBan::ban(conn, &community_user_ban_form)
117     })
118     .await??;
119
120     // Also unsubscribe them from the community, if they are subscribed
121     let community_follower_form = CommunityFollowerForm {
122       community_id: community.id,
123       person_id: blocked_user.id,
124       pending: false,
125     };
126     blocking(context.pool(), move |conn: &'_ _| {
127       CommunityFollower::unfollow(conn, &community_follower_form)
128     })
129     .await?
130     .ok();
131
132     Ok(())
133   }
134 }