package io.github.skippyall.minions.input;

import eu.pb4.sgui.api.elements.GuiElementBuilder;
import eu.pb4.sgui.api.gui.AnvilInputGui;
import net.minecraft.item.Items;
import net.minecraft.screen.AnvilScreenHandler;
import net.minecraft.server.network.ServerPlayerEntity;
import net.minecraft.text.Text;

import java.util.concurrent.CompletableFuture;
import java.util.function.Function;
import java.util.function.Predicate;

public class TextInput<T> extends AnvilInputGui {
    private final GuiElementBuilder valid = new GuiElementBuilder()
            .setItem(Items.EMERALD_BLOCK)
            .setName(Text.literal("OK"))
            .setCallback(this::onConfirm);

    private final GuiElementBuilder invalid = new GuiElementBuilder()
            .setItem(Items.REDSTONE_BLOCK);
    private final Function<String, CompletableFuture<Result<T, Text>>> parser;
    private final CompletableFuture<T> future;
    private Result<T, Text> result;
    private boolean isConfirm;

    public TextInput(ServerPlayerEntity player, Text title, String defaultValue, Function<String, CompletableFuture<Result<T, Text>>> parser, CompletableFuture<T> future) {
        super(player, false);
        setTitle(title);
        setDefaultInputValue(defaultValue);
        this.parser = parser;
        this.future = future;

        updateConfirmButton(defaultValue);
    }

    public static <T> CompletableFuture<T> inputSync(ServerPlayerEntity player, Text title, String defaultValue, Function<String, Result<T, Text>> parser) {
        return input(player, title, defaultValue, (String string) -> CompletableFuture.completedFuture(parser.apply(string)));
    }

    public static <T> CompletableFuture<T> input(ServerPlayerEntity player, Text title, String defaultValue, Function<String, CompletableFuture<Result<T, Text>>> parser) {
        CompletableFuture<T> future = new CompletableFuture<>();
        new TextInput<>(player, title, defaultValue, parser, future).open();
        return future;
    }

    public static CompletableFuture<String> inputString(ServerPlayerEntity player, Text title, String defaultValue) {
        return inputSync(player, title, defaultValue, Result.Success::new);
    }

    public static CompletableFuture<Integer> inputInt(ServerPlayerEntity player, Text title, String defaultValue) {
        return inputSync(player, title, defaultValue, string -> Result.wrapCustomError(() -> Integer.valueOf(string), Text.translatable("minions.command.input.int.fail")));
    }

    public static CompletableFuture<Float> inputFloat(ServerPlayerEntity player, Text title, String defaultValue) {
        return inputSync(player, title, defaultValue, string -> Result.wrapCustomError(() -> Float.valueOf(string), Text.translatable("minions.command.input.float.fail")));
    }

    @Override
    public void onInput(String input) {
        updateConfirmButton(input);
    }

    public void updateConfirmButton(String input) {
        parser.apply(input).thenAccept(result -> {
            this.result = result;
            if(result.isSuccess()) {
                setSlot(AnvilScreenHandler.OUTPUT_ID, valid);
            } else {
                Text text = result.getErrorOrThrow();
                setSlot(AnvilScreenHandler.OUTPUT_ID, invalid.setName(text));
            }
        });
    }

    @Override
    public void onClose() {
        if(!future.isDone() && !isConfirm) {
            future.cancel(false);
        }
    }

    public void onConfirm() {
        if(result != null) {
            result.ifSuccess(success -> {
                isConfirm = true;
                close();
                future.complete(success);
            });
        }
    }
}