SpigotAttributeModifiersItemComponentLoader.java

package fr.maxlego08.menu.loader.components.spigot;

import fr.maxlego08.menu.api.MenuPlugin;
import fr.maxlego08.menu.api.attribute.AttributeMergeStrategy;
import fr.maxlego08.menu.api.attribute.AttributeWrapper;
import fr.maxlego08.menu.api.configuration.Configuration;
import fr.maxlego08.menu.api.context.MenuItemStackContext;
import fr.maxlego08.menu.api.itemstack.ItemComponent;
import fr.maxlego08.menu.api.itemstack.components.AttributeModifiersComponent;
import fr.maxlego08.menu.api.loader.ItemComponentLoader;
import fr.maxlego08.menu.zcore.logger.Logger;
import org.bukkit.NamespacedKey;
import org.bukkit.Registry;
import org.bukkit.attribute.Attribute;
import org.bukkit.attribute.AttributeModifier;
import org.bukkit.configuration.ConfigurationSection;
import org.bukkit.configuration.file.YamlConfiguration;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;

import java.io.File;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

public class SpigotAttributeModifiersItemComponentLoader extends ItemComponentLoader {
    private final MenuPlugin plugin;

    public SpigotAttributeModifiersItemComponentLoader(MenuPlugin plugin){
        super("attribute-modifiers");
        this.plugin = plugin;
    }

    @Override
    public @Nullable ItemComponent load(@NotNull MenuItemStackContext context, @NotNull File file, @NotNull YamlConfiguration configuration, @NotNull String path, @Nullable ConfigurationSection componentSection) {
        if (componentSection == null) return null;
        String mergeStrategyStr = componentSection.getString("attribute-merge-strategy", "");
        AttributeMergeStrategy mergeStrategy;
        try {
            mergeStrategy = AttributeMergeStrategy.valueOf(mergeStrategyStr.toUpperCase());
        } catch (IllegalArgumentException e) {
            mergeStrategy = AttributeMergeStrategy.ADD;
        }
        List<Map<?, ?>> mapList = componentSection.getMapList("modifiers");
        List<AttributeWrapper> modifiersWrapper = new ArrayList<>();
        for (Map<?, ?> rawMap : mapList) {
            @SuppressWarnings("unchecked")
            Map<String, Object> map = (Map<String, Object>) rawMap;
            String type = (String) map.get("type");
            if (type == null) continue;
            Attribute attribute;
            try {
                NamespacedKey key = NamespacedKey.fromString(type.toLowerCase());
                if (key == null) continue;
                attribute = Registry.ATTRIBUTE.get(key);
            } catch (IllegalArgumentException e) {
                continue;
            }
            if (attribute == null) continue;
            try {
                AttributeModifier deserialize = AttributeModifier.deserialize(map);
                modifiersWrapper.add(new AttributeWrapper(attribute, deserialize.getOperation(), deserialize.getAmount(), deserialize.getSlotGroup(), deserialize.getKey()));
            } catch (IllegalArgumentException e) {
                if (Configuration.enableDebug){
                    Logger.info("Error deserializing attribute modifier for attribute " + attribute.name() + ": " + e.getMessage());
                    e.printStackTrace();
                }

            }
        }
        return modifiersWrapper.isEmpty() ? null : new AttributeModifiersComponent(this.plugin,modifiersWrapper, mergeStrategy);
    }
}