Use cached native via GameObject accessors

Author: joeyhodge

shared/sdk/REComponent.hpp

@@ -3,24 +3,20 @@
 #include "ReClass.hpp"
 
 namespace utility::re_component {
+    namespace detail {
+        ::REGameObject* get_game_object_via_property(::REComponent* comp);
+    }
+
     static auto get_game_object(::REComponent* comp) {
         if (comp == nullptr) {
             return (::REGameObject*)nullptr;
         }
 
-        if (comp->ownerGameObject != nullptr) {
-            return comp->ownerGameObject;
-        }
-
-        if (auto reflected = utility::re_managed_object::get_field<::REGameObject*>(comp, "GameObject"); reflected != nullptr) {
-            return reflected;
-        }
-
-        if (auto reflected = utility::re_managed_object::get_field<::REGameObject*>(comp, "OwnerGameObject"); reflected != nullptr) {
-            return reflected;
+        if (auto game_object = detail::get_game_object_via_property(comp); game_object != nullptr) {
+            return game_object;
         }
 
-        return utility::re_managed_object::get_field<::REGameObject*>(comp, "_OwnerGameObject_k__BackingField");
+        return comp->ownerGameObject;
     }
 
     static auto get_valid(::REComponent* comp) {

shared/sdk/REGameObject.cpp

@@ -1,7 +1,25 @@
 #include "RETypeDB.hpp"
 
+#include "REComponent.hpp"
 #include "REGameObject.hpp"
 
+namespace utility::re_component::detail {
+::REGameObject* get_game_object_via_property(::REComponent* comp) {
+    if (comp == nullptr) {
+        return nullptr;
+    }
+
+    static const auto component_t = sdk::find_type_definition("via.Component");
+    static const auto get_game_object_method = component_t != nullptr ? component_t->get_method("get_GameObject") : nullptr;
+
+    if (get_game_object_method == nullptr) {
+        return nullptr;
+    }
+
+    return get_game_object_method->call<::REGameObject*>(sdk::get_thread_context(), comp);
+}
+}
+
 namespace utility::re_game_object {
 std::string get_name(REGameObject* obj) {
     if (obj == nullptr) {
@@ -33,33 +51,15 @@ RETransform* get_transform(REGameObject* obj) {
         return nullptr;
     }
 
-    if (obj->transform != nullptr) {
-        return obj->transform;
-    }
-
     static const auto game_object_t = sdk::find_type_definition("via.GameObject");
     static const auto get_transform_fn = game_object_t != nullptr ? game_object_t->get_method("get_Transform") : nullptr;
-    static const auto transform_field = game_object_t != nullptr ? game_object_t->get_field("Transform") : nullptr;
-    static const auto transform_field_lower = game_object_t != nullptr ? game_object_t->get_field("transform") : nullptr;
 
     if (get_transform_fn != nullptr) {
         if (auto transform = get_transform_fn->call<::RETransform*>(sdk::get_thread_context(), obj); transform != nullptr) {
             return transform;
         }
     }
 
-    if (transform_field != nullptr) {
-        if (auto transform = transform_field->get_data<::RETransform*>(obj); transform != nullptr) {
-            return transform;
-        }
-    }
-
-    if (transform_field_lower != nullptr) {
-        if (auto transform = transform_field_lower->get_data<::RETransform*>(obj); transform != nullptr) {
-            return transform;
-        }
-    }
-
-    return nullptr;
+    return obj->transform;
 }
 }