From 4c878facbe9fb64a6a3d48df44e4c85d704db575 Mon Sep 17 00:00:00 2001
From: vytskalt <vytskalt@protonmail.com>
Date: Mon, 26 Feb 2024 19:49:20 +0200
Subject: [PATCH] Add RenderType enum wrapper

---
 .../protocol/events/AbstractStructure.java       | 11 +++++++++++
 .../protocol/wrappers/EnumWrappers.java          | 16 ++++++++++++++++
 2 files changed, 27 insertions(+)

diff --git a/src/main/java/com/comphenix/protocol/events/AbstractStructure.java b/src/main/java/com/comphenix/protocol/events/AbstractStructure.java
index 5b94a603..4a16d15c 100644
--- a/src/main/java/com/comphenix/protocol/events/AbstractStructure.java
+++ b/src/main/java/com/comphenix/protocol/events/AbstractStructure.java
@@ -867,6 +867,17 @@ public abstract class AbstractStructure {
                 EnumWrappers.getDisplaySlotConverter());
     }
 
+    /**
+     * Retrieve a read/write structure for the RenderType enum.
+     * @return A modifier for RenderType enum fields.
+     */
+    public StructureModifier<EnumWrappers.RenderType> getRenderTypes() {
+        return structureModifier.withType(
+                EnumWrappers.getRenderTypeClass(),
+                EnumWrappers.getRenderTypeConverter());
+    }
+
+
     /**
      * Retrieve a read/write structure for the MinecraftKey class.
      * @return A modifier for MinecraftKey fields.
diff --git a/src/main/java/com/comphenix/protocol/wrappers/EnumWrappers.java b/src/main/java/com/comphenix/protocol/wrappers/EnumWrappers.java
index dfb46dff..10d2f2ff 100644
--- a/src/main/java/com/comphenix/protocol/wrappers/EnumWrappers.java
+++ b/src/main/java/com/comphenix/protocol/wrappers/EnumWrappers.java
@@ -485,6 +485,11 @@ public abstract class EnumWrappers {
         TEAM_WHITE;
     }
 
+    public enum RenderType {
+        INTEGER,
+        HEARTS
+    }
+
     private static Class<?> PROTOCOL_CLASS = null;
     private static Class<?> CLIENT_COMMAND_CLASS = null;
     private static Class<?> CHAT_VISIBILITY_CLASS = null;
@@ -507,6 +512,7 @@ public abstract class EnumWrappers {
     private static Class<?> CHAT_TYPE_CLASS = null;
     private static Class<?> ENTITY_POSE_CLASS = null;
     private static Class<?> DISPLAY_SLOT_CLASS = null;
+    private static Class<?> RENDER_TYPE_CLASS = null;
 
     private static boolean INITIALIZED = false;
     private static Map<Class<?>, EquivalentConverter<?>> FROM_NATIVE = new HashMap<>();
@@ -591,6 +597,7 @@ public abstract class EnumWrappers {
         CHAT_TYPE_CLASS = getEnum(PacketType.Play.Server.CHAT.getPacketClass(), 0);
         ENTITY_POSE_CLASS = MinecraftReflection.getNullableNMS("world.entity.EntityPose", "world.entity.Pose", "EntityPose");
         DISPLAY_SLOT_CLASS = MinecraftReflection.getNullableNMS("world.scores.DisplaySlot");
+        RENDER_TYPE_CLASS = MinecraftReflection.getNullableNMS("world.scores.criteria.IScoreboardCriteria$EnumScoreboardHealthDisplay", "IScoreboardCriteria$EnumScoreboardHealthDisplay");
 
         associate(PROTOCOL_CLASS, Protocol.class, getProtocolConverter());
         associate(CLIENT_COMMAND_CLASS, ClientCommand.class, getClientCommandConverter());
@@ -768,6 +775,11 @@ public abstract class EnumWrappers {
         return DISPLAY_SLOT_CLASS;
     }
 
+    public static Class<?> getRenderTypeClass() {
+        initialize();
+        return RENDER_TYPE_CLASS;
+    }
+
     // Get the converters
     public static EquivalentConverter<Protocol> getProtocolConverter() {
         return new EnumConverter<>(getProtocolClass(), Protocol.class);
@@ -853,6 +865,10 @@ public abstract class EnumWrappers {
         return new EnumConverter<>(getDisplaySlotClass(), DisplaySlot.class);
     }
 
+    public static EquivalentConverter<RenderType> getRenderTypeConverter() {
+        return new EnumConverter<>(getRenderTypeClass(), RenderType.class);
+    }
+
     /**
      * @since 1.13+
      * @return {@link EnumConverter} or null (if bellow 1.13 / nms EnumPose class cannot be found)