commit a3619e92f73faf29e6dc3014d35a71596c1d2764
author David Tolnay <dtolnay@gmail.com> Fri Nov 27 14:13:15 2020 -0800
committer David Tolnay <dtolnay@gmail.com> Fri Nov 27 15:59:03 2020 -0800
tree 7e67a286f643539ccb9d3ec00b53fe20c16899ac
parent b3d7bb12fb55dee7b09095b1ecefe02c8a92c0dd

Prepare for PartialEq derives on structs

macro/src/derive.rs

diff --git a/macro/src/derive.rs b/macro/src/derive.rs
index 22c05cc..d896afb 100644
--- a/macro/src/derive.rs
+++ b/macro/src/derive.rs
@@ -2,7 +2,7 @@
 use proc_macro2::{Ident, Span, TokenStream};
 use quote::{quote, quote_spanned, ToTokens};
 
-pub fn expand_struct(strct: &Struct) -> TokenStream {
+pub fn expand_struct(strct: &Struct, actual_derives: &mut Option<TokenStream>) -> TokenStream {
     let mut expanded = TokenStream::new();
 
     for derive in &strct.derives {
@@ -14,6 +14,8 @@
         }
     }
 
+    *actual_derives = None;
+
     expanded
 }
 

macro/src/expand.rs

diff --git a/macro/src/expand.rs b/macro/src/expand.rs
index cc55101..62ab95f 100644
--- a/macro/src/expand.rs
+++ b/macro/src/expand.rs
@@ -134,10 +134,12 @@
         let vis = Token![pub](field.ident.span());
         quote!(#vis #field)
     });
-    let derived_traits = derive::expand_struct(strct);
+    let mut derives = None;
+    let derived_traits = derive::expand_struct(strct, &mut derives);
 
     quote! {
         #doc
+        #derives
         #[repr(C)]
         pub struct #ident {
             #(#fields,)*