minecraft-kajetans-mod/src/main/java/me/km/networking/ModPacketHandler.java

package me.km.networking;

import me.km.KajetansMod;
import me.km.inventory.ModInventory;
import net.minecraft.entity.player.ServerPlayerEntity;
import net.minecraft.util.ResourceLocation;
import net.minecraftforge.fml.network.NetworkRegistry;
import net.minecraftforge.fml.network.PacketDistributor;
import net.minecraftforge.fml.network.simple.SimpleChannel;

public class ModPacketHandler {
    private static final String PROTOCOL_VERSION = "1";

    private static final SimpleChannel INSTANCE = NetworkRegistry.ChannelBuilder
            .named(new ResourceLocation(KajetansMod.MODID, "main"))
            .clientAcceptedVersions(PROTOCOL_VERSION::equals)
            .serverAcceptedVersions(PROTOCOL_VERSION::equals)
            .networkProtocolVersion(() -> PROTOCOL_VERSION)
            .simpleChannel();

    public static void init() {
        int id = 0;
        INSTANCE.registerMessage(id++, PlayerDisplay.class, PlayerDisplay::writeBytes, PlayerDisplay::fromBytes, PlayerDisplay::handle);
        INSTANCE.registerMessage(id++, FunctionKey.class, FunctionKey::writeBytes, FunctionKey::fromBytes, FunctionKey::handle);
        INSTANCE.registerMessage(id++, CustomInventory.class, CustomInventory::writeBytes, CustomInventory::fromBytes, CustomInventory::handle);
        INSTANCE.registerMessage(id++, StatusDisplay.class, StatusDisplay::writeBytes, StatusDisplay::fromBytes, StatusDisplay::handle);
        INSTANCE.registerMessage(id++, PlayerHead.class, PlayerHead::writeBytes, PlayerHead::fromBytes, PlayerHead::handle);
        INSTANCE.registerMessage(id++, ItemStackDisplay.class, ItemStackDisplay::writeBytes, ItemStackDisplay::fromBytes, ItemStackDisplay::handle);
        INSTANCE.registerMessage(id++, PlayerLeftClickEmpty.class, PlayerLeftClickEmpty::writeBytes, PlayerLeftClickEmpty::fromBytes, PlayerLeftClickEmpty::handle);
    }

    private static void sendToPlayer(ServerPlayerEntity p, Object msg) {
        INSTANCE.send(PacketDistributor.PLAYER.with(() -> p), msg);
    }

    private static void sendToServer(Object msg) {
        INSTANCE.sendToServer(msg);
    }

    public static void sendToDisplay(ServerPlayerEntity p, byte action, byte index, String text) {
        sendToPlayer(p, new PlayerDisplay(action, index, text));
    }

    public static void sendFunctionKey(int key) {
        sendToServer(new FunctionKey(key));
    }

    public static void sendCustomInventory(ServerPlayerEntity p, int id, String title, ModInventory inv) {
        sendToPlayer(p, new CustomInventory(id, title, inv));
    }

    public static void addStatus(ServerPlayerEntity p, byte index, String text) {
        sendToPlayer(p, new StatusDisplay((byte) 1, index, text));
    }

    public static void addTimedStatus(ServerPlayerEntity p, byte index, String text, int time) {
        sendToPlayer(p, new StatusDisplay((byte) 1, index, time, text));
    }

    public static void removeStatus(ServerPlayerEntity p, byte index) {
        sendToPlayer(p, new StatusDisplay((byte) 2, index, ""));
    }

    public static void clearStatus(ServerPlayerEntity p) {
        sendToPlayer(p, new StatusDisplay((byte) 3, (byte) 0, ""));
    }

    public static void sendToHead(ServerPlayerEntity p, byte action, byte index, String name, int x, int y, byte scale) {
        sendToPlayer(p, new PlayerHead(action, index, name, x, y, scale));
    }

    public static void sendLeftClickEmpty() {
        sendToServer(new PlayerLeftClickEmpty());
    }

    public static void setItemStackIndex(ServerPlayerEntity p, byte index, int iconIndex, int count) {
        sendToPlayer(p, new ItemStackDisplay(index, iconIndex, count));
    }

    public static void setItemStackActive(ServerPlayerEntity p, boolean active) {
        sendToPlayer(p, new ItemStackDisplay(active ? (byte) -1 : (byte) -2, -1, -1));
    }

    public static void clearItemStacks(ServerPlayerEntity p) {
        sendToPlayer(p, new ItemStackDisplay((byte) -3, -1, -1));
    }
}