rocket_codegen/bang/
export.rs

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
use std::hash::Hash;

use devise::Spanned;
use devise::ext::SpanDiagnosticExt;
use proc_macro2::{TokenStream, TokenTree, Punct};

use crate::syn_ext::IdentExt;

pub fn _macro(input: proc_macro::TokenStream) -> devise::Result<TokenStream> {
    let mac: syn::ItemMacro = syn::parse(input)?;
    let macro_name = match mac.ident {
        Some(ident) => ident,
        None => return Err(mac.span().error("expected `macro_rules!`")),
    };

    // We rename the actual `macro_export` macro so we don't accidentally use it
    // internally from the auto-imported crate root macro namespace.
    let (attrs, def) = (mac.attrs, mac.mac);
    let internal_name = macro_name.prepend("___internal_");
    let mod_name = macro_name.uniqueify_with(|mut hasher| def.hash(&mut hasher));

    let macro_rules_tokens = def.tokens.clone();
    let decl_macro_tokens: TokenStream = def.tokens.into_iter()
        .map(|t| match t {
            TokenTree::Punct(p) if p.as_char() == ';' => {
                let mut token = Punct::new(',', p.spacing());
                token.set_span(p.span());
                TokenTree::Punct(token)
            },
            _ => t,
        })
        .collect();

    // Only try using the `macro` syntax on nightly/dev or when we don't know.
    let export = match version_check::is_feature_flaggable() {
        Some(true) | None => quote! {
            #(#attrs)*
            #[cfg(all(nightly, doc))]
            pub macro #macro_name {
                #decl_macro_tokens
            }

            #[cfg(not(all(nightly, doc)))]
            pub use #mod_name::#internal_name as #macro_name;
        },
        Some(false) => quote! {
            pub use #mod_name::#internal_name as #macro_name;
        }
    };

    Ok(quote! {
        #[allow(non_snake_case)]
        mod #mod_name {
            #[doc(hidden)]
            #[macro_export]
            macro_rules! #internal_name {
                #macro_rules_tokens
            }

            pub use #internal_name;
        }

        #export
    })
}