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

import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.client.gui.components.AbstractWidget;
import net.minecraft.client.gui.components.Checkbox;
import net.minecraft.client.gui.components.FocusableTextWidget;
import net.minecraft.client.gui.components.StringWidget;
import net.minecraft.client.gui.layouts.FrameLayout;
import net.minecraft.client.gui.layouts.Layout;
import net.minecraft.client.gui.layouts.LinearLayout;
import net.minecraft.client.gui.screens.Screen;
import net.minecraft.network.chat.Component;
import org.jetbrains.annotations.Nullable;

@Environment(EnvType.CLIENT)
public abstract class WarningScreen extends Screen {
	private static final int MESSAGE_PADDING = 100;
	private final Component message;
	@Nullable
	private final Component check;
	private final Component narration;
	@Nullable
	protected Checkbox stopShowing;
	@Nullable
	private FocusableTextWidget messageWidget;
	private final FrameLayout layout;

	protected WarningScreen(Component title, Component content, Component narration) {
		this(title, content, null, narration);
	}

	protected WarningScreen(Component title, Component content, @Nullable Component check, Component narration) {
		super(title);
		this.message = content;
		this.check = check;
		this.narration = narration;
		this.layout = new FrameLayout(0, 0, this.width, this.height);
	}

	protected abstract Layout addFooterButtons();

	@Override
	protected void init() {
		LinearLayout linearLayout = this.layout.addChild(LinearLayout.vertical().spacing(8));
		linearLayout.defaultCellSetting().alignHorizontallyCenter();
		linearLayout.addChild(new StringWidget(this.getTitle(), this.font));
		this.messageWidget = linearLayout.addChild(
			new FocusableTextWidget(this.width - 100, this.message, this.font, 12), layoutSettings -> layoutSettings.padding(12)
		);
		this.messageWidget.setCentered(false);
		LinearLayout linearLayout2 = linearLayout.addChild(LinearLayout.vertical().spacing(8));
		linearLayout2.defaultCellSetting().alignHorizontallyCenter();
		if (this.check != null) {
			this.stopShowing = linearLayout2.addChild(Checkbox.builder(this.check, this.font).build());
		}

		linearLayout2.addChild(this.addFooterButtons());
		this.layout.visitWidgets(guiEventListener -> {
			AbstractWidget var10000 = this.addRenderableWidget(guiEventListener);
		});
		this.repositionElements();
	}

	@Override
	protected void repositionElements() {
		if (this.messageWidget != null) {
			this.messageWidget.setMaxWidth(this.width - 100);
		}

		this.layout.arrangeElements();
		FrameLayout.centerInRectangle(this.layout, this.getRectangle());
	}

	@Override
	public Component getNarrationMessage() {
		return this.narration;
	}
}