studio.magemonkey.divinity.stats.items.requirements.ItemRequirements Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of divinity Show documentation
Custom items, combat, and more!
The newest version!
package studio.magemonkey.divinity.stats.items.requirements;

import studio.magemonkey.codex.util.ItemUT;
import studio.magemonkey.divinity.stats.items.requirements.api.ItemRequirement;
import studio.magemonkey.divinity.stats.items.requirements.api.UserRequirement;
import org.bukkit.entity.Player;
import org.bukkit.inventory.ItemStack;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;

import java.util.Collection;
import java.util.HashMap;
import java.util.Map;

public class ItemRequirements {

    public static final Map> USER_REQUIREMENTS;
    public static final Map> ITEM_REQUIREMENTS;

    static {
        USER_REQUIREMENTS = new HashMap<>();
        ITEM_REQUIREMENTS = new HashMap<>();
    }

    public static void clear() {
        USER_REQUIREMENTS.clear();
        ITEM_REQUIREMENTS.clear();
    }

    public static void registerUserRequirement(@NotNull UserRequirement req) {
        USER_REQUIREMENTS.put(req.getId(), req);
    }

    @NotNull
    public static Collection> getUserRequirements() {
        return USER_REQUIREMENTS.values();
    }

    @Nullable
    public static UserRequirement getUserRequirement(@NotNull String id) {
        return USER_REQUIREMENTS.get(id.toLowerCase());
    }

    public static > boolean isRegisteredUser(@NotNull Class clazz) {
        return getUserRequirement(clazz) != null;
    }

    public static > boolean isRegisteredItem(@NotNull Class clazz) {
        return ItemRequirements.getItemRequirement(clazz) != null;
    }

    @SuppressWarnings("unchecked")
    @Nullable
    public static  T getUserRequirement(@NotNull Class clazz) {
        for (UserRequirement stat : ItemRequirements.getUserRequirements()) {
            Class clazz2 = stat.getClass();
            if (clazz.isAssignableFrom(clazz2)) {
                return (T) stat;
            }
        }
        return null;
    }

    public static void registerItemRequirement(@NotNull ItemRequirement req) {
        ITEM_REQUIREMENTS.put(req.getId(), req);
    }

    @NotNull
    public static Collection> getItemRequirements() {
        return ITEM_REQUIREMENTS.values();
    }

    @Nullable
    public static ItemRequirement getItemRequirement(@NotNull String id) {
        return ITEM_REQUIREMENTS.get(id.toLowerCase());
    }

    @SuppressWarnings("unchecked")
    @Nullable
    public static > T getItemRequirement(@NotNull Class clazz) {
        for (ItemRequirement stat : ItemRequirements.getItemRequirements()) {
            Class clazz2 = stat.getClass();
            if (clazz.isAssignableFrom(clazz2)) {
                return (T) stat;
            }
        }
        return null;
    }

    public static boolean canApply(
            @NotNull Player p, @NotNull ItemStack src, @NotNull ItemStack target) {

        for (ItemRequirement req : ItemRequirements.getItemRequirements()) {
            if (req.isApplied(src) && !req.canApply(src, target)) {
                req.getApplyMessage(src, target)
                        .replace("%target%", ItemUT.getItemName(target))
                        .replace("%source%", ItemUT.getItemName(src))
                        .replace("%player%", p.getName())
                        .send(p);
                return false;
            }
        }
        return true;
    }
}