UltimateStacker/src/main/java/com/songoda/ultimatestacker/stackable/entity/custom/CustomEntityManager.java

package com.songoda.ultimatestacker.stackable.entity.custom;

import com.songoda.ultimatestacker.settings.Settings;
import com.songoda.ultimatestacker.stackable.entity.custom.entities.MythicMobsCustomEntity;
import org.bukkit.Bukkit;
import org.bukkit.entity.Entity;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

public class CustomEntityManager {

    public CustomEntityManager() {
        load();
    }

    private final List<CustomEntity> registeredCustomEntities = new ArrayList<>();

    public void load() {
        if (isEnabled("MythicMobs"))
            registeredCustomEntities.add(new MythicMobsCustomEntity());
    }

    public boolean isEnabled(String plugin) {
        return Bukkit.getPluginManager().isPluginEnabled(plugin)
                && Settings.ENABLED_CUSTOM_ENTITY_PLUGINS.getStringList().contains(plugin);
    }

    public CustomEntity getCustomEntity(Entity entity) {
        for (CustomEntity customEntity : registeredCustomEntities) {
            if (customEntity.isMatchingType(entity) && customEntity.isCustomEntity(entity)) {
                if (Settings.BLACKLISTED_CUSTOM_ENTITIES.getStringList()
                        .contains((customEntity.getPluginName() + "_" + customEntity.getNBTIdentifier(entity)).toLowerCase()))
                    continue;
                return customEntity;
            }
        }
        return null;
    }

    public List<CustomEntity> getRegisteredCustomEntities() {
        return Collections.unmodifiableList(registeredCustomEntities);
    }

    public boolean isCustomEntity(Entity entity) {
        return getCustomEntity(entity) != null && getCustomEntity(entity).isCustomEntity(entity);
    }
}