Add support of unbreaking enchant for durability removal

EpiCanard · Oct 25, 2021 · 1fdc566 · 1fdc566
1 parent 39b5e1a
commit 1fdc566
Showing 1 changed file with 13 additions and 10 deletions.
diff --git a/src/main/java/net/gkid117/choptree/ChopTreeBlockListener.java b/src/main/java/net/gkid117/choptree/ChopTreeBlockListener.java
@@ -4,10 +4,13 @@
 import java.util.Iterator;
 import java.util.LinkedList;
 import java.util.List;
+import java.util.Random;
+
 import org.bukkit.Material;
 import org.bukkit.World;
 import org.bukkit.block.Block;
 import org.bukkit.block.BlockFace;
+import org.bukkit.enchantments.Enchantment;
 import org.bukkit.entity.Player;
 import org.bukkit.event.EventHandler;
 import org.bukkit.event.EventPriority;
@@ -19,6 +22,7 @@ public class ChopTreeBlockListener implements Listener {
   public static Player pubplayer = null;
   public static ChopTree plugin;
   private int leafRadius;
+  private final Random random = new Random();
 
   public ChopTreeBlockListener(ChopTree instance) {
     plugin = instance;
@@ -364,24 +368,23 @@ public void moveLeavesDown(List<Block> blocks)
     }
   }
 
-  public boolean breaksTool(Player player, ItemStack item)
-  {
-    if ((item != null) && 
-      (isTool(item.getType())))
-    {
+  public boolean breaksTool(Player player, ItemStack item) {
+    if (item != null && isTool(item.getType()) && shouldTakeDamage(item)) {
       short damage = item.getDurability();
-      if (isAxe(item.getType())) {
-        damage = (short)(damage + 1);
-      } else {
-        damage = (short)(damage + 2);
-      }
+      damage += (isAxe(item.getType())) ? 1 : 2;
       if (damage >= item.getType().getMaxDurability()) {
         return true;
       }
       item.setDurability(damage);
     }
     return false;
   }
+
+  private Boolean shouldTakeDamage(ItemStack item) {
+    int level = item.getEnchantmentLevel(Enchantment.DURABILITY);
+    int rate = 100 / (level + 1);
+    return random.nextInt(100) < rate;
+  }
 
   public boolean isTool(Material mat)
   {