package net.minecraft.client.gui.components.toasts;

import com.google.common.collect.ImmutableList;
import java.util.List;
import java.util.stream.Stream;
import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.ChatFormatting;
import net.minecraft.client.Minecraft;
import net.minecraft.client.gui.Font;
import net.minecraft.client.gui.GuiGraphicsExtractor;
import net.minecraft.client.renderer.RenderPipelines;
import net.minecraft.network.chat.Component;
import net.minecraft.resources.Identifier;
import net.minecraft.util.FormattedCharSequence;
import net.minecraft.world.level.ChunkPos;
import org.jspecify.annotations.Nullable;

@Environment(EnvType.CLIENT)
public class SystemToast implements Toast {
	private static final Identifier BACKGROUND_SPRITE = Identifier.withDefaultNamespace("toast/system");
	private static final int MAX_LINE_SIZE = 200;
	private static final int LINE_SPACING = 12;
	private static final int MARGIN = 10;
	private static final int TEXT_X_START = 18;
	private final SystemToast.SystemToastId id;
	private List<FormattedCharSequence> titleLines;
	private List<FormattedCharSequence> messageLines;
	private long lastChanged;
	private boolean changed;
	private int width;
	private boolean forceHide;
	private Toast.Visibility wantedVisibility = Toast.Visibility.HIDE;

	public SystemToast(final SystemToast.SystemToastId id, final Component title, @Nullable final Component message) {
		this.id = id;
		this.update(title, message);
	}

	private static List<FormattedCharSequence> nullToEmpty(@Nullable final Component text) {
		return (List<FormattedCharSequence>)(text == null ? ImmutableList.of() : splitToLength(text));
	}

	private static List<FormattedCharSequence> splitToLength(final Component text) {
		return Minecraft.getInstance().font.split(text, 200);
	}

	@Override
	public int width() {
		return this.width;
	}

	@Override
	public int height() {
		int titleHeight = (this.titleLines.size() - 1) * 12;
		int messageHeight = Math.max(this.messageLines.size(), 1) * 12;
		return 20 + titleHeight + messageHeight;
	}

	public void forceHide() {
		this.forceHide = true;
	}

	@Override
	public Toast.Visibility getWantedVisibility() {
		return this.wantedVisibility;
	}

	@Override
	public void update(final ToastManager manager, final long fullyVisibleForMs) {
		if (this.changed) {
			this.lastChanged = fullyVisibleForMs;
			this.changed = false;
		}

		double timeToDisplayUpdate = this.id.displayTime * manager.getNotificationDisplayTimeMultiplier();
		long timeSinceUpdate = fullyVisibleForMs - this.lastChanged;
		this.wantedVisibility = !this.forceHide && timeSinceUpdate < timeToDisplayUpdate ? Toast.Visibility.SHOW : Toast.Visibility.HIDE;
	}

	@Override
	public void extractRenderState(final GuiGraphicsExtractor graphics, final Font font, final long fullyVisibleForMs) {
		graphics.blitSprite(RenderPipelines.GUI_TEXTURED, BACKGROUND_SPRITE, 0, 0, this.width(), this.height());
		if (this.messageLines.isEmpty()) {
			this.extractTextLines(graphics, font, this.titleLines, 12, -256);
		} else {
			this.extractTextLines(graphics, font, this.titleLines, 7, -256);
			this.extractTextLines(graphics, font, this.messageLines, 7 + this.titleLines.size() * 12, -1);
		}
	}

	private void extractTextLines(
		final GuiGraphicsExtractor graphics, final Font font, final List<FormattedCharSequence> textLines, final int yStart, final int textColor
	) {
		for (int i = 0; i < textLines.size(); i++) {
			graphics.text(font, (FormattedCharSequence)textLines.get(i), 18, yStart + i * 12, textColor, false);
		}
	}

	public void reset(final Component title, @Nullable final Component message) {
		this.update(title, message);
		this.changed = true;
	}

	private void update(final Component title, @Nullable final Component message) {
		this.titleLines = splitToLength(title);
		this.messageLines = nullToEmpty(message);
		this.recalculateWidth();
	}

	public void recalculateWidth() {
		int width = Math.max(160, Stream.concat(this.titleLines.stream(), this.messageLines.stream()).mapToInt(Minecraft.getInstance().font::width).max().orElse(200));
		this.width = width + 30;
	}

	public SystemToast.SystemToastId getToken() {
		return this.id;
	}

	public static void add(final ToastManager toastManager, final SystemToast.SystemToastId id, final Component title, @Nullable final Component message) {
		toastManager.addToast(new SystemToast(id, title, message));
	}

	public static void addOrUpdate(final ToastManager toastManager, final SystemToast.SystemToastId id, final Component title, @Nullable final Component message) {
		SystemToast toast = toastManager.getToast(SystemToast.class, id);
		if (toast == null) {
			add(toastManager, id, title, message);
		} else {
			toast.reset(title, message);
		}
	}

	public static void forceHide(final ToastManager toastManager, final SystemToast.SystemToastId id) {
		SystemToast toast = toastManager.getToast(SystemToast.class, id);
		if (toast != null) {
			toast.forceHide();
		}
	}

	public static void onWorldAccessFailure(final Minecraft minecraft, final String levelId) {
		add(
			minecraft.gui.toastManager(),
			SystemToast.SystemToastId.WORLD_ACCESS_FAILURE,
			Component.translatable("selectWorld.access_failure"),
			Component.literal(levelId)
		);
	}

	public static void onWorldDeleteFailure(final Minecraft minecraft, final String levelId) {
		add(
			minecraft.gui.toastManager(),
			SystemToast.SystemToastId.WORLD_ACCESS_FAILURE,
			Component.translatable("selectWorld.delete_failure"),
			Component.literal(levelId)
		);
	}

	public static void onPackCopyFailure(final Minecraft minecraft, final String extraInfo) {
		add(minecraft.gui.toastManager(), SystemToast.SystemToastId.PACK_COPY_FAILURE, Component.translatable("pack.copyFailure"), Component.literal(extraInfo));
	}

	public static void onFileDropFailure(final Minecraft minecraft, final int count) {
		add(
			minecraft.gui.toastManager(),
			SystemToast.SystemToastId.FILE_DROP_FAILURE,
			Component.translatable("gui.fileDropFailure.title"),
			Component.translatable("gui.fileDropFailure.detail", count)
		);
	}

	public static void onLowDiskSpace(final Minecraft minecraft) {
		addOrUpdate(
			minecraft.gui.toastManager(),
			SystemToast.SystemToastId.LOW_DISK_SPACE,
			Component.translatable("chunk.toast.lowDiskSpace"),
			Component.translatable("chunk.toast.lowDiskSpace.description")
		);
	}

	public static void onChunkLoadFailure(final Minecraft minecraft, final ChunkPos pos) {
		addOrUpdate(
			minecraft.gui.toastManager(),
			SystemToast.SystemToastId.CHUNK_LOAD_FAILURE,
			Component.translatable("chunk.toast.loadFailure", Component.translationArg(pos)).withStyle(ChatFormatting.RED),
			Component.translatable("chunk.toast.checkLog")
		);
	}

	public static void onChunkSaveFailure(final Minecraft minecraft, final ChunkPos pos) {
		addOrUpdate(
			minecraft.gui.toastManager(),
			SystemToast.SystemToastId.CHUNK_SAVE_FAILURE,
			Component.translatable("chunk.toast.saveFailure", Component.translationArg(pos)).withStyle(ChatFormatting.RED),
			Component.translatable("chunk.toast.checkLog")
		);
	}

	@Environment(EnvType.CLIENT)
	public static class SystemToastId {
		public static final SystemToast.SystemToastId NARRATOR_TOGGLE = new SystemToast.SystemToastId();
		public static final SystemToast.SystemToastId WORLD_BACKUP = new SystemToast.SystemToastId();
		public static final SystemToast.SystemToastId PACK_LOAD_FAILURE = new SystemToast.SystemToastId();
		public static final SystemToast.SystemToastId WORLD_ACCESS_FAILURE = new SystemToast.SystemToastId();
		public static final SystemToast.SystemToastId PACK_COPY_FAILURE = new SystemToast.SystemToastId();
		public static final SystemToast.SystemToastId FILE_DROP_FAILURE = new SystemToast.SystemToastId();
		public static final SystemToast.SystemToastId PERIODIC_NOTIFICATION = new SystemToast.SystemToastId();
		public static final SystemToast.SystemToastId LOW_DISK_SPACE = new SystemToast.SystemToastId(10000L);
		public static final SystemToast.SystemToastId CHUNK_LOAD_FAILURE = new SystemToast.SystemToastId();
		public static final SystemToast.SystemToastId CHUNK_SAVE_FAILURE = new SystemToast.SystemToastId();
		public static final SystemToast.SystemToastId UNSECURE_SERVER_WARNING = new SystemToast.SystemToastId(10000L);
		private final long displayTime;

		public SystemToastId(final long displayTime) {
			this.displayTime = displayTime;
		}

		public SystemToastId() {
			this(5000L);
		}
	}
}