crates/apub/src/objects/community.rs

   1 use crate::{
   2   check_apub_id_valid_with_strictness,
   3   fetch_local_site_data,
   4   objects::instance::fetch_instance_actor_for_object,
   5   protocol::{
   6     objects::{group::Group, Endpoints, LanguageTag},
   7     ImageObject,
   8     Source,
   9   },
  10 };
  11 use activitypub_federation::{
  12   config::Data,
  13   kinds::actor::GroupType,
  14   traits::{Actor, Object},
  15 };
  16 use chrono::NaiveDateTime;
  17 use itertools::Itertools;
  18 use lemmy_api_common::{
  19   context::LemmyContext,
  20   utils::{generate_featured_url, generate_moderators_url, generate_outbox_url},
  21 };
  22 use lemmy_db_schema::{
  23   source::{
  24     actor_language::CommunityLanguage,
  25     community::{Community, CommunityUpdateForm},
  26   },
  27   traits::{ApubActor, Crud},
  28 };
  29 use lemmy_db_views_actor::structs::CommunityFollowerView;
  30 use lemmy_utils::{
  31   error::LemmyError,
  32   utils::{markdown::markdown_to_html, time::convert_datetime},
  33 };
  34 use std::ops::Deref;
  35 use tracing::debug;
  36 use url::Url;
  37
  38 #[derive(Clone, Debug)]
  39 pub struct ApubCommunity(Community);
  40
  41 impl Deref for ApubCommunity {
  42   type Target = Community;
  43   fn deref(&self) -> &Self::Target {
  44     &self.0
  45   }
  46 }
  47
  48 impl From<Community> for ApubCommunity {
  49   fn from(c: Community) -> Self {
  50     ApubCommunity(c)
  51   }
  52 }
  53
  54 #[async_trait::async_trait]
  55 impl Object for ApubCommunity {
  56   type DataType = LemmyContext;
  57   type Kind = Group;
  58   type Error = LemmyError;
  59
  60   fn last_refreshed_at(&self) -> Option<NaiveDateTime> {
  61     Some(self.last_refreshed_at)
  62   }
  63
  64   #[tracing::instrument(skip_all)]
  65   async fn read_from_id(
  66     object_id: Url,
  67     context: &Data<Self::DataType>,
  68   ) -> Result<Option<Self>, LemmyError> {
  69     Ok(
  70       Community::read_from_apub_id(context.pool(), &object_id.into())
  71         .await?
  72         .map(Into::into),
  73     )
  74   }
  75
  76   #[tracing::instrument(skip_all)]
  77   async fn delete(self, context: &Data<Self::DataType>) -> Result<(), LemmyError> {
  78     let form = CommunityUpdateForm::builder().deleted(Some(true)).build();
  79     Community::update(context.pool(), self.id, &form).await?;
  80     Ok(())
  81   }
  82
  83   #[tracing::instrument(skip_all)]
  84   async fn into_json(self, data: &Data<Self::DataType>) -> Result<Group, LemmyError> {
  85     let community_id = self.id;
  86     let langs = CommunityLanguage::read(data.pool(), community_id).await?;
  87     let language = LanguageTag::new_multiple(langs, data.pool()).await?;
  88
  89     let group = Group {
  90       kind: GroupType::Group,
  91       id: self.id().into(),
  92       preferred_username: self.name.clone(),
  93       name: Some(self.title.clone()),
  94       summary: self.description.as_ref().map(|b| markdown_to_html(b)),
  95       source: self.description.clone().map(Source::new),
  96       icon: self.icon.clone().map(ImageObject::new),
  97       image: self.banner.clone().map(ImageObject::new),
  98       sensitive: Some(self.nsfw),
  99       featured: Some(generate_featured_url(&self.actor_id)?.into()),
 100       inbox: self.inbox_url.clone().into(),
 101       outbox: generate_outbox_url(&self.actor_id)?.into(),
 102       followers: self.followers_url.clone().into(),
 103       endpoints: self.shared_inbox_url.clone().map(|s| Endpoints {
 104         shared_inbox: s.into(),
 105       }),
 106       public_key: self.public_key(),
 107       language,
 108       published: Some(convert_datetime(self.published)),
 109       updated: self.updated.map(convert_datetime),
 110       posting_restricted_to_mods: Some(self.posting_restricted_to_mods),
 111       attributed_to: Some(generate_moderators_url(&self.actor_id)?.into()),
 112     };
 113     Ok(group)
 114   }
 115
 116   #[tracing::instrument(skip_all)]
 117   async fn verify(
 118     group: &Group,
 119     expected_domain: &Url,
 120     context: &Data<Self::DataType>,
 121   ) -> Result<(), LemmyError> {
 122     group.verify(expected_domain, context).await
 123   }
 124
 125   /// Converts a `Group` to `Community`, inserts it into the database and updates moderators.
 126   #[tracing::instrument(skip_all)]
 127   async fn from_json(
 128     group: Group,
 129     context: &Data<Self::DataType>,
 130   ) -> Result<ApubCommunity, LemmyError> {
 131     let instance_id = fetch_instance_actor_for_object(&group.id, context).await?;
 132
 133     let form = Group::into_insert_form(group.clone(), instance_id);
 134     let languages = LanguageTag::to_language_id_multiple(group.language, context.pool()).await?;
 135
 136     let community = Community::create(context.pool(), &form).await?;
 137     CommunityLanguage::update(context.pool(), languages, community.id).await?;
 138
 139     let community: ApubCommunity = community.into();
 140
 141     // Fetching mods and outbox is not necessary for Lemmy to work, so ignore errors. Besides,
 142     // we need to ignore these errors so that tests can work entirely offline.
 143     group
 144       .outbox
 145       .dereference(&community, context)
 146       .await
 147       .map_err(|e| debug!("{}", e))
 148       .ok();
 149
 150     if let Some(moderators) = group.attributed_to {
 151       moderators
 152         .dereference(&community, context)
 153         .await
 154         .map_err(|e| debug!("{}", e))
 155         .ok();
 156     }
 157
 158     Ok(community)
 159   }
 160 }
 161
 162 impl Actor for ApubCommunity {
 163   fn id(&self) -> Url {
 164     self.actor_id.inner().clone()
 165   }
 166
 167   fn public_key_pem(&self) -> &str {
 168     &self.public_key
 169   }
 170
 171   fn private_key_pem(&self) -> Option<String> {
 172     self.private_key.clone()
 173   }
 174
 175   fn inbox(&self) -> Url {
 176     self.inbox_url.clone().into()
 177   }
 178
 179   fn shared_inbox(&self) -> Option<Url> {
 180     self.shared_inbox_url.clone().map(Into::into)
 181   }
 182 }
 183
 184 impl ApubCommunity {
 185   /// For a given community, returns the inboxes of all followers.
 186   #[tracing::instrument(skip_all)]
 187   pub(crate) async fn get_follower_inboxes(
 188     &self,
 189     context: &LemmyContext,
 190   ) -> Result<Vec<Url>, LemmyError> {
 191     let id = self.id;
 192
 193     let local_site_data = fetch_local_site_data(context.pool()).await?;
 194     let follows = CommunityFollowerView::for_community(context.pool(), id).await?;
 195     let inboxes: Vec<Url> = follows
 196       .into_iter()
 197       .filter(|f| !f.follower.local)
 198       .map(|f| {
 199         f.follower
 200           .shared_inbox_url
 201           .unwrap_or(f.follower.inbox_url)
 202           .into()
 203       })
 204       .unique()
 205       .filter(|inbox: &Url| inbox.host_str() != Some(&context.settings().hostname))
 206       // Don't send to blocked instances
 207       .filter(|inbox| {
 208         check_apub_id_valid_with_strictness(inbox, false, &local_site_data, context.settings())
 209           .is_ok()
 210       })
 211       .collect();
 212
 213     Ok(inboxes)
 214   }
 215 }
 216
 217 #[cfg(test)]
 218 pub(crate) mod tests {
 219   use super::*;
 220   use crate::{
 221     objects::{instance::tests::parse_lemmy_instance, tests::init_context},
 222     protocol::tests::file_to_json_object,
 223   };
 224   use activitypub_federation::fetch::collection_id::CollectionId;
 225   use lemmy_db_schema::{source::site::Site, traits::Crud};
 226   use serial_test::serial;
 227
 228   pub(crate) async fn parse_lemmy_community(context: &Data<LemmyContext>) -> ApubCommunity {
 229     // use separate counter so this doesnt affect tests
 230     let context2 = context.reset_request_count();
 231     let mut json: Group = file_to_json_object("assets/lemmy/objects/group.json").unwrap();
 232     // change these links so they dont fetch over the network
 233     json.attributed_to = None;
 234     json.outbox =
 235       CollectionId::parse("https://enterprise.lemmy.ml/c/tenforward/not_outbox").unwrap();
 236
 237     let url = Url::parse("https://enterprise.lemmy.ml/c/tenforward").unwrap();
 238     ApubCommunity::verify(&json, &url, &context2).await.unwrap();
 239     let community = ApubCommunity::from_json(json, &context2).await.unwrap();
 240     // this makes one requests to the (intentionally broken) outbox collection
 241     assert_eq!(context2.request_count(), 1);
 242     community
 243   }
 244
 245   #[tokio::test]
 246   #[serial]
 247   async fn test_parse_lemmy_community() {
 248     let context = init_context().await;
 249     let site = parse_lemmy_instance(&context).await;
 250     let community = parse_lemmy_community(&context).await;
 251
 252     assert_eq!(community.title, "Ten Forward");
 253     assert!(!community.local);
 254     assert_eq!(community.description.as_ref().unwrap().len(), 132);
 255
 256     Community::delete(context.pool(), community.id)
 257       .await
 258       .unwrap();
 259     Site::delete(context.pool(), site.id).await.unwrap();
 260   }
 261 }