AstralRinth/theseus_macros/src/lib.rs

extern crate proc_macro;

use proc_macro::TokenStream;
use quote::quote;
use syn::{parse_macro_input, ItemFn};

#[proc_macro_attribute]
pub fn debug_pin(_attr: TokenStream, item: TokenStream) -> TokenStream {
    let input = parse_macro_input!(item as ItemFn);

    let attrs = &input.attrs;
    let vis = &input.vis;
    let sig = &input.sig;
    let body = &input.block;

    #[cfg(debug_assertions)]
    let result = quote! {
        #(#attrs)*
        #vis #sig {
            Box::pin(async move {
                #body
            }).await
        }
    };
    #[cfg(not(debug_assertions))]
    let result = quote! {
        #(#attrs)*
        #vis #sig {
            #body
        }
    };

    TokenStream::from(result)
}