package net.minecraft.server.commands;

import com.mojang.brigadier.CommandDispatcher;
import com.mojang.brigadier.arguments.DoubleArgumentType;
import com.mojang.brigadier.exceptions.CommandSyntaxException;
import com.mojang.brigadier.exceptions.DynamicCommandExceptionType;
import com.mojang.brigadier.suggestion.SuggestionProvider;
import net.minecraft.commands.CommandSourceStack;
import net.minecraft.commands.Commands;
import net.minecraft.commands.SharedSuggestionProvider;
import net.minecraft.commands.arguments.IdentifierArgument;
import net.minecraft.network.chat.Component;
import net.minecraft.resources.Identifier;
import net.minecraft.server.MinecraftServer;
import net.minecraft.world.Stopwatch;
import net.minecraft.world.Stopwatches;

public class StopwatchCommand {
	private static final DynamicCommandExceptionType ERROR_ALREADY_EXISTS = new DynamicCommandExceptionType(
		id -> Component.translatableEscape("commands.stopwatch.already_exists", id)
	);
	public static final DynamicCommandExceptionType ERROR_DOES_NOT_EXIST = new DynamicCommandExceptionType(
		id -> Component.translatableEscape("commands.stopwatch.does_not_exist", id)
	);
	public static final SuggestionProvider<CommandSourceStack> SUGGEST_STOPWATCHES = (c, p) -> SharedSuggestionProvider.suggestResource(
		c.getSource().getServer().getStopwatches().ids(), p
	);

	public static void register(final CommandDispatcher<CommandSourceStack> dispatcher) {
		dispatcher.register(
			Commands.literal("stopwatch")
				.requires(Commands.hasPermission(Commands.LEVEL_GAMEMASTERS))
				.then(
					Commands.literal("create")
						.then(Commands.argument("id", IdentifierArgument.id()).executes(c -> createStopwatch(c.getSource(), IdentifierArgument.getId(c, "id"))))
				)
				.then(
					Commands.literal("query")
						.then(
							Commands.argument("id", IdentifierArgument.id())
								.suggests(SUGGEST_STOPWATCHES)
								.then(
									Commands.argument("scale", DoubleArgumentType.doubleArg())
										.executes(c -> queryStopwatch(c.getSource(), IdentifierArgument.getId(c, "id"), DoubleArgumentType.getDouble(c, "scale")))
								)
								.executes(c -> queryStopwatch(c.getSource(), IdentifierArgument.getId(c, "id"), 1.0))
						)
				)
				.then(
					Commands.literal("restart")
						.then(
							Commands.argument("id", IdentifierArgument.id())
								.suggests(SUGGEST_STOPWATCHES)
								.executes(c -> restartStopwatch(c.getSource(), IdentifierArgument.getId(c, "id")))
						)
				)
				.then(
					Commands.literal("remove")
						.then(
							Commands.argument("id", IdentifierArgument.id())
								.suggests(SUGGEST_STOPWATCHES)
								.executes(c -> removeStopwatch(c.getSource(), IdentifierArgument.getId(c, "id")))
						)
				)
		);
	}

	private static int createStopwatch(final CommandSourceStack source, final Identifier id) throws CommandSyntaxException {
		MinecraftServer server = source.getServer();
		Stopwatches stopwatches = server.getStopwatches();
		Stopwatch now = new Stopwatch(Stopwatches.currentTime());
		if (!stopwatches.add(id, now)) {
			throw ERROR_ALREADY_EXISTS.create(id);
		} else {
			source.sendSuccess(() -> Component.translatable("commands.stopwatch.create.success", Component.translationArg(id)), true);
			return 1;
		}
	}

	private static int queryStopwatch(final CommandSourceStack source, final Identifier id, final double scale) throws CommandSyntaxException {
		MinecraftServer server = source.getServer();
		Stopwatches stopwatches = server.getStopwatches();
		Stopwatch stopwatch = stopwatches.get(id);
		if (stopwatch == null) {
			throw ERROR_DOES_NOT_EXIST.create(id);
		} else {
			long currentTime = Stopwatches.currentTime();
			double elapsedSeconds = stopwatch.elapsedSeconds(currentTime);
			source.sendSuccess(() -> Component.translatable("commands.stopwatch.query", Component.translationArg(id), elapsedSeconds), true);
			return (int)(elapsedSeconds * scale);
		}
	}

	private static int restartStopwatch(final CommandSourceStack source, final Identifier id) throws CommandSyntaxException {
		MinecraftServer server = source.getServer();
		Stopwatches stopwatches = server.getStopwatches();
		if (!stopwatches.update(id, stopwatch -> new Stopwatch(Stopwatches.currentTime()))) {
			throw ERROR_DOES_NOT_EXIST.create(id);
		} else {
			source.sendSuccess(() -> Component.translatable("commands.stopwatch.restart.success", Component.translationArg(id)), true);
			return 1;
		}
	}

	private static int removeStopwatch(final CommandSourceStack source, final Identifier id) throws CommandSyntaxException {
		MinecraftServer server = source.getServer();
		Stopwatches stopwatches = server.getStopwatches();
		if (!stopwatches.remove(id)) {
			throw ERROR_DOES_NOT_EXIST.create(id);
		} else {
			source.sendSuccess(() -> Component.translatable("commands.stopwatch.remove.success", Component.translationArg(id)), true);
			return 1;
		}
	}
}