correctly handle the "ALL" config enchant entry

Author: zefir-git

src/main/java/pro/cloudnode/smp/enchantbookplus/ConfigEnchantmentEntry.java

@@ -12,43 +12,43 @@
 import java.util.Optional;
 import java.util.stream.Collectors;
 
-public final class ConfigEnchantmentEntry {
+public class ConfigEnchantmentEntry {
     /**
      * Name of the enchantment.
      */
-    private final @NotNull String name;
+    public final @NotNull String name;
 
     /**
      * Maximum level of the enchantment.
      */
-    private final @Nullable Integer maxLevel;
+    protected final @Nullable Integer maxLevel;
 
     /**
      * Max level relative
      */
-    private final boolean maxLevelRelative;
+    protected final boolean maxLevelRelative;
 
     /**
      * Cost of the enchantment.
      */
-    private final int cost;
+    protected final int cost;
 
     /**
      * Multiply cost by level.
      */
-    private final boolean multiplyCostByLevel;
+    protected final boolean multiplyCostByLevel;
 
     /**
      * Name of the enchantment.
      */
-    public @NotNull String getName() {
+    public final @NotNull String getName() {
         return name;
     }
 
     /**
      * Maximum level of the enchantment.
      */
-    public @NotNull Optional<@NotNull Integer> getMaxLevel() {
+    public final @NotNull Optional<@NotNull Integer> getMaxLevel() {
         if (Optional.ofNullable(maxLevel).isEmpty()) return Optional.empty();
         if (maxLevelRelative) return Optional.of(getEnchantment().getMaxLevel() + maxLevel);
         return Optional.of(maxLevel);
@@ -57,21 +57,21 @@ public final class ConfigEnchantmentEntry {
     /**
      * Cost of the enchantment.
      */
-    public int getCost() {
+    public final int getCost() {
         return cost;
     }
 
     /**
      * Multiply cost by level.
      */
-    public boolean getMultiplyCostByLevel() {
+    public final boolean getMultiplyCostByLevel() {
         return multiplyCostByLevel;
     }
 
     /**
      * Get enchantment
      */
-    public Enchantment getEnchantment() {
+    public final Enchantment getEnchantment() {
         return Enchantment.getByKey(NamespacedKey.minecraft(name));
     }
 
@@ -80,7 +80,7 @@ public Enchantment getEnchantment() {
      *
      * @param enchantment The enchantment
      */
-    public boolean isEnchantment(final @NotNull Enchantment enchantment) {
+    public final boolean isEnchantment(final @NotNull Enchantment enchantment) {
         return name.equalsIgnoreCase(enchantment.getKey().getKey());
     }
 
@@ -192,4 +192,18 @@ private static boolean isValidConfigValue(final @Nullable Object configValue) {
         if (!isValidConfigValue(configValue)) throw new IllegalArgumentException("Invalid config value");
         return configArray((ArrayList<HashMap<String, Object>>) configValue);
     }
+
+    public static final class AllConfigEnchantmentEntry extends ConfigEnchantmentEntry {
+        private AllConfigEnchantmentEntry(final @Nullable Integer maxLevel, final boolean maxLevelRelative, final int cost, final boolean multiplyCostByLevel) {
+            super("ALL", maxLevel, maxLevelRelative, cost, multiplyCostByLevel);
+        }
+
+        public static @NotNull AllConfigEnchantmentEntry from(final @NotNull ConfigEnchantmentEntry configEnchantmentEntry) {
+            return new AllConfigEnchantmentEntry(configEnchantmentEntry.maxLevel, configEnchantmentEntry.maxLevelRelative, configEnchantmentEntry.cost, configEnchantmentEntry.multiplyCostByLevel);
+        }
+
+        public @NotNull ConfigEnchantmentEntry enchant(final @NotNull Enchantment enchantment) {
+            return new ConfigEnchantmentEntry(enchantment.getKey().getKey(), this.maxLevel, maxLevelRelative, cost, multiplyCostByLevel);
+        }
+    }
 }

src/main/java/pro/cloudnode/smp/enchantbookplus/EnchantBookPlus.java

@@ -3,13 +3,15 @@
 import org.bukkit.enchantments.Enchantment;
 import org.bukkit.plugin.java.JavaPlugin;
 import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
 import pro.cloudnode.smp.enchantbookplus.event.PrepareAnvil;
 
 import java.util.ArrayList;
 import java.util.List;
 import java.util.Objects;
 import java.util.Optional;
 import java.util.logging.Level;
+import java.util.stream.Collectors;
 
 public final class EnchantBookPlus extends JavaPlugin {
     /**
@@ -38,28 +40,46 @@ private void registerEvents() {
         return configEnchantments;
     }
 
+    /**
+     * "ALL" enchantment cache
+     */
+    private @Nullable ConfigEnchantmentEntry.AllConfigEnchantmentEntry allConfigEnchantment;
+
+    /**
+     * "ALL" enchantment cache
+     */
+    public @NotNull Optional<ConfigEnchantmentEntry.@NotNull AllConfigEnchantmentEntry> getAllConfigEnchantment() {
+        return Optional.ofNullable(allConfigEnchantment);
+    }
+
     /**
      * Get enchantment from cache
      *
      * @param enchantment The Minecraft enchantment
      */
     public @NotNull Optional<@NotNull ConfigEnchantmentEntry> getConfigEnchantment(final @NotNull Enchantment enchantment) {
         final @NotNull Optional<@NotNull ConfigEnchantmentEntry> entry = getConfigEnchantments().stream().filter(c -> c.isEnchantment(enchantment)).findFirst();
-        return entry.isEmpty() ? getConfigEnchantments().stream().filter(c -> c.getName().equalsIgnoreCase("ALL")).findFirst() : entry;
+        getLogger().info(configEnchantments.stream().map(c -> c.name).collect(Collectors.joining(", ")));
+        return entry.isEmpty() ? getAllConfigEnchantment().map(a -> a.enchant(enchantment)) : entry;
     }
 
     /**
      * Reload config
      */
     public void reload() {
         reloadConfig();
+        final @NotNull List<@NotNull ConfigEnchantmentEntry> enchants;
         try {
-            configEnchantments = ConfigEnchantmentEntry.config(getConfig().get("enchantments"));
+            enchants = ConfigEnchantmentEntry.config(getConfig().get("enchantments"));
         }
         catch (final @NotNull Exception exception) {
             getLogger().log(Level.SEVERE, "Failed to load config", exception);
             getServer().getPluginManager().disablePlugin(this);
+            return;
         }
+        allConfigEnchantment = enchants.stream()
+                .filter(c -> c.name.equalsIgnoreCase("ALL")).findFirst().map(ConfigEnchantmentEntry.AllConfigEnchantmentEntry::from).orElse(null);
+        configEnchantments = enchants.stream().filter(c -> !c.name.equalsIgnoreCase("ALL")).collect(Collectors.toList());
     }
 
     @Override