Merge pull request #116 from DeterminateSystems/env

Support getting agent token from BUILDKITE_AGENT_TOKEN env var
buildkite · Jul 16, 2021 · c260bc1 · c260bc1
2 parents 1d0ac29 + 6c40b47
commit c260bc1
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/main.go b/main.go
@@ -52,8 +52,12 @@ func main() {
 	}
 
 	if *token == "" {
-		fmt.Println("Must provide a value for -token")
-		os.Exit(1)
+		if bkToken := os.Getenv("BUILDKITE_AGENT_TOKEN"); bkToken != "" {
+			*token = bkToken
+		} else {
+			fmt.Println("Must provide a token")
+			os.Exit(1)
+		}
 	}
 
 	var err error