Merge pull request #161 from ovotech/fix-ssm

Export cobra vars and add switch for key vs keyId names

Author: Chris Every

pkg/location/circleci.go

@@ -41,12 +41,14 @@ func (circle CircleCI) Write(serviceAccountName string, keyWrapper KeyWrapper, c
 	provider := keyWrapper.KeyProvider
 
 	var keyEnvVar string
-	if keyEnvVar, err = getVarNameFromProvider(provider, circle.KeyEnvVar); err != nil {
+	var idValue bool
+	if keyEnvVar, err = getVarNameFromProvider(provider, circle.KeyEnvVar, idValue); err != nil {
 		return
 	}
 
 	var keyIDEnvVar string
-	if keyIDEnvVar, err = getVarNameFromProvider(provider, circle.KeyIDEnvVar); err != nil {
+	idValue = true
+	if keyIDEnvVar, err = getVarNameFromProvider(provider, circle.KeyIDEnvVar, idValue); err != nil {
 		return
 	}
 

pkg/location/locations.go

@@ -68,15 +68,19 @@ func envVarDefaultsFromProvider(provider string) (envVarDefaults envVarDefaults,
 	return
 }
 
-func getVarNameFromProvider(provider, suppliedVarName string) (envName string, err error) {
+func getVarNameFromProvider(provider, suppliedVarName string, idValue bool) (envName string, err error) {
 	if len(suppliedVarName) > 0 {
 		envName = suppliedVarName
 	} else {
 		var defaultEnvVar envVarDefaults
 		if defaultEnvVar, err = envVarDefaultsFromProvider(provider); err != nil {
 			return
 		}
-		envName = defaultEnvVar.keyEnvVar
+		if idValue {
+			envName = defaultEnvVar.keyIDEnvVar
+		} else {
+			envName = defaultEnvVar.keyEnvVar
+		}
 	}
 	return
 }

pkg/location/ssm.go

@@ -42,12 +42,14 @@ func (ssm Ssm) Write(serviceAccountName string, keyWrapper KeyWrapper, creds cre
 	}
 
 	var keyEnvVar string
-	if keyEnvVar, err = getVarNameFromProvider(provider, ssm.keyParamName); err != nil {
+	var idValue bool
+	if keyEnvVar, err = getVarNameFromProvider(provider, ssm.keyParamName, idValue); err != nil {
 		return
 	}
 
 	var keyIDEnvVar string
-	if keyIDEnvVar, err = getVarNameFromProvider(provider, ssm.keyIDParamName); err != nil {
+	idValue = true
+	if keyIDEnvVar, err = getVarNameFromProvider(provider, ssm.keyIDParamName, idValue); err != nil {
 		return
 	}