crates/apub/src/activity_lists.rs

   1 use crate::{
   2   objects::community::ApubCommunity,
   3   protocol::{
   4     activities::{
   5       block::{block_user::BlockUser, undo_block_user::UndoBlockUser},
   6       community::{
   7         add_mod::AddMod,
   8         announce::{AnnounceActivity, RawAnnouncableActivities},
   9         remove_mod::RemoveMod,
  10         report::Report,
  11         update::UpdateCommunity,
  12       },
  13       create_or_update::{
  14         chat_message::CreateOrUpdateChatMessage,
  15         note::CreateOrUpdateNote,
  16         page::CreateOrUpdatePage,
  17       },
  18       deletion::{delete::Delete, delete_user::DeleteUser, undo_delete::UndoDelete},
  19       following::{accept::AcceptFollow, follow::Follow, undo_follow::UndoFollow},
  20       voting::{undo_vote::UndoVote, vote::Vote},
  21     },
  22     objects::page::Page,
  23     InCommunity,
  24   },
  25 };
  26 use activitypub_federation::{data::Data, deser::context::WithContext, traits::ActivityHandler};
  27 use lemmy_api_common::context::LemmyContext;
  28 use lemmy_utils::error::LemmyError;
  29 use serde::{Deserialize, Serialize};
  30 use url::Url;
  31
  32 #[derive(Debug, Deserialize, Serialize)]
  33 #[serde(untagged)]
  34 #[enum_delegate::implement(ActivityHandler)]
  35 pub enum SharedInboxActivities {
  36   PersonInboxActivities(Box<WithContext<PersonInboxActivities>>),
  37   GroupInboxActivities(Box<WithContext<GroupInboxActivities>>),
  38 }
  39
  40 #[derive(Debug, Deserialize, Serialize)]
  41 #[serde(untagged)]
  42 #[enum_delegate::implement(ActivityHandler)]
  43 pub enum GroupInboxActivities {
  44   Follow(Follow),
  45   UndoFollow(UndoFollow),
  46   Report(Report),
  47   // This is a catch-all and needs to be last
  48   AnnouncableActivities(RawAnnouncableActivities),
  49 }
  50
  51 #[derive(Clone, Debug, Deserialize, Serialize)]
  52 #[serde(untagged)]
  53 #[enum_delegate::implement(ActivityHandler)]
  54 pub enum PersonInboxActivities {
  55   Follow(Follow),
  56   AcceptFollow(AcceptFollow),
  57   UndoFollow(UndoFollow),
  58   CreateOrUpdatePrivateMessage(CreateOrUpdateChatMessage),
  59   Delete(Delete),
  60   UndoDelete(UndoDelete),
  61   AnnounceActivity(AnnounceActivity),
  62 }
  63
  64 /// This is necessary for user inbox, which can also receive some "announcable" activities,
  65 /// eg a comment mention. This needs to be a separate enum so that announcables received in shared
  66 /// inbox can fall through to be parsed as GroupInboxActivities::AnnouncableActivities.
  67 #[derive(Clone, Debug, Deserialize, Serialize)]
  68 #[serde(untagged)]
  69 #[enum_delegate::implement(ActivityHandler)]
  70 pub enum PersonInboxActivitiesWithAnnouncable {
  71   PersonInboxActivities(Box<PersonInboxActivities>),
  72   AnnouncableActivities(Box<AnnouncableActivities>),
  73 }
  74
  75 #[derive(Clone, Debug, Deserialize, Serialize)]
  76 #[serde(untagged)]
  77 #[enum_delegate::implement(ActivityHandler)]
  78 pub enum AnnouncableActivities {
  79   CreateOrUpdateComment(CreateOrUpdateNote),
  80   CreateOrUpdatePost(CreateOrUpdatePage),
  81   Vote(Vote),
  82   UndoVote(UndoVote),
  83   Delete(Delete),
  84   UndoDelete(UndoDelete),
  85   UpdateCommunity(UpdateCommunity),
  86   BlockUser(BlockUser),
  87   UndoBlockUser(UndoBlockUser),
  88   AddMod(AddMod),
  89   RemoveMod(RemoveMod),
  90   // For compatibility with Pleroma/Mastodon (send only)
  91   Page(Page),
  92 }
  93
  94 #[derive(Clone, Debug, Deserialize, Serialize)]
  95 #[serde(untagged)]
  96 #[enum_delegate::implement(ActivityHandler)]
  97 #[allow(clippy::enum_variant_names)]
  98 pub enum SiteInboxActivities {
  99   BlockUser(BlockUser),
 100   UndoBlockUser(UndoBlockUser),
 101   DeleteUser(DeleteUser),
 102 }
 103
 104 #[async_trait::async_trait(?Send)]
 105 impl InCommunity for AnnouncableActivities {
 106   #[tracing::instrument(skip(self, context))]
 107   async fn community(
 108     &self,
 109     context: &LemmyContext,
 110     request_counter: &mut i32,
 111   ) -> Result<ApubCommunity, LemmyError> {
 112     use AnnouncableActivities::*;
 113     match self {
 114       CreateOrUpdateComment(a) => a.community(context, request_counter).await,
 115       CreateOrUpdatePost(a) => a.community(context, request_counter).await,
 116       Vote(a) => a.community(context, request_counter).await,
 117       UndoVote(a) => a.community(context, request_counter).await,
 118       Delete(a) => a.community(context, request_counter).await,
 119       UndoDelete(a) => a.community(context, request_counter).await,
 120       UpdateCommunity(a) => a.community(context, request_counter).await,
 121       BlockUser(a) => a.community(context, request_counter).await,
 122       UndoBlockUser(a) => a.community(context, request_counter).await,
 123       AddMod(a) => a.community(context, request_counter).await,
 124       RemoveMod(a) => a.community(context, request_counter).await,
 125       Page(_) => unimplemented!(),
 126     }
 127   }
 128 }
 129
 130 #[cfg(test)]
 131 mod tests {
 132   use crate::{
 133     activity_lists::{
 134       GroupInboxActivities,
 135       PersonInboxActivities,
 136       PersonInboxActivitiesWithAnnouncable,
 137       SiteInboxActivities,
 138     },
 139     protocol::tests::{test_json, test_parse_lemmy_item},
 140   };
 141
 142   #[test]
 143   fn test_group_inbox() {
 144     test_parse_lemmy_item::<GroupInboxActivities>("assets/lemmy/activities/following/follow.json")
 145       .unwrap();
 146     test_parse_lemmy_item::<GroupInboxActivities>(
 147       "assets/lemmy/activities/create_or_update/create_note.json",
 148     )
 149     .unwrap();
 150   }
 151
 152   #[test]
 153   fn test_person_inbox() {
 154     test_parse_lemmy_item::<PersonInboxActivities>("assets/lemmy/activities/following/accept.json")
 155       .unwrap();
 156     test_parse_lemmy_item::<PersonInboxActivitiesWithAnnouncable>(
 157       "assets/lemmy/activities/create_or_update/create_note.json",
 158     )
 159     .unwrap();
 160     test_parse_lemmy_item::<PersonInboxActivitiesWithAnnouncable>(
 161       "assets/lemmy/activities/create_or_update/create_private_message.json",
 162     )
 163     .unwrap();
 164     test_json::<PersonInboxActivitiesWithAnnouncable>("assets/mastodon/activities/follow.json")
 165       .unwrap();
 166   }
 167
 168   #[test]
 169   fn test_site_inbox() {
 170     test_parse_lemmy_item::<SiteInboxActivities>(
 171       "assets/lemmy/activities/deletion/delete_user.json",
 172     )
 173     .unwrap();
 174   }
 175 }