crates/apub/src/objects/mod.rs

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