fix for color and name of the buttons in new level traits

Chasmical · Jun 3, 2021 · 00d9feb · 00d9feb
1 parent 2a05ee5
commit 00d9feb
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/RogueLibsCore/Hooks/Unlocks/DisplayedUnlock.cs b/RogueLibsCore/Hooks/Unlocks/DisplayedUnlock.cs
@@ -93,7 +93,7 @@ private int GetStateNum()
 		/// <param name="normal">Disabled/not selected state.</param>
 		protected void UpdateButton(bool isEnabledOrSelected, UnlockButtonState selected, UnlockButtonState normal)
 		{
-			Text = GetName();
+			Text = GetFancyName();
 			if (IsUnlocked)
 			{
 				State = isEnabledOrSelected ? selected : normal;

diff --git a/RogueLibsCore/Hooks/Unlocks/TraitUnlock.cs b/RogueLibsCore/Hooks/Unlocks/TraitUnlock.cs
@@ -136,10 +136,9 @@ public override void OnPushedButton()
 					ScrollingMenu menu = ((CustomScrollingMenu)Menu).Menu;
 					if (menu.twitchCountdownOn) return;
 					State = UnlockButtonState.Selected;
-					if (gc.twitchMode || (gc.sessionDataBig.twitchOn && gc.sessionDataBig.twitchTraits))
-					{
+					UpdateMenu();
+					if (gc.twitchMode || gc.sessionDataBig.twitchOn && gc.sessionDataBig.twitchTraits)
 						gc.twitchFunctions.ShowWinner(Name, "StatusEffect", Menu.Agent.isPlayer);
-					}
 					if (Name == "EnduranceTrait")
 					{
 						Menu.Agent.SetEndurance(Menu.Agent.enduranceStatMod + 1, true);