diff --git a/bot.go b/bot.go index 42992e9..32cd046 100644 --- a/bot.go +++ b/bot.go @@ -32,10 +32,15 @@ var ( func main() { 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.") + println("Please add token as environment variable (MENSABOT_TOKEN) or cli argument.") return } - dg, err := discordgo.New(fmt.Sprintf("Bot %s", os.Args[1])) + + if len(os.Args) > 0 { + token = os.Args[0] + } + + dg, err := discordgo.New(fmt.Sprintf("Bot %s", token)) if err != nil { fmt.Println("error creating discord bot: ", err) return