mcMMO/src/main/java/com/gmail/nossr50/util/EnchantmentMapper.java

package com.gmail.nossr50.util;

import com.gmail.nossr50.mcMMO;
import org.bukkit.Registry;
import org.bukkit.enchantments.Enchantment;

public class EnchantmentMapper {
    private final mcMMO pluginRef;
    private final Enchantment efficiency;
    private final Enchantment unbreaking;
    private final Enchantment infinity;
    private final Enchantment featherFalling;
    private final Enchantment luckOfTheSea;

    public EnchantmentMapper(mcMMO pluginRef) {
        this.pluginRef = pluginRef;
        this.efficiency = initEfficiency();
        this.unbreaking = initUnbreaking();
        this.infinity = initInfinity();
        this.featherFalling = initFeatherFalling();
        this.luckOfTheSea = initLuckOfTheSea();
    }

    private Enchantment initLuckOfTheSea() {
        // TODO: Make use of match if present
//        if (Registry.ENCHANTMENT.match("luck_of_the_sea") != null) {
//            return Registry.ENCHANTMENT.match("luck_of_the_sea");
//        }

        // Look for the enchantment by name
        for (Enchantment enchantment : Registry.ENCHANTMENT) {
            if (enchantment.getKey().getKey().equalsIgnoreCase("LUCK_OF_THE_SEA")
                    || enchantment.getKey().getKey().equalsIgnoreCase("LUCK")
                    || enchantment.getName().equalsIgnoreCase("LUCK_OF_THE_SEA")
                    || enchantment.getName().equalsIgnoreCase("LUCK")) {
                return enchantment;
            }
        }

        pluginRef.getLogger().severe("Unable to find the Luck of the Sea enchantment, " +
                "mcMMO will not function properly.");
        throw new IllegalStateException("Unable to find the Luck of the Sea enchantment");
    }

    private Enchantment initFeatherFalling() {
//        if (Registry.ENCHANTMENT.match("feather_falling") != null) {
//            return Registry.ENCHANTMENT.match("feather_falling");
//        }

        // Look for the enchantment by name
        for (Enchantment enchantment : Registry.ENCHANTMENT) {
            if (enchantment.getKey().getKey().equalsIgnoreCase("FEATHER_FALLING")
                    || enchantment.getKey().getKey().equalsIgnoreCase("PROTECTION_FALL")
                    || enchantment.getName().equalsIgnoreCase("FEATHER_FALLING")
                    || enchantment.getName().equalsIgnoreCase("PROTECTION_FALL")) {
                return enchantment;
            }
        }

        pluginRef.getLogger().severe("Unable to find the Feather Falling enchantment, " +
                "mcMMO will not function properly.");
        throw new IllegalStateException("Unable to find the Feather Falling enchantment");
    }

    private Enchantment initInfinity() {
//        if (Registry.ENCHANTMENT.match("infinity") != null) {
//            return Registry.ENCHANTMENT.match("infinity");
//        }

        // Look for the enchantment by name
        for (Enchantment enchantment : Registry.ENCHANTMENT) {
            if (enchantment.getKey().getKey().equalsIgnoreCase("INFINITY")
                    || enchantment.getKey().getKey().equalsIgnoreCase("ARROW_INFINITE")
                    || enchantment.getName().equalsIgnoreCase("INFINITY")
                    || enchantment.getName().equalsIgnoreCase("ARROW_INFINITE")) {
                return enchantment;
            }
        }

        pluginRef.getLogger().severe("Unable to find the Infinity enchantment, " +
                "mcMMO will not function properly.");
        throw new IllegalStateException("Unable to find the Infinity enchantment");
    }

    private Enchantment initEfficiency() {
//        if (Registry.ENCHANTMENT.match("efficiency") != null) {
//            return Registry.ENCHANTMENT.match("efficiency");
//        }

        // Look for the enchantment by name
        for (Enchantment enchantment : Registry.ENCHANTMENT) {
            if (enchantment.getKey().getKey().equalsIgnoreCase("EFFICIENCY")
                    || enchantment.getKey().getKey().equalsIgnoreCase("DIG_SPEED")
                    || enchantment.getName().equalsIgnoreCase("EFFICIENCY")
                    || enchantment.getName().equalsIgnoreCase("DIG_SPEED")) {
                return enchantment;
            }
        }

        pluginRef.getLogger().severe("Unable to find the Efficiency enchantment, " +
                "mcMMO will not function properly.");
        throw new IllegalStateException("Unable to find the Efficiency enchantment");
    }

    private Enchantment initUnbreaking() {
//        if (Registry.ENCHANTMENT.match("unbreaking") != null) {
//            return Registry.ENCHANTMENT.match("unbreaking");
//        }

        // Look for the enchantment by name
        for (Enchantment enchantment : Registry.ENCHANTMENT) {
            if (enchantment.getKey().getKey().equalsIgnoreCase("UNBREAKING")
                    || enchantment.getKey().getKey().equalsIgnoreCase("DURABILITY")
                    || enchantment.getName().equalsIgnoreCase("UNBREAKING")
                    || enchantment.getName().equalsIgnoreCase("DURABILITY")) {
                return enchantment;
            }
        }

        pluginRef.getLogger().severe("Unable to find the Unbreaking enchantment, " +
                "mcMMO will not function properly.");
        throw new IllegalStateException("Unable to find the Unbreaking enchantment");
    }

    /**
     * Get the efficiency enchantment
     *
     * @return The efficiency enchantment
     */
    public Enchantment getEfficiency() {
        return efficiency;
    }

    public Enchantment getUnbreaking() {
        return unbreaking;
    }

    public Enchantment getInfinity() {
        return infinity;
    }

    public Enchantment getFeatherFalling() {
        return featherFalling;
    }

    public Enchantment getLuckOfTheSea() {
        return luckOfTheSea;
    }
}