1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17

mod expand;
mod types;
mod parse;
mod crate_refs;

use proc_macro::TokenStream;
use syn::{parse_macro_input, ItemMod, LitStr};

#[proc_macro_attribute]
pub fn hook_module(args: TokenStream, input: TokenStream) -> TokenStream {
    let ast = parse_macro_input!(input as ItemMod);
    let args = parse_macro_input!(args as LitStr);
    
    let stream = expand::expand(&ast, &args)
        .unwrap_or_else(syn::Error::into_compile_error);
    stream.into()
}