Essentials/providers/NMSReflectionProvider/src/main/java/net/ess3/nms/refl/ReflUtil.java

336 lines
12 KiB
Java

package net.ess3.nms.refl;
import com.google.common.base.Objects;
import com.google.common.base.Preconditions;
import com.google.common.collect.HashBasedTable;
import com.google.common.collect.Table;
import org.bukkit.Bukkit;
import java.lang.reflect.Constructor;
import java.lang.reflect.Field;
import java.lang.reflect.Method;
import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
public final class ReflUtil {
public static final NMSVersion V1_12_R1 = NMSVersion.fromString("v1_12_R1");
public static final NMSVersion V1_11_R1 = NMSVersion.fromString("v1_11_R1");
public static final NMSVersion V1_17_R1 = NMSVersion.fromString("v1_17_R1");
public static final NMSVersion V1_18_R1 = NMSVersion.fromString("v1_18_R1");
public static final NMSVersion V1_19_R1 = NMSVersion.fromString("v1_19_R1");
public static final NMSVersion V1_19_R2 = NMSVersion.fromString("v1_19_R2");
private static final Map<String, Class<?>> classCache = new HashMap<>();
private static final Table<Class<?>, String, Method> methodCache = HashBasedTable.create();
private static final Table<Class<?>, MethodParams, Method> methodParamCache = HashBasedTable.create();
private static final Table<Class<?>, String, Field> fieldCache = HashBasedTable.create();
private static final Map<Class<?>, Constructor<?>> constructorCache = new HashMap<>();
private static final Table<Class<?>, ConstructorParams, Constructor<?>> constructorParamCache = HashBasedTable.create();
private static NMSVersion nmsVersionObject;
private static String nmsVersion;
private ReflUtil() {
}
public static String getNMSVersion() {
if (nmsVersion == null) {
final String name = Bukkit.getServer().getClass().getName();
final String[] parts = name.split("\\.");
if (parts.length > 3) {
return nmsVersion = parts[3];
}
// We're not on craftbukkit, return an empty string so we can silently fail
return nmsVersion = "";
}
return nmsVersion;
}
public static NMSVersion getNmsVersionObject() {
if (nmsVersionObject == null) {
try {
nmsVersionObject = NMSVersion.fromString(getNMSVersion());
} catch (final IllegalArgumentException e) {
try {
Class.forName("org.bukkit.craftbukkit.CraftServer");
nmsVersionObject = new NMSVersion(99, 99, 99); // Mojang Dev Mappings
} catch (final ClassNotFoundException ignored) {
throw e;
}
}
}
return nmsVersionObject;
}
public static boolean isMojMap() {
return getNmsVersionObject().getMajor() == 99;
}
public static Class<?> getNMSClass(final String className) {
return getClassCached("net.minecraft.server" + (ReflUtil.getNmsVersionObject().isLowerThan(ReflUtil.V1_17_R1) ? "." + getNMSVersion() : "") + "." + className);
}
public static Class<?> getOBCClass(final String className) {
return getClassCached("org.bukkit.craftbukkit" + (getNmsVersionObject().getMajor() == 99 ? "" : ("." + getNMSVersion())) + "." + className);
}
public static Class<?> getClassCached(final String className) {
if (classCache.containsKey(className)) {
return classCache.get(className);
}
try {
final Class<?> classForName = Class.forName(className);
classCache.put(className, classForName);
return classForName;
} catch (final ClassNotFoundException e) {
return null;
}
}
public static Method getMethodCached(final Class<?> clazz, final String methodName) {
if (methodCache.contains(clazz, methodName)) {
return methodCache.get(clazz, methodName);
}
try {
final Method method = clazz.getDeclaredMethod(methodName);
method.setAccessible(true);
methodCache.put(clazz, methodName, method);
return method;
} catch (final NoSuchMethodException e) {
return null;
}
}
public static Method getMethodCached(final Class<?> clazz, final String methodName, final Class<?>... params) {
final MethodParams methodParams = new MethodParams(methodName, params);
if (methodParamCache.contains(clazz, methodParams)) {
return methodParamCache.get(clazz, methodParams);
}
try {
final Method method = clazz.getDeclaredMethod(methodName, params);
method.setAccessible(true);
methodParamCache.put(clazz, methodParams, method);
return method;
} catch (final NoSuchMethodException e) {
return null;
}
}
public static Field getFieldCached(final Class<?> clazz, final String fieldName) {
if (fieldCache.contains(clazz, fieldName)) {
return fieldCache.get(clazz, fieldName);
}
try {
final Field field = clazz.getDeclaredField(fieldName);
field.setAccessible(true);
fieldCache.put(clazz, fieldName, field);
return field;
} catch (final NoSuchFieldException e) {
return null;
}
}
public static Constructor<?> getConstructorCached(final Class<?> clazz) {
if (constructorCache.containsKey(clazz)) {
return constructorCache.get(clazz);
}
try {
final Constructor<?> constructor = clazz.getDeclaredConstructor();
constructor.setAccessible(true);
constructorCache.put(clazz, constructor);
return constructor;
} catch (final NoSuchMethodException e) {
return null;
}
}
public static Constructor<?> getConstructorCached(final Class<?> clazz, final Class<?>... params) {
final ConstructorParams constructorParams = new ConstructorParams(params);
if (constructorParamCache.contains(clazz, constructorParams)) {
return constructorParamCache.get(clazz, constructorParams);
}
try {
final Constructor<?> constructor = clazz.getDeclaredConstructor(params);
constructor.setAccessible(true);
constructorParamCache.put(clazz, constructorParams, constructor);
return constructor;
} catch (final NoSuchMethodException e) {
return null;
}
}
// Adapted from @minecrafter
private static class MethodParams {
private final String name;
private final Class<?>[] params;
MethodParams(final String name, final Class<?>[] params) {
this.name = name;
this.params = params;
}
// Ugly autogenned Lombok code
@Override
public boolean equals(final Object o) {
if (o == this) {
return true;
}
if (!(o instanceof MethodParams)) {
return false;
}
final MethodParams that = (MethodParams) o;
if (!that.canEqual(this)) {
return false;
}
final Object thisName = this.name;
final Object thatName = that.name;
if (thisName == null) {
if (thatName == null) {
return Arrays.deepEquals(this.params, that.params);
}
} else if (thisName.equals(thatName)) {
return Arrays.deepEquals(this.params, that.params);
}
return false;
}
boolean canEqual(final Object that) {
return that instanceof MethodParams;
}
@Override
public int hashCode() {
int result = 1;
final Object thisName = this.name;
result = result * 31 + ((thisName == null) ? 0 : thisName.hashCode());
result = result * 31 + Arrays.deepHashCode(this.params);
return result;
}
}
// Necessary for deepequals
private static class ConstructorParams {
private final Class<?>[] params;
ConstructorParams(final Class<?>[] params) {
this.params = params;
}
@Override
public boolean equals(final Object o) {
if (this == o) return true;
if (o == null || getClass() != o.getClass()) return false;
final ConstructorParams that = (ConstructorParams) o;
return Arrays.deepEquals(params, that.params);
}
@Override
public int hashCode() {
return Arrays.deepHashCode(params);
}
}
/**
* https://gist.github.com/SupaHam/dad1db6406596c5f8e4b221ff473831c
*
* @author SupaHam (<a href="https://github.com/SupaHam">https://github.com/SupaHam</a>)
*/
public static final class NMSVersion implements Comparable<NMSVersion> {
private static final Pattern VERSION_PATTERN = Pattern.compile("^v(\\d+)_(\\d+)_R(\\d+)");
private final int major;
private final int minor;
private final int release;
private NMSVersion(final int major, final int minor, final int release) {
this.major = major;
this.minor = minor;
this.release = release;
}
public static NMSVersion fromString(final String string) {
Preconditions.checkNotNull(string, "string cannot be null.");
Matcher matcher = VERSION_PATTERN.matcher(string);
if (!matcher.matches()) {
if (!Bukkit.getName().equals("Essentials Fake Server")) {
throw new IllegalArgumentException(string + " is not in valid version format. e.g. v1_10_R1");
}
matcher = VERSION_PATTERN.matcher(V1_12_R1.toString());
Preconditions.checkArgument(matcher.matches(), string + " is not in valid version format. e.g. v1_10_R1");
}
return new NMSVersion(Integer.parseInt(matcher.group(1)), Integer.parseInt(matcher.group(2)), Integer.parseInt(matcher.group(3)));
}
public boolean isHigherThan(final NMSVersion o) {
return compareTo(o) > 0;
}
public boolean isHigherThanOrEqualTo(final NMSVersion o) {
return compareTo(o) >= 0;
}
public boolean isLowerThan(final NMSVersion o) {
return compareTo(o) < 0;
}
public boolean isLowerThanOrEqualTo(final NMSVersion o) {
return compareTo(o) <= 0;
}
public int getMajor() {
return major;
}
public int getMinor() {
return minor;
}
public int getRelease() {
return release;
}
@Override
public boolean equals(final Object o) {
if (this == o) {
return true;
}
if (o == null || getClass() != o.getClass()) {
return false;
}
final NMSVersion that = (NMSVersion) o;
return major == that.major &&
minor == that.minor &&
release == that.release;
}
@Override
public int hashCode() {
return Objects.hashCode(major, minor, release);
}
@Override
public String toString() {
return "v" + major + "_" + minor + "_R" + release;
}
@Override
public int compareTo(final NMSVersion o) {
if (major < o.major) {
return -1;
} else if (major > o.major) {
return 1;
} else { // equal major
if (minor < o.minor) {
return -1;
} else if (minor > o.minor) {
return 1;
} else {
return Integer.compare(release, o.release);
}
}
}
}
}