package net.minecraft.client.multiplayer;

import com.mojang.logging.LogUtils;
import it.unimi.dsi.fastutil.objects.Object2ObjectOpenHashMap;
import java.util.Map;
import java.util.Optional;
import java.util.Map.Entry;
import net.minecraft.advancements.AdvancementHolder;
import net.minecraft.advancements.AdvancementNode;
import net.minecraft.advancements.AdvancementProgress;
import net.minecraft.advancements.AdvancementTree;
import net.minecraft.advancements.DisplayInfo;
import net.minecraft.client.Minecraft;
import net.minecraft.client.gui.components.toasts.AdvancementToast;
import net.minecraft.client.telemetry.WorldSessionTelemetryManager;
import net.minecraft.network.protocol.game.ClientboundUpdateAdvancementsPacket;
import net.minecraft.network.protocol.game.ServerboundSeenAdvancementsPacket;
import net.minecraft.resources.Identifier;
import org.jspecify.annotations.Nullable;
import org.slf4j.Logger;

public class ClientAdvancements {
	private static final Logger LOGGER = LogUtils.getLogger();
	private final Minecraft minecraft;
	private final WorldSessionTelemetryManager telemetryManager;
	private final AdvancementTree tree = new AdvancementTree();
	private final Map<AdvancementHolder, AdvancementProgress> progress = new Object2ObjectOpenHashMap<>();
	@Nullable
	private ClientAdvancements.Listener listener;
	@Nullable
	private AdvancementHolder selectedTab;

	public ClientAdvancements(final Minecraft minecraft, final WorldSessionTelemetryManager telemetryManager) {
		this.minecraft = minecraft;
		this.telemetryManager = telemetryManager;
	}

	public void update(final ClientboundUpdateAdvancementsPacket packet) {
		if (packet.shouldReset()) {
			this.tree.clear();
			this.progress.clear();
		}

		this.tree.remove(packet.getRemoved());
		this.tree.addAll(packet.getAdded());

		for (Entry<Identifier, AdvancementProgress> entry : packet.getProgress().entrySet()) {
			AdvancementNode node = this.tree.get((Identifier)entry.getKey());
			if (node != null) {
				AdvancementProgress progress = (AdvancementProgress)entry.getValue();
				progress.update(node.advancement().requirements());
				this.progress.put(node.holder(), progress);
				if (this.listener != null) {
					this.listener.onUpdateAdvancementProgress(node, progress);
				}

				if (!packet.shouldReset() && progress.isDone()) {
					if (this.minecraft.level != null) {
						this.telemetryManager.onAdvancementDone(this.minecraft.level, node.holder());
					}

					Optional<DisplayInfo> display = node.advancement().display();
					if (packet.shouldShowAdvancements() && display.isPresent() && ((DisplayInfo)display.get()).shouldShowToast()) {
						this.minecraft.gui.toastManager().addToast(new AdvancementToast(node.holder()));
					}
				}
			} else {
				LOGGER.warn("Server informed client about progress for unknown advancement {}", entry.getKey());
			}
		}
	}

	public AdvancementTree getTree() {
		return this.tree;
	}

	public void setSelectedTab(@Nullable final AdvancementHolder selectedTab, final boolean tellServer) {
		ClientPacketListener connection = this.minecraft.getConnection();
		if (connection != null && selectedTab != null && tellServer) {
			connection.send(ServerboundSeenAdvancementsPacket.openedTab(selectedTab));
		}

		if (this.selectedTab != selectedTab) {
			this.selectedTab = selectedTab;
			if (this.listener != null) {
				this.listener.onSelectedTabChanged(selectedTab);
			}
		}
	}

	public void setListener(@Nullable final ClientAdvancements.Listener listener) {
		this.listener = listener;
		this.tree.setListener(listener);
		if (listener != null) {
			this.progress.forEach((holder, progress) -> {
				AdvancementNode node = this.tree.get(holder);
				if (node != null) {
					listener.onUpdateAdvancementProgress(node, progress);
				}
			});
			listener.onSelectedTabChanged(this.selectedTab);
		}
	}

	@Nullable
	public AdvancementHolder get(final Identifier id) {
		AdvancementNode node = this.tree.get(id);
		return node != null ? node.holder() : null;
	}

	public interface Listener extends net.minecraft.advancements.AdvancementTree.Listener {
		void onUpdateAdvancementProgress(AdvancementNode advancement, AdvancementProgress progress);

		void onSelectedTabChanged(@Nullable AdvancementHolder selectedTab);
	}
}