relay/src/main.rs

use actix_rt::Arbiter;
use actix_web::{
    middleware::{Compress, Logger},
    web, App, HttpServer,
};

mod apub;
mod args;
mod config;
mod data;
mod db;
mod error;
mod jobs;
mod middleware;
mod notify;
mod requests;
mod routes;

use self::{
    args::Args,
    config::Config,
    data::{ActorCache, Media, State},
    db::Db,
    jobs::{create_server, create_workers},
    middleware::{DebugPayload, RelayResolver},
    routes::{actor, inbox, index, nodeinfo, nodeinfo_meta, statics},
};

#[actix_rt::main]
async fn main() -> Result<(), anyhow::Error> {
    dotenv::dotenv().ok();

    let config = Config::build()?;

    if config.debug() {
        std::env::set_var(
            "RUST_LOG",
            "debug,tokio_postgres=info,h2=info,trust_dns_resolver=info,trust_dns_proto=info,rustls=info,html5ever=info",
        )
    } else {
        std::env::set_var("RUST_LOG", "info")
    }

    if config.pretty_log() {
        pretty_env_logger::init();
    } else {
        env_logger::init();
    }

    let db = Db::build(&config)?;

    let args = Args::new();

    if args.jobs_only() && args.no_jobs() {
        return Err(anyhow::Error::msg(
            "Either the server or the jobs must be run",
        ));
    }

    if !args.blocks().is_empty() || !args.whitelists().is_empty() {
        if args.undo() {
            db.remove_blocks(args.blocks()).await?;
            db.remove_whitelists(args.whitelists()).await?;
        } else {
            db.add_blocks(args.blocks()).await?;
            db.add_whitelists(args.whitelists()).await?;
        }
        return Ok(());
    }

    let media = Media::new(db.clone());
    let state = State::hydrate(config.clone(), &db).await?;
    let actors = ActorCache::new(db.clone());
    let job_server = create_server();

    notify::Notifier::new(config.database_url().parse()?)
        .register(notify::NewBlocks(state.clone()))
        .register(notify::NewWhitelists(state.clone()))
        .register(notify::NewListeners(state.clone(), job_server.clone()))
        .register(notify::NewActors(actors.clone()))
        .register(notify::NewNodes(state.node_cache()))
        .register(notify::RmBlocks(state.clone()))
        .register(notify::RmWhitelists(state.clone()))
        .register(notify::RmListeners(state.clone()))
        .register(notify::RmActors(actors.clone()))
        .register(notify::RmNodes(state.node_cache()))
        .start();

    if args.jobs_only() {
        for _ in 0..num_cpus::get() {
            let state = state.clone();
            let actors = actors.clone();
            let job_server = job_server.clone();
            let media = media.clone();
            let config = config.clone();
            let db = db.clone();

            Arbiter::new().exec_fn(move || {
                create_workers(db, state, actors, job_server, media, config);
            });
        }
        actix_rt::signal::ctrl_c().await?;
        return Ok(());
    }

    let no_jobs = args.no_jobs();

    let bind_address = config.bind_address();
    HttpServer::new(move || {
        if !no_jobs {
            create_workers(
                db.clone(),
                state.clone(),
                actors.clone(),
                job_server.clone(),
                media.clone(),
                config.clone(),
            );
        }

        App::new()
            .wrap(Compress::default())
            .wrap(Logger::default())
            .data(db.clone())
            .data(state.clone())
            .data(state.requests())
            .data(actors.clone())
            .data(config.clone())
            .data(job_server.clone())
            .data(media.clone())
            .service(web::resource("/").route(web::get().to(index)))
            .service(web::resource("/media/{path}").route(web::get().to(routes::media)))
            .service(
                web::resource("/inbox")
                    .wrap(config.digest_middleware())
                    .wrap(config.signature_middleware(state.requests(), actors.clone()))
                    .wrap(DebugPayload(config.debug()))
                    .route(web::post().to(inbox)),
            )
            .service(web::resource("/actor").route(web::get().to(actor)))
            .service(web::resource("/nodeinfo/2.0.json").route(web::get().to(nodeinfo)))
            .service(
                web::scope("/.well-known")
                    .service(actix_webfinger::scoped::<RelayResolver>())
                    .service(web::resource("/nodeinfo").route(web::get().to(nodeinfo_meta))),
            )
            .service(web::resource("/static/{filename}").route(web::get().to(statics)))
    })
    .bind(bind_address)?
    .run()
    .await?;
    Ok(())
}

include!(concat!(env!("OUT_DIR"), "/templates.rs"));