src/main.rs

   1 #[macro_use]
   2 extern crate diesel_migrations;
   3
   4 use actix::prelude::*;
   5 use actix_web::*;
   6 use diesel::{
   7   r2d2::{ConnectionManager, Pool},
   8   PgConnection,
   9 };
  10 use lemmy_api::match_websocket_operation;
  11 use lemmy_api_common::blocking;
  12 use lemmy_apub::activity_queue::create_activity_queue;
  13 use lemmy_db_queries::get_database_url_from_env;
  14 use lemmy_routes::{feeds, images, nodeinfo, webfinger};
  15 use lemmy_server::{code_migrations::run_advanced_migrations, scheduled_tasks};
  16 use lemmy_utils::{
  17   rate_limit::{rate_limiter::RateLimiter, RateLimit},
  18   settings::structs::Settings,
  19   LemmyError,
  20 };
  21 use lemmy_websocket::{chat_server::ChatServer, LemmyContext};
  22 use reqwest::Client;
  23 use std::{sync::Arc, thread};
  24 use tokio::sync::Mutex;
  25
  26 embed_migrations!();
  27
  28 #[actix_web::main]
  29 async fn main() -> Result<(), LemmyError> {
  30   env_logger::init();
  31   let settings = Settings::get();
  32
  33   // Set up the r2d2 connection pool
  34   let db_url = match get_database_url_from_env() {
  35     Ok(url) => url,
  36     Err(_) => settings.get_database_url(),
  37   };
  38   let manager = ConnectionManager::<PgConnection>::new(&db_url);
  39   let pool = Pool::builder()
  40     .max_size(settings.database().pool_size)
  41     .build(manager)
  42     .unwrap_or_else(|_| panic!("Error connecting to {}", db_url));
  43
  44   // Run the migrations from code
  45   blocking(&pool, move |conn| {
  46     embedded_migrations::run(conn)?;
  47     run_advanced_migrations(conn)?;
  48     Ok(()) as Result<(), LemmyError>
  49   })
  50   .await??;
  51
  52   let pool2 = pool.clone();
  53   thread::spawn(move || {
  54     scheduled_tasks::setup(pool2);
  55   });
  56
  57   // Set up the rate limiter
  58   let rate_limiter = RateLimit {
  59     rate_limiter: Arc::new(Mutex::new(RateLimiter::default())),
  60   };
  61
  62   println!(
  63     "Starting http server at {}:{}",
  64     settings.bind(),
  65     settings.port()
  66   );
  67
  68   let activity_queue = create_activity_queue();
  69   let chat_server = ChatServer::startup(
  70     pool.clone(),
  71     rate_limiter.clone(),
  72     |c, i, o, d| Box::pin(match_websocket_operation(c, i, o, d)),
  73     Client::default(),
  74     activity_queue.clone(),
  75   )
  76   .start();
  77
  78   // Create Http server with websocket support
  79   HttpServer::new(move || {
  80     let context = LemmyContext::create(
  81       pool.clone(),
  82       chat_server.to_owned(),
  83       Client::default(),
  84       activity_queue.to_owned(),
  85     );
  86     let rate_limiter = rate_limiter.clone();
  87     App::new()
  88       .wrap(middleware::Logger::default())
  89       .data(context)
  90       // The routes
  91       .configure(|cfg| lemmy_api_crud::routes::config(cfg, &rate_limiter))
  92       .configure(|cfg| lemmy_api::routes::config(cfg, &rate_limiter))
  93       .configure(lemmy_apub::routes::config)
  94       .configure(feeds::config)
  95       .configure(|cfg| images::config(cfg, &rate_limiter))
  96       .configure(nodeinfo::config)
  97       .configure(webfinger::config)
  98   })
  99   .bind((settings.bind(), settings.port()))?
 100   .run()
 101   .await?;
 102
 103   Ok(())
 104 }