/*
 * Copyright (C) 2022-2023 NotEnoughUpdates contributors
 *
 * This file is part of NotEnoughUpdates.
 *
 * NotEnoughUpdates is free software: you can redistribute it
 * and/or modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation, either
 * version 3 of the License, or (at your option) any later version.
 *
 * NotEnoughUpdates is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with NotEnoughUpdates. If not, see <https://www.gnu.org/licenses/>.
 */

package io.github.moulberry.notenoughupdates.miscgui.minionhelper.loaders;

import com.google.gson.JsonArray;
import com.google.gson.JsonElement;
import com.google.gson.JsonObject;
import io.github.moulberry.notenoughupdates.core.util.StringUtils;
import io.github.moulberry.notenoughupdates.events.ProfileDataLoadedEvent;
import io.github.moulberry.notenoughupdates.miscgui.minionhelper.ApiData;
import io.github.moulberry.notenoughupdates.miscgui.minionhelper.MinionHelperManager;
import io.github.moulberry.notenoughupdates.profileviewer.ProfileViewer;
import io.github.moulberry.notenoughupdates.util.Constants;
import io.github.moulberry.notenoughupdates.util.SBInfo;
import io.github.moulberry.notenoughupdates.util.Utils;
import lombok.Getter;
import net.minecraft.client.Minecraft;
import net.minecraft.client.entity.EntityPlayerSP;
import net.minecraft.item.ItemStack;
import net.minecraftforge.fml.common.eventhandler.SubscribeEvent;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;

public class MinionHelperApiLoader {
	private final MinionHelperManager manager;
	private boolean collectionApiEnabled = true;
	@Getter
	private boolean readyToUse = false;
	@Getter
	private ApiData apiData = null;
	@Getter
	private boolean notifyNoCollectionApi = false;
	@Getter
	private boolean invalidApiKey = true;

	public MinionHelperApiLoader(MinionHelperManager manager) {
		this.manager = manager;
	}

	private String getUuid() {
		EntityPlayerSP thePlayer = Minecraft.getMinecraft().thePlayer;
		if (thePlayer == null) return null;

		String debugPlayerUuid = manager.getDebugPlayerUuid();
		if (debugPlayerUuid != null) return debugPlayerUuid;

		return thePlayer.getUniqueID().toString().replace("-", "");
	}

	@SubscribeEvent
	public void onApiDataLoaded(ProfileDataLoadedEvent event) {
		JsonObject data = event.getData();
		if (data == null) {
			invalidApiKey = true;
			return;
		}
		invalidApiKey = false;

		if (!data.has("success") || !data.get("success").getAsBoolean()) return;
		JsonArray profiles = data.getAsJsonArray("profiles");
		for (JsonElement element : profiles) {
			JsonObject profile = element.getAsJsonObject();
			String profileName = profile.get("cute_name").getAsString();
			JsonObject members = profile.getAsJsonObject("members");
			JsonObject player = members.getAsJsonObject(getUuid());

			String debugProfileName = manager.getDebugProfileName();
			String currentProfile = debugProfileName != null ? debugProfileName : SBInfo.getInstance().currentProfile;

			if (profileName.equals(currentProfile)) {
				readData(player, members);
				return;
			}
		}
	}

	private void readData(JsonObject player, JsonObject members) {
		int magesReputation = 0;
		int barbariansReputation = 0;
		if (player.has("nether_island_player_data")) {
			JsonObject netherData = player.getAsJsonObject("nether_island_player_data");
			if (netherData.has("mages_reputation")) {
				magesReputation = netherData.get("mages_reputation").getAsInt();
			}
			if (netherData.has("barbarians_reputation")) {
				barbariansReputation = netherData.get("barbarians_reputation").getAsInt();
			}
		}

		apiData = new ApiData(
			getCollections(player),
			getSlayers(player),
			magesReputation,
			barbariansReputation,
			!collectionApiEnabled,
			loadCraftedMinions(members),
			loadPeltCount(player)
		);

		manager.reloadData();
		readyToUse = true;
	}

	private int loadPeltCount(JsonObject player) {
		int localPelts = manager.getLocalPelts();
		if (localPelts != -1) return localPelts;

		return Utils.getElementAsInt(Utils.getElement(player, "quests.trapper_quest.pelt_count"), 0);
	}

	private Map<String, Integer> getSlayers(JsonObject player) {
		JsonObject slayerLeveling = Constants.LEVELING.getAsJsonObject("slayer_xp");

		Map<String, Integer> slayerTier = new HashMap<>();
		JsonObject slayerBosses = Utils
			.getElementOrDefault(player, "slayer.slayer_bosses", new JsonObject())
			.getAsJsonObject();
		for (Map.Entry<String, JsonElement> entry : slayerBosses.entrySet()) {
			String name = entry.getKey();
			JsonObject slayerEntry = entry.getValue().getAsJsonObject();
			if (slayerEntry.has("xp")) {
				long xp = slayerEntry.get("xp").getAsLong();

				int tier = 0;
				for (JsonElement element : slayerLeveling.getAsJsonArray(name)) {
					int needForLevel = element.getAsInt();
					if (xp >= needForLevel) {
						tier++;
					} else {
						break;
					}
				}
				slayerTier.put(name, tier);
			}
		}
		return slayerTier;
	}

	private Map<String, Integer> getCollections(JsonObject player) {
		Map<String, Integer> highestCollectionTier = new HashMap<>();

		JsonElement collectionTiers = Utils.getElement(player, "player_data.unlocked_coll_tiers");
		if (collectionTiers != null) {
			for (JsonElement element : collectionTiers.getAsJsonArray()) {
				String text = element.getAsString();
				String[] split = text.split("_");
				int level = Integer.parseInt(split[split.length - 1]);
				String name = StringUtils.removeLastWord(text, "_");

				//Because skyblock is good in naming things
				LinkedHashMap<String, ItemStack> collectionMap = ProfileViewer.getCollectionToCollectionDisplayMap();
				if (collectionMap.containsKey(name)) {
					ItemStack itemStack = collectionMap.get(name);
					String displayName = itemStack.getDisplayName();
					name = Utils.cleanColour(displayName);
					name = manager.formatInternalName(name);
				} else {
					//Doing this since there is no space in the profile viewer gui for more entries in collectionToCollectionDisplayMap
					if (name.equals("SAND:1")) name = "RED_SAND";
					if (name.equals("MYCEL")) name = "MYCELIUM";
				}

				level = Math.max(highestCollectionTier.getOrDefault(name, 0), level);
				highestCollectionTier.put(name, level);
			}
			if (!collectionApiEnabled) {
				Utils.addChatMessage("§e[NEU Minion Helper] Collection API detected!");
			}
			collectionApiEnabled = true;
		} else {
			if (collectionApiEnabled) {
				notifyNoCollectionApi = true;
			}
			collectionApiEnabled = false;
		}
		return highestCollectionTier;
	}

	private List<String> loadCraftedMinions(JsonObject members) {
		List<String> craftedMinions = new ArrayList<>();
		for (Map.Entry<String, JsonElement> entry : members.entrySet()) {
			JsonObject value = entry.getValue().getAsJsonObject();
			for (JsonElement e : Utils
				.getElementOrDefault(value, "player_data.crafted_generators", new JsonArray())
				.getAsJsonArray()) {
					String rawGenerator = e.getAsString();
					String[] split = rawGenerator.split("_");
					String tier = split[split.length - 1];
					String name = rawGenerator.substring(0, rawGenerator.length() - tier.length() - 1);
					String internalName = name + "_GENERATOR_" + tier;
					craftedMinions.add(internalName);
				}
		}

		return craftedMinions;
	}

	public void setDirty() {
		readyToUse = false;
	}

	public void prepareProfileSwitch() {
		readyToUse = false;
	}

	public void onProfileSwitch() {
		apiData = null;
		collectionApiEnabled = true;
	}

	public boolean isCollectionApiDisabled() {
		return apiData != null && apiData.isCollectionApiDisabled();
	}

	public void resetData() {
		apiData = null;
	}

	public void setNotifyNoCollectionApi(boolean notifyNoCollectionApi) {
		this.notifyNoCollectionApi = notifyNoCollectionApi;
	}

}