package net.minecraft.client.gui.screens;

import com.mojang.text2speech.Narrator;
import net.minecraft.client.Minecraft;
import net.minecraft.client.NarratorStatus;
import net.minecraft.client.Options;
import net.minecraft.client.gui.GuiGraphicsExtractor;
import net.minecraft.client.gui.components.Button;
import net.minecraft.client.gui.components.CommonButtons;
import net.minecraft.client.gui.components.CycleButton;
import net.minecraft.client.gui.components.FocusableTextWidget;
import net.minecraft.client.gui.components.LogoRenderer;
import net.minecraft.client.gui.components.SpriteIconButton;
import net.minecraft.client.gui.layouts.GridLayout;
import net.minecraft.client.gui.layouts.HeaderAndFooterLayout;
import net.minecraft.client.gui.layouts.LinearLayout;
import net.minecraft.client.gui.screens.options.AccessibilityOptionsScreen;
import net.minecraft.client.gui.screens.options.LanguageSelectScreen;
import net.minecraft.client.gui.screens.options.SoundOptionsScreen;
import net.minecraft.network.chat.CommonComponents;
import net.minecraft.network.chat.Component;
import net.minecraft.resources.Identifier;
import net.minecraft.sounds.SoundSource;
import net.minecraft.util.Mth;
import net.minecraft.util.Util;
import org.jspecify.annotations.Nullable;

public class AccessibilityOnboardingScreen extends Screen {
	private static final Component TITLE = Component.translatable("accessibility.onboarding.screen.title");
	private static final Component ONBOARDING_NARRATOR_MESSAGE = Component.translatable("accessibility.onboarding.screen.narrator");
	private static final int PADDING = 4;
	private static final int TITLE_PADDING = 16;
	private static final float FADE_OUT_TIME = 1000.0F;
	private static final int TEXT_WIDGET_WIDTH = 374;
	private final LogoRenderer logoRenderer;
	private final Options options;
	private final boolean narratorAvailable;
	private boolean hasNarrated;
	private float timer;
	private final Runnable onClose;
	private final HeaderAndFooterLayout layout = new HeaderAndFooterLayout(this, this.initTitleYPos(), 33);
	@Nullable
	private FocusableTextWidget focusableTextWidget;
	private float fadeInStart;
	private boolean fadingIn = true;
	private float fadeOutStart;

	public AccessibilityOnboardingScreen(final Options options, final Runnable onClose) {
		super(TITLE);
		this.options = options;
		this.onClose = onClose;
		this.logoRenderer = new LogoRenderer(true);
		this.narratorAvailable = Minecraft.getInstance().getNarrator().isActive();
	}

	@Override
	public void init() {
		LinearLayout content = this.layout.addToContents(LinearLayout.vertical());
		content.defaultCellSetting().alignHorizontallyCenter().padding(4);
		this.focusableTextWidget = content.addChild(FocusableTextWidget.builder(this.title, this.font).maxWidth(374).build(), w -> w.padding(8));
		GridLayout grid = content.addChild(new GridLayout());
		grid.defaultCellSetting().padding(4);
		GridLayout.RowHelper rowHelper = grid.createRowHelper(2);
		if (this.options.narrator().createButton(this.options) instanceof CycleButton<?> cycleButton) {
			this.narratorButton = (CycleButton<NarratorStatus>)cycleButton;
			this.narratorButton.active = this.narratorAvailable;
			rowHelper.addChild(this.narratorButton);
		}

		rowHelper.addChild(
			SpriteIconButton.builder(Component.translatable("options.sounds"), button -> this.closeAndSetScreen(new SoundOptionsScreen(this, this.options)), false)
				.width(150)
				.sprite(Identifier.withDefaultNamespace("icon/music_notes"), 16, 16)
				.build()
		);
		rowHelper.addChild(CommonButtons.accessibility(150, button -> this.closeAndSetScreen(new AccessibilityOptionsScreen(this, this.minecraft.options)), false));
		rowHelper.addChild(
			CommonButtons.language(
				150, button -> this.closeAndSetScreen(new LanguageSelectScreen(this, this.minecraft.options, this.minecraft.getLanguageManager())), false
			)
		);
		this.layout.addToFooter(Button.builder(CommonComponents.GUI_CONTINUE, button -> this.onClose()).build());
		this.layout.visitWidgets(this::addRenderableWidget);
		this.repositionElements();
	}

	@Override
	protected void repositionElements() {
		if (this.focusableTextWidget != null) {
			this.focusableTextWidget.updateHeight();
		}

		this.layout.arrangeElements();
	}

	@Override
	protected void setInitialFocus() {
		if (this.narratorAvailable && this.narratorButton != null) {
			this.setInitialFocus(this.narratorButton);
		} else {
			super.setInitialFocus();
		}
	}

	private int initTitleYPos() {
		return 90;
	}

	@Override
	public void onClose() {
		if (this.fadeOutStart == 0.0F) {
			this.fadeOutStart = (float)Util.getMillis();
		}
	}

	private void closeAndSetScreen(final Screen screen) {
		this.close(false, () -> this.minecraft.gui.setScreen(screen));
	}

	private void close(final boolean onboardingFinished, final Runnable runnable) {
		if (onboardingFinished) {
			this.options.onboardingAccessibilityFinished();
		}

		Narrator.getNarrator().clear();
		runnable.run();
	}

	@Override
	public void extractRenderState(final GuiGraphicsExtractor graphics, final int mouseX, final int mouseY, final float a) {
		super.extractRenderState(graphics, mouseX, mouseY, a);
		this.handleInitialNarrationDelay();
		if (this.fadeInStart == 0.0F && this.fadingIn) {
			this.fadeInStart = (float)Util.getMillis();
		}

		if (this.fadeInStart > 0.0F) {
			float fade = ((float)Util.getMillis() - this.fadeInStart) / 2000.0F;
			float widgetAlpha = 1.0F;
			if (fade >= 1.0F) {
				this.fadingIn = false;
				this.fadeInStart = 0.0F;
			} else {
				fade = Mth.clamp(fade, 0.0F, 1.0F);
				widgetAlpha = Mth.clampedMap(fade, 0.5F, 1.0F, 0.0F, 1.0F);
			}

			this.fadeWidgets(widgetAlpha);
		}

		if (this.fadeOutStart > 0.0F) {
			float fade = 1.0F - ((float)Util.getMillis() - this.fadeOutStart) / 1000.0F;
			float widgetAlpha = 0.0F;
			if (fade <= 0.0F) {
				this.fadeOutStart = 0.0F;
				this.close(true, this.onClose);
			} else {
				fade = Mth.clamp(fade, 0.0F, 1.0F);
				widgetAlpha = Mth.clampedMap(fade, 0.5F, 1.0F, 0.0F, 1.0F);
			}

			this.fadeWidgets(widgetAlpha);
		}

		this.logoRenderer.extractRenderState(graphics, this.width, 1.0F);
	}

	@Override
	protected boolean panoramaShouldSpin() {
		return false;
	}

	private void handleInitialNarrationDelay() {
		if (!this.hasNarrated && this.narratorAvailable) {
			if (this.timer < 40.0F) {
				this.timer++;
			} else if (this.minecraft.isWindowActive()) {
				Narrator.getNarrator().say(ONBOARDING_NARRATOR_MESSAGE.getString(), true, this.minecraft.options.getFinalSoundSourceVolume(SoundSource.VOICE));
				this.hasNarrated = true;
			}
		}
	}
}