use crate::{ objects::{community::ApubCommunity, person::ApubPerson}, protocol::objects::{group::Group, person::Person}, }; use activitypub_federation::{ config::Data, traits::{Actor, Object}, }; use chrono::NaiveDateTime; use lemmy_api_common::context::LemmyContext; use lemmy_utils::error::LemmyError; use serde::{Deserialize, Serialize}; use url::Url; #[derive(Clone, Debug)] pub enum UserOrCommunity { User(ApubPerson), Community(ApubCommunity), } #[derive(Serialize, Deserialize, Clone, Debug)] #[serde(untagged)] pub enum PersonOrGroup { Person(Person), Group(Group), } #[derive(Serialize, Deserialize, Clone, Debug, PartialEq, Eq)] pub enum PersonOrGroupType { Person, Group, } #[async_trait::async_trait] impl Object for UserOrCommunity { type DataType = LemmyContext; type Kind = PersonOrGroup; type Error = LemmyError; fn last_refreshed_at(&self) -> Option { Some(match self { UserOrCommunity::User(p) => p.last_refreshed_at, UserOrCommunity::Community(p) => p.last_refreshed_at, }) } #[tracing::instrument(skip_all)] async fn read_from_id( object_id: Url, data: &Data, ) -> Result, LemmyError> { let person = ApubPerson::read_from_id(object_id.clone(), data).await?; Ok(match person { Some(o) => Some(UserOrCommunity::User(o)), None => ApubCommunity::read_from_id(object_id, data) .await? .map(UserOrCommunity::Community), }) } #[tracing::instrument(skip_all)] async fn delete(self, data: &Data) -> Result<(), LemmyError> { match self { UserOrCommunity::User(p) => p.delete(data).await, UserOrCommunity::Community(p) => p.delete(data).await, } } async fn into_json(self, _data: &Data) -> Result { unimplemented!() } #[tracing::instrument(skip_all)] async fn verify( apub: &Self::Kind, expected_domain: &Url, data: &Data, ) -> Result<(), LemmyError> { match apub { PersonOrGroup::Person(a) => ApubPerson::verify(a, expected_domain, data).await, PersonOrGroup::Group(a) => ApubCommunity::verify(a, expected_domain, data).await, } } #[tracing::instrument(skip_all)] async fn from_json(apub: Self::Kind, data: &Data) -> Result { Ok(match apub { PersonOrGroup::Person(p) => UserOrCommunity::User(ApubPerson::from_json(p, data).await?), PersonOrGroup::Group(p) => { UserOrCommunity::Community(ApubCommunity::from_json(p, data).await?) } }) } } impl Actor for UserOrCommunity { fn id(&self) -> Url { match self { UserOrCommunity::User(u) => u.id(), UserOrCommunity::Community(c) => c.id(), } } fn public_key_pem(&self) -> &str { match self { UserOrCommunity::User(p) => p.public_key_pem(), UserOrCommunity::Community(p) => p.public_key_pem(), } } fn private_key_pem(&self) -> Option { match self { UserOrCommunity::User(p) => p.private_key_pem(), UserOrCommunity::Community(p) => p.private_key_pem(), } } fn inbox(&self) -> Url { unimplemented!() } }