site/src/main.rs

use anyhow::Result;
use hyper::{header::CONTENT_TYPE, Body, Response};
use prometheus::{Encoder, TextEncoder};
use std::sync::Arc;
use warp::{path, Filter};

pub mod app;
pub mod handlers;
pub mod post;
pub mod signalboost;

use app::State;

const APPLICATION_NAME: &str = concat!(env!("CARGO_PKG_NAME"), "/", env!("CARGO_PKG_VERSION"));

fn with_state(
    state: Arc<State>,
) -> impl Filter<Extract = (Arc<State>,), Error = std::convert::Infallible> + Clone {
    warp::any().map(move || state.clone())
}

#[tokio::main]
async fn main() -> Result<()> {
    pretty_env_logger::init();
    log::info!("starting up");

    let state = Arc::new(app::init(
        std::env::var("CONFIG_FNAME")
            .unwrap_or("./config.dhall".into())
            .as_str()
            .into(),
    )?);

    let healthcheck = warp::get().and(warp::path(".within").and(warp::path("health")).map(|| "OK"));

    let blog = {
        let base = warp::path!("blog" / ..);
        let index = base
            .and(warp::path::end())
            .and(with_state(state.clone()))
            .and_then(handlers::blog::index);
        let series = base.and(
            warp::path!("series").and(with_state(state.clone()).and_then(handlers::blog::series)),
        );
        let series_view = base.and(
            warp::path!("series" / String)
                .and(with_state(state.clone()))
                .and(warp::get())
                .and_then(handlers::blog::series_view),
        );
        let post_view = base.and(
            warp::path!(String)
                .and(with_state(state.clone()))
                .and(warp::get())
                .and_then(handlers::blog::post_view),
        );

        index.or(series.or(series_view)).or(post_view)
    };

    let gallery = {
        let base = warp::path!("gallery" / ..);
        let index = base
            .and(warp::path::end())
            .and(with_state(state.clone()))
            .and_then(handlers::gallery::index);
        let post_view = base.and(
            warp::path!(String)
                .and(with_state(state.clone()))
                .and(warp::get())
                .and_then(handlers::gallery::post_view),
        );

        index.or(post_view)
    };

    let talks = {
        let base = warp::path!("talks" / ..);
        let index = base
            .and(warp::path::end())
            .and(with_state(state.clone()))
            .and_then(handlers::talks::index);
        let post_view = base.and(
            warp::path!(String)
                .and(with_state(state.clone()))
                .and(warp::get())
                .and_then(handlers::talks::post_view),
        );

        index.or(post_view)
    };

    let static_pages = {
        let contact = warp::path!("contact").and_then(handlers::contact);
        let feeds = warp::path!("feeds").and_then(handlers::feeds);
        let resume = warp::path!("resume")
            .and(with_state(state.clone()))
            .and_then(handlers::resume);
        let signalboost = warp::path!("signalboost")
            .and(with_state(state.clone()))
            .and_then(handlers::signalboost);

        contact.or(feeds).or(resume).or(signalboost)
    };

    let files = {
        let files = warp::path("static").and(warp::fs::dir("./static"));
        let css = warp::path("css").and(warp::fs::dir("./css"));
        let sw = warp::path("sw.js").and(warp::fs::file("./static/js/sw.js"));
        let robots = warp::path("robots.txt").and(warp::fs::file("./static/robots.txt"));

        files.or(css).or(sw).or(robots)
    };

    let metrics_endpoint = warp::path("metrics").and(warp::path::end()).map(move || {
        let encoder = TextEncoder::new();
        let metric_families = prometheus::gather();
        let mut buffer = vec![];
        encoder.encode(&metric_families, &mut buffer).unwrap();
        Response::builder()
            .status(200)
            .header(CONTENT_TYPE, encoder.format_type())
            .body(Body::from(buffer))
            .unwrap()
    });

    let site = files
        .or(warp::get().and(path::end().and_then(handlers::index)))
        .or(static_pages)
        .or(blog)
        .or(gallery)
        .or(talks)
        .or(healthcheck)
        .or(metrics_endpoint)
        .map(|reply| {
            warp::reply::with_header(
                reply,
                "X-Hacker",
                "If you are reading this, check out /signalboost to find people for your team",
            )
        })
        .map(|reply| warp::reply::with_header(reply, "X-Clacks-Overhead", "GNU Ashlynn"))
        .with(warp::log(APPLICATION_NAME))
        .recover(handlers::rejection);

    warp::serve(site).run(([127, 0, 0, 1], 3030)).await;

    Ok(())
}

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