load app token from FIRETAIL_APP_TOKEN

Author: TheTeaCat

src/main.go

@@ -13,7 +13,8 @@ import (
 const DefaultFiretailApiUrl string = "https://api.logging.eu-west-1.prod.firetail.app/logs/aws/appsync"
 
 var firetailApiUrl string
-var firetailApiToken string
+var firetailToken string
+var firetailTokenHeaderName string = "x-ft-api-key"
 var regionName string
 
 func loadEnvVars() error {
@@ -23,10 +24,14 @@ func loadEnvVars() error {
 		firetailApiUrl = DefaultFiretailApiUrl
 	}
 
-	var firetailApiTokenSet bool
-	firetailApiToken, firetailApiTokenSet = os.LookupEnv("FIRETAIL_API_TOKEN")
-	if !firetailApiTokenSet {
-		return errors.New("FIRETAIL_API_TOKEN is not set")
+	var firetailTokenSet bool
+	firetailToken, firetailTokenSet = os.LookupEnv("FIRETAIL_API_TOKEN")
+	if !firetailTokenSet {
+		firetailToken, firetailTokenSet = os.LookupEnv("FIRETAIL_APP_TOKEN")
+		firetailTokenHeaderName = "x-ft-app-key"
+	}
+	if !firetailTokenSet {
+		return errors.New("Neither FIRETAIL_API_TOKEN or FIRETAIL_APP_TOKEN are set. Cannot continue.")
 	}
 
 	var regionNameExists bool
@@ -35,7 +40,7 @@ func loadEnvVars() error {
 		regionName, regionNameExists = os.LookupEnv("AWS_DEFAULT_REGION")
 	}
 	if !regionNameExists {
-		return errors.New("Neither AWS_REGION nor AWS_DEFAULT_REGION are set. Cannot continue.")
+		return errors.New("Neither AWS_REGION or AWS_DEFAULT_REGION are set. Cannot continue.")
 	}
 
 	return nil