NaturesAura/src/main/java/de/ellpeck/naturesaura/misc/WorldData.java

package de.ellpeck.naturesaura.misc;

import de.ellpeck.naturesaura.Helper;
import de.ellpeck.naturesaura.api.NaturesAuraAPI;
import de.ellpeck.naturesaura.api.misc.IWorldData;
import de.ellpeck.naturesaura.blocks.tiles.ItemStackHandlerNA;
import net.minecraft.nbt.NBTBase;
import net.minecraft.nbt.NBTTagCompound;
import net.minecraft.nbt.NBTTagList;
import net.minecraft.util.EnumFacing;
import net.minecraft.world.World;
import net.minecraftforge.common.capabilities.Capability;

import javax.annotation.Nonnull;
import javax.annotation.Nullable;
import java.util.HashMap;
import java.util.Map;

public class WorldData implements IWorldData {
    private final World world;
    private final Map<String, ItemStackHandlerNA> enderStorages = new HashMap<>();

    public WorldData(World world) {
        this.world = world;
    }

    @Override
    public boolean hasCapability(@Nonnull Capability<?> capability, @Nullable EnumFacing facing) {
        return capability == NaturesAuraAPI.capWorldData;
    }

    @Nullable
    @Override
    public <T> T getCapability(@Nonnull Capability<T> capability, @Nullable EnumFacing facing) {
        return capability == NaturesAuraAPI.capWorldData ? (T) this : null;
    }

    @Override
    public NBTTagCompound serializeNBT() {
        NBTTagCompound compound = new NBTTagCompound();

        NBTTagList storages = new NBTTagList();
        for (Map.Entry<String, ItemStackHandlerNA> entry : this.enderStorages.entrySet()) {
            ItemStackHandlerNA handler = entry.getValue();
            if (Helper.isEmpty(handler))
                continue;
            NBTTagCompound storageComp = handler.serializeNBT();
            storageComp.setString("name", entry.getKey());
            storages.appendTag(storageComp);
        }
        compound.setTag("storages", storages);

        return compound;
    }

    @Override
    public void deserializeNBT(NBTTagCompound compound) {
        this.enderStorages.clear();
        NBTTagList storages = compound.getTagList("storages", 10);
        for (NBTBase base : storages) {
            NBTTagCompound storageComp = (NBTTagCompound) base;
            ItemStackHandlerNA storage = this.getEnderStorage(storageComp.getString("name"));
            storage.deserializeNBT(storageComp);
        }
    }

    @Override
    public ItemStackHandlerNA getEnderStorage(String name) {
        return this.enderStorages.computeIfAbsent(name, n -> new ItemStackHandlerNA(27));
    }
}