Modified token getter

This commit is contained in:
Anthony Robert committed 2019-03-17 16:20:10 -04:00
1 parent 3885367a5a
commit d95b4c40d0
2 files changed
+13 -10

No files matched your search

+12 -10
View File
@@ -5,13 +5,13 @@ import net.dv8tion.jda.core.JDA;
import net.dv8tion.jda.core.JDABuilder; import net.dv8tion.jda.core.JDABuilder;
import net.dv8tion.jda.core.entities.Game; import net.dv8tion.jda.core.entities.Game;
import netneutrality.db.DBController; import netneutrality.db.DBController;
import netneutrality.db.controllers.DBSettings;
import netneutrality.handler.ConsoleHandler; import netneutrality.handler.ConsoleHandler;
import netneutrality.handler.EditHandler; import netneutrality.handler.EditHandler;
import netneutrality.handler.LogHandler; import netneutrality.handler.LogHandler;
import netneutrality.handler.NewCommandListener; import netneutrality.handler.NewCommandListener;
import netneutrality.http.server.WebApi; import netneutrality.http.server.WebApi;
import netneutrality.music.MusicManager; import netneutrality.music.MusicManager;
import netneutrality.utils.Settings;
import org.slf4j.Logger; import org.slf4j.Logger;
import org.slf4j.LoggerFactory; import org.slf4j.LoggerFactory;
@@ -20,20 +20,22 @@ import javax.security.auth.login.LoginException;
public class MusicBot { public class MusicBot {
private static JDA jda; private static JDA jda;
public static Logger LOGGER = LoggerFactory.getLogger(NewCommandListener.class); public static Logger LOGGER = LoggerFactory.getLogger(MusicBot.class);
public static void main(String[] args) { public static void main(String[] args) {
System.setProperty("log4j.skipJansi", "false"); System.setProperty("log4j.skipJansi", "false");
DBController.init(); DBController.init();
String token = ((DBSettings) DBController.get("settings")).getSettingsByKey("token");
try { try {
jda = new JDABuilder(AccountType.BOT).setToken(token).buildBlocking(); jda = new JDABuilder(AccountType.BOT).setToken(Settings.TOKEN).build();
} catch (LoginException | IllegalArgumentException | InterruptedException e) { jda.awaitReady();
LOGGER.error(e.getMessage()); } catch (LoginException | IllegalArgumentException e) {
LogHandler.log(e); LogHandler.log(e);
} catch (InterruptedException e) {
LOGGER.error(e.getMessage());
} }
jda.getPresence().setGame(Game.watching("me load...")); jda.getPresence().setGame(Game.watching("me load..."));
jda.addEventListener(new NewCommandListener(), new EditHandler()); jda.addEventListener(new NewCommandListener(), new EditHandler());
@@ -48,7 +50,6 @@ public class MusicBot {
case "quit": case "quit":
shutdown(); shutdown();
System.exit(0);
break; break;
} }
@@ -56,12 +57,13 @@ public class MusicBot {
} }
public static void shutdown() { public static void shutdown() {
System.out.println("Shutting down..."); LOGGER.info("[Net Neutrality] Shutting down...");
MusicManager.disconnectPlayers(); MusicManager.disconnectPlayers();
disconnect(); disconnect();
DBController.shutdown(); DBController.shutdown();
ThreadManager.shutdownExecutor(); ThreadManager.shutdownExecutor();
System.out.println("Shutdown complete"); LOGGER.info("[Net Neutrality] Shutdown complete");
System.exit(0);
} }
public static JDA getJDA() { public static JDA getJDA() {
@@ -11,6 +11,7 @@ public class Settings {
public static final String DATETIMEFORMAT = "yyyy-MM-dd HH:mm:ss"; public static final String DATETIMEFORMAT = "yyyy-MM-dd HH:mm:ss";
public static String PREFIX = ((DBSettings) DBController.get("settings")).getSettingsByKey("prefix"); public static String PREFIX = ((DBSettings) DBController.get("settings")).getSettingsByKey("prefix");
public static final String TOKEN = ((DBSettings) DBController.get("settings")).getSettingsByKey("token");
public static User getCreator() { public static User getCreator() {
return MusicBot.getJDA().getUserById(((DBSettings) DBController.get("settings")).getSettingsByKey("owner")); return MusicBot.getJDA().getUserById(((DBSettings) DBController.get("settings")).getSettingsByKey("owner"));