From 47a2ff06148bbefccd0f29e6db6174f3be5e6bf1 Mon Sep 17 00:00:00 2001
From: kolaente <konrad@kola-entertainments.de>
Date: Fri, 12 Apr 2019 09:41:08 +0200
Subject: [PATCH] Sync files with github

---
 bot.go | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/bot.go b/bot.go
index 5b76c6e..42992e9 100644
--- a/bot.go
+++ b/bot.go
@@ -30,16 +30,21 @@ var (
 )
 
 func main() {
-	if len(os.Args) <= 0 {
-		println("Please add token as cli argument.")
+	token := os.Getenv("MENSABOT_TOKEN")
+	if len(token) == 0 || token == "token" || len(os.Args) <= 0 {
+		println("Please add token as environment variable: (MENSABOT_TOKEN). or cli argument.")
 		return
 	}
 	dg, err := discordgo.New(fmt.Sprintf("Bot %s", os.Args[1]))
-	canteens, err := openmensa.GetCanteens(175)
 	if err != nil {
 		fmt.Println("error creating discord bot: ", err)
 		return
 	}
+	canteens, err := openmensa.GetCanteens(175)
+	if err != nil {
+		fmt.Println("error getting mensa: ", err)
+		return
+	}
 	dg.AddHandler(messageCreate)
 	canteen = canteens[0]