Merge branch 'main' of github.com:csunibo/informabot

csunibo · Jul 29, 2023 · ba546c0 · ba546c0
2 parents 1958d1b + fb5e9c5
commit ba546c0
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/main.go b/main.go
@@ -7,9 +7,13 @@ import (
 	"github.com/csunibo/informabot/bot"
 )
 
+const tokenKey = "TOKEN"
+
 func main() {
-	if len(os.Args) < 2 {
-		log.Fatalf("Usage: go run main.go <telegram-bot-token>")
+	token, found := os.LookupEnv(tokenKey)
+	if !found {
+		log.Fatal("token not found. please set the TOKEN environment variable")
 	}
-	bot.StartInformaBot(os.Args[1], false)
+
+	bot.StartInformaBot(token, false)
 }