chore: add NetworkHelper#enumStreamCodec method

Like NeoForgeExtraCodecs method, but we need it cross-platform...
FTBTeam · May 10, 2024 · cb74b03 · cb74b03
1 parent e2af3ee
commit cb74b03
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/common/src/main/java/dev/ftb/mods/ftblibrary/util/NetworkHelper.java b/common/src/main/java/dev/ftb/mods/ftblibrary/util/NetworkHelper.java
@@ -3,6 +3,7 @@
 import dev.architectury.networking.NetworkManager;
 import dev.architectury.platform.Platform;
 import dev.architectury.utils.Env;
+import net.minecraft.network.FriendlyByteBuf;
 import net.minecraft.network.RegistryFriendlyByteBuf;
 import net.minecraft.network.codec.StreamCodec;
 import net.minecraft.network.protocol.common.custom.CustomPacketPayload;
@@ -19,4 +20,18 @@ public static <T extends CustomPacketPayload> void registerS2C(CustomPacketPaylo
             NetworkManager.registerS2CPayloadType(type, codec);
         }
     }
+
+    public static <B extends FriendlyByteBuf, V extends Enum<V>> StreamCodec<B, V> enumStreamCodec(Class<V> enumClass) {
+        return new StreamCodec<>() {
+            @Override
+            public V decode(B buf) {
+                return buf.readEnum(enumClass);
+            }
+
+            @Override
+            public void encode(B buf, V value) {
+                buf.writeEnum(value);
+            }
+        };
+    }
 }