package net.minecraft.server.commands;

import com.mojang.brigadier.CommandDispatcher;
import com.mojang.brigadier.arguments.FloatArgumentType;
import com.mojang.brigadier.builder.LiteralArgumentBuilder;
import com.mojang.brigadier.builder.RequiredArgumentBuilder;
import com.mojang.brigadier.exceptions.CommandSyntaxException;
import com.mojang.brigadier.exceptions.SimpleCommandExceptionType;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import net.minecraft.commands.CommandSourceStack;
import net.minecraft.commands.Commands;
import net.minecraft.commands.arguments.EntityArgument;
import net.minecraft.commands.arguments.IdentifierArgument;
import net.minecraft.commands.arguments.coordinates.Vec3Argument;
import net.minecraft.commands.synchronization.SuggestionProviders;
import net.minecraft.core.Holder;
import net.minecraft.network.chat.Component;
import net.minecraft.network.protocol.game.ClientboundSoundPacket;
import net.minecraft.resources.Identifier;
import net.minecraft.server.level.ServerLevel;
import net.minecraft.server.level.ServerPlayer;
import net.minecraft.sounds.SoundEvent;
import net.minecraft.sounds.SoundSource;
import net.minecraft.util.Mth;
import net.minecraft.world.phys.Vec3;
import org.jspecify.annotations.Nullable;

public class PlaySoundCommand {
	private static final SimpleCommandExceptionType ERROR_TOO_FAR = new SimpleCommandExceptionType(Component.translatable("commands.playsound.failed"));

	public static void register(final CommandDispatcher<CommandSourceStack> dispatcher) {
		RequiredArgumentBuilder<CommandSourceStack, Identifier> name = Commands.argument("sound", IdentifierArgument.id())
			.suggests(SuggestionProviders.cast(SuggestionProviders.AVAILABLE_SOUNDS))
			.executes(
				c -> playSound(
					c.getSource(),
					getCallingPlayerAsCollection(c.getSource().getPlayer()),
					IdentifierArgument.getId(c, "sound"),
					SoundSource.MASTER,
					c.getSource().getPosition(),
					1.0F,
					1.0F,
					0.0F
				)
			);

		for (SoundSource source : SoundSource.values()) {
			name.then(source(source));
		}

		dispatcher.register(Commands.literal("playsound").requires(Commands.hasPermission(Commands.LEVEL_GAMEMASTERS)).then(name));
	}

	private static LiteralArgumentBuilder<CommandSourceStack> source(final SoundSource source) {
		return Commands.literal(source.getName())
			.executes(
				c -> playSound(
					c.getSource(),
					getCallingPlayerAsCollection(c.getSource().getPlayer()),
					IdentifierArgument.getId(c, "sound"),
					source,
					c.getSource().getPosition(),
					1.0F,
					1.0F,
					0.0F
				)
			)
			.then(
				Commands.argument("targets", EntityArgument.players())
					.executes(
						c -> playSound(
							c.getSource(), EntityArgument.getPlayers(c, "targets"), IdentifierArgument.getId(c, "sound"), source, c.getSource().getPosition(), 1.0F, 1.0F, 0.0F
						)
					)
					.then(
						Commands.argument("pos", Vec3Argument.vec3())
							.executes(
								c -> playSound(
									c.getSource(), EntityArgument.getPlayers(c, "targets"), IdentifierArgument.getId(c, "sound"), source, Vec3Argument.getVec3(c, "pos"), 1.0F, 1.0F, 0.0F
								)
							)
							.then(
								Commands.argument("volume", FloatArgumentType.floatArg(0.0F))
									.executes(
										c -> playSound(
											c.getSource(),
											EntityArgument.getPlayers(c, "targets"),
											IdentifierArgument.getId(c, "sound"),
											source,
											Vec3Argument.getVec3(c, "pos"),
											c.<Float>getArgument("volume", Float.class),
											1.0F,
											0.0F
										)
									)
									.then(
										Commands.argument("pitch", FloatArgumentType.floatArg(0.0F, 2.0F))
											.executes(
												c -> playSound(
													c.getSource(),
													EntityArgument.getPlayers(c, "targets"),
													IdentifierArgument.getId(c, "sound"),
													source,
													Vec3Argument.getVec3(c, "pos"),
													c.<Float>getArgument("volume", Float.class),
													c.<Float>getArgument("pitch", Float.class),
													0.0F
												)
											)
											.then(
												Commands.argument("minVolume", FloatArgumentType.floatArg(0.0F, 1.0F))
													.executes(
														c -> playSound(
															c.getSource(),
															EntityArgument.getPlayers(c, "targets"),
															IdentifierArgument.getId(c, "sound"),
															source,
															Vec3Argument.getVec3(c, "pos"),
															c.<Float>getArgument("volume", Float.class),
															c.<Float>getArgument("pitch", Float.class),
															c.<Float>getArgument("minVolume", Float.class)
														)
													)
											)
									)
							)
					)
			);
	}

	private static Collection<ServerPlayer> getCallingPlayerAsCollection(@Nullable final ServerPlayer player) {
		return player != null ? List.of(player) : List.of();
	}

	private static int playSound(
		final CommandSourceStack source,
		final Collection<ServerPlayer> players,
		final Identifier sound,
		final SoundSource soundSource,
		final Vec3 position,
		final float volume,
		final float pitch,
		final float minVolume
	) throws CommandSyntaxException {
		Holder<SoundEvent> soundHolder = Holder.direct(SoundEvent.createVariableRangeEvent(sound));
		double maxDistSqr = Mth.square(soundHolder.value().getRange(volume));
		ServerLevel level = source.getLevel();
		long seed = level.getRandom().nextLong();
		List<ServerPlayer> playedFor = new ArrayList();

		for (ServerPlayer player : players) {
			if (player.level() == level) {
				double deltaX = position.x - player.getX();
				double deltaY = position.y - player.getY();
				double deltaZ = position.z - player.getZ();
				double distSqr = deltaX * deltaX + deltaY * deltaY + deltaZ * deltaZ;
				Vec3 localPosition = position;
				float localVolume = volume;
				if (distSqr > maxDistSqr) {
					if (minVolume <= 0.0F) {
						continue;
					}

					double distance = Math.sqrt(distSqr);
					localPosition = new Vec3(player.getX() + deltaX / distance * 2.0, player.getY() + deltaY / distance * 2.0, player.getZ() + deltaZ / distance * 2.0);
					localVolume = minVolume;
				}

				player.connection
					.send(new ClientboundSoundPacket(soundHolder, soundSource, localPosition.x(), localPosition.y(), localPosition.z(), localVolume, pitch, seed));
				playedFor.add(player);
			}
		}

		int count = playedFor.size();
		if (count == 0) {
			throw ERROR_TOO_FAR.create();
		} else {
			if (count == 1) {
				source.sendSuccess(
					() -> Component.translatable("commands.playsound.success.single", Component.translationArg(sound), ((ServerPlayer)playedFor.getFirst()).getDisplayName()),
					true
				);
			} else {
				source.sendSuccess(() -> Component.translatable("commands.playsound.success.multiple", Component.translationArg(sound), count), true);
			}

			return count;
		}
	}
}