dremoc-macro 0.1.6

Procedural macros for dremoc
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
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74

use heck::{ToPascalCase, ToSnakeCase};
use proc_macro2::TokenStream as TokenStream2;
use quote::{quote, ToTokens};
use std::fmt::Display;
use syn::{Attribute, Ident};

pub fn to_pascal_case(ident: &Ident) -> Ident {
    to_pascal_case_ext(ident, "", "")
}

pub fn to_pascal_case_ext(ident: &Ident, prefix: impl Display, postfix: impl Display) -> Ident {
    let new_ident = format!("{} {} {}", prefix, ident, postfix);

    Ident::new(&new_ident.to_pascal_case(), ident.span())
}

pub fn to_snake_case(ident: &Ident) -> Ident {
    to_snake_case_ext(ident, "", "")
}

pub fn to_snake_case_ext(ident: &Ident, prefix: impl Display, postfix: impl Display) -> Ident {
    let new_ident = format!("{} {} {}", prefix, ident, postfix);

    Ident::new(&new_ident.to_snake_case(), ident.span())
}

pub fn hide_ident(ident: &Ident) -> Ident {
    let new_ident = format!("__{}", ident);

    Ident::new(&new_ident, ident.span())
}

pub fn attribute_tokens(attrs: &[Attribute]) -> TokenStream2 {
    let mut tokens = quote! {};
    for attr in attrs {
        attr.to_tokens(&mut tokens);
    }
    tokens
}

pub fn trait_macro_attribute_tokens(send: bool) -> TokenStream2 {
    #[cfg(feature = "use-async-trait")]
    return if send {
        quote! { #[::dremoc::service::async_trait::async_trait] }
    } else {
        quote! { #[::dremoc::service::async_trait::async_trait(?Send)] }
    };

    #[cfg(feature = "use-trait-variant")]
    return if send {
        quote! { #[::dremoc::service::trait_variant::make(Send)] }
    } else {
        quote! { #[::dremoc::service::trait_variant::make] }
    };

    #[cfg(not(any(feature = "use-async-trait", feature = "use-trait-variant")))]
    return quote! {};
}

pub fn impl_macro_attribute_tokens(send: bool) -> TokenStream2 {
    #[cfg(feature = "use-async-trait")]
    return if send {
        quote! { #[::dremoc::service::async_trait::async_trait] }
    } else {
        quote! { #[::dremoc::service::async_trait::async_trait(?Send)] }
    };

    #[cfg(not(feature = "use-async-trait"))]
    return if send {
        quote! {}
    } else {
        quote! {}
    };
}