aboutsummaryrefslogtreecommitdiff
path: root/derive-macro/src/lib.rs
diff options
context:
space:
mode:
authorLizzy Fleckenstein <eliasfleckenstein@web.de>2023-02-06 16:30:55 +0100
committerLizzy Fleckenstein <eliasfleckenstein@web.de>2023-02-06 16:50:23 +0100
commitdbe6934449b4db1162079cbc39d89d51728bdda1 (patch)
treef1bef1cff37aea91c74d5702847610bbb5cb3c58 /derive-macro/src/lib.rs
parent97385f084e01bf935f8040a1616ed465f9d7db11 (diff)
downloadgenerate-random-dbe6934449b4db1162079cbc39d89d51728bdda1.tar.xz
cargo fmt
Diffstat (limited to 'derive-macro/src/lib.rs')
-rw-r--r--derive-macro/src/lib.rs15
1 files changed, 10 insertions, 5 deletions
diff --git a/derive-macro/src/lib.rs b/derive-macro/src/lib.rs
index 9aa93f5..8c77e61 100644
--- a/derive-macro/src/lib.rs
+++ b/derive-macro/src/lib.rs
@@ -2,10 +2,10 @@
//! 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::{DeriveInput, Data, Fields};
+use syn::{Data, DeriveInput, Fields};
-mod handle_struct;
mod handle_enum;
+mod handle_struct;
#[proc_macro_derive(GenerateRandom, attributes(weight))]
pub fn derive_generate_random(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
@@ -14,14 +14,17 @@ pub fn derive_generate_random(input: proc_macro::TokenStream) -> proc_macro::Tok
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()
+ }
+ .into()
}
fn generate_fields(fields: Fields) -> proc_macro2::TokenStream {
use quote::quote;
match fields {
Fields::Named(fields) => {
- let fields = fields.named.into_iter()
+ let fields = fields
+ .named
+ .into_iter()
.map(|field| {
let field = field.ident.unwrap();
quote! {
@@ -32,7 +35,9 @@ fn generate_fields(fields: Fields) -> proc_macro2::TokenStream {
quote! { { #fields } }
}
Fields::Unnamed(fields) => {
- let fields = fields.unnamed.into_iter()
+ let fields = fields
+ .unnamed
+ .into_iter()
.map(|_field| {
quote! {
generate_random::GenerateRandom::generate_random(rng),