Implements delete cluster and toolbox dumper command

upup/pkg/fi/cloudup/vsphere/vsphere_cloud.go

@@ -327,6 +327,98 @@ func (c *VSphereCloud) FindVMUUID(vm *string) (string, error) {
 	return vmResult.Config.Uuid, nil
 }
 
+// GetVirtualMachines returns the VMs where the VM name matches the strings in the argument
+func (c *VSphereCloud) GetVirtualMachines(args []string) ([]*object.VirtualMachine, error) {
+	var out []*object.VirtualMachine
+
+	// List virtual machines
+	if len(args) == 0 {
+		return nil, errors.New("no argument")
+	}
+
+	f := find.NewFinder(c.Client.Client, true)
+	ctx, cancel := context.WithCancel(context.Background())
+	defer cancel()
+
+	dc, err := f.Datacenter(ctx, c.Datacenter)
+	if err != nil {
+		return nil, err
+	}
+	f.SetDatacenter(dc)
+
+	var nfe error
+
+	// List virtual machines for every argument
+	for _, arg := range args {
+		vms, err := f.VirtualMachineList(context.TODO(), arg)
+		if err != nil {
+			if _, ok := err.(*find.NotFoundError); ok {
+				// Let caller decide how to handle NotFoundError
+				nfe = err
+				continue
+			}
+			return nil, err
+		}
+
+		out = append(out, vms...)
+	}
+
+	return out, nfe
+}
+
+func (c *VSphereCloud) DeleteCloudInitISO(vm *string) error {
+	f := find.NewFinder(c.Client.Client, true)
+	ctx, cancel := context.WithCancel(context.Background())
+	defer cancel()
+
+	dc, err := f.Datacenter(ctx, c.Datacenter)
+	if err != nil {
+		return err
+	}
+	f.SetDatacenter(dc)
+
+	vmRef, err := f.VirtualMachine(ctx, *vm)
+	if err != nil {
+		return err
+	}
+
+	var refs []types.ManagedObjectReference
+	refs = append(refs, vmRef.Reference())
+	var vmResult mo.VirtualMachine
+
+	pc := property.DefaultCollector(c.Client.Client)
+	err = pc.RetrieveOne(ctx, vmRef.Reference(), []string{"datastore"}, &vmResult)
+	if err != nil {
+		glog.Fatalf("Unable to retrieve VM summary for VM %s", *vm)
+	}
+	glog.V(4).Infof("vm property collector result :%+v\n", vmResult)
+
+	// We expect the VM to be on only 1 datastore
+	dsRef := vmResult.Datastore[0].Reference()
+	var dsResult mo.Datastore
+	err = pc.RetrieveOne(ctx, dsRef, []string{"summary"}, &dsResult)
+	if err != nil {
+		glog.Fatalf("Unable to retrieve datastore summary for datastore  %s", dsRef)
+	}
+	glog.V(4).Infof("datastore property collector result :%+v\n", dsResult)
+	dsObj, err := f.Datastore(ctx, dsResult.Summary.Name)
+	if err != nil {
+		return err
+	}
+	isoFileName := getCloudInitFileName(*vm)
+	fileManager := dsObj.NewFileManager(dc, false)
+	err = fileManager.DeleteFile(ctx, isoFileName)
+	if err != nil {
+		if types.IsFileNotFound(err) {
+			glog.Warningf("ISO file not found: %q", isoFileName)
+			return nil
+		}
+		return err
+	}
+	glog.V(2).Infof("Deleted ISO file %q", isoFileName)
+	return nil
+}
+
 func getCloudInitFileName(vmName string) string {
 	return vmName + "/" + cloudInitFile
 }

upup/pkg/kutil/delete_cluster_vsphere.go

@@ -17,10 +17,117 @@ limitations under the License.
 package kutil
 
 import (
-	"fmt"
+	"context"
+	"github.com/golang/glog"
+	"github.com/vmware/govmomi/find"
+	"github.com/vmware/govmomi/object"
+	"k8s.io/kops/upup/pkg/fi"
+	"k8s.io/kops/upup/pkg/fi/cloudup/vsphere"
 )
 
-func (c *AwsCluster) listResourcesVSphere() (map[string]*ResourceTracker, error) {
-	fmt.Print("listResourcesVSphere to be implemented")
-	return nil, nil
+const (
+	typeVM = "VM"
+)
+
+type clusterDiscoveryVSphere struct {
+	cloud        fi.Cloud
+	vsphereCloud *vsphere.VSphereCloud
+	clusterName  string
+}
+
+type vsphereListFn func() ([]*ResourceTracker, error)
+
+func (c *AwsCluster) listResourcesVSphere() (map[string]*ResourceTracker, error) {
+	vsphereCloud := c.Cloud.(*vsphere.VSphereCloud)
+
+	resources := make(map[string]*ResourceTracker)
+
+	d := &clusterDiscoveryVSphere{
+		cloud:        c.Cloud,
+		vsphereCloud: vsphereCloud,
+		clusterName:  c.ClusterName,
+	}
+
+	listFunctions := []vsphereListFn{
+		d.listVMs,
+	}
+
+	for _, fn := range listFunctions {
+		trackers, err := fn()
+		if err != nil {
+			return nil, err
+		}
+		for _, t := range trackers {
+			resources[GetResourceTrackerKey(t)] = t
+		}
+	}
+
+	return resources, nil
+}
+
+func (d *clusterDiscoveryVSphere) listVMs() ([]*ResourceTracker, error) {
+	c := d.vsphereCloud
+
+	regexForMasterVMs := "*" + "." + "masters" + "." + d.clusterName + "*"
+	regexForNodeVMs := "nodes" + "." + d.clusterName + "*"
+
+	vms, err := c.GetVirtualMachines([]string{regexForMasterVMs, regexForNodeVMs})
+	if err != nil {
+		if _, ok := err.(*find.NotFoundError); !ok {
+			return nil, err
+		}
+		glog.Warning(err)
+	}
+
+	var trackers []*ResourceTracker
+	for _, vm := range vms {
+		tracker := &ResourceTracker{
+			Name:    vm.Name(),
+			ID:      vm.Name(),
+			Type:    typeVM,
+			deleter: deleteVM,
+			Dumper:  DumpVMInfo,
+			obj:     vm,
+		}
+		trackers = append(trackers, tracker)
+	}
+	return trackers, nil
+}
+
+func deleteVM(cloud fi.Cloud, r *ResourceTracker) error {
+	vsphereCloud := cloud.(*vsphere.VSphereCloud)
+
+	vm := r.obj.(*object.VirtualMachine)
+
+	task, err := vm.PowerOff(context.TODO())
+	if err != nil {
+		return err
+	}
+	task.Wait(context.TODO())
+
+	vsphereCloud.DeleteCloudInitISO(fi.String(vm.Name()))
+
+	task, err = vm.Destroy(context.TODO())
+	if err != nil {
+		return err
+	}
+
+	err = task.Wait(context.TODO())
+	if err != nil {
+		glog.Fatalf("Destroy VM failed: %q", err)
+	}
+
+	return nil
+}
+
+func DumpVMInfo(r *ResourceTracker) (interface{}, error) {
+	data := make(map[string]interface{})
+	data["id"] = r.ID
+	data["type"] = r.Type
+	data["raw"] = r.obj
+	return data, nil
+}
+
+func GetResourceTrackerKey(t *ResourceTracker) string {
+	return t.Type + ":" + t.ID
 }