package me.xmrvizzy.skyblocker.skyblock.spidersden;

import com.google.gson.JsonArray;
import com.google.gson.JsonElement;
import com.google.gson.JsonObject;
import com.google.gson.JsonParser;
import me.xmrvizzy.skyblocker.SkyblockerMod;
import me.xmrvizzy.skyblocker.config.SkyblockerConfig;
import me.xmrvizzy.skyblocker.utils.PosUtils;
import me.xmrvizzy.skyblocker.utils.Utils;
import me.xmrvizzy.skyblocker.utils.render.RenderHelper;
import net.fabricmc.fabric.api.client.event.lifecycle.v1.ClientLifecycleEvents;
import net.fabricmc.fabric.api.client.message.v1.ClientReceiveMessageEvents;
import net.fabricmc.fabric.api.client.rendering.v1.WorldRenderContext;
import net.fabricmc.fabric.api.client.rendering.v1.WorldRenderEvents;
import net.minecraft.client.MinecraftClient;
import net.minecraft.entity.player.PlayerEntity;
import net.minecraft.text.Text;
import net.minecraft.util.DyeColor;
import net.minecraft.util.Identifier;
import net.minecraft.util.math.BlockPos;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.*;
import java.util.*;

public class Relics {
    private static final Logger LOGGER = LoggerFactory.getLogger(Relics.class);
    private static int totalRelics = 0;
    private static final List<BlockPos> relics = new ArrayList<>();
    private static final Map<String, Set<BlockPos>> foundRelics = new HashMap<>();

    public static void init() {
        ClientLifecycleEvents.CLIENT_STARTED.register(Relics::onClientStarted);
        ClientLifecycleEvents.CLIENT_STOPPING.register(Relics::saveFoundRelics);
        ClientReceiveMessageEvents.GAME.register(Relics::onChatMessage);
        WorldRenderEvents.AFTER_TRANSLUCENT.register(Relics::render);
    }

    private static void onClientStarted(MinecraftClient client) {
        try (BufferedReader reader = client.getResourceManager().openAsReader(new Identifier(SkyblockerMod.NAMESPACE, "spidersden/relics.json"))) {
            for (Map.Entry<String, JsonElement> json : JsonParser.parseReader(reader).getAsJsonObject().asMap().entrySet()) {
                if (json.getKey().equals("total")) {
                    totalRelics = json.getValue().getAsInt();
                } else if (json.getKey().equals("locations")) {
                    for (JsonElement locationJson : json.getValue().getAsJsonArray().asList()) {
                        JsonObject posData = locationJson.getAsJsonObject();
                        relics.add(new BlockPos(posData.get("x").getAsInt(), posData.get("y").getAsInt(), posData.get("z").getAsInt()));
                    }
                }
            }
            LOGGER.info("[Skyblocker] Loaded relics locations");
        } catch (IOException e) {
            LOGGER.error("[Skyblocker] Failed to load relics locations", e);
        }

        try (BufferedReader reader = new BufferedReader(new FileReader(SkyblockerMod.CONFIG_DIR.resolve("found_relics.json").toFile()))) {
            for (Map.Entry<String, JsonElement> profileJson : JsonParser.parseReader(reader).getAsJsonObject().asMap().entrySet()) {
                Set<BlockPos> foundRelicsForProfile = new HashSet<>();
                for (JsonElement foundRelicsJson : profileJson.getValue().getAsJsonArray().asList()) {
                    foundRelicsForProfile.add(PosUtils.parsePosString(foundRelicsJson.getAsString()));
                }
                foundRelics.put(profileJson.getKey(), foundRelicsForProfile);
            }
            LOGGER.debug("[Skyblocker] Loaded found relics");
        } catch (FileNotFoundException ignored) {
        } catch (IOException e) {
            LOGGER.error("[Skyblocker] Failed to load found relics", e);
        }
    }

    private static void saveFoundRelics(MinecraftClient client) {
        try (BufferedWriter writer = new BufferedWriter(new FileWriter(SkyblockerMod.CONFIG_DIR.resolve("found_relics.json").toFile()))) {
            JsonObject json = new JsonObject();
            for (Map.Entry<String, Set<BlockPos>> foundRelicsForProfile : foundRelics.entrySet()) {
                JsonArray foundRelicsJson = new JsonArray();
                for (BlockPos foundRelic : foundRelicsForProfile.getValue()) {
                    foundRelicsJson.add(PosUtils.getPosString(foundRelic));
                }
                json.add(foundRelicsForProfile.getKey(), foundRelicsJson);
            }
            SkyblockerMod.GSON.toJson(json, writer);
            LOGGER.debug("[Skyblocker] Saved found relics");
        } catch (IOException e) {
            LOGGER.error("[Skyblocker] Failed to write found relics to file", e);
        }
    }

    private static void onChatMessage(Text text, boolean overlay) {
        String message = text.getString();
        if (message.equals("You've already found this relic!") || message.startsWith("+10,000 Coins! (") && message.endsWith("/28 Relics)")) {
            markClosestRelicFound();
        }
    }

    private static void render(WorldRenderContext context) {
        SkyblockerConfig.Relics config = SkyblockerConfig.get().locations.spidersDen.relics;

        if (config.enableRelicsHelper && Utils.getLocationRaw().equals("combat_1")) {
            for (BlockPos fairySoulPos : relics) {
                boolean isRelicFound = isRelicFound(fairySoulPos);
                if (isRelicFound && !config.highlightFoundRelics) continue;
                float[] colorComponents = isRelicFound ? DyeColor.BROWN.getColorComponents() : DyeColor.YELLOW.getColorComponents();
                RenderHelper.renderFilledThroughWallsWithBeaconBeam(context, fairySoulPos, colorComponents, 0.5F);
            }
        }
    }

    private static boolean isRelicFound(BlockPos pos) {
        Set<BlockPos> foundRelicsForProfile = foundRelics.get(Utils.getProfile());
        return (foundRelicsForProfile == null) ? false : foundRelicsForProfile.contains(pos);
    }

    private static void markClosestRelicFound() {
        PlayerEntity player = MinecraftClient.getInstance().player;
        if (player == null) {
            LOGGER.warn("[Skyblocker] Failed to mark closest relic as found because player is null");
            return;
        }
        relics.stream()
            .filter((relicPos) -> !Relics.isRelicFound(relicPos))
            .min(Comparator.comparingDouble(relicPos -> relicPos.getSquaredDistance(player.getPos())))
            .filter(relicPos -> relicPos.getSquaredDistance(player.getPos()) <= 16)
            .ifPresent(relicPos -> {
                foundRelics.computeIfAbsent(Utils.getProfile(), profileKey -> new HashSet<>());
                foundRelics.get(Utils.getProfile()).add(relicPos);
            });
    }
}