package net.minecraft.client.gui.screens.telemetry;

import java.util.ArrayList;
import java.util.Comparator;
import java.util.List;
import java.util.function.DoubleConsumer;
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.gui.components.AbstractScrollArea;
import net.minecraft.client.gui.components.AbstractTextAreaWidget;
import net.minecraft.client.gui.components.MultiLineTextWidget;
import net.minecraft.client.gui.layouts.Layout;
import net.minecraft.client.gui.layouts.LinearLayout;
import net.minecraft.client.gui.layouts.SpacerElement;
import net.minecraft.client.gui.narration.NarratedElementType;
import net.minecraft.client.gui.narration.NarrationElementOutput;
import net.minecraft.client.telemetry.TelemetryEventType;
import net.minecraft.client.telemetry.TelemetryProperty;
import net.minecraft.network.chat.Component;
import net.minecraft.network.chat.MutableComponent;
import org.jspecify.annotations.Nullable;

public class TelemetryEventWidget extends AbstractTextAreaWidget {
	private static final int HEADER_HORIZONTAL_PADDING = 32;
	private static final String TELEMETRY_REQUIRED_TRANSLATION_KEY = "telemetry.event.required";
	private static final String TELEMETRY_OPTIONAL_TRANSLATION_KEY = "telemetry.event.optional";
	private static final String TELEMETRY_OPTIONAL_DISABLED_TRANSLATION_KEY = "telemetry.event.optional.disabled";
	private static final Component PROPERTY_TITLE = Component.translatable("telemetry_info.property_title").withStyle(ChatFormatting.UNDERLINE);
	private final Font font;
	private TelemetryEventWidget.Content content;
	@Nullable
	private DoubleConsumer onScrolledListener;

	public TelemetryEventWidget(final int x, final int y, final int width, final int height, final Font font) {
		super(x, y, width, height, Component.empty(), AbstractScrollArea.defaultSettings(9));
		this.font = font;
		this.content = this.buildContent(Minecraft.getInstance().telemetryOptInExtra());
	}

	public void onOptInChanged(final boolean optIn) {
		this.content = this.buildContent(optIn);
		this.refreshScrollAmount();
	}

	public void updateLayout() {
		this.content = this.buildContent(Minecraft.getInstance().telemetryOptInExtra());
		this.refreshScrollAmount();
	}

	private TelemetryEventWidget.Content buildContent(final boolean hasOptedIn) {
		TelemetryEventWidget.ContentBuilder content = new TelemetryEventWidget.ContentBuilder(this.containerWidth());
		List<TelemetryEventType> eventTypes = new ArrayList(TelemetryEventType.values());
		eventTypes.sort(Comparator.comparing(TelemetryEventType::isOptIn));

		for (int i = 0; i < eventTypes.size(); i++) {
			TelemetryEventType eventType = (TelemetryEventType)eventTypes.get(i);
			boolean isDisabled = eventType.isOptIn() && !hasOptedIn;
			this.addEventType(content, eventType, isDisabled);
			if (i < eventTypes.size() - 1) {
				content.addSpacer(9);
			}
		}

		return content.build();
	}

	public void setOnScrolledListener(@Nullable final DoubleConsumer listener) {
		this.onScrolledListener = listener;
	}

	@Override
	public void setScrollAmount(final double scrollAmount) {
		super.setScrollAmount(scrollAmount);
		if (this.onScrolledListener != null) {
			this.onScrolledListener.accept(this.scrollAmount());
		}
	}

	@Override
	protected int getInnerHeight() {
		return this.content.container().getHeight();
	}

	@Override
	protected void extractContents(final GuiGraphicsExtractor graphics, final int mouseX, final int mouseY, final float a) {
		int top = this.getInnerTop();
		int left = this.getInnerLeft();
		graphics.pose().pushMatrix();
		graphics.pose().translate(left, top);
		this.content.container().visitWidgets(widget -> widget.extractRenderState(graphics, mouseX, mouseY, a));
		graphics.pose().popMatrix();
	}

	@Override
	protected void updateWidgetNarration(final NarrationElementOutput output) {
		output.add(NarratedElementType.TITLE, this.content.narration());
	}

	private Component grayOutIfDisabled(final Component component, final boolean isDisabled) {
		return (Component)(isDisabled ? component.copy().withStyle(ChatFormatting.GRAY) : component);
	}

	private void addEventType(final TelemetryEventWidget.ContentBuilder builder, final TelemetryEventType eventType, final boolean isDisabled) {
		String titleTranslationPattern = eventType.isOptIn()
			? (isDisabled ? "telemetry.event.optional.disabled" : "telemetry.event.optional")
			: "telemetry.event.required";
		builder.addHeader(this.font, this.grayOutIfDisabled(Component.translatable(titleTranslationPattern, new Object[]{eventType.title()}), isDisabled));
		builder.addHeader(this.font, eventType.description().withStyle(ChatFormatting.GRAY));
		builder.addSpacer(9 / 2);
		builder.addLine(this.font, this.grayOutIfDisabled(PROPERTY_TITLE, isDisabled), 2);
		this.addEventTypeProperties(eventType, builder, isDisabled);
	}

	private void addEventTypeProperties(final TelemetryEventType eventType, final TelemetryEventWidget.ContentBuilder content, final boolean isDisabled) {
		for (TelemetryProperty<?> property : eventType.properties()) {
			content.addLine(this.font, this.grayOutIfDisabled(property.title(), isDisabled));
		}
	}

	private int containerWidth() {
		return this.width - this.totalInnerPadding();
	}

	private record Content(Layout container, Component narration) {
	}

	private static class ContentBuilder {
		private final int width;
		private final LinearLayout layout;
		private final MutableComponent narration = Component.empty();

		public ContentBuilder(final int width) {
			this.width = width;
			this.layout = LinearLayout.vertical();
			this.layout.defaultCellSetting().alignHorizontallyLeft();
			this.layout.addChild(SpacerElement.width(width));
		}

		public void addLine(final Font font, final Component line) {
			this.addLine(font, line, 0);
		}

		public void addLine(final Font font, final Component line, final int paddingBottom) {
			this.layout.addChild(new MultiLineTextWidget(line, font).setMaxWidth(this.width), s -> s.paddingBottom(paddingBottom));
			this.narration.append(line).append("\n");
		}

		public void addHeader(final Font font, final Component line) {
			this.layout
				.addChild(new MultiLineTextWidget(line, font).setMaxWidth(this.width - 64).setCentered(true), s -> s.alignHorizontallyCenter().paddingHorizontal(32));
			this.narration.append(line).append("\n");
		}

		public void addSpacer(final int height) {
			this.layout.addChild(SpacerElement.height(height));
		}

		public TelemetryEventWidget.Content build() {
			this.layout.arrangeElements();
			return new TelemetryEventWidget.Content(this.layout, this.narration);
		}
	}
}