2 extensions::context::lemmy_context,
3 fetcher::person::get_or_fetch_and_upsert_person,
4 objects::{create_tombstone, FromApub, Source, ToApub},
8 object::{kind::NoteType, Tombstone},
13 use chrono::{DateTime, FixedOffset};
14 use lemmy_api_common::blocking;
16 values::{MediaTypeHtml, MediaTypeMarkdown},
19 use lemmy_db_queries::{ApubObject, Crud, DbPool};
20 use lemmy_db_schema::source::{
22 private_message::{PrivateMessage, PrivateMessageForm},
24 use lemmy_utils::{utils::convert_datetime, LemmyError};
25 use lemmy_websocket::LemmyContext;
26 use serde::{Deserialize, Serialize};
27 use serde_with::skip_serializing_none;
30 #[skip_serializing_none]
31 #[derive(Clone, Debug, Deserialize, Serialize)]
32 #[serde(rename_all = "camelCase")]
34 #[serde(rename = "@context")]
35 context: OneOrMany<AnyBase>,
38 pub(crate) attributed_to: Url,
41 media_type: MediaTypeHtml,
43 published: DateTime<FixedOffset>,
44 updated: Option<DateTime<FixedOffset>>,
50 pub(crate) fn id_unchecked(&self) -> &Url {
53 pub(crate) fn id(&self, expected_domain: &Url) -> Result<&Url, LemmyError> {
54 verify_domains_match(&self.id, expected_domain)?;
58 pub(crate) async fn verify(
60 context: &LemmyContext,
61 request_counter: &mut i32,
62 ) -> Result<(), LemmyError> {
63 verify_domains_match(&self.attributed_to, &self.id)?;
65 get_or_fetch_and_upsert_person(&self.attributed_to, context, request_counter).await?;
67 return Err(anyhow!("Person is banned from site").into());
73 #[async_trait::async_trait(?Send)]
74 impl ToApub for PrivateMessage {
77 async fn to_apub(&self, pool: &DbPool) -> Result<Note, LemmyError> {
78 let creator_id = self.creator_id;
79 let creator = blocking(pool, move |conn| Person::read(conn, creator_id)).await??;
81 let recipient_id = self.recipient_id;
82 let recipient = blocking(pool, move |conn| Person::read(conn, recipient_id)).await??;
85 context: lemmy_context(),
86 r#type: NoteType::Note,
87 id: self.ap_id.clone().into(),
88 attributed_to: creator.actor_id.into_inner(),
89 to: recipient.actor_id.into(),
90 content: self.content.clone(),
91 media_type: MediaTypeHtml::Html,
93 content: self.content.clone(),
94 media_type: MediaTypeMarkdown::Markdown,
96 published: convert_datetime(self.published),
97 updated: self.updated.map(convert_datetime),
98 unparsed: Default::default(),
103 fn to_tombstone(&self) -> Result<Tombstone, LemmyError> {
106 self.ap_id.to_owned().into(),
113 #[async_trait::async_trait(?Send)]
114 impl FromApub for PrivateMessage {
115 type ApubType = Note;
119 context: &LemmyContext,
120 expected_domain: &Url,
121 request_counter: &mut i32,
122 ) -> Result<PrivateMessage, LemmyError> {
123 let ap_id = Some(note.id(expected_domain)?.clone().into());
125 get_or_fetch_and_upsert_person(¬e.attributed_to, context, request_counter).await?;
126 let recipient = get_or_fetch_and_upsert_person(¬e.to, context, request_counter).await?;
128 let form = PrivateMessageForm {
129 creator_id: creator.id,
130 recipient_id: recipient.id,
131 content: note.source.content.clone(),
132 published: Some(note.published.naive_local()),
133 updated: note.updated.map(|u| u.to_owned().naive_local()),
140 blocking(context.pool(), move |conn| {
141 PrivateMessage::upsert(conn, &form)