Use audience field to federate items in groups (fixes #2464) (#2584)

[lemmy.git] / crates / apub / src / protocol / objects / note.rs

diff --git a/crates/apub/src/protocol/objects/note.rs b/crates/apub/src/protocol/objects/note.rs
index 727b98994941175846cd007cccf33058544926a4..21f7bcf38b67ffc30c12d7c7001afe3cc1d5f82a 100644 (file)
--- a/crates/apub/src/protocol/objects/note.rs
+++ b/crates/apub/src/protocol/objects/note.rs
@@ -1,9 +1,10 @@
 use crate::{
+  activities::verify_community_matches,
   fetcher::post_or_comment::PostOrComment,
   local_instance,
   mentions::MentionOrValue,
-  objects::{comment::ApubComment, person::ApubPerson, post::ApubPost},
-  protocol::{objects::LanguageTag, Source},
+  objects::{comment::ApubComment, community::ApubCommunity, person::ApubPerson, post::ApubPost},
+  protocol::{objects::LanguageTag, InCommunity, Source},
 };
 use activitypub_federation::{
   core::object_id::ObjectId,
@@ -14,7 +15,10 @@ use activitypub_federation::{
 };
 use activitystreams_kinds::object::NoteType;
 use chrono::{DateTime, FixedOffset};
-use lemmy_db_schema::{source::post::Post, traits::Crud};
+use lemmy_db_schema::{
+  source::{community::Community, post::Post},
+  traits::Crud,
+};
 use lemmy_utils::error::LemmyError;
 use lemmy_websocket::LemmyContext;
 use serde::{Deserialize, Serialize};
@@ -46,6 +50,7 @@ pub struct Note {
   // lemmy extension
   pub(crate) distinguished: Option<bool>,
   pub(crate) language: Option<LanguageTag>,
+  pub(crate) audience: Option<ObjectId<ApubCommunity>>,
 }
 
 impl Note {
@@ -75,3 +80,24 @@ impl Note {
     }
   }
 }
+
+#[async_trait::async_trait(?Send)]
+impl InCommunity for Note {
+  async fn community(
+    &self,
+    context: &LemmyContext,
+    request_counter: &mut i32,
+  ) -> Result<ApubCommunity, LemmyError> {
+    let (post, _) = self.get_parents(context, request_counter).await?;
+    let community_id = post.community_id;
+    if let Some(audience) = &self.audience {
+      let audience = audience
+        .dereference(context, local_instance(context).await, request_counter)
+        .await?;
+      verify_community_matches(&audience, community_id)?;
+      Ok(audience)
+    } else {
+      Ok(Community::read(context.pool(), community_id).await?.into())
+    }
+  }
+}