use super::Decompression;
use crate::compression_utils::AcceptEncoding;
use tower_layer::Layer;
#[derive(Debug, Default, Clone)]
pub struct DecompressionLayer {
accept: AcceptEncoding,
}
impl<S> Layer<S> for DecompressionLayer {
type Service = Decompression<S>;
fn layer(&self, service: S) -> Self::Service {
Decompression {
inner: service,
accept: self.accept,
}
}
}
impl DecompressionLayer {
pub fn new() -> Self {
Default::default()
}
#[cfg(feature = "decompression-gzip")]
pub fn gzip(mut self, enable: bool) -> Self {
self.accept.set_gzip(enable);
self
}
#[cfg(feature = "decompression-deflate")]
pub fn deflate(mut self, enable: bool) -> Self {
self.accept.set_deflate(enable);
self
}
#[cfg(feature = "decompression-br")]
pub fn br(mut self, enable: bool) -> Self {
self.accept.set_br(enable);
self
}
pub fn no_gzip(mut self) -> Self {
self.accept.set_gzip(false);
self
}
pub fn no_deflate(mut self) -> Self {
self.accept.set_deflate(false);
self
}
pub fn no_br(mut self) -> Self {
self.accept.set_br(false);
self
}
}