preserve other macros (#39)

src/type_state.rs

@@ -124,6 +124,13 @@ pub fn type_state_inner(args: TokenStream, input: TokenStream) -> TokenStream {
         .map(|ident| quote!(::core::marker::PhantomData<fn() -> #ident>))
         .collect::<Vec<_>>();
 
+    // Get the struct's attributes (other macros) excluding the #[type_state] macro
+    let attrs: Vec<_> = input_struct
+        .attrs
+        .iter()
+        .filter(|attr| !attr.path().is_ident("type_state"))
+        .collect();
+
     // Generate the final output
     let output = quote! {
         mod #sealed_mod_name {
@@ -138,6 +145,7 @@ pub fn type_state_inner(args: TokenStream, input: TokenStream) -> TokenStream {
 
         #(#trait_impls)*
 
+        #(#attrs)*
         #[allow(clippy::type_complexity)]
         #visibility struct #struct_name<#combined_generics>
         #merged_where_clause

tests/simple_example.rs

@@ -14,6 +14,7 @@ enum Race {
     Human,
 }
 
+#[derive(Debug)]
 #[type_state(states = (Initial, RaceSet, LevelSet, SkillSlotsSet), slots = (Initial))]
 struct PlayerBuilder {
     race: Option<Race>,
@@ -133,4 +134,11 @@ mod tests {
         assert_eq!(player.level, another_player.level);
         assert_eq!(player.skill_slots, another_player.skill_slots);
     }
+
+    #[test]
+    fn other_macros_are_preserved() {
+        let player = PlayerBuilder::new();
+        println!("{:?}", player); // ensures `#[derive(Debug)]` is preserved
+        assert_eq!(player.level, None);
+    }
 }