Add getFirst<Getter|Setter> API

CitizensDev · Jun 13, 2021 · b199013 · b199013
1 parent bd19c9e
commit b199013
Showing 1 changed file with 38 additions and 0 deletions.
diff --git a/main/src/main/java/net/citizensnpcs/util/NMS.java b/main/src/main/java/net/citizensnpcs/util/NMS.java
@@ -164,6 +164,32 @@ public static MethodHandle getFinalSetter(Class<?> clazz, String field, boolean
         return null;
     }
 
+    private static Field getFirstFieldMatchingType(Class<?> clazz, Class<?> type) {
+        Field found = null;
+        for (Field field : clazz.getDeclaredFields()) {
+            if (field.getType() == type) {
+                found = field;
+                break;
+            }
+        }
+        if (found != null) {
+            found.setAccessible(true);
+        }
+        return found;
+    }
+
+    public static MethodHandle getFirstGetter(Class<?> clazz, Class<?> type) {
+        try {
+            Field found = getFirstFieldMatchingType(clazz, type);
+            if (found == null)
+                return null;
+            return LOOKUP.unreflectGetter(found);
+        } catch (Exception e) {
+            Messaging.logTr(Messages.ERROR_GETTING_FIELD, type, e.getLocalizedMessage());
+        }
+        return null;
+    }
+
     public static MethodHandle getFirstMethodHandle(Class<?> clazz, boolean log, Class<?>... params) {
         if (clazz == null)
             return null;
@@ -195,6 +221,18 @@ public static MethodHandle getFirstMethodHandle(Class<?> clazz, boolean log, Cla
         return null;
     }
 
+    public static MethodHandle getFirstSetter(Class<?> clazz, Class<?> type) {
+        try {
+            Field found = getFirstFieldMatchingType(clazz, type);
+            if (found == null)
+                return null;
+            return LOOKUP.unreflectSetter(found);
+        } catch (Exception e) {
+            Messaging.logTr(Messages.ERROR_GETTING_FIELD, type, e.getLocalizedMessage());
+        }
+        return null;
+    }
+
     public static GameProfileRepository getGameProfileRepository() {
         return BRIDGE.getGameProfileRepository();
     }