// 
// Decompiled by Procyon v0.6.0
// 

package com.hypixel.hytale.server.core.modules.item;

import com.hypixel.hytale.assetstore.AssetMap;
import java.util.Set;
import com.hypixel.hytale.assetstore.AssetUpdateQuery;
import java.util.Iterator;
import com.hypixel.hytale.protocol.ItemBase;
import it.unimi.dsi.fastutil.objects.Object2ObjectOpenHashMap;
import com.hypixel.hytale.protocol.UpdateType;
import com.hypixel.hytale.protocol.packets.assets.UpdateItems;
import com.hypixel.hytale.protocol.Packet;
import javax.annotation.Nonnull;
import java.util.Map;
import com.hypixel.hytale.assetstore.map.DefaultAssetMap;
import com.hypixel.hytale.server.core.asset.type.item.config.Item;
import com.hypixel.hytale.server.core.asset.packet.AssetPacketGenerator;

public class ItemPacketGenerator extends AssetPacketGenerator<String, Item, DefaultAssetMap<String, Item>>
{
    @Nonnull
    @Override
    public Packet generateInitPacket(final DefaultAssetMap<String, Item> assetMap, @Nonnull final Map<String, Item> assets) {
        final UpdateItems packet = new UpdateItems();
        packet.type = UpdateType.Init;
        packet.items = new Object2ObjectOpenHashMap<String, ItemBase>();
        for (final Map.Entry<String, Item> entry : assets.entrySet()) {
            packet.items.put(entry.getKey(), entry.getValue().toPacket());
        }
        packet.updateModels = true;
        packet.updateIcons = true;
        return packet;
    }
    
    @Nonnull
    @Override
    public Packet generateUpdatePacket(final DefaultAssetMap<String, Item> assetMap, @Nonnull final Map<String, Item> loadedAssets, @Nonnull final AssetUpdateQuery query) {
        final UpdateItems packet = new UpdateItems();
        packet.type = UpdateType.AddOrUpdate;
        packet.items = new Object2ObjectOpenHashMap<String, ItemBase>();
        for (final Map.Entry<String, Item> entry : loadedAssets.entrySet()) {
            packet.items.put(entry.getKey(), entry.getValue().toPacket());
        }
        final AssetUpdateQuery.RebuildCache rebuildCache = query.getRebuildCache();
        packet.updateModels = (rebuildCache.isBlockTextures() || rebuildCache.isModels());
        packet.updateIcons = rebuildCache.isItemIcons();
        return packet;
    }
    
    @Nonnull
    @Override
    public Packet generateRemovePacket(final DefaultAssetMap<String, Item> assetMap, @Nonnull final Set<String> removed, @Nonnull final AssetUpdateQuery query) {
        final UpdateItems packet = new UpdateItems();
        packet.type = UpdateType.Remove;
        packet.removedItems = removed.toArray(String[]::new);
        final AssetUpdateQuery.RebuildCache rebuildCache = query.getRebuildCache();
        packet.updateModels = (rebuildCache.isBlockTextures() || rebuildCache.isModels());
        packet.updateIcons = rebuildCache.isItemIcons();
        return packet;
    }
}