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