fix: resolve components for mocks

Valheim-Modding · Apr 23, 2024 · 72d054a · 72d054a
1 parent d16425f
commit 72d054a
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/JotunnLib/Managers/PrefabManager.cs b/JotunnLib/Managers/PrefabManager.cs
@@ -629,6 +629,19 @@ internal static bool TryFindAssetInSelfOrChildComponents(GameObject unityObject,
                 return false;
             }
 
+            // search for the component on the object
+            if (objectType.IsSubclassOf(typeof(Component)))
+            {
+                var component = unityObject.GetComponent(objectType);
+
+                if (component)
+                {
+                    asset = component;
+                    return true;
+                }
+            }
+
+            // search for the asset inside a component
             foreach (var component in unityObject.GetComponents<Component>())
             {
                 if (!(component is Transform))
@@ -640,6 +653,7 @@ internal static bool TryFindAssetInSelfOrChildComponents(GameObject unityObject,
                 }
             }
 
+            // recursively search in children
             foreach (Transform tf in unityObject.transform)
             {
                 if (TryFindAssetInSelfOrChildComponents(tf.gameObject, objectType, out asset))