rocket-include-handlebars 0.12.7

#[cfg(debug_assertions)]
use std::sync::{Mutex, MutexGuard};

#[cfg(debug_assertions)]
use crate::rocket::data::Data;
use crate::rocket::fairing::{Fairing, Info, Kind};
#[cfg(debug_assertions)]
use crate::rocket::request::Request;
use crate::rocket::Rocket;
#[cfg(debug_assertions)]
use crate::rocket::State;

#[cfg(debug_assertions)]
use crate::ReloadableHandlebars;

#[cfg(not(debug_assertions))]
use crate::Handlebars;

use crate::HandlebarsContextManager;

const FAIRING_NAME: &str = "Handlebars";

/// The fairing of `HandlebarsResponse`.
#[cfg(debug_assertions)]
pub struct HandlebarsResponseFairing {
    pub(crate) custom_callback:
        Box<dyn Fn(&mut MutexGuard<ReloadableHandlebars>) -> usize + Send + Sync + 'static>,
}

/// The fairing of `HandlebarsResponse`.
#[cfg(not(debug_assertions))]
pub struct HandlebarsResponseFairing {
    pub(crate) custom_callback: Box<dyn Fn(&mut Handlebars) -> usize + Send + Sync + 'static>,
}

impl Fairing for HandlebarsResponseFairing {
    #[cfg(debug_assertions)]
    fn info(&self) -> Info {
        Info {
            name: FAIRING_NAME,
            kind: Kind::Attach | Kind::Request,
        }
    }

    #[cfg(not(debug_assertions))]
    fn info(&self) -> Info {
        Info {
            name: FAIRING_NAME,
            kind: Kind::Attach,
        }
    }

    #[cfg(debug_assertions)]
    fn on_attach(&self, rocket: Rocket) -> Result<Rocket, Rocket> {
        let handlebars = Mutex::new(ReloadableHandlebars::new());

        let cache_capacity = (self.custom_callback)(&mut handlebars.lock().unwrap());

        let state = HandlebarsContextManager::new(handlebars, cache_capacity);

        Ok(rocket.manage(state))
    }

    #[cfg(not(debug_assertions))]
    fn on_attach(&self, rocket: Rocket) -> Result<Rocket, Rocket> {
        let mut handlebars = Handlebars::new();

        handlebars_helpers!(handlebars);

        let cache_capacity = (self.custom_callback)(&mut handlebars);

        let state = HandlebarsContextManager::new(handlebars, cache_capacity);

        Ok(rocket.manage(state))
    }

    #[cfg(debug_assertions)]
    fn on_request(&self, req: &mut Request, _data: &Data) {
        let cm = req
            .guard::<State<HandlebarsContextManager>>()
            .expect("HandlebarsContextManager registered in on_attach");

        cm.handlebars.lock().unwrap().reload_if_needed().unwrap();
    }
}