Derive macro should also work on enums (#47)

yewstack · Nov 11, 2023 · 085211b · 085211b
1 parent 8714ccf
commit 085211b
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/implicit-clone-derive/src/lib.rs b/implicit-clone-derive/src/lib.rs
@@ -2,9 +2,9 @@ use quote::quote;
 
 #[proc_macro_derive(ImplicitClone)]
 pub fn derive_implicit_clone(item: proc_macro::TokenStream) -> proc_macro::TokenStream {
-    let syn::ItemStruct {
+    let syn::DeriveInput {
         ident, generics, ..
-    } = syn::parse_macro_input!(item as syn::ItemStruct);
+    } = syn::parse_macro_input!(item as syn::DeriveInput);
     let (_impl_generics, ty_generics, where_clause) = generics.split_for_impl();
     let generics = generics
         .params

diff --git a/implicit-clone-derive/tests/function_component_attr/derive-pass.rs b/implicit-clone-derive/tests/function_component_attr/derive-pass.rs
@@ -9,6 +9,14 @@ struct StructWithGenerics<T>(T);
 #[derive(Clone, ImplicitClone)]
 struct StructWithGenericsWithBounds<T: PartialEq>(T);
 
+#[derive(Clone, ImplicitClone)]
+enum ExampleEnum {}
+
+#[derive(Clone, ImplicitClone)]
+enum EnumWithGenerics<T> {
+    Variant(T),
+}
+
 fn main() {
     let _ = ImplicitClone::implicit_clone(&ExampleStruct);
 }