package de.hysky.skyblocker.skyblock.dwarven;

import de.hysky.skyblocker.SkyblockerMod;
import de.hysky.skyblocker.config.SkyblockerConfigManager;
import de.hysky.skyblocker.utils.Utils;
import de.hysky.skyblocker.utils.scheduler.Scheduler;
import it.unimi.dsi.fastutil.ints.IntIntPair;
import net.fabricmc.fabric.api.client.command.v2.ClientCommandManager;
import net.fabricmc.fabric.api.client.command.v2.ClientCommandRegistrationCallback;
import net.fabricmc.fabric.api.client.rendering.v1.HudRenderCallback;
import net.minecraft.client.MinecraftClient;
import net.minecraft.client.gui.DrawContext;
import net.minecraft.client.util.math.MatrixStack;
import net.minecraft.util.Identifier;
import net.minecraft.util.math.MathHelper;
import net.minecraft.util.math.RotationAxis;

import java.awt.*;
import java.util.Arrays;
import java.util.Map;

public class CrystalsHud {
    private static final MinecraftClient CLIENT = MinecraftClient.getInstance();
    protected static final Identifier MAP_TEXTURE = new Identifier(SkyblockerMod.NAMESPACE, "textures/gui/crystals_map.png"); 
    private static final Identifier MAP_ICON = new Identifier("textures/map/map_icons.png");
    private static final String[] SMALL_LOCATIONS = { "Fairy Grotto", "King", "Corleone" };

    public static boolean visible = false;

    public static void init() {
        ClientCommandRegistrationCallback.EVENT.register((dispatcher, registryAccess) -> dispatcher.register(ClientCommandManager.literal("skyblocker")
                .then(ClientCommandManager.literal("hud")
                        .then(ClientCommandManager.literal("crystals")
                                .executes(Scheduler.queueOpenScreenCommand(CrystalsHudConfigScreen::new))))));

        HudRenderCallback.EVENT.register((context, tickDelta) -> {
            if (!SkyblockerConfigManager.get().locations.dwarvenMines.crystalsHud.enabled
                    || CLIENT.player == null
                    || !visible) {
                return;
            }
            render(context, tickDelta, SkyblockerConfigManager.get().locations.dwarvenMines.crystalsHud.x,
                    SkyblockerConfigManager.get().locations.dwarvenMines.crystalsHud.y);
        });
    }

    protected static IntIntPair getDimensionsForConfig() {
        int size = (int) (62 * SkyblockerConfigManager.get().locations.dwarvenMines.crystalsHud.mapScaling);
        return IntIntPair.of(size, size);
    }


    /**
     * Renders the map to the players UI. renders the background image ({@link CrystalsHud#MAP_TEXTURE}) of the map then if enabled special locations on the map. then finally the player to the map.
     *
     * @param context DrawContext to draw map to
     * @param tickDelta For interpolating the player's yaw for map marker
     * @param hudX Top left X coordinate of the map
     * @param hudY Top left Y coordinate of the map
     */
    private static void render(DrawContext context, float tickDelta, int hudX, int hudY) {
        float scale = SkyblockerConfigManager.get().locations.dwarvenMines.crystalsHud.mapScaling;

        //make sure the map renders infront of some stuff - improve this in the future with better layering (1.20.5?)
        //and set position and scale
        MatrixStack matrices = context.getMatrices();
        matrices.push();
        matrices.translate(hudX, hudY, 200f);
        matrices.scale(scale, scale, 0f);

        //draw map texture
        context.drawTexture(MAP_TEXTURE, 0, 0, 0, 0, 62, 62, 62, 62);

        //if enabled add waypoint locations to map
        if (SkyblockerConfigManager.get().locations.dwarvenMines.crystalsHud.showLocations) {
            Map<String,CrystalsWaypoint> ActiveWaypoints=  CrystalsLocationsManager.activeWaypoints;

            for (CrystalsWaypoint waypoint : ActiveWaypoints.values()) {
                Color waypointColor = waypoint.category.color;
                IntIntPair renderPos = transformLocation(waypoint.pos.getX(),waypoint.pos.getZ());
                int locationSize = SkyblockerConfigManager.get().locations.dwarvenMines.crystalsHud.locationSize;

                if (Arrays.asList(SMALL_LOCATIONS).contains(waypoint.name.getString())) {//if small location half the location size
                    locationSize = locationSize / 2;
                }

                //fill square of size locationSize around the coordinates of the location
                context.fill(renderPos.firstInt() - locationSize / 2, renderPos.secondInt() - locationSize / 2, renderPos.firstInt() + locationSize / 2, renderPos.secondInt() + locationSize / 2, waypointColor.getRGB());
            }
        }

        //draw player on map
        if (CLIENT.player == null || CLIENT.getNetworkHandler() == null) {
            return;
        }

        //get player location
        double playerX = CLIENT.player.getX();
        double playerZ = CLIENT.player.getZ();
        float playerRotation = CLIENT.player.getYaw(); //TODO make the transitions more rough?
        IntIntPair renderPos = transformLocation(playerX,playerZ);

        int renderX = renderPos.firstInt() - 2;
        int renderY = renderPos.secondInt() - 3;

        //position, scale and rotate the player marker
        matrices.translate(renderX, renderY, 0f);
        matrices.scale(0.75f, 0.75f, 0f);
        matrices.multiply(RotationAxis.POSITIVE_Z.rotationDegrees(playerRotation + 180f), 2, 3, 0);

        //draw marker on map
        context.drawTexture(MAP_ICON, 0, 0, 2, 0, 5, 7, 128, 128);

        //todo add direction (can not work out how to rotate)
        matrices.pop();
    }

    /**
     * Converts an X and Z coordinate in the crystal hollow to a X and Y coordinate on the map.
     *
     * @param x the world X coordinate
     * @param z the world Z coordinate
     * @return the pair of values for x and y
     */
    protected static IntIntPair transformLocation(double x, double z) {
        //converts an x and z to a location on the map
        int transformedX = (int)((x - 202) / 621 * 62);
        int transformedY = (int)((z - 202) / 621 * 62);
        transformedX = MathHelper.clamp(transformedX, 0, 62);
        transformedY = MathHelper.clamp(transformedY, 0, 62);

        return IntIntPair.of(transformedX,transformedY);
    }

    /**
     * Works out if the crystals map should be rendered and sets {@link CrystalsHud#visible} accordingly.
     *
     */
    public static void update() {
        if (CLIENT.player == null || CLIENT.getNetworkHandler() == null || !SkyblockerConfigManager.get().locations.dwarvenMines.crystalsHud.enabled) {
            visible = false;
            return;
        }

        //get if the player is in the crystals
        visible = Utils.isInCrystalHollows();
    }
}