1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
#![deny(clippy::all)]

mod config;
mod handlers;
mod stats;

use std::env;

use ::config::Config;
use actix_governor::{Governor, GovernorConfigBuilder};
use actix_web::{middleware::Logger, web, App, HttpServer};
use actix_web_lab::web as web_lab;
use dotenv::dotenv;
use handlers::index_page;
use tokio_postgres::NoTls;

use crate::config::ServerConfig;
use crate::handlers::status_get_api;

use stats::Stats;

fn scoped_config(cfg: &mut web::ServiceConfig) {
    cfg.service(index_page);
}

fn api_scoped_config(cfg: &mut web::ServiceConfig) {
    cfg.service(status_get_api);
}

#[actix_web::main]
async fn main() -> std::io::Result<()> {
    std::env::set_var("RUST_LOG", "debug");
    std::env::set_var("RUST_BACKTRACE", "1");
    env_logger::init();

    dotenv().ok();

    let config_ = Config::builder()
        .add_source(::config::Environment::default().separator("__"))
        .build()
        .unwrap();

    let config: ServerConfig = config_.try_deserialize().unwrap();

    let pool = config.pg.create_pool(None, NoTls).unwrap();

    // Allow bursts with up to 10 requests per IP address
    // and replenishes one element every two seconds
    let governor_conf = GovernorConfigBuilder::default()
        .per_second(2)
        .burst_size(10)
        .finish()
        .unwrap();

    let base_addr = env::var("BASE_ADDR").expect("BASE_ADDR not set"); // BASE_ADDR could be replaced with ServerConfig.server_addr

    let server = HttpServer::new(move || {
        let logger = Logger::default();
        App::new()
            .wrap(logger)
            // Enable Governor middleware
            .wrap(Governor::new(&governor_conf))
            .app_data(web::Data::new(pool.clone()))
            .service(web::scope("/api").configure(api_scoped_config))
            .service(web::scope("/stats").configure(scoped_config))
            .service(web_lab::Redirect::new(
                "/",
                format!("{}{}", base_addr, "/stats"),
            ))
    })
    .bind(config.server_addr.clone())?
    .run();
    println!("Server running at http://{}/", config.server_addr);

    server.await
}