package net.minecraft.client.gui.components;

import java.util.ArrayList;
import java.util.List;
import net.minecraft.client.gui.ActiveTextCollector;
import net.minecraft.client.gui.Font;
import net.minecraft.client.gui.TextAlignment;
import net.minecraft.locale.Language;
import net.minecraft.network.chat.CommonComponents;
import net.minecraft.network.chat.Component;
import net.minecraft.network.chat.FormattedText;
import net.minecraft.util.FormattedCharSequence;
import org.jspecify.annotations.Nullable;

public interface MultiLineLabel {
	MultiLineLabel EMPTY = new MultiLineLabel() {
		@Override
		public int visitLines(final TextAlignment align, final int anchorX, final int topY, final int lineHeight, final ActiveTextCollector output) {
			return topY;
		}

		@Override
		public int getLineCount() {
			return 0;
		}

		@Override
		public int getWidth() {
			return 0;
		}
	};

	static MultiLineLabel create(final Font font, final Component... messages) {
		return create(font, Integer.MAX_VALUE, Integer.MAX_VALUE, messages);
	}

	static MultiLineLabel create(final Font font, final int maxWidth, final Component... messages) {
		return create(font, maxWidth, Integer.MAX_VALUE, messages);
	}

	static MultiLineLabel create(final Font font, final Component message, final int maxWidth) {
		return create(font, maxWidth, Integer.MAX_VALUE, message);
	}

	static MultiLineLabel create(final Font font, final int maxWidth, final int maxLines, final Component... messages) {
		return messages.length == 0
			? EMPTY
			: new MultiLineLabel() {
				@Nullable
				private List<MultiLineLabel.TextAndWidth> cachedTextAndWidth;
				@Nullable
				private Language splitWithLanguage;

				@Override
				public int visitLines(final TextAlignment align, final int anchorX, final int topY, final int lineHeight, final ActiveTextCollector output) {
					int y = topY;

					for (MultiLineLabel.TextAndWidth splitLine : this.getSplitMessage()) {
						int leftX = align.calculateLeft(anchorX, splitLine.width);
						output.accept(leftX, y, splitLine.text);
						y += lineHeight;
					}

					return y;
				}

				private List<MultiLineLabel.TextAndWidth> getSplitMessage() {
					Language currentLanguage = Language.getInstance();
					if (this.cachedTextAndWidth != null && currentLanguage == this.splitWithLanguage) {
						return this.cachedTextAndWidth;
					} else {
						this.splitWithLanguage = currentLanguage;
						List<FormattedText> splitMessage = new ArrayList();

						for (Component message : messages) {
							splitMessage.addAll(font.splitIgnoringLanguage(message, maxWidth));
						}

						this.cachedTextAndWidth = new ArrayList();
						int actualMaxLines = Math.min(splitMessage.size(), maxLines);
						List<FormattedText> linesToAdd = splitMessage.subList(0, actualMaxLines);

						for (int i = 0; i < linesToAdd.size(); i++) {
							FormattedText formattedText = (FormattedText)linesToAdd.get(i);
							FormattedCharSequence formattedCharSequence = Language.getInstance().getVisualOrder(formattedText);
							if (i == linesToAdd.size() - 1 && actualMaxLines == maxLines && actualMaxLines != splitMessage.size()) {
								FormattedText clippedText = font.substrByWidth(formattedText, font.width(formattedText) - font.width(CommonComponents.ELLIPSIS));
								FormattedText withEllipsis = FormattedText.composite(
									new FormattedText[]{clippedText, CommonComponents.ELLIPSIS.copy().withStyle(messages[messages.length - 1].getStyle())}
								);
								this.cachedTextAndWidth.add(new MultiLineLabel.TextAndWidth(Language.getInstance().getVisualOrder(withEllipsis), font.width(withEllipsis)));
							} else {
								this.cachedTextAndWidth.add(new MultiLineLabel.TextAndWidth(formattedCharSequence, font.width(formattedCharSequence)));
							}
						}

						return this.cachedTextAndWidth;
					}
				}

				@Override
				public int getLineCount() {
					return this.getSplitMessage().size();
				}

				@Override
				public int getWidth() {
					return Math.min(maxWidth, this.getSplitMessage().stream().mapToInt(MultiLineLabel.TextAndWidth::width).max().orElse(0));
				}
			};
	}

	int visitLines(TextAlignment align, int anchorX, int topY, int lineHeight, ActiveTextCollector output);

	int getLineCount();

	int getWidth();

	public record TextAndWidth(FormattedCharSequence text, int width) {
	}
}