package net.minecraft.server.commands;

import com.google.common.collect.Lists;
import com.mojang.brigadier.CommandDispatcher;
import com.mojang.logging.LogUtils;
import java.util.Collection;
import net.minecraft.commands.CommandSourceStack;
import net.minecraft.commands.Commands;
import net.minecraft.network.chat.Component;
import net.minecraft.server.MinecraftServer;
import net.minecraft.server.packs.repository.PackRepository;
import net.minecraft.world.level.storage.WorldData;
import org.slf4j.Logger;

public class ReloadCommand {
	private static final Logger LOGGER = LogUtils.getLogger();

	public static void reloadPacks(final Collection<String> selectedPacks, final CommandSourceStack source) {
		source.getServer().reloadResources(selectedPacks).exceptionally(throwable -> {
			LOGGER.warn("Failed to execute reload", throwable);
			source.sendFailure(Component.translatable("commands.reload.failure"));
			return null;
		});
	}

	private static Collection<String> discoverNewPacks(final PackRepository packRepository, final WorldData worldData, final Collection<String> currentPacks) {
		packRepository.reload();
		Collection<String> selected = Lists.<String>newArrayList(currentPacks);
		Collection<String> disabled = worldData.getDataConfiguration().dataPacks().getDisabled();

		for (String pack : packRepository.getAvailableIds()) {
			if (!disabled.contains(pack) && !selected.contains(pack)) {
				selected.add(pack);
			}
		}

		return selected;
	}

	public static void register(final CommandDispatcher<CommandSourceStack> dispatcher) {
		dispatcher.register(Commands.literal("reload").requires(Commands.hasPermission(Commands.LEVEL_GAMEMASTERS)).executes(s -> {
			CommandSourceStack source = s.getSource();
			MinecraftServer server = source.getServer();
			PackRepository packRepository = server.getPackRepository();
			WorldData worldData = server.getWorldData();
			Collection<String> currentPacks = packRepository.getSelectedIds();
			Collection<String> newSelectedPacks = discoverNewPacks(packRepository, worldData, currentPacks);
			source.sendSuccess(() -> Component.translatable("commands.reload.success"), true);
			reloadPacks(newSelectedPacks, source);
			return 0;
		}));
	}
}