package com.enhancedtnt.network; import com.enhancedtnt.EnhancedTNTSurvival; import com.enhancedtnt.screen.SuperTNTScreenHandler; import com.enhancedtnt.util.PlayerStatsManager; import net.fabricmc.api.EnvType; import net.fabricmc.api.Environment; import net.fabricmc.fabric.api.client.networking.v1.ClientPlayNetworking; import net.fabricmc.fabric.api.networking.v1.PayloadTypeRegistry; import net.fabricmc.fabric.api.networking.v1.ServerPlayNetworking; import net.minecraft.item.ItemStack; import net.minecraft.item.Items; import net.minecraft.network.RegistryByteBuf; import net.minecraft.network.codec.PacketCodec; import net.minecraft.network.codec.PacketCodecs; import net.minecraft.network.packet.CustomPayload; import net.minecraft.util.Identifier; public class ModNetworking { public static final CustomPayload.Id ADD_GUNPOWDER_ID = new CustomPayload.Id<>(Identifier.of(EnhancedTNTSurvival.MOD_ID, "add_gunpowder")); public static final CustomPayload.Id DETONATE_ID = new CustomPayload.Id<>(Identifier.of(EnhancedTNTSurvival.MOD_ID, "detonate")); public static final CustomPayload.Id SYNC_STATS_ID = new CustomPayload.Id<>(Identifier.of(EnhancedTNTSurvival.MOD_ID, "sync_stats")); public static final CustomPayload.Id UPDATE_GUNPOWDER_ID = new CustomPayload.Id<>(Identifier.of(EnhancedTNTSurvival.MOD_ID, "update_gunpowder")); public record AddGunpowderPayload() implements CustomPayload { public static final PacketCodec CODEC = PacketCodec.unit(new AddGunpowderPayload()); @Override public Id getId() { return ADD_GUNPOWDER_ID; } } public record DetonatePayload() implements CustomPayload { public static final PacketCodec CODEC = PacketCodec.unit(new DetonatePayload()); @Override public Id getId() { return DETONATE_ID; } } public record SyncStatsPayload(float energy, float metallic) implements CustomPayload { public static final PacketCodec CODEC = PacketCodec.tuple( PacketCodecs.FLOAT, SyncStatsPayload::energy, PacketCodecs.FLOAT, SyncStatsPayload::metallic, SyncStatsPayload::new ); @Override public Id getId() { return SYNC_STATS_ID; } } public record UpdateGunpowderPayload(int amount) implements CustomPayload { public static final PacketCodec CODEC = PacketCodec.tuple( PacketCodecs.INTEGER, UpdateGunpowderPayload::amount, UpdateGunpowderPayload::new ); @Override public Id getId() { return UPDATE_GUNPOWDER_ID; } } public static void registerServerReceivers() { PayloadTypeRegistry.playC2S().register(ADD_GUNPOWDER_ID, AddGunpowderPayload.CODEC); PayloadTypeRegistry.playC2S().register(DETONATE_ID, DetonatePayload.CODEC); PayloadTypeRegistry.playS2C().register(SYNC_STATS_ID, SyncStatsPayload.CODEC); PayloadTypeRegistry.playS2C().register(UPDATE_GUNPOWDER_ID, UpdateGunpowderPayload.CODEC); ServerPlayNetworking.registerGlobalReceiver(ADD_GUNPOWDER_ID, (payload, context) -> { context.server().execute(() -> { if (context.player().currentScreenHandler instanceof SuperTNTScreenHandler handler) { ItemStack stack = handler.getInventory().getStack(0); if (stack.isOf(Items.GUNPOWDER) && !stack.isEmpty()) { int toAdd = Math.min(stack.getCount(), 192 - handler.getGunpowderAmount()); if (toAdd > 0) { handler.addGunpowder(toAdd); stack.decrement(toAdd); ServerPlayNetworking.send(context.player(), new UpdateGunpowderPayload(handler.getGunpowderAmount())); } } } }); }); ServerPlayNetworking.registerGlobalReceiver(DETONATE_ID, (payload, context) -> { context.server().execute(() -> { if (context.player().currentScreenHandler instanceof SuperTNTScreenHandler handler) { handler.detonate(context.player()); } }); }); } @Environment(EnvType.CLIENT) public static void registerClientReceivers() { ClientPlayNetworking.registerGlobalReceiver(SYNC_STATS_ID, (payload, context) -> { context.client().execute(() -> { PlayerStatsManager.setClientEnergy(payload.energy()); PlayerStatsManager.setClientMetallic(payload.metallic()); }); }); ClientPlayNetworking.registerGlobalReceiver(UPDATE_GUNPOWDER_ID, (payload, context) -> { context.client().execute(() -> { if (context.player().currentScreenHandler instanceof SuperTNTScreenHandler handler) { handler.setGunpowderAmount(payload.amount()); } }); }); } }