package com.mojang.realmsclient.gui.screens;

import com.google.common.collect.ImmutableList;
import com.mojang.realmsclient.dto.RealmsWorldOptions;
import com.mojang.realmsclient.dto.RealmsServer.WorldType;
import java.util.List;
import java.util.function.Consumer;
import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.client.gui.GuiGraphics;
import net.minecraft.client.gui.components.AbstractSliderButton;
import net.minecraft.client.gui.components.Button;
import net.minecraft.client.gui.components.CycleButton;
import net.minecraft.client.gui.components.EditBox;
import net.minecraft.client.gui.components.CycleButton.OnValueChange;
import net.minecraft.network.chat.CommonComponents;
import net.minecraft.network.chat.Component;
import net.minecraft.realms.RealmsLabel;
import net.minecraft.realms.RealmsScreen;
import net.minecraft.util.Mth;
import net.minecraft.world.Difficulty;
import net.minecraft.world.level.GameType;

@Environment(EnvType.CLIENT)
public class RealmsSlotOptionsScreen extends RealmsScreen {
	private static final int DEFAULT_DIFFICULTY = 2;
	public static final List<Difficulty> DIFFICULTIES = ImmutableList.of(Difficulty.PEACEFUL, Difficulty.EASY, Difficulty.NORMAL, Difficulty.HARD);
	private static final int DEFAULT_GAME_MODE = 0;
	public static final List<GameType> GAME_MODES = ImmutableList.of(GameType.SURVIVAL, GameType.CREATIVE, GameType.ADVENTURE);
	private static final Component NAME_LABEL = Component.translatable("mco.configure.world.edit.slot.name");
	static final Component SPAWN_PROTECTION_TEXT = Component.translatable("mco.configure.world.spawnProtection");
	private EditBox nameEdit;
	protected final RealmsConfigureWorldScreen parentScreen;
	private int column1X;
	private int columnWidth;
	private final RealmsWorldOptions options;
	private final WorldType worldType;
	private Difficulty difficulty;
	private GameType gameMode;
	private final String defaultSlotName;
	private String worldName;
	private boolean pvp;
	private boolean spawnMonsters;
	int spawnProtection;
	private boolean commandBlocks;
	private boolean forceGameMode;
	RealmsSlotOptionsScreen.SettingsSlider spawnProtectionButton;

	public RealmsSlotOptionsScreen(RealmsConfigureWorldScreen parent, RealmsWorldOptions options, WorldType worldType, int activeSlot) {
		super(Component.translatable("mco.configure.world.buttons.options"));
		this.parentScreen = parent;
		this.options = options;
		this.worldType = worldType;
		this.difficulty = findByIndex(DIFFICULTIES, options.difficulty, 2);
		this.gameMode = findByIndex(GAME_MODES, options.gameMode, 0);
		this.defaultSlotName = options.getDefaultSlotName(activeSlot);
		this.setWorldName(options.getSlotName(activeSlot));
		if (worldType == WorldType.NORMAL) {
			this.pvp = options.pvp;
			this.spawnProtection = options.spawnProtection;
			this.forceGameMode = options.forceGameMode;
			this.spawnMonsters = options.spawnMonsters;
			this.commandBlocks = options.commandBlocks;
		} else {
			this.pvp = true;
			this.spawnProtection = 0;
			this.forceGameMode = false;
			this.spawnMonsters = true;
			this.commandBlocks = true;
		}
	}

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

	private static <T> T findByIndex(List<T> list, int index, int fallback) {
		try {
			return (T)list.get(index);
		} catch (IndexOutOfBoundsException var4) {
			return (T)list.get(fallback);
		}
	}

	private static <T> int findIndex(List<T> list, T object, int fallback) {
		int i = list.indexOf(object);
		return i == -1 ? fallback : i;
	}

	@Override
	public void init() {
		this.columnWidth = 170;
		this.column1X = this.width / 2 - this.columnWidth;
		int i = this.width / 2 + 10;
		if (this.worldType != WorldType.NORMAL) {
			Component component;
			if (this.worldType == WorldType.ADVENTUREMAP) {
				component = Component.translatable("mco.configure.world.edit.subscreen.adventuremap");
			} else if (this.worldType == WorldType.INSPIRATION) {
				component = Component.translatable("mco.configure.world.edit.subscreen.inspiration");
			} else {
				component = Component.translatable("mco.configure.world.edit.subscreen.experience");
			}

			this.addLabel(new RealmsLabel(component, this.width / 2, 26, 16711680));
		}

		this.nameEdit = this.addWidget(
			new EditBox(this.minecraft.font, this.column1X, row(1), this.columnWidth, 20, null, Component.translatable("mco.configure.world.edit.slot.name"))
		);
		this.nameEdit.setValue(this.worldName);
		this.nameEdit.setResponder(this::setWorldName);
		CycleButton<Boolean> cycleButton = this.addRenderableWidget(
			CycleButton.onOffBuilder(this.pvp)
				.create(i, row(1), this.columnWidth, 20, Component.translatable("mco.configure.world.pvp"), (cycleButtonx, boolean_) -> this.pvp = boolean_)
		);
		this.addRenderableWidget(
			CycleButton.<GameType>builder(GameType::getShortDisplayName)
				.withValues(GAME_MODES)
				.withInitialValue(this.gameMode)
				.create(this.column1X, row(3), this.columnWidth, 20, Component.translatable("selectWorld.gameMode"), (cycleButtonx, gameType) -> this.gameMode = gameType)
		);
		this.spawnProtectionButton = this.addRenderableWidget(
			new RealmsSlotOptionsScreen.SettingsSlider(i, row(3), this.columnWidth, this.spawnProtection, 0.0F, 16.0F)
		);
		Component component2 = Component.translatable("mco.configure.world.spawn_toggle.message");
		CycleButton<Boolean> cycleButton2 = CycleButton.onOffBuilder(this.difficulty != Difficulty.PEACEFUL && this.spawnMonsters)
			.create(
				i,
				row(5),
				this.columnWidth,
				20,
				Component.translatable("mco.configure.world.spawnMonsters"),
				this.confirmDangerousOption(component2, boolean_ -> this.spawnMonsters = boolean_)
			);
		this.addRenderableWidget(
			CycleButton.<Difficulty>builder(Difficulty::getDisplayName)
				.withValues(DIFFICULTIES)
				.withInitialValue(this.difficulty)
				.create(this.column1X, row(5), this.columnWidth, 20, Component.translatable("options.difficulty"), (cycleButton2x, difficulty) -> {
					this.difficulty = difficulty;
					if (this.worldType == WorldType.NORMAL) {
						boolean bl = this.difficulty != Difficulty.PEACEFUL;
						cycleButton2.active = bl;
						cycleButton2.setValue(bl && this.spawnMonsters);
					}
				})
		);
		this.addRenderableWidget(cycleButton2);
		CycleButton<Boolean> cycleButton3 = this.addRenderableWidget(
			CycleButton.onOffBuilder(this.forceGameMode)
				.create(
					this.column1X,
					row(7),
					this.columnWidth,
					20,
					Component.translatable("mco.configure.world.forceGameMode"),
					(cycleButtonx, boolean_) -> this.forceGameMode = boolean_
				)
		);
		CycleButton<Boolean> cycleButton4 = this.addRenderableWidget(
			CycleButton.onOffBuilder(this.commandBlocks)
				.create(
					i, row(7), this.columnWidth, 20, Component.translatable("mco.configure.world.commandBlocks"), (cycleButtonx, boolean_) -> this.commandBlocks = boolean_
				)
		);
		if (this.worldType != WorldType.NORMAL) {
			cycleButton.active = false;
			cycleButton2.active = false;
			this.spawnProtectionButton.active = false;
			cycleButton4.active = false;
			cycleButton3.active = false;
		}

		if (this.difficulty == Difficulty.PEACEFUL) {
			cycleButton2.active = false;
		}

		this.addRenderableWidget(
			Button.builder(Component.translatable("mco.configure.world.buttons.done"), button -> this.saveSettings())
				.bounds(this.column1X, row(13), this.columnWidth, 20)
				.build()
		);
		this.addRenderableWidget(Button.builder(CommonComponents.GUI_CANCEL, button -> this.onClose()).bounds(i, row(13), this.columnWidth, 20).build());
	}

	private OnValueChange<Boolean> confirmDangerousOption(Component question, Consumer<Boolean> onPress) {
		return (cycleButton, boolean_) -> {
			if (boolean_) {
				onPress.accept(true);
			} else {
				this.minecraft.setScreen(RealmsPopups.warningPopupScreen(this, question, popupScreen -> {
					onPress.accept(false);
					popupScreen.onClose();
				}));
			}
		};
	}

	@Override
	public Component getNarrationMessage() {
		return CommonComponents.joinForNarration(this.getTitle(), this.createLabelNarration());
	}

	@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, 17, -1);
		guiGraphics.drawString(this.font, NAME_LABEL, this.column1X + this.columnWidth / 2 - this.font.width(NAME_LABEL) / 2, row(0) - 5, -1);
		this.nameEdit.render(guiGraphics, mouseX, mouseY, partialTick);
	}

	private void setWorldName(String name) {
		if (name.equals(this.defaultSlotName)) {
			this.worldName = "";
		} else {
			this.worldName = name;
		}
	}

	private void saveSettings() {
		int i = findIndex(DIFFICULTIES, this.difficulty, 2);
		int j = findIndex(GAME_MODES, this.gameMode, 0);
		if (this.worldType != WorldType.ADVENTUREMAP && this.worldType != WorldType.EXPERIENCE && this.worldType != WorldType.INSPIRATION) {
			boolean bl = this.worldType == WorldType.NORMAL && this.difficulty != Difficulty.PEACEFUL && this.spawnMonsters;
			this.parentScreen
				.saveSlotSettings(
					new RealmsWorldOptions(
						this.pvp,
						bl,
						this.spawnProtection,
						this.commandBlocks,
						i,
						j,
						this.options.hardcore,
						this.forceGameMode,
						this.worldName,
						this.options.version,
						this.options.compatibility
					)
				);
		} else {
			this.parentScreen
				.saveSlotSettings(
					new RealmsWorldOptions(
						this.options.pvp,
						this.options.spawnMonsters,
						this.options.spawnProtection,
						this.options.commandBlocks,
						i,
						j,
						this.options.hardcore,
						this.options.forceGameMode,
						this.worldName,
						this.options.version,
						this.options.compatibility
					)
				);
		}
	}

	@Environment(EnvType.CLIENT)
	class SettingsSlider extends AbstractSliderButton {
		private final double minValue;
		private final double maxValue;

		public SettingsSlider(final int x, final int y, final int width, final int value, final float minValue, final float maxValue) {
			super(x, y, width, 20, CommonComponents.EMPTY, 0.0);
			this.minValue = minValue;
			this.maxValue = maxValue;
			this.value = (Mth.clamp((float)value, minValue, maxValue) - minValue) / (maxValue - minValue);
			this.updateMessage();
		}

		@Override
		public void applyValue() {
			if (RealmsSlotOptionsScreen.this.spawnProtectionButton.active) {
				RealmsSlotOptionsScreen.this.spawnProtection = (int)Mth.lerp(Mth.clamp(this.value, 0.0, 1.0), this.minValue, this.maxValue);
			}
		}

		@Override
		protected void updateMessage() {
			this.setMessage(
				CommonComponents.optionNameValue(
					RealmsSlotOptionsScreen.SPAWN_PROTECTION_TEXT,
					(Component)(RealmsSlotOptionsScreen.this.spawnProtection == 0
						? CommonComponents.OPTION_OFF
						: Component.literal(String.valueOf(RealmsSlotOptionsScreen.this.spawnProtection)))
				)
			);
		}
	}
}