🐛 fix(InventoryFilter): Fix #58. Check against display name aswell as…

… material type
ProfElements · Apr 10, 2021 · e62bcb6 · e62bcb6
1 parent 2d2d05b
commit e62bcb6
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/main/java/me/profelements/dynatech/listeners/InventoryFilterListener.java b/src/main/java/me/profelements/dynatech/listeners/InventoryFilterListener.java
@@ -13,7 +13,10 @@
 import org.bukkit.inventory.ItemStack;
 
 import javax.annotation.Nonnull;
+
+import java.util.ArrayList;
 import java.util.EnumSet;
+import java.util.List;
 import java.util.Set;
 
 public class InventoryFilterListener implements Listener {
@@ -46,19 +49,22 @@ private void checkAndFilter(Player p) {
     }
 
     private void filterInventory(@Nonnull Player p, @Nonnull PlayerBackpack backpack) {
+        List<String> blacklistedStrings = new ArrayList<String>();
         for (ItemStack item : backpack.getInventory().getContents()) {
             if (item != null && item.getType() != Material.AIR) {
                 blacklistedMaterials.add(item.getType());
+                blacklistedStrings.add(item.getItemMeta().getDisplayName());
             }
 
             if (backpack.getInventory().isEmpty()) {
                 blacklistedMaterials.clear();
+                blacklistedStrings.clear();
             }
         }
 
         //CANT DROP AIR SO HAVE TO ITERATE THROUGH THE INVENTORY
         for (ItemStack item : p.getInventory().getStorageContents()) {
-            if (item != null && blacklistedMaterials.contains(item.getType())) {
+            if (item != null && blacklistedMaterials.contains(item.getType()) && blacklistedStrings.contains(item.getItemMeta().getDisplayName())) {
                     item.setAmount(0);
                     break;
             }