httproxide 0.2.0

Rusted HTTP router reverse-proxy
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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101

#[cfg(feature = "auth-layer")]
pub mod auth;
#[cfg(feature = "auth-request-layer")]
pub mod auth_request;
#[cfg(feature = "ip-filter-layer")]
pub mod ip_filter;
#[cfg(feature = "limit-layer")]
pub mod limit;
#[cfg(feature = "set-header-layer")]
pub mod set_header;
#[cfg(feature = "set-status-layer")]
pub mod set_status;

use serde::{Deserialize, Serialize};
use tower::Layer;

use crate::target::{IntoTarget, Target};

pub trait ApplyLayer {
    fn apply_layer(&self, t: Target) -> Target;
}

impl<L> ApplyLayer for L
where
    L: Layer<Target>,
    L::Service: IntoTarget,
{
    fn apply_layer(&self, t: Target) -> Target {
        self.layer(t).into_target()
    }
}

#[derive(Clone, Debug, Serialize, Deserialize, PartialEq, Eq, Hash)]
#[serde(tag = "layer_type")]
pub enum BuiltinLayerConfig {
    #[cfg(feature = "auth-layer")]
    RequireBasicAuth(auth::RequireBasicAuthConfig),
    #[cfg(feature = "auth-layer")]
    RequireBearerAuth(auth::RequireBearerAuthConfig),
    #[cfg(feature = "auth-request-layer")]
    AuthRequest(auth_request::AuthRequestConfig),
    #[cfg(feature = "set-header-layer")]
    SetResponseHeader(set_header::SetHeaderConfig),
    #[cfg(feature = "set-header-layer")]
    SetRequestHeader(set_header::SetHeaderConfig),
    #[cfg(feature = "set-status-layer")]
    SetStatus(set_status::SetStatusConfig),
    #[cfg(feature = "ip-filter-layer")]
    IpFilter(ip_filter::IpFilterConfig),
    #[cfg(feature = "limit-layer")]
    RequestBodyLimit(limit::RequestBodyLimitConfig),
}

pub fn layer_config(s: Target, cfg: serde_yaml::Value) -> anyhow::Result<Target> {
    let target = match serde_yaml::from_value(cfg) {
        Ok(builtin) => match builtin {
            #[cfg(feature = "auth-layer")]
            BuiltinLayerConfig::RequireBasicAuth(config) => {
                auth::basic_from_config(config)?.apply_layer(s)
            }
            #[cfg(feature = "auth-layer")]
            BuiltinLayerConfig::RequireBearerAuth(config) => {
                auth::bearer_from_config(config)?.apply_layer(s)
            }
            #[cfg(feature = "auth-request-layer")]
            BuiltinLayerConfig::AuthRequest(config) => {
                auth_request::from_config(config)?.apply_layer(s)
            }
            #[cfg(feature = "set-header-layer")]
            BuiltinLayerConfig::SetResponseHeader(config) => {
                set_header::response_from_config(config)?.apply_layer(s)
            }
            #[cfg(feature = "set-header-layer")]
            BuiltinLayerConfig::SetRequestHeader(config) => {
                set_header::request_from_config(config)?.apply_layer(s)
            }
            #[cfg(feature = "set-status-layer")]
            BuiltinLayerConfig::SetStatus(config) => {
                set_status::from_config(config)?.apply_layer(s)
            }
            #[cfg(feature = "ip-filter-layer")]
            BuiltinLayerConfig::IpFilter(config) => ip_filter::from_config(config)?.apply_layer(s),
            #[cfg(feature = "limit-layer")]
            BuiltinLayerConfig::RequestBodyLimit(config) => {
                limit::from_config(config)?.apply_layer(s)
            }
        },
        Err(e) => anyhow::bail!("{}", e),
    };
    Ok(target)
}

pub fn layers_config(
    mut s: Target,
    cfg: impl IntoIterator<Item = serde_yaml::Value>,
) -> anyhow::Result<Target> {
    for layer in cfg.into_iter() {
        s = layer_config(s, layer)?;
    }
    Ok(s)
}