Added OnInitialize callsite back in

shdwp · Apr 14, 2020 · 470dd0c · 470dd0c
1 parent abb162e
commit 470dd0c
Showing 1 changed file with 23 additions and 1 deletion.
diff --git a/UIExtenderLibModule/ViewModel/Patches/GauntletPartyScreenCallsites.cs b/UIExtenderLibModule/ViewModel/Patches/GauntletPartyScreenCallsites.cs
@@ -3,6 +3,7 @@
 using System.Reflection;
 using HarmonyLib;
 using SandBox.GauntletUI;
+using TaleWorlds.CampaignSystem.ViewModelCollection;
 using TaleWorlds.Core;
 
 namespace UIExtenderLibModule.ViewModel.Patches
@@ -23,4 +24,25 @@ public static MethodBase TargetMethod()
 
         public static IEnumerable<CodeInstruction> Transpiler(IEnumerable<CodeInstruction> input) => ViewModelPatchUtil.TranspilerForVMInstantiation(input);
     }
-}
+
+    // OnInitialize - only runs once each time party screen is opened BUT DOES NOT RUN ON THE FIRST OPEN
+    [HarmonyPatch]
+    public static class GauntletPartyScreenOnInitializeCallsite
+    {
+        public static MethodBase TargetMethod()
+        {
+            Type type = typeof(GauntletPartyScreen);
+            MethodInfo interfaceMethod = type.GetInterface(nameof(IGameStateListener)).GetMethod("OnInitialize");
+            if (interfaceMethod == null) return null;
+
+            InterfaceMapping map = type.GetInterfaceMap(interfaceMethod.DeclaringType ?? throw new NullReferenceException("Cannot find GauntletPartyScreen IGameStateListener.OnInitialize method"));
+            return map.TargetMethods[Array.IndexOf(map.InterfaceMethods, interfaceMethod)];
+        }
+
+        public static void Prefix()
+        {
+            ViewModelComponent component = UIExtenderLibModule.SharedInstance.ViewModelComponent;
+            component.RefreshMixinsForTypes(new[] {typeof(PartyVM)});
+        }
+    }
+}