Use keys specified in env vars for GCP and AWS

Signed-off-by: Davanum Srinivas <davanum@gmail.com>
This commit is contained in:
Davanum Srinivas 2024-08-30 07:56:46 -04:00
parent d5c56dcf98
commit ef76bf9164
No known key found for this signature in database
GPG Key ID: 80D83A796103BF59
1 changed files with 17 additions and 0 deletions

View File

@ -51,6 +51,12 @@ func (d *deployer) initialize() error {
switch d.CloudProvider {
case "aws":
if d.SSHPrivateKeyPath == "" {
d.SSHPrivateKeyPath = os.Getenv("AWS_SSH_PRIVATE_KEY_FILE")
}
if d.SSHPublicKeyPath == "" {
d.SSHPublicKeyPath = os.Getenv("AWS_SSH_PUBLIC_KEY_FILE")
}
if d.SSHPrivateKeyPath == "" || d.SSHPublicKeyPath == "" {
publicKeyPath, privateKeyPath, err := util.CreateSSHKeyPair(d.ClusterName)
if err != nil {
@ -90,6 +96,12 @@ func (d *deployer) initialize() error {
d.GCPProject = resource.Name
klog.V(1).Infof("Got project %s from boskos", d.GCPProject)
if d.SSHPrivateKeyPath == "" {
d.SSHPrivateKeyPath = os.Getenv("GCE_SSH_PRIVATE_KEY_FILE")
}
if d.SSHPublicKeyPath == "" {
d.SSHPublicKeyPath = os.Getenv("GCE_SSH_PUBLIC_KEY_FILE")
}
if d.SSHPrivateKeyPath == "" && d.SSHPublicKeyPath == "" {
privateKey, publicKey, err := gce.SetupSSH(d.GCPProject)
if err != nil {
@ -98,12 +110,15 @@ func (d *deployer) initialize() error {
d.SSHPrivateKeyPath = privateKey
d.SSHPublicKeyPath = publicKey
}
d.createBucket = true
} else if d.SSHPrivateKeyPath == "" && os.Getenv("KUBE_SSH_KEY_PATH") != "" {
d.SSHPrivateKeyPath = os.Getenv("KUBE_SSH_KEY_PATH")
}
}
klog.V(1).Infof("Using SSH keypair: [%s,%s]", d.SSHPrivateKeyPath, d.SSHPublicKeyPath)
if d.commonOptions.ShouldBuild() {
if err := d.verifyBuildFlags(); err != nil {
return fmt.Errorf("init failed to check build flags: %v", err)
@ -113,6 +128,8 @@ func (d *deployer) initialize() error {
if d.SSHUser == "" {
d.SSHUser = os.Getenv("KUBE_SSH_USER")
}
klog.V(1).Infof("Using SSH user: [%s]", d.SSHUser)
if d.TerraformVersion != "" {
t, err := target.NewTerraform(d.TerraformVersion)
if err != nil {