DreamEnchantmentRegistry.java

Go to the documentation of this file.

/*
 * MIT License
 *
 * Copyright (c) 2025 Dreamfire Studio
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in all
 * copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 */
package com.dreamfirestudios.dreamcore.DreamEnchantment;
 
import org.bukkit.NamespacedKey;
import org.bukkit.enchantments.Enchantment;
import org.bukkit.inventory.ItemStack;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
 
import java.util.*;
 
 
public final class DreamEnchantmentRegistry {
 
    private DreamEnchantmentRegistry() {}
 
    private static final Map<NamespacedKey, IDreamEnchantment> ENCHANTS = new LinkedHashMap<>();
 
    private static final Map<NamespacedKey, Enchantment> WRAPPERS = new LinkedHashMap<>();
 
 
    public static synchronized @NotNull Enchantment register(@NotNull IDreamEnchantment impl) {
        var key = impl.getKey();
        var existing = WRAPPERS.get(key);
        if (existing != null) return existing;
 
        ENCHANTS.put(key, impl);
        var wrapper = impl.returnEnchantment();
        WRAPPERS.put(key, wrapper);
        return wrapper;
    }
 
 
    public static @Nullable Enchantment getWrapper(@NotNull NamespacedKey key) {
        return WRAPPERS.get(key);
    }
 
 
    public static @Nullable IDreamEnchantment get(@NotNull NamespacedKey key) {
        return ENCHANTS.get(key);
    }
 
 
    public static @NotNull Collection<Enchantment> allWrappers() {
        return Collections.unmodifiableCollection(WRAPPERS.values());
    }
 
 
    public static @NotNull Collection<IDreamEnchantment> all() {
        return Collections.unmodifiableCollection(ENCHANTS.values());
    }
 
    // ---------- item helpers ----------
 
 
    public static @NotNull List<IDreamEnchantment> findOn(@Nullable ItemStack stack) {
        if (stack == null || stack.getItemMeta() == null) return List.of();
        var meta = stack.getItemMeta();
        var out = new ArrayList<IDreamEnchantment>();
        for (var impl : ENCHANTS.values()) {
            var ench = impl.returnEnchantment(); // NOTE: creates a new wrapper per call (see suggestion).
            if (meta.hasEnchant(ench)) out.add(impl);
        }
        return Collections.unmodifiableList(out);
    }
 
 
    public static boolean add(@NotNull ItemStack stack, @NotNull NamespacedKey key) {
        var impl = ENCHANTS.get(key);
        return impl != null && impl.addToItem(stack);
    }
}