package net.minecraft.client.gui.screens.social;

import com.google.common.collect.Maps;
import com.google.common.collect.Sets;
import com.mojang.authlib.GameProfile;
import com.mojang.authlib.minecraft.UserApiService;
import java.util.Map;
import java.util.Set;
import java.util.UUID;
import java.util.concurrent.CompletableFuture;
import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.client.Minecraft;
import net.minecraft.client.multiplayer.PlayerInfo;
import net.minecraft.util.Util;

@Environment(EnvType.CLIENT)
public class PlayerSocialManager {
	private final Minecraft minecraft;
	private final Set<UUID> hiddenPlayers = Sets.<UUID>newHashSet();
	private final UserApiService service;
	private final Map<String, UUID> discoveredNamesToUUID = Maps.<String, UUID>newHashMap();
	private boolean onlineMode;
	private CompletableFuture<?> pendingBlockListRefresh = CompletableFuture.completedFuture(null);

	public PlayerSocialManager(final Minecraft minecraft, final UserApiService service) {
		this.minecraft = minecraft;
		this.service = service;
	}

	public void hidePlayer(final UUID id) {
		this.hiddenPlayers.add(id);
	}

	public void showPlayer(final UUID id) {
		this.hiddenPlayers.remove(id);
	}

	public boolean shouldHideMessageFrom(final UUID id) {
		return this.isHidden(id) || this.isBlocked(id);
	}

	public boolean isHidden(final UUID id) {
		return this.hiddenPlayers.contains(id);
	}

	public void startOnlineMode() {
		this.onlineMode = true;
		this.pendingBlockListRefresh = this.pendingBlockListRefresh.thenRunAsync(this.service::refreshBlockList, Util.ioPool());
	}

	public void stopOnlineMode() {
		this.onlineMode = false;
	}

	public boolean isBlocked(final UUID id) {
		if (!this.onlineMode) {
			return false;
		} else {
			this.pendingBlockListRefresh.join();
			return this.service.isBlockedPlayer(id);
		}
	}

	public Set<UUID> getHiddenPlayers() {
		return this.hiddenPlayers;
	}

	public UUID getDiscoveredUUID(final String name) {
		return (UUID)this.discoveredNamesToUUID.getOrDefault(name, Util.NIL_UUID);
	}

	public void addPlayer(final PlayerInfo info) {
		GameProfile gameProfile = info.getProfile();
		this.discoveredNamesToUUID.put(gameProfile.name(), gameProfile.id());
		if (this.minecraft.gui.screen() instanceof SocialInteractionsScreen screen) {
			screen.onAddPlayer(info);
		}
	}

	public void removePlayer(final UUID id) {
		if (this.minecraft.gui.screen() instanceof SocialInteractionsScreen screen) {
			screen.onRemovePlayer(id);
		}
	}
}