Make urban command support multiple entries

2022-12-20 00:34:49 +01:00
parent 0f54211ecd
commit 4ef42ffa9e
4 changed files with 423 additions and 21 deletions
--- a/src/main/java/wtf/beatrice/hidekobot/commands/base/CoinFlip.java
+++ b/src/main/java/wtf/beatrice/hidekobot/commands/base/CoinFlip.java
@@ -61,8 +61,6 @@ public class CoinFlip

    public static void trackAndRestrict(Message replyMessage, User user)
    {
-        String replyMessageId = replyMessage.getId();
-
        Cache.getDatabaseSource().queueDisabling(replyMessage);
        Cache.getDatabaseSource().trackRanCommandReply(replyMessage, user);
    }
--- a/src/main/java/wtf/beatrice/hidekobot/commands/message/UrbanDictionaryCommand.java
+++ b/src/main/java/wtf/beatrice/hidekobot/commands/message/UrbanDictionaryCommand.java
@@ -2,7 +2,16 @@ package wtf.beatrice.hidekobot.commands.message;

 import net.dv8tion.jda.api.EmbedBuilder;
 import net.dv8tion.jda.api.Permission;
+import net.dv8tion.jda.api.entities.MessageEmbed;
+import net.dv8tion.jda.api.entities.User;
+import net.dv8tion.jda.api.entities.emoji.Emoji;
+import net.dv8tion.jda.api.events.interaction.component.ButtonInteractionEvent;
 import net.dv8tion.jda.api.events.message.MessageReceivedEvent;
+import net.dv8tion.jda.api.interactions.components.ActionRow;
+import net.dv8tion.jda.api.interactions.components.ItemComponent;
+import net.dv8tion.jda.api.interactions.components.buttons.Button;
+import org.apache.commons.lang3.SerializationException;
+import org.apache.commons.text.StringEscapeUtils;
 import org.apache.commons.text.WordUtils;
 import org.jetbrains.annotations.Nullable;
 import org.jsoup.Jsoup;
@@ -10,13 +19,11 @@ import org.jsoup.nodes.Document;
 import org.jsoup.nodes.Element;
 import org.jsoup.select.Elements;
 import wtf.beatrice.hidekobot.Cache;
+import wtf.beatrice.hidekobot.datasources.DatabaseSource;
 import wtf.beatrice.hidekobot.objects.commands.MessageCommand;

-import java.io.IOException;
-import java.util.ArrayList;
-import java.util.Arrays;
-import java.util.LinkedList;
-import java.util.List;
+import java.io.*;
+import java.util.*;

 public class UrbanDictionaryCommand implements MessageCommand
 {
@@ -37,6 +44,34 @@ public class UrbanDictionaryCommand implements MessageCommand
        return false;
    }

+    static final String baseUrl = "https://www.urbandictionary.com/define.php?term=";
+    static final Button previousPageButton = Button.primary("urban_previouspage", "Back")
+            .withEmoji(Emoji.fromFormatted("⬅️"));
+    static final Button nextPageButton = Button.primary("urban_nextpage", "Next")
+            .withEmoji(Emoji.fromFormatted("➡️"));
+
+    private static MessageEmbed buildEmbed(String term,
+                                           String url,
+                                           User author,
+                                           String meaning,
+                                           String example,
+                                           String contributor,
+                                           String date,
+                                           int page)
+    {
+        EmbedBuilder embedBuilder = new EmbedBuilder();
+        embedBuilder.setColor(Cache.getBotColor());
+        embedBuilder.setTitle(term + ", on Urban Dictionary", url);
+        embedBuilder.setAuthor(author.getAsTag(), null, author.getAvatarUrl());
+        embedBuilder.addField("Definition", meaning, false);
+        embedBuilder.addField("Example", example, false);
+        embedBuilder.addField("Submission",
+                "*Entry " + (page+1) + " | Sent by " + contributor + " on " + date + "*",
+                false);
+
+        return embedBuilder.build();
+    }
+
    @Override
    public void runCommand(MessageReceivedEvent event, String label, String[] args)
    {
@@ -65,9 +100,9 @@ public class UrbanDictionaryCommand implements MessageCommand
        // cut it to length to avoid abuse
        if (term.length() > 64) term = term.substring(0, 64);

-        String url = "https://www.urbandictionary.com/define.php?term=" + term;
+        String url = baseUrl + term;

-        Document doc = null;
+        Document doc;

        try {
            doc = Jsoup.connect(url).get();
@@ -76,10 +111,10 @@ public class UrbanDictionaryCommand implements MessageCommand
            return;
        }

-        List<String> contributorsNames = new ArrayList<>();
-        List<String> submissionDates = new ArrayList<>();
        List<String> plaintextMeanings = new ArrayList<>();
        List<String> plaintextExamples = new ArrayList<>();
+        List<String> contributorsNames = new ArrayList<>();
+        List<String> submissionDates = new ArrayList<>();

        Elements definitions = doc.getElementsByClass("definition");
        for(Element definition : definitions)
@@ -94,6 +129,8 @@ public class UrbanDictionaryCommand implements MessageCommand
                String text = meaning.html()
                        .replaceAll("<br\\s*?>", "\n") // keep newlines
                        .replaceAll("<.*?>", ""); // remove all other html tags
+                // this is used to fix eg. &amp; being shown literally instead of being parsed
+                text = StringEscapeUtils.unescapeHtml4(text);
                // discord only allows 1024 characters for embed fields
                if(text.length() > 1024) text = text.substring(0, 1020) + "...";
                plaintextMeanings.add(text);
@@ -110,6 +147,8 @@ public class UrbanDictionaryCommand implements MessageCommand
                String text = example.html()
                        .replaceAll("<br\\s*?>", "\n") // keep newlines
                        .replaceAll("<.*?>", ""); // remove all other html tags
+                // this is used to fix eg. &amp; being shown literally instead of being parsed
+                text = StringEscapeUtils.unescapeHtml4(text);
                // discord only allows 1024 characters for embed fields
                if(text.length() > 1024) text = text.substring(0, 1020) + "...";
                plaintextExamples.add(text);
@@ -140,16 +179,134 @@ public class UrbanDictionaryCommand implements MessageCommand
        term = term.replaceAll("\\+", " ");
        term = WordUtils.capitalizeFully(term);

-        EmbedBuilder embedBuilder = new EmbedBuilder();
-        embedBuilder.setColor(Cache.getBotColor());
-        embedBuilder.setTitle(term + ", on Urban Dictionary", url);
-        embedBuilder.setAuthor(event.getAuthor().getAsTag(), null, event.getAuthor().getAvatarUrl());
-        embedBuilder.addField("Definition", plaintextMeanings.get(0), false);
-        embedBuilder.addField("Example", plaintextExamples.get(0), false);
-        embedBuilder.addField("Submission",
-                "*sent by " + contributorsNames.get(0) + " on " + submissionDates.get(0) + "*",
-                false);

-        event.getChannel().sendMessageEmbeds(embedBuilder.build()).queue();
+
+        String serializedMeanings = serialize(plaintextMeanings);
+        String serializedExamples = serialize(plaintextExamples);
+        String serializedContributors = serialize(contributorsNames);
+        String serializedDates = serialize(submissionDates);
+
+        // disable next page if we only have one result
+        Button nextPageBtnLocal = nextPageButton;
+        if(submissionDates.size() == 1) nextPageBtnLocal = nextPageBtnLocal.asDisabled();
+
+        MessageEmbed embed = buildEmbed(term, url, event.getAuthor(), plaintextMeanings.get(0),
+                plaintextExamples.get(0), contributorsNames.get(0), submissionDates.get(0), 0);
+
+        // copy term for async thing
+        final String finalTerm = term;
+        event.getChannel()
+                .sendMessageEmbeds(embed)
+                .addActionRow(previousPageButton.asDisabled(), //disabled by default because we're on page 0
+                        nextPageBtnLocal)
+                .queue(message ->
+        {
+
+            Cache.getDatabaseSource().queueDisabling(message);
+            Cache.getDatabaseSource().trackRanCommandReply(message, event.getAuthor());
+            Cache.getDatabaseSource().trackUrban(serializedMeanings,
+                    serializedExamples,
+                    serializedContributors,
+                    serializedDates,
+                    message,
+                    finalTerm);
+        });
+
+    }
+
+    private String serialize(List dataList) {
+
+        try (ByteArrayOutputStream bo = new ByteArrayOutputStream();
+             ObjectOutputStream so = new ObjectOutputStream(bo)) {
+            so.writeObject(dataList);
+            so.flush();
+            return Base64.getEncoder().encodeToString(bo.toByteArray());
+        }
+        catch (IOException ignored) {}
+        return null;
+    }
+
+    private static ArrayList deserialize(String dataStr) {
+
+        byte[] b = Base64.getDecoder().decode(dataStr);
+        ByteArrayInputStream bi = new ByteArrayInputStream(b);
+        ObjectInputStream si;
+        try {
+            si = new ObjectInputStream(bi);
+            return ArrayList.class.cast(si.readObject());
+        }
+        catch (IOException | ClassNotFoundException e) {
+            throw new SerializationException("Error during deserialization", e);
+        }
+    }
+
+
+    public static void changePage(ButtonInteractionEvent event, boolean increase)
+    {
+
+
+        String messageId = event.getMessageId();
+        DatabaseSource database = Cache.getDatabaseSource();
+
+        // check if the user interacting is the same one who ran the command
+        if (!(database.isUserTrackedFor(event.getUser().getId(), messageId))) {
+            event.reply("❌ You did not run this command!").setEphemeral(true).queue();
+            return;
+        }
+
+        // get current page and calculate how many pages there are
+        int page = Cache.getDatabaseSource().getUrbanPage(messageId);
+        int pages;
+
+
+        String serializedMeanings = database.getUrbanMeanings(messageId);
+        List<String> meanings = deserialize(serializedMeanings);
+        String serializedExamples = database.getUrbanExamples(messageId);
+        List<String> examples = deserialize(serializedExamples);
+        String serializedContributors = database.getUrbanContributors(messageId);
+        List<String> contributors = deserialize(serializedContributors);
+        String serializedDates = database.getUrbanDates(messageId);
+        List<String> dates = deserialize(serializedDates);
+        String term = database.getUrbanTerm(messageId);
+        String url = baseUrl + term;
+
+        // count how many pages there are
+        pages = meanings.size();
+
+        // move to new page
+        if(increase)
+            page++;
+        else page--;
+
+        MessageEmbed updatedEmbed = buildEmbed(term, url, event.getUser(),
+                meanings.get(page), examples.get(page), contributors.get(page),
+                dates.get(page), page);
+
+
+
+        List<ItemComponent> components = new ArrayList<>();
+
+        if(page > 0)
+        {
+            components.add(previousPageButton.asEnabled());
+        } else {
+            components.add(previousPageButton.asDisabled());
+        }
+
+        if(page + 1 == pages)
+        {
+            components.add(nextPageButton.asDisabled());
+        } else {
+            components.add(nextPageButton.asEnabled());
+        }
+
+        ActionRow currentRow = ActionRow.of(components);
+        List<ActionRow> actionRows = new ArrayList<>(Collections.singletonList(currentRow));
+
+        event.getMessage().editMessageEmbeds(updatedEmbed).complete();
+        event.editComponents(actionRows).complete();
+        database.setUrbanPage(messageId, page);
+        database.resetExpiryTimestamp(messageId);
+
    }
 }