rama-net 0.3.0

rama network types and utilities
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
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152

use crate::std::borrow::Cow;

use crate::http::uri::{UriMatchError, UriMatchReplace};
use crate::uri::Uri;

use rama_utils::macros::all_the_tuples_no_last_special_case;

/// Apply fallthrough to slices, arrays, tuples or vectors.
pub struct UriMatchReplaceFallthrough<R>(pub R);

macro_rules! impl_uri_match_replace_on_fallthrough_slice {
    () => {
        fn match_replace_uri<'a>(
            &self,
            mut uri: Cow<'a, Uri>,
        ) -> Result<Cow<'a, Uri>, UriMatchError<'a>> {
            let mut found = false;
            for rule in self.0.iter() {
                match rule.match_replace_uri(uri) {
                    Ok(new_uri) => {
                        uri = new_uri;
                        found = true;
                    }
                    Err(UriMatchError::NoMatch(original_uri)) => uri = original_uri,
                    Err(UriMatchError::Unexpected(err)) => {
                        return Err(UriMatchError::Unexpected(err));
                    }
                }
            }
            if found {
                Ok(uri)
            } else {
                Err(UriMatchError::NoMatch(uri))
            }
        }
    };
}

impl<R: UriMatchReplace, const N: usize> UriMatchReplace for UriMatchReplaceFallthrough<[R; N]> {
    impl_uri_match_replace_on_fallthrough_slice!();
}

impl<R: UriMatchReplace> UriMatchReplace for UriMatchReplaceFallthrough<&[R]> {
    impl_uri_match_replace_on_fallthrough_slice!();
}

impl<R: UriMatchReplace> UriMatchReplace for UriMatchReplaceFallthrough<Vec<R>> {
    impl_uri_match_replace_on_fallthrough_slice!();
}

macro_rules! impl_uri_match_replace_on_fallthrough_tuple {
    ( $($ty:ident),* $(,)? ) => {
        impl<$($ty),+> UriMatchReplace for UriMatchReplaceFallthrough<($($ty),+,)>
        where
            $(
                $ty: UriMatchReplace,
            )+
        {
            fn match_replace_uri<'a>(&self, mut uri: Cow<'a, Uri>) -> Result<Cow<'a, Uri>, UriMatchError<'a>> {
                let Self((
                    $($ty),+
                    ,
                )) = self;
                let mut found = false;
                $(
                    match $ty.match_replace_uri(uri) {
                        Ok(new_uri) => {
                            uri = new_uri;
                            found = true;
                        }
                        Err(UriMatchError::NoMatch(original_uri)) => uri = original_uri,
                        Err(UriMatchError::Unexpected(err)) => {
                            return Err(UriMatchError::Unexpected(err));
                        }
                    }
                )+
                if found {
                    Ok(uri)
                } else {
                    Err(UriMatchError::NoMatch(uri))
                }
            }
        }
    };
}

all_the_tuples_no_last_special_case!(impl_uri_match_replace_on_fallthrough_tuple);

#[cfg(test)]
mod tests {
    use super::*;
    use crate::http::uri::{UriMatchReplaceScheme, match_replace::UriMatchReplaceNever};

    #[test]
    fn fallthrough_slices_simple() {
        let input = [
            UriMatchReplaceScheme::replace("fpt".parse().unwrap(), "fpts".parse().unwrap()),
            UriMatchReplaceScheme::http_to_https(),
            UriMatchReplaceScheme::replace("foo".parse().unwrap(), "bar".parse().unwrap()),
            UriMatchReplaceScheme::replace("https".parse().unwrap(), "baz".parse().unwrap()),
            UriMatchReplaceScheme::replace("fpt".parse().unwrap(), "fpts".parse().unwrap()),
        ];

        //slice
        let uri = UriMatchReplaceFallthrough(input.as_slice())
            .match_replace_uri(Cow::Owned(Uri::from_static("http://example.com")))
            .unwrap();
        assert_eq!("baz://example.com", uri.to_string());

        // vec
        let uri = UriMatchReplaceFallthrough(input.to_vec())
            .match_replace_uri(Cow::Owned(Uri::from_static("http://example.com")))
            .unwrap();
        assert_eq!("baz://example.com", uri.to_string());

        // arr
        let uri = UriMatchReplaceFallthrough(input.clone())
            .match_replace_uri(Cow::Owned(Uri::from_static("http://example.com")))
            .unwrap();
        assert_eq!("baz://example.com", uri.to_string());

        // 1 no-error test w/ arr
        match input.match_replace_uri(Cow::Owned(Uri::from_static("gopher://example.com"))) {
            Ok(found) => panic!("unexpected match found: {found}"),
            Err(UriMatchError::NoMatch(_)) => (), // good,
            Err(UriMatchError::Unexpected(err)) => panic!("unexpected error: {err}"),
        }
    }

    #[test]
    fn fallthrough_tuple_simple() {
        let input = (
            UriMatchReplaceNever::new(),
            UriMatchReplaceScheme::http_to_https(),
            UriMatchReplaceNever::new(),
            UriMatchReplaceScheme::replace("https".parse().unwrap(), "baz".parse().unwrap()),
            UriMatchReplaceNever::new(),
        );

        let uri = UriMatchReplaceFallthrough(input.clone())
            .match_replace_uri(Cow::Owned(Uri::from_static("http://example.com")))
            .unwrap();
        assert_eq!("baz://example.com", uri.to_string());

        // 1 no-error test w/ arr
        match input.match_replace_uri(Cow::Owned(Uri::from_static("gopher://example.com"))) {
            Ok(found) => panic!("unexpected match found: {found}"),
            Err(UriMatchError::NoMatch(_)) => (), // good,
            Err(UriMatchError::Unexpected(err)) => panic!("unexpected error: {err}"),
        }
    }
}