class="kw">package com.hypixel.hytale.codec.store;

class="kw">import com.hypixel.hytale.codec.Codec;
class="kw">import java.util.Map;
class="kw">import java.util.concurrent.ConcurrentHashMap;
class="kw">import java.util.function.Supplier;
class="kw">import javax.annotation.Nullable;

class="kw">public class CodecStore {
   class="kw">public class="kw">static class="kw">final CodecStore STATIC = new CodecStore();
   class="kw">private class="kw">final CodecStore parent;
   class="kw">private class="kw">final Map<CodecKey<?>, Codec<?>> codecs = new ConcurrentHashMap<>();
   class="kw">private class="kw">final Map<CodecKey<?>, Supplier<Codec<?>>> codecSuppliers = new ConcurrentHashMap<>();

   class="kw">public CodecStore() {
      this.parent = STATIC;
   }

   class="kw">public CodecStore(CodecStore var1) {
      this.parent = var1;
   }

   @Nullable
   class="kw">public <T> Codec<T> getCodec(CodecKey<T> var1) {
      Codec var2 = (Codec<T>)this.codecs.get(var1);
      if (var2 != null) {
         class="kw">return var2;
      }

      Supplier var3 = this.codecSuppliers.get(var1);
      if (var3 != null) {
         var2 = (Codec<T>)var3.get();
      }

      if (var2 != null) {
         class="kw">return var2;
      } else {
         class="kw">return this.parent != null ? this.parent.getCodec(var1) : null;
      }
   }

   class="kw">public <T> void putCodec(CodecKey<T> var1, Codec<T> var2) {
      this.codecs.put(var1, var2);
   }

   class="kw">public <T> Codec<?> removeCodec(CodecKey<T> var1) {
      class="kw">return this.codecs.remove(var1);
   }

   class="kw">public <T> void putCodecSupplier(CodecKey<T> var1, Supplier<Codec<T>> var2) {
      this.codecSuppliers.put(var1, var2);
   }

   class="kw">public <T> Supplier<Codec<?>> removeCodecSupplier(CodecKey<T> var1) {
      class="kw">return this.codecSuppliers.remove(var1);
   }
}