crates/apub/src/objects/mod.rs

   1 use crate::protocol::{ImageObject, Source};
   2 use html2md::parse_html;
   3 use lemmy_apub_lib::verify::verify_domains_match;
   4 use lemmy_utils::LemmyError;
   5 use url::Url;
   6
   7 pub mod comment;
   8 pub mod community;
   9 pub mod instance;
  10 pub mod person;
  11 pub mod post;
  12 pub mod private_message;
  13
  14 pub(crate) fn get_summary_from_string_or_source(
  15   raw: &Option<String>,
  16   source: &Option<Source>,
  17 ) -> Option<String> {
  18   if let Some(source) = &source {
  19     Some(source.content.clone())
  20   } else {
  21     raw.as_ref().map(|s| parse_html(s))
  22   }
  23 }
  24
  25 pub fn verify_image_domain_matches(a: &Url, b: &Option<ImageObject>) -> Result<(), LemmyError> {
  26   if let Some(b) = b {
  27     verify_domains_match(a, &b.url)
  28   } else {
  29     Ok(())
  30   }
  31 }
  32
  33 #[cfg(test)]
  34 pub(crate) mod tests {
  35   use actix::Actor;
  36   use diesel::{
  37     r2d2::{ConnectionManager, Pool},
  38     PgConnection,
  39   };
  40   use lemmy_apub_lib::activity_queue::create_activity_queue;
  41   use lemmy_db_schema::{
  42     establish_unpooled_connection,
  43     get_database_url_from_env,
  44     source::secret::Secret,
  45   };
  46   use lemmy_utils::{
  47     rate_limit::{rate_limiter::RateLimiter, RateLimit},
  48     request::build_user_agent,
  49     settings::structs::Settings,
  50     LemmyError,
  51   };
  52   use lemmy_websocket::{chat_server::ChatServer, LemmyContext};
  53   use reqwest::Client;
  54   use reqwest_middleware::ClientBuilder;
  55   use std::sync::Arc;
  56   use tokio::sync::Mutex;
  57
  58   // TODO: would be nice if we didnt have to use a full context for tests.
  59   //       or at least write a helper function so this code is shared with main.rs
  60   pub(crate) fn init_context() -> LemmyContext {
  61     let client = reqwest::Client::new().into();
  62     // activity queue isnt used in tests, so worker count makes no difference
  63     let queue_manager = create_activity_queue(client, 4);
  64     let activity_queue = queue_manager.queue_handle().clone();
  65     // call this to run migrations
  66     establish_unpooled_connection();
  67     let settings = Settings::init().unwrap();
  68     let rate_limiter = RateLimit {
  69       rate_limiter: Arc::new(Mutex::new(RateLimiter::default())),
  70       rate_limit_config: settings.rate_limit.to_owned().unwrap_or_default(),
  71     };
  72     let client = Client::builder()
  73       .user_agent(build_user_agent(&settings))
  74       .build()
  75       .unwrap();
  76
  77     let client = ClientBuilder::new(client).build();
  78     let secret = Secret {
  79       id: 0,
  80       jwt_secret: "".to_string(),
  81     };
  82     let db_url = match get_database_url_from_env() {
  83       Ok(url) => url,
  84       Err(_) => settings.get_database_url(),
  85     };
  86     let manager = ConnectionManager::<PgConnection>::new(&db_url);
  87     let pool = Pool::builder()
  88       .max_size(settings.database.pool_size)
  89       .build(manager)
  90       .unwrap_or_else(|_| panic!("Error connecting to {}", db_url));
  91     async fn x() -> Result<String, LemmyError> {
  92       Ok("".to_string())
  93     }
  94     let chat_server = ChatServer::startup(
  95       pool.clone(),
  96       rate_limiter,
  97       |_, _, _, _| Box::pin(x()),
  98       |_, _, _, _| Box::pin(x()),
  99       client.clone(),
 100       activity_queue.clone(),
 101       settings.clone(),
 102       secret.clone(),
 103     )
 104     .start();
 105     LemmyContext::create(pool, chat_server, client, activity_queue, settings, secret)
 106   }
 107 }