crates/apub/src/activity_lists.rs

   1 use crate::{
   2   activities::community::announce::GetCommunity,
   3   objects::community::ApubCommunity,
   4   protocol::{
   5     activities::{
   6       block::{block_user::BlockUser, undo_block_user::UndoBlockUser},
   7       community::{
   8         add_mod::AddMod,
   9         announce::{AnnounceActivity, RawAnnouncableActivities},
  10         remove_mod::RemoveMod,
  11         report::Report,
  12         update::UpdateCommunity,
  13       },
  14       create_or_update::{
  15         comment::CreateOrUpdateComment,
  16         post::CreateOrUpdatePost,
  17         private_message::CreateOrUpdatePrivateMessage,
  18       },
  19       deletion::{delete::Delete, delete_user::DeleteUser, undo_delete::UndoDelete},
  20       following::{accept::AcceptFollow, follow::Follow, undo_follow::UndoFollow},
  21       voting::{undo_vote::UndoVote, vote::Vote},
  22     },
  23     objects::page::Page,
  24   },
  25 };
  26 use activitypub_federation::{data::Data, deser::context::WithContext, traits::ActivityHandler};
  27 use lemmy_utils::error::LemmyError;
  28 use lemmy_websocket::LemmyContext;
  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   AcceptFollow(AcceptFollow),
  56   UndoFollow(UndoFollow),
  57   FollowCommunity(Follow),
  58   CreateOrUpdatePrivateMessage(CreateOrUpdatePrivateMessage),
  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(CreateOrUpdateComment),
  80   CreateOrUpdatePost(CreateOrUpdatePost),
  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 GetCommunity for AnnouncableActivities {
 106   #[tracing::instrument(skip(self, context))]
 107   async fn get_community(
 108     &self,
 109     context: &LemmyContext,
 110     request_counter: &mut i32,
 111   ) -> Result<ApubCommunity, LemmyError> {
 112     use AnnouncableActivities::*;
 113     let community = match self {
 114       CreateOrUpdateComment(a) => a.get_community(context, request_counter).await?,
 115       CreateOrUpdatePost(a) => a.get_community(context, request_counter).await?,
 116       Vote(a) => a.get_community(context, request_counter).await?,
 117       UndoVote(a) => a.get_community(context, request_counter).await?,
 118       Delete(a) => a.get_community(context, request_counter).await?,
 119       UndoDelete(a) => a.get_community(context, request_counter).await?,
 120       UpdateCommunity(a) => a.get_community(context, request_counter).await?,
 121       BlockUser(a) => a.get_community(context, request_counter).await?,
 122       UndoBlockUser(a) => a.get_community(context, request_counter).await?,
 123       AddMod(a) => a.get_community(context, request_counter).await?,
 124       RemoveMod(a) => a.get_community(context, request_counter).await?,
 125       Page(_) => unimplemented!(),
 126     };
 127     Ok(community)
 128   }
 129 }
 130
 131 #[cfg(test)]
 132 mod tests {
 133   use crate::{
 134     activity_lists::{
 135       GroupInboxActivities,
 136       PersonInboxActivities,
 137       PersonInboxActivitiesWithAnnouncable,
 138       SiteInboxActivities,
 139     },
 140     protocol::tests::test_parse_lemmy_item,
 141   };
 142
 143   #[test]
 144   fn test_group_inbox() {
 145     test_parse_lemmy_item::<GroupInboxActivities>("assets/lemmy/activities/following/follow.json")
 146       .unwrap();
 147     test_parse_lemmy_item::<GroupInboxActivities>(
 148       "assets/lemmy/activities/create_or_update/create_note.json",
 149     )
 150     .unwrap();
 151   }
 152
 153   #[test]
 154   fn test_person_inbox() {
 155     test_parse_lemmy_item::<PersonInboxActivities>("assets/lemmy/activities/following/accept.json")
 156       .unwrap();
 157     test_parse_lemmy_item::<PersonInboxActivitiesWithAnnouncable>(
 158       "assets/lemmy/activities/create_or_update/create_note.json",
 159     )
 160     .unwrap();
 161     test_parse_lemmy_item::<PersonInboxActivitiesWithAnnouncable>(
 162       "assets/lemmy/activities/create_or_update/create_private_message.json",
 163     )
 164     .unwrap();
 165   }
 166
 167   #[test]
 168   fn test_site_inbox() {
 169     test_parse_lemmy_item::<SiteInboxActivities>(
 170       "assets/lemmy/activities/deletion/delete_user.json",
 171     )
 172     .unwrap();
 173   }
 174 }