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

import net.minecraft.client.gui.components.CycleButton;
import net.minecraft.network.chat.Component;
import net.minecraft.network.protocol.game.ServerboundSetCommandBlockPacket;
import net.minecraft.world.level.BaseCommandBlock;
import net.minecraft.world.level.block.entity.CommandBlockEntity;
import net.minecraft.world.level.block.entity.CommandBlockEntity.Mode;

public class CommandBlockEditScreen extends AbstractCommandBlockEditScreen {
	private final CommandBlockEntity autoCommandBlock;
	private CycleButton<Mode> modeButton;
	private CycleButton<Boolean> conditionalButton;
	private CycleButton<Boolean> autoexecButton;
	private Mode mode = Mode.REDSTONE;
	private boolean conditional;
	private boolean autoexec;

	public CommandBlockEditScreen(final CommandBlockEntity commandBlock) {
		this.autoCommandBlock = commandBlock;
	}

	@Override
	protected BaseCommandBlock getCommandBlock() {
		return this.autoCommandBlock.getCommandBlock();
	}

	@Override
	protected int getPreviousY() {
		return 135;
	}

	@Override
	protected void init() {
		super.init();
		this.enableControls(false);
	}

	@Override
	protected void addExtraControls() {
		this.modeButton = this.addRenderableWidget(
			CycleButton.<Mode>builder(mode -> {
					return switch (mode) {
						case SEQUENCE -> Component.translatable("advMode.mode.sequence");
						case AUTO -> Component.translatable("advMode.mode.auto");
						case REDSTONE -> Component.translatable("advMode.mode.redstone");
						default -> throw new MatchException(null, null);
					};
				}, this.mode)
				.withValues(Mode.values())
				.displayOnlyValue()
				.create(this.width / 2 - 50 - 100 - 4, 165, 100, 20, Component.translatable("advMode.mode"), (button, value) -> this.mode = value)
		);
		this.conditionalButton = this.addRenderableWidget(
			CycleButton.booleanBuilder(Component.translatable("advMode.mode.conditional"), Component.translatable("advMode.mode.unconditional"), this.conditional)
				.displayOnlyValue()
				.create(this.width / 2 - 50, 165, 100, 20, Component.translatable("advMode.type"), (button, value) -> this.conditional = value)
		);
		this.autoexecButton = this.addRenderableWidget(
			CycleButton.booleanBuilder(Component.translatable("advMode.mode.autoexec.bat"), Component.translatable("advMode.mode.redstoneTriggered"), this.autoexec)
				.displayOnlyValue()
				.create(this.width / 2 + 50 + 4, 165, 100, 20, Component.translatable("advMode.triggering"), (button, value) -> this.autoexec = value)
		);
	}

	private void enableControls(final boolean state) {
		this.doneButton.active = state;
		this.outputButton.active = state;
		this.modeButton.active = state;
		this.conditionalButton.active = state;
		this.autoexecButton.active = state;
	}

	public void updateGui() {
		BaseCommandBlock commandBlock = this.autoCommandBlock.getCommandBlock();
		this.commandEdit.setValue(commandBlock.getCommand());
		boolean trackOutput = commandBlock.isTrackOutput();
		this.mode = this.autoCommandBlock.getMode();
		this.conditional = this.autoCommandBlock.isConditional();
		this.autoexec = this.autoCommandBlock.isAutomatic();
		this.outputButton.setValue(trackOutput);
		this.modeButton.setValue(this.mode);
		this.conditionalButton.setValue(this.conditional);
		this.autoexecButton.setValue(this.autoexec);
		this.updatePreviousOutput(trackOutput);
		this.enableControls(true);
	}

	@Override
	public void resize(final int width, final int height) {
		super.resize(width, height);
		this.enableControls(true);
	}

	@Override
	protected void populateAndSendPacket() {
		this.minecraft
			.getConnection()
			.send(
				new ServerboundSetCommandBlockPacket(
					this.autoCommandBlock.getBlockPos(),
					this.commandEdit.getValue(),
					this.mode,
					this.autoCommandBlock.getCommandBlock().isTrackOutput(),
					this.conditional,
					this.autoexec
				)
			);
	}
}