five 0.2.6

A rust crate to support DCI natively in rust
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

use crate::analysis::{GenericsInfo, ImplBlockInfo};

use super::{function_descriptor::CompiledFunctionDescription, Compiled, Compiler};

#[allow(dead_code)]
#[derive(Clone)]
pub struct CompiledImplBlock {
    pub generics: GenericsInfo,
    pub for_lifetimes: Option<syn::Lifetime>,
    pub implemented_traits: Vec<syn::Path>,
    pub functions: Vec<CompiledFunctionDescription>,
    pub self_ty: syn::Type,
    pub attrs: Vec<syn::Attribute>,
}

impl Compiled<ImplBlockInfo> for CompiledImplBlock {
    fn emit(&self) -> proc_macro2::TokenStream {
        use quote::quote;

        let functions = self.functions.iter().map(|func| {
            // If we're implementing a trait, strip the pub modifier
            if !self.implemented_traits.is_empty() {
                // Convert to string, remove "pub ", and parse back
                let tokens = func.emit().to_string().replace("pub ", "");
                tokens.parse().unwrap()
            } else {
                func.emit()
            }
        });

        let impl_generics = &self.generics.get_params();
        let impl_where = &self.generics.get_where_clause();

        // Split the generic parameters into just the type parameters without bounds
        let type_params = impl_generics.iter().map(|param| match param {
            syn::GenericParam::Type(t) => &t.ident,
            _ => panic!("Unexpected generic parameter type"),
        });

        // Only add angle brackets if we have generic parameters
        let impl_generic_tokens = if !impl_generics.is_empty() {
            quote!(<#(#impl_generics),*>)
        } else {
            quote!()
        };

        let type_generic_tokens = if !impl_generics.is_empty() {
            quote!(<#(#type_params),*>)
        } else {
            quote!()
        };

        // Handle implemented traits
        let impl_trait = if !self.implemented_traits.is_empty() {
            let traits = &self.implemented_traits;
            quote!(#(#traits)+* for)
        } else {
            quote!()
        };
        let attrs = &self.attrs;
        quote! {
            #(#attrs)*
            impl #impl_generic_tokens #impl_trait Context #type_generic_tokens #impl_where {
                #(#functions)*
            }
        }
    }
}

impl Compiler<ImplBlockInfo> for ImplBlockInfo {
    fn compile(&self) -> CompiledImplBlock {
        CompiledImplBlock {
            generics: self.generics.clone(),
            for_lifetimes: self.for_lifetimes.clone(),
            implemented_traits: self.implemented_traits.clone(),
            functions: self.functions.iter().map(|f| f.compile()).collect(),
            self_ty: self.self_ty.clone(),
            attrs: self.attrs.clone(),
        }
    }

    type Output = CompiledImplBlock;
}