2 extern crate diesel_migrations;
7 r2d2::{ConnectionManager, Pool},
10 use lemmy_api::match_websocket_operation;
11 use lemmy_apub::activity_queue::create_activity_queue;
12 use lemmy_db_queries::get_database_url_from_env;
13 use lemmy_server::{code_migrations::run_advanced_migrations, routes::*, scheduled_tasks};
14 use lemmy_structs::blocking;
16 rate_limit::{rate_limiter::RateLimiter, RateLimit},
20 use lemmy_websocket::{chat_server::ChatServer, LemmyContext};
22 use std::{sync::Arc, thread};
23 use tokio::sync::Mutex;
28 async fn main() -> Result<(), LemmyError> {
30 let settings = Settings::get();
32 // Set up the r2d2 connection pool
33 let db_url = match get_database_url_from_env() {
35 Err(_) => settings.get_database_url(),
37 let manager = ConnectionManager::<PgConnection>::new(&db_url);
38 let pool = Pool::builder()
39 .max_size(settings.database.pool_size)
41 .unwrap_or_else(|_| panic!("Error connecting to {}", db_url));
43 // Run the migrations from code
44 blocking(&pool, move |conn| {
45 embedded_migrations::run(conn)?;
46 run_advanced_migrations(conn)?;
47 Ok(()) as Result<(), LemmyError>
51 let pool2 = pool.clone();
52 thread::spawn(move || {
53 scheduled_tasks::setup(pool2);
56 // Set up the rate limiter
57 let rate_limiter = RateLimit {
58 rate_limiter: Arc::new(Mutex::new(RateLimiter::default())),
62 "Starting http server at {}:{}",
63 settings.bind, settings.port
66 let activity_queue = create_activity_queue();
67 let chat_server = ChatServer::startup(
70 |c, i, o, d| Box::pin(match_websocket_operation(c, i, o, d)),
72 activity_queue.clone(),
76 // Create Http server with websocket support
77 HttpServer::new(move || {
78 let context = LemmyContext::create(
80 chat_server.to_owned(),
82 activity_queue.to_owned(),
84 let rate_limiter = rate_limiter.clone();
86 .wrap(middleware::Logger::default())
89 .configure(|cfg| api::config(cfg, &rate_limiter))
90 .configure(federation::config)
91 .configure(feeds::config)
92 .configure(|cfg| images::config(cfg, &rate_limiter))
93 .configure(nodeinfo::config)
94 .configure(webfinger::config)
95 .service(actix_files::Files::new("/docs", Settings::get().docs_dir))
97 .bind((settings.bind, settings.port))?