use axum::{extract::Query, http::header, response::IntoResponse, routing::get, Router};
use std::collections::HashMap;
use tera::Tera;

#[tokio::main]
async fn main() {
    // build our application with a single route
    let app = Router::new()
        .route("/", get(index))
        .route("/search", get(search));

    // run it with hyper on localhost:3000
    axum::Server::bind(&"0.0.0.0:3000".parse().unwrap())
        .serve(app.into_make_service())
        .await
        .unwrap();
}

async fn index() -> impl IntoResponse {
    // Create a new Tera instance and add a template from a string
    let tera = tera_create();
    // Prepare the context with some data
    let context = tera::Context::new();

    // Render the template with the given context
    let rendered = tera.render("index.html.tera", &context).unwrap();

    header(rendered)
}

async fn search(Query(params): Query<HashMap<String, String>>) -> impl IntoResponse {
    let tera = tera_create();
    // Prepare the context with some data
    let mut context = tera::Context::new();
    context.insert("query", params.get("q").unwrap());

    let rendered = tera.render("search.html.tera", &context).unwrap();

    header(rendered)
}

fn tera_create() -> Tera {
    let tera = match Tera::new("templates/**/*.html") {
        Ok(tera) => tera,
        Err(error) => {
            println!("Parsing error(s): {}", error);
            ::std::process::exit(1);
        }
    };

    tera
}

fn header(rendered: impl ToString) -> impl IntoResponse {
    (
        [(header::CONTENT_TYPE, "text/html; charset=utf-8")],
        rendered.to_string(),
    )
}