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

mod expand;
mod parse;
mod utils;

use proc_macro::TokenStream;
use syn::Error;

fn parse_expand(is_mut: bool, attr: TokenStream, item: TokenStream) -> TokenStream {
    let output = parse::parse(is_mut, attr.into(), item.into())
        .and_then(expand::expand)
        .unwrap_or_else(Error::into_compile_error);
    #[cfg(feature = "debug")]
    utils::debug(output.clone()).unwrap();
    output.into()
}

#[proc_macro_attribute]
pub fn impl_expr(attr: TokenStream, item: TokenStream) -> TokenStream {
    parse_expand(false, attr, item)
}

#[proc_macro_attribute]
pub fn impl_expr_mut(attr: TokenStream, item: TokenStream) -> TokenStream {
    parse_expand(true, attr, item)
}