Add jenkins public key to instance

Author: davidz627

test/remote/run_remote/run_remote.go

@@ -20,6 +20,7 @@ import (
 	"context"
 	"flag"
 	"fmt"
+	"io/ioutil"
 	"math/rand"
 	"net/http"
 	"os"
@@ -320,6 +321,18 @@ func createInstance(serviceAccount string) (string, error) {
 		glog.Infof("Compute service GOT instance %v: %#v", i.Name, gotInstance)
 	}
 
+	pubkey, ok := os.LookupEnv("JENKINS_GCE_SSH_PUBLIC_KEY_FILE")
+	if ok {
+		glog.Infof("Running on Jenkins and JENKINS_GCE_SSH_PUBLIC_KEY_FILE set")
+		// If we're on CI add ServiceAccount Keys to the instance
+		err = addPubKeyToInstance(*project, *zone, i.Name, pubkey)
+		if err != nil {
+			return "", fmt.Errorf("could not add Jenkins Public Key %v to instance %v: %v", pubkey, i.Name, err)
+		}
+	} else {
+		glog.Infof("JENKINS_GCE_SSH_PUBLIC_KEY_FILE not set, not adding SSH Public Key to Instance")
+	}
+
 	then := time.Now()
 	err = wait.Poll(15*time.Second, 10*time.Minute, func() (bool, error) {
 		glog.V(2).Infof("Waiting for instance %v to come up. %v elapsed", name, time.Since(then))
@@ -359,6 +372,49 @@ func createInstance(serviceAccount string) (string, error) {
 	return name, nil
 }
 
+func addPubKeyToInstance(project, zone, name, pubKeyFile string) error {
+	found := false
+	i, err := computeService.Instances.Get(project, zone, name).Do()
+	if err != nil {
+		return err
+	}
+	fingerprint := i.Metadata.Fingerprint
+	items := i.Metadata.Items
+	for _, item := range items {
+		if item.Key == "ssh-keys" {
+			found = true
+		}
+	}
+	newKeys := ""
+	if found {
+		// Append these to newKeys first
+		glog.Infof("Found existing public keys on instance %v", name)
+	}
+	glog.Infof("Public key file: %v", pubKeyFile)
+	publicKey, err := ioutil.ReadFile(pubKeyFile)
+	if err != nil {
+		return err
+	}
+
+	newKeys += string(publicKey) + "\n"
+	glog.Infof("New Keys: %v", newKeys)
+	newMeta := &compute.Metadata{
+		Fingerprint: fingerprint,
+		Items: []*compute.MetadataItems{
+			&compute.MetadataItems{
+				Key:   "ssh-keys",
+				Value: &newKeys,
+			},
+		},
+	}
+	_, err = computeService.Instances.SetMetadata(project, zone, name, newMeta).Do()
+	if err != nil {
+		return err
+	}
+	return nil
+
+}
+
 func getexternalIP(instance *compute.Instance) string {
 	for i := range instance.NetworkInterfaces {
 		ni := instance.NetworkInterfaces[i]