class="kw">package com.hypixel.hytale.server.core.asset.type.entityeffect;

class="kw">import com.hypixel.hytale.assetstore.map.IndexedLookupTableAssetMap;
class="kw">import com.hypixel.hytale.protocol.ToClientPacket;
class="kw">import com.hypixel.hytale.protocol.UpdateType;
class="kw">import com.hypixel.hytale.protocol.packets.assets.UpdateEntityEffects;
class="kw">import com.hypixel.hytale.server.core.asset.packet.SimpleAssetPacketGenerator;
class="kw">import com.hypixel.hytale.server.core.asset.type.entityeffect.config.EntityEffect;
class="kw">import it.unimi.dsi.fastutil.ints.Int2ObjectOpenHashMap;
class="kw">import java.util.Map;
class="kw">import java.util.Set;
class="kw">import java.util.Map.Entry;
class="kw">import javax.annotation.Nonnull;

class="kw">public class EntityEffectPacketGenerator class="kw">extends SimpleAssetPacketGenerator<String, EntityEffect, IndexedLookupTableAssetMap<String, EntityEffect>> {
   class="kw">public EntityEffectPacketGenerator() {
   }

   @Nonnull
   class="kw">public ToClientPacket generateInitPacket(@Nonnull IndexedLookupTableAssetMap<String, EntityEffect> var1, @Nonnull Map<String, EntityEffect> var2) {
      UpdateEntityEffects var3 = new UpdateEntityEffects();
      var3.type = UpdateType.Init;
      var3.entityEffects = new Int2ObjectOpenHashMap(var2.size());

      for (Entry var5 : var2.entrySet()) {
         String var6 = var5.getKey();
         int var7 = var1.getIndex(var6);
         if (var7 == Integer.MIN_VALUE) {
            throw new IllegalArgumentException("Unknown key! " + var6);
         }

         var3.entityEffects.put(var7, var5.getValue().toPacket());
      }

      var3.maxId = var1.getNextIndex();
      class="kw">return var3;
   }

   @Nonnull
   class="kw">protected ToClientPacket generateUpdatePacket(@Nonnull IndexedLookupTableAssetMap<String, EntityEffect> var1, @Nonnull Map<String, EntityEffect> var2) {
      UpdateEntityEffects var3 = new UpdateEntityEffects();
      var3.type = UpdateType.AddOrUpdate;
      var3.entityEffects = new Int2ObjectOpenHashMap(var2.size());

      for (Entry var5 : var2.entrySet()) {
         String var6 = var5.getKey();
         int var7 = var1.getIndex(var6);
         if (var7 == Integer.MIN_VALUE) {
            throw new IllegalArgumentException("Unknown key! " + var6);
         }

         var3.entityEffects.put(var7, var5.getValue().toPacket());
      }

      var3.maxId = var1.getNextIndex();
      class="kw">return var3;
   }

   @Nonnull
   class="kw">protected ToClientPacket generateRemovePacket(@Nonnull IndexedLookupTableAssetMap<String, EntityEffect> var1, @Nonnull Set<String> var2) {
      UpdateEntityEffects var3 = new UpdateEntityEffects();
      var3.type = UpdateType.Remove;
      var3.entityEffects = new Int2ObjectOpenHashMap(var2.size());

      for (String var5 : var2) {
         int var6 = var1.getIndex(var5);
         if (var6 == Integer.MIN_VALUE) {
            throw new IllegalArgumentException("Unknown key! " + var5);
         }

         var3.entityEffects.put(var6, null);
      }

      var3.maxId = var1.getNextIndex();
      class="kw">return var3;
   }
}