package net.minecraft.client.gui.screens;

import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.client.GameNarrator;
import net.minecraft.client.gui.GuiGraphicsExtractor;
import net.minecraft.network.chat.Component;
import net.minecraft.util.ProgressListener;
import org.jspecify.annotations.Nullable;

@Environment(EnvType.CLIENT)
public class ProgressScreen extends Screen implements ProgressListener {
	@Nullable
	private Component header;
	@Nullable
	private Component stage;
	private int progress;
	private boolean stop;
	private final boolean clearScreenAfterStop;

	public ProgressScreen(final boolean clearScreenAfterStop) {
		super(GameNarrator.NO_TITLE);
		this.clearScreenAfterStop = clearScreenAfterStop;
	}

	@Override
	public boolean shouldCloseOnEsc() {
		return false;
	}

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

	@Override
	public void progressStartNoAbort(final Component string) {
		this.progressStart(string);
	}

	@Override
	public void progressStart(final Component string) {
		this.header = string;
		this.progressStage(Component.translatable("menu.working"));
	}

	@Override
	public void progressStage(final Component string) {
		this.stage = string;
		this.progressStagePercentage(0);
	}

	@Override
	public void progressStagePercentage(final int i) {
		this.progress = i;
	}

	@Override
	public void stop() {
		this.stop = true;
	}

	@Override
	public void extractRenderState(final GuiGraphicsExtractor graphics, final int mouseX, final int mouseY, final float a) {
		if (this.stop) {
			if (this.clearScreenAfterStop) {
				this.minecraft.gui.setScreen(null);
			}
		} else {
			super.extractRenderState(graphics, mouseX, mouseY, a);
			if (this.header != null) {
				graphics.centeredText(this.font, this.header, this.width / 2, 70, -1);
			}

			if (this.stage != null && this.progress != 0) {
				graphics.centeredText(this.font, Component.empty().append(this.stage).append(" " + this.progress + "%"), this.width / 2, 90, -1);
			}
		}
	}
}