use super::{StaticContextManager, StaticResources, StaticResponse};
use crate::rocket::{
fairing::{Fairing, Info, Kind},
Build, Rocket,
};
const FAIRING_NAME: &str = "Static Resources";
pub struct StaticResponseFairing {
pub(crate) custom_callback: Box<dyn Fn(&mut StaticResources) + Send + Sync + 'static>,
}
#[rocket::async_trait]
impl Fairing for StaticResponseFairing {
#[inline]
fn info(&self) -> Info {
Info {
name: FAIRING_NAME, kind: Kind::Ignite
}
}
#[inline]
async fn on_ignite(&self, rocket: Rocket<Build>) -> Result<Rocket<Build>, Rocket<Build>> {
let mut resources = StaticResources::new();
(self.custom_callback)(&mut resources);
let state = StaticContextManager::new(resources);
Ok(rocket.manage(state))
}
}
impl StaticResponse {
#[inline]
pub fn fairing<F>(f: F) -> impl Fairing
where
F: Fn(&mut StaticResources) + Send + Sync + 'static, {
StaticResponseFairing {
custom_callback: Box::new(f)
}
}
}