92 lines
3.5 KiB
Java
92 lines
3.5 KiB
Java
package net.minecraftforge.fml.network;
|
|
|
|
import io.netty.buffer.Unpooled;
|
|
import it.unimi.dsi.fastutil.objects.Reference2ReferenceArrayMap;
|
|
import net.minecraft.network.Packet;
|
|
import net.minecraft.network.PacketBuffer;
|
|
import net.minecraft.network.login.client.CPacketCustomPayloadLogin;
|
|
import net.minecraft.network.login.server.SPacketCustomPayloadLogin;
|
|
import net.minecraft.network.play.client.CPacketCustomPayload;
|
|
import net.minecraft.network.play.server.SPacketCustomPayload;
|
|
import net.minecraft.util.ResourceLocation;
|
|
import net.minecraftforge.fml.UnsafeHacks;
|
|
|
|
import java.lang.reflect.Field;
|
|
import java.lang.reflect.Modifier;
|
|
import java.util.Arrays;
|
|
import java.util.Optional;
|
|
import java.util.function.Function;
|
|
import java.util.function.Supplier;
|
|
import java.util.stream.Collectors;
|
|
import java.util.stream.Stream;
|
|
|
|
public interface ICustomPacket<T extends Packet<?>> {
|
|
@SuppressWarnings("OptionalUsedAsFieldOrParameterType")
|
|
enum Fields {
|
|
CPACKETCUSTOMPAYLOAD(CPacketCustomPayload.class),
|
|
SPACKETCUSTOMPAYLOAD(SPacketCustomPayload.class),
|
|
CPACKETCUSTOMLOGIN(CPacketCustomPayloadLogin.class),
|
|
SPACKETCUSTOMLOGIN(SPacketCustomPayloadLogin.class),
|
|
;
|
|
|
|
static final Reference2ReferenceArrayMap<Class<?>, Fields> lookup;
|
|
static {
|
|
lookup = Stream.of(values()).
|
|
collect(Collectors.toMap(Fields::getClazz, Function.identity(), (m1, m2)->m1, Reference2ReferenceArrayMap::new));
|
|
}
|
|
|
|
private final Class<?> clazz;
|
|
|
|
final Optional<Field> data;
|
|
final Optional<Field> channel;
|
|
final Optional<Field> index;
|
|
|
|
Fields(Class<?> customPacketClass)
|
|
{
|
|
this.clazz = customPacketClass;
|
|
Field[] fields = customPacketClass.getDeclaredFields();
|
|
data = Arrays.stream(fields).filter(f-> !Modifier.isStatic(f.getModifiers()) && f.getType() == PacketBuffer.class).findFirst();
|
|
channel = Arrays.stream(fields).filter(f->!Modifier.isStatic(f.getModifiers()) && f.getType() == ResourceLocation.class).findFirst();
|
|
index = Arrays.stream(fields).filter(f->!Modifier.isStatic(f.getModifiers()) && f.getType() == int.class).findFirst();
|
|
}
|
|
|
|
private Class<?> getClazz()
|
|
{
|
|
return clazz;
|
|
}
|
|
}
|
|
|
|
default PacketBuffer getData() {
|
|
return Fields.lookup.get(this.getClass()).data.map(f->UnsafeHacks.<PacketBuffer>getField(f, this)).orElse(null);
|
|
}
|
|
|
|
default ResourceLocation getName() {
|
|
return Fields.lookup.get(this.getClass()).channel.map(f->UnsafeHacks.<ResourceLocation>getField(f, this)).orElse(FMLLoginWrapper.WRAPPER);
|
|
}
|
|
|
|
default int getIndex() {
|
|
return Fields.lookup.get(this.getClass()).index.map(f->UnsafeHacks.getIntField(f, this)).orElse(Integer.MIN_VALUE);
|
|
}
|
|
|
|
default void setData(PacketBuffer buffer) {
|
|
Fields.lookup.get(this.getClass()).data.ifPresent(f->UnsafeHacks.setField(f, this, buffer));
|
|
}
|
|
|
|
default void setName(ResourceLocation channelName) {
|
|
Fields.lookup.get(this.getClass()).channel.ifPresent(f->UnsafeHacks.setField(f, this, channelName));
|
|
}
|
|
|
|
default void setIndex(int index) {
|
|
Fields.lookup.get(this.getClass()).index.ifPresent(f->UnsafeHacks.setIntField(f, this, index));
|
|
}
|
|
|
|
default NetworkDirection getDirection() {
|
|
return NetworkDirection.directionFor(this.getClass());
|
|
}
|
|
|
|
@SuppressWarnings("unchecked")
|
|
default T getThis() {
|
|
return (T)this;
|
|
}
|
|
}
|