From 944d6137d4f6347ef8787aaf04b00d2fb2f91774 Mon Sep 17 00:00:00 2001 From: pocketjawa Date: Sat, 27 Apr 2019 01:56:31 -0400 Subject: [PATCH] Force using a bot token. --- buttsbot.go | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/buttsbot.go b/buttsbot.go index 78a14a4..eee62a0 100644 --- a/buttsbot.go +++ b/buttsbot.go @@ -12,11 +12,15 @@ import ( var Session, _ = discordgo.New() var testPtr = flag.Bool("test", false, "Test mode") +var bot_token string func init() { - Session.Token = os.Getenv("DISCORD_TOKEN") - if Session.Token == "" { - flag.StringVar(&Session.Token, "t", "", "Discord Authentication Token") + if os.Getenv("DISCORD_TOKEN") != "" { + Session.Token = "Bot " + os.Getenv("DISCORD_TOKEN") + } else { + flag.StringVar(&bot_token, "t", "", "Discord Authentication Token") + flag.Parse() + Session.Token = "Bot " + bot_token } }