From 5cad29c75402fe7194f1ad19975633c05cd6fa9a Mon Sep 17 00:00:00 2001
From: Nassim Jahnke <nassim@njahnke.dev>
Date: Fri, 24 Feb 2023 10:07:25 +0100
Subject: [PATCH] Expose PacketTypesProvider in Protocol

---
 .../viaversion/api/protocol/AbstractProtocol.java     |  5 +++++
 .../viaversion/viaversion/api/protocol/Protocol.java  | 11 +++++++++++
 2 files changed, 16 insertions(+)

diff --git a/api/src/main/java/com/viaversion/viaversion/api/protocol/AbstractProtocol.java b/api/src/main/java/com/viaversion/viaversion/api/protocol/AbstractProtocol.java
index 6bc7b7aaa..4c93eb4cb 100644
--- a/api/src/main/java/com/viaversion/viaversion/api/protocol/AbstractProtocol.java
+++ b/api/src/main/java/com/viaversion/viaversion/api/protocol/AbstractProtocol.java
@@ -404,6 +404,11 @@ public abstract class AbstractProtocol<CU extends ClientboundPacketType, CM exte
         }
     }
 
+    @Override
+    public PacketTypesProvider<CU, CM, SM, SU> getPacketTypesProvider() {
+        return packetTypesProvider;
+    }
+
     @Override
     public @Nullable <T> T get(Class<T> objectClass) {
         //noinspection unchecked
diff --git a/api/src/main/java/com/viaversion/viaversion/api/protocol/Protocol.java b/api/src/main/java/com/viaversion/viaversion/api/protocol/Protocol.java
index de493719a..18de97e1d 100644
--- a/api/src/main/java/com/viaversion/viaversion/api/protocol/Protocol.java
+++ b/api/src/main/java/com/viaversion/viaversion/api/protocol/Protocol.java
@@ -22,6 +22,7 @@
  */
 package com.viaversion.viaversion.api.protocol;
 
+import com.google.common.annotations.Beta;
 import com.viaversion.viaversion.api.connection.UserConnection;
 import com.viaversion.viaversion.api.data.MappingData;
 import com.viaversion.viaversion.api.platform.providers.ViaProviders;
@@ -30,6 +31,7 @@ import com.viaversion.viaversion.api.protocol.packet.Direction;
 import com.viaversion.viaversion.api.protocol.packet.PacketWrapper;
 import com.viaversion.viaversion.api.protocol.packet.ServerboundPacketType;
 import com.viaversion.viaversion.api.protocol.packet.State;
+import com.viaversion.viaversion.api.protocol.packet.provider.PacketTypesProvider;
 import com.viaversion.viaversion.api.protocol.remapper.PacketHandler;
 import com.viaversion.viaversion.api.protocol.remapper.PacketRemapper;
 import com.viaversion.viaversion.api.rewriter.EntityRewriter;
@@ -236,6 +238,15 @@ public interface Protocol<CU extends ClientboundPacketType, CM extends Clientbou
      */
     void transform(Direction direction, State state, PacketWrapper packetWrapper) throws Exception;
 
+    /**
+     * Returns a packet type provider for this protocol to get packet types by id.
+     * Depending on the Protocol, not every state may be populated.
+     *
+     * @return the packet types provider
+     */
+    @Beta
+    PacketTypesProvider<CU, CM, SM, SU> getPacketTypesProvider();
+
     /**
      * Returns a cached object by the given type if present.
      *