chore: add Icon.STREAM_CODEC, NetworkHelper.sendToAll

FTBTeam · May 18, 2024 · 00f3c16 · 00f3c16
1 parent 8179d9b
commit 00f3c16
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 1 deletion.
diff --git a/common/src/main/java/dev/ftb/mods/ftblibrary/icon/Icon.java b/common/src/main/java/dev/ftb/mods/ftblibrary/icon/Icon.java
@@ -2,9 +2,10 @@
 
 import com.google.gson.JsonElement;
 import com.google.gson.JsonPrimitive;
-import dev.ftb.mods.ftblibrary.FTBLibrary;
 import dev.ftb.mods.ftblibrary.config.ImageResourceConfig;
 import dev.ftb.mods.ftblibrary.math.PixelBuffer;
+import net.minecraft.network.FriendlyByteBuf;
+import net.minecraft.network.codec.StreamCodec;
 import net.minecraft.resources.ResourceLocation;
 import org.jetbrains.annotations.Nullable;
 
@@ -19,6 +20,18 @@ public static Color4I empty() {
 		return Color4I.EMPTY_ICON;
 	}
 
+	public static final StreamCodec<FriendlyByteBuf,Icon> STREAM_CODEC = new StreamCodec<>() {
+        @Override
+        public Icon decode(FriendlyByteBuf buf) {
+			return Icon.getIcon(buf.readUtf());
+        }
+
+        @Override
+        public void encode(FriendlyByteBuf buf, Icon icon) {
+			buf.writeUtf(icon.toString());
+        }
+    };
+
 	public static Icon getIcon(@Nullable JsonElement json) {
 		if (json == null || json.isJsonNull()) {
 			return empty();

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
@@ -7,6 +7,7 @@
 import net.minecraft.network.RegistryFriendlyByteBuf;
 import net.minecraft.network.codec.StreamCodec;
 import net.minecraft.network.protocol.common.custom.CustomPacketPayload;
+import net.minecraft.server.MinecraftServer;
 
 public class NetworkHelper {
     public static <T extends CustomPacketPayload> void registerC2S(CustomPacketPayload.Type<T> type, StreamCodec<? super RegistryFriendlyByteBuf, T> codec, NetworkManager.NetworkReceiver<T> handler) {
@@ -21,6 +22,10 @@ public static <T extends CustomPacketPayload> void registerS2C(CustomPacketPaylo
         }
     }
 
+    public static <T extends CustomPacketPayload> void sendToAll(MinecraftServer server, T packet) {
+        NetworkManager.sendToPlayers(server.getPlayerList().getPlayers(), packet);
+    }
+
     public static <B extends FriendlyByteBuf, V extends Enum<V>> StreamCodec<B, V> enumStreamCodec(Class<V> enumClass) {
         return new StreamCodec<>() {
             @Override