Add the greetings backend and /greet command

Author: TechnoVisionDev

src/main/java/technobot/commands/CommandRegistry.java

@@ -10,6 +10,7 @@
 import org.jetbrains.annotations.NotNull;
 import technobot.TechnoBot;
 import technobot.commands.fun.*;
+import technobot.commands.greetings.GreetCommand;
 import technobot.commands.levels.*;
 import technobot.commands.music.*;
 import technobot.commands.staff.*;
@@ -43,6 +44,9 @@ public class CommandRegistry extends ListenerAdapter {
      */
     public CommandRegistry(TechnoBot bot) {
         mapCommand(
+                //Greeting commands
+                new GreetCommand(bot),
+
                 //Fun commands
                 new JokeCommand(bot),
                 new MemeCommand(bot),

src/main/java/technobot/commands/greetings/GreetCommand.java

@@ -0,0 +1,52 @@
+package technobot.commands.greetings;
+
+import net.dv8tion.jda.api.Permission;
+import net.dv8tion.jda.api.entities.Guild;
+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.OptionData;
+import technobot.TechnoBot;
+import technobot.commands.Category;
+import technobot.commands.Command;
+import technobot.data.GuildData;
+import technobot.handlers.GreetingHandler;
+import technobot.util.embeds.EmbedUtils;
+
+/**
+ * Command that configures auto greetings.
+ *
+ * @author TechnoVision
+ */
+public class GreetCommand extends Command {
+
+    public GreetCommand(TechnoBot bot) {
+        super(bot);
+        this.name = "greet";
+        this.description = "Sets a message to be sent to the welcome channel when a member joins.";
+        this.category = Category.GREETINGS;
+        this.args.add(new OptionData(OptionType.STRING, "greeting", "The message to send as a greeting"));
+        this.permission = Permission.MANAGE_SERVER;
+    }
+
+    @Override
+    public void execute(SlashCommandInteractionEvent event) {
+        event.deferReply().queue();
+        Guild guild = event.getGuild();
+        GreetingHandler greetingHandler = GuildData.get(guild).greetingHandler;
+        OptionMapping greetingOption = event.getOption("greeting");
+
+        // Remove greeting message
+        if (greetingOption == null) {
+            greetingHandler.removeGreet();
+            String text = EmbedUtils.BLUE_X + " Greeting message removed!";
+            event.getHook().sendMessageEmbeds(EmbedUtils.createDefault(text)).queue();
+            return;
+        }
+
+        // Set greeting message
+        greetingHandler.setGreet(greetingOption.getAsString());
+        String text = EmbedUtils.BLUE_TICK + " Greeting message updated!";
+        event.getHook().sendMessageEmbeds(EmbedUtils.createDefault(text)).queue();
+    }
+}

src/main/java/technobot/data/Database.java

@@ -11,11 +11,8 @@
 import org.bson.codecs.pojo.PojoCodecProvider;
 import org.bson.conversions.Bson;
 import org.jetbrains.annotations.NotNull;
-import technobot.data.cache.Config;
-import technobot.data.cache.Leveling;
-import technobot.data.cache.Suggestion;
+import technobot.data.cache.*;
 import technobot.data.cache.moderation.Moderation;
-import technobot.data.cache.Starboard;
 
 import static org.bson.codecs.configuration.CodecRegistries.fromProviders;
 import static org.bson.codecs.configuration.CodecRegistries.fromRegistries;
@@ -33,6 +30,7 @@ public class Database {
     public @NotNull MongoCollection<Starboard> starboard;
     public @NotNull MongoCollection<Leveling> leveling;
     public @NotNull MongoCollection<Config> config;
+    public @NotNull MongoCollection<Greetings> greetings;
 
     /**
      * Connect to database using MongoDB URI and
@@ -57,12 +55,14 @@ public Database(String uri) {
         starboard = database.getCollection("starboard", Starboard.class);
         leveling = database.getCollection("leveling", Leveling.class);
         config = database.getCollection("config", Config.class);
+        greetings = database.getCollection("greetings", Greetings.class);
 
         Bson guildIndex = Indexes.descending("guild");
         suggestions.createIndex(guildIndex);
         moderation.createIndex(guildIndex);
         starboard.createIndex(guildIndex);
         config.createIndex(guildIndex);
+        greetings.createIndex(guildIndex);
         leveling.createIndex(Indexes.compoundIndex(guildIndex, Indexes.descending("user")));
     }
 }

src/main/java/technobot/data/GuildData.java

@@ -33,6 +33,7 @@ public class GuildData {
     public ModerationHandler moderationHandler;
     public StarboardHandler starboardHandler;
     public LevelingHandler levelingHandler;
+    public GreetingHandler greetingHandler;
 
     /**
      * Represents the local memory cache of guild data stored in the MongoDB databases.
@@ -46,6 +47,7 @@ private GuildData(Guild guild) {
         moderationHandler = new ModerationHandler(bot, guild);
         starboardHandler = new StarboardHandler(bot, guild);
         levelingHandler = new LevelingHandler(bot, guild);
+        greetingHandler = new GreetingHandler(bot, guild);
 
         // Setup guild config
         config = bot.database.config.find(Filters.eq("guild", guild.getIdLong())).first();

src/main/java/technobot/data/cache/Greetings.java

@@ -0,0 +1,69 @@
+package technobot.data.cache;
+
+import org.bson.codecs.pojo.annotations.BsonProperty;
+
+/**
+ * POJO object that stores server greeting/farewell data.
+ *
+ * @author TechnoVision
+ */
+public class Greetings {
+
+    private long guild;
+
+    @BsonProperty("welcome_channel")
+    private Long welcomeChannel;
+
+    private String greeting;
+
+    private String farewell;
+
+    @BsonProperty("join_dm")
+    private String joinDM;
+
+    public Greetings() { }
+
+    public Greetings(long guild) {
+        this.guild = guild;
+    }
+
+    public long getGuild() {
+        return guild;
+    }
+
+    public void setGuild(long guild) {
+        this.guild = guild;
+    }
+
+    public Long getWelcomeChannel() {
+        return welcomeChannel;
+    }
+
+    public void setWelcomeChannel(Long welcomeChannel) {
+        this.welcomeChannel = welcomeChannel;
+    }
+
+    public String getGreeting() {
+        return greeting;
+    }
+
+    public void setGreeting(String greeting) {
+        this.greeting = greeting;
+    }
+
+    public String getFarewell() {
+        return farewell;
+    }
+
+    public void setFarewell(String farewell) {
+        this.farewell = farewell;
+    }
+
+    public String getJoinDM() {
+        return joinDM;
+    }
+
+    public void setJoinDM(String joinDM) {
+        this.joinDM = joinDM;
+    }
+}

src/main/java/technobot/handlers/GreetingHandler.java

@@ -0,0 +1,52 @@
+package technobot.handlers;
+
+import com.mongodb.client.model.Filters;
+import com.mongodb.client.model.Updates;
+import net.dv8tion.jda.api.entities.Guild;
+import org.bson.conversions.Bson;
+import technobot.TechnoBot;
+import technobot.data.cache.Greetings;
+
+/**
+ * Handles server messages (greeting, farewell, joinDM, etc).
+ *
+ * @author TechnoVision
+ */
+public class GreetingHandler {
+
+    private final Guild guild;
+    private final TechnoBot bot;
+    private final Bson filter;
+    private Greetings greetings;
+
+    public GreetingHandler(TechnoBot bot, Guild guild) {
+        this.bot = bot;
+        this.guild = guild;
+
+        // Get POJO object from database
+        filter = Filters.eq("guild", guild.getIdLong());
+        this.greetings = bot.database.greetings.find(filter).first();
+        if (greetings == null) {
+            greetings = new Greetings(guild.getIdLong());
+            bot.database.greetings.insertOne(greetings);
+        }
+    }
+
+    /**
+     * Set a greeting message for this server.
+     *
+     * @param msg the message to send on join.
+     */
+    public void setGreet(String msg) {
+        greetings.setGreeting(msg);
+        bot.database.greetings.updateOne(filter, Updates.set("greeting", msg));
+    }
+
+    /**
+     * Remove greeting message from this server.
+     */
+    public void removeGreet() {
+        greetings.setGreeting(null);
+        bot.database.greetings.updateOne(filter, Updates.unset("greeting"));
+    }
+}