feat(engine): change enum reflection get_variant_reflection fn to get_variant_index

1 files changed, 15 insertions, 4 deletions

diff --git a/engine-macros/src/reflection/enum_impl.rs b/engine-macros/src/reflection/enum_impl.rs
index 1cc8f0a..4e27c2b 100644
--- a/engine-macros/src/reflection/enum_impl.rs
+++ b/engine-macros/src/reflection/enum_impl.rs
@@ -110,15 +110,26 @@ fn generate_impls(
                                 #get_default_value_fn
                             },
                             cast_dyn_any: |ptr| ptr.cast::<Self>(),
-                            get_variant_reflection: |target| {
+                            get_variant_index: |target| {
                                 use std::any::Any;
+                                use #engine_crate_path::reflection::Reflection;
                                 use #engine_crate_path::reflection::EnumReflectionExt;
 
                                 let target = target.downcast_ref::<Self>()?;
 
-                                Some(<Self as EnumReflectionExt>::get_variant_reflection(
-                                    target
-                                ))
+                                let enum_reflection = unsafe {
+                                    <Self as Reflection>::TYPE_REFLECTION
+                                        .as_enum()
+                                        .unwrap_unchecked()
+                                };
+
+                                Some(enum_reflection.variants
+                                    .element_offset(
+                                        match target {
+                                            #(#variant_lookup_match_arms),*
+                                        }
+                                    )
+                                    .unwrap())
                             }
                         }
                     )