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::ActivityCommonFields,
   5   insert_activity,
   6   objects::community::ApubCommunity,
   7   protocol::activities::{community::announce::AnnounceActivity, CreateOrUpdateType},
   8 };
   9 use activitystreams_kinds::{activity::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 use tracing::debug;
  18
  19 #[async_trait::async_trait(?Send)]
  20 pub(crate) trait GetCommunity {
  21   async fn get_community(
  22     &self,
  23     context: &LemmyContext,
  24     request_counter: &mut i32,
  25   ) -> Result<ApubCommunity, LemmyError>;
  26 }
  27
  28 impl AnnounceActivity {
  29   pub(crate) fn new(
  30     object: AnnouncableActivities,
  31     community: &ApubCommunity,
  32     context: &LemmyContext,
  33   ) -> Result<AnnounceActivity, LemmyError> {
  34     Ok(AnnounceActivity {
  35       actor: ObjectId::new(community.actor_id()),
  36       to: vec![public()],
  37       object,
  38       cc: vec![community.followers_url.clone().into()],
  39       kind: AnnounceType::Announce,
  40       id: generate_activity_id(
  41         &AnnounceType::Announce,
  42         &context.settings().get_protocol_and_hostname(),
  43       )?,
  44       unparsed: Default::default(),
  45     })
  46   }
  47
  48   #[tracing::instrument(skip_all)]
  49   pub async fn send(
  50     object: AnnouncableActivities,
  51     community: &ApubCommunity,
  52     context: &LemmyContext,
  53   ) -> Result<(), LemmyError> {
  54     let announce = AnnounceActivity::new(object.clone(), community, context)?;
  55     let inboxes = community.get_follower_inboxes(context).await?;
  56     send_lemmy_activity(
  57       context,
  58       &announce,
  59       &announce.id,
  60       community,
  61       inboxes.clone(),
  62       false,
  63     )
  64     .await?;
  65
  66     // Pleroma and Mastodon can't handle activities like Announce/Create/Page. So for
  67     // compatibility, we also send Announce/Page so that they can follow Lemmy communities.
  68     use AnnouncableActivities::*;
  69     let object = match object {
  70       CreateOrUpdatePost(c) if c.kind == CreateOrUpdateType::Create => Page(c.object),
  71       _ => return Ok(()),
  72     };
  73     let announce_compat = AnnounceActivity::new(object, community, context)?;
  74     send_lemmy_activity(
  75       context,
  76       &announce_compat,
  77       &announce_compat.id,
  78       community,
  79       inboxes,
  80       false,
  81     )
  82     .await?;
  83     Ok(())
  84   }
  85 }
  86
  87 #[async_trait::async_trait(?Send)]
  88 impl ActivityHandler for AnnounceActivity {
  89   type DataType = LemmyContext;
  90
  91   #[tracing::instrument(skip_all)]
  92   async fn verify(
  93     &self,
  94     context: &Data<LemmyContext>,
  95     request_counter: &mut i32,
  96   ) -> Result<(), LemmyError> {
  97     verify_is_public(&self.to, &self.cc)?;
  98     verify_activity(&self.id, self.actor.inner(), &context.settings())?;
  99     self.object.verify(context, request_counter).await?;
 100     Ok(())
 101   }
 102
 103   #[tracing::instrument(skip_all)]
 104   async fn receive(
 105     self,
 106     context: &Data<LemmyContext>,
 107     request_counter: &mut i32,
 108   ) -> Result<(), LemmyError> {
 109     // TODO: this can probably be implemented in a cleaner way
 110     match self.object {
 111       // Dont insert these into activities table, as they are not activities.
 112       AnnouncableActivities::Page(_) => {}
 113       _ => {
 114         let object_value = serde_json::to_value(&self.object)?;
 115         let object_data: ActivityCommonFields = serde_json::from_value(object_value.to_owned())?;
 116
 117         let insert =
 118           insert_activity(&object_data.id, object_value, false, true, context.pool()).await?;
 119         if !insert {
 120           debug!(
 121             "Received duplicate activity in announce {}",
 122             object_data.id.to_string()
 123           );
 124           return Ok(());
 125         }
 126       }
 127     }
 128     self.object.receive(context, request_counter).await
 129   }
 130 }