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
use proc_macro::TokenStream; use proc_macro_error::proc_macro_error; use quote::quote; use syn::spanned::Spanned; #[proc_macro_attribute] #[proc_macro_error] pub fn test(_attr: TokenStream, item: TokenStream) -> TokenStream { let input = syn::parse_macro_input!(item as syn::ItemFn); let ret = &input.sig.output; let name = &input.sig.ident; let body = &input.block; let attrs = &input.attrs; let vis = &input.vis; if input.sig.asyncness.is_none() { proc_macro_error::abort!(input.span(), "the function must be declared as 'async'"); } let result = quote! { #[test] #(#attrs)* #vis fn #name() #ret { async_metronome::run(async { #body }); } }; result.into() }