-#[macro_use]
-extern crate diesel_migrations;
-
-use actix::prelude::*;
-use actix_web::{
- body::Body,
- dev::{Service, ServiceRequest, ServiceResponse},
- http::{
- header::{CACHE_CONTROL, CONTENT_TYPE},
- HeaderValue,
- },
- *,
-};
-use diesel::{
- r2d2::{ConnectionManager, Pool},
- PgConnection,
-};
-use lazy_static::lazy_static;
-use lemmy_db::get_database_url_from_env;
-use lemmy_rate_limit::{rate_limiter::RateLimiter, RateLimit};
-use lemmy_server::{
- apub::activity_queue::create_activity_queue,
- code_migrations::run_advanced_migrations,
- routes::*,
- websocket::chat_server::ChatServer,
- LemmyContext,
-};
-use lemmy_structs::blocking;
-use lemmy_utils::{settings::Settings, LemmyError, CACHE_CONTROL_REGEX};
-use reqwest::Client;
-use std::sync::Arc;
-use tokio::sync::Mutex;
-
-lazy_static! {
- // static ref CACHE_CONTROL_VALUE: String = format!("public, max-age={}", 365 * 24 * 60 * 60);
- // Test out 1 hour here, this is breaking some things
- static ref CACHE_CONTROL_VALUE: String = format!("public, max-age={}", 60 * 60);
-}
-
-embed_migrations!();
-
-#[actix_web::main]
-async fn main() -> Result<(), LemmyError> {
- env_logger::init();
- let settings = Settings::get();
-
- // Set up the r2d2 connection pool
- let db_url = match get_database_url_from_env() {
- Ok(url) => url,
- Err(_) => settings.get_database_url(),
- };
- let manager = ConnectionManager::<PgConnection>::new(&db_url);
- let pool = Pool::builder()
- .max_size(settings.database.pool_size)
- .build(manager)
- .unwrap_or_else(|_| panic!("Error connecting to {}", db_url));
-
- // Run the migrations from code
- blocking(&pool, move |conn| {
- embedded_migrations::run(conn)?;
- run_advanced_migrations(conn)?;
- Ok(()) as Result<(), LemmyError>
- })
- .await??;
-
- // Set up the rate limiter
- let rate_limiter = RateLimit {
- rate_limiter: Arc::new(Mutex::new(RateLimiter::default())),
- };
-
- println!(
- "Starting http server at {}:{}",
- settings.bind, settings.port
- );
-
- let activity_queue = create_activity_queue();
- let chat_server = ChatServer::startup(
- pool.clone(),
- rate_limiter.clone(),
- Client::default(),
- activity_queue.clone(),
- )
- .start();
-
- // Create Http server with websocket support
- HttpServer::new(move || {
- let context = LemmyContext::create(
- pool.clone(),
- chat_server.to_owned(),
- Client::default(),
- activity_queue.to_owned(),
- );
- let rate_limiter = rate_limiter.clone();
- App::new()
- .wrap_fn(add_cache_headers)
- .wrap(middleware::Logger::default())
- .data(context)
- // The routes
- .configure(|cfg| api::config(cfg, &rate_limiter))
- .configure(federation::config)
- .configure(feeds::config)
- .configure(|cfg| images::config(cfg, &rate_limiter))
- .configure(nodeinfo::config)
- .configure(webfinger::config)
- .service(actix_files::Files::new("/docs", "/app/documentation"))
- })
- .bind((settings.bind, settings.port))?
- .run()
- .await?;
-
- Ok(())
-}
-
-fn add_cache_headers<S>(
- req: ServiceRequest,
- srv: &mut S,
-) -> impl Future<Output = Result<ServiceResponse, Error>>
-where
- S: Service<Request = ServiceRequest, Response = ServiceResponse<Body>, Error = Error>,
-{
- let fut = srv.call(req);
- async move {
- let mut res = fut.await?;
- if let Some(content_type) = res.headers().get(CONTENT_TYPE) {
- if CACHE_CONTROL_REGEX.is_match(content_type.to_str().unwrap()) {
- let header_val = HeaderValue::from_static(&CACHE_CONTROL_VALUE);
- res.headers_mut().insert(CACHE_CONTROL, header_val);
- }
- }
- Ok(res)
+use lemmy_server::{init_logging, start_lemmy_server};
+use lemmy_utils::{error::LemmyError, settings::SETTINGS};
+
+#[tokio::main]
+pub async fn main() -> Result<(), LemmyError> {
+ init_logging(&SETTINGS.opentelemetry_url)?;
+ #[cfg(not(feature = "embed-pictrs"))]
+ start_lemmy_server().await?;
+ #[cfg(feature = "embed-pictrs")]
+ {
+ let pictrs_port = &SETTINGS
+ .pictrs_config()
+ .unwrap_or_default()
+ .url
+ .port()
+ .unwrap_or(8080);
+ let pictrs_address = ["127.0.0.1", &pictrs_port.to_string()].join(":");
+ pict_rs::ConfigSource::memory(serde_json::json!({
+ "server": {
+ "address": pictrs_address
+ },
+ "old_db": {
+ "path": "./pictrs/old"
+ },
+ "repo": {
+ "type": "sled",
+ "path": "./pictrs/sled-repo"
+ },
+ "store": {
+ "type": "filesystem",
+ "path": "./pictrs/files"
+ }
+ }))
+ .init::<&str>(None)
+ .expect("initialize pictrs config");
+ let (lemmy, pictrs) = tokio::join!(start_lemmy_server(), pict_rs::run());
+ lemmy?;
+ pictrs.expect("run pictrs");
}
+ Ok(())
}