Add Rules button to Health Overview, closes Jaxe-Dev#5

meiskam · Feb 12, 2021 · 19ddf5f · 19ddf5f
1 parent 5de5ba6
commit 19ddf5f
Show file tree

Hide file tree

Showing 4 changed files with 33 additions and 20 deletions.
diff --git a/Source/Patch/RimWorld_HealthCardUtility_DrawOverviewTab.cs b/Source/Patch/RimWorld_HealthCardUtility_DrawOverviewTab.cs
@@ -0,0 +1,31 @@
+using HarmonyLib;
+using PawnRules.Data;
+using PawnRules.Interface;
+using RimWorld;
+using System.Collections.Generic;
+using System.Linq;
+using System.Reflection;
+using Verse;
+
+namespace PawnRules.Patch
+{
+    [HarmonyPatch(typeof(HealthCardUtility), "DrawOverviewTab")]
+    internal static class RimWorld_HealthCardUtility_DrawOverviewTab
+    {
+        private static void Postfix(ref float __result, UnityEngine.Rect leftRect, Pawn pawn, float curY)
+        {
+            if (!Registry.IsActive) { return; }
+
+            FloatMenu overviewWindowStack = Find.WindowStack.FloatMenu;
+
+            if (overviewWindowStack == null) { return; }
+
+            List<FloatMenuOption> overviewWindowStackOptions = (List<FloatMenuOption>) typeof(FloatMenu).GetField("options", BindingFlags.NonPublic | BindingFlags.Instance).GetValue(overviewWindowStack);
+            if (overviewWindowStackOptions?.Count > 0 && overviewWindowStackOptions.Last().Label.Equals("ManageFoodRestrictions".Translate()))
+            {
+                overviewWindowStack.Close(false);
+                Dialog_Rules.Open(pawn);
+            }
+        }
+    }
+}
diff --git a/Source/Patch/RimWorld_Pawn_FoodRestrictionTracker_Configurable.cs b/Source/Patch/RimWorld_Pawn_FoodRestrictionTracker_Configurable.cs
diff --git a/Source/Patch/RimWorld_Pawn_FoodRestrictionTracker_CurrentFoodRestriction.cs b/Source/Patch/RimWorld_Pawn_FoodRestrictionTracker_CurrentFoodRestriction.cs
@@ -11,7 +11,7 @@ private static bool Prefix(ref FoodRestriction __result)
         {
             if (!Registry.IsActive) { return true; }
 
-            __result = new FoodRestriction(-1, Lang.Get("FoodRestrictionsOverridden"));
+            __result = new FoodRestriction(-1, Lang.Get("Gizmo.EditRulesLabel"));
             return false;
         }
     }

diff --git a/Source/PawnRules.csproj b/Source/PawnRules.csproj
@@ -52,6 +52,7 @@
     </Reference>
   </ItemGroup>
   <ItemGroup>
+    <Compile Include="Patch\RimWorld_HealthCardUtility_DrawOverviewTab.cs" />
     <Compile Include="Patch\RimWorld_PawnColumnWorker_FoodRestriction_Compare.cs" />
     <Compile Include="Patch\RimWorld_PawnColumnWorker_FoodRestriction_DoAssignFoodRestrictionButtons.cs" />
     <Compile Include="Patch\RimWorld_PawnColumnWorker_FoodRestriction_DoHeader.cs" />
@@ -94,7 +95,6 @@
     <Compile Include="Patch\RimWorld_GenConstruct_CanConstruct.cs" />
     <Compile Include="Patch\RimWorld_InteractionWorker_RomanceAttempt_RandomSelectionWeight.cs" />
     <Compile Include="Patch\RimWorld_InteractionWorker_RomanceAttempt_SuccessChance.cs" />
-    <Compile Include="Patch\RimWorld_Pawn_FoodRestrictionTracker_Configurable.cs" />
     <Compile Include="Patch\RimWorld_Pawn_FoodRestrictionTracker_CurrentFoodRestriction.cs" />
     <Compile Include="Patch\RimWorld_Pawn_GuestTracker_SetGuestStatus.cs" />
     <Compile Include="Patch\RimWorld_PawnUtility_TrySpawnHatchedOrBornPawn.cs" />