61 lines
2.9 KiB
Java
61 lines
2.9 KiB
Java
package net.minecraft.network.protocol.common.custom;
|
|
|
|
import io.netty.buffer.ByteBuf;
|
|
import java.util.List;
|
|
import java.util.Map;
|
|
import java.util.stream.Collectors;
|
|
import net.minecraft.network.FriendlyByteBuf;
|
|
import net.minecraft.network.codec.StreamCodec;
|
|
import net.minecraft.network.codec.StreamDecoder;
|
|
import net.minecraft.network.codec.StreamMemberEncoder;
|
|
import net.minecraft.resources.ResourceLocation;
|
|
|
|
public interface CustomPacketPayload {
|
|
CustomPacketPayload.Type<? extends CustomPacketPayload> type();
|
|
|
|
static <B extends ByteBuf, T extends CustomPacketPayload> StreamCodec<B, T> codec(StreamMemberEncoder<B, T> encoder, StreamDecoder<B, T> decoder) {
|
|
return StreamCodec.ofMember(encoder, decoder);
|
|
}
|
|
|
|
static <T extends CustomPacketPayload> CustomPacketPayload.Type<T> createType(String id) {
|
|
return new CustomPacketPayload.Type<>(ResourceLocation.withDefaultNamespace(id));
|
|
}
|
|
|
|
static <B extends FriendlyByteBuf> StreamCodec<B, CustomPacketPayload> codec(
|
|
CustomPacketPayload.FallbackProvider<B> fallbackProvider, List<CustomPacketPayload.TypeAndCodec<? super B, ?>> typeAndCodecs
|
|
) {
|
|
final Map<ResourceLocation, StreamCodec<? super B, ? extends CustomPacketPayload>> map = (Map<ResourceLocation, StreamCodec<? super B, ? extends CustomPacketPayload>>)typeAndCodecs.stream()
|
|
.collect(Collectors.toUnmodifiableMap(typeAndCodec -> typeAndCodec.type().id(), CustomPacketPayload.TypeAndCodec::codec));
|
|
return new StreamCodec<B, CustomPacketPayload>() {
|
|
private StreamCodec<? super B, ? extends CustomPacketPayload> findCodec(ResourceLocation resourceLocation) {
|
|
StreamCodec<? super B, ? extends CustomPacketPayload> streamCodec = (StreamCodec<? super B, ? extends CustomPacketPayload>)map.get(resourceLocation);
|
|
return streamCodec != null ? streamCodec : fallbackProvider.create(resourceLocation);
|
|
}
|
|
|
|
private <T extends CustomPacketPayload> void writeCap(B buffer, CustomPacketPayload.Type<T> type, CustomPacketPayload payload) {
|
|
buffer.writeResourceLocation(type.id());
|
|
StreamCodec<B, T> streamCodec = this.findCodec(type.id);
|
|
streamCodec.encode(buffer, (T)payload);
|
|
}
|
|
|
|
public void encode(B friendlyByteBuf, CustomPacketPayload customPacketPayload) {
|
|
this.writeCap(friendlyByteBuf, customPacketPayload.type(), customPacketPayload);
|
|
}
|
|
|
|
public CustomPacketPayload decode(B friendlyByteBuf) {
|
|
ResourceLocation resourceLocation = friendlyByteBuf.readResourceLocation();
|
|
return (CustomPacketPayload)this.findCodec(resourceLocation).decode(friendlyByteBuf);
|
|
}
|
|
};
|
|
}
|
|
|
|
public interface FallbackProvider<B extends FriendlyByteBuf> {
|
|
StreamCodec<B, ? extends CustomPacketPayload> create(ResourceLocation resourceLocation);
|
|
}
|
|
|
|
public record Type<T extends CustomPacketPayload>(ResourceLocation id) {
|
|
}
|
|
|
|
public record TypeAndCodec<B extends FriendlyByteBuf, T extends CustomPacketPayload>(CustomPacketPayload.Type<T> type, StreamCodec<B, T> codec) {
|
|
}
|
|
}
|