spire_enum_macros 1.2.0

Procedural macros distributed by the crate `spire_enum`.
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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98

#![allow(clippy::too_many_arguments)]
use super::*;

pub fn tokenize_enum_ref<'a>(
    vis: &Visibility,
    enum_ident: &Ident,
    gen_lf_params: &TokenStream,
    gen_lf_args: &TokenStream,
    where_clause: Optional<&WhereClause>,
    var_cfgs: &[&Any<Attribute<CfgMeta>>],
    var_tys: &[&'a Type],
    var_idents: &[&'a Ident],
) -> (Ident, Type, TokenStream) {
    let lf = Lifetime::new("'_r", Span::call_site());
    let ident = format_ident!("{enum_ident}Ref");
    let ty: Type = parse_quote! { #ident::#gen_lf_args };

    let mut def = {
        let docs = docs_tokens(format!(
            "An enum that mirrors the variants of [`{enum_ident}`], except this \
			 holds a reference to the variant's type instead of owning it."
        ));

        quote! {
            #docs
            #vis enum #ident #gen_lf_params #where_clause {
                #(
                    #var_cfgs
                    #var_idents (&#lf #var_tys)
                ),*
            }
        }
    };

    for ((var_cfg, var_ty), var_ident) in var_cfgs.iter().zip(var_tys.iter()).zip(var_idents.iter())
    {
        def.extend(quote! {
            #var_cfg
            impl #gen_lf_params ::core::convert::From<&#lf #var_ty> for #ident #gen_lf_params
            #where_clause
            {
                fn from(var: &#lf #var_ty) -> Self {
                    Self::#var_ident(var)
                }
            }
        });
    }

    (ident, ty, def)
}

pub fn tokenize_enum_mut<'a>(
    vis: &Visibility,
    enum_ident: &Ident,
    gen_lf_params: &TokenStream,
    gen_lf_args: &TokenStream,
    where_clause: Optional<&WhereClause>,
    var_cfgs: &[&Any<Attribute<CfgMeta>>],
    var_tys: &[&'a Type],
    var_idents: &[&'a Ident],
) -> (Ident, Type, TokenStream) {
    let lf = Lifetime::new("'_r", Span::call_site());
    let ident = format_ident!("{enum_ident}Mut");
    let ty: Type = parse_quote! { #ident::#gen_lf_args };

    let mut def = {
        let docs = docs_tokens(format!(
            "An enum that mirrors the variants of [`{enum_ident}`], except this \
            holds mutable references to the variants instead of owning their values."
        ));

        quote! {
            #docs
            #vis enum #ident #gen_lf_params #where_clause {
                #(
                    #var_cfgs
                    #var_idents (&#lf mut #var_tys)
                ),*
            }
        }
    };

    for ((var_cfg, var_ty), var_ident) in var_cfgs.iter().zip(var_tys.iter()).zip(var_idents.iter())
    {
        def.extend(quote! {
            #var_cfg
            impl #gen_lf_params ::core::convert::From<&#lf mut #var_ty> for #ident #gen_lf_params
            #where_clause
            {
                fn from(var: &#lf mut #var_ty) -> Self {
                    Self::#var_ident(var)
                }
            }
        });
    }

    (ident, ty, def)
}