crates/apub/src/objects/mod.rs

   1 use activitystreams::{
   2   base::BaseExt,
   3   object::{kind::ImageType, Tombstone, TombstoneExt},
   4 };
   5 use anyhow::anyhow;
   6 use chrono::NaiveDateTime;
   7 use lemmy_apub_lib::values::MediaTypeMarkdown;
   8 use lemmy_utils::{utils::convert_datetime, LemmyError};
   9 use url::Url;
  10
  11 pub mod comment;
  12 pub mod community;
  13 pub mod person;
  14 pub mod post;
  15 pub mod private_message;
  16
  17 #[derive(Clone, Debug, serde::Deserialize, serde::Serialize)]
  18 #[serde(rename_all = "camelCase")]
  19 pub struct Source {
  20   content: String,
  21   media_type: MediaTypeMarkdown,
  22 }
  23
  24 #[derive(Clone, Debug, serde::Deserialize, serde::Serialize)]
  25 #[serde(rename_all = "camelCase")]
  26 pub struct ImageObject {
  27   #[serde(rename = "type")]
  28   kind: ImageType,
  29   url: Url,
  30 }
  31
  32 /// Updated is actually the deletion time
  33 fn create_tombstone<T>(
  34   deleted: bool,
  35   object_id: Url,
  36   updated: Option<NaiveDateTime>,
  37   former_type: T,
  38 ) -> Result<Tombstone, LemmyError>
  39 where
  40   T: ToString,
  41 {
  42   if deleted {
  43     if let Some(updated) = updated {
  44       let mut tombstone = Tombstone::new();
  45       tombstone.set_id(object_id);
  46       tombstone.set_former_type(former_type.to_string());
  47       tombstone.set_deleted(convert_datetime(updated));
  48       Ok(tombstone)
  49     } else {
  50       Err(anyhow!("Cant convert to tombstone because updated time was None.").into())
  51     }
  52   } else {
  53     Err(anyhow!("Cant convert object to tombstone if it wasnt deleted").into())
  54   }
  55 }
  56
  57 #[cfg(test)]
  58 mod tests {
  59   use super::*;
  60   use actix::Actor;
  61   use diesel::{
  62     r2d2::{ConnectionManager, Pool},
  63     PgConnection,
  64   };
  65   use lemmy_apub_lib::activity_queue::create_activity_queue;
  66   use lemmy_db_schema::{
  67     establish_unpooled_connection,
  68     get_database_url_from_env,
  69     source::secret::Secret,
  70   };
  71   use lemmy_utils::{
  72     rate_limit::{rate_limiter::RateLimiter, RateLimit},
  73     request::build_user_agent,
  74     settings::structs::Settings,
  75   };
  76   use lemmy_websocket::{chat_server::ChatServer, LemmyContext};
  77   use reqwest::Client;
  78   use serde::de::DeserializeOwned;
  79   use std::{fs::File, io::BufReader, sync::Arc};
  80   use tokio::sync::Mutex;
  81
  82   // TODO: would be nice if we didnt have to use a full context for tests.
  83   //       or at least write a helper function so this code is shared with main.rs
  84   pub(crate) fn init_context() -> LemmyContext {
  85     // call this to run migrations
  86     establish_unpooled_connection();
  87     let settings = Settings::init().unwrap();
  88     let rate_limiter = RateLimit {
  89       rate_limiter: Arc::new(Mutex::new(RateLimiter::default())),
  90       rate_limit_config: settings.rate_limit.to_owned().unwrap_or_default(),
  91     };
  92     let client = Client::builder()
  93       .user_agent(build_user_agent(&settings))
  94       .build()
  95       .unwrap();
  96     let activity_queue = create_activity_queue();
  97     let secret = Secret {
  98       id: 0,
  99       jwt_secret: "".to_string(),
 100     };
 101     let db_url = match get_database_url_from_env() {
 102       Ok(url) => url,
 103       Err(_) => settings.get_database_url(),
 104     };
 105     let manager = ConnectionManager::<PgConnection>::new(&db_url);
 106     let pool = Pool::builder()
 107       .max_size(settings.database.pool_size)
 108       .build(manager)
 109       .unwrap_or_else(|_| panic!("Error connecting to {}", db_url));
 110     async fn x() -> Result<String, LemmyError> {
 111       Ok("".to_string())
 112     }
 113     let chat_server = ChatServer::startup(
 114       pool.clone(),
 115       rate_limiter,
 116       |_, _, _, _| Box::pin(x()),
 117       |_, _, _, _| Box::pin(x()),
 118       client.clone(),
 119       activity_queue.clone(),
 120       settings.clone(),
 121       secret.clone(),
 122     )
 123     .start();
 124     LemmyContext::create(pool, chat_server, client, activity_queue, settings, secret)
 125   }
 126
 127   pub(crate) fn file_to_json_object<T: DeserializeOwned>(path: &str) -> T {
 128     let file = File::open(path).unwrap();
 129     let reader = BufReader::new(file);
 130     serde_json::from_reader(reader).unwrap()
 131   }
 132 }