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

mod uri;
mod uri_parsing;
mod test_guide;
mod export;

pub mod typed_stream;

use devise::Result;
use syn::{Path, punctuated::Punctuated, parse::Parser, Token};
use syn::spanned::Spanned;
use proc_macro2::TokenStream;

fn struct_maker_vec(
    input: proc_macro::TokenStream,
    ty: TokenStream,
    map: impl Fn(TokenStream) -> TokenStream,
) -> Result<TokenStream> {
    use crate::exports::_Vec;

    // Parse a comma-separated list of paths.
    let paths = <Punctuated<Path, Token![,]>>::parse_terminated.parse(input)?;
    let exprs = paths.iter().map(|path| {
        let expr = map(quote_spanned!(path.span() => ___struct));
        quote_spanned!(path.span() => {
            let ___struct = #path {};
            let ___item: #ty = #expr;
            ___item
        })
    });

    Ok(quote!({
        let ___vec: #_Vec<#ty> = vec![#(#exprs),*];
        ___vec
    }))
}

pub fn routes_macro(input: proc_macro::TokenStream) -> TokenStream {
    struct_maker_vec(input, quote!(::rocket::Route), |e| quote!(#e.into_route()))
        .unwrap_or_else(|diag| diag.emit_as_expr_tokens())
}

pub fn catchers_macro(input: proc_macro::TokenStream) -> TokenStream {
    struct_maker_vec(input, quote!(::rocket::Catcher), |e| quote!(#e.into_catcher()))
        .unwrap_or_else(|diag| diag.emit_as_expr_tokens())
}

pub fn uri_macro(input: proc_macro::TokenStream) -> TokenStream {
    uri::_uri_macro(input.into())
        .unwrap_or_else(|diag| diag.emit_as_expr_tokens_or(quote! {
            rocket::http::uri::Origin::ROOT
        }))
}

pub fn uri_internal_macro(input: proc_macro::TokenStream) -> TokenStream {
    // TODO: Ideally we would generate a perfect `Origin::ROOT` so that we don't
    // assist in propagating further errors. Alas, we can't set the span to the
    // invocation of `uri!` without access to `span.parent()`, and
    // `Span::call_site()` here points to the `#[route]`, immediate caller,
    // generating a rather confusing error message when there's a type-mismatch.
    uri::_uri_internal_macro(input.into())
        .unwrap_or_else(|diag| diag.emit_as_expr_tokens_or(quote! {
            rocket::http::uri::Origin::ROOT
        }))
}

pub fn guide_tests_internal(input: proc_macro::TokenStream) -> TokenStream {
    test_guide::_macro(input)
        .unwrap_or_else(|diag| diag.emit_as_item_tokens())
}

pub fn export_internal(input: proc_macro::TokenStream) -> TokenStream {
    export::_macro(input)
        .unwrap_or_else(|diag| diag.emit_as_item_tokens())
}

pub fn typed_stream(input: proc_macro::TokenStream) -> TokenStream {
    typed_stream::_macro(input)
        .unwrap_or_else(|diag| diag.emit_as_item_tokens())
}