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.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.GuiGraphics;
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.jetbrains.annotations.Nullable;

@Environment(EnvType.CLIENT)
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(int x, int y, int width, int height, Font font) {
		super(x, y, width, height, Component.empty());
		this.font = font;
		this.content = this.buildContent(Minecraft.getInstance().telemetryOptInExtra());
	}

	public void onOptInChanged(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(boolean optIn) {
		TelemetryEventWidget.ContentBuilder contentBuilder = new TelemetryEventWidget.ContentBuilder(this.containerWidth());
		List<TelemetryEventType> list = new ArrayList(TelemetryEventType.values());
		list.sort(Comparator.comparing(TelemetryEventType::isOptIn));

		for (int i = 0; i < list.size(); i++) {
			TelemetryEventType telemetryEventType = (TelemetryEventType)list.get(i);
			boolean bl = telemetryEventType.isOptIn() && !optIn;
			this.addEventType(contentBuilder, telemetryEventType, bl);
			if (i < list.size() - 1) {
				contentBuilder.addSpacer(9);
			}
		}

		return contentBuilder.build();
	}

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

	@Override
	public void setScrollAmount(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 double scrollRate() {
		return 9.0;
	}

	@Override
	protected void renderContents(GuiGraphics guiGraphics, int mouseX, int mouseY, float partialTick) {
		int i = this.getInnerTop();
		int j = this.getInnerLeft();
		guiGraphics.pose().pushPose();
		guiGraphics.pose().translate((double)j, (double)i, 0.0);
		this.content.container().visitWidgets(abstractWidget -> abstractWidget.render(guiGraphics, mouseX, mouseY, partialTick));
		guiGraphics.pose().popPose();
	}

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

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

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

	private void addEventTypeProperties(TelemetryEventType eventType, TelemetryEventWidget.ContentBuilder contentBuilder, boolean disabled) {
		for (TelemetryProperty<?> telemetryProperty : eventType.properties()) {
			contentBuilder.addLine(this.font, this.grayOutIfDisabled(telemetryProperty.title(), disabled));
		}
	}

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

	@Environment(EnvType.CLIENT)
	record Content(Layout container, Component narration) {
	}

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

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

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

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

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

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

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