summaryrefslogtreecommitdiffstats
path: root/vendor/derive_more/src/mul_helpers.rs
blob: 985d64fe8c2137c89cefa9e7806f94656cd6ae46 (plain)
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
use crate::utils::{add_where_clauses_for_new_ident, MultiFieldData, RefType};
use proc_macro2::TokenStream;
use quote::quote;
use syn::{Generics, Ident};

pub fn generics_and_exprs(
    multi_field_data: MultiFieldData,
    scalar_ident: &Ident,
    type_where_clauses: TokenStream,
    ref_type: RefType,
) -> (Generics, Vec<TokenStream>) {
    let MultiFieldData {
        fields,
        casted_traits,
        members,
        method_ident,
        ..
    } = multi_field_data;
    let reference = ref_type.reference();
    let exprs: Vec<_> = casted_traits
        .iter()
        .zip(members)
        .map(
            |(casted_trait, member)| quote!(#casted_trait::#method_ident(#reference #member, rhs)),
        )
        .collect();

    let new_generics = add_where_clauses_for_new_ident(
        &multi_field_data.state.input.generics,
        &fields,
        scalar_ident,
        type_where_clauses,
        true,
    );
    (new_generics, exprs)
}