isolate mod component creation

FEZModding · Mar 9, 2023 · 874a3de · 874a3de
1 parent 959db0f
commit 874a3de
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 9 deletions.
diff --git a/Source/Hat.cs b/Source/Hat.cs
@@ -198,6 +198,10 @@ public void InitalizeAssemblies()
             {
                 mod.InitializeAssembly();
             }
+            foreach (var mod in Mods)
+            {
+                mod.InitializeComponents();
+            }
             Logger.Log("HAT", "Assembly initialization completed!");
         }
 
@@ -214,7 +218,7 @@ public void InitalizeComponents()
         {
             foreach(var mod in Mods)
             {
-                mod.InitializeComponents();
+                mod.InjectComponents();
             }
             Logger.Log("HAT", "Component initialization completed!");
         }

diff --git a/Source/Mod.cs b/Source/Mod.cs
@@ -90,8 +90,21 @@ public void InitializeAssets()
             }
         }
 
-        // injects custom components of this mod into the game
+
         public void InitializeComponents()
+        {
+            if (RawAssembly == null || Assembly == null) return;
+
+            foreach (Type type in Assembly.GetExportedTypes())
+            {
+                if (!typeof(IGameComponent).IsAssignableFrom(type) || !type.IsPublic) continue;
+                var gameComponent = (IGameComponent)Activator.CreateInstance(type, new object[] { ModLoader.Game });
+                Components.Add(gameComponent);
+            }
+        }
+
+        // injects custom components of this mod into the game
+        public void InjectComponents()
         {
             // add game components
             foreach (var component in Components)
@@ -102,17 +115,11 @@ public void InitializeComponents()
             }
         }
 
+        // loads mod's assemblies
         public void InitializeAssembly()
         {
             if (RawAssembly == null) return;
             Assembly = Assembly.Load(RawAssembly);
-
-            foreach (Type type in Assembly.GetExportedTypes())
-            {
-                if (!typeof(IGameComponent).IsAssignableFrom(type) || !type.IsPublic) continue;
-                var gameComponent = (IGameComponent)Activator.CreateInstance(type, new object[] { ModLoader.Game });
-                Components.Add(gameComponent);
-            }
         }
 
         public void Dispose()