use crate::{ check_is_apub_id_valid, fetcher::{get_or_fetch_and_upsert_community, get_or_fetch_and_upsert_user}, inbox::community_inbox::check_community_or_site_ban, }; use activitystreams::{ base::{AsBase, BaseExt, ExtendsExt}, markers::Base, mime::{FromStrError, Mime}, object::{ApObjectExt, Object, ObjectExt, Tombstone, TombstoneExt}, }; use anyhow::{anyhow, Context}; use chrono::NaiveDateTime; use lemmy_db::{ApubObject, Crud, DbPool}; use lemmy_structs::blocking; use lemmy_utils::{location_info, settings::Settings, utils::convert_datetime, LemmyError}; use lemmy_websocket::LemmyContext; use url::Url; pub(crate) mod comment; pub(crate) mod community; pub(crate) mod post; pub(crate) mod private_message; pub(crate) mod user; /// Trait for converting an object or actor into the respective ActivityPub type. #[async_trait::async_trait(?Send)] pub(crate) trait ToApub { type ApubType; async fn to_apub(&self, pool: &DbPool) -> Result; fn to_tombstone(&self) -> Result; } #[async_trait::async_trait(?Send)] pub(crate) trait FromApub { type ApubType; /// Converts an object from ActivityPub type to Lemmy internal type. /// /// * `apub` The object to read from /// * `context` LemmyContext which holds DB pool, HTTP client etc /// * `expected_domain` Domain where the object was received from async fn from_apub( apub: &Self::ApubType, context: &LemmyContext, expected_domain: Url, request_counter: &mut i32, ) -> Result where Self: Sized; } #[async_trait::async_trait(?Send)] pub(in crate::objects) trait FromApubToForm { async fn from_apub( apub: &ApubType, context: &LemmyContext, expected_domain: Url, request_counter: &mut i32, ) -> Result where Self: Sized; } /// Updated is actually the deletion time fn create_tombstone( deleted: bool, object_id: &str, updated: Option, former_type: T, ) -> Result where T: ToString, { if deleted { if let Some(updated) = updated { let mut tombstone = Tombstone::new(); tombstone.set_id(object_id.parse()?); tombstone.set_former_type(former_type.to_string()); tombstone.set_deleted(convert_datetime(updated)); Ok(tombstone) } else { Err(anyhow!("Cant convert to tombstone because updated time was None.").into()) } } else { Err(anyhow!("Cant convert object to tombstone if it wasnt deleted").into()) } } pub(in crate::objects) fn check_object_domain( apub: &T, expected_domain: Url, ) -> Result where T: Base + AsBase, { let domain = expected_domain.domain().context(location_info!())?; let object_id = apub.id(domain)?.context(location_info!())?; check_is_apub_id_valid(&object_id)?; Ok(object_id.to_string()) } pub(in crate::objects) fn set_content_and_source( object: &mut T, markdown_text: &str, ) -> Result<(), LemmyError> where T: ApObjectExt + ObjectExt + AsBase, { let mut source = Object::<()>::new_none_type(); source .set_content(markdown_text) .set_media_type(mime_markdown()?); object.set_source(source.into_any_base()?); // set `content` to markdown for compatibility with older Lemmy versions // TODO: change this to HTML in a while object.set_content(markdown_text); object.set_media_type(mime_markdown()?); //object.set_content(markdown_to_html(markdown_text)); Ok(()) } pub(in crate::objects) fn get_source_markdown_value( object: &T, ) -> Result, LemmyError> where T: ApObjectExt + ObjectExt + AsBase, { let content = object .content() .map(|s| s.as_single_xsd_string()) .flatten() .map(|s| s.to_string()); if content.is_some() { let source = object.source(); // updated lemmy version, read markdown from `source.content` if let Some(source) = source { let source = Object::<()>::from_any_base(source.to_owned())?.context(location_info!())?; check_is_markdown(source.media_type())?; let source_content = source .content() .map(|s| s.as_single_xsd_string()) .flatten() .context(location_info!())? .to_string(); return Ok(Some(source_content)); } // older lemmy version, read markdown from `content` // TODO: remove this after a while else { return Ok(content); } } Ok(None) } pub(in crate::objects) fn mime_markdown() -> Result { "text/markdown".parse() } pub(in crate::objects) fn check_is_markdown(mime: Option<&Mime>) -> Result<(), LemmyError> { let mime = mime.context(location_info!())?; if !mime.eq(&mime_markdown()?) { Err(LemmyError::from(anyhow!( "Lemmy only supports markdown content" ))) } else { Ok(()) } } /// Converts an ActivityPub object (eg `Note`) to a database object (eg `Comment`). If an object /// with the same ActivityPub ID already exists in the database, it is returned directly. Otherwise /// the apub object is parsed, inserted and returned. pub(in crate::objects) async fn get_object_from_apub( from: &From, context: &LemmyContext, expected_domain: Url, request_counter: &mut i32, ) -> Result where From: BaseExt, To: ApubObject + Crud + Send + 'static, ToForm: FromApubToForm + Send + 'static, { let object_id = from.id_unchecked().context(location_info!())?.to_owned(); let domain = object_id.domain().context(location_info!())?; // if its a local object, return it directly from the database if Settings::get().hostname == domain { let object = blocking(context.pool(), move |conn| { To::read_from_apub_id(conn, object_id.as_str()) }) .await??; Ok(object) } // otherwise parse and insert, assuring that it comes from the right domain else { let to_form = ToForm::from_apub(&from, context, expected_domain, request_counter).await?; let to = blocking(context.pool(), move |conn| To::upsert(conn, &to_form)).await??; Ok(to) } } pub(in crate::objects) async fn check_object_for_community_or_site_ban( object: &T, context: &LemmyContext, request_counter: &mut i32, ) -> Result<(), LemmyError> where T: ObjectExt, { let user_id = object .attributed_to() .context(location_info!())? .as_single_xsd_any_uri() .context(location_info!())?; let user = get_or_fetch_and_upsert_user(user_id, context, request_counter).await?; let community_id = object .to() .context(location_info!())? .as_single_xsd_any_uri() .context(location_info!())?; let community = get_or_fetch_and_upsert_community(community_id, context, request_counter).await?; check_community_or_site_ban(&user, &community, context.pool()).await }