package com.mojang.realmsclient.gui.screens;

import com.mojang.blaze3d.pipeline.RenderTarget;
import com.mojang.blaze3d.systems.RenderSystem;
import java.util.Collection;
import java.util.List;
import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.client.gui.GuiGraphics;
import net.minecraft.client.gui.components.Button;
import net.minecraft.client.gui.components.FittingMultiLineTextWidget;
import net.minecraft.client.gui.components.ImageButton;
import net.minecraft.client.gui.components.Tooltip;
import net.minecraft.client.gui.components.WidgetSprites;
import net.minecraft.client.gui.screens.ConfirmLinkScreen;
import net.minecraft.client.gui.screens.Screen;
import net.minecraft.client.renderer.RenderType;
import net.minecraft.network.chat.Component;
import net.minecraft.realms.RealmsScreen;
import net.minecraft.resources.ResourceLocation;
import net.minecraft.server.packs.resources.ResourceManager;
import net.minecraft.util.CommonLinks;
import org.jetbrains.annotations.Nullable;

@Environment(EnvType.CLIENT)
public class AddRealmPopupScreen extends RealmsScreen {
	private static final Component POPUP_TEXT = Component.translatable("mco.selectServer.popup");
	private static final Component CLOSE_TEXT = Component.translatable("mco.selectServer.close");
	private static final ResourceLocation BACKGROUND_SPRITE = ResourceLocation.withDefaultNamespace("popup/background");
	private static final ResourceLocation TRIAL_AVAILABLE_SPRITE = ResourceLocation.withDefaultNamespace("icon/trial_available");
	private static final WidgetSprites CROSS_BUTTON_SPRITES = new WidgetSprites(
		ResourceLocation.withDefaultNamespace("widget/cross_button"), ResourceLocation.withDefaultNamespace("widget/cross_button_highlighted")
	);
	private static final int BG_TEXTURE_WIDTH = 236;
	private static final int BG_TEXTURE_HEIGHT = 34;
	private static final int BG_BORDER_SIZE = 6;
	private static final int IMAGE_WIDTH = 195;
	private static final int IMAGE_HEIGHT = 152;
	private static final int BUTTON_SPACING = 4;
	private static final int PADDING = 10;
	private static final int WIDTH = 320;
	private static final int HEIGHT = 172;
	private static final int TEXT_WIDTH = 100;
	private static final int BUTTON_WIDTH = 99;
	private static final int CAROUSEL_SWITCH_INTERVAL = 100;
	private static List<ResourceLocation> carouselImages = List.of();
	private final Screen backgroundScreen;
	private final boolean trialAvailable;
	@Nullable
	private Button createTrialButton;
	private int carouselIndex;
	private int carouselTick;

	public AddRealmPopupScreen(Screen backgroundScreen, boolean trialAvailable) {
		super(POPUP_TEXT);
		this.backgroundScreen = backgroundScreen;
		this.trialAvailable = trialAvailable;
	}

	public static void updateCarouselImages(ResourceManager resourceManager) {
		Collection<ResourceLocation> collection = resourceManager.listResources(
				"textures/gui/images", resourceLocation -> resourceLocation.getPath().endsWith(".png")
			)
			.keySet();
		carouselImages = collection.stream().filter(resourceLocation -> resourceLocation.getNamespace().equals("realms")).toList();
	}

	@Override
	protected void init() {
		this.backgroundScreen.resize(this.minecraft, this.width, this.height);
		if (this.trialAvailable) {
			this.createTrialButton = this.addRenderableWidget(
				Button.builder(Component.translatable("mco.selectServer.trial"), ConfirmLinkScreen.confirmLink(this, CommonLinks.START_REALMS_TRIAL))
					.bounds(this.right() - 10 - 99, this.bottom() - 10 - 4 - 40, 99, 20)
					.build()
			);
		}

		this.addRenderableWidget(
			Button.builder(Component.translatable("mco.selectServer.buy"), ConfirmLinkScreen.confirmLink(this, CommonLinks.BUY_REALMS))
				.bounds(this.right() - 10 - 99, this.bottom() - 10 - 20, 99, 20)
				.build()
		);
		ImageButton imageButton = this.addRenderableWidget(
			new ImageButton(this.left() + 4, this.top() + 4, 14, 14, CROSS_BUTTON_SPRITES, button -> this.onClose(), CLOSE_TEXT)
		);
		imageButton.setTooltip(Tooltip.create(CLOSE_TEXT));
		int i = 142 - (this.trialAvailable ? 40 : 20);
		FittingMultiLineTextWidget fittingMultiLineTextWidget = new FittingMultiLineTextWidget(
			this.right() - 10 - 100, this.top() + 10, 100, i, POPUP_TEXT, this.font
		);
		if (fittingMultiLineTextWidget.showingScrollBar()) {
			fittingMultiLineTextWidget.setWidth(94);
		}

		this.addRenderableWidget(fittingMultiLineTextWidget);
	}

	@Override
	public void tick() {
		super.tick();
		if (++this.carouselTick > 100) {
			this.carouselTick = 0;
			this.carouselIndex = (this.carouselIndex + 1) % carouselImages.size();
		}
	}

	@Override
	public void render(GuiGraphics guiGraphics, int mouseX, int mouseY, float partialTick) {
		super.render(guiGraphics, mouseX, mouseY, partialTick);
		if (this.createTrialButton != null) {
			renderDiamond(guiGraphics, this.createTrialButton);
		}
	}

	public static void renderDiamond(GuiGraphics guiGraphics, Button button) {
		int i = 8;
		guiGraphics.pose().pushPose();
		guiGraphics.pose().translate(0.0F, 0.0F, 110.0F);
		guiGraphics.blitSprite(
			RenderType::guiTextured, TRIAL_AVAILABLE_SPRITE, button.getX() + button.getWidth() - 8 - 4, button.getY() + button.getHeight() / 2 - 4, 8, 8
		);
		guiGraphics.pose().popPose();
	}

	@Override
	public void renderBackground(GuiGraphics guiGraphics, int mouseX, int mouseY, float partialTick) {
		this.backgroundScreen.render(guiGraphics, -1, -1, partialTick);
		guiGraphics.flush();
		RenderTarget renderTarget = this.minecraft.getMainRenderTarget();
		RenderSystem.getDevice().createCommandEncoder().clearDepthTexture(renderTarget.getDepthTexture(), 1.0);
		this.clearTooltipForNextRenderPass();
		this.renderTransparentBackground(guiGraphics);
		guiGraphics.blitSprite(RenderType::guiTextured, BACKGROUND_SPRITE, this.left(), this.top(), 320, 172);
		if (!carouselImages.isEmpty()) {
			guiGraphics.blit(
				RenderType::guiTextured, (ResourceLocation)carouselImages.get(this.carouselIndex), this.left() + 10, this.top() + 10, 0.0F, 0.0F, 195, 152, 195, 152
			);
		}
	}

	private int left() {
		return (this.width - 320) / 2;
	}

	private int top() {
		return (this.height - 172) / 2;
	}

	private int right() {
		return this.left() + 320;
	}

	private int bottom() {
		return this.top() + 172;
	}

	@Override
	public void onClose() {
		this.minecraft.setScreen(this.backgroundScreen);
	}
}