package net.minecraft.server.dialog;

import com.mojang.serialization.MapCodec;
import com.mojang.serialization.codecs.RecordCodecBuilder;
import java.util.List;
import java.util.Optional;
import net.minecraft.server.dialog.action.Action;

public record ConfirmationDialog(@Override CommonDialogData common, ActionButton yesButton, ActionButton noButton) implements SimpleDialog {
    public static final MapCodec<ConfirmationDialog> MAP_CODEC = RecordCodecBuilder.mapCodec(
        instance -> instance.group(
                CommonDialogData.MAP_CODEC.forGetter(ConfirmationDialog::common),
                ActionButton.CODEC.fieldOf("yes").forGetter(ConfirmationDialog::yesButton),
                ActionButton.CODEC.fieldOf("no").forGetter(ConfirmationDialog::noButton)
            )
            .apply(instance, ConfirmationDialog::new)
    );

    @Override
    public MapCodec<ConfirmationDialog> codec() {
        return MAP_CODEC;
    }

    @Override
    public Optional<Action> onCancel() {
        return this.noButton.action();
    }

    @Override
    public List<ActionButton> mainActions() {
        return List.of(this.yesButton, this.noButton);
    }
}
