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

import com.google.common.collect.ImmutableList;
import java.text.DecimalFormat;
import java.text.DecimalFormatSymbols;
import java.util.Locale;
import java.util.Objects;
import net.minecraft.client.gui.GuiGraphicsExtractor;
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.screens.Screen;
import net.minecraft.client.input.CharacterEvent;
import net.minecraft.client.input.KeyEvent;
import net.minecraft.core.BlockPos;
import net.minecraft.core.Vec3i;
import net.minecraft.network.chat.CommonComponents;
import net.minecraft.network.chat.Component;
import net.minecraft.network.protocol.game.ServerboundSetStructureBlockPacket;
import net.minecraft.world.level.block.Blocks;
import net.minecraft.world.level.block.Mirror;
import net.minecraft.world.level.block.Rotation;
import net.minecraft.world.level.block.entity.StructureBlockEntity;
import net.minecraft.world.level.block.entity.StructureBlockEntity.UpdateType;
import net.minecraft.world.level.block.state.properties.StructureMode;

public class StructureBlockEditScreen extends Screen {
	private static final Component NAME_LABEL = Component.translatable("structure_block.structure_name");
	private static final Component POSITION_LABEL = Component.translatable("structure_block.position");
	private static final Component SIZE_LABEL = Component.translatable("structure_block.size");
	private static final Component INTEGRITY_LABEL = Component.translatable("structure_block.integrity");
	private static final Component CUSTOM_DATA_LABEL = Component.translatable("structure_block.custom_data");
	private static final Component INCLUDE_ENTITIES_LABEL = Component.translatable("structure_block.include_entities");
	private static final Component STRICT_LABEL = Component.translatable("structure_block.strict");
	private static final Component DETECT_SIZE_LABEL = Component.translatable("structure_block.detect_size");
	private static final Component SHOW_AIR_LABEL = Component.translatable("structure_block.show_air");
	private static final Component SHOW_BOUNDING_BOX_LABEL = Component.translatable("structure_block.show_boundingbox");
	private static final ImmutableList<StructureMode> ALL_MODES = ImmutableList.copyOf(StructureMode.values());
	private static final ImmutableList<StructureMode> DEFAULT_MODES = (ImmutableList<StructureMode>)ALL_MODES.stream()
		.filter(m -> m != StructureMode.DATA)
		.collect(ImmutableList.toImmutableList());
	private final StructureBlockEntity structure;
	private Mirror initialMirror = Mirror.NONE;
	private Rotation initialRotation = Rotation.NONE;
	private StructureMode initialMode = StructureMode.DATA;
	private boolean initialEntityIgnoring;
	private boolean initialStrict;
	private boolean initialShowAir;
	private boolean initialShowBoundingBox;
	private EditBox nameEdit;
	private EditBox posXEdit;
	private EditBox posYEdit;
	private EditBox posZEdit;
	private EditBox sizeXEdit;
	private EditBox sizeYEdit;
	private EditBox sizeZEdit;
	private EditBox integrityEdit;
	private EditBox seedEdit;
	private EditBox dataEdit;
	private Button saveButton;
	private Button loadButton;
	private Button rot0Button;
	private Button rot90Button;
	private Button rot180Button;
	private Button rot270Button;
	private Button detectButton;
	private CycleButton<Boolean> includeEntitiesButton;
	private CycleButton<Boolean> strictButton;
	private CycleButton<Mirror> mirrorButton;
	private CycleButton<Boolean> toggleAirButton;
	private CycleButton<Boolean> toggleBoundingBox;
	private final DecimalFormat decimalFormat = new DecimalFormat("0.0###", DecimalFormatSymbols.getInstance(Locale.ROOT));

	public StructureBlockEditScreen(final StructureBlockEntity structure) {
		super(Component.translatable(Blocks.STRUCTURE_BLOCK.getDescriptionId()));
		this.structure = structure;
	}

	private void onDone() {
		if (this.sendToServer(UpdateType.UPDATE_DATA)) {
			this.minecraft.gui.setScreen(null);
		}
	}

	private void onCancel() {
		this.structure.setMirror(this.initialMirror);
		this.structure.setRotation(this.initialRotation);
		this.structure.setMode(this.initialMode);
		this.structure.setIgnoreEntities(this.initialEntityIgnoring);
		this.structure.setStrict(this.initialStrict);
		this.structure.setShowAir(this.initialShowAir);
		this.structure.setShowBoundingBox(this.initialShowBoundingBox);
		this.minecraft.gui.setScreen(null);
	}

	@Override
	protected void init() {
		this.addRenderableWidget(Button.builder(CommonComponents.GUI_DONE, button -> this.onDone()).bounds(this.width / 2 - 4 - 150, 210, 150, 20).build());
		this.addRenderableWidget(Button.builder(CommonComponents.GUI_CANCEL, button -> this.onCancel()).bounds(this.width / 2 + 4, 210, 150, 20).build());
		this.initialMirror = this.structure.getMirror();
		this.initialRotation = this.structure.getRotation();
		this.initialMode = this.structure.getMode();
		this.initialEntityIgnoring = this.structure.isIgnoreEntities();
		this.initialStrict = this.structure.isStrict();
		this.initialShowAir = this.structure.getShowAir();
		this.initialShowBoundingBox = this.structure.getShowBoundingBox();
		this.saveButton = this.addRenderableWidget(Button.builder(Component.translatable("structure_block.button.save"), button -> {
			if (this.structure.getMode() == StructureMode.SAVE) {
				this.sendToServer(UpdateType.SAVE_AREA);
				this.minecraft.gui.setScreen(null);
			}
		}).bounds(this.width / 2 + 4 + 100, 185, 50, 20).build());
		this.loadButton = this.addRenderableWidget(Button.builder(Component.translatable("structure_block.button.load"), button -> {
			if (this.structure.getMode() == StructureMode.LOAD) {
				this.sendToServer(UpdateType.LOAD_AREA);
				this.minecraft.gui.setScreen(null);
			}
		}).bounds(this.width / 2 + 4 + 100, 185, 50, 20).build());
		this.addRenderableWidget(
			CycleButton.<StructureMode>builder(value -> Component.translatable("structure_block.mode." + value.getSerializedName()), this.initialMode)
				.withValues(DEFAULT_MODES, ALL_MODES)
				.displayOnlyValue()
				.create(this.width / 2 - 4 - 150, 185, 50, 20, Component.literal("MODE"), (button, value) -> {
					this.structure.setMode(value);
					this.updateMode(value);
				})
		);
		this.detectButton = this.addRenderableWidget(Button.builder(Component.translatable("structure_block.button.detect_size"), button -> {
			if (this.structure.getMode() == StructureMode.SAVE) {
				this.sendToServer(UpdateType.SCAN_AREA);
				this.minecraft.gui.setScreen(null);
			}
		}).bounds(this.width / 2 + 4 + 100, 120, 50, 20).build());
		this.includeEntitiesButton = this.addRenderableWidget(
			CycleButton.onOffBuilder(!this.structure.isIgnoreEntities())
				.displayOnlyValue()
				.create(this.width / 2 + 4 + 100, 160, 50, 20, INCLUDE_ENTITIES_LABEL, (button, value) -> this.structure.setIgnoreEntities(!value))
		);
		this.strictButton = this.addRenderableWidget(
			CycleButton.onOffBuilder(this.structure.isStrict())
				.displayOnlyValue()
				.create(this.width / 2 + 4 + 100, 120, 50, 20, STRICT_LABEL, (button, value) -> this.structure.setStrict(value))
		);
		this.mirrorButton = this.addRenderableWidget(
			CycleButton.builder(Mirror::symbol, this.initialMirror)
				.withValues(Mirror.values())
				.displayOnlyValue()
				.create(this.width / 2 - 20, 185, 40, 20, Component.literal("MIRROR"), (button, value) -> this.structure.setMirror(value))
		);
		this.toggleAirButton = this.addRenderableWidget(
			CycleButton.onOffBuilder(this.structure.getShowAir())
				.displayOnlyValue()
				.create(this.width / 2 + 4 + 100, 80, 50, 20, SHOW_AIR_LABEL, (button, value) -> this.structure.setShowAir(value))
		);
		this.toggleBoundingBox = this.addRenderableWidget(
			CycleButton.onOffBuilder(this.structure.getShowBoundingBox())
				.displayOnlyValue()
				.create(this.width / 2 + 4 + 100, 80, 50, 20, SHOW_BOUNDING_BOX_LABEL, (button, value) -> this.structure.setShowBoundingBox(value))
		);
		this.rot0Button = this.addRenderableWidget(Button.builder(Component.literal("0"), button -> {
			this.structure.setRotation(Rotation.NONE);
			this.updateDirectionButtons();
		}).bounds(this.width / 2 - 1 - 40 - 1 - 40 - 20, 185, 40, 20).build());
		this.rot90Button = this.addRenderableWidget(Button.builder(Component.literal("90"), button -> {
			this.structure.setRotation(Rotation.CLOCKWISE_90);
			this.updateDirectionButtons();
		}).bounds(this.width / 2 - 1 - 40 - 20, 185, 40, 20).build());
		this.rot180Button = this.addRenderableWidget(Button.builder(Component.literal("180"), button -> {
			this.structure.setRotation(Rotation.CLOCKWISE_180);
			this.updateDirectionButtons();
		}).bounds(this.width / 2 + 1 + 20, 185, 40, 20).build());
		this.rot270Button = this.addRenderableWidget(Button.builder(Component.literal("270"), button -> {
			this.structure.setRotation(Rotation.COUNTERCLOCKWISE_90);
			this.updateDirectionButtons();
		}).bounds(this.width / 2 + 1 + 40 + 1 + 20, 185, 40, 20).build());
		this.nameEdit = new EditBox(this.font, this.width / 2 - 152, 40, 300, 20, Component.translatable("structure_block.structure_name")) {
			{
				Objects.requireNonNull(StructureBlockEditScreen.this);
			}

			@Override
			public boolean charTyped(final CharacterEvent event) {
				return !StructureBlockEditScreen.this.isValidCharacterForName(this.getValue(), event.codepoint(), this.getCursorPosition())
					? false
					: super.charTyped(event);
			}
		};
		this.nameEdit.setMaxLength(128);
		this.nameEdit.setValue(this.structure.getStructureName());
		this.addWidget(this.nameEdit);
		BlockPos minPos = this.structure.getStructurePos();
		this.posXEdit = new EditBox(this.font, this.width / 2 - 152, 80, 80, 20, Component.translatable("structure_block.position.x"));
		this.posXEdit.setMaxLength(15);
		this.posXEdit.setValue(Integer.toString(minPos.getX()));
		this.addWidget(this.posXEdit);
		this.posYEdit = new EditBox(this.font, this.width / 2 - 72, 80, 80, 20, Component.translatable("structure_block.position.y"));
		this.posYEdit.setMaxLength(15);
		this.posYEdit.setValue(Integer.toString(minPos.getY()));
		this.addWidget(this.posYEdit);
		this.posZEdit = new EditBox(this.font, this.width / 2 + 8, 80, 80, 20, Component.translatable("structure_block.position.z"));
		this.posZEdit.setMaxLength(15);
		this.posZEdit.setValue(Integer.toString(minPos.getZ()));
		this.addWidget(this.posZEdit);
		Vec3i maxPos = this.structure.getStructureSize();
		this.sizeXEdit = new EditBox(this.font, this.width / 2 - 152, 120, 80, 20, Component.translatable("structure_block.size.x"));
		this.sizeXEdit.setMaxLength(15);
		this.sizeXEdit.setValue(Integer.toString(maxPos.getX()));
		this.addWidget(this.sizeXEdit);
		this.sizeYEdit = new EditBox(this.font, this.width / 2 - 72, 120, 80, 20, Component.translatable("structure_block.size.y"));
		this.sizeYEdit.setMaxLength(15);
		this.sizeYEdit.setValue(Integer.toString(maxPos.getY()));
		this.addWidget(this.sizeYEdit);
		this.sizeZEdit = new EditBox(this.font, this.width / 2 + 8, 120, 80, 20, Component.translatable("structure_block.size.z"));
		this.sizeZEdit.setMaxLength(15);
		this.sizeZEdit.setValue(Integer.toString(maxPos.getZ()));
		this.addWidget(this.sizeZEdit);
		this.integrityEdit = new EditBox(this.font, this.width / 2 - 152, 120, 80, 20, Component.translatable("structure_block.integrity.integrity"));
		this.integrityEdit.setMaxLength(15);
		this.integrityEdit.setValue(this.decimalFormat.format(this.structure.getIntegrity()));
		this.addWidget(this.integrityEdit);
		this.seedEdit = new EditBox(this.font, this.width / 2 - 72, 120, 80, 20, Component.translatable("structure_block.integrity.seed"));
		this.seedEdit.setMaxLength(31);
		this.seedEdit.setValue(Long.toString(this.structure.getSeed()));
		this.addWidget(this.seedEdit);
		this.dataEdit = new EditBox(this.font, this.width / 2 - 152, 120, 240, 20, Component.translatable("structure_block.custom_data"));
		this.dataEdit.setMaxLength(128);
		this.dataEdit.setValue(this.structure.getMetaData());
		this.addWidget(this.dataEdit);
		this.updateDirectionButtons();
		this.updateMode(this.initialMode);
	}

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

	@Override
	public void resize(final int width, final int height) {
		String oldNameEdit = this.nameEdit.getValue();
		String oldPosXEdit = this.posXEdit.getValue();
		String oldPosYEdit = this.posYEdit.getValue();
		String oldPosZEdit = this.posZEdit.getValue();
		String oldSizeXEdit = this.sizeXEdit.getValue();
		String oldSizeYEdit = this.sizeYEdit.getValue();
		String oldSizeZEdit = this.sizeZEdit.getValue();
		String oldIntegrityEdit = this.integrityEdit.getValue();
		String oldSeedEdit = this.seedEdit.getValue();
		String oldDataEdit = this.dataEdit.getValue();
		this.init(width, height);
		this.nameEdit.setValue(oldNameEdit);
		this.posXEdit.setValue(oldPosXEdit);
		this.posYEdit.setValue(oldPosYEdit);
		this.posZEdit.setValue(oldPosZEdit);
		this.sizeXEdit.setValue(oldSizeXEdit);
		this.sizeYEdit.setValue(oldSizeYEdit);
		this.sizeZEdit.setValue(oldSizeZEdit);
		this.integrityEdit.setValue(oldIntegrityEdit);
		this.seedEdit.setValue(oldSeedEdit);
		this.dataEdit.setValue(oldDataEdit);
	}

	private void updateDirectionButtons() {
		this.rot0Button.active = true;
		this.rot90Button.active = true;
		this.rot180Button.active = true;
		this.rot270Button.active = true;
		switch (this.structure.getRotation()) {
			case NONE:
				this.rot0Button.active = false;
				break;
			case CLOCKWISE_180:
				this.rot180Button.active = false;
				break;
			case COUNTERCLOCKWISE_90:
				this.rot270Button.active = false;
				break;
			case CLOCKWISE_90:
				this.rot90Button.active = false;
		}
	}

	private void updateMode(final StructureMode mode) {
		this.nameEdit.setVisible(false);
		this.posXEdit.setVisible(false);
		this.posYEdit.setVisible(false);
		this.posZEdit.setVisible(false);
		this.sizeXEdit.setVisible(false);
		this.sizeYEdit.setVisible(false);
		this.sizeZEdit.setVisible(false);
		this.integrityEdit.setVisible(false);
		this.seedEdit.setVisible(false);
		this.dataEdit.setVisible(false);
		this.saveButton.visible = false;
		this.loadButton.visible = false;
		this.detectButton.visible = false;
		this.includeEntitiesButton.visible = false;
		this.strictButton.visible = false;
		this.mirrorButton.visible = false;
		this.rot0Button.visible = false;
		this.rot90Button.visible = false;
		this.rot180Button.visible = false;
		this.rot270Button.visible = false;
		this.toggleAirButton.visible = false;
		this.toggleBoundingBox.visible = false;
		switch (mode) {
			case SAVE:
				this.nameEdit.setVisible(true);
				this.posXEdit.setVisible(true);
				this.posYEdit.setVisible(true);
				this.posZEdit.setVisible(true);
				this.sizeXEdit.setVisible(true);
				this.sizeYEdit.setVisible(true);
				this.sizeZEdit.setVisible(true);
				this.saveButton.visible = true;
				this.detectButton.visible = true;
				this.includeEntitiesButton.visible = true;
				this.strictButton.visible = false;
				this.toggleAirButton.visible = true;
				break;
			case LOAD:
				this.nameEdit.setVisible(true);
				this.posXEdit.setVisible(true);
				this.posYEdit.setVisible(true);
				this.posZEdit.setVisible(true);
				this.integrityEdit.setVisible(true);
				this.seedEdit.setVisible(true);
				this.loadButton.visible = true;
				this.includeEntitiesButton.visible = true;
				this.strictButton.visible = true;
				this.mirrorButton.visible = true;
				this.rot0Button.visible = true;
				this.rot90Button.visible = true;
				this.rot180Button.visible = true;
				this.rot270Button.visible = true;
				this.toggleBoundingBox.visible = true;
				this.updateDirectionButtons();
				break;
			case CORNER:
				this.nameEdit.setVisible(true);
				break;
			case DATA:
				this.dataEdit.setVisible(true);
		}
	}

	private boolean sendToServer(final UpdateType updateType) {
		BlockPos offset = new BlockPos(
			this.parseCoordinate(this.posXEdit.getValue()), this.parseCoordinate(this.posYEdit.getValue()), this.parseCoordinate(this.posZEdit.getValue())
		);
		Vec3i size = new Vec3i(
			this.parseCoordinate(this.sizeXEdit.getValue()), this.parseCoordinate(this.sizeYEdit.getValue()), this.parseCoordinate(this.sizeZEdit.getValue())
		);
		float integrity = this.parseIntegrity(this.integrityEdit.getValue());
		long seed = this.parseSeed(this.seedEdit.getValue());
		this.minecraft
			.getConnection()
			.send(
				new ServerboundSetStructureBlockPacket(
					this.structure.getBlockPos(),
					updateType,
					this.structure.getMode(),
					this.nameEdit.getValue(),
					offset,
					size,
					this.structure.getMirror(),
					this.structure.getRotation(),
					this.dataEdit.getValue(),
					this.structure.isIgnoreEntities(),
					this.structure.isStrict(),
					this.structure.getShowAir(),
					this.structure.getShowBoundingBox(),
					integrity,
					seed
				)
			);
		return true;
	}

	private long parseSeed(final String value) {
		try {
			return Long.valueOf(value);
		} catch (NumberFormatException var3) {
			return 0L;
		}
	}

	private float parseIntegrity(final String value) {
		try {
			return Float.valueOf(value);
		} catch (NumberFormatException var3) {
			return 1.0F;
		}
	}

	private int parseCoordinate(final String value) {
		try {
			return Integer.parseInt(value);
		} catch (NumberFormatException var3) {
			return 0;
		}
	}

	@Override
	public void onClose() {
		this.onCancel();
	}

	@Override
	public boolean keyPressed(final KeyEvent event) {
		if (super.keyPressed(event)) {
			return true;
		} else if (event.isConfirmation()) {
			this.onDone();
			return true;
		} else {
			return false;
		}
	}

	@Override
	public void extractRenderState(final GuiGraphicsExtractor graphics, final int mouseX, final int mouseY, final float a) {
		super.extractRenderState(graphics, mouseX, mouseY, a);
		StructureMode mode = this.structure.getMode();
		graphics.centeredText(this.font, this.title, this.width / 2, 10, -1);
		if (mode != StructureMode.DATA) {
			graphics.text(this.font, NAME_LABEL, this.width / 2 - 153, 30, -6250336);
			this.nameEdit.extractRenderState(graphics, mouseX, mouseY, a);
		}

		if (mode == StructureMode.LOAD || mode == StructureMode.SAVE) {
			graphics.text(this.font, POSITION_LABEL, this.width / 2 - 153, 70, -6250336);
			this.posXEdit.extractRenderState(graphics, mouseX, mouseY, a);
			this.posYEdit.extractRenderState(graphics, mouseX, mouseY, a);
			this.posZEdit.extractRenderState(graphics, mouseX, mouseY, a);
			graphics.text(this.font, INCLUDE_ENTITIES_LABEL, this.width / 2 + 154 - this.font.width(INCLUDE_ENTITIES_LABEL), 150, -6250336);
		}

		if (mode == StructureMode.SAVE) {
			graphics.text(this.font, SIZE_LABEL, this.width / 2 - 153, 110, -6250336);
			this.sizeXEdit.extractRenderState(graphics, mouseX, mouseY, a);
			this.sizeYEdit.extractRenderState(graphics, mouseX, mouseY, a);
			this.sizeZEdit.extractRenderState(graphics, mouseX, mouseY, a);
			graphics.text(this.font, DETECT_SIZE_LABEL, this.width / 2 + 154 - this.font.width(DETECT_SIZE_LABEL), 110, -6250336);
			graphics.text(this.font, SHOW_AIR_LABEL, this.width / 2 + 154 - this.font.width(SHOW_AIR_LABEL), 70, -6250336);
		}

		if (mode == StructureMode.LOAD) {
			graphics.text(this.font, INTEGRITY_LABEL, this.width / 2 - 153, 110, -6250336);
			this.integrityEdit.extractRenderState(graphics, mouseX, mouseY, a);
			this.seedEdit.extractRenderState(graphics, mouseX, mouseY, a);
			graphics.text(this.font, STRICT_LABEL, this.width / 2 + 154 - this.font.width(STRICT_LABEL), 110, -6250336);
			graphics.text(this.font, SHOW_BOUNDING_BOX_LABEL, this.width / 2 + 154 - this.font.width(SHOW_BOUNDING_BOX_LABEL), 70, -6250336);
		}

		if (mode == StructureMode.DATA) {
			graphics.text(this.font, CUSTOM_DATA_LABEL, this.width / 2 - 153, 110, -6250336);
			this.dataEdit.extractRenderState(graphics, mouseX, mouseY, a);
		}

		graphics.text(this.font, mode.getDisplayName(), this.width / 2 - 153, 174, -6250336);
	}

	@Override
	public boolean isPauseScreen() {
		return false;
	}

	@Override
	public boolean isInGameUi() {
		return true;
	}
}