only add damage to valid armor

closes #6

Author: hyperdefined

src/main/java/lol/hyper/toolstats/events/EntityDamage.java

@@ -126,7 +126,9 @@ public void onDamage(EntityDamageByEntityEvent event) {
             PlayerInventory inventory = player.getInventory();
             for (ItemStack armor : inventory.getArmorContents()) {
                 if (armor != null) {
-                    updateArmorDamage(armor, event.getDamage());
+                    if (isArmor(armor.getType().toString().toLowerCase(Locale.ROOT))) {
+                        updateArmorDamage(armor, event.getDamage());
+                    }
                 }
             }
         }
@@ -144,7 +146,9 @@ public void onDamage(EntityDamageEvent event) {
             PlayerInventory inventory = player.getInventory();
             for (ItemStack armor : inventory.getArmorContents()) {
                 if (armor != null) {
-                    updateArmorDamage(armor, event.getDamage());
+                    if (isArmor(armor.getType().toString().toLowerCase(Locale.ROOT))) {
+                        updateArmorDamage(armor, event.getDamage());
+                    }
                 }
             }
         }
@@ -162,7 +166,9 @@ public void onDamage(EntityDamageByBlockEvent event) {
             PlayerInventory inventory = player.getInventory();
             for (ItemStack armor : inventory.getArmorContents()) {
                 if (armor != null) {
-                    updateArmorDamage(armor, event.getFinalDamage());
+                    if (isArmor(armor.getType().toString().toLowerCase(Locale.ROOT))) {
+                        updateArmorDamage(armor, event.getDamage());
+                    }
                 }
             }
         }
@@ -351,4 +357,13 @@ private void updateArmorDamage(ItemStack itemStack, double damage) {
         }
         itemStack.setItemMeta(meta);
     }
+
+    /**
+     * Check if item is an armor piece.
+     * @param itemType The item type, not name.
+     * @return If the item is an armor piece.
+     */
+    private boolean isArmor(String itemType) {
+        return itemType.endsWith("_helmet") || itemType.endsWith("_chestplate") || itemType.endsWith("_leggings") || itemType.endsWith("_boots");
+    }
 }