add support of environment variables for the name of the table and th…

…e key alias
Versent · Dec 6, 2016 · fca2698 · fca2698
1 parent 70a80c0
commit fca2698
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cmd/unicreds/main.go b/cmd/unicreds/main.go
@@ -24,8 +24,8 @@ var (
 	region  = app.Flag("region", "Configure the AWS region").Short('r').String()
 	profile = app.Flag("profile", "Configure the AWS profile").Short('p').String()
 
-	dynamoTable = app.Flag("table", "DynamoDB table.").Default("credential-store").Short('t').String()
-	alias       = app.Flag("alias", "KMS key alias.").Default("alias/credstash").Short('k').String()
+	dynamoTable = app.Flag("table", "DynamoDB table.").Default("credential-store").OverrideDefaultFromEnvar("UNICREDS_TABLE").Short('t').String()
+	alias       = app.Flag("alias", "KMS key alias.").Default("alias/credstash").OverrideDefaultFromEnvar("UNICREDS_ALIAS").Short('k').String()
 	encContext  = encryptionContext(app.Flag("enc-context", "Add a key value pair to the encryption context.").Short('E'))
 
 	// commands