BuilderObjectMapHelper.java

package com.hypixel.hytale.server.npc.asset.builder;

import com.google.gson.JsonElement;
import com.hypixel.hytale.server.npc.util.expression.ExecutionContext;
import it.unimi.dsi.fastutil.objects.Object2ObjectLinkedOpenHashMap;
import java.util.Map;
import java.util.Objects;
import java.util.function.BiFunction;
import java.util.function.Function;
import javax.annotation.Nonnull;
import javax.annotation.Nullable;

public class BuilderObjectMapHelper<K, V> extends BuilderObjectArrayHelper<Map<K, V>, V> {
   private Function<V, K> id;

   public BuilderObjectMapHelper(Class classType, Function<V, K> id, BuilderContext owner) {
      super(classType, owner);
      this.id = id;
   }

   @Nullable
   public Map<K, V> build(@Nonnull BuilderSupport builderSupport) {
      if (this.hasNoElements()) {
         return null;
      } else {
         Map<K, V> objects = new Object2ObjectLinkedOpenHashMap<K, V>();

         for(BuilderObjectReferenceHelper<V> builderObjectReferenceHelper : this.builders) {
            if (!builderObjectReferenceHelper.excludeFromRegularBuild()) {
               V value = builderObjectReferenceHelper.build(builderSupport);
               K key = (K)this.id.apply(value);
               if (objects.containsKey(key)) {
                  String var10002 = String.valueOf(key);
                  throw new IllegalArgumentException("Duplicate key \"" + var10002 + "\" at " + this.getBreadCrumbs() + ": " + this.builderParameters.getFileName());
               }

               objects.put(key, value);
            }
         }

         return objects;
      }
   }

   public void readConfig(@Nonnull JsonElement data, @Nonnull BuilderManager builderManager, @Nonnull BuilderParameters builderParameters, @Nonnull BuilderValidationHelper builderValidationHelper) {
      super.readConfig(data, builderManager, builderParameters, builderValidationHelper);
   }

   @Nullable
   public <T, U> T testEach(@Nonnull BiFunction<Builder<V>, U, T> test, @Nonnull BuilderManager builderManager, ExecutionContext executionContext, U meta, T successResult, T emptyResult, Builder<?> parentSpawnable) {
      if (this.hasNoElements()) {
         return emptyResult;
      } else {
         for(BuilderObjectReferenceHelper<V> builderObjectReferenceHelper : this.builders) {
            T result = (T)test.apply(builderObjectReferenceHelper.getBuilder(builderManager, executionContext, parentSpawnable), meta);
            if (!Objects.equals(result, successResult)) {
               return result;
            }
         }

         return successResult;
      }
   }
}