From 586e4add9f84077cc303a1a035012f846fd4f70c Mon Sep 17 00:00:00 2001
From: Auxilor <william.favierparsons1@gmail.com>
Date: Sun, 18 Apr 2021 21:13:07 +0100
Subject: [PATCH] Optimised ArmorSlot#getSlot

---
 .../willfp/ecoarmor/sets/meta/ArmorSlot.java  | 37 ++++++++-----------
 1 file changed, 16 insertions(+), 21 deletions(-)

diff --git a/eco-core/core-plugin/src/main/java/com/willfp/ecoarmor/sets/meta/ArmorSlot.java b/eco-core/core-plugin/src/main/java/com/willfp/ecoarmor/sets/meta/ArmorSlot.java
index 4097d3a..0efa279 100644
--- a/eco-core/core-plugin/src/main/java/com/willfp/ecoarmor/sets/meta/ArmorSlot.java
+++ b/eco-core/core-plugin/src/main/java/com/willfp/ecoarmor/sets/meta/ArmorSlot.java
@@ -56,29 +56,24 @@ public enum ArmorSlot {
         }
 
         Material material = itemStack.getType();
-        String name = material.name().toLowerCase();
+        String[] split = material.name().toLowerCase().split("_");
+        String name = split[split.length - 1];
 
-        if (name.endsWith("helmet") || name.endsWith("head")) {
-            return HELMET;
+        switch (name) {
+            case "helmet":
+            case "head":
+                return HELMET;
+            case "chestplate":
+                return CHESTPLATE;
+            case "elytra":
+                return ELYTRA;
+            case "leggings":
+                return LEGGINGS;
+            case "boots":
+                return BOOTS;
+            default:
+                return null;
         }
-
-        if (name.endsWith("chestplate")) {
-            return CHESTPLATE;
-        }
-
-        if (name.endsWith("elytra")) {
-            return ELYTRA;
-        }
-
-        if (name.endsWith("leggings")) {
-            return LEGGINGS;
-        }
-
-        if (name.endsWith("boots")) {
-            return BOOTS;
-        }
-
-        return null;
     }
 
     /**