crates/apub/src/activities/community/add_mod.rs

   1 use crate::{
   2   activities::{
   3     community::{
   4       announce::GetCommunity,
   5       get_community_from_moderators_url,
   6       send_activity_in_community,
   7     },
   8     generate_activity_id,
   9     verify_activity,
  10     verify_add_remove_moderator_target,
  11     verify_is_public,
  12     verify_mod_action,
  13     verify_person_in_community,
  14   },
  15   activity_lists::AnnouncableActivities,
  16   generate_moderators_url,
  17   objects::{community::ApubCommunity, person::ApubPerson},
  18   protocol::activities::community::add_mod::AddMod,
  19 };
  20 use activitystreams_kinds::{activity::AddType, public};
  21 use lemmy_api_common::blocking;
  22 use lemmy_apub_lib::{
  23   data::Data,
  24   object_id::ObjectId,
  25   traits::{ActivityHandler, ActorType},
  26 };
  27 use lemmy_db_schema::{
  28   source::community::{CommunityModerator, CommunityModeratorForm},
  29   traits::Joinable,
  30 };
  31 use lemmy_utils::LemmyError;
  32 use lemmy_websocket::LemmyContext;
  33
  34 impl AddMod {
  35   #[tracing::instrument(skip_all)]
  36   pub async fn send(
  37     community: &ApubCommunity,
  38     added_mod: &ApubPerson,
  39     actor: &ApubPerson,
  40     context: &LemmyContext,
  41   ) -> Result<(), LemmyError> {
  42     let id = generate_activity_id(
  43       AddType::Add,
  44       &context.settings().get_protocol_and_hostname(),
  45     )?;
  46     let add = AddMod {
  47       actor: ObjectId::new(actor.actor_id()),
  48       to: vec![public()],
  49       object: ObjectId::new(added_mod.actor_id()),
  50       target: generate_moderators_url(&community.actor_id)?.into(),
  51       cc: vec![community.actor_id()],
  52       kind: AddType::Add,
  53       id: id.clone(),
  54       unparsed: Default::default(),
  55     };
  56
  57     let activity = AnnouncableActivities::AddMod(add);
  58     let inboxes = vec![added_mod.shared_inbox_or_inbox_url()];
  59     send_activity_in_community(activity, &id, actor, community, inboxes, context).await
  60   }
  61 }
  62
  63 #[async_trait::async_trait(?Send)]
  64 impl ActivityHandler for AddMod {
  65   type DataType = LemmyContext;
  66
  67   #[tracing::instrument(skip_all)]
  68   async fn verify(
  69     &self,
  70     context: &Data<LemmyContext>,
  71     request_counter: &mut i32,
  72   ) -> Result<(), LemmyError> {
  73     verify_is_public(&self.to, &self.cc)?;
  74     verify_activity(&self.id, self.actor.inner(), &context.settings())?;
  75     let community = self.get_community(context, request_counter).await?;
  76     verify_person_in_community(&self.actor, &community, context, request_counter).await?;
  77     verify_mod_action(&self.actor, &community, context, request_counter).await?;
  78     verify_add_remove_moderator_target(&self.target, &community)?;
  79     Ok(())
  80   }
  81
  82   #[tracing::instrument(skip_all)]
  83   async fn receive(
  84     self,
  85     context: &Data<LemmyContext>,
  86     request_counter: &mut i32,
  87   ) -> Result<(), LemmyError> {
  88     let community = self.get_community(context, request_counter).await?;
  89     let new_mod = self
  90       .object
  91       .dereference(context, context.client(), request_counter)
  92       .await?;
  93
  94     // If we had to refetch the community while parsing the activity, then the new mod has already
  95     // been added. Skip it here as it would result in a duplicate key error.
  96     let new_mod_id = new_mod.id;
  97     let moderated_communities = blocking(context.pool(), move |conn| {
  98       CommunityModerator::get_person_moderated_communities(conn, new_mod_id)
  99     })
 100     .await??;
 101     if !moderated_communities.contains(&community.id) {
 102       let form = CommunityModeratorForm {
 103         community_id: community.id,
 104         person_id: new_mod.id,
 105       };
 106       blocking(context.pool(), move |conn| {
 107         CommunityModerator::join(conn, &form)
 108       })
 109       .await??;
 110     }
 111     // TODO: send websocket notification about added mod
 112     Ok(())
 113   }
 114 }
 115
 116 #[async_trait::async_trait(?Send)]
 117 impl GetCommunity for AddMod {
 118   #[tracing::instrument(skip_all)]
 119   async fn get_community(
 120     &self,
 121     context: &LemmyContext,
 122     request_counter: &mut i32,
 123   ) -> Result<ApubCommunity, LemmyError> {
 124     get_community_from_moderators_url(&self.target, context, request_counter).await
 125   }
 126 }