cgp-macro-lib 0.7.0

Context-generic programming core component macros implemented as a library.
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

use alloc::vec::Vec;

use quote::{ToTokens, quote};
use syn::punctuated::Punctuated;
use syn::token::Comma;
use syn::{Ident, ItemTrait, TraitItem, TypeParamBound, parse2};

use crate::parse::parse_is_provider_params;
use crate::replace_self::{
    iter_parse_and_replace_self_type, parse_and_replace_self_type,
    replace_self_receiver_in_signature, replace_self_var, to_snake_case_ident,
};

pub fn derive_provider_trait(
    component_name: &Ident,
    component_params: &Punctuated<Ident, Comma>,
    consumer_trait: &ItemTrait,
    provider_name: &Ident,
    context_type: &Ident,
) -> syn::Result<ItemTrait> {
    let mut provider_trait = consumer_trait.clone();

    provider_trait.ident = provider_name.clone();

    // Add generic parameter `Context` to the front of generics
    {
        provider_trait
            .generics
            .params
            .insert(0, parse2(quote!(#context_type))?);
    }

    let local_assoc_types: Vec<Ident> = provider_trait
        .items
        .iter()
        .filter_map(|item| {
            if let TraitItem::Type(assoc_type) = item {
                Some(assoc_type.ident.clone())
            } else {
                None
            }
        })
        .collect();

    // Turn the supertrait constraints into `Context` constraints in the `where` clause
    {
        let context_constraints = iter_parse_and_replace_self_type(
            provider_trait.supertraits.clone(),
            context_type,
            &local_assoc_types,
        )?;

        let is_provider_params = parse_is_provider_params(&consumer_trait.generics)?;

        let provider_supertrait: TypeParamBound = parse2(quote!(
            IsProviderFor< #component_name < #component_params >, #context_type, ( #is_provider_params ) >
        ))?;

        provider_trait.supertraits = Punctuated::from_iter([provider_supertrait]);

        if !context_constraints.is_empty() {
            match &mut provider_trait.generics.where_clause {
                Some(where_clause) => {
                    let mut predicates = iter_parse_and_replace_self_type(
                        where_clause.predicates.clone(),
                        context_type,
                        &local_assoc_types,
                    )?;

                    predicates.push(parse2(quote! {
                        #context_type : #context_constraints
                    })?);

                    where_clause.predicates = predicates;
                }
                _ => {
                    provider_trait.generics.where_clause = Some(parse2(quote! {
                        where #context_type : #context_constraints
                    })?);
                }
            }
        }
    }

    // Replace self type and argument into context type argument
    {
        let context_var = to_snake_case_ident(context_type);

        for item in provider_trait.items.iter_mut() {
            let mut replaced_item =
                parse_and_replace_self_type(item, context_type, &local_assoc_types)?;

            if let TraitItem::Fn(func) = &mut replaced_item {
                replace_self_receiver_in_signature(
                    &mut func.sig,
                    &context_var,
                    context_type.to_token_stream(),
                );

                if let Some(block) = &mut func.default {
                    let replaced = replace_self_var(block.to_token_stream(), &context_var);
                    *block = parse2(replaced)?;
                }
            }

            *item = replaced_item;
        }
    }

    Ok(provider_trait)
}