crates/apub/src/activities/following/accept.rs

   1 use crate::{
   2   activities::{
   3     following::follow::FollowCommunity,
   4     generate_activity_id,
   5     verify_activity,
   6     verify_community,
   7   },
   8   context::lemmy_context,
   9   fetcher::object_id::ObjectId,
  10   send_lemmy_activity,
  11 };
  12 use activitystreams::{
  13   activity::kind::AcceptType,
  14   base::AnyBase,
  15   primitives::OneOrMany,
  16   unparsed::Unparsed,
  17 };
  18 use lemmy_api_common::blocking;
  19 use lemmy_apub_lib::{
  20   data::Data,
  21   traits::{ActivityFields, ActivityHandler, ActorType},
  22   verify::verify_urls_match,
  23 };
  24 use lemmy_db_queries::Followable;
  25 use lemmy_db_schema::source::{
  26   community::{Community, CommunityFollower},
  27   person::Person,
  28 };
  29 use lemmy_utils::LemmyError;
  30 use lemmy_websocket::LemmyContext;
  31 use serde::{Deserialize, Serialize};
  32 use url::Url;
  33
  34 #[derive(Clone, Debug, Deserialize, Serialize, ActivityFields)]
  35 #[serde(rename_all = "camelCase")]
  36 pub struct AcceptFollowCommunity {
  37   actor: ObjectId<Community>,
  38   to: ObjectId<Person>,
  39   object: FollowCommunity,
  40   #[serde(rename = "type")]
  41   kind: AcceptType,
  42   id: Url,
  43   #[serde(rename = "@context")]
  44   context: OneOrMany<AnyBase>,
  45   #[serde(flatten)]
  46   unparsed: Unparsed,
  47 }
  48
  49 impl AcceptFollowCommunity {
  50   pub async fn send(
  51     follow: FollowCommunity,
  52     context: &LemmyContext,
  53     request_counter: &mut i32,
  54   ) -> Result<(), LemmyError> {
  55     let community = follow.object.dereference_local(context).await?;
  56     let person = follow
  57       .actor
  58       .clone()
  59       .dereference(context, request_counter)
  60       .await?;
  61     let accept = AcceptFollowCommunity {
  62       actor: ObjectId::new(community.actor_id()),
  63       to: ObjectId::new(person.actor_id()),
  64       object: follow,
  65       kind: AcceptType::Accept,
  66       id: generate_activity_id(
  67         AcceptType::Accept,
  68         &context.settings().get_protocol_and_hostname(),
  69       )?,
  70       context: lemmy_context(),
  71       unparsed: Default::default(),
  72     };
  73     let inbox = vec![person.inbox_url.into()];
  74     send_lemmy_activity(context, &accept, &accept.id, &community, inbox, true).await
  75   }
  76 }
  77
  78 /// Handle accepted follows
  79 #[async_trait::async_trait(?Send)]
  80 impl ActivityHandler for AcceptFollowCommunity {
  81   type DataType = LemmyContext;
  82   async fn verify(
  83     &self,
  84     context: &Data<LemmyContext>,
  85     request_counter: &mut i32,
  86   ) -> Result<(), LemmyError> {
  87     verify_activity(self, &context.settings())?;
  88     verify_urls_match(self.to.inner(), self.object.actor())?;
  89     verify_urls_match(self.actor(), self.object.to.inner())?;
  90     verify_community(&self.actor, context, request_counter).await?;
  91     self.object.verify(context, request_counter).await?;
  92     Ok(())
  93   }
  94
  95   async fn receive(
  96     self,
  97     context: &Data<LemmyContext>,
  98     request_counter: &mut i32,
  99   ) -> Result<(), LemmyError> {
 100     let actor = self.actor.dereference(context, request_counter).await?;
 101     let to = self.to.dereference(context, request_counter).await?;
 102     // This will throw an error if no follow was requested
 103     blocking(context.pool(), move |conn| {
 104       CommunityFollower::follow_accepted(conn, actor.id, to.id)
 105     })
 106     .await??;
 107
 108     Ok(())
 109   }
 110 }