class="kw">package com.hypixel.hytale.server.core.io.stream;

class="kw">import com.hypixel.hytale.protocol.io.ChannelConnection;
class="kw">import com.hypixel.hytale.protocol.io.ConnectionHandler;
class="kw">import com.hypixel.hytale.protocol.packets.stream.StreamType;
class="kw">import com.hypixel.hytale.server.core.io.PacketHandler;
class="kw">import io.netty.handler.codec.quic.QuicStreamPriority;
class="kw">import java.util.Map;
class="kw">import java.util.concurrent.ConcurrentHashMap;
class="kw">import javax.annotation.Nonnull;
class="kw">import javax.annotation.Nullable;

class="kw">public class StreamManager {
   class="kw">public class="kw">static class="kw">final int MAX_AUXILIARY_STREAMS = 4;
   class="kw">public class="kw">static class="kw">final QuicStreamPriority GAME_STREAM_PRIORITY = new QuicStreamPriority(0, false);
   class="kw">public class="kw">static class="kw">final QuicStreamPriority DEFAULT_AUXILIARY_PRIORITY = new QuicStreamPriority(64, false);
   class="kw">private class="kw">static class="kw">final StreamManager INSTANCE = new StreamManager();
   class="kw">private class="kw">final Map<StreamType, StreamManager.StreamRegistration> handlers = new ConcurrentHashMap<>();

   StreamManager() {
   }

   @Nonnull
   class="kw">public class="kw">static StreamManager getInstance() {
      class="kw">return INSTANCE;
   }

   class="kw">public void registerHandler(@Nonnull StreamType var1, @Nonnull StreamManager.StreamHandlerFactory var2) {
      this.registerHandler(var1, var2, DEFAULT_AUXILIARY_PRIORITY);
   }

   class="kw">public void registerHandler(@Nonnull StreamType var1, @Nonnull StreamManager.StreamHandlerFactory var2, @Nonnull QuicStreamPriority var3) {
      if (var1 == StreamType.Game) {
         throw new IllegalArgumentException("Cannot register handler for Game stream type - it uses the main pipeline");
      }

      this.handlers.put(var1, new StreamManager.StreamRegistration(var2, var3));
   }

   class="kw">public void unregisterHandler(@Nonnull StreamType var1) {
      this.handlers.remove(var1);
   }

   class="kw">public boolean isSupported(@Nonnull StreamType var1) {
      class="kw">return var1 == StreamType.Game || this.handlers.containsKey(var1);
   }

   @Nullable
   class="kw">public ConnectionHandler createHandler(@Nonnull StreamType var1, @Nonnull PacketHandler var2, @Nonnull ChannelConnection var3) {
      StreamManager.StreamRegistration var4 = this.handlers.get(var1);
      class="kw">return var4 != null ? var4.factory().create(var2, var3) : null;
   }

   class="kw">public void clearAll() {
      this.handlers.clear();
   }

   @Nonnull
   class="kw">public QuicStreamPriority getStreamPriority(@Nonnull StreamType var1) {
      StreamManager.StreamRegistration var2 = this.handlers.get(var1);
      class="kw">return var2 != null ? var2.priority() : DEFAULT_AUXILIARY_PRIORITY;
   }

   @FunctionalInterface
   class="kw">public class="kw">interface StreamHandlerFactory {
      @Nonnull
      ConnectionHandler create(@Nonnull PacketHandler var1, @Nonnull ChannelConnection var2);
   }

   class="kw">private record StreamRegistration(@Nonnull StreamManager.StreamHandlerFactory factory, @Nonnull QuicStreamPriority priority) {
      class="kw">private StreamRegistration {
      }
   }
}