use crate::{ activities::{ following::follow::FollowCommunity, generate_activity_id, send_lemmy_activity, verify_activity, }, fetcher::object_id::ObjectId, objects::{community::ApubCommunity, person::ApubPerson}, }; use activitystreams::{activity::kind::AcceptType, unparsed::Unparsed}; use lemmy_api_common::blocking; use lemmy_apub_lib::{ data::Data, traits::{ActivityFields, ActivityHandler, ActorType}, verify::verify_urls_match, }; use lemmy_db_schema::{source::community::CommunityFollower, traits::Followable}; use lemmy_utils::LemmyError; use lemmy_websocket::LemmyContext; use serde::{Deserialize, Serialize}; use url::Url; #[derive(Clone, Debug, Deserialize, Serialize, ActivityFields)] #[serde(rename_all = "camelCase")] pub struct AcceptFollowCommunity { actor: ObjectId, to: [ObjectId; 1], object: FollowCommunity, #[serde(rename = "type")] kind: AcceptType, id: Url, #[serde(flatten)] unparsed: Unparsed, } impl AcceptFollowCommunity { pub async fn send( follow: FollowCommunity, context: &LemmyContext, request_counter: &mut i32, ) -> Result<(), LemmyError> { let community = follow.object.dereference_local(context).await?; let person = follow .actor .clone() .dereference(context, request_counter) .await?; let accept = AcceptFollowCommunity { actor: ObjectId::new(community.actor_id()), to: [ObjectId::new(person.actor_id())], object: follow, kind: AcceptType::Accept, id: generate_activity_id( AcceptType::Accept, &context.settings().get_protocol_and_hostname(), )?, unparsed: Default::default(), }; let inbox = vec![person.inbox_url()]; send_lemmy_activity(context, &accept, &accept.id, &community, inbox, true).await } } /// Handle accepted follows #[async_trait::async_trait(?Send)] impl ActivityHandler for AcceptFollowCommunity { type DataType = LemmyContext; async fn verify( &self, context: &Data, request_counter: &mut i32, ) -> Result<(), LemmyError> { verify_activity(self, &context.settings())?; verify_urls_match(self.to[0].inner(), self.object.actor())?; verify_urls_match(self.actor(), self.object.to[0].inner())?; self.object.verify(context, request_counter).await?; Ok(()) } async fn receive( self, context: &Data, request_counter: &mut i32, ) -> Result<(), LemmyError> { let actor = self.actor.dereference(context, request_counter).await?; let to = self.to[0].dereference(context, request_counter).await?; // This will throw an error if no follow was requested blocking(context.pool(), move |conn| { CommunityFollower::follow_accepted(conn, actor.id, to.id) }) .await??; Ok(()) } }