Merge pull request #316 from slickwarren/cwarren/v2.9/add-corral-bastion

[v2.9] add corralBastion back as a func

clients/corral/corral.go

@@ -18,10 +18,12 @@ import (
 )
 
 const (
-	debugFlag           = "--trace"
-	skipCleanupFlag     = "--skip-cleanup"
-	corralPrivateSSHKey = "corral_private_key"
-	corralPublicSSHKey  = "corral_public_key"
+	debugFlag               = "--trace"
+	skipCleanupFlag         = "--skip-cleanup"
+	corralPrivateSSHKey     = "corral_private_key"
+	corralPublicSSHKey      = "corral_public_key"
+	corralRegistryIP        = "registry_ip"
+	corralRegistryPrivateIP = "registry_private_ip"
 )
 
 // GetCorralEnvVar gets corral environment variables
@@ -345,3 +347,23 @@ func SetCorralSSHKeys(corralName string) error {
 
 	return UpdateCorralConfig(corralPublicSSHKey, publicSSHkey)
 }
+
+// SetCorralBastion is a helper function that will set the corral bastion private and pulic addresses previously generated by `corralName`
+func SetCorralBastion(corralName string) error {
+	bastion_ip, err := GetCorralEnvVar(corralName, corralRegistryIP)
+	if err != nil {
+		return err
+	}
+
+	err = UpdateCorralConfig(corralRegistryIP, bastion_ip)
+	if err != nil {
+		return err
+	}
+
+	bastion_internal_ip, err := GetCorralEnvVar(corralName, corralRegistryPrivateIP)
+	if err != nil {
+		return err
+	}
+
+	return UpdateCorralConfig(corralRegistryPrivateIP, bastion_internal_ip)
+}