From ecd5ed31a3a5a30c04fc71aa3cbd620b069b4cc5 Mon Sep 17 00:00:00 2001
From: Sebastien <seb6596@gmail.com>
Date: Mon, 31 Dec 2018 15:08:19 +0100
Subject: [PATCH] Check ENV var before start

---
 src/main/java/net/Broken/Init.java    | 54 +++++++++++++++++++++++++++
 src/main/java/net/Broken/MainBot.java | 11 +++++-
 2 files changed, 63 insertions(+), 2 deletions(-)

diff --git a/src/main/java/net/Broken/Init.java b/src/main/java/net/Broken/Init.java
index b775937..c2f87f5 100644
--- a/src/main/java/net/Broken/Init.java
+++ b/src/main/java/net/Broken/Init.java
@@ -113,4 +113,58 @@ public class Init {
 
 
     }
+
+
+    public static boolean checkEnv(){
+        boolean ok = true;
+
+
+        if(System.getenv("PORT") == null){
+            logger.fatal("Missing PORT ENV variable.");
+            ok = false;
+        }
+
+        if(System.getenv("DB_URL") == null){
+            logger.fatal("Missing DB_URL ENV variable.");
+            ok = false;
+        }
+
+        if(System.getenv("DB_USER") == null){
+            logger.fatal("Missing DB_USER ENV variable.");
+            ok = false;
+        }
+
+
+        if(System.getenv("DB_PWD") == null){
+            logger.fatal("Missing DB_PWD ENV variable.");
+            ok = false;
+        }
+
+        if(System.getenv("OAUTH_URL") == null){
+            logger.fatal("Missing OAUTH_URL ENV variable.");
+            ok = false;
+        }
+
+        if(System.getenv("DISCORD_TOKEN") == null){
+            logger.fatal("Missing DISCORD_TOKEN ENV variable.");
+            ok = false;
+        }
+
+        if(System.getenv("GOOGLE_API_KEY") == null){
+            logger.fatal("Missing GOOGLE_API_KEY ENV variable.");
+            ok = false;
+        }
+
+        if(System.getenv("RANDOM_API_KEY") == null){
+            logger.fatal("Missing GOOGLE_API_KEY ENV variable.");
+            ok = false;
+        }
+
+        if(System.getenv("LOG_LEVEL") == null){
+            logger.fatal("Missing LOG_LEVEL ENV variable.");
+            ok = false;
+        }
+
+        return ok;
+    }
 }
diff --git a/src/main/java/net/Broken/MainBot.java b/src/main/java/net/Broken/MainBot.java
index 9c18801..fd59813 100644
--- a/src/main/java/net/Broken/MainBot.java
+++ b/src/main/java/net/Broken/MainBot.java
@@ -51,14 +51,19 @@ public class MainBot {
 
     public static void main(String[] args) {
 
+
+        if(!Init.checkEnv())
+            System.exit(1);
+
         logger.info("=======================================");
         logger.info("--------------Starting Bot-------------");
         logger.info("=======================================");
-
         if(System.getenv("DEV")!= null){
             dev = Boolean.parseBoolean(System.getenv("DEV"));
         }
-        String token = System.getenv("TOKEN");
+
+
+        String token = System.getenv("DISCORD_TOKEN");
 
         jda = Init.initJda(token);
 
@@ -134,4 +139,6 @@ public class MainBot {
         }
     }
 
+
+
 }