package net.minecraft.network.chat;

import com.google.common.collect.Lists;
import com.mojang.brigadier.Message;
import com.mojang.brigadier.exceptions.CommandSyntaxException;
import com.mojang.datafixers.DataFixUtils;
import java.util.Collection;
import java.util.List;
import java.util.Optional;
import java.util.function.Function;
import javax.annotation.CheckReturnValue;
import net.minecraft.ChatFormatting;
import net.minecraft.locale.Language;
import net.minecraft.network.chat.contents.TranslatableContents;
import org.jspecify.annotations.Nullable;

public class ComponentUtils {
	public static final String DEFAULT_SEPARATOR_TEXT = ", ";
	public static final Component DEFAULT_SEPARATOR = Component.literal(", ").withStyle(ChatFormatting.GRAY);
	public static final Component DEFAULT_NO_STYLE_SEPARATOR = Component.literal(", ");

	@CheckReturnValue
	public static MutableComponent mergeStyles(final MutableComponent component, final Style style) {
		if (style.isEmpty()) {
			return component;
		} else {
			Style inner = component.getStyle();
			if (inner.isEmpty()) {
				return component.setStyle(style);
			} else {
				return inner.equals(style) ? component : component.setStyle(inner.applyTo(style));
			}
		}
	}

	@CheckReturnValue
	public static Component mergeStyles(final Component component, final Style style) {
		if (style.isEmpty()) {
			return component;
		} else {
			Style inner = component.getStyle();
			if (inner.isEmpty()) {
				return component.copy().setStyle(style);
			} else {
				return (Component)(inner.equals(style) ? component : component.copy().setStyle(inner.applyTo(style)));
			}
		}
	}

	public static Optional<MutableComponent> resolve(final ResolutionContext context, final Optional<Component> component, final int recursionDepth) throws CommandSyntaxException {
		return component.isPresent() ? Optional.of(resolve(context, (Component)component.get(), recursionDepth)) : Optional.empty();
	}

	public static MutableComponent resolve(final ResolutionContext context, final Component component) throws CommandSyntaxException {
		return resolve(context, component, 0);
	}

	public static MutableComponent resolve(final ResolutionContext context, final Component component, final int recursionDepth) throws CommandSyntaxException {
		if (recursionDepth > context.depthLimit()) {
			return switch (context.depthLimitBehavior()) {
				case DISCARD_REMAINING -> CommonComponents.ELLIPSIS.copy();
				case STOP_PROCESSING_AND_COPY_REMAINING -> component.copy();
			};
		} else {
			MutableComponent result = component.getContents().resolve(context, recursionDepth + 1);

			for (Component sibling : component.getSiblings()) {
				result.append(resolve(context, sibling, recursionDepth + 1));
			}

			return result.withStyle(resolveStyle(context, component.getStyle(), recursionDepth));
		}
	}

	private static Style resolveStyle(final ResolutionContext context, final Style style, final int recursionDepth) throws CommandSyntaxException {
		if (style.getHoverEvent() instanceof HoverEvent.ShowText(Component resolved)) {
			HoverEvent resolvedx = new HoverEvent.ShowText(resolve(context, resolved, recursionDepth + 1));
			return style.withHoverEvent(resolvedx);
		} else {
			return style;
		}
	}

	public static Component formatList(final Collection<String> values) {
		return formatAndSortList(values, v -> Component.literal(v).withStyle(ChatFormatting.GREEN));
	}

	public static <T extends Comparable<T>> Component formatAndSortList(final Collection<T> values, final Function<T, Component> formatter) {
		if (values.isEmpty()) {
			return CommonComponents.EMPTY;
		} else if (values.size() == 1) {
			return (Component)formatter.apply((Comparable)values.iterator().next());
		} else {
			List<T> sorted = Lists.<T>newArrayList(values);
			sorted.sort(Comparable::compareTo);
			return formatList(sorted, formatter);
		}
	}

	public static <T> Component formatList(final Collection<? extends T> values, final Function<T, Component> formatter) {
		return formatList(values, DEFAULT_SEPARATOR, formatter);
	}

	public static <T> MutableComponent formatList(
		final Collection<? extends T> values, final Optional<? extends Component> separator, final Function<T, Component> formatter
	) {
		return formatList(values, DataFixUtils.orElse(separator, DEFAULT_SEPARATOR), formatter);
	}

	public static Component formatList(final Collection<? extends Component> values, final Component separator) {
		return formatList(values, separator, Function.identity());
	}

	public static <T> MutableComponent formatList(final Collection<? extends T> values, final Component separator, final Function<T, Component> formatter) {
		if (values.isEmpty()) {
			return Component.empty();
		} else if (values.size() == 1) {
			return ((Component)formatter.apply(values.iterator().next())).copy();
		} else {
			MutableComponent result = Component.empty();
			boolean first = true;

			for (T value : values) {
				if (!first) {
					result.append(separator);
				}

				result.append((Component)formatter.apply(value));
				first = false;
			}

			return result;
		}
	}

	public static MutableComponent wrapInSquareBrackets(final Component inner) {
		return Component.translatable("chat.square_brackets", inner);
	}

	public static Component fromMessage(final Message message) {
		return (Component)(message instanceof Component component ? component : Component.literal(message.getString()));
	}

	public static boolean isTranslationResolvable(@Nullable final Component component) {
		if (component != null && component.getContents() instanceof TranslatableContents translatable) {
			String key = translatable.getKey();
			String fallback = translatable.getFallback();
			return fallback != null || Language.getInstance().has(key);
		} else {
			return true;
		}
	}

	public static MutableComponent copyOnClickText(final String text) {
		return wrapInSquareBrackets(
			Component.literal(text)
				.withStyle(
					s -> s.withColor(ChatFormatting.GREEN)
						.withClickEvent(new ClickEvent.CopyToClipboard(text))
						.withHoverEvent(new HoverEvent.ShowText(Component.translatable("chat.copy.click")))
						.withInsertion(text)
				)
		);
	}
}