clean up ssh keys when cleaning up e2e test resources

hack/do-janitor/do-janitor.go

@@ -20,6 +20,7 @@ import (
 	"context"
 	"log"
 	"os"
+	"strings"
 	"time"
 
 	"github.com/digitalocean/godo"
@@ -104,6 +105,24 @@ func main() {
 		}
 	}
 
+	keys, err := keyList(ctx, client)
+	if err != nil {
+		log.Fatalf("failed to list keys: %+v", err.Error())
+	}
+
+	for _, key := range keys {
+		if strings.HasPrefix(key.Name, "capdo-") {
+			log.Printf("%s contains capdo prefix so it's safe to terminate\n", key.Name)
+			_, err := client.Keys.DeleteByID(ctx, key.ID)
+			if err != nil {
+				log.Printf("failed to delete key %s: %+v\n", key.Name, err.Error())
+				continue
+			}
+
+			log.Printf("key %s terminated\n", key.Name)
+		}
+	}
+
 	log.Println("Completed DO Janitor")
 	os.Exit(0)
 }
@@ -190,3 +209,31 @@ func volumeList(ctx context.Context, client *godo.Client) ([]godo.Volume, error)
 
 	return list, nil
 }
+
+func keyList(ctx context.Context, client *godo.Client) ([]godo.Key, error) {
+	list := []godo.Key{}
+
+	// create options. initially, these will be blank.
+	opt := &godo.ListOptions{}
+	for {
+		keys, resp, err := client.Keys.List(ctx, opt)
+		if err != nil {
+			return nil, err
+		}
+
+		list = append(list, keys...)
+
+		if resp.Links == nil || resp.Links.IsLastPage() {
+			break
+		}
+
+		page, err := resp.Links.CurrentPage()
+		if err != nil {
+			return nil, err
+		}
+
+		opt.Page = page + 1
+	}
+
+	return list, nil
+}