package net.minecraft.server.commands;

import com.mojang.brigadier.CommandDispatcher;
import com.mojang.brigadier.exceptions.CommandSyntaxException;
import com.mojang.brigadier.exceptions.SimpleCommandExceptionType;
import java.util.function.Predicate;
import net.minecraft.commands.CommandBuildContext;
import net.minecraft.commands.CommandSourceStack;
import net.minecraft.commands.Commands;
import net.minecraft.commands.arguments.blocks.BlockInput;
import net.minecraft.commands.arguments.blocks.BlockStateArgument;
import net.minecraft.commands.arguments.coordinates.BlockPosArgument;
import net.minecraft.core.BlockPos;
import net.minecraft.network.chat.Component;
import net.minecraft.server.level.ServerLevel;
import net.minecraft.world.Clearable;
import net.minecraft.world.level.block.entity.BlockEntity;
import net.minecraft.world.level.block.state.pattern.BlockInWorld;
import net.minecraft.world.level.levelgen.structure.BoundingBox;
import org.jetbrains.annotations.Nullable;

public class SetBlockCommand {
	private static final SimpleCommandExceptionType ERROR_FAILED = new SimpleCommandExceptionType(Component.translatable("commands.setblock.failed"));

	public static void register(CommandDispatcher<CommandSourceStack> dispatcher, CommandBuildContext context) {
		dispatcher.register(
			Commands.literal("setblock")
				.requires(commandSourceStack -> commandSourceStack.hasPermission(2))
				.then(
					Commands.argument("pos", BlockPosArgument.blockPos())
						.then(
							Commands.argument("block", BlockStateArgument.block(context))
								.executes(
									commandContext -> setBlock(
										commandContext.getSource(),
										BlockPosArgument.getLoadedBlockPos(commandContext, "pos"),
										BlockStateArgument.getBlock(commandContext, "block"),
										SetBlockCommand.Mode.REPLACE,
										null
									)
								)
								.then(
									Commands.literal("destroy")
										.executes(
											commandContext -> setBlock(
												commandContext.getSource(),
												BlockPosArgument.getLoadedBlockPos(commandContext, "pos"),
												BlockStateArgument.getBlock(commandContext, "block"),
												SetBlockCommand.Mode.DESTROY,
												null
											)
										)
								)
								.then(
									Commands.literal("keep")
										.executes(
											commandContext -> setBlock(
												commandContext.getSource(),
												BlockPosArgument.getLoadedBlockPos(commandContext, "pos"),
												BlockStateArgument.getBlock(commandContext, "block"),
												SetBlockCommand.Mode.REPLACE,
												blockInWorld -> blockInWorld.getLevel().isEmptyBlock(blockInWorld.getPos())
											)
										)
								)
								.then(
									Commands.literal("replace")
										.executes(
											commandContext -> setBlock(
												commandContext.getSource(),
												BlockPosArgument.getLoadedBlockPos(commandContext, "pos"),
												BlockStateArgument.getBlock(commandContext, "block"),
												SetBlockCommand.Mode.REPLACE,
												null
											)
										)
								)
						)
				)
		);
	}

	private static int setBlock(CommandSourceStack source, BlockPos pos, BlockInput state, SetBlockCommand.Mode mode, @Nullable Predicate<BlockInWorld> predicate) throws CommandSyntaxException {
		ServerLevel serverLevel = source.getLevel();
		if (predicate != null && !predicate.test(new BlockInWorld(serverLevel, pos, true))) {
			throw ERROR_FAILED.create();
		} else {
			boolean bl;
			if (mode == SetBlockCommand.Mode.DESTROY) {
				serverLevel.destroyBlock(pos, true);
				bl = !state.getState().isAir() || !serverLevel.getBlockState(pos).isAir();
			} else {
				BlockEntity blockEntity = serverLevel.getBlockEntity(pos);
				Clearable.tryClear(blockEntity);
				bl = true;
			}

			if (bl && !state.place(serverLevel, pos, 2)) {
				throw ERROR_FAILED.create();
			} else {
				serverLevel.blockUpdated(pos, state.getState().getBlock());
				source.sendSuccess(() -> Component.translatable("commands.setblock.success", pos.getX(), pos.getY(), pos.getZ()), true);
				return 1;
			}
		}
	}

	public interface Filter {
		@Nullable
		BlockInput filter(BoundingBox boundingBox, BlockPos blockPos, BlockInput blockInput, ServerLevel serverLevel);
	}

	public static enum Mode {
		REPLACE,
		DESTROY;
	}
}