hyperlane-macros 21.0.2

A comprehensive collection of procedural macros for building HTTP servers with enhanced functionality. This crate provides attribute macros that simplify HTTP request handling, protocol validation, response management, and request data extraction.
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

use crate::*;

/// Filters requests matching the specified Referer header.
/// Supports both single and multiple referer value checks.
///
/// # Arguments
///
/// - `TokenStream` - The attribute token stream.
/// - `TokenStream` - The input token stream to process.
/// - `Position` - The position to inject the code.
///
/// # Returns
///
/// - `TokenStream` - The expanded token stream with Referer filter.
pub(crate) fn referer_macro(
    attr: TokenStream,
    item: TokenStream,
    position: Position,
) -> TokenStream {
    let multi_referer: MultiRefererData = parse_macro_input!(attr as MultiRefererData);
    inject(position, item, |context: &Ident, _: &Ident| {
        let statements = multi_referer.referer_values.iter().map(|referer_value| {
            quote! {
                let referer: Option<::hyperlane::RequestHeadersValueItem> = #context.get_request().try_get_header_back(::hyperlane::REFERER);
                if let Some(referer_header) = referer {
                    if referer_header != #referer_value {
                        return ::hyperlane::Status::Reject;
                    }
                } else {
                    return ::hyperlane::Status::Reject;
                }
            }
        });
        quote! {
            #(#statements)*
        }
    })
}

/// Rejects requests matching the specified Referer header.
/// Supports both single and multiple referer value checks.
///
/// # Arguments
///
/// - `TokenStream` - The attribute token stream.
/// - `TokenStream` - The input token stream to process.
/// - `Position` - The position to inject the code.
///
/// # Returns
///
/// - `TokenStream` - The expanded token stream with Referer rejection filter.
pub(crate) fn reject_referer_macro(
    attr: TokenStream,
    item: TokenStream,
    position: Position,
) -> TokenStream {
    let multi_referer: MultiRefererData = parse_macro_input!(attr as MultiRefererData);
    inject(position, item, |context: &Ident, _: &Ident| {
        let statements = multi_referer.referer_values.iter().map(|referer_value| {
            quote! {
                let referer: Option<::hyperlane::RequestHeadersValueItem> = #context.get_request().try_get_header_back(::hyperlane::REFERER);
                if let Some(referer_header) = referer {
                    if referer_header == #referer_value {
                        return ::hyperlane::Status::Reject;
                    }
                }
            }
        });
        quote! {
            #(#statements)*
        }
    })
}