package com.mojang.realmsclient.gui;

import com.mojang.realmsclient.RealmsMainScreen;
import com.mojang.realmsclient.dto.RealmsServer;
import com.mojang.realmsclient.dto.RealmsWorldOptions;
import com.mojang.realmsclient.util.RealmsTextureManager;
import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.client.Minecraft;
import net.minecraft.client.gui.Font;
import net.minecraft.client.gui.GuiGraphics;
import net.minecraft.client.gui.components.Button;
import net.minecraft.client.gui.components.Tooltip;
import net.minecraft.client.renderer.RenderType;
import net.minecraft.network.chat.CommonComponents;
import net.minecraft.network.chat.Component;
import net.minecraft.network.chat.MutableComponent;
import net.minecraft.resources.ResourceLocation;
import net.minecraft.util.ARGB;
import org.jetbrains.annotations.Nullable;

@Environment(EnvType.CLIENT)
public class RealmsWorldSlotButton extends Button {
	private static final ResourceLocation SLOT_FRAME_SPRITE = ResourceLocation.withDefaultNamespace("widget/slot_frame");
	private static final ResourceLocation CHECKMARK_SPRITE = ResourceLocation.withDefaultNamespace("icon/checkmark");
	public static final ResourceLocation EMPTY_SLOT_LOCATION = ResourceLocation.withDefaultNamespace("textures/gui/realms/empty_frame.png");
	public static final ResourceLocation DEFAULT_WORLD_SLOT_1 = ResourceLocation.withDefaultNamespace("textures/gui/title/background/panorama_0.png");
	public static final ResourceLocation DEFAULT_WORLD_SLOT_2 = ResourceLocation.withDefaultNamespace("textures/gui/title/background/panorama_2.png");
	public static final ResourceLocation DEFAULT_WORLD_SLOT_3 = ResourceLocation.withDefaultNamespace("textures/gui/title/background/panorama_3.png");
	private static final Component SLOT_ACTIVE_TOOLTIP = Component.translatable("mco.configure.world.slot.tooltip.active");
	private static final Component SWITCH_TO_MINIGAME_SLOT_TOOLTIP = Component.translatable("mco.configure.world.slot.tooltip.minigame");
	private static final Component SWITCH_TO_WORLD_SLOT_TOOLTIP = Component.translatable("mco.configure.world.slot.tooltip");
	static final Component MINIGAME = Component.translatable("mco.worldSlot.minigame");
	private static final int WORLD_NAME_MAX_WIDTH = 64;
	private static final String DOTS = "...";
	private final int slotIndex;
	@Nullable
	private RealmsWorldSlotButton.State state;

	public RealmsWorldSlotButton(int x, int y, int width, int height, int slotIndex, Button.OnPress onPress) {
		super(x, y, width, height, CommonComponents.EMPTY, onPress, DEFAULT_NARRATION);
		this.slotIndex = slotIndex;
	}

	@Nullable
	public RealmsWorldSlotButton.State getState() {
		return this.state;
	}

	public void setServerData(RealmsServer serverData) {
		this.state = new RealmsWorldSlotButton.State(serverData, this.slotIndex);
		this.setTooltipAndNarration(this.state, serverData.minigameName);
	}

	private void setTooltipAndNarration(RealmsWorldSlotButton.State state, @Nullable String minigameName) {
		Component component = switch (state.action) {
			case SWITCH_SLOT -> state.minigame ? SWITCH_TO_MINIGAME_SLOT_TOOLTIP : SWITCH_TO_WORLD_SLOT_TOOLTIP;
			case JOIN -> SLOT_ACTIVE_TOOLTIP;
			default -> null;
		};
		if (component != null) {
			this.setTooltip(Tooltip.create(component));
		}

		MutableComponent mutableComponent = Component.literal(state.slotName);
		if (state.minigame && minigameName != null) {
			mutableComponent = mutableComponent.append(CommonComponents.SPACE).append(minigameName);
		}

		this.setMessage(mutableComponent);
	}

	static RealmsWorldSlotButton.Action getAction(RealmsServer realmsServer, boolean isCurrentlyActiveSlot, boolean minigame) {
		if (isCurrentlyActiveSlot && !realmsServer.expired && realmsServer.state != RealmsServer.State.UNINITIALIZED) {
			return RealmsWorldSlotButton.Action.JOIN;
		} else {
			return isCurrentlyActiveSlot || minigame && realmsServer.expired ? RealmsWorldSlotButton.Action.NOTHING : RealmsWorldSlotButton.Action.SWITCH_SLOT;
		}
	}

	@Override
	public void renderWidget(GuiGraphics guiGraphics, int mouseX, int mouseY, float partialTick) {
		if (this.state != null) {
			int i = this.getX();
			int j = this.getY();
			boolean bl = this.isHoveredOrFocused();
			ResourceLocation resourceLocation;
			if (this.state.minigame) {
				resourceLocation = RealmsTextureManager.worldTemplate(String.valueOf(this.state.imageId), this.state.image);
			} else if (this.state.empty) {
				resourceLocation = EMPTY_SLOT_LOCATION;
			} else if (this.state.image != null && this.state.imageId != -1L) {
				resourceLocation = RealmsTextureManager.worldTemplate(String.valueOf(this.state.imageId), this.state.image);
			} else if (this.slotIndex == 1) {
				resourceLocation = DEFAULT_WORLD_SLOT_1;
			} else if (this.slotIndex == 2) {
				resourceLocation = DEFAULT_WORLD_SLOT_2;
			} else if (this.slotIndex == 3) {
				resourceLocation = DEFAULT_WORLD_SLOT_3;
			} else {
				resourceLocation = EMPTY_SLOT_LOCATION;
			}

			int k = -1;
			if (this.state.isCurrentlyActiveSlot) {
				k = ARGB.colorFromFloat(1.0F, 0.56F, 0.56F, 0.56F);
			}

			guiGraphics.blit(RenderType::guiTextured, resourceLocation, i + 3, j + 3, 0.0F, 0.0F, 74, 74, 74, 74, 74, 74, k);
			if (bl && this.state.action != RealmsWorldSlotButton.Action.NOTHING) {
				guiGraphics.blitSprite(RenderType::guiTextured, SLOT_FRAME_SPRITE, i, j, 80, 80);
			} else if (this.state.isCurrentlyActiveSlot) {
				guiGraphics.blitSprite(RenderType::guiTextured, SLOT_FRAME_SPRITE, i, j, 80, 80, ARGB.colorFromFloat(1.0F, 0.8F, 0.8F, 0.8F));
			} else {
				guiGraphics.blitSprite(RenderType::guiTextured, SLOT_FRAME_SPRITE, i, j, 80, 80, ARGB.colorFromFloat(1.0F, 0.56F, 0.56F, 0.56F));
			}

			if (this.state.isCurrentlyActiveSlot) {
				guiGraphics.blitSprite(RenderType::guiTextured, CHECKMARK_SPRITE, i + 67, j + 4, 9, 8);
			}

			if (this.state.hardcore) {
				guiGraphics.blitSprite(RenderType::guiTextured, RealmsMainScreen.HARDCORE_MODE_SPRITE, i + 3, j + 4, 9, 8);
			}

			Font font = Minecraft.getInstance().font;
			String string = this.state.slotName;
			if (font.width(string) > 64) {
				string = font.plainSubstrByWidth(string, 64 - font.width("...")) + "...";
			}

			guiGraphics.drawCenteredString(font, string, i + 40, j + 66, -1);
			guiGraphics.drawCenteredString(
				font, RealmsMainScreen.getVersionComponent(this.state.slotVersion, this.state.compatibility.isCompatible()), i + 40, j + 80 + 2, -1
			);
		}
	}

	@Environment(EnvType.CLIENT)
	public static enum Action {
		NOTHING,
		SWITCH_SLOT,
		JOIN;
	}

	@Environment(EnvType.CLIENT)
	public static class State {
		final boolean isCurrentlyActiveSlot;
		final String slotName;
		final String slotVersion;
		final RealmsServer.Compatibility compatibility;
		final long imageId;
		@Nullable
		final String image;
		public final boolean empty;
		public final boolean minigame;
		public final RealmsWorldSlotButton.Action action;
		public final boolean hardcore;

		public State(RealmsServer server, int slot) {
			this.minigame = slot == 4;
			if (this.minigame) {
				this.isCurrentlyActiveSlot = server.isMinigameActive();
				this.slotName = RealmsWorldSlotButton.MINIGAME.getString();
				this.imageId = server.minigameId;
				this.image = server.minigameImage;
				this.empty = server.minigameId == -1;
				this.slotVersion = "";
				this.compatibility = RealmsServer.Compatibility.UNVERIFIABLE;
				this.hardcore = false;
			} else {
				RealmsWorldOptions realmsWorldOptions = (RealmsWorldOptions)server.slots.get(slot);
				this.isCurrentlyActiveSlot = server.activeSlot == slot && !server.isMinigameActive();
				this.slotName = realmsWorldOptions.getSlotName(slot);
				this.imageId = realmsWorldOptions.templateId;
				this.image = realmsWorldOptions.templateImage;
				this.empty = realmsWorldOptions.empty;
				this.slotVersion = realmsWorldOptions.version;
				this.compatibility = realmsWorldOptions.compatibility;
				this.hardcore = realmsWorldOptions.hardcore;
			}

			this.action = RealmsWorldSlotButton.getAction(server, this.isCurrentlyActiveSlot, this.minigame);
		}
	}
}