From 5fd3e5bc7303094be440e081b651d27dfa615728 Mon Sep 17 00:00:00 2001
From: Max <mjdodd01@comcast.net>
Date: Wed, 5 Mar 2025 17:28:34 +0100
Subject: [PATCH] Fixed selected chest rewards not being disabled

---
 .../Assets/Scripts/UI/ChestPopUpController.cs          | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/PuzzleGameProject/Assets/Scripts/UI/ChestPopUpController.cs b/PuzzleGameProject/Assets/Scripts/UI/ChestPopUpController.cs
index 4af1dbe..47b5409 100644
--- a/PuzzleGameProject/Assets/Scripts/UI/ChestPopUpController.cs
+++ b/PuzzleGameProject/Assets/Scripts/UI/ChestPopUpController.cs
@@ -11,8 +11,12 @@ namespace UI
         public event Action ArmorAndDiamondSelected;
 
         private Button _blackDie;
+        private bool _blackDieAvailable = true;
         private Button _key;
+        private bool _keyAvailable = true;
         private Button _armorAndDiamond;
+        private bool _armorAndDiamondAvailable = true;
+
         
         private void OnEnable()
         {
@@ -20,27 +24,33 @@ namespace UI
             
             _blackDie = root.Q<Button>("BlackDie");
             _blackDie.clicked += () => BlackDieSelected?.Invoke();
+            _blackDie.SetEnabled(_blackDieAvailable);
             
             _key = root.Q<Button>("Key");
             _key.clicked += () => KeySelected?.Invoke();
+            _key.SetEnabled(_keyAvailable);
             
             _armorAndDiamond = root.Q<Button>("ArmorAndDiamond");
             _armorAndDiamond.clicked += () => ArmorAndDiamondSelected?.Invoke();
+            _armorAndDiamond.SetEnabled(_armorAndDiamondAvailable);
         }
 
         public void DisableBlackDie()
         {
             _blackDie.SetEnabled(false);
+            _blackDieAvailable = false;
         }
 
         public void DisableKey()
         {
             _key.SetEnabled(false);
+            _keyAvailable = false;
         }
 
         public void DisableArmorAndDiamond()
         {
             _armorAndDiamond.SetEnabled(false);
+            _armorAndDiamondAvailable = false;
         }
     }
 }