package net.minecraft.client.gui.screens;

import it.unimi.dsi.fastutil.objects.Object2IntMap;
import it.unimi.dsi.fastutil.objects.Object2IntOpenHashMap;
import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.client.GameNarrator;
import net.minecraft.client.Minecraft;
import net.minecraft.client.gui.GuiGraphicsExtractor;
import net.minecraft.client.gui.components.debug.DebugScreenEntries;
import net.minecraft.client.gui.narration.NarratedElementType;
import net.minecraft.client.gui.narration.NarrationElementOutput;
import net.minecraft.client.gui.render.TextureSetup;
import net.minecraft.client.multiplayer.LevelLoadTracker;
import net.minecraft.client.renderer.RenderPipelines;
import net.minecraft.client.renderer.blockentity.AbstractEndPortalRenderer;
import net.minecraft.client.renderer.texture.AbstractTexture;
import net.minecraft.client.renderer.texture.TextureAtlasSprite;
import net.minecraft.client.renderer.texture.TextureManager;
import net.minecraft.network.chat.Component;
import net.minecraft.server.level.progress.ChunkLoadStatusView;
import net.minecraft.util.ARGB;
import net.minecraft.util.Mth;
import net.minecraft.util.Util;
import net.minecraft.world.level.block.Blocks;
import net.minecraft.world.level.chunk.status.ChunkStatus;
import org.jspecify.annotations.Nullable;

@Environment(EnvType.CLIENT)
public class LevelLoadingScreen extends Screen {
	private static final Component DOWNLOADING_TERRAIN_TEXT = Component.translatable("multiplayer.downloadingTerrain");
	private static final Component READY_TO_PLAY_TEXT = Component.translatable("narrator.ready_to_play");
	private static final long NARRATION_DELAY_MS = 2000L;
	private static final int PROGRESS_BAR_WIDTH = 200;
	private LevelLoadTracker loadTracker;
	private float smoothedProgress;
	private long lastNarration = -1L;
	private LevelLoadingScreen.Reason reason;
	@Nullable
	private TextureAtlasSprite cachedNetherPortalSprite;
	private static final Object2IntMap<ChunkStatus> COLORS = Util.make(new Object2IntOpenHashMap<>(), map -> {
		map.defaultReturnValue(0);
		map.put(ChunkStatus.EMPTY, 5526612);
		map.put(ChunkStatus.STRUCTURE_STARTS, 10066329);
		map.put(ChunkStatus.STRUCTURE_REFERENCES, 6250897);
		map.put(ChunkStatus.BIOMES, 8434258);
		map.put(ChunkStatus.NOISE, 13750737);
		map.put(ChunkStatus.SURFACE, 7497737);
		map.put(ChunkStatus.CARVERS, 3159410);
		map.put(ChunkStatus.FEATURES, 2213376);
		map.put(ChunkStatus.INITIALIZE_LIGHT, 13421772);
		map.put(ChunkStatus.LIGHT, 16769184);
		map.put(ChunkStatus.SPAWN, 15884384);
		map.put(ChunkStatus.FULL, 16777215);
	});

	public LevelLoadingScreen(final LevelLoadTracker loadTracker, final LevelLoadingScreen.Reason reason) {
		super(GameNarrator.NO_TITLE);
		this.loadTracker = loadTracker;
		this.reason = reason;
	}

	public void update(final LevelLoadTracker loadTracker, final LevelLoadingScreen.Reason reason) {
		this.loadTracker = loadTracker;
		this.reason = reason;
	}

	@Override
	public boolean shouldCloseOnEsc() {
		return false;
	}

	@Override
	protected boolean shouldNarrateNavigation() {
		return false;
	}

	@Override
	protected void updateNarratedWidget(final NarrationElementOutput output) {
		if (this.loadTracker.hasProgress()) {
			output.add(NarratedElementType.TITLE, Component.translatable("loading.progress", Mth.floor(this.loadTracker.serverProgress() * 100.0F)));
		}
	}

	@Override
	public void tick() {
		super.tick();
		this.smoothedProgress = this.smoothedProgress + (this.loadTracker.serverProgress() - this.smoothedProgress) * 0.2F;
		if (this.loadTracker.isLevelReady()) {
			this.onClose();
		}
	}

	@Override
	public void extractRenderState(final GuiGraphicsExtractor graphics, final int mouseX, final int mouseY, final float a) {
		super.extractRenderState(graphics, mouseX, mouseY, a);
		long current = Util.getMillis();
		if (current - this.lastNarration > 2000L) {
			this.lastNarration = current;
			this.triggerImmediateNarration(true);
		}

		int xCenter = this.width / 2;
		int yCenter = this.height / 2;
		ChunkLoadStatusView statusView = this.loadTracker.statusView();
		int textTop;
		if (statusView != null) {
			int size = 2;
			extractChunksForRendering(graphics, xCenter, yCenter, 2, 0, statusView);
			textTop = yCenter - statusView.radius() * 2 - 9 * 3;
		} else {
			textTop = yCenter - 50;
		}

		graphics.centeredText(this.font, DOWNLOADING_TERRAIN_TEXT, xCenter, textTop, -1);
		if (this.loadTracker.hasProgress()) {
			this.drawProgressBar(graphics, xCenter - 100, textTop + 9 + 3, 200, 2, this.smoothedProgress);
		}
	}

	private void drawProgressBar(final GuiGraphicsExtractor graphics, final int left, final int top, final int width, final int height, final float progress) {
		graphics.fill(left, top, left + width, top + height, -16777216);
		graphics.fill(left, top, left + Math.round(progress * width), top + height, -16711936);
	}

	public static void extractChunksForRendering(
		final GuiGraphicsExtractor graphics, final int xCenter, final int yCenter, final int size, final int margin, final ChunkLoadStatusView statusView
	) {
		int width = size + margin;
		int diameter = statusView.radius() * 2 + 1;
		int totalWidth = diameter * width - margin;
		int xStart = xCenter - totalWidth / 2;
		int yStart = yCenter - totalWidth / 2;
		if (Minecraft.getInstance().debugEntries.isCurrentlyEnabled(DebugScreenEntries.VISUALIZE_CHUNKS_ON_SERVER)) {
			int centerWidth = width / 2 + 1;
			graphics.fill(xCenter - centerWidth, yCenter - centerWidth, xCenter + centerWidth, yCenter + centerWidth, -65536);
		}

		for (int x = 0; x < diameter; x++) {
			for (int z = 0; z < diameter; z++) {
				ChunkStatus status = statusView.get(x, z);
				int xCellStart = xStart + x * width;
				int yCellStart = yStart + z * width;
				graphics.fill(xCellStart, yCellStart, xCellStart + size, yCellStart + size, ARGB.opaque(COLORS.getInt(status)));
			}
		}
	}

	@Override
	public void extractBackground(final GuiGraphicsExtractor graphics, final int mouseX, final int mouseY, final float a) {
		switch (this.reason) {
			case NETHER_PORTAL:
				graphics.blitSprite(RenderPipelines.GUI_OPAQUE_TEXTURED_BACKGROUND, this.getNetherPortalSprite(), 0, 0, graphics.guiWidth(), graphics.guiHeight());
				break;
			case END_PORTAL:
				TextureManager textureManager = Minecraft.getInstance().getTextureManager();
				AbstractTexture skyTexture = textureManager.getTexture(AbstractEndPortalRenderer.END_SKY_LOCATION);
				AbstractTexture portalTexture = textureManager.getTexture(AbstractEndPortalRenderer.END_PORTAL_LOCATION);
				TextureSetup textureSetup = TextureSetup.doubleTexture(
					skyTexture.getTextureView(), skyTexture.getSampler(), portalTexture.getTextureView(), portalTexture.getSampler()
				);
				graphics.fill(RenderPipelines.END_PORTAL, textureSetup, 0, 0, this.width, this.height);
				break;
			case OTHER:
				this.extractPanorama(graphics, a);
				this.extractBlurredBackground(graphics);
				this.extractMenuBackground(graphics);
		}
	}

	private TextureAtlasSprite getNetherPortalSprite() {
		if (this.cachedNetherPortalSprite != null) {
			return this.cachedNetherPortalSprite;
		} else {
			this.cachedNetherPortalSprite = this.minecraft
				.getModelManager()
				.getBlockStateModelSet()
				.getParticleMaterial(Blocks.NETHER_PORTAL.defaultBlockState())
				.sprite();
			return this.cachedNetherPortalSprite;
		}
	}

	@Override
	public void onClose() {
		this.minecraft.getNarrator().saySystemNow(READY_TO_PLAY_TEXT);
		super.onClose();
	}

	@Override
	public boolean isPauseScreen() {
		return false;
	}

	@Environment(EnvType.CLIENT)
	public static enum Reason {
		NETHER_PORTAL,
		END_PORTAL,
		OTHER;
	}
}