Add Bukkit 1.21.6 support

Author: bloodmc

platform-bukkit/src/main/java/net/kyori/adventure/platform/bukkit/MinecraftComponentSerializer.java

@@ -29,6 +29,8 @@
 import java.lang.reflect.Field;
 import java.lang.reflect.Method;
 import java.lang.reflect.Modifier;
+import java.lang.reflect.ParameterizedType;
+import java.lang.reflect.Type;
 import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.Comparator;
@@ -95,7 +97,12 @@ public static boolean isSupported() {
     findMcClassName("network.chat.IChatBaseComponent"),
     findMcClassName("network.chat.Component")
   );
+  private static final @Nullable Class<?> CLASS_COMPONENT_SERIALIZATION = findClass(findMcClassName("network.chat.ComponentSerialization"));
   private static final @Nullable Class<?> CLASS_CRAFT_REGISTRY = findCraftClass("CraftRegistry");
+  private static final @Nullable Class<?> CLASS_HOLDERLOOKUP_PROVIDER = findClass(
+    findMcClassName("core.HolderLookup$Provider"), // Paper mapping
+    findMcClassName("core.HolderLookup$a") // Spigot mapping
+  );
   private static final @Nullable Class<?> CLASS_REGISTRY_ACCESS = findClass(
     findMcClassName("core.IRegistryCustom"),
     findMcClassName("core.RegistryAccess")
@@ -109,6 +116,10 @@ public static boolean isSupported() {
   private static final MethodHandle TEXT_SERIALIZER_SERIALIZE;
   private static final MethodHandle TEXT_SERIALIZER_DESERIALIZE_TREE;
   private static final MethodHandle TEXT_SERIALIZER_SERIALIZE_TREE;
+  private static final MethodHandle COMPONENTSERIALIZATION_CODEC_ENCODE;
+  private static final MethodHandle COMPONENTSERIALIZATION_CODEC_DECODE;
+  private static final Object REGISTRY_ACCESS_EMPTY;
+  private static final MethodHandle CREATE_SERIALIZATION_CONTEXT;
 
   static {
     Object gson = null;
@@ -117,6 +128,10 @@ public static boolean isSupported() {
     MethodHandle textSerializerSerialize = null;
     MethodHandle textSerializerDeserializeTree = null;
     MethodHandle textSerializerSerializeTree = null;
+    MethodHandle codecEncode = null;
+    MethodHandle codecDecode = null;
+    Object registryAccessEmpty = null;
+    MethodHandle createContext = null;
 
     try {
       if (CLASS_JSON_PARSER != null) {
@@ -218,6 +233,42 @@ public static boolean isSupported() {
           }
         }
       }
+      if (CLASS_REGISTRY_ACCESS != null) {
+        for (Field field : CLASS_REGISTRY_ACCESS.getDeclaredFields()) {
+          field.setAccessible(true);
+          if (Modifier.isStatic(field.getModifiers()) && CLASS_REGISTRY_ACCESS.isAssignableFrom(field.getType())) {
+            registryAccessEmpty = field.get(null);
+            break;
+          }
+        }
+        if (registryAccessEmpty != null && CLASS_HOLDERLOOKUP_PROVIDER != null) {
+          for (Method m : CLASS_HOLDERLOOKUP_PROVIDER.getDeclaredMethods()) {
+            m.setAccessible(true);
+            if (m.getParameterCount() == 1 && m.getParameterTypes()[0].getSimpleName().equals("DynamicOps") && m.getReturnType().getSimpleName().contains("RegistryOps")) {
+              createContext = lookup().unreflect(m).bindTo(registryAccessEmpty);
+              break;
+            }
+          }
+        }
+      }
+      if (CLASS_COMPONENT_SERIALIZATION != null) {
+        for (Field f : CLASS_COMPONENT_SERIALIZATION.getDeclaredFields()) {
+          if (Modifier.isStatic(f.getModifiers()) && f.getType().getSimpleName().equals("Codec")) {
+            f.setAccessible(true);
+            Object codecInstance = f.get(null);
+            Class<?> codecClass = codecInstance.getClass();
+            for (Method m : codecClass.getDeclaredMethods()) {
+              if (m.getName().equals("decode")) {
+                codecDecode = lookup().unreflect(m).bindTo(codecInstance);
+              } else if (m.getName().equals("encode")) {
+                codecEncode = lookup().unreflect(m).bindTo(codecInstance);
+              }
+            }
+            break;
+          }
+        }
+      }
+
     } catch (final Throwable error) {
       INITIALIZATION_ERROR.set(new UnsupportedOperationException("Error occurred during initialization", error));
     }
@@ -228,9 +279,13 @@ public static boolean isSupported() {
     TEXT_SERIALIZER_SERIALIZE = textSerializerSerialize;
     TEXT_SERIALIZER_DESERIALIZE_TREE = textSerializerDeserializeTree;
     TEXT_SERIALIZER_SERIALIZE_TREE = textSerializerSerializeTree;
+    COMPONENTSERIALIZATION_CODEC_ENCODE = codecEncode;
+    COMPONENTSERIALIZATION_CODEC_DECODE = codecDecode;
+    REGISTRY_ACCESS_EMPTY = registryAccessEmpty;
+    CREATE_SERIALIZATION_CONTEXT = createContext;
   }
 
-  private static final boolean SUPPORTED = MC_TEXT_GSON != null || (TEXT_SERIALIZER_DESERIALIZE != null && TEXT_SERIALIZER_SERIALIZE != null) || (TEXT_SERIALIZER_DESERIALIZE_TREE != null && TEXT_SERIALIZER_SERIALIZE_TREE != null);
+  private static final boolean SUPPORTED = MC_TEXT_GSON != null || (TEXT_SERIALIZER_DESERIALIZE != null && TEXT_SERIALIZER_SERIALIZE != null) || (TEXT_SERIALIZER_DESERIALIZE_TREE != null && TEXT_SERIALIZER_SERIALIZE_TREE != null) || (COMPONENTSERIALIZATION_CODEC_ENCODE != null && COMPONENTSERIALIZATION_CODEC_DECODE != null && REGISTRY_ACCESS_EMPTY != null && CREATE_SERIALIZATION_CONTEXT != null);
 
   @Override
   public @NotNull Component deserialize(final @NotNull Object input) {
@@ -242,6 +297,13 @@ public static boolean isSupported() {
         element = TEXT_SERIALIZER_SERIALIZE_TREE.invoke(input);
       } else if (MC_TEXT_GSON != null) {
         element = ((Gson) MC_TEXT_GSON).toJsonTree(input);
+      } else if (COMPONENTSERIALIZATION_CODEC_ENCODE != null && REGISTRY_ACCESS_EMPTY != null && CREATE_SERIALIZATION_CONTEXT != null) {
+        final Object jsonOps = Class.forName("com.mojang.serialization.JsonOps").getField("INSTANCE").get(null);
+        final Object serializationContext = CREATE_SERIALIZATION_CONTEXT.invoke(REGISTRY_ACCESS_EMPTY, jsonOps);
+        Object result = COMPONENTSERIALIZATION_CODEC_ENCODE.invoke(serializationContext, input, null);
+        Method getOrThrow = result.getClass().getMethod("getOrThrow", java.util.function.Function.class);
+        Object jsonElement = getOrThrow.invoke(result, (java.util.function.Function<Throwable, RuntimeException>) RuntimeException::new);
+        return gson().serializer().fromJson(jsonElement.toString(), Component.class);
       } else {
         return gson().deserialize((String) TEXT_SERIALIZER_SERIALIZE.invoke(input));
       }
@@ -268,6 +330,15 @@ public static boolean isSupported() {
       }
     } else {
       try {
+        if (COMPONENTSERIALIZATION_CODEC_DECODE != null && REGISTRY_ACCESS_EMPTY != null && CREATE_SERIALIZATION_CONTEXT != null) {
+          final Object jsonOps = Class.forName("com.mojang.serialization.JsonOps").getField("INSTANCE").get(null);
+          final Object serializationContext = CREATE_SERIALIZATION_CONTEXT.invoke(jsonOps);
+          Object result = COMPONENTSERIALIZATION_CODEC_DECODE.invoke(serializationContext, gson().serializeToTree(component));
+          Method getOrThrow = result.getClass().getMethod("getOrThrow", java.util.function.Function.class);
+          Object pair = getOrThrow.invoke(result, (java.util.function.Function<Throwable, RuntimeException>) RuntimeException::new);
+          Method getFirst = pair.getClass().getMethod("getFirst");
+          return getFirst.invoke(pair);
+        }
         return TEXT_SERIALIZER_DESERIALIZE.invoke(gson().serialize(component));
       } catch (final Throwable error) {
         throw new UnsupportedOperationException(error);