package net.minecraft.server.commands;

import com.mojang.brigadier.CommandDispatcher;
import com.mojang.brigadier.context.CommandContext;
import java.util.Collection;
import java.util.Collections;
import net.minecraft.commands.CommandSourceStack;
import net.minecraft.commands.Commands;
import net.minecraft.commands.arguments.EntityArgument;
import net.minecraft.commands.arguments.GameModeArgument;
import net.minecraft.network.chat.Component;
import net.minecraft.server.level.ServerPlayer;
import net.minecraft.server.permissions.PermissionCheck;
import net.minecraft.server.permissions.Permissions;
import net.minecraft.world.level.GameType;
import net.minecraft.world.level.gamerules.GameRules;

public class GameModeCommand {
	public static final PermissionCheck PERMISSION_CHECK = new PermissionCheck.Require(Permissions.COMMANDS_GAMEMASTER);

	public static void register(final CommandDispatcher<CommandSourceStack> dispatcher) {
		dispatcher.register(
			Commands.literal("gamemode")
				.requires(Commands.hasPermission(PERMISSION_CHECK))
				.then(
					Commands.argument("gamemode", GameModeArgument.gameMode())
						.executes(c -> setMode(c, Collections.singleton(c.getSource().getPlayerOrException()), GameModeArgument.getGameMode(c, "gamemode")))
						.then(
							Commands.argument("target", EntityArgument.players())
								.executes(c -> setMode(c, EntityArgument.getPlayers(c, "target"), GameModeArgument.getGameMode(c, "gamemode")))
						)
				)
		);
	}

	private static void logGamemodeChange(final CommandSourceStack source, final ServerPlayer target, final GameType newType) {
		Component mode = Component.translatable("gameMode." + newType.getName());
		if (source.getEntity() == target) {
			source.sendSuccess(() -> Component.translatable("commands.gamemode.success.self", mode), true);
		} else {
			if (source.getLevel().getGameRules().get(GameRules.SEND_COMMAND_FEEDBACK)) {
				target.sendSystemMessage(Component.translatable("gameMode.changed", mode));
			}

			source.sendSuccess(() -> Component.translatable("commands.gamemode.success.other", target.getDisplayName(), mode), true);
		}
	}

	private static int setMode(final CommandContext<CommandSourceStack> context, final Collection<ServerPlayer> players, final GameType type) {
		int count = 0;

		for (ServerPlayer player : players) {
			if (setGameMode(context.getSource(), player, type)) {
				count++;
			}
		}

		return count;
	}

	public static void setGameMode(final ServerPlayer player, final GameType type) {
		setGameMode(player.createCommandSourceStack(), player, type);
	}

	private static boolean setGameMode(final CommandSourceStack source, final ServerPlayer player, final GameType type) {
		if (player.setGameMode(type)) {
			logGamemodeChange(source, player, type);
			return true;
		} else {
			return false;
		}
	}
}