Adds an option for toggling whether reforging removing enchantments

src/main/java/net/knarcraft/blacksmith/config/NPCSetting.java

@@ -17,6 +17,12 @@ public enum NPCSetting {
      */
     FAIL_CHANCE("failReforgeChance", SettingValueType.PERCENTAGE, 10, "failReforgeChance"),
 
+    /**
+     * The setting for whether failing a reforging should downgrade/remove enchantments as well
+     */
+    FAIL_REMOVE_ENCHANTMENTS("failReforgeRemovesEnchantments", SettingValueType.BOOLEAN, false,
+            "failReforgeRemovesEnchantments"),
+
     /**
      * The setting for the chance of an additional enchantment being added
      */

src/main/java/net/knarcraft/blacksmith/config/NPCSettings.java

@@ -258,6 +258,15 @@ public class NPCSettings {
         return asInt(NPCSetting.FAIL_CHANCE);
     }
 
+    /**
+     * Gets whether a failed reforging should remove/downgrade enchantments
+     *
+     * @return <p>Whether enchantments should be removed</p>
+     */
+    public boolean getFailRemovesEnchantments() {
+        return asBoolean(NPCSetting.FAIL_REMOVE_ENCHANTMENTS);
+    }
+
     /**
      * Gets the chance for adding an extra enchantment to an item
      *
@@ -282,7 +291,7 @@ public class NPCSettings {
      * @return <p>Whether to drop reforged items on the ground</p>
      */
     public boolean getDropItem() {
-        return ConfigHelper.asBoolean(getValue(NPCSetting.DROP_ITEM));
+        return asBoolean(NPCSetting.DROP_ITEM);
     }
 
     /**
@@ -318,7 +327,7 @@ public class NPCSettings {
      * @return <p>True if this blacksmith is able to repair anvils</p>
      */
     public boolean getRepairAnvils() {
-        return ConfigHelper.asBoolean(getValue(NPCSetting.REPAIR_ANVILS));
+        return asBoolean(NPCSetting.REPAIR_ANVILS);
     }
 
     /**
@@ -343,6 +352,16 @@ public class NPCSettings {
         return getValue(setting).toString();
     }
 
+    /**
+     * Gets the boolean value of the given setting
+     *
+     * @param setting <p>The setting to get the value of</p>
+     * @return <p>The value of the given setting as a boolean</p>
+     */
+    private boolean asBoolean(NPCSetting setting) {
+        return ConfigHelper.asBoolean(getValue(setting));
+    }
+
     /**
      * Gets the value of a setting, using the default if not set
      *

src/main/java/net/knarcraft/blacksmith/trait/ReforgeSession.java

@@ -133,14 +133,19 @@ public class ReforgeSession implements Runnable {
             itemToReforge.setType(Material.ANVIL);
         }
 
-        // Add random enchantments
+        //See if a random roll (0-99) is less than extraEnchantmentChance, and add a random enchantment
         int roll = random.nextInt(100);
-        if (!(roll < config.getExtraEnchantmentChance() &&
-                itemToReforge.getEnchantments().keySet().size() < config.getMaxEnchantments())) {
-            // Abort if randomness isn't on our side, or if max enchantments has been reached
-            return;
+        if (roll < config.getExtraEnchantmentChance() &&
+                itemToReforge.getEnchantments().keySet().size() < config.getMaxEnchantments() &&
+                !ItemHelper.isAnvil(itemToReforge.getType(), false)) {
+            addRandomEnchantment();
         }
+    }
 
+    /**
+     * Adds a random enchantment to the currently reforged item
+     */
+    private void addRandomEnchantment() {
         //Find usable enchantments first
         List<Enchantment> usableEnchantments = new ArrayList<>();
         for (String enchantmentName : enchantments) {
@@ -176,19 +181,11 @@ public class ReforgeSession implements Runnable {
      * The method to run when a blacksmith fails re-forging an item
      */
     private void failReforge() {
-        // Remove or downgrade existing enchantments
-        for (Enchantment enchantment : itemToReforge.getEnchantments().keySet()) {
-            if (random.nextBoolean()) {
-                itemToReforge.removeEnchantment(enchantment);
-            } else {
-                if (itemToReforge.getEnchantmentLevel(enchantment) > 1) {
-                    itemToReforge.removeEnchantment(enchantment);
-                    itemToReforge.addEnchantment(enchantment, 1);
-                }
-            }
+        if (config.getFailRemovesEnchantments()) {
+            removeOrDowngradeEnchantments();
         }
 
-        // Damage the item
+        //Damage the item
         short currentItemDurability = ItemHelper.getDurability(itemToReforge);
         short newDurability = (short) (currentItemDurability + (currentItemDurability * random.nextInt(8)));
         short maxDurability = itemToReforge.getType().getMaxDurability();
@@ -200,6 +197,22 @@ public class ReforgeSession implements Runnable {
         updateDamage(itemToReforge, maxDurability - newDurability);
     }
 
+    /**
+     * Removes or downgrades all enchantments for the currently reforged item
+     */
+    private void removeOrDowngradeEnchantments() {
+        //Remove or downgrade existing enchantments
+        for (Enchantment enchantment : itemToReforge.getEnchantments().keySet()) {
+            //Completely remove the enchantment, downgrade it, or keep it if lucky and already level 1
+            if (random.nextBoolean()) {
+                itemToReforge.removeEnchantment(enchantment);
+            } else if (itemToReforge.getEnchantmentLevel(enchantment) > 1) {
+                itemToReforge.removeEnchantment(enchantment);
+                itemToReforge.addEnchantment(enchantment, 1);
+            }
+        }
+    }
+
     /**
      * Updates the damage done to an item
      *