package net.minecraft.client.gui.components;

import com.google.common.collect.Maps;
import java.util.Map;
import java.util.Objects;
import java.util.UUID;
import net.minecraft.client.Minecraft;
import net.minecraft.client.gui.GuiGraphicsExtractor;
import net.minecraft.client.renderer.RenderPipelines;
import net.minecraft.network.chat.Component;
import net.minecraft.network.protocol.game.ClientboundBossEventPacket;
import net.minecraft.network.protocol.game.ClientboundBossEventPacket.Handler;
import net.minecraft.resources.Identifier;
import net.minecraft.util.Mth;
import net.minecraft.util.profiling.Profiler;
import net.minecraft.util.profiling.ProfilerFiller;
import net.minecraft.world.BossEvent;
import net.minecraft.world.BossEvent.BossBarColor;
import net.minecraft.world.BossEvent.BossBarOverlay;

public class BossHealthOverlay {
	private static final int BAR_WIDTH = 182;
	private static final int BAR_HEIGHT = 5;
	private static final Identifier[] BAR_BACKGROUND_SPRITES = new Identifier[]{
		Identifier.withDefaultNamespace("boss_bar/pink_background"),
		Identifier.withDefaultNamespace("boss_bar/blue_background"),
		Identifier.withDefaultNamespace("boss_bar/red_background"),
		Identifier.withDefaultNamespace("boss_bar/green_background"),
		Identifier.withDefaultNamespace("boss_bar/yellow_background"),
		Identifier.withDefaultNamespace("boss_bar/purple_background"),
		Identifier.withDefaultNamespace("boss_bar/white_background")
	};
	private static final Identifier[] BAR_PROGRESS_SPRITES = new Identifier[]{
		Identifier.withDefaultNamespace("boss_bar/pink_progress"),
		Identifier.withDefaultNamespace("boss_bar/blue_progress"),
		Identifier.withDefaultNamespace("boss_bar/red_progress"),
		Identifier.withDefaultNamespace("boss_bar/green_progress"),
		Identifier.withDefaultNamespace("boss_bar/yellow_progress"),
		Identifier.withDefaultNamespace("boss_bar/purple_progress"),
		Identifier.withDefaultNamespace("boss_bar/white_progress")
	};
	private static final Identifier[] OVERLAY_BACKGROUND_SPRITES = new Identifier[]{
		Identifier.withDefaultNamespace("boss_bar/notched_6_background"),
		Identifier.withDefaultNamespace("boss_bar/notched_10_background"),
		Identifier.withDefaultNamespace("boss_bar/notched_12_background"),
		Identifier.withDefaultNamespace("boss_bar/notched_20_background")
	};
	private static final Identifier[] OVERLAY_PROGRESS_SPRITES = new Identifier[]{
		Identifier.withDefaultNamespace("boss_bar/notched_6_progress"),
		Identifier.withDefaultNamespace("boss_bar/notched_10_progress"),
		Identifier.withDefaultNamespace("boss_bar/notched_12_progress"),
		Identifier.withDefaultNamespace("boss_bar/notched_20_progress")
	};
	private final Minecraft minecraft;
	private final Map<UUID, LerpingBossEvent> events = Maps.<UUID, LerpingBossEvent>newLinkedHashMap();

	public BossHealthOverlay(final Minecraft minecraft) {
		this.minecraft = minecraft;
	}

	public void extractRenderState(final GuiGraphicsExtractor graphics) {
		if (!this.events.isEmpty()) {
			graphics.nextStratum();
			ProfilerFiller profiler = Profiler.get();
			profiler.push("bossHealth");
			int screenWidth = graphics.guiWidth();
			int yOffset = 12;

			for (LerpingBossEvent event : this.events.values()) {
				int xLeft = screenWidth / 2 - 91;
				this.extractBar(graphics, xLeft, yOffset, event);
				Component msg = event.getName();
				int width = this.minecraft.font.width(msg);
				int x = screenWidth / 2 - width / 2;
				int y = yOffset - 9;
				graphics.text(this.minecraft.font, msg, x, y, -1);
				yOffset += 10 + 9;
				if (yOffset >= graphics.guiHeight() / 3) {
					break;
				}
			}

			profiler.pop();
		}
	}

	private void extractBar(final GuiGraphicsExtractor graphics, final int x, final int y, final BossEvent event) {
		this.extractBar(graphics, x, y, event, 182, BAR_BACKGROUND_SPRITES, OVERLAY_BACKGROUND_SPRITES);
		int width = Mth.lerpDiscrete(event.getProgress(), 0, 182);
		if (width > 0) {
			this.extractBar(graphics, x, y, event, width, BAR_PROGRESS_SPRITES, OVERLAY_PROGRESS_SPRITES);
		}
	}

	private void extractBar(
		final GuiGraphicsExtractor graphics,
		final int x,
		final int y,
		final BossEvent event,
		final int width,
		final Identifier[] sprites,
		final Identifier[] overlaySprites
	) {
		graphics.blitSprite(RenderPipelines.GUI_TEXTURED, sprites[event.getColor().ordinal()], 182, 5, 0, 0, x, y, width, 5);
		if (event.getOverlay() != BossBarOverlay.PROGRESS) {
			graphics.blitSprite(RenderPipelines.GUI_TEXTURED, overlaySprites[event.getOverlay().ordinal() - 1], 182, 5, 0, 0, x, y, width, 5);
		}
	}

	public void update(final ClientboundBossEventPacket packet) {
		packet.dispatch(
			new Handler() {
				{
					Objects.requireNonNull(BossHealthOverlay.this);
				}

				public void add(
					final UUID id,
					final Component name,
					final float progress,
					final BossBarColor color,
					final BossBarOverlay overlay,
					final boolean darkenScreen,
					final boolean playMusic,
					final boolean createWorldFog
				) {
					BossHealthOverlay.this.events.put(id, new LerpingBossEvent(id, name, progress, color, overlay, darkenScreen, playMusic, createWorldFog));
				}

				public void remove(final UUID id) {
					BossHealthOverlay.this.events.remove(id);
				}

				public void updateProgress(final UUID id, final float progress) {
					((LerpingBossEvent)BossHealthOverlay.this.events.get(id)).setProgress(progress);
				}

				public void updateName(final UUID id, final Component name) {
					((LerpingBossEvent)BossHealthOverlay.this.events.get(id)).setName(name);
				}

				public void updateStyle(final UUID id, final BossBarColor color, final BossBarOverlay overlay) {
					LerpingBossEvent event = (LerpingBossEvent)BossHealthOverlay.this.events.get(id);
					event.setColor(color);
					event.setOverlay(overlay);
				}

				public void updateProperties(final UUID id, final boolean darkenScreen, final boolean playMusic, final boolean createWorldFog) {
					LerpingBossEvent event = (LerpingBossEvent)BossHealthOverlay.this.events.get(id);
					event.setDarkenScreen(darkenScreen);
					event.setPlayBossMusic(playMusic);
					event.setCreateWorldFog(createWorldFog);
				}
			}
		);
	}

	public void reset() {
		this.events.clear();
	}

	public boolean shouldPlayMusic() {
		if (!this.events.isEmpty()) {
			for (BossEvent event : this.events.values()) {
				if (event.shouldPlayBossMusic()) {
					return true;
				}
			}
		}

		return false;
	}

	public boolean shouldDarkenScreen() {
		if (!this.events.isEmpty()) {
			for (BossEvent event : this.events.values()) {
				if (event.shouldDarkenScreen()) {
					return true;
				}
			}
		}

		return false;
	}

	public boolean shouldCreateWorldFog() {
		if (!this.events.isEmpty()) {
			for (BossEvent event : this.events.values()) {
				if (event.shouldCreateWorldFog()) {
					return true;
				}
			}
		}

		return false;
	}
}