]> Untitled Git - lemmy.git/blob - crates/apub/src/activities/community/update.rs
Reduce stack memory usage in apub code
[lemmy.git] / crates / apub / src / activities / community / update.rs
1 use crate::{
2   activities::{
3     community::{announce::GetCommunity, send_to_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::update::UpdateCommunity,
13 };
14 use activitystreams::{activity::kind::UpdateType, public};
15 use lemmy_api_common::blocking;
16 use lemmy_apub_lib::{
17   data::Data,
18   object_id::ObjectId,
19   traits::{ActivityHandler, ActorType, ApubObject},
20 };
21 use lemmy_db_schema::{
22   source::community::{Community, CommunityForm},
23   traits::Crud,
24 };
25 use lemmy_utils::LemmyError;
26 use lemmy_websocket::{send::send_community_ws_message, LemmyContext, UserOperationCrud};
27
28 impl UpdateCommunity {
29   pub async fn send(
30     community: ApubCommunity,
31     actor: &ApubPerson,
32     context: &LemmyContext,
33   ) -> Result<(), LemmyError> {
34     let id = generate_activity_id(
35       UpdateType::Update,
36       &context.settings().get_protocol_and_hostname(),
37     )?;
38     let update = UpdateCommunity {
39       actor: ObjectId::new(actor.actor_id()),
40       to: vec![public()],
41       object: Box::new(community.clone().into_apub(context).await?),
42       cc: vec![community.actor_id()],
43       kind: UpdateType::Update,
44       id: id.clone(),
45       unparsed: Default::default(),
46     };
47
48     let activity = AnnouncableActivities::UpdateCommunity(update);
49     send_to_community(activity, &id, actor, &community, vec![], context).await
50   }
51 }
52
53 #[async_trait::async_trait(?Send)]
54 impl ActivityHandler for UpdateCommunity {
55   type DataType = LemmyContext;
56   async fn verify(
57     &self,
58     context: &Data<LemmyContext>,
59     request_counter: &mut i32,
60   ) -> Result<(), LemmyError> {
61     verify_is_public(&self.to)?;
62     verify_activity(&self.id, self.actor.inner(), &context.settings())?;
63     let community = self.get_community(context, request_counter).await?;
64     verify_person_in_community(&self.actor, &community, context, request_counter).await?;
65     verify_mod_action(&self.actor, &community, context, request_counter).await?;
66     Ok(())
67   }
68
69   async fn receive(
70     self,
71     context: &Data<LemmyContext>,
72     request_counter: &mut i32,
73   ) -> Result<(), LemmyError> {
74     let community = self.get_community(context, request_counter).await?;
75
76     let updated_community = self
77       .object
78       .into_form(&community.actor_id.clone().into(), &context.settings())
79       .await?;
80     let cf = CommunityForm {
81       name: updated_community.name,
82       title: updated_community.title,
83       description: updated_community.description,
84       nsfw: updated_community.nsfw,
85       // TODO: icon and banner would be hosted on the other instance, ideally we would copy it to ours
86       icon: updated_community.icon,
87       banner: updated_community.banner,
88       ..CommunityForm::default()
89     };
90     let updated_community = blocking(context.pool(), move |conn| {
91       Community::update(conn, community.id, &cf)
92     })
93     .await??;
94
95     send_community_ws_message(
96       updated_community.id,
97       UserOperationCrud::EditCommunity,
98       None,
99       None,
100       context,
101     )
102     .await?;
103     Ok(())
104   }
105 }
106
107 #[async_trait::async_trait(?Send)]
108 impl GetCommunity for UpdateCommunity {
109   async fn get_community(
110     &self,
111     context: &LemmyContext,
112     request_counter: &mut i32,
113   ) -> Result<ApubCommunity, LemmyError> {
114     let cid = ObjectId::new(self.object.id.clone());
115     cid.dereference(context, request_counter).await
116   }
117 }