/*
 * Decompiled with CFR 0.152.
 */
package net.minecraft.client.gui.screens.dialog;

import java.util.HashMap;
import java.util.Map;
import java.util.Optional;
import java.util.function.Consumer;
import java.util.function.Supplier;
import net.minecraft.client.gui.components.Button;
import net.minecraft.client.gui.components.Tooltip;
import net.minecraft.client.gui.layouts.LayoutElement;
import net.minecraft.client.gui.screens.dialog.DialogScreen;
import net.minecraft.client.gui.screens.dialog.input.InputControlHandlers;
import net.minecraft.network.chat.ClickEvent;
import net.minecraft.server.dialog.ActionButton;
import net.minecraft.server.dialog.CommonButtonData;
import net.minecraft.server.dialog.Input;
import net.minecraft.server.dialog.action.Action;

public class DialogControlSet {
    public static final Supplier<Optional<ClickEvent>> EMPTY_ACTION = Optional::empty;
    private final DialogScreen<?> screen;
    private final Map<String, Action.ValueGetter> valueGetters = new HashMap<String, Action.ValueGetter>();

    public DialogControlSet(DialogScreen<?> screen) {
        this.screen = screen;
    }

    public void addInput(Input data, Consumer<LayoutElement> output) {
        String key = data.key();
        InputControlHandlers.createHandler(data.control(), this.screen, (element, valueGetter) -> {
            this.valueGetters.put(key, valueGetter);
            output.accept(element);
        });
    }

    private static Button.Builder createDialogButton(CommonButtonData data, Button.OnPress clickAction) {
        Button.Builder result = Button.builder(data.label(), clickAction);
        result.width(data.width());
        if (data.tooltip().isPresent()) {
            result = result.tooltip(Tooltip.create(data.tooltip().get()));
        }
        return result;
    }

    public Supplier<Optional<ClickEvent>> bindAction(Optional<Action> maybeAction) {
        if (maybeAction.isPresent()) {
            Action action = maybeAction.get();
            return () -> action.createAction(this.valueGetters);
        }
        return EMPTY_ACTION;
    }

    public Button.Builder createActionButton(ActionButton actionButton) {
        Supplier<Optional<ClickEvent>> action = this.bindAction(actionButton.action());
        return DialogControlSet.createDialogButton(actionButton.button(), button -> this.screen.runAction((Optional)action.get()));
    }
}