orma-derive 0.3.3

Derive macros for Orma
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
26
27
28
29
30
31
32

use proc_macro2::TokenStream;
use quote::quote;

fn maybe_ident(arg: &syn::FnArg) -> Option<&syn::Ident> {
    match arg {
        syn::FnArg::Typed(syn::PatType { pat, .. }) => match pat.as_ref() {
            syn::Pat::Ident(ident) => Some(&ident.ident),
            _ => None,
        },
        _ => None,
    }
}

pub fn test_impl(mut item: syn::ItemFn) -> TokenStream {
    let aa = std::mem::take(&mut item.sig.inputs);
    let vis = &item.vis;
    let sig = &item.sig;
    let block = &item.block;
    let ident = format!("{}", sig.ident);

    let args = aa.iter().filter_map(maybe_ident);

    quote! {
        #[::tokio::test]
        #vis #sig {
            #(
                let #args = #args (#ident);
            )*
            #block
        }
    }
}