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

import java.util.List;
import javax.annotation.Nullable;
import net.minecraft.client.gui.GuiGraphics;
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.core.BlockPos;
import net.minecraft.network.chat.CommonComponents;
import net.minecraft.network.chat.Component;
import net.minecraft.network.protocol.game.ServerboundSetTestBlockPacket;
import net.minecraft.world.level.block.Blocks;
import net.minecraft.world.level.block.entity.TestBlockEntity;
import net.minecraft.world.level.block.state.properties.TestBlockMode;
import net.minecraftforge.api.distmarker.Dist;
import net.minecraftforge.api.distmarker.OnlyIn;

@OnlyIn(Dist.CLIENT)
public class TestBlockEditScreen extends Screen {
    private static final List<TestBlockMode> MODES = List.of(TestBlockMode.values());
    private static final Component TITLE = Component.translatable(Blocks.TEST_BLOCK.getDescriptionId());
    private static final Component MESSAGE_LABEL = Component.translatable("test_block.message");
    private final BlockPos position;
    private TestBlockMode mode;
    private String message;
    @Nullable
    private EditBox messageEdit;

    public TestBlockEditScreen(TestBlockEntity p_394367_) {
        super(TITLE);
        this.position = p_394367_.getBlockPos();
        this.mode = p_394367_.getMode();
        this.message = p_394367_.getMessage();
    }

    @Override
    public void init() {
        this.messageEdit = new EditBox(this.font, this.width / 2 - 152, 80, 240, 20, Component.translatable("test_block.message"));
        this.messageEdit.setMaxLength(128);
        this.messageEdit.setValue(this.message);
        this.addRenderableWidget(this.messageEdit);
        this.setInitialFocus(this.messageEdit);
        this.updateMode(this.mode);
        this.addRenderableWidget(
            CycleButton.builder(TestBlockMode::getDisplayName)
                .withValues(MODES)
                .displayOnlyValue()
                .withInitialValue(this.mode)
                .create(this.width / 2 - 4 - 150, 185, 50, 20, TITLE, (p_396452_, p_393954_) -> this.updateMode(p_393954_))
        );
        this.addRenderableWidget(
            Button.builder(CommonComponents.GUI_DONE, p_392720_ -> this.onDone()).bounds(this.width / 2 - 4 - 150, 210, 150, 20).build()
        );
        this.addRenderableWidget(Button.builder(CommonComponents.GUI_CANCEL, p_391429_ -> this.onCancel()).bounds(this.width / 2 + 4, 210, 150, 20).build());
    }

    @Override
    public void render(GuiGraphics p_393246_, int p_397000_, int p_391691_, float p_397480_) {
        super.render(p_393246_, p_397000_, p_391691_, p_397480_);
        p_393246_.drawCenteredString(this.font, this.title, this.width / 2, 10, 16777215);
        if (this.mode != TestBlockMode.START) {
            p_393246_.drawString(this.font, MESSAGE_LABEL, this.width / 2 - 153, 70, 10526880);
        }

        p_393246_.drawString(this.font, this.mode.getDetailedMessage(), this.width / 2 - 153, 174, 10526880);
    }

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

    private void onDone() {
        this.message = this.messageEdit.getValue();
        this.minecraft.getConnection().send(new ServerboundSetTestBlockPacket(this.position, this.mode, this.message));
        this.onClose();
    }

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

    private void onCancel() {
        this.minecraft.setScreen(null);
    }

    private void updateMode(TestBlockMode p_396502_) {
        this.mode = p_396502_;
        this.messageEdit.visible = p_396502_ != TestBlockMode.START;
    }

    @Override
    public void renderBackground(GuiGraphics p_397094_, int p_395762_, int p_391364_, float p_394776_) {
        this.renderTransparentBackground(p_397094_);
    }
}