package com.xxdraggy.utils.input;

import com.comphenix.protocol.PacketType;
import com.comphenix.protocol.ProtocolLibrary;
import com.comphenix.protocol.ProtocolManager;
import com.comphenix.protocol.events.PacketAdapter;
import com.comphenix.protocol.events.PacketEvent;
import org.bukkit.Bukkit;
import org.bukkit.Location;
import org.bukkit.entity.Player;
import org.bukkit.plugin.java.JavaPlugin;

import java.util.HashMap;
import java.util.Map;
import java.util.logging.Level;

public class InputManager {
    private static JavaPlugin plugin;
    private static ProtocolManager manager;
    public static ProtocolManager getManager() {
        return InputManager.manager;
    }

    private static final Map<Player, InputGui> currentInputs = new HashMap<>();
    public static Map<Player, InputGui> getCurrentInputs() {
        return currentInputs;
    }

    public static void register(JavaPlugin plugin) {
        InputManager.plugin = plugin;
        InputManager.manager = ProtocolLibrary.getProtocolManager();

        Bukkit.getLogger().log(Level.INFO, "[BukkitUtils/InputController] Registered listener!");

        if(InputManager.isInstalled()) {
            InputManager.listen();
        }
        else {
            Bukkit.getLogger().log(Level.SEVERE, "[BukkitUtils/InputController] Could not find ProtocolLib in the plugin folder.\nPlease install it from https://github.com/dmulloy2/ProtocolLib");
        }
    }

    public static boolean isInstalled() {
        return ProtocolLibrary.getProtocolManager() != null;
    }

    private static void listen() {
        InputManager.getManager().addPacketListener(new PacketAdapter(plugin, PacketType.Play.Client.UPDATE_SIGN) {
            @Override
            public void onPacketReceiving(PacketEvent event) {
                Player player = event.getPlayer();
                InputGui gui = currentInputs.remove(player);

                if (gui == null) {
                    return;
                }

                event.setCancelled(true);

                boolean success = gui.getChecker().test(player, event.getPacket().getStringArrays().read(0));

                if (!success && gui.isReopen() && !gui.isForceClose()) {
                    Bukkit.getScheduler().runTaskLater(plugin, () -> gui.open(player), 2L);
                }
                else {
                    gui.getCallback().apply(player, event.getPacket().getStringArrays().read(0));
                }

                Bukkit.getScheduler().runTaskLater(plugin, () -> {
                    if (player.isOnline()) {
                        Location location = gui.getBlockPosition().toLocation(player.getWorld());

                        player.sendBlockChange(location, location.getBlock().getBlockData());
                    }
                }, 2L);
            }
        });
    }
}

package com.xxdraggy.utils.input;

import com.comphenix.protocol.PacketType;
import com.comphenix.protocol.events.PacketContainer;
import com.comphenix.protocol.wrappers.BlockPosition;
import com.comphenix.protocol.wrappers.nbt.NbtCompound;
import com.xxdraggy.utils.data.SignType;
import org.bukkit.ChatColor;
import org.bukkit.Location;
import org.bukkit.entity.Player;

import java.util.ArrayList;
import java.util.List;
import java.util.Objects;
import java.util.function.BiFunction;
import java.util.function.BiPredicate;

public class InputGui {
    private SignType signType = SignType.Birch;
    public void setSignType(SignType signType) {
        this.signType = signType;
    }
    public SignType getSignType() {
        return signType;
    }

    private List<String> lines = new ArrayList<>();
    public void setLines(List<String> lines) {
        this.lines = lines;
    }
    public List<String> getLines() {
        return lines;
    }

    private BiFunction<Player, String[], Void> callback = (player, strings) -> null;
    public void setCallback(BiFunction<Player, String[], Void> callback) {
        this.callback = callback;
    }
    public BiFunction<Player, String[], Void> getCallback() {
        return callback;
    }

    private BiPredicate<Player, String[]> checker = (player, strings) -> true;
    public void setChecker(BiPredicate<Player, String[]> checker) {
        this.checker = checker;
    }
    public BiPredicate<Player, String[]> getChecker() {
        return checker;
    }

    private boolean reopen = true;
    public void setReopen(boolean reopen) {
        this.reopen = reopen;
    }
    public boolean isReopen() {
        return reopen;
    }

    private boolean forceClose;
    public void setForceClose(boolean forceClose) {
        this.forceClose = forceClose;
    }
    public boolean isForceClose() {
        return forceClose;
    }

    private BlockPosition blockPosition;
    public BlockPosition getBlockPosition() {
        return blockPosition;
    }

    public void open(Player player) {
        Objects.requireNonNull(player, "Player must be given!");

        if (!player.isOnline())
            return;

        PacketContainer openSign = InputManager.getManager().createPacket(PacketType.Play.Server.OPEN_SIGN_EDITOR);
        PacketContainer signData = InputManager.getManager().createPacket(PacketType.Play.Server.TILE_ENTITY_DATA);

        Location playerLocation = player.getLocation();

        this.blockPosition = new BlockPosition(
                playerLocation.getBlockX(),
                playerLocation.getBlockY()
                        +
                        (255 - playerLocation.getBlockY()),
                playerLocation.getBlockZ());

        player.sendBlockChange(this.blockPosition.toLocation(playerLocation.getWorld()), signType.getBlock().createBlockData());

        openSign.getBlockPositionModifier().write(0, this.blockPosition);

        NbtCompound signNBT = (NbtCompound) signData.getNbtModifier().read(0);

        for (int line = 0; line < 4; line++) {
            signNBT.put(
                    "text"
                    +
                    (line + 1),
                    this.lines.size() > line
                    ?
                    String.format("{\"text\":\"%s\"}", ChatColor.translateAlternateColorCodes('&', this.lines.get(line)))
                    :
                    ""
            );
        }

        signNBT.put("x", this.blockPosition.getX());
        signNBT.put("y", this.blockPosition.getY());
        signNBT.put("z", this.blockPosition.getZ());
        signNBT.put("id", signType.getId());

        signData.getBlockPositionModifier().write(0, this.blockPosition);
        signData.getNbtModifier().write(0, signNBT);

        InputManager.getManager().sendServerPacket(player, signData);
        InputManager.getManager().sendServerPacket(player, openSign);

        InputManager.getCurrentInputs().put(player, this);
    }

    public void close(Player player) {
        this.close(player, false);
    }

    public void close(Player player, boolean forceClose) {
        this.setForceClose(forceClose);

        if (player.isOnline())
            player.closeInventory();
    }
}