Implemented armor and diamond ability

PuzzleGameProject/Assets/Scripts/Abilities/ArmorAbility.cs

@@ -0,0 +1,46 @@
+using System;
+using UnityEngine;
+
+namespace Abilities
+{
+    public class ArmorAbility : MonoBehaviour
+    {
+        public event Action MarkAbilityUsed;
+        public event Action ArmorAbilityGained;
+        
+        [SerializeField] private Player player;
+        [SerializeField] private int maxUses;
+        private int currentUses;
+        
+        private bool _abilityAcive;
+        private void OnEnable()
+        {
+            ChestRewardSelection.DiamondAndLifeSelected += HandleAbilityAcitvated;
+            player.DamageTaken += HandleDamageDealt;
+        }
+
+        private void HandleDamageDealt(int damage)
+        {
+            for (int i = 0; i < damage; i++)
+            {
+                if (_abilityAcive)
+                {
+                    player.Heal(1);
+                    currentUses--;
+                    MarkAbilityUsed?.Invoke();
+                    if (currentUses < 1)
+                    {
+                        _abilityAcive = false;
+                    }
+                }
+            }
+        }
+
+        private void HandleAbilityAcitvated()
+        {
+            ArmorAbilityGained?.Invoke();
+            _abilityAcive = true;
+            currentUses = maxUses;
+        }
+    }
+}