Make love calculator also support slash commands

2022-12-20 22:32:40 +01:00
parent 174b78704f
commit 1c82d2b53b
4 changed files with 102 additions and 22 deletions
--- a/src/main/java/wtf/beatrice/hidekobot/commands/slash/LoveCalculatorCommand.java
+++ b/src/main/java/wtf/beatrice/hidekobot/commands/slash/LoveCalculatorCommand.java
@@ -0,0 +1,61 @@
+package wtf.beatrice.hidekobot.commands.slash;
+
+import net.dv8tion.jda.api.entities.MessageEmbed;
+import net.dv8tion.jda.api.entities.User;
+import net.dv8tion.jda.api.events.interaction.command.SlashCommandInteractionEvent;
+import net.dv8tion.jda.api.interactions.commands.OptionMapping;
+import net.dv8tion.jda.api.interactions.commands.OptionType;
+import net.dv8tion.jda.api.interactions.commands.build.CommandData;
+import net.dv8tion.jda.api.interactions.commands.build.Commands;
+import org.jetbrains.annotations.NotNull;
+import wtf.beatrice.hidekobot.commands.base.LoveCalculator;
+import wtf.beatrice.hidekobot.objects.commands.SlashCommandImpl;
+
+public class LoveCalculatorCommand extends SlashCommandImpl
+{
+    @Override
+    public CommandData getSlashCommandData()
+    {
+
+        return Commands.slash("lovecalc",
+                "Calculate how much two people love each other.")
+
+                .addOption(OptionType.MENTIONABLE,
+                        "first",
+                        "The first person to account for",
+                        true)
+
+                .addOption(OptionType.MENTIONABLE,
+                        "second",
+                        "The second person to account for",
+                        false);
+    }
+
+    @Override
+    public void runSlashCommand(@NotNull SlashCommandInteractionEvent event)
+    {
+        User firstUser, secondUser;
+
+        OptionMapping firsUserArg = event.getOption("first");
+        if(firsUserArg != null)
+        {
+            firstUser = firsUserArg.getAsUser();
+        } else {
+            event.reply("\uD83D\uDE22 I need to know who to check! Please mention them.")
+                    .setEphemeral(true)
+                    .queue();
+            return;
+        }
+
+        OptionMapping secondUserArg = event.getOption("second");
+        if(secondUserArg != null)
+        {
+            secondUser = secondUserArg.getAsUser();
+        } else {
+            secondUser = event.getUser();
+        }
+
+        MessageEmbed embed = LoveCalculator.buildEmbed(event.getUser(), firstUser, secondUser);
+        event.replyEmbeds(embed).queue();
+    }
+}