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

extern crate proc_macro;

mod args;
mod throws;

use proc_macro::TokenStream;

use args::Args;
use throws::Throws;

#[proc_macro_attribute]
pub fn throws(args: TokenStream, input: TokenStream) -> TokenStream {
    let args = syn::parse_macro_input!(args as Args);
    Throws::new(Some(args)).fold(input)
}

#[proc_macro_attribute]
pub fn try_fn(args: TokenStream, input: TokenStream) -> TokenStream {
    assert!(args.to_string() == "", "try_fn does not take arguments");
    Throws::new(None).fold(input)
}