crates/apub/src/objects/mod.rs

   1 use crate::protocol::Source;
   2 use activitypub_federation::deser::values::MediaTypeMarkdownOrHtml;
   3 use anyhow::anyhow;
   4 use html2md::parse_html;
   5 use lemmy_utils::{error::LemmyError, settings::structs::Settings};
   6 use url::Url;
   7
   8 pub mod comment;
   9 pub mod community;
  10 pub mod instance;
  11 pub mod person;
  12 pub mod post;
  13 pub mod private_message;
  14
  15 pub(crate) fn read_from_string_or_source(
  16   content: &str,
  17   media_type: &Option<MediaTypeMarkdownOrHtml>,
  18   source: &Option<Source>,
  19 ) -> String {
  20   if let Some(s) = source {
  21     // markdown sent by lemmy in source field
  22     s.content.clone()
  23   } else if media_type == &Some(MediaTypeMarkdownOrHtml::Markdown) {
  24     // markdown sent by peertube in content field
  25     content.to_string()
  26   } else {
  27     // otherwise, convert content html to markdown
  28     parse_html(content)
  29   }
  30 }
  31
  32 pub(crate) fn read_from_string_or_source_opt(
  33   content: &Option<String>,
  34   media_type: &Option<MediaTypeMarkdownOrHtml>,
  35   source: &Option<Source>,
  36 ) -> Option<String> {
  37   content
  38     .as_ref()
  39     .map(|content| read_from_string_or_source(content, media_type, source))
  40 }
  41
  42 /// When for example a Post is made in a remote community, the community will send it back,
  43 /// wrapped in Announce. If we simply receive this like any other federated object, overwrite the
  44 /// existing, local Post. In particular, it will set the field local = false, so that the object
  45 /// can't be fetched from the Activitypub HTTP endpoint anymore (which only serves local objects).
  46 pub(crate) fn verify_is_remote_object(id: &Url, settings: &Settings) -> Result<(), LemmyError> {
  47   let local_domain = settings.get_hostname_without_port()?;
  48   if id.domain() == Some(&local_domain) {
  49     Err(anyhow!("cant accept local object from remote instance").into())
  50   } else {
  51     Ok(())
  52   }
  53 }
  54
  55 #[cfg(test)]
  56 pub(crate) mod tests {
  57   use actix::Actor;
  58   use anyhow::anyhow;
  59   use lemmy_api_common::{
  60     request::build_user_agent,
  61     websocket::chat_server::ChatServer,
  62     LemmyContext,
  63   };
  64   use lemmy_db_schema::{source::secret::Secret, utils::build_db_pool_for_tests};
  65   use lemmy_utils::{
  66     error::LemmyError,
  67     rate_limit::{RateLimitCell, RateLimitConfig},
  68     settings::SETTINGS,
  69   };
  70   use reqwest::{Client, Request, Response};
  71   use reqwest_middleware::{ClientBuilder, Middleware, Next};
  72   use task_local_extensions::Extensions;
  73
  74   struct BlockedMiddleware;
  75
  76   /// A reqwest middleware which blocks all requests
  77   #[async_trait::async_trait]
  78   impl Middleware for BlockedMiddleware {
  79     async fn handle(
  80       &self,
  81       _req: Request,
  82       _extensions: &mut Extensions,
  83       _next: Next<'_>,
  84     ) -> reqwest_middleware::Result<Response> {
  85       Err(anyhow!("Network requests not allowed").into())
  86     }
  87   }
  88
  89   // TODO: would be nice if we didnt have to use a full context for tests.
  90   pub(crate) async fn init_context() -> LemmyContext {
  91     async fn x() -> Result<String, LemmyError> {
  92       Ok(String::new())
  93     }
  94     // call this to run migrations
  95     let pool = build_db_pool_for_tests().await;
  96
  97     let settings = SETTINGS.clone();
  98     let client = Client::builder()
  99       .user_agent(build_user_agent(&settings))
 100       .build()
 101       .unwrap();
 102
 103     let client = ClientBuilder::new(client).with(BlockedMiddleware).build();
 104     let secret = Secret {
 105       id: 0,
 106       jwt_secret: String::new(),
 107     };
 108
 109     let rate_limit_config = RateLimitConfig::builder().build();
 110     let rate_limit_cell = RateLimitCell::new(rate_limit_config).await;
 111
 112     let chat_server = ChatServer::startup(
 113       pool.clone(),
 114       |_, _, _, _| Box::pin(x()),
 115       |_, _, _, _| Box::pin(x()),
 116       client.clone(),
 117       settings.clone(),
 118       secret.clone(),
 119       rate_limit_cell.clone(),
 120     )
 121     .start();
 122     LemmyContext::create(
 123       pool,
 124       chat_server,
 125       client,
 126       settings,
 127       secret,
 128       rate_limit_cell.clone(),
 129     )
 130   }
 131 }