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_add_remove_moderator_target,
  10     verify_is_public,
  11     verify_mod_action,
  12     verify_person_in_community,
  13   },
  14   activity_lists::AnnouncableActivities,
  15   generate_moderators_url,
  16   local_instance,
  17   objects::{community::ApubCommunity, person::ApubPerson},
  18   protocol::activities::community::add_mod::AddMod,
  19   ActorType,
  20 };
  21 use activitypub_federation::{
  22   core::object_id::ObjectId,
  23   data::Data,
  24   traits::{ActivityHandler, Actor},
  25 };
  26 use activitystreams_kinds::{activity::AddType, public};
  27 use lemmy_db_schema::{
  28   source::{
  29     community::{CommunityModerator, CommunityModeratorForm},
  30     moderator::{ModAddCommunity, ModAddCommunityForm},
  31   },
  32   traits::{Crud, Joinable},
  33 };
  34 use lemmy_utils::error::LemmyError;
  35 use lemmy_websocket::LemmyContext;
  36 use url::Url;
  37
  38 impl AddMod {
  39   #[tracing::instrument(skip_all)]
  40   pub async fn send(
  41     community: &ApubCommunity,
  42     added_mod: &ApubPerson,
  43     actor: &ApubPerson,
  44     context: &LemmyContext,
  45   ) -> Result<(), LemmyError> {
  46     let id = generate_activity_id(
  47       AddType::Add,
  48       &context.settings().get_protocol_and_hostname(),
  49     )?;
  50     let add = AddMod {
  51       actor: ObjectId::new(actor.actor_id()),
  52       to: vec![public()],
  53       object: ObjectId::new(added_mod.actor_id()),
  54       target: generate_moderators_url(&community.actor_id)?.into(),
  55       cc: vec![community.actor_id()],
  56       kind: AddType::Add,
  57       id: id.clone(),
  58       unparsed: Default::default(),
  59     };
  60
  61     let activity = AnnouncableActivities::AddMod(add);
  62     let inboxes = vec![added_mod.shared_inbox_or_inbox()];
  63     send_activity_in_community(activity, actor, community, inboxes, context).await
  64   }
  65 }
  66
  67 #[async_trait::async_trait(?Send)]
  68 impl ActivityHandler for AddMod {
  69   type DataType = LemmyContext;
  70   type Error = LemmyError;
  71
  72   fn id(&self) -> &Url {
  73     &self.id
  74   }
  75
  76   fn actor(&self) -> &Url {
  77     self.actor.inner()
  78   }
  79
  80   #[tracing::instrument(skip_all)]
  81   async fn verify(
  82     &self,
  83     context: &Data<LemmyContext>,
  84     request_counter: &mut i32,
  85   ) -> Result<(), LemmyError> {
  86     verify_is_public(&self.to, &self.cc)?;
  87     let community = self.get_community(context, request_counter).await?;
  88     verify_person_in_community(&self.actor, &community, context, request_counter).await?;
  89     verify_mod_action(
  90       &self.actor,
  91       self.object.inner(),
  92       community.id,
  93       context,
  94       request_counter,
  95     )
  96     .await?;
  97     verify_add_remove_moderator_target(&self.target, &community)?;
  98     Ok(())
  99   }
 100
 101   #[tracing::instrument(skip_all)]
 102   async fn receive(
 103     self,
 104     context: &Data<LemmyContext>,
 105     request_counter: &mut i32,
 106   ) -> Result<(), LemmyError> {
 107     let community = self.get_community(context, request_counter).await?;
 108     let new_mod = self
 109       .object
 110       .dereference(context, local_instance(context).await, request_counter)
 111       .await?;
 112
 113     // If we had to refetch the community while parsing the activity, then the new mod has already
 114     // been added. Skip it here as it would result in a duplicate key error.
 115     let new_mod_id = new_mod.id;
 116     let moderated_communities =
 117       CommunityModerator::get_person_moderated_communities(context.pool(), new_mod_id).await?;
 118     if !moderated_communities.contains(&community.id) {
 119       let form = CommunityModeratorForm {
 120         community_id: community.id,
 121         person_id: new_mod.id,
 122       };
 123       CommunityModerator::join(context.pool(), &form).await?;
 124
 125       // write mod log
 126       let actor = self
 127         .actor
 128         .dereference(context, local_instance(context).await, request_counter)
 129         .await?;
 130       let form = ModAddCommunityForm {
 131         mod_person_id: actor.id,
 132         other_person_id: new_mod.id,
 133         community_id: community.id,
 134         removed: Some(false),
 135       };
 136       ModAddCommunity::create(context.pool(), &form).await?;
 137     }
 138     // TODO: send websocket notification about added mod
 139     Ok(())
 140   }
 141 }
 142
 143 #[async_trait::async_trait(?Send)]
 144 impl GetCommunity for AddMod {
 145   #[tracing::instrument(skip_all)]
 146   async fn get_community(
 147     &self,
 148     context: &LemmyContext,
 149     request_counter: &mut i32,
 150   ) -> Result<ApubCommunity, LemmyError> {
 151     get_community_from_moderators_url(&self.target, context, request_counter).await
 152   }
 153 }