rama 0.3.0-rc1

modular service framework
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

//! Reusable CLI [`Service`]s.
//!
//! These services are used by rama's binary
//! distribution and that can be added to your own rama-driven CLI application as well.
//!
//! [`Service`]: crate::Service

pub mod echo;
pub mod fs;
pub mod geo;
pub mod http_security;
pub mod ip;

use crate::{
    cli::ForwardKind,
    combinators::Either7,
    http::{
        headers::forwarded::{
            CFConnectingIp, ClientIp, TrueClientIp, XClientIp, XForwardedFor, XRealIp,
        },
        layer::forwarded::GetForwardedHeaderLayer,
    },
};

/// The forwarded-header extraction layer selected by a [`ForwardKind`].
///
/// `None` means no extraction: the default, and `HaProxy` (which forwards at
/// the transport layer instead).
type HttpForwardedLayer = Option<
    Either7<
        GetForwardedHeaderLayer,
        GetForwardedHeaderLayer<XForwardedFor>,
        GetForwardedHeaderLayer<XClientIp>,
        GetForwardedHeaderLayer<ClientIp>,
        GetForwardedHeaderLayer<XRealIp>,
        GetForwardedHeaderLayer<CFConnectingIp>,
        GetForwardedHeaderLayer<TrueClientIp>,
    >,
>;

/// Build the HTTP forwarded-header extraction layer selected by `forward`,
/// shared by the echo and fs CLI services.
pub(super) fn http_forwarded_layer(forward: Option<&ForwardKind>) -> HttpForwardedLayer {
    match forward {
        None | Some(ForwardKind::HaProxy) => None,
        Some(ForwardKind::Forwarded) => Some(Either7::A(GetForwardedHeaderLayer::forwarded())),
        Some(ForwardKind::XForwardedFor) => {
            Some(Either7::B(GetForwardedHeaderLayer::x_forwarded_for()))
        }
        Some(ForwardKind::XClientIp) => {
            Some(Either7::C(GetForwardedHeaderLayer::<XClientIp>::new()))
        }
        Some(ForwardKind::ClientIp) => Some(Either7::D(GetForwardedHeaderLayer::<ClientIp>::new())),
        Some(ForwardKind::XRealIp) => Some(Either7::E(GetForwardedHeaderLayer::<XRealIp>::new())),
        Some(ForwardKind::CFConnectingIp) => {
            Some(Either7::F(GetForwardedHeaderLayer::<CFConnectingIp>::new()))
        }
        Some(ForwardKind::TrueClientIp) => {
            Some(Either7::G(GetForwardedHeaderLayer::<TrueClientIp>::new()))
        }
    }
}