clean up ssh keys when cleaning up e2e test resources

hack/do-janitor/do-janitor.go

@@ -104,6 +104,25 @@ func main() {
 		}
 	}
 
+	keys, err := keyList(ctx, client)
+	if err != nil {
+		log.Fatalf("failed to list keys: %+v", err.Error())
+	}
+
+	capdoStr := "capdo" // using this for repeated comparisons in the loop below
+	for _, key := range keys {
+		// we only care to cleanup keys that start with "capdo"
+		if len(key.Name) >= len(capdoStr) && key.Name[0:len(capdoStr)] == capdoStr {
+			log.Printf("%s contains capdo at the start of it's name 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
+}