package net.minecraft.network.chat;

import com.google.common.collect.Lists;
import java.util.List;
import java.util.function.UnaryOperator;
import net.minecraft.ChatFormatting;
import net.minecraft.locale.Language;
import net.minecraft.util.FormattedCharSequence;
import org.jspecify.annotations.Nullable;

public final class MutableComponent implements Component {
	private final ComponentContents contents;
	private final List<Component> siblings;
	private Style style;
	private FormattedCharSequence visualOrderText = FormattedCharSequence.EMPTY;
	@Nullable
	private Language decomposedWith;

	MutableComponent(final ComponentContents contents, final List<Component> siblings, final Style style) {
		this.contents = contents;
		this.siblings = siblings;
		this.style = style;
	}

	public static MutableComponent create(final ComponentContents contents) {
		return new MutableComponent(contents, Lists.<Component>newArrayList(), Style.EMPTY);
	}

	@Override
	public ComponentContents getContents() {
		return this.contents;
	}

	@Override
	public List<Component> getSiblings() {
		return this.siblings;
	}

	public MutableComponent setStyle(final Style style) {
		this.style = style;
		return this;
	}

	@Override
	public Style getStyle() {
		return this.style;
	}

	public MutableComponent append(final String text) {
		return text.isEmpty() ? this : this.append(Component.literal(text));
	}

	public MutableComponent append(final Component component) {
		this.siblings.add(component);
		return this;
	}

	public MutableComponent withStyle(final UnaryOperator<Style> updater) {
		this.setStyle((Style)updater.apply(this.getStyle()));
		return this;
	}

	public MutableComponent withStyle(final Style patch) {
		this.setStyle(patch.applyTo(this.getStyle()));
		return this;
	}

	public MutableComponent withStyle(final ChatFormatting... formats) {
		this.setStyle(this.getStyle().applyFormats(formats));
		return this;
	}

	public MutableComponent withStyle(final ChatFormatting format) {
		this.setStyle(this.getStyle().applyFormat(format));
		return this;
	}

	public MutableComponent withColor(final int color) {
		this.setStyle(this.getStyle().withColor(color));
		return this;
	}

	public MutableComponent withColor(final TextColor color) {
		this.setStyle(this.getStyle().withColor(color));
		return this;
	}

	public MutableComponent withoutShadow() {
		this.setStyle(this.getStyle().withoutShadow());
		return this;
	}

	@Override
	public FormattedCharSequence getVisualOrderText() {
		Language currentLanguage = Language.getInstance();
		if (this.decomposedWith != currentLanguage) {
			this.visualOrderText = currentLanguage.getVisualOrder(this);
			this.decomposedWith = currentLanguage;
		}

		return this.visualOrderText;
	}

	public boolean equals(final Object o) {
		return this == o
			? true
			: o instanceof MutableComponent that && this.contents.equals(that.contents) && this.style.equals(that.style) && this.siblings.equals(that.siblings);
	}

	public int hashCode() {
		int result = 1;
		result = 31 * result + this.contents.hashCode();
		result = 31 * result + this.style.hashCode();
		return 31 * result + this.siblings.hashCode();
	}

	public String toString() {
		StringBuilder result = new StringBuilder(this.contents.toString());
		boolean hasStyle = !this.style.isEmpty();
		boolean hasSiblings = !this.siblings.isEmpty();
		if (hasStyle || hasSiblings) {
			result.append('[');
			if (hasStyle) {
				result.append("style=");
				result.append(this.style);
			}

			if (hasStyle && hasSiblings) {
				result.append(", ");
			}

			if (hasSiblings) {
				result.append("siblings=");
				result.append(this.siblings);
			}

			result.append(']');
		}

		return result.toString();
	}
}