efibs / mc-lasertag-mod

public class LoadMapCommand extends ServerFeedbackCommand {

    @Override

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

        // Get the game managers

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

        var arenaManager = gameManager.getArenaManager();

        var mapTranslatableName = StringArgumentType.getString(context, "map");

        var arenaTypeOptional = Arrays.stream(ArenaType.values())

                .filter(m -> m.translatableName.equals(mapTranslatableName))

                .findFirst();

        var proceduralTypeOptional = Arrays.stream(ProceduralArenaType.values())

                .filter(m -> m.translatableName.equals(mapTranslatableName))

                .findFirst();

        // If is procedural map

        if (proceduralTypeOptional.isPresent()) {

            arenaTypeOptional = Optional.of(ArenaType.PROCEDURAL);

        }

        if (arenaTypeOptional.isEmpty()) {

            return Optional.of(new CommandFeedback(Text.literal("Could not find map.").formatted(Formatting.RED), false, false));

        }

        try {

            arenaManager.loadArena(arenaTypeOptional.get(), proceduralTypeOptional.orElse(ProceduralArenaType.SMALL_2V2));

        } catch (Exception e) {

            return Optional.of(new CommandFeedback(Text.literal("Unexpected error while loading map: " + e.getMessage()).formatted(Formatting.RED), false, false));

        }

        return Optional.of(new CommandFeedback(Text.literal("Map loaded."), false, false));

    }

    public static void register(LiteralArgumentBuilder<ServerCommandSource> lab) {

        lab.then(literal("loadMap")

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

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

                        .suggests(MapSuggestionProvider.getInstance())

                        .executes(new LoadMapCommand())));