diff --git a/src/main/java/com/bbn/hadder/Hadder.java b/src/main/java/com/bbn/hadder/Hadder.java index 157d555..7065fe3 100644 --- a/src/main/java/com/bbn/hadder/Hadder.java +++ b/src/main/java/com/bbn/hadder/Hadder.java @@ -55,7 +55,7 @@ public class Hadder { new CommandListener(), new GuildJoinListener(), new GuildLeaveListener(), - new BotListListener(), + new GuildListener(), new ReadyListener()); try { diff --git a/src/main/java/com/bbn/hadder/Rethink.java b/src/main/java/com/bbn/hadder/Rethink.java index 7853082..68fa912 100644 --- a/src/main/java/com/bbn/hadder/Rethink.java +++ b/src/main/java/com/bbn/hadder/Rethink.java @@ -72,6 +72,18 @@ public class Rethink { return out; } + public static String insertUser(String id) { + String out = ""; + try { + Cursor cursor = r.table("user") + .insert(r.hashMap("id", id) + .with("prefix", "h.") + ).run(conn); + out = cursor.next().toString(); + } catch (ClassCastException ignored) {} + return out; + } + public static void setup() { try { r.dbCreate("Hadder").run(conn); diff --git a/src/main/java/com/bbn/hadder/listener/BotListListener.java b/src/main/java/com/bbn/hadder/listener/BotListListener.java deleted file mode 100644 index 2d57cbc..0000000 --- a/src/main/java/com/bbn/hadder/listener/BotListListener.java +++ /dev/null @@ -1,23 +0,0 @@ -package com.bbn.hadder.listener; - -/* - * @author Skidder / GregTCLTK - */ - -import net.dv8tion.jda.api.events.guild.GuildJoinEvent; -import net.dv8tion.jda.api.events.guild.GuildLeaveEvent; -import net.dv8tion.jda.api.hooks.ListenerAdapter; - -public class BotListListener extends ListenerAdapter { - - public void onGuildJoin(GuildJoinEvent event) { - - - } - - public void onGuildLeave(GuildLeaveEvent event) { - - - - } -} diff --git a/src/main/java/com/bbn/hadder/listener/GuildJoinListener.java b/src/main/java/com/bbn/hadder/listener/GuildJoinListener.java deleted file mode 100644 index cdea693..0000000 --- a/src/main/java/com/bbn/hadder/listener/GuildJoinListener.java +++ /dev/null @@ -1,16 +0,0 @@ -package com.bbn.hadder.listener; - -/* - * @author Skidder / GregTCLTK - */ - -import com.bbn.hadder.Rethink; -import net.dv8tion.jda.api.events.guild.GuildJoinEvent; -import net.dv8tion.jda.api.hooks.ListenerAdapter; -import org.jetbrains.annotations.NotNull; - -public class GuildJoinListener extends ListenerAdapter { - public void onGuildJoin(@NotNull GuildJoinEvent event) { - Rethink.insertServer(event.getGuild().getId()); - } -} diff --git a/src/main/java/com/bbn/hadder/listener/GuildLeaveListener.java b/src/main/java/com/bbn/hadder/listener/GuildLeaveListener.java deleted file mode 100644 index fc65d2e..0000000 --- a/src/main/java/com/bbn/hadder/listener/GuildLeaveListener.java +++ /dev/null @@ -1,16 +0,0 @@ -package com.bbn.hadder.listener; - -/* - * @author Skidder / GregTCLTK - */ - -import net.dv8tion.jda.api.events.guild.GuildLeaveEvent; -import net.dv8tion.jda.api.hooks.ListenerAdapter; -import org.jetbrains.annotations.NotNull; - -public class GuildLeaveListener extends ListenerAdapter { - public void onGuildLeave(@NotNull GuildLeaveEvent event) { - - - } -} diff --git a/src/main/java/com/bbn/hadder/listener/GuildListener.java b/src/main/java/com/bbn/hadder/listener/GuildListener.java new file mode 100644 index 0000000..7da1d4f --- /dev/null +++ b/src/main/java/com/bbn/hadder/listener/GuildListener.java @@ -0,0 +1,28 @@ +package com.bbn.hadder.listener; + +/* + * @author Skidder / GregTCLTK + */ + +import com.bbn.hadder.Rethink; +import net.dv8tion.jda.api.events.guild.GuildJoinEvent; +import net.dv8tion.jda.api.events.guild.GuildLeaveEvent; +import net.dv8tion.jda.api.events.guild.member.GuildMemberJoinEvent; +import net.dv8tion.jda.api.hooks.ListenerAdapter; + +public class GuildListener extends ListenerAdapter { + + public void onGuildJoin(GuildJoinEvent event) { + Rethink.insertServer(event.getGuild().getId()); + } + + public void onGuildLeave(GuildLeaveEvent event) { + + } + + public void onGuildMemberJoin(GuildMemberJoinEvent event) { + if (!event.getUser().getId().equals(event.getJDA().getSelfUser().getId())) { + Rethink.insertUser(event.getUser().getId()); + } + } +} diff --git a/src/main/java/com/bbn/hadder/listener/ReadyListener.java b/src/main/java/com/bbn/hadder/listener/ReadyListener.java index 3072f77..4d24a2b 100644 --- a/src/main/java/com/bbn/hadder/listener/ReadyListener.java +++ b/src/main/java/com/bbn/hadder/listener/ReadyListener.java @@ -2,6 +2,7 @@ package com.bbn.hadder.listener; import com.bbn.hadder.Rethink; import net.dv8tion.jda.api.entities.Guild; +import net.dv8tion.jda.api.entities.User; import net.dv8tion.jda.api.events.ReadyEvent; import net.dv8tion.jda.api.hooks.ListenerAdapter; @@ -12,6 +13,11 @@ public class ReadyListener extends ListenerAdapter { @Override public void onReady(@Nonnull ReadyEvent event) { Rethink.setup(); + for (User user : event.getJDA().getUsers()) { + if (!user.getId().equals(event.getJDA().getSelfUser().getId())) { + Rethink.insertUser(user.getId()); + } + } for (Guild g : event.getJDA().getGuilds()) { Rethink.insertServer(g.getId()); }