ClientboundDebugRenderAddPacket.java

package net.hollowcube.compat.moulberrytweaks.packets;

import net.hollowcube.compat.api.packet.ClientboundModPacket;
import net.hollowcube.compat.moulberrytweaks.MoulberryTweaksAPI;
import net.hollowcube.compat.moulberrytweaks.debugrender.DebugShape;
import net.kyori.adventure.key.Key;
import net.minestom.server.network.NetworkBuffer;
import net.minestom.server.network.NetworkBufferTemplate;
import org.intellij.lang.annotations.MagicConstant;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;

public record ClientboundDebugRenderAddPacket(
        @Nullable Key key,
        @NotNull DebugShape shape,
        @MagicConstant(flagsFromClass = DebugShape.class) int flags,
        int lifetimeTicks
) implements ClientboundModPacket<ClientboundDebugRenderAddPacket> {
    public static final Type<ClientboundDebugRenderAddPacket> TYPE = Type.of(
            MoulberryTweaksAPI.DEBUG_RENDER_CHANNEL, "add",
            NetworkBufferTemplate.template(
                    NetworkBuffer.KEY.optional(), ClientboundDebugRenderAddPacket::key,
                    DebugShape.NETWORK_TYPE, ClientboundDebugRenderAddPacket::shape,
                    NetworkBuffer.VAR_INT, ClientboundDebugRenderAddPacket::flags,
                    NetworkBuffer.VAR_INT, ClientboundDebugRenderAddPacket::lifetimeTicks,
                    ClientboundDebugRenderAddPacket::new)
    );

    @Override
    public Type<ClientboundDebugRenderAddPacket> getType() {
        return TYPE;
    }
}

ClientboundDebugRenderClearNamespacePacket.java

package net.hollowcube.compat.moulberrytweaks.packets;

import net.hollowcube.compat.api.packet.ClientboundModPacket;
import net.hollowcube.compat.moulberrytweaks.MoulberryTweaksAPI;
import net.minestom.server.network.NetworkBuffer;
import net.minestom.server.network.NetworkBufferTemplate;
import org.jetbrains.annotations.NotNull;

public record ClientboundDebugRenderClearNamespacePacket(
        @NotNull String namespace
) implements ClientboundModPacket<ClientboundDebugRenderClearNamespacePacket> {
    public static final Type<ClientboundDebugRenderClearNamespacePacket> TYPE = Type.of(
            MoulberryTweaksAPI.DEBUG_RENDER_CHANNEL, "clear_namespace",
            NetworkBufferTemplate.template(
                    NetworkBuffer.STRING, ClientboundDebugRenderClearNamespacePacket::namespace,
                    ClientboundDebugRenderClearNamespacePacket::new)
    );

    @Override
    public Type<ClientboundDebugRenderClearNamespacePacket> getType() {
        return TYPE;
    }
}

ClientboundDebugRenderClearPacket.java

package net.hollowcube.compat.moulberrytweaks.packets;

import net.hollowcube.compat.api.packet.ClientboundModPacket;
import net.hollowcube.compat.moulberrytweaks.MoulberryTweaksAPI;
import net.minestom.server.network.NetworkBufferTemplate;

public record ClientboundDebugRenderClearPacket() implements ClientboundModPacket<ClientboundDebugRenderClearPacket> {
    public static final Type<ClientboundDebugRenderClearPacket> TYPE = Type.of(
            MoulberryTweaksAPI.DEBUG_RENDER_CHANNEL, "clear",
            NetworkBufferTemplate.template(ClientboundDebugRenderClearPacket::new)
    );

    @Override
    public Type<ClientboundDebugRenderClearPacket> getType() {
        return TYPE;
    }
}

ClientboundDebugRenderRemovePacket.java

package net.hollowcube.compat.moulberrytweaks.packets;

import net.hollowcube.compat.api.packet.ClientboundModPacket;
import net.hollowcube.compat.moulberrytweaks.MoulberryTweaksAPI;
import net.kyori.adventure.key.Key;
import net.minestom.server.network.NetworkBuffer;
import net.minestom.server.network.NetworkBufferTemplate;
import org.jetbrains.annotations.NotNull;

public record ClientboundDebugRenderRemovePacket(
        @NotNull Key key
) implements ClientboundModPacket<ClientboundDebugRenderRemovePacket> {
    public static final Type<ClientboundDebugRenderRemovePacket> TYPE = Type.of(
            MoulberryTweaksAPI.DEBUG_RENDER_CHANNEL, "remove",
            NetworkBufferTemplate.template(
                    NetworkBuffer.KEY, ClientboundDebugRenderRemovePacket::key,
                    ClientboundDebugRenderRemovePacket::new)
    );

    @Override
    public Type<ClientboundDebugRenderRemovePacket> getType() {
        return TYPE;
    }
}

DebugShape.java

package net.hollowcube.compat.moulberrytweaks.debugrender;

import net.kyori.adventure.text.Component;
import net.minestom.server.coordinate.Point;
import net.minestom.server.network.NetworkBuffer;
import net.minestom.server.network.NetworkBufferTemplate;
import org.jetbrains.annotations.NotNull;

import java.util.List;

public sealed interface DebugShape {
    @NotNull NetworkBuffer.Type<DebugShape> NETWORK_TYPE = NetworkBuffer.STRING.unionType(DebugShape::typeFromName, DebugShape::type);

    int FLAG_SHOW_THROUGH_WALLS = 1;
    int FLAG_FULL_OPACITY_BEHIND_WALLS = 2;
    int FLAG_WIREFRAME = 4;
    int FLAG_NO_SHADE = 8;
    int FLAG_SHOW_AXIS = 16;

    record Box(
            @NotNull Point center, @NotNull Point size, float[] rotation,
            int faceArgb, int lineArgb, float lineThickness
    ) implements DebugShape {
        public static final String TYPE = "box";
        public static final NetworkBuffer.Type<Box> NETWORK_TYPE = NetworkBufferTemplate.template(
                NetworkBuffer.VECTOR3D, Box::center,
                NetworkBuffer.VECTOR3D, Box::size,
                NetworkBuffer.QUATERNION, Box::rotation,
                NetworkBuffer.INT, Box::faceArgb,
                NetworkBuffer.INT, Box::lineArgb,
                NetworkBuffer.FLOAT, Box::lineThickness,
                Box::new);

        @Override
        public @NotNull String type() {
            return TYPE;
        }
    }

    record Ellipsoid(
            @NotNull Point center, @NotNull Point size, float[] rotation,
            int argb, int detail
    ) implements DebugShape {
        public static final String TYPE = "ellipsoid";
        public static final NetworkBuffer.Type<Ellipsoid> NETWORK_TYPE = NetworkBufferTemplate.template(
                NetworkBuffer.VECTOR3D, Ellipsoid::center,
                NetworkBuffer.VECTOR3D, Ellipsoid::size,
                NetworkBuffer.QUATERNION, Ellipsoid::rotation,
                NetworkBuffer.INT, Ellipsoid::argb,
                NetworkBuffer.VAR_INT, Ellipsoid::detail,
                Ellipsoid::new);

        @Override
        public @NotNull String type() {
            return TYPE;
        }
    }

    record LineStrip(@NotNull List<Point> points, int argb, float lineThickness) implements DebugShape {
        public static final String TYPE = "line_strip";
        public static final NetworkBuffer.Type<LineStrip> NETWORK_TYPE = NetworkBufferTemplate.template(
                NetworkBuffer.VECTOR3D.list(), LineStrip::points,
                NetworkBuffer.INT, LineStrip::argb,
                NetworkBuffer.FLOAT, LineStrip::lineThickness,
                LineStrip::new);

        @Override
        public @NotNull String type() {
            return TYPE;
        }
    }

    record Quad(
            @NotNull Point one, @NotNull Point two,
            @NotNull Point three, @NotNull Point four,
            int argb
    ) implements DebugShape {
        public static final String TYPE = "quad";
        public static final NetworkBuffer.Type<Quad> NETWORK_TYPE = NetworkBufferTemplate.template(
                NetworkBuffer.VECTOR3D, Quad::one,
                NetworkBuffer.VECTOR3D, Quad::two,
                NetworkBuffer.VECTOR3D, Quad::three,
                NetworkBuffer.VECTOR3D, Quad::four,
                NetworkBuffer.INT, Quad::argb,
                Quad::new);

        @Override
        public @NotNull String type() {
            return TYPE;
        }
    }

    record Text(
            @NotNull Point position, @NotNull Component component,
            boolean shadow, int backgroundColor
    ) implements DebugShape {
        public static final String TYPE = "text";
        public static final NetworkBuffer.Type<Text> NETWORK_TYPE = NetworkBufferTemplate.template(
                NetworkBuffer.VECTOR3D, Text::position,
                NetworkBuffer.COMPONENT, Text::component,
                NetworkBuffer.BOOLEAN, Text::shadow,
                NetworkBuffer.INT, Text::backgroundColor,
                Text::new);

        @Override
        public @NotNull String type() {
            return TYPE;
        }
    }

    @NotNull String type();

    @SuppressWarnings({"unchecked", "RedundantCast"})
    private static @NotNull NetworkBuffer.Type<DebugShape> typeFromName(@NotNull String type) {
        // Gross cast will be fixed once minestom correctly takes ? extends DebugShape
        return (NetworkBuffer.Type<DebugShape>) (Object) switch (type) {
            case Box.TYPE -> Box.NETWORK_TYPE;
            case Ellipsoid.TYPE -> Ellipsoid.NETWORK_TYPE;
            case LineStrip.TYPE -> LineStrip.NETWORK_TYPE;
            case Quad.TYPE -> Quad.NETWORK_TYPE;
            case Text.TYPE -> Text.NETWORK_TYPE;
            default -> throw new IllegalArgumentException("Unknown debug shape type: " + type);
        };
    }
}