Fix usage in `fn main`.

This commit is contained in:
daxpedda 2019-12-09 12:08:38 +01:00
parent d7800612a5
commit cd6007d483
No known key found for this signature in database
GPG Key ID: 43D62A3EA388E46F
1 changed files with 4 additions and 2 deletions

View File

@ -17,10 +17,10 @@ use quote::quote;
#[proc_macro_attribute]
#[cfg(not(test))] // Work around for rust-lang/rust#62127
pub fn main(_: TokenStream, item: TokenStream) -> TokenStream {
let input = syn::parse_macro_input!(item as syn::ItemFn);
let mut input = syn::parse_macro_input!(item as syn::ItemFn);
let attrs = &input.attrs;
let vis = &input.vis;
let sig = &input.sig;
let sig = &mut input.sig;
let body = &input.block;
let name = &sig.ident;
@ -30,6 +30,8 @@ pub fn main(_: TokenStream, item: TokenStream) -> TokenStream {
.into();
}
sig.asyncness = None;
(quote! {
#(#attrs)*
#vis #sig {