use crate::{ activities::{ generate_activity_id, send_lemmy_activity, verify_person, verify_person_in_community, }, local_instance, objects::{community::ApubCommunity, person::ApubPerson}, protocol::activities::following::{accept::AcceptFollowCommunity, follow::FollowCommunity}, ActorType, }; use activitypub_federation::{core::object_id::ObjectId, data::Data, traits::ActivityHandler}; use activitystreams_kinds::activity::FollowType; use lemmy_api_common::utils::blocking; use lemmy_db_schema::{ source::community::{CommunityFollower, CommunityFollowerForm}, traits::Followable, }; use lemmy_utils::error::LemmyError; use lemmy_websocket::LemmyContext; use url::Url; impl FollowCommunity { pub(in crate::activities::following) fn new( actor: &ApubPerson, community: &ApubCommunity, context: &LemmyContext, ) -> Result { Ok(FollowCommunity { actor: ObjectId::new(actor.actor_id()), object: ObjectId::new(community.actor_id()), kind: FollowType::Follow, id: generate_activity_id( FollowType::Follow, &context.settings().get_protocol_and_hostname(), )?, unparsed: Default::default(), }) } #[tracing::instrument(skip_all)] pub async fn send( actor: &ApubPerson, community: &ApubCommunity, context: &LemmyContext, ) -> Result<(), LemmyError> { let community_follower_form = CommunityFollowerForm { community_id: community.id, person_id: actor.id, pending: true, }; blocking(context.pool(), move |conn| { CommunityFollower::follow(conn, &community_follower_form).ok() }) .await?; let follow = FollowCommunity::new(actor, community, context)?; let inbox = vec![community.inbox_url.clone().into()]; send_lemmy_activity(context, &follow, &follow.id, actor, inbox, true).await } } #[async_trait::async_trait(?Send)] impl ActivityHandler for FollowCommunity { type DataType = LemmyContext; type Error = LemmyError; fn id(&self) -> &Url { &self.id } fn actor(&self) -> &Url { self.actor.inner() } #[tracing::instrument(skip_all)] async fn verify( &self, context: &Data, request_counter: &mut i32, ) -> Result<(), LemmyError> { verify_person(&self.actor, context, request_counter).await?; let community = self .object .dereference::(context, local_instance(context), request_counter) .await?; verify_person_in_community(&self.actor, &community, context, request_counter).await?; Ok(()) } #[tracing::instrument(skip_all)] async fn receive( self, context: &Data, request_counter: &mut i32, ) -> Result<(), LemmyError> { let person = self .actor .dereference::(context, local_instance(context), request_counter) .await?; let community = self .object .dereference::(context, local_instance(context), request_counter) .await?; let community_follower_form = CommunityFollowerForm { community_id: community.id, person_id: person.id, pending: false, }; // This will fail if they're already a follower, but ignore the error. blocking(context.pool(), move |conn| { CommunityFollower::follow(conn, &community_follower_form).ok() }) .await?; AcceptFollowCommunity::send(self, context, request_counter).await } }