mod handlers;

use warp::{
    Filter,
    Rejection,
    Reply,
    body
};
use env_logger;
use u_lib::{
    MASTER_PORT,
    contracts::*,
    client::network::Paths
};
use uuid::Uuid;


fn get_content<M: ToMsg>() -> impl Filter<Extract = (Message<'static, M>,),
                                Error = Rejection> + Clone {
    body::content_length_limit(1024*64).and(body::json::<Message<M>>())
}


#[tokio::main]
async fn main() {
    env_logger::init();
    let base_db = Storage::new();
    let db = warp::any().map(move || base_db.clone());

    let new_client = warp::post()
        .and(warp::path(Paths::init))
        .and(get_content::<ClientInfo>())
        .and(db.clone())
        .and_then(handlers::add_client);

    let ls = warp::get()
        .and(warp::path(Paths::ls))
        .and(db.clone())
        .and_then(handlers::ls);

    let get_jobs = warp::get()
        .and(warp::path(Paths::get_jobs))
        .and(db.clone())
        .and_then(handlers::get_jobs);

    let set_jobs = warp::post()
        .and(warp::path(Paths::set_jobs))
        .and(warp::path::param::<Uuid>().map(Some))
        .and(get_content::<IterWrap<JobMetaStorage>>())
        .and(db.clone())
        .and_then(handlers::set_jobs);

    let get_job_results = warp::get()
        .and(warp::path(Paths::get_job_results))
        .and(warp::path::param::<Uuid>())
        .and(db.clone())
        .and_then(handlers::get_job_results);

    let report = warp::post()
        .and(warp::path(Paths::report))
        .and(get_content::<IterWrap<Vec<JobResult>>>())
        .and(db.clone())
        .and_then(handlers::report);

    let auth_token = warp::header::exact("authorization", "Bearer 123qwe");

    let agent_zone = new_client
        .or(get_jobs)
        .or(report)
    ;

    let auth_zone = auth_token
        .and(ls
            .or(set_jobs)
            .or(get_job_results)
        );

    let routes = auth_zone
        .or(agent_zone)
        .with(warp::log("warp"));
    warp::serve(routes)
        .run(([0,0,0,0], MASTER_PORT)).await;
}