use http::HeaderMap;
pub trait HeaderExtractor {
type Error;
fn extract_header<'a>(&self, headers: &'a HeaderMap) -> Result<&'a str, Self::Error>;
}
pub trait HeaderExtractorExt: Sized + HeaderExtractor {
fn map_err<Fn>(self, map_err: Fn) -> MapError<Self, Fn>;
}
impl<T> HeaderExtractorExt for T
where
T: Sized + HeaderExtractor,
{
fn map_err<Fn>(self, map_err: Fn) -> MapError<Self, Fn> {
MapError::new(self, map_err)
}
}
#[derive(Debug, Clone)]
pub struct MapError<T, Fn> {
inner: T,
map_err: Fn,
}
impl<T, Fn> MapError<T, Fn> {
pub const fn new(inner: T, map_err: Fn) -> Self {
Self { inner, map_err }
}
}
impl<H, Fn, E> HeaderExtractor for MapError<H, Fn>
where
H: HeaderExtractor + Sync,
Fn: FnOnce(H::Error) -> E + Clone + Sync,
{
type Error = E;
fn extract_header<'a>(&self, headers: &'a HeaderMap) -> Result<&'a str, Self::Error> {
self.inner
.extract_header(headers)
.map_err(self.map_err.clone())
}
}