package net.minecraft.client.gui.screens;

import it.unimi.dsi.fastutil.booleans.BooleanConsumer;
import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.client.Minecraft;
import net.minecraft.client.gui.GuiGraphics;
import net.minecraft.client.gui.components.Button;
import net.minecraft.client.gui.components.EditBox;
import net.minecraft.client.multiplayer.ServerData;
import net.minecraft.client.multiplayer.resolver.ServerAddress;
import net.minecraft.network.chat.CommonComponents;
import net.minecraft.network.chat.Component;

@Environment(EnvType.CLIENT)
public class DirectJoinServerScreen extends Screen {
	private static final Component ENTER_IP_LABEL = Component.translatable("addServer.enterIp");
	private Button selectButton;
	private final ServerData serverData;
	private EditBox ipEdit;
	private final BooleanConsumer callback;
	private final Screen lastScreen;

	public DirectJoinServerScreen(Screen lastScreen, BooleanConsumer callback, ServerData serverData) {
		super(Component.translatable("selectServer.direct"));
		this.lastScreen = lastScreen;
		this.serverData = serverData;
		this.callback = callback;
	}

	@Override
	public boolean keyPressed(int keyCode, int scanCode, int modifiers) {
		if (!this.selectButton.active || this.getFocused() != this.ipEdit || keyCode != 257 && keyCode != 335) {
			return super.keyPressed(keyCode, scanCode, modifiers);
		} else {
			this.onSelect();
			return true;
		}
	}

	@Override
	protected void init() {
		this.ipEdit = new EditBox(this.font, this.width / 2 - 100, 116, 200, 20, Component.translatable("addServer.enterIp"));
		this.ipEdit.setMaxLength(128);
		this.ipEdit.setValue(this.minecraft.options.lastMpIp);
		this.ipEdit.setResponder(string -> this.updateSelectButtonStatus());
		this.addWidget(this.ipEdit);
		this.selectButton = this.addRenderableWidget(
			Button.builder(Component.translatable("selectServer.select"), button -> this.onSelect())
				.bounds(this.width / 2 - 100, this.height / 4 + 96 + 12, 200, 20)
				.build()
		);
		this.addRenderableWidget(
			Button.builder(CommonComponents.GUI_CANCEL, button -> this.callback.accept(false)).bounds(this.width / 2 - 100, this.height / 4 + 120 + 12, 200, 20).build()
		);
		this.updateSelectButtonStatus();
	}

	@Override
	protected void setInitialFocus() {
		this.setInitialFocus(this.ipEdit);
	}

	@Override
	public void resize(Minecraft minecraft, int width, int height) {
		String string = this.ipEdit.getValue();
		this.init(minecraft, width, height);
		this.ipEdit.setValue(string);
	}

	private void onSelect() {
		this.serverData.ip = this.ipEdit.getValue();
		this.callback.accept(true);
	}

	@Override
	public void onClose() {
		this.minecraft.setScreen(this.lastScreen);
	}

	@Override
	public void removed() {
		this.minecraft.options.lastMpIp = this.ipEdit.getValue();
		this.minecraft.options.save();
	}

	private void updateSelectButtonStatus() {
		this.selectButton.active = ServerAddress.isValidAddress(this.ipEdit.getValue());
	}

	@Override
	public void render(GuiGraphics guiGraphics, int mouseX, int mouseY, float partialTick) {
		super.render(guiGraphics, mouseX, mouseY, partialTick);
		guiGraphics.drawCenteredString(this.font, this.title, this.width / 2, 20, 16777215);
		guiGraphics.drawString(this.font, ENTER_IP_LABEL, this.width / 2 - 100 + 1, 100, 10526880);
		this.ipEdit.render(guiGraphics, mouseX, mouseY, partialTick);
	}
}