package moe.oko.Kiafumi.command.utility;

import moe.oko.Kiafumi.command.CommandClass;
import moe.oko.Kiafumi.util.CommandInfo;
import moe.oko.Kiafumi.util.CommandType;
import moe.oko.Kiafumi.util.EmbedUI;
import net.dv8tion.jda.api.EmbedBuilder;
import net.dv8tion.jda.api.events.interaction.command.SlashCommandInteractionEvent;
import net.dv8tion.jda.api.interactions.commands.OptionType;
import org.bitnick.net.duckduckgo.WebSearch;
import org.bitnick.net.duckduckgo.entity.SearchResult;

import java.time.ZonedDateTime;
import java.util.ArrayList;
import java.util.List;

import static moe.oko.Kiafumi.util.LoggingManager.slashLog;

/**
 * Helpful Search Command (Uses DDG API)
 * @author Kay
 */
public class DuckCommand extends CommandClass {
    private boolean enabled = true;

    @Override
    public boolean isEnabled() {
        return enabled;
    }

    @Override
    public String getName() {
        return "Duck";
    }

    @Override
    public void newCommand(String name, SlashCommandInteractionEvent e) {
        if ("search".equals(name)) {
            e.deferReply().queue();
            String option = e.getOption("query").getAsString();
            slashLog(e, "with search \"" + option + "\".");
            WebSearch ws = WebSearch.instanceOf();
            SearchResult sr;
            try {
                sr = ws.instantAnswerSearch(option);
            } catch (Exception ex) {
                EmbedBuilder eb = new EmbedBuilder()
                        .setColor(EmbedUI.FAILURE)
                        .setTitle("Query Failed")
                        .setDescription("Couldn't find any results.")
                        .setFooter("Query: " + option)
                        .setTimestamp(ZonedDateTime.now());
                e.getHook().sendMessageEmbeds(eb.build()).queue();
                return;
            }
            EmbedBuilder eb = new EmbedBuilder()
                    .setColor(EmbedUI.INFO)
                    .setTitle(sr.getTitle(), sr.getUrl().toString())
                    .setDescription(sr.getDescription())
                    .setFooter("Query: " + option)
                    .setTimestamp(ZonedDateTime.now());
            e.getHook().sendMessageEmbeds(eb.build()).queue();
        }
    }

    @Override
    public List<CommandInfo> getSlashCommandInfo() {
        List<CommandInfo> cil = new ArrayList<>();
        CommandInfo ci = new CommandInfo("search", "Looks up with DuckDuckGo your query!", CommandType.COMMAND);
        ci.addOption("query", "The query to be searched", OptionType.STRING, true);
        cil.add(ci);
        return cil;
    }
}