crates/apub/src/objects/private_message.rs

   1 use crate::{
   2   check_is_apub_id_valid,
   3   extensions::context::lemmy_context,
   4   fetcher::user::get_or_fetch_and_upsert_user,
   5   objects::{
   6     check_object_domain,
   7     create_tombstone,
   8     get_object_from_apub,
   9     get_source_markdown_value,
  10     set_content_and_source,
  11     FromApub,
  12     FromApubToForm,
  13     ToApub,
  14   },
  15   NoteExt,
  16 };
  17 use activitystreams::{
  18   object::{kind::NoteType, ApObject, Note, Tombstone},
  19   prelude::*,
  20 };
  21 use anyhow::Context;
  22 use lemmy_db_queries::{Crud, DbPool};
  23 use lemmy_db_schema::source::{
  24   private_message::{PrivateMessage, PrivateMessageForm},
  25   user::User_,
  26 };
  27 use lemmy_structs::blocking;
  28 use lemmy_utils::{location_info, utils::convert_datetime, LemmyError};
  29 use lemmy_websocket::LemmyContext;
  30 use url::Url;
  31
  32 #[async_trait::async_trait(?Send)]
  33 impl ToApub for PrivateMessage {
  34   type ApubType = NoteExt;
  35
  36   async fn to_apub(&self, pool: &DbPool) -> Result<NoteExt, LemmyError> {
  37     let mut private_message = ApObject::new(Note::new());
  38
  39     let creator_id = self.creator_id;
  40     let creator = blocking(pool, move |conn| User_::read(conn, creator_id)).await??;
  41
  42     let recipient_id = self.recipient_id;
  43     let recipient = blocking(pool, move |conn| User_::read(conn, recipient_id)).await??;
  44
  45     private_message
  46       .set_many_contexts(lemmy_context()?)
  47       .set_id(Url::parse(&self.ap_id.to_owned())?)
  48       .set_published(convert_datetime(self.published))
  49       .set_to(recipient.actor_id)
  50       .set_attributed_to(creator.actor_id);
  51
  52     set_content_and_source(&mut private_message, &self.content)?;
  53
  54     if let Some(u) = self.updated {
  55       private_message.set_updated(convert_datetime(u));
  56     }
  57
  58     Ok(private_message)
  59   }
  60
  61   fn to_tombstone(&self) -> Result<Tombstone, LemmyError> {
  62     create_tombstone(self.deleted, &self.ap_id, self.updated, NoteType::Note)
  63   }
  64 }
  65
  66 #[async_trait::async_trait(?Send)]
  67 impl FromApub for PrivateMessage {
  68   type ApubType = NoteExt;
  69
  70   async fn from_apub(
  71     note: &NoteExt,
  72     context: &LemmyContext,
  73     expected_domain: Url,
  74     request_counter: &mut i32,
  75   ) -> Result<PrivateMessage, LemmyError> {
  76     get_object_from_apub(note, context, expected_domain, request_counter).await
  77   }
  78 }
  79
  80 #[async_trait::async_trait(?Send)]
  81 impl FromApubToForm<NoteExt> for PrivateMessageForm {
  82   async fn from_apub(
  83     note: &NoteExt,
  84     context: &LemmyContext,
  85     expected_domain: Url,
  86     request_counter: &mut i32,
  87   ) -> Result<PrivateMessageForm, LemmyError> {
  88     let creator_actor_id = note
  89       .attributed_to()
  90       .context(location_info!())?
  91       .clone()
  92       .single_xsd_any_uri()
  93       .context(location_info!())?;
  94
  95     let creator = get_or_fetch_and_upsert_user(&creator_actor_id, context, request_counter).await?;
  96     let recipient_actor_id = note
  97       .to()
  98       .context(location_info!())?
  99       .clone()
 100       .single_xsd_any_uri()
 101       .context(location_info!())?;
 102     let recipient =
 103       get_or_fetch_and_upsert_user(&recipient_actor_id, context, request_counter).await?;
 104     let ap_id = note.id_unchecked().context(location_info!())?.to_string();
 105     check_is_apub_id_valid(&Url::parse(&ap_id)?)?;
 106
 107     let content = get_source_markdown_value(note)?.context(location_info!())?;
 108
 109     Ok(PrivateMessageForm {
 110       creator_id: creator.id,
 111       recipient_id: recipient.id,
 112       content,
 113       published: note.published().map(|u| u.to_owned().naive_local()),
 114       updated: note.updated().map(|u| u.to_owned().naive_local()),
 115       deleted: None,
 116       read: None,
 117       ap_id: Some(check_object_domain(note, expected_domain)?),
 118       local: false,
 119     })
 120   }
 121 }