Made Omni-breaker enchantable with unbreaking

Author: NegaNote

src/main/java/net/neganote/gtutilities/common/item/OmniBreakerItem.java

@@ -12,6 +12,8 @@
 import net.minecraft.world.item.Item;
 import net.minecraft.world.item.ItemStack;
 import net.minecraft.world.item.TooltipFlag;
+import net.minecraft.world.item.enchantment.Enchantment;
+import net.minecraft.world.item.enchantment.Enchantments;
 import net.minecraft.world.level.Level;
 import net.minecraft.world.level.block.state.BlockState;
 
@@ -66,15 +68,37 @@ public boolean mineBlock(@NotNull ItemStack pStack, @NotNull Level pLevel, @NotN
 
         var electricItem = Objects.requireNonNull(GTCapabilityHelper.getElectricItem(pStack));
 
+        var unbreaking = getAllEnchantments(pStack).getOrDefault(Enchantments.UNBREAKING, 0);
+        double chance = 1.0f / (unbreaking + 1);
+
+        double rand = Math.random();
+
         if (electricItem.getCharge() >= GTValues.VEX[tier]) {
-            // Only discharge if possible to discharge the full amount
-            electricItem.discharge(GTValues.VEX[tier], tier, true, false, false);
+            // Only discharge if possible to discharge the full amount and unbreaking chance doesn't proc
+            if (rand <= chance) {
+                electricItem.discharge(GTValues.VEX[tier], tier, true, false, false);
+            }
             return true;
         } else {
             return false;
         }
     }
 
+    @Override
+    public int getEnchantmentValue(ItemStack stack) {
+        return 22;
+    }
+
+    @Override
+    public boolean canApplyAtEnchantingTable(ItemStack stack, Enchantment enchantment) {
+        return enchantment == Enchantments.UNBREAKING;
+    }
+
+    @Override
+    public boolean isEnchantable(ItemStack stack) {
+        return true;
+    }
+
     @Override
     public void appendHoverText(@NotNull ItemStack stack, @Nullable Level level,
                                 @NotNull List<Component> tooltipComponents, @NotNull TooltipFlag isAdvanced) {

src/main/java/net/neganote/gtutilities/common/item/PrecisionBreakBehavior.java

@@ -7,6 +7,7 @@
 import net.minecraft.core.BlockPos;
 import net.minecraft.world.InteractionResult;
 import net.minecraft.world.item.context.UseOnContext;
+import net.minecraft.world.item.enchantment.Enchantments;
 import net.minecraft.world.level.Level;
 import net.minecraft.world.level.block.state.BlockState;
 
@@ -29,11 +30,18 @@ public InteractionResult useOn(UseOnContext context) {
                 return InteractionResult.PASS;
             }
 
+            int unbreaking = context.getItemInHand().getItem().getAllEnchantments(context.getItemInHand())
+                    .getOrDefault(Enchantments.UNBREAKING, 0);
+            double chance = 1.0f / (unbreaking + 1);
+            double rand = Math.random();
+
             var electricItem = GTCapabilityHelper.getElectricItem(context.getItemInHand());
 
             if (electricItem != null) {
                 if (electricItem.getCharge() >= GTValues.VEX[tier]) {
-                    electricItem.discharge(GTValues.VEX[tier], tier, true, false, false);
+                    if (rand >= chance) {
+                        electricItem.discharge(GTValues.VEX[tier], tier, true, false, false);
+                    }
                 } else {
                     return InteractionResult.PASS;
                 }