path: root/src/main.rs

#![feature(lazy_cell)]

use dotenv::var;
use log::LevelFilter;
use std::collections::BTreeMap;
use std::fs::File;
use std::sync::Arc;
use lfm_embed::{STATE, ResponseCtx};
use lfm_embed::font::FontQuery;
use warp::Filter;

#[derive(serde::Deserialize, Debug)]
struct UserQuery {
    #[serde(default)]
    theme: Option<Arc<str>>,
    #[serde(flatten)]
    #[serde(default)]
    font: Option<FontQuery>,
    #[serde(flatten)]
    rest: BTreeMap<String, String>
}

#[tokio::main]
async fn main() {
    env_logger::Builder::new()
        .filter_level(LevelFilter::Warn)
        .parse_filters(&var("LFME_LOG_LEVEL").unwrap_or_default())
        .target(
            var("LFME_LOG_FILE").ok()
                .map(
                    |f| env_logger::Target::Pipe(
                        Box::new(File::options()
                                 .append(true)
                                 .open(f)
                                 .expect("couldn't open LFME_LOG_FILE")))
                )
                .unwrap_or(env_logger::Target::Stderr)
        ).init();

    std::sync::LazyLock::force(&STATE);

    let user = warp::path!("user" / String)
        .and(warp::query::<UserQuery>())
        .then(|s, q: UserQuery| async move {
            log::debug!(target: "lfm::server::user", "Handling request for user `{s}` with {q:?}");
            let (ctx, dur) = STATE.get_userinfo(&s).await;
            let ResponseCtx(mut data, status) = (ctx, q.font, q.rest).into();
            
            let theme = q.theme.filter(|a| STATE.handlebars().has_template(&a)).unwrap_or_else(|| STATE.default_theme());
            log::debug!(target: "lfm::server::user", "Using theme {theme}");
            warp::reply::with_header(
                warp::reply::with_header(
                    warp::reply::with_status(
                        warp::reply::html(
                            STATE.handlebars().render(&theme, &data).unwrap()
                        ), status
                    ), "Refresh", dur.as_secs()
                ), "X-Selected-Theme", theme.as_ref()
            )
        });

    warp::serve(user)
        .bind(([127,0,0,1], STATE.port())).await;
}