efibs / mc-lasertag-mod

import com.mojang.brigadier.arguments.StringArgumentType;

import com.mojang.brigadier.builder.LiteralArgumentBuilder;

import com.mojang.brigadier.context.CommandContext;

import de.pewpewproject.lasertag.command.CommandFeedback;

import de.pewpewproject.lasertag.command.ServerFeedbackCommand;

import de.pewpewproject.lasertag.command.suggestions.LasertagGameModeSuggestionProvider;

import de.pewpewproject.lasertag.lasertaggame.gamemode.GameModes;

import net.minecraft.server.command.ServerCommandSource;

import net.minecraft.text.Text;

import net.minecraft.util.Formatting;

import java.util.Optional;

import static com.mojang.brigadier.arguments.StringArgumentType.word;

import static net.minecraft.server.command.CommandManager.argument;

import static net.minecraft.server.command.CommandManager.literal;

/**

 * The lasertag game mode command

 *

 * @author Étienne Muser

 */

public class LasertagGameModeCommand extends ServerFeedbackCommand {

    @Override

    protected Optional<CommandFeedback> execute(CommandContext<ServerCommandSource> context) {

        // Get the game managers

        var gameManager = context.getSource().getWorld().getServerLasertagManager();

        var gameModeManager = gameManager.getGameModeManager();

        // Get the game mode translatable name

        var gameModeTranslatableName = StringArgumentType.getString(context, "gamemode");

        // Get the new game mode

        var newGameMode = GameModes.GAME_MODES.get(gameModeTranslatableName);

        // Sanity check

        if (newGameMode == null) {

            return Optional.of(new CommandFeedback(Text.literal("Invalid game mode.").formatted(Formatting.RED), false, false));

        }

        // Set the game mode

        gameModeManager.setGameMode(newGameMode);

        // Translate the game mode name

        var translatedGameModeName = Text.translatable(gameModeTranslatableName).getString();

        return Optional.of(new CommandFeedback(Text.literal("Game mode changed to '" + translatedGameModeName + "'."), false, true));

    }

    static void register(LiteralArgumentBuilder<ServerCommandSource> lab) {

        lab.then(literal("gamemode")

                        .executes(LasertagGameModeCommand::executeWithoutArgs)

                        .then(argument("gamemode", word())

                                .requires(s -> s.hasPermissionLevel(1))

                                .suggests(LasertagGameModeSuggestionProvider.getInstance())

                                .executes(new LasertagGameModeCommand())));

    }

}