2 objects::{comment::ApubComment, post::ApubPost},
3 protocol::objects::{note::Note, page::Page},
5 use chrono::NaiveDateTime;
6 use lemmy_apub_lib::traits::ApubObject;
7 use lemmy_utils::LemmyError;
8 use lemmy_websocket::LemmyContext;
9 use serde::Deserialize;
12 #[derive(Clone, Debug)]
13 pub enum PostOrComment {
15 Comment(Box<ApubComment>),
18 #[derive(Deserialize)]
25 #[async_trait::async_trait(?Send)]
26 impl ApubObject for PostOrComment {
27 type DataType = LemmyContext;
28 type ApubType = PageOrNote;
29 type TombstoneType = ();
31 fn last_refreshed_at(&self) -> Option<NaiveDateTime> {
35 // TODO: this can probably be implemented using a single sql query
36 async fn read_from_apub_id(
38 data: &Self::DataType,
39 ) -> Result<Option<Self>, LemmyError> {
40 let post = ApubPost::read_from_apub_id(object_id.clone(), data).await?;
42 Some(o) => Some(PostOrComment::Post(Box::new(o))),
43 None => ApubComment::read_from_apub_id(object_id, data)
45 .map(|c| PostOrComment::Comment(Box::new(c))),
49 async fn delete(self, data: &Self::DataType) -> Result<(), LemmyError> {
51 PostOrComment::Post(p) => p.delete(data).await,
52 PostOrComment::Comment(c) => c.delete(data).await,
56 async fn into_apub(self, _data: &Self::DataType) -> Result<Self::ApubType, LemmyError> {
60 fn to_tombstone(&self) -> Result<Self::TombstoneType, LemmyError> {
65 apub: &Self::ApubType,
66 expected_domain: &Url,
67 data: &Self::DataType,
68 request_counter: &mut i32,
69 ) -> Result<(), LemmyError> {
71 PageOrNote::Page(a) => ApubPost::verify(a, expected_domain, data, request_counter).await,
72 PageOrNote::Note(a) => ApubComment::verify(a, expected_domain, data, request_counter).await,
78 context: &LemmyContext,
79 request_counter: &mut i32,
80 ) -> Result<Self, LemmyError> {
82 PageOrNote::Page(p) => PostOrComment::Post(Box::new(
83 ApubPost::from_apub(p, context, request_counter).await?,
85 PageOrNote::Note(n) => PostOrComment::Comment(Box::new(
86 ApubComment::from_apub(n, context, request_counter).await?,
93 pub(crate) fn ap_id(&self) -> Url {
95 PostOrComment::Post(p) => p.ap_id.clone(),
96 PostOrComment::Comment(c) => c.ap_id.clone(),