use rocket::fairing::{AdHoc, Fairing, Info, Kind};
use rocket::http::Header;
use rocket::{Data, Request, Response};
use rocket_contrib::serve::StaticFiles;
use rusty_ulid::generate_ulid_string;

#[derive(Default)]
pub struct RequestId;

impl Fairing for RequestId {
    fn info(&self) -> Info {
        Info {
            name: "Request ID",
            kind: Kind::Request | Kind::Response,
        }
    }

    fn on_request(&self, request: &mut Request, _: &Data) {
        match request.headers().get_one("X-Request-Id") {
            Some(_) => {}
            None => {
                let reqid = generate_ulid_string();
                request.add_header(Header::new("X-Request-Id", reqid.clone()));
            }
        };
    }

    fn on_response(&self, request: &Request, response: &mut Response) {
        match request.headers().get_one("X-Request-Id") {
            Some(reqid) => {
                response.set_header(Header::new("X-Request-Id", format!("{}", reqid)));
            }
            None => {}
        };
    }
}

pub fn static_files() -> AdHoc {
    AdHoc::on_attach("Static fileserver", |rocket| {
        let asset_path = rocket.config().get_string("asset_path").unwrap();

        Ok(rocket.mount("/static", StaticFiles::from(asset_path)))
    })
}