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

import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.client.Minecraft;
import net.minecraft.client.gui.components.CycleButton;
import net.minecraft.client.gui.components.LockIconButton;
import net.minecraft.client.gui.layouts.EqualSpacingLayout;
import net.minecraft.client.gui.layouts.LayoutElement;
import net.minecraft.client.gui.screens.ConfirmScreen;
import net.minecraft.client.gui.screens.Screen;
import net.minecraft.network.chat.Component;
import net.minecraft.network.protocol.game.ServerboundChangeDifficultyPacket;
import net.minecraft.network.protocol.game.ServerboundLockDifficultyPacket;
import net.minecraft.world.Difficulty;
import net.minecraft.world.level.Level;

@Environment(EnvType.CLIENT)
public record DifficultyButtons(LayoutElement layout, CycleButton<Difficulty> difficultyButton, LockIconButton lockButton, Level level) {
	public static DifficultyButtons create(final Minecraft minecraft, final Level level, final Screen screen) {
		CycleButton<Difficulty> difficultyButton = CycleButton.builder(Difficulty::getDisplayName, level.getDifficulty())
			.withValues(Difficulty.values())
			.create(
				0,
				0,
				150,
				20,
				Component.translatable("options.difficulty"),
				(button, value) -> minecraft.getConnection().send(new ServerboundChangeDifficultyPacket(value))
			);
		LockIconButton lockButton = new LockIconButton(
			0,
			0,
			button -> minecraft.gui
				.setScreen(
					new ConfirmScreen(
						result -> onLockCallback(result, minecraft, screen, difficultyButton, (LockIconButton)button),
						Component.translatable("difficulty.lock.title"),
						Component.translatable("difficulty.lock.question", level.getLevelData().getDifficulty().getDisplayName())
					)
				)
		);
		difficultyButton.setWidth(difficultyButton.getWidth() - lockButton.getWidth());
		lockButton.setLocked(isDifficultyLocked(level));
		lockButton.active = !lockButton.isLocked() && playerHasPermissionToChangeDifficulty(minecraft);
		difficultyButton.active = !lockButton.isLocked() && playerHasPermissionToChangeDifficulty(minecraft);
		EqualSpacingLayout linearLayout = new EqualSpacingLayout(150, 0, EqualSpacingLayout.Orientation.HORIZONTAL);
		linearLayout.addChild(difficultyButton);
		linearLayout.addChild(lockButton);
		return new DifficultyButtons(linearLayout, difficultyButton, lockButton, level);
	}

	public void refresh(final Minecraft minecraft) {
		this.difficultyButton.setValue(this.level.getDifficulty());
		this.lockButton.setLocked(isDifficultyLocked(this.level));
		this.lockButton.active = !this.lockButton.isLocked() && playerHasPermissionToChangeDifficulty(minecraft);
		this.difficultyButton.active = !this.lockButton.isLocked() && playerHasPermissionToChangeDifficulty(minecraft);
	}

	private static boolean isDifficultyLocked(final Level level) {
		return level.getLevelData().isDifficultyLocked() || level.getLevelData().isHardcore();
	}

	private static boolean playerHasPermissionToChangeDifficulty(final Minecraft minecraft) {
		return minecraft.hasSingleplayerServer();
	}

	private static void onLockCallback(
		final boolean result, final Minecraft minecraft, final Screen screen, final CycleButton<Difficulty> difficultyButton, final LockIconButton lockButton
	) {
		minecraft.gui.setScreen(screen);
		if (result) {
			minecraft.getConnection().send(new ServerboundLockDifficultyPacket(true));
			lockButton.setLocked(true);
			lockButton.active = false;
			difficultyButton.active = false;
		}
	}
}