tower-canonical-redirect 0.1.0

A Tower middleware to enforce canonical hosts in HTTP requests
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54

use crate::http::uri::Origin;
use crate::layer::builder::CanonicalRedirectLayerBuilder;
use crate::service::CanonicalRedirect;
use http::HeaderName;
use http::uri::InvalidUri;
use std::sync::Arc;
use tower::Layer;

pub mod builder;

#[derive(Clone, Debug)]
pub struct CanonicalRedirectLayer
{
	canonical_origin: Origin,
	temporary_origins: Arc<[Origin]>,
	proto_headers: Arc<[HeaderName]>,
	host_headers: Arc<[HeaderName]>,
}

impl<S> Layer<S> for CanonicalRedirectLayer
{
	type Service = CanonicalRedirect<S>;

	fn layer(&self, inner: S) -> Self::Service
	{
		Self::Service {
			inner,
			canonical_origin: self.canonical_origin.clone(),
			temporary_origins: self.temporary_origins.clone(),
			proto_headers: self.proto_headers.clone(),
			host_headers: self.host_headers.clone(),
		}
	}
}

impl CanonicalRedirectLayer
{
	pub fn new(origin: impl AsRef<str>) -> Result<Self, InvalidUri>
	{
		Ok(Self {
			canonical_origin: origin.as_ref().parse()?,
			temporary_origins: Default::default(),
			proto_headers: Default::default(),
			host_headers: Default::default(),
		})
	}

	pub fn builder<'a, S>(origin: &'a S) -> CanonicalRedirectLayerBuilder<'a>
	where
		S: AsRef<str> + ?Sized,
	{
		CanonicalRedirectLayerBuilder::new(origin)
	}
}