]> Untitled Git - lemmy.git/blob - crates/apub/src/activities/community/block_user.rs
Don't drop error context when adding a message to errors (#1958)
[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   #[tracing::instrument(skip_all)]
56   pub async fn send(
57     community: &ApubCommunity,
58     target: &ApubPerson,
59     actor: &ApubPerson,
60     context: &LemmyContext,
61   ) -> Result<(), LemmyError> {
62     let block = BlockUserFromCommunity::new(community, target, actor, context)?;
63     let block_id = block.id.clone();
64
65     let activity = AnnouncableActivities::BlockUserFromCommunity(block);
66     let inboxes = vec![target.shared_inbox_or_inbox_url()];
67     send_activity_in_community(activity, &block_id, actor, community, inboxes, context).await
68   }
69 }
70
71 #[async_trait::async_trait(?Send)]
72 impl ActivityHandler for BlockUserFromCommunity {
73   type DataType = LemmyContext;
74
75   #[tracing::instrument(skip_all)]
76   async fn verify(
77     &self,
78     context: &Data<LemmyContext>,
79     request_counter: &mut i32,
80   ) -> Result<(), LemmyError> {
81     verify_is_public(&self.to, &self.cc)?;
82     verify_activity(&self.id, self.actor.inner(), &context.settings())?;
83     let community = self.get_community(context, request_counter).await?;
84     verify_person_in_community(&self.actor, &community, context, request_counter).await?;
85     verify_mod_action(&self.actor, &community, context, request_counter).await?;
86     Ok(())
87   }
88
89   #[tracing::instrument(skip_all)]
90   async fn receive(
91     self,
92     context: &Data<LemmyContext>,
93     request_counter: &mut i32,
94   ) -> Result<(), LemmyError> {
95     let community = self.get_community(context, request_counter).await?;
96     let blocked_user = self.object.dereference(context, request_counter).await?;
97
98     let community_user_ban_form = CommunityPersonBanForm {
99       community_id: community.id,
100       person_id: blocked_user.id,
101     };
102
103     blocking(context.pool(), move |conn: &'_ _| {
104       CommunityPersonBan::ban(conn, &community_user_ban_form)
105     })
106     .await??;
107
108     // Also unsubscribe them from the community, if they are subscribed
109     let community_follower_form = CommunityFollowerForm {
110       community_id: community.id,
111       person_id: blocked_user.id,
112       pending: false,
113     };
114     blocking(context.pool(), move |conn: &'_ _| {
115       CommunityFollower::unfollow(conn, &community_follower_form)
116     })
117     .await?
118     .ok();
119
120     Ok(())
121   }
122 }
123
124 #[async_trait::async_trait(?Send)]
125 impl GetCommunity for BlockUserFromCommunity {
126   #[tracing::instrument(skip_all)]
127   async fn get_community(
128     &self,
129     context: &LemmyContext,
130     request_counter: &mut i32,
131   ) -> Result<ApubCommunity, LemmyError> {
132     self.target.dereference(context, request_counter).await
133   }
134 }