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