derive-macro/src/lib.rs


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

//! This crate provide the [`GenerateRandom`] derive macro
//! that implements the trait of the same name from the `generate-random` crate.
//! Refer to the documentation of that crate for more information.

use syn::{Data, DeriveInput, Fields};

mod handle_enum;
mod handle_struct;

#[proc_macro_derive(GenerateRandom, attributes(weight))]
pub fn derive_generate_random(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
    let input: DeriveInput = syn::parse(input).unwrap();
    match input.data {
        Data::Struct(ty) => handle_struct::generate(&input.ident, ty),
        Data::Enum(ty) => handle_enum::generate(&input.ident, ty),
        Data::Union(_) => panic!("Unions are not supported"),
    }
    .into()
}

fn generate_fields(fields: Fields) -> proc_macro2::TokenStream {
    use quote::quote;
    match fields {
        Fields::Named(fields) => {
            let fields = fields
                .named
                .into_iter()
                .map(|field| {
                    let field = field.ident.unwrap();
                    quote! {
                        #field: generate_random::GenerateRandom::generate_random(rng),
                    }
                })
                .collect::<proc_macro2::TokenStream>();
            quote! { { #fields } }
        }
        Fields::Unnamed(fields) => {
            let fields = fields
                .unnamed
                .into_iter()
                .map(|_field| {
                    quote! {
                        generate_random::GenerateRandom::generate_random(rng),
                    }
                })
                .collect::<proc_macro2::TokenStream>();
            quote! { ( #fields ) }
        }
        Fields::Unit => quote! {},
    }
}