Quests/core/src/main/java/me/pikamug/quests/convo/misc/QuestAcceptPrompt.java

package me.pikamug.quests.convo.misc;

import me.pikamug.quests.BukkitQuestsPlugin;
import me.pikamug.quests.player.Quester;
import me.pikamug.quests.util.BukkitLang;
import net.md_5.bungee.api.chat.ClickEvent;
import net.md_5.bungee.api.chat.TextComponent;
import org.browsit.conversations.api.Conversations;
import org.bukkit.Bukkit;
import org.bukkit.ChatColor;
import org.bukkit.entity.Player;
import org.jetbrains.annotations.NotNull;

import java.util.UUID;

public class QuestAcceptPrompt {

    private final UUID uuid;
    private final BukkitQuestsPlugin plugin;

    public QuestAcceptPrompt(final UUID uuid, BukkitQuestsPlugin plugin) {
        super();
        this.uuid = uuid;
        this.plugin = plugin;
    }

    public int getSize() {
        return 2;
    }

    public String getTitle() {
        return null;
    }

    @SuppressWarnings("unused")
    public ChatColor getNumberColor(final int number) {
        switch (number) {
            case 1:
                return ChatColor.GREEN;
            case 2:
                return ChatColor.RED;
            default:
                return null;
        }
    }

    @SuppressWarnings("unused")
    public String getSelectionText(final int number) {
        switch (number) {
            case 1:
                return ChatColor.GREEN + BukkitLang.get("yesWord");
            case 2:
                return ChatColor.RED + BukkitLang.get("noWord");
            default:
                return null;
        }
    }

    public String getQueryText() {
        return BukkitLang.get("acceptQuest");
    }

    public @NotNull String getPromptText(final @NotNull UUID uuid) {
        if (plugin == null) {
            return ChatColor.YELLOW + BukkitLang.get("itemCreateCriticalError");
        }

        // TODO resolve
        /*final MiscPostQuestAcceptEvent event = new MiscPostQuestAcceptEvent(this);
        plugin.getServer().getPluginManager().callEvent(event);*/

        if (!plugin.getConfigSettings().canClickablePrompts()) {
            return ChatColor.YELLOW + getQueryText() + "  " + ChatColor.GREEN
                    + getSelectionText(1) + ChatColor.RESET + " / " + getSelectionText(2);
        }

        final TextComponent component = new TextComponent("");
        component.addExtra(ChatColor.YELLOW + getQueryText() + "  " + ChatColor.GREEN);
        final TextComponent yes = new TextComponent(getSelectionText(1));
        yes.setClickEvent(new ClickEvent(ClickEvent.Action.RUN_COMMAND, "/quests choice " + BukkitLang.get("yesWord")));
        component.addExtra(yes);
        component.addExtra(ChatColor.RESET + " / ");
        final TextComponent no = new TextComponent(getSelectionText(2));
        no.setClickEvent(new ClickEvent(ClickEvent.Action.RUN_COMMAND, "/quests choice " + BukkitLang.get("noWord")));
        component.addExtra(no);

        Bukkit.getPlayer(uuid).spigot().sendMessage(component);
        return "";
    }

    public void acceptInput(final @NotNull UUID uuid, final String input) {
        if (plugin == null || input == null) {
            return;
        }
        final Player player = Bukkit.getPlayer(uuid);
        if (input.equalsIgnoreCase("1") || input.equalsIgnoreCase("y")
                || input.equalsIgnoreCase(BukkitLang.get("yesWord"))
                || input.equalsIgnoreCase(BukkitLang.get(player, "yesWord"))) {
            final Quester quester = plugin.getQuester(uuid);
            if (quester == null) {
                plugin.getLogger().info("Ended conversation because quester for " + player.getName() + "was null");
                return;
            }
            final String questIdToTake = quester.getQuestIdToTake();
            if (plugin.getQuestById(questIdToTake) == null) {
                plugin.getLogger().warning(player.getName() + " attempted to take quest ID \"" + questIdToTake
                        + "\" but something went wrong");
                player.sendMessage(ChatColor.RED
                        + "Something went wrong! Please report issue to an administrator.");
            } else {
                quester.takeQuest(plugin.getQuestById(questIdToTake), false);
            }
        } else if (input.equalsIgnoreCase("2") || input.equalsIgnoreCase("n")
                || input.equalsIgnoreCase(BukkitLang.get("noWord"))
                || input.equalsIgnoreCase(BukkitLang.get(player, "noWord"))) {
            BukkitLang.send(player, ChatColor.YELLOW + BukkitLang.get("cancelled"));
        } else {
            final String msg = BukkitLang.get(player, "questInvalidChoice")
                    .replace("<yes>", BukkitLang.get(player, "yesWord"))
                    .replace("<no>", BukkitLang.get(player, "noWord"));
            BukkitLang.send(player, ChatColor.RED + msg);
        }
    }

    public void start() {
        Conversations.create(uuid)
                .prompt(getPromptText(uuid), String.class, prompt -> prompt
                .converter(String::valueOf)
                .fetch((input, sender) -> acceptInput(uuid, input)))
                .start();
    }
}