use std::env;

use handlebars::Handlebars;
use serde_json::json;
use futures_core::future::BoxFuture;
use futures_util::FutureExt;
use mime;
use northstar::{Server, Request, Response, Body, GEMINI_PORT, GEMINI_MIME};
use anyhow::{anyhow};
use tokio;

fn serve_pdf(_: Request) -> BoxFuture<'static, anyhow::Result<Response>> {
    async move {
        let response = northstar::util::serve_file("./static/files/ritesh-chitlangi-2018.pdf", &mime::APPLICATION_PDF).await?;

        Ok(response)
    }
    .boxed()
}

fn gemini_index(_: Request) -> BoxFuture<'static, anyhow::Result<Response>> {
    async move {
        let json = &json!({
            "name": env::var("MY_NAME").unwrap(),
        });

        let mut hb = Handlebars::new();

        hb.register_template_file("index", "./templates/index.gmi.hbs").unwrap_or_else (|err| {
            anyhow!("{}", err);
        });

        match hb.render("index", json) {
            Ok(contents) => Ok(Response::success(&GEMINI_MIME, Body::from(contents))),
            Err(err) => Err(anyhow!("{}", err))
        }
    }
    .boxed()
}

#[tokio::main]
async fn main() {
    env_logger::init();

    let gemini_port = match env::var("GEMINI_PORT") {
        Ok(port) => port.parse::<u16>().unwrap(),
        Err(_) => GEMINI_PORT
    };

    let _gemini_server = match Server::bind(("0.0.0.0", gemini_port))
        .add_route("/", gemini_index)
        .add_route("/index.gmi", gemini_index)
        .add_route("/resume.pdf", serve_pdf)
        .serve()
        .await {
            Ok(_) => Ok(()),
            Err(e) => {
                println!("Error: {}", e.to_string());
                Err(e.to_string())
            }
    };
}