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.level.block.state.BlockState;
import net.minecraft.world.level.block.state.pattern.BlockInWorld;
import org.jspecify.annotations.Nullable;

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

	public static void register(final CommandDispatcher<CommandSourceStack> dispatcher, final CommandBuildContext context) {
		Predicate<BlockInWorld> filter = b -> b.getLevel().isEmptyBlock(b.getPos());
		dispatcher.register(
			Commands.literal("setblock")
				.requires(Commands.hasPermission(Commands.LEVEL_GAMEMASTERS))
				.then(
					Commands.argument("pos", BlockPosArgument.blockPos())
						.then(
							Commands.argument("block", BlockStateArgument.block(context))
								.executes(
									c -> setBlock(
										c.getSource(), BlockPosArgument.getLoadedBlockPos(c, "pos"), BlockStateArgument.getBlock(c, "block"), SetBlockCommand.Mode.REPLACE, null, false
									)
								)
								.then(
									Commands.literal("destroy")
										.executes(
											c -> setBlock(
												c.getSource(), BlockPosArgument.getLoadedBlockPos(c, "pos"), BlockStateArgument.getBlock(c, "block"), SetBlockCommand.Mode.DESTROY, null, false
											)
										)
								)
								.then(
									Commands.literal("keep")
										.executes(
											c -> setBlock(
												c.getSource(), BlockPosArgument.getLoadedBlockPos(c, "pos"), BlockStateArgument.getBlock(c, "block"), SetBlockCommand.Mode.REPLACE, filter, false
											)
										)
								)
								.then(
									Commands.literal("replace")
										.executes(
											c -> setBlock(
												c.getSource(), BlockPosArgument.getLoadedBlockPos(c, "pos"), BlockStateArgument.getBlock(c, "block"), SetBlockCommand.Mode.REPLACE, null, false
											)
										)
								)
								.then(
									Commands.literal("strict")
										.executes(
											c -> setBlock(
												c.getSource(), BlockPosArgument.getLoadedBlockPos(c, "pos"), BlockStateArgument.getBlock(c, "block"), SetBlockCommand.Mode.REPLACE, null, true
											)
										)
								)
						)
				)
		);
	}

	private static int setBlock(
		final CommandSourceStack source,
		final BlockPos pos,
		final BlockInput block,
		final SetBlockCommand.Mode mode,
		@Nullable final Predicate<BlockInWorld> predicate,
		final boolean strict
	) throws CommandSyntaxException {
		ServerLevel level = source.getLevel();
		if (level.isDebug()) {
			throw ERROR_FAILED.create();
		} else if (predicate != null && !predicate.test(new BlockInWorld(level, pos, true))) {
			throw ERROR_FAILED.create();
		} else {
			boolean placeNeeded;
			if (mode == SetBlockCommand.Mode.DESTROY) {
				level.destroyBlock(pos, true);
				placeNeeded = !block.getState().isAir() || !level.getBlockState(pos).isAir();
			} else {
				placeNeeded = true;
			}

			BlockState oldState = level.getBlockState(pos);
			if (placeNeeded && !block.place(level, pos, 2 | (strict ? 816 : 256))) {
				throw ERROR_FAILED.create();
			} else {
				if (!strict) {
					level.updateNeighboursOnBlockSet(pos, oldState);
				}

				source.sendSuccess(() -> Component.translatable("commands.setblock.success", pos.getX(), pos.getY(), pos.getZ()), true);
				return 1;
			}
		}
	}

	public static enum Mode {
		REPLACE,
		DESTROY;
	}
}