derive_more-impl 2.1.1

Internal implementation of `derive_more` crate
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
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140

use crate::utils::{
    add_extra_where_clauses, numbered_vars, panic_one_field, SingleFieldData, State,
};
use proc_macro2::TokenStream;
use quote::quote;
use syn::{parse::Result, Data, DeriveInput};

/// Provides the hook to expand `#[derive(Deref)]` into an implementation of `Deref`
pub fn expand(input: &DeriveInput, trait_name: &'static str) -> Result<TokenStream> {
    match input.data {
        Data::Struct(_) => expand_struct(input, trait_name),
        Data::Enum(_) => expand_enum(input, trait_name),
        _ => panic!("only structs and enums can use `derive({trait_name})`"),
    }
}

fn expand_enum(input: &DeriveInput, trait_name: &'static str) -> Result<TokenStream> {
    let state = State::with_field_ignore_and_forward(
        input,
        trait_name,
        trait_name.to_lowercase(),
    )?;

    let trait_path = &state.trait_path;
    let enum_name = &input.ident;

    let mut target = None;
    let mut match_arms = vec![];
    let mut predicates = vec![];

    for variant_state in state.enabled_variant_data().variant_states.into_iter() {
        let data = variant_state.enabled_fields_data();
        if data.fields.len() != 1 {
            panic_one_field(variant_state.trait_name, &variant_state.trait_attr);
        };

        let vars = numbered_vars(variant_state.fields.len(), "");
        let matcher = data.matcher(&data.field_indexes, &vars);

        let info = data.infos[0].clone();
        let field_type = data.field_types[0];

        let (target_, var) = if info.forward {
            let casted_trait = data.casted_traits[0].clone();
            predicates.push(quote! { #field_type: #trait_path });
            (
                quote! { #casted_trait::Target },
                quote! { #casted_trait::deref(__0) },
            )
        } else {
            (quote! { #field_type }, quote! { __0 })
        };

        if target.is_none() {
            target = Some(target_);
        }

        match_arms.push(quote! {
            #matcher => #var
        });
    }

    let target = target.unwrap();

    let generics = if predicates.is_empty() {
        &input.generics
    } else {
        &add_extra_where_clauses(&input.generics, quote! { where #(#predicates),* })
    };

    let (imp_generics, type_generics, where_clause) = generics.split_for_impl();

    Ok(quote! {
        #[allow(deprecated)] // omit warnings on deprecated fields/variants
        #[allow(unreachable_code)] // omit warnings for `!` and other unreachable types
        #[automatically_derived]
        impl #imp_generics #trait_path for #enum_name #type_generics #where_clause {
            type Target = #target;

            #[inline]
            fn deref(&self) -> &Self::Target {
                match self {
                    #(#match_arms),*
                }
            }
        }
    })
}

fn expand_struct(input: &DeriveInput, trait_name: &'static str) -> Result<TokenStream> {
    let state = State::with_field_ignore_and_forward(
        input,
        trait_name,
        trait_name.to_lowercase(),
    )?;
    let SingleFieldData {
        input_type,
        field_type,
        trait_path,
        casted_trait,
        ty_generics,
        member,
        info,
        ..
    } = state.assert_single_enabled_field();

    let (target, body, generics) = if info.forward {
        (
            quote! { #casted_trait::Target },
            quote! { #casted_trait::deref(&#member) },
            add_extra_where_clauses(
                &input.generics,
                quote! {
                    where #field_type: #trait_path
                },
            ),
        )
    } else {
        (
            quote! { #field_type },
            quote! { &#member },
            input.generics.clone(),
        )
    };
    let (impl_generics, _, where_clause) = generics.split_for_impl();

    Ok(quote! {
        #[allow(deprecated)] // omit warnings on deprecated fields/variants
        #[allow(unreachable_code)] // omit warnings for `!` and other unreachable types
        #[automatically_derived]
        impl #impl_generics #trait_path for #input_type #ty_generics #where_clause {
            type Target = #target;

            #[inline]
            fn deref(&self) -> &Self::Target {
                #body
            }
        }
    })
}