mcMMO/src/main/java/com/gmail/nossr50/mods/ModChecks.java

package com.gmail.nossr50.mods;

import org.bukkit.block.BlockState;
import org.bukkit.entity.Entity;
import org.bukkit.inventory.ItemStack;

import com.gmail.nossr50.config.Config;
import com.gmail.nossr50.mods.config.CustomArmorConfig;
import com.gmail.nossr50.mods.config.CustomBlocksConfig;
import com.gmail.nossr50.mods.config.CustomEntityConfig;
import com.gmail.nossr50.mods.config.CustomToolsConfig;
import com.gmail.nossr50.mods.datatypes.CustomBlock;
import com.gmail.nossr50.mods.datatypes.CustomEntity;
import com.gmail.nossr50.mods.datatypes.CustomItem;
import com.gmail.nossr50.mods.datatypes.CustomTool;

public final class ModChecks {
    private static Config configInstance = Config.getInstance();
    private static boolean customToolsEnabled = configInstance.getToolModsEnabled();
    private static boolean customArmorEnabled = configInstance.getArmorModsEnabled();
    private static boolean customBlocksEnabled = configInstance.getBlockModsEnabled();
    private static boolean customEntitiesEnabled = configInstance.getEntityModsEnabled();

    private ModChecks() {}

    /**
     * Get the custom armor associated with an item.
     *
     * @param item The item to check
     * @return the armor if it exists, null otherwise
     */
    public static CustomItem getArmorFromItemStack(ItemStack item) {
        return CustomArmorConfig.getInstance().customArmor.get(item.getTypeId());
    }

    /**
     * Get the custom tool associated with an item.
     *
     * @param item The item to check
     * @return the tool if it exists, null otherwise
     */
    public static CustomTool getToolFromItemStack(ItemStack item) {
        return CustomToolsConfig.getInstance().customTools.get(item.getTypeId());
    }

    /**
     * Get the custom block associated with an block.
     *
     * @param blockState The block to check
     * @return the block if it exists, null otherwise
     */
    public static CustomBlock getCustomBlock(BlockState blockState) {
        if (customBlocksEnabled) {
            ItemStack item = blockState.getData().toItemStack();

            if (CustomBlocksConfig.getInstance().customItems.contains(item)) {
                for (CustomBlock block : CustomBlocksConfig.getInstance().customBlocks) {
                    if ((block.getItemID() == blockState.getTypeId()) && (block.getDataValue() == blockState.getRawData())) {
                        return block;
                    }
                }
            }
        }

        return null;
    }

    public static CustomEntity getCustomEntity(Entity entity) {
        if (!CustomEntityConfig.getInstance().customEntityIds.contains(entity.getEntityId()) && !CustomEntityConfig.getInstance().customEntityTypes.contains(entity.getType())) {
            return null;
        }

        for (CustomEntity customEntity : CustomEntityConfig.getInstance().customEntities) {
            if ((customEntity.getEntityID() == entity.getEntityId()) && (customEntity.getEntityType() == entity.getType())) {
                return customEntity;
            }
        }

        return null;
    }

    /**
     * Check if a custom block is a woodcutting block.
     *
     * @param block The block to check
     * @return true if the block represents a log, false otherwise
     */
    public static boolean isCustomWoodcuttingBlock(BlockState blockState) {
        if (customBlocksEnabled) {
            ItemStack item = blockState.getData().toItemStack();

            if (CustomBlocksConfig.getInstance().customWoodcuttingBlocks.contains(item)) {
                for (CustomBlock block : CustomBlocksConfig.getInstance().customBlocks) {
                    if ((block.getItemID() == blockState.getTypeId()) && (block.getDataValue() == blockState.getRawData())) {
                        return true;
                    }
                }
            }
        }

        return false;
    }

    /**
     * Check if a custom block should not activate abilites.
     *
     * @param block The block to check
     * @return true if the block represents an ability block, false otherwise
     */
    public static boolean isCustomAbilityBlock(BlockState blockState) {
        if (customBlocksEnabled) {
            ItemStack item = blockState.getData().toItemStack();

            if (CustomBlocksConfig.getInstance().customAbilityBlocks.contains(item)) {
                for (CustomBlock block : CustomBlocksConfig.getInstance().customBlocks) {
                    if ((block.getItemID() == blockState.getTypeId()) && (block.getDataValue() == blockState.getRawData())) {
                        return true;
                    }
                }
            }
        }

        return false;
    }

    /**
     * Check if a custom block is a mining block.
     *
     * @param block The block to check
     * @return true if the block is custom, false otherwise
     */
    public static boolean isCustomMiningBlock(BlockState blockState) {
        if (customBlocksEnabled) {
            ItemStack item = blockState.getData().toItemStack();

            if (CustomBlocksConfig.getInstance().customMiningBlocks.contains(item)) {
                for (CustomBlock block : CustomBlocksConfig.getInstance().customBlocks) {
                    if ((block.getItemID() == blockState.getTypeId()) && (block.getDataValue() == blockState.getRawData())) {
                        return true;
                    }
                }
            }
        }

        return false;
    }

    /**
     * Check if a custom block is an excavation block.
     *
     * @param block The block to check
     * @return true if the block is custom, false otherwise
     */
    public static boolean isCustomExcavationBlock(BlockState blockState) {
        if (customBlocksEnabled) {
            ItemStack item = blockState.getData().toItemStack();

            if (CustomBlocksConfig.getInstance().customExcavationBlocks.contains(item)) {
                for (CustomBlock block : CustomBlocksConfig.getInstance().customBlocks) {
                    if ((block.getItemID() == blockState.getTypeId()) && (block.getDataValue() == blockState.getRawData())) {
                        return true;
                    }
                }
            }
        }

        return false;
    }

    /**
     * Check if a custom block is an herbalism block.
     *
     * @param blockState The block to check
     * @return true if the block is custom, false otherwise
     */
    public static boolean isCustomHerbalismBlock(BlockState blockState) {
        if (customBlocksEnabled) {
            ItemStack item = blockState.getData().toItemStack();

            if (CustomBlocksConfig.getInstance().customHerbalismBlocks.contains(item)) {
                for (CustomBlock block : CustomBlocksConfig.getInstance().customBlocks) {
                    if ((block.getItemID() == blockState.getTypeId()) && (block.getDataValue() == blockState.getRawData())) {
                        return true;
                    }
                }
            }
        }

        return false;
    }

    /**
     * Check if a custom block is a leaf block.
     *
     * @param block The block to check
     * @return true if the block represents leaves, false otherwise
     */
    public static boolean isCustomLeafBlock(BlockState blockState) {
        if (customBlocksEnabled) {
            ItemStack item = blockState.getData().toItemStack();

            if (CustomBlocksConfig.getInstance().customLeaves.contains(item)) {
                for (CustomBlock block : CustomBlocksConfig.getInstance().customBlocks) {
                    if ((block.getItemID() == blockState.getTypeId()) && (block.getDataValue() == blockState.getRawData())) {
                        return true;
                    }
                }
            }
        }

        return false;
    }

    /**
     * Check if a custom block is a log block.
     *
     * @param block The block to check
     * @return true if the block represents a log, false otherwise
     */
    public static boolean isCustomLogBlock(BlockState blockState) {
        if (customBlocksEnabled) {
            ItemStack item = blockState.getData().toItemStack();

            if (CustomBlocksConfig.getInstance().customLogs.contains(item)) {
                for (CustomBlock block : CustomBlocksConfig.getInstance().customBlocks) {
                    if ((block.getItemID() == blockState.getTypeId()) && (block.getDataValue() == blockState.getRawData())) {
                        return true;
                    }
                }
            }
        }

        return false;
    }

    /**
     * Check if a custom block is an ore block.
     *
     * @param blockState The block to check
     * @return true if the block represents an ore, false otherwise
     */
    public static boolean isCustomOreBlock(BlockState blockState) {
        if (customBlocksEnabled) {
            ItemStack item = blockState.getData().toItemStack();

            if (CustomBlocksConfig.getInstance().customOres.contains(item)) {
                for (CustomBlock block : CustomBlocksConfig.getInstance().customBlocks) {
                    if ((block.getItemID() == blockState.getTypeId()) && (block.getDataValue() == blockState.getRawData())) {
                        return true;
                    }
                }
            }
        }

        return false;
    }

    /**
     * Checks to see if an item is a custom tool.
     *
     * @param item Item to check
     * @return true if the item is a custom tool, false otherwise
     */
    public static boolean isCustomTool(ItemStack item) {
        if (customToolsEnabled && CustomToolsConfig.getInstance().customTools.containsKey(item.getTypeId())) {
            return true;
        }

        return false;
    }

    /**
     * Checks to see if an item is custom armor.
     *
     * @param item Item to check
     * @return true if the item is custom armor, false otherwise
     */
    public static boolean isCustomArmor(ItemStack item) {
        if (customArmorEnabled && CustomArmorConfig.getInstance().customArmor.containsKey(item.getTypeId())) {
            return true;
        }

        return false;
    }

    public static boolean isCustomEntity(Entity entity) {
        if (customEntitiesEnabled && CustomEntityConfig.getInstance().customEntityIds.contains(entity.getEntityId())) {
            return true;
        }

        return false;
    }
}