package net.minecraft.server.commands;

import com.mojang.brigadier.CommandDispatcher;
import com.mojang.brigadier.arguments.IntegerArgumentType;
import com.mojang.brigadier.exceptions.CommandSyntaxException;
import com.mojang.brigadier.exceptions.Dynamic2CommandExceptionType;
import com.mojang.brigadier.exceptions.DynamicCommandExceptionType;
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.entity.Entity;
import net.minecraft.world.entity.LivingEntity;
import net.minecraft.world.item.ItemStack;
import net.minecraft.world.item.enchantment.Enchantment;
import net.minecraft.world.item.enchantment.EnchantmentHelper;

public class EnchantCommand {
	private static final DynamicCommandExceptionType ERROR_NOT_LIVING_ENTITY = new DynamicCommandExceptionType(
		object -> Component.translatableEscape("commands.enchant.failed.entity", object)
	);
	private static final DynamicCommandExceptionType ERROR_NO_ITEM = new DynamicCommandExceptionType(
		object -> Component.translatableEscape("commands.enchant.failed.itemless", object)
	);
	private static final DynamicCommandExceptionType ERROR_INCOMPATIBLE = new DynamicCommandExceptionType(
		object -> Component.translatableEscape("commands.enchant.failed.incompatible", object)
	);
	private static final Dynamic2CommandExceptionType ERROR_LEVEL_TOO_HIGH = new Dynamic2CommandExceptionType(
		(object, object2) -> Component.translatableEscape("commands.enchant.failed.level", object, object2)
	);
	private static final SimpleCommandExceptionType ERROR_NOTHING_HAPPENED = new SimpleCommandExceptionType(Component.translatable("commands.enchant.failed"));

	public static void register(CommandDispatcher<CommandSourceStack> dispatcher, CommandBuildContext context) {
		dispatcher.register(
			Commands.literal("enchant")
				.requires(commandSourceStack -> commandSourceStack.hasPermission(2))
				.then(
					Commands.argument("targets", EntityArgument.entities())
						.then(
							Commands.argument("enchantment", ResourceArgument.resource(context, Registries.ENCHANTMENT))
								.executes(
									commandContext -> enchant(
										commandContext.getSource(), EntityArgument.getEntities(commandContext, "targets"), ResourceArgument.getEnchantment(commandContext, "enchantment"), 1
									)
								)
								.then(
									Commands.argument("level", IntegerArgumentType.integer(0))
										.executes(
											commandContext -> enchant(
												commandContext.getSource(),
												EntityArgument.getEntities(commandContext, "targets"),
												ResourceArgument.getEnchantment(commandContext, "enchantment"),
												IntegerArgumentType.getInteger(commandContext, "level")
											)
										)
								)
						)
				)
		);
	}

	private static int enchant(CommandSourceStack source, Collection<? extends Entity> targets, Holder<Enchantment> enchantment, int level) throws CommandSyntaxException {
		Enchantment enchantment2 = enchantment.value();
		if (level > enchantment2.getMaxLevel()) {
			throw ERROR_LEVEL_TOO_HIGH.create(level, enchantment2.getMaxLevel());
		} else {
			int i = 0;

			for (Entity entity : targets) {
				if (entity instanceof LivingEntity livingEntity) {
					ItemStack itemStack = livingEntity.getMainHandItem();
					if (!itemStack.isEmpty()) {
						if (enchantment2.canEnchant(itemStack)
							&& EnchantmentHelper.isEnchantmentCompatible(EnchantmentHelper.getEnchantmentsForCrafting(itemStack).keySet(), enchantment)) {
							itemStack.enchant(enchantment, level);
							i++;
						} else if (targets.size() == 1) {
							throw ERROR_INCOMPATIBLE.create(itemStack.getHoverName().getString());
						}
					} else if (targets.size() == 1) {
						throw ERROR_NO_ITEM.create(livingEntity.getName().getString());
					}
				} else if (targets.size() == 1) {
					throw ERROR_NOT_LIVING_ENTITY.create(entity.getName().getString());
				}
			}

			if (i == 0) {
				throw ERROR_NOTHING_HAPPENED.create();
			} else {
				if (targets.size() == 1) {
					source.sendSuccess(
						() -> Component.translatable(
							"commands.enchant.success.single", Enchantment.getFullname(enchantment, level), ((Entity)targets.iterator().next()).getDisplayName()
						),
						true
					);
				} else {
					source.sendSuccess(() -> Component.translatable("commands.enchant.success.multiple", Enchantment.getFullname(enchantment, level), targets.size()), true);
				}

				return i;
			}
		}
	}
}