ret_ty_attr 0.1.1

Specify the return type depending on configuration conditional checks.
Documentation 
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

use proc_macro as pm;
use proc_macro2 as pm2;
use quote::ToTokens;

#[proc_macro_attribute]
pub fn ret_ty(meta: pm::TokenStream, input: pm::TokenStream) -> pm::TokenStream {
    fn inner(meta: pm2::TokenStream, input: pm2::TokenStream) -> syn::Result<pm2::TokenStream> {
        let mut input = syn::parse2::<syn::ItemFn>(input)?;
        let ret_ty: syn::ReturnType = if meta.is_empty() {
            syn::ReturnType::Default
        } else {
            syn::parse2::<syn::ReturnType>(meta)?
        };
        let syn::ItemFn {
            sig: syn::Signature { output, .. },
            ..
        } = &mut input;
        *output = ret_ty;
        Ok(input.into_token_stream())
    }
    match inner(meta.into(), input.into()) {
        Ok(output) => output.into(),
        Err(error) => error.into_compile_error().into(),
    }
}