1 use activitystreams::object::kind::ImageType;
2 use html2md::parse_html;
3 use lemmy_apub_lib::values::MediaTypeMarkdown;
4 use serde::{Deserialize, Serialize};
11 pub mod private_message;
14 #[derive(Clone, Debug, Deserialize, Serialize)]
15 #[serde(rename_all = "camelCase")]
18 media_type: MediaTypeMarkdown,
21 #[derive(Clone, Debug, Deserialize, Serialize)]
22 #[serde(rename_all = "camelCase")]
23 pub struct ImageObject {
24 #[serde(rename = "type")]
29 fn get_summary_from_string_or_source(
31 source: &Option<Source>,
33 if let Some(source) = &source {
34 Some(source.content.clone())
36 raw.as_ref().map(|s| parse_html(s))
44 r2d2::{ConnectionManager, Pool},
47 use lemmy_apub_lib::activity_queue::create_activity_queue;
48 use lemmy_db_schema::{
49 establish_unpooled_connection,
50 get_database_url_from_env,
51 source::secret::Secret,
54 rate_limit::{rate_limiter::RateLimiter, RateLimit},
55 request::build_user_agent,
56 settings::structs::Settings,
59 use lemmy_websocket::{chat_server::ChatServer, LemmyContext};
61 use serde::de::DeserializeOwned;
62 use std::{fs::File, io::BufReader, sync::Arc};
63 use tokio::sync::Mutex;
65 // TODO: would be nice if we didnt have to use a full context for tests.
66 // or at least write a helper function so this code is shared with main.rs
67 pub(crate) fn init_context() -> LemmyContext {
68 // call this to run migrations
69 establish_unpooled_connection();
70 let settings = Settings::init().unwrap();
71 let rate_limiter = RateLimit {
72 rate_limiter: Arc::new(Mutex::new(RateLimiter::default())),
73 rate_limit_config: settings.rate_limit.to_owned().unwrap_or_default(),
75 let client = Client::builder()
76 .user_agent(build_user_agent(&settings))
79 let activity_queue = create_activity_queue();
82 jwt_secret: "".to_string(),
84 let db_url = match get_database_url_from_env() {
86 Err(_) => settings.get_database_url(),
88 let manager = ConnectionManager::<PgConnection>::new(&db_url);
89 let pool = Pool::builder()
90 .max_size(settings.database.pool_size)
92 .unwrap_or_else(|_| panic!("Error connecting to {}", db_url));
93 async fn x() -> Result<String, LemmyError> {
96 let chat_server = ChatServer::startup(
99 |_, _, _, _| Box::pin(x()),
100 |_, _, _, _| Box::pin(x()),
102 activity_queue.clone(),
107 LemmyContext::create(pool, chat_server, client, activity_queue, settings, secret)
110 pub(crate) fn file_to_json_object<T: DeserializeOwned>(path: &str) -> T {
111 let file = File::open(path).unwrap();
112 let reader = BufReader::new(file);
113 serde_json::from_reader(reader).unwrap()