Flywheel/src/main/java/com/jozufozu/flywheel/config/BooleanConfigCommand.java

package com.jozufozu.flywheel.config;

import com.mojang.brigadier.Command;
import com.mojang.brigadier.builder.ArgumentBuilder;

import net.minecraft.command.CommandSource;
import net.minecraft.command.Commands;
import net.minecraft.entity.player.ServerPlayerEntity;
import net.minecraftforge.fml.network.PacketDistributor;

public class BooleanConfigCommand {

	private final String name;

	private final BooleanConfig value;

	public BooleanConfigCommand(String name, BooleanConfig value) {
		this.name = name;
		this.value = value;
	}

	public ArgumentBuilder<CommandSource, ?> register() {
		return Commands.literal(name)
				.executes(context -> {
					ServerPlayerEntity player = context.getSource().asPlayer();
					FlwPackets.channel.send(
							PacketDistributor.PLAYER.with(() -> player),
							new SConfigureBooleanPacket(value, BooleanDirective.DISPLAY)
					);
					return Command.SINGLE_SUCCESS;
				})
				.then(Commands.literal("on").executes(context -> {
					ServerPlayerEntity player = context.getSource().asPlayer();
					FlwPackets.channel.send(
							PacketDistributor.PLAYER.with(() -> player),
							new SConfigureBooleanPacket(value, BooleanDirective.TRUE)
					);
					return Command.SINGLE_SUCCESS;
				}))
				.then(Commands.literal("off").executes(context -> {
					ServerPlayerEntity player = context.getSource().asPlayer();
					FlwPackets.channel.send(
							PacketDistributor.PLAYER.with(() -> player),
							new SConfigureBooleanPacket(value, BooleanDirective.FALSE)
					);
					return Command.SINGLE_SUCCESS;
				}));
	}
}