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

// pub fn replace_self_assoc_type

use syn::punctuated::Punctuated;
use syn::token::Plus;
use syn::visit_mut::{VisitMut, visit_type_mut, visit_type_param_bound_mut};
use syn::{Ident, TraitItemType, Type, TypeParamBound, parse_quote};

pub fn get_bounds_and_replace_self_assoc_type(
    item_type: &TraitItemType,
) -> Punctuated<TypeParamBound, Plus> {
    let mut bounds = item_type.bounds.clone();
    let mut visitor = ReplaceSelfAssocTypeVisitor {
        type_ident: item_type.ident.clone(),
    };

    for bound in &mut bounds {
        visit_type_param_bound_mut(&mut visitor, bound);
    }

    bounds
}

pub struct ReplaceSelfAssocTypeVisitor {
    pub type_ident: Ident,
}

impl VisitMut for ReplaceSelfAssocTypeVisitor {
    fn visit_type_mut(&mut self, node: &mut Type) {
        if let Type::Path(type_path) = node {
            let type_ident = &self.type_ident;
            if type_path == &parse_quote! { Self :: #type_ident } {
                *node = parse_quote!(#type_ident);
                return;
            }
        }

        visit_type_mut(self, node);
    }
}