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

   1 use crate::{
   2   activities::{generate_activity_id, send_lemmy_activity, verify_activity, verify_is_public},
   3   activity_lists::AnnouncableActivities,
   4   http::{is_activity_already_known, ActivityCommonFields},
   5   insert_activity,
   6   objects::community::ApubCommunity,
   7   protocol::activities::{community::announce::AnnounceActivity, CreateOrUpdateType},
   8 };
   9 use activitystreams::{activity::kind::AnnounceType, public};
  10 use lemmy_apub_lib::{
  11   data::Data,
  12   object_id::ObjectId,
  13   traits::{ActivityHandler, ActorType},
  14 };
  15 use lemmy_utils::LemmyError;
  16 use lemmy_websocket::LemmyContext;
  17
  18 #[async_trait::async_trait(?Send)]
  19 pub(crate) trait GetCommunity {
  20   async fn get_community(
  21     &self,
  22     context: &LemmyContext,
  23     request_counter: &mut i32,
  24   ) -> Result<ApubCommunity, LemmyError>;
  25 }
  26
  27 impl AnnounceActivity {
  28   fn new(
  29     object: AnnouncableActivities,
  30     community: &ApubCommunity,
  31     context: &LemmyContext,
  32   ) -> Result<AnnounceActivity, LemmyError> {
  33     Ok(AnnounceActivity {
  34       actor: ObjectId::new(community.actor_id()),
  35       to: vec![public()],
  36       object,
  37       cc: vec![community.followers_url.clone().into()],
  38       kind: AnnounceType::Announce,
  39       id: generate_activity_id(
  40         &AnnounceType::Announce,
  41         &context.settings().get_protocol_and_hostname(),
  42       )?,
  43       unparsed: Default::default(),
  44     })
  45   }
  46
  47   pub async fn send(
  48     object: AnnouncableActivities,
  49     community: &ApubCommunity,
  50     context: &LemmyContext,
  51   ) -> Result<(), LemmyError> {
  52     let announce = AnnounceActivity::new(object.clone(), community, context)?;
  53     let inboxes = community.get_follower_inboxes(context).await?;
  54     send_lemmy_activity(
  55       context,
  56       &announce,
  57       &announce.id,
  58       community,
  59       inboxes.clone(),
  60       false,
  61     )
  62     .await?;
  63
  64     // Pleroma and Mastodon can't handle activities like Announce/Create/Page. So for
  65     // compatibility, we also send Announce/Page so that they can follow Lemmy communities.
  66     use AnnouncableActivities::*;
  67     let object = match object {
  68       CreateOrUpdatePost(c) if c.kind == CreateOrUpdateType::Create => Page(c.object),
  69       _ => return Ok(()),
  70     };
  71     let announce_compat = AnnounceActivity::new(object, community, context)?;
  72     send_lemmy_activity(
  73       context,
  74       &announce_compat,
  75       &announce_compat.id,
  76       community,
  77       inboxes,
  78       false,
  79     )
  80     .await?;
  81     Ok(())
  82   }
  83 }
  84
  85 #[async_trait::async_trait(?Send)]
  86 impl ActivityHandler for AnnounceActivity {
  87   type DataType = LemmyContext;
  88   async fn verify(
  89     &self,
  90     context: &Data<LemmyContext>,
  91     request_counter: &mut i32,
  92   ) -> Result<(), LemmyError> {
  93     verify_is_public(&self.to, &self.cc)?;
  94     verify_activity(&self.id, self.actor.inner(), &context.settings())?;
  95     self.object.verify(context, request_counter).await?;
  96     Ok(())
  97   }
  98
  99   async fn receive(
 100     self,
 101     context: &Data<LemmyContext>,
 102     request_counter: &mut i32,
 103   ) -> Result<(), LemmyError> {
 104     let object_value = serde_json::to_value(&self.object)?;
 105     let object_data: ActivityCommonFields = serde_json::from_value(object_value.to_owned())?;
 106
 107     if is_activity_already_known(context.pool(), &object_data.id).await? {
 108       return Ok(());
 109     }
 110     insert_activity(&object_data.id, object_value, false, true, context.pool()).await?;
 111     self.object.receive(context, request_counter).await
 112   }
 113 }