Adds check for required environment variables

dollarshaveclub · Jun 25, 2018 · 42f3239 · 42f3239
1 parent b4f5ec0
commit 42f3239
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/cmd/root.go b/cmd/root.go
@@ -39,6 +39,9 @@ var rootCmd = &cobra.Command{
 
 		switch directoryBackend {
 		case "github":
+			if os.Getenv("GITHUB_TOKEN") == "" {
+				errorAndExit(errors.New("You must set the GITHUB_TOKEN environment variable"), 1)
+			}
 			dirState, err = directory.NewGitHub(Org)
 			if err != nil {
 				errorAndExit(fmt.Errorf("unable to get directory client: %+v", err), 1)
@@ -49,6 +52,9 @@ var rootCmd = &cobra.Command{
 
 		switch storageBackend {
 		case "vault":
+			if os.Getenv("VAULT_TOKEN") == "" {
+				errorAndExit(errors.New("You must set the VAULT_TOKEN environment variable with a valid Vault token"), 1)
+			}
 			storageClient, err = storage.NewVault()
 			if err != nil {
 				errorAndExit(fmt.Errorf("unable to get storage client: %+v", err), 1)