too many if statements

Assets/Scripts/Entity.cs

@@ -11,15 +11,20 @@ public class Entity : MonoBehaviour
     [Header("Movement")]
     public int maxMovement;
     public TileObject currentTile;
-    public float moveAnimSpeed;
+    [SerializeField] private float moveAnimSpeed;
 
     [Header("Flags")] 
     public bool canMove = true;
     public bool invincible;
 
+    [Header("Animation")] 
+    [SerializeField] private SpriteRenderer[] spriteRenderers;
+    [SerializeField] private float damageColorChangeSpeed;
+
     public virtual void TakeDamage(float damage)
     {
         health -= damage;
+        StartCoroutine(DamageAnimation());
         if (health <= 0)
         {
             OnKillEffects();
@@ -61,4 +66,20 @@ public class Entity : MonoBehaviour
         }
         FinishedMovement();
     }
+    private IEnumerator DamageAnimation()
+    {
+        float currentState = 0;
+        while (currentState < 1)
+        {
+            currentState += Time.deltaTime * damageColorChangeSpeed;
+            foreach (SpriteRenderer spritepart in spriteRenderers)
+            {
+                if (spritepart)
+                {
+                    spritepart.color = Color.Lerp(Color.gray, Color.white, currentState);
+                }
+            }
+            yield return null;
+        }
+    }
 }