bbgun 0.1.4

A builder for builders running builders!
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

use darling::{ast, util, FromDeriveInput, FromField};
use proc_macro::TokenStream;
use quote::quote;
use syn::*;

#[derive(Debug, FromField)]
#[darling(attributes(bbgun))]
struct BuilderField {
    ident: Option<Ident>,
    ty: Type,
    #[darling(default)]
    transform: Option<Path>,
    #[darling(default)]
    skip: bool,
}

#[derive(Debug, FromDeriveInput)]
#[darling(attributes(bbgun), supports(struct_named))]
struct Builder {
    ident: Ident,
    data: ast::Data<util::Ignored, BuilderField>,
    #[darling(default)]
    build_func: Option<Path>,
    #[darling(default)]
    builds_to: Option<Path>,
    #[darling(default)]
    mutable_builder: bool,
}

#[proc_macro_derive(Builder, attributes(bbgun))]
pub fn derive(input: TokenStream) -> TokenStream {
    let input = parse_macro_input!(input as DeriveInput);
    let builder = Builder::from_derive_input(&input).unwrap();
    let name = &builder.ident;

    let fields = match builder.data {
        ast::Data::Struct(fields) => fields,
        _ => panic!("Only structs are supported"),
    };

    let field_setters = fields.iter().filter(|x| !x.skip).map(|field| {
        let field_name = field.ident.as_ref().unwrap();
        let field_type = &field.ty;
        let subsetter = if let Some(transform) = &field.transform {
            quote! {
                #transform(#field_name)
            }
        } else {
            quote! {
                #field_name
            }
        };

        if builder.mutable_builder {
            quote! {
                fn #field_name(&mut self, #field_name: #field_type) -> &mut Self {
                    self.#field_name = #subsetter;
                    self
                }
            }
        } else {
            quote! {
                fn #field_name(self, #field_name: #field_type) -> Self {
                    Self {
                        #field_name: #subsetter,
                        ..self
                    }
                }
            }
        }
    });

    let build_func = if builder.build_func.is_some() && builder.builds_to.is_some() {
        let bf = builder.build_func.as_ref().unwrap();
        let bt = builder.builds_to.as_ref().unwrap();
        quote! {
            fn build(&self) -> #bt {
                    #bf(self)
            }
        }
    } else {
        quote! {}
    };

    TokenStream::from(quote! {
        #[automatically_derived]
        impl #name {
            #build_func

            #(#field_setters)*
        }
    })
}