package net.minecraft.server.commands;

import com.mojang.authlib.GameProfile;
import com.mojang.brigadier.CommandDispatcher;
import com.mojang.brigadier.exceptions.CommandSyntaxException;
import com.mojang.brigadier.exceptions.SimpleCommandExceptionType;
import java.util.Collection;
import net.minecraft.commands.CommandSourceStack;
import net.minecraft.commands.Commands;
import net.minecraft.commands.arguments.GameProfileArgument;
import net.minecraft.commands.arguments.MessageArgument;
import net.minecraft.network.chat.Component;
import net.minecraft.server.level.ServerPlayer;
import net.minecraft.server.players.UserBanList;
import net.minecraft.server.players.UserBanListEntry;
import org.jetbrains.annotations.Nullable;

public class BanPlayerCommands {
	private static final SimpleCommandExceptionType ERROR_ALREADY_BANNED = new SimpleCommandExceptionType(Component.translatable("commands.ban.failed"));

	public static void register(CommandDispatcher<CommandSourceStack> dispatcher) {
		dispatcher.register(
			Commands.literal("ban")
				.requires(commandSourceStack -> commandSourceStack.hasPermission(3))
				.then(
					Commands.argument("targets", GameProfileArgument.gameProfile())
						.executes(commandContext -> banPlayers(commandContext.getSource(), GameProfileArgument.getGameProfiles(commandContext, "targets"), null))
						.then(
							Commands.argument("reason", MessageArgument.message())
								.executes(
									commandContext -> banPlayers(
										commandContext.getSource(), GameProfileArgument.getGameProfiles(commandContext, "targets"), MessageArgument.getMessage(commandContext, "reason")
									)
								)
						)
				)
		);
	}

	private static int banPlayers(CommandSourceStack source, Collection<GameProfile> gameProfiles, @Nullable Component reason) throws CommandSyntaxException {
		UserBanList userBanList = source.getServer().getPlayerList().getBans();
		int i = 0;

		for (GameProfile gameProfile : gameProfiles) {
			if (!userBanList.isBanned(gameProfile)) {
				UserBanListEntry userBanListEntry = new UserBanListEntry(gameProfile, null, source.getTextName(), null, reason == null ? null : reason.getString());
				userBanList.add(userBanListEntry);
				i++;
				source.sendSuccess(() -> Component.translatable("commands.ban.success", Component.literal(gameProfile.getName()), userBanListEntry.getReason()), true);
				ServerPlayer serverPlayer = source.getServer().getPlayerList().getPlayer(gameProfile.getId());
				if (serverPlayer != null) {
					serverPlayer.connection.disconnect(Component.translatable("multiplayer.disconnect.banned"));
				}
			}
		}

		if (i == 0) {
			throw ERROR_ALREADY_BANNED.create();
		} else {
			return i;
		}
	}
}