minecraft-src/net/minecraft/server/commands/EffectCommands.java

package net.minecraft.server.commands;

import com.google.common.collect.ImmutableList;
import com.mojang.brigadier.CommandDispatcher;
import com.mojang.brigadier.arguments.BoolArgumentType;
import com.mojang.brigadier.arguments.IntegerArgumentType;
import com.mojang.brigadier.exceptions.CommandSyntaxException;
import com.mojang.brigadier.exceptions.SimpleCommandExceptionType;
import java.util.Collection;
import net.minecraft.commands.CommandBuildContext;
import net.minecraft.commands.CommandSourceStack;
import net.minecraft.commands.Commands;
import net.minecraft.commands.arguments.EntityArgument;
import net.minecraft.commands.arguments.ResourceArgument;
import net.minecraft.core.Holder;
import net.minecraft.core.registries.Registries;
import net.minecraft.network.chat.Component;
import net.minecraft.world.effect.MobEffect;
import net.minecraft.world.effect.MobEffectInstance;
import net.minecraft.world.entity.Entity;
import net.minecraft.world.entity.LivingEntity;
import org.jetbrains.annotations.Nullable;

public class EffectCommands {
	private static final SimpleCommandExceptionType ERROR_GIVE_FAILED = new SimpleCommandExceptionType(Component.translatable("commands.effect.give.failed"));
	private static final SimpleCommandExceptionType ERROR_CLEAR_EVERYTHING_FAILED = new SimpleCommandExceptionType(
		Component.translatable("commands.effect.clear.everything.failed")
	);
	private static final SimpleCommandExceptionType ERROR_CLEAR_SPECIFIC_FAILED = new SimpleCommandExceptionType(
		Component.translatable("commands.effect.clear.specific.failed")
	);

	public static void register(CommandDispatcher<CommandSourceStack> dispatcher, CommandBuildContext context) {
		dispatcher.register(
			Commands.literal("effect")
				.requires(commandSourceStack -> commandSourceStack.hasPermission(2))
				.then(
					Commands.literal("clear")
						.executes(commandContext -> clearEffects(commandContext.getSource(), ImmutableList.of(commandContext.getSource().getEntityOrException())))
						.then(
							Commands.argument("targets", EntityArgument.entities())
								.executes(commandContext -> clearEffects(commandContext.getSource(), EntityArgument.getEntities(commandContext, "targets")))
								.then(
									Commands.argument("effect", ResourceArgument.resource(context, Registries.MOB_EFFECT))
										.executes(
											commandContext -> clearEffect(
												commandContext.getSource(), EntityArgument.getEntities(commandContext, "targets"), ResourceArgument.getMobEffect(commandContext, "effect")
											)
										)
								)
						)
				)
				.then(
					Commands.literal("give")
						.then(
							Commands.argument("targets", EntityArgument.entities())
								.then(
									Commands.argument("effect", ResourceArgument.resource(context, Registries.MOB_EFFECT))
										.executes(
											commandContext -> giveEffect(
												commandContext.getSource(),
												EntityArgument.getEntities(commandContext, "targets"),
												ResourceArgument.getMobEffect(commandContext, "effect"),
												null,
												0,
												true
											)
										)
										.then(
											Commands.argument("seconds", IntegerArgumentType.integer(1, 1000000))
												.executes(
													commandContext -> giveEffect(
														commandContext.getSource(),
														EntityArgument.getEntities(commandContext, "targets"),
														ResourceArgument.getMobEffect(commandContext, "effect"),
														IntegerArgumentType.getInteger(commandContext, "seconds"),
														0,
														true
													)
												)
												.then(
													Commands.argument("amplifier", IntegerArgumentType.integer(0, 255))
														.executes(
															commandContext -> giveEffect(
																commandContext.getSource(),
																EntityArgument.getEntities(commandContext, "targets"),
																ResourceArgument.getMobEffect(commandContext, "effect"),
																IntegerArgumentType.getInteger(commandContext, "seconds"),
																IntegerArgumentType.getInteger(commandContext, "amplifier"),
																true
															)
														)
														.then(
															Commands.argument("hideParticles", BoolArgumentType.bool())
																.executes(
																	commandContext -> giveEffect(
																		commandContext.getSource(),
																		EntityArgument.getEntities(commandContext, "targets"),
																		ResourceArgument.getMobEffect(commandContext, "effect"),
																		IntegerArgumentType.getInteger(commandContext, "seconds"),
																		IntegerArgumentType.getInteger(commandContext, "amplifier"),
																		!BoolArgumentType.getBool(commandContext, "hideParticles")
																	)
																)
														)
												)
										)
										.then(
											Commands.literal("infinite")
												.executes(
													commandContext -> giveEffect(
														commandContext.getSource(),
														EntityArgument.getEntities(commandContext, "targets"),
														ResourceArgument.getMobEffect(commandContext, "effect"),
														-1,
														0,
														true
													)
												)
												.then(
													Commands.argument("amplifier", IntegerArgumentType.integer(0, 255))
														.executes(
															commandContext -> giveEffect(
																commandContext.getSource(),
																EntityArgument.getEntities(commandContext, "targets"),
																ResourceArgument.getMobEffect(commandContext, "effect"),
																-1,
																IntegerArgumentType.getInteger(commandContext, "amplifier"),
																true
															)
														)
														.then(
															Commands.argument("hideParticles", BoolArgumentType.bool())
																.executes(
																	commandContext -> giveEffect(
																		commandContext.getSource(),
																		EntityArgument.getEntities(commandContext, "targets"),
																		ResourceArgument.getMobEffect(commandContext, "effect"),
																		-1,
																		IntegerArgumentType.getInteger(commandContext, "amplifier"),
																		!BoolArgumentType.getBool(commandContext, "hideParticles")
																	)
																)
														)
												)
										)
								)
						)
				)
		);
	}

	private static int giveEffect(
		CommandSourceStack source, Collection<? extends Entity> targets, Holder<MobEffect> effect, @Nullable Integer seconds, int amplifier, boolean showParticles
	) throws CommandSyntaxException {
		MobEffect mobEffect = effect.value();
		int i = 0;
		int j;
		if (seconds != null) {
			if (mobEffect.isInstantenous()) {
				j = seconds;
			} else if (seconds == -1) {
				j = -1;
			} else {
				j = seconds * 20;
			}
		} else if (mobEffect.isInstantenous()) {
			j = 1;
		} else {
			j = 600;
		}

		for (Entity entity : targets) {
			if (entity instanceof LivingEntity) {
				MobEffectInstance mobEffectInstance = new MobEffectInstance(effect, j, amplifier, false, showParticles);
				if (((LivingEntity)entity).addEffect(mobEffectInstance, source.getEntity())) {
					i++;
				}
			}
		}

		if (i == 0) {
			throw ERROR_GIVE_FAILED.create();
		} else {
			if (targets.size() == 1) {
				source.sendSuccess(
					() -> Component.translatable(
						"commands.effect.give.success.single", mobEffect.getDisplayName(), ((Entity)targets.iterator().next()).getDisplayName(), j / 20
					),
					true
				);
			} else {
				source.sendSuccess(() -> Component.translatable("commands.effect.give.success.multiple", mobEffect.getDisplayName(), targets.size(), j / 20), true);
			}

			return i;
		}
	}

	private static int clearEffects(CommandSourceStack source, Collection<? extends Entity> targets) throws CommandSyntaxException {
		int i = 0;

		for (Entity entity : targets) {
			if (entity instanceof LivingEntity && ((LivingEntity)entity).removeAllEffects()) {
				i++;
			}
		}

		if (i == 0) {
			throw ERROR_CLEAR_EVERYTHING_FAILED.create();
		} else {
			if (targets.size() == 1) {
				source.sendSuccess(
					() -> Component.translatable("commands.effect.clear.everything.success.single", ((Entity)targets.iterator().next()).getDisplayName()), true
				);
			} else {
				source.sendSuccess(() -> Component.translatable("commands.effect.clear.everything.success.multiple", targets.size()), true);
			}

			return i;
		}
	}

	private static int clearEffect(CommandSourceStack source, Collection<? extends Entity> targets, Holder<MobEffect> effect) throws CommandSyntaxException {
		MobEffect mobEffect = effect.value();
		int i = 0;

		for (Entity entity : targets) {
			if (entity instanceof LivingEntity && ((LivingEntity)entity).removeEffect(effect)) {
				i++;
			}
		}

		if (i == 0) {
			throw ERROR_CLEAR_SPECIFIC_FAILED.create();
		} else {
			if (targets.size() == 1) {
				source.sendSuccess(
					() -> Component.translatable(
						"commands.effect.clear.specific.success.single", mobEffect.getDisplayName(), ((Entity)targets.iterator().next()).getDisplayName()
					),
					true
				);
			} else {
				source.sendSuccess(() -> Component.translatable("commands.effect.clear.specific.success.multiple", mobEffect.getDisplayName(), targets.size()), true);
			}

			return i;
		}
	}
}