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_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   CreateOrUpdatePrivateMessage(CreateOrUpdateChatMessage),
  58   Delete(Delete),
  59   UndoDelete(UndoDelete),
  60   AnnounceActivity(AnnounceActivity),
  61 }
  62
  63 /// This is necessary for user inbox, which can also receive some "announcable" activities,
  64 /// eg a comment mention. This needs to be a separate enum so that announcables received in shared
  65 /// inbox can fall through to be parsed as GroupInboxActivities::AnnouncableActivities.
  66 #[derive(Clone, Debug, Deserialize, Serialize)]
  67 #[serde(untagged)]
  68 #[enum_delegate::implement(ActivityHandler)]
  69 pub enum PersonInboxActivitiesWithAnnouncable {
  70   PersonInboxActivities(Box<PersonInboxActivities>),
  71   AnnouncableActivities(Box<AnnouncableActivities>),
  72 }
  73
  74 #[derive(Clone, Debug, Deserialize, Serialize)]
  75 #[serde(untagged)]
  76 #[enum_delegate::implement(ActivityHandler)]
  77 pub enum AnnouncableActivities {
  78   CreateOrUpdateComment(CreateOrUpdateNote),
  79   CreateOrUpdatePost(CreateOrUpdatePage),
  80   Vote(Vote),
  81   UndoVote(UndoVote),
  82   Delete(Delete),
  83   UndoDelete(UndoDelete),
  84   UpdateCommunity(UpdateCommunity),
  85   BlockUser(BlockUser),
  86   UndoBlockUser(UndoBlockUser),
  87   AddMod(AddMod),
  88   RemoveMod(RemoveMod),
  89   // For compatibility with Pleroma/Mastodon (send only)
  90   Page(Page),
  91 }
  92
  93 #[derive(Clone, Debug, Deserialize, Serialize)]
  94 #[serde(untagged)]
  95 #[enum_delegate::implement(ActivityHandler)]
  96 #[allow(clippy::enum_variant_names)]
  97 pub enum SiteInboxActivities {
  98   BlockUser(BlockUser),
  99   UndoBlockUser(UndoBlockUser),
 100   DeleteUser(DeleteUser),
 101 }
 102
 103 #[async_trait::async_trait(?Send)]
 104 impl InCommunity for AnnouncableActivities {
 105   #[tracing::instrument(skip(self, context))]
 106   async fn community(
 107     &self,
 108     context: &LemmyContext,
 109     request_counter: &mut i32,
 110   ) -> Result<ApubCommunity, LemmyError> {
 111     use AnnouncableActivities::*;
 112     match self {
 113       CreateOrUpdateComment(a) => a.community(context, request_counter).await,
 114       CreateOrUpdatePost(a) => a.community(context, request_counter).await,
 115       Vote(a) => a.community(context, request_counter).await,
 116       UndoVote(a) => a.community(context, request_counter).await,
 117       Delete(a) => a.community(context, request_counter).await,
 118       UndoDelete(a) => a.community(context, request_counter).await,
 119       UpdateCommunity(a) => a.community(context, request_counter).await,
 120       BlockUser(a) => a.community(context, request_counter).await,
 121       UndoBlockUser(a) => a.community(context, request_counter).await,
 122       AddMod(a) => a.community(context, request_counter).await,
 123       RemoveMod(a) => a.community(context, request_counter).await,
 124       Page(_) => unimplemented!(),
 125     }
 126   }
 127 }
 128
 129 #[cfg(test)]
 130 mod tests {
 131   use crate::{
 132     activity_lists::{
 133       GroupInboxActivities,
 134       PersonInboxActivities,
 135       PersonInboxActivitiesWithAnnouncable,
 136       SiteInboxActivities,
 137     },
 138     protocol::tests::test_parse_lemmy_item,
 139   };
 140
 141   #[test]
 142   fn test_group_inbox() {
 143     test_parse_lemmy_item::<GroupInboxActivities>("assets/lemmy/activities/following/follow.json")
 144       .unwrap();
 145     test_parse_lemmy_item::<GroupInboxActivities>(
 146       "assets/lemmy/activities/create_or_update/create_note.json",
 147     )
 148     .unwrap();
 149   }
 150
 151   #[test]
 152   fn test_person_inbox() {
 153     test_parse_lemmy_item::<PersonInboxActivities>("assets/lemmy/activities/following/accept.json")
 154       .unwrap();
 155     test_parse_lemmy_item::<PersonInboxActivitiesWithAnnouncable>(
 156       "assets/lemmy/activities/create_or_update/create_note.json",
 157     )
 158     .unwrap();
 159     test_parse_lemmy_item::<PersonInboxActivitiesWithAnnouncable>(
 160       "assets/lemmy/activities/create_or_update/create_private_message.json",
 161     )
 162     .unwrap();
 163   }
 164
 165   #[test]
 166   fn test_site_inbox() {
 167     test_parse_lemmy_item::<SiteInboxActivities>(
 168       "assets/lemmy/activities/deletion/delete_user.json",
 169     )
 170     .unwrap();
 171   }
 172 }