implement vc inventory scanning

2024-09-30 10:36:23 +10:00
parent a91642b450
commit c4eedb55b7
8 changed files with 533 additions and 37 deletions
--- a/internal/vcenter/vcenter.go
+++ b/internal/vcenter/vcenter.go
@@ -8,6 +8,7 @@ import (
 	"net/url"
 	"os"
 	"path"
+	"strings"

 	"github.com/vmware/govmomi"
 	"github.com/vmware/govmomi/find"
@@ -20,6 +21,7 @@ import (

 type Vcenter struct {
 	Logger      *slog.Logger
+	Vurl        string
 	ctx         context.Context
 	client      *govmomi.Client
 	credentials *VcenterLogin
@@ -61,6 +63,7 @@ func (v *Vcenter) Login(vUrl string) error {
 	if err != nil {
 		log.Fatalf("Error parsing vCenter URL: %s", err)
 	}
+	v.Vurl = vUrl

 	u.User = url.UserPassword(v.credentials.Username, v.credentials.Password)

@@ -108,6 +111,236 @@ func (v *Vcenter) Logout() error {

 }

+func (v *Vcenter) GetAllVmReferences() ([]*object.VirtualMachine, error) {
+	var results []*object.VirtualMachine
+	finder := find.NewFinder(v.client.Client, true)
+
+	m := view.NewManager(v.client.Client)
+
+	vms, err := m.CreateContainerView(v.ctx, v.client.ServiceContent.RootFolder, []string{"VirtualMachine"}, true)
+	if err != nil {
+		return nil, err
+	}
+	defer vms.Destroy(v.ctx)
+
+	// List all datacenters
+	datacenters, err := finder.DatacenterList(v.ctx, "*")
+	if err != nil {
+		return nil, fmt.Errorf("failed to list datacenters: %w", err)
+	}
+
+	for _, dc := range datacenters {
+		v.Logger.Debug("Getting VMs in", "datacenter", dc.Name())
+		// Set the current datacenter
+		finder.SetDatacenter(dc)
+
+		// Get the list of all virtual machines in the current datacenter
+		vms, err := finder.VirtualMachineList(v.ctx, "*")
+		if err != nil {
+			v.Logger.Error("Failed to list VMs in", "datacenter", dc.Name(), "error", err)
+			continue
+		}
+
+		for _, vm := range vms {
+			//vmRef := vm.Reference()
+			//v.Logger.Debug("result", "vm", vm, "MoRef", vmRef, "path", vm.InventoryPath)
+			results = append(results, vm)
+		}
+
+	}
+
+	v.Logger.Debug("Found VM references", "count", len(results))
+
+	return results, err
+}
+
+func (v *Vcenter) ConvertObjToMoVM(vmObj *object.VirtualMachine) (*mo.VirtualMachine, error) {
+	// Use the InventoryPath to extract the datacenter name and VM path
+	inventoryPath := vmObj.InventoryPath
+	parts := strings.SplitN(inventoryPath, "/", 3)
+
+	if len(parts) < 2 {
+		return nil, fmt.Errorf("invalid InventoryPath: %s", inventoryPath)
+	}
+
+	// The first part of the path is the datacenter name
+	datacenterName := parts[1]
+
+	// Finder to search for datacenter and VM
+	finder := find.NewFinder(v.client.Client, true)
+
+	// Find the specific datacenter by name
+	datacenter, err := finder.Datacenter(v.ctx, fmt.Sprintf("/%s", datacenterName))
+	if err != nil {
+		return nil, fmt.Errorf("failed to find datacenter %s: %w", datacenterName, err)
+	}
+
+	// Set the found datacenter in the finder
+	finder.SetDatacenter(datacenter)
+
+	// Now retrieve the VM using its ManagedObjectReference
+	vmRef := vmObj.Reference()
+
+	// Retrieve the full mo.VirtualMachine object for the reference
+	var moVM mo.VirtualMachine
+	err = v.client.RetrieveOne(v.ctx, vmRef, nil, &moVM)
+	if err != nil {
+		return nil, fmt.Errorf("failed to retrieve VM %s in datacenter %s: %w", vmObj.Name(), datacenterName, err)
+	}
+
+	// Return the found mo.VirtualMachine object
+	v.Logger.Debug("Found VM in datacenter", "vm_name", moVM.Name, "dc_name", datacenterName)
+	return &moVM, nil
+}
+
+func (v *Vcenter) ConvertObjToMoHost(hostObj *object.HostSystem) (*mo.HostSystem, error) {
+	// Use the InventoryPath to extract the datacenter name and Host path
+	inventoryPath := hostObj.InventoryPath
+	parts := strings.SplitN(inventoryPath, "/", 3)
+	v.Logger.Debug("inventory path", "parts", parts)
+
+	if len(parts) < 2 {
+		return nil, fmt.Errorf("invalid InventoryPath: %s", inventoryPath)
+	}
+
+	// The first part of the path is the datacenter name
+	datacenterName := parts[1]
+
+	// Finder to search for datacenter and VM
+	finder := find.NewFinder(v.client.Client, true)
+
+	// Find the specific datacenter by name
+	datacenter, err := finder.Datacenter(v.ctx, fmt.Sprintf("/%s", datacenterName))
+	if err != nil {
+		return nil, fmt.Errorf("failed to find datacenter %s: %w", datacenterName, err)
+	}
+
+	// Set the found datacenter in the finder
+	finder.SetDatacenter(datacenter)
+
+	// Now retrieve the VM using its ManagedObjectReference
+	hostRef := hostObj.Reference()
+
+	// Retrieve the full mo.HostSystem object for the reference
+	var moHost mo.HostSystem
+	err = v.client.RetrieveOne(v.ctx, hostRef, nil, &moHost)
+	if err != nil {
+		return nil, fmt.Errorf("failed to retrieve Host %s in datacenter %s: %w", hostObj.Name(), datacenterName, err)
+	}
+
+	// Return the found mo.HostSystem object
+	v.Logger.Debug("Found Host in datacenter", "host_name", moHost.Name, "dc_name", datacenterName)
+	return &moHost, nil
+}
+
+func (v *Vcenter) GetHostSystemObject(hostRef types.ManagedObjectReference) (*mo.HostSystem, error) {
+	finder := find.NewFinder(v.client.Client, true)
+
+	// List all datacenters
+	datacenters, err := finder.DatacenterList(v.ctx, "*")
+	if err != nil {
+		return nil, fmt.Errorf("failed to list datacenters: %w", err)
+	}
+
+	for _, dc := range datacenters {
+		v.Logger.Debug("Checking dc for host", "name", dc.Name(), "hostRef", hostRef.String())
+		// Set the current datacenter
+		finder.SetDatacenter(dc)
+
+		var hs mo.HostSystem
+		err := v.client.RetrieveOne(v.ctx, hostRef, nil, &hs)
+		if err != nil {
+			return nil, err
+		} else {
+			v.Logger.Debug("Found hostsystem", "name", hs.Name)
+			return &hs, nil
+		}
+	}
+
+	return nil, nil
+}
+
+// Function to find the cluster or compute resource from a host reference
+func (v *Vcenter) GetClusterFromHost(hostRef *types.ManagedObjectReference) (string, error) {
+	// Get the host object
+	host, err := v.GetHostSystemObject(*hostRef)
+	if err != nil {
+		v.Logger.Error("cant get host", "error", err)
+		return "", err
+	}
+
+	v.Logger.Debug("host parent", "parent", host.Parent)
+
+	if host.Parent.Type == "ClusterComputeResource" {
+		// Retrieve properties of the compute resource
+		var moCompute mo.ComputeResource
+		err = v.client.RetrieveOne(v.ctx, *host.Parent, nil, &moCompute)
+		if err != nil {
+			return "", fmt.Errorf("failed to retrieve compute resource: %w", err)
+		}
+		v.Logger.Debug("VM is on host in cluster/compute resource", "name", moCompute.Name)
+		return moCompute.Name, nil
+	}
+
+	return "", nil
+}
+
+// Function to determine the datacenter a VM belongs to
+func (v *Vcenter) GetDatacenterForVM(vm mo.VirtualMachine) (string, error) {
+	// Start with the VM's parent reference
+	ref := vm.Reference()
+
+	// Traverse the inventory hierarchy upwards to find the datacenter
+	for {
+		// Get the parent reference of the current object
+		parentRef, err := v.getParent(ref)
+		if err != nil {
+			return "", fmt.Errorf("failed to get parent object: %w", err)
+		}
+
+		// If we get a nil parent reference, it means we've hit the root without finding the datacenter
+		if parentRef == nil {
+			return "", fmt.Errorf("failed to find datacenter for VM")
+		}
+
+		// Check if the parent is a Datacenter
+		switch parentRef.Type {
+		case "Datacenter":
+			// If we found a Datacenter, retrieve its properties
+			datacenter := object.NewDatacenter(v.client.Client, *parentRef)
+			var moDC mo.Datacenter
+			err = v.client.RetrieveOne(v.ctx, datacenter.Reference(), nil, &moDC)
+			if err != nil {
+				return "", fmt.Errorf("failed to retrieve datacenter: %w", err)
+			}
+
+			//log.Printf("VM is in datacenter: %s", moDC.Name)
+			v.Logger.Debug("VM datacenter found", "vm_name", vm.Name, "dc_name", moDC.Name)
+			return moDC.Name, nil
+
+		default:
+			// Continue traversing upwards if not a Datacenter
+			ref = *parentRef
+		}
+	}
+}
+
+// Helper function to get the parent ManagedObjectReference of a given object
+func (v *Vcenter) getParent(ref types.ManagedObjectReference) (*types.ManagedObjectReference, error) {
+	// Retrieve the object's properties
+	var obj mo.ManagedEntity
+	err := v.client.RetrieveOne(v.ctx, ref, []string{"parent"}, &obj)
+	if err != nil {
+		return nil, fmt.Errorf("failed to retrieve parent of object: %w", err)
+	}
+
+	// Return the parent reference
+	if obj.Parent != nil {
+		return obj.Parent, nil
+	}
+	return nil, nil
+}
+
 func (v *Vcenter) FindVMByName(vmName string) ([]mo.VirtualMachine, error) {
 	m := view.NewManager(v.client.Client)

@@ -183,10 +416,9 @@ func (v *Vcenter) FindVMByID(vmID string) (*VmProperties, error) {
 	return nil, fmt.Errorf("VM with ID %s not found in any datacenter", vmID)
 }

-func (v *Vcenter) FindVMByIDWithDatacenter(vmID string, dcID string) (*VmProperties, error) {
-	//var dcName string
+func (v *Vcenter) FindVMByIDWithDatacenter(vmID string, dcID string) (*mo.VirtualMachine, error) {
 	var err error
-	resourcePool := ""
+	//resourcePool := ""
 	v.Logger.Debug("searching for vm id", "vm_id", vmID, "datacenter_id", dcID)

 	finder := find.NewFinder(v.client.Client, true)
@@ -218,24 +450,28 @@ func (v *Vcenter) FindVMByIDWithDatacenter(vmID string, dcID string) (*VmPropert
 	if err == nil {
 		v.Logger.Debug("Found VM")

-		// Retrieve the resource pool the VM is in
-		if vm.ResourcePool != nil {
-			rp := object.NewResourcePool(v.client.Client, *vm.ResourcePool)
-			rpName, err := rp.ObjectName(v.ctx)
-			if err != nil {
-				v.Logger.Error("failed to get resource pool name", "error", err)
-			} else {
-				v.Logger.Debug("Found resource pool name", "rp_name", rpName)
-				resourcePool = rpName
+		/*
+			// Retrieve the resource pool the VM is in
+			if vm.ResourcePool != nil {
+				rp := object.NewResourcePool(v.client.Client, *vm.ResourcePool)
+				rpName, err := rp.ObjectName(v.ctx)
+				if err != nil {
+					v.Logger.Error("failed to get resource pool name", "error", err)
+				} else {
+					v.Logger.Debug("Found resource pool name", "rp_name", rpName)
+					resourcePool = rpName
+				}
 			}
+		*/

-		}
-
-		return &VmProperties{
-			//Datacenter: dcName,
-			Vm:           vm,
-			ResourcePool: resourcePool,
-		}, nil
+		return &vm, nil
+		/*
+			return &VmProperties{
+				//Datacenter: dcName,
+				Vm:           vm,
+				ResourcePool: resourcePool,
+			}, nil
+		*/
 	} else if _, ok := err.(*find.NotFoundError); !ok {
 		// If the error is not a NotFoundError, return it
 		//return nil, fmt.Errorf("failed to retrieve VM with ID %s in datacenter %s: %w", vmID, dc.Name(), err)
@@ -244,9 +480,23 @@ func (v *Vcenter) FindVMByIDWithDatacenter(vmID string, dcID string) (*VmPropert
 	} else {
 		return nil, fmt.Errorf("failed to retrieve VM: %w", err)
 	}
+}

-	//v.Logger.Info("Unable to find vm in datacenter", "vm_id", vmID, "datacenter_id", dcID)
-	//return nil, nil
+// Helper function to retrieve the resource pool for the VM
+func (v *Vcenter) GetVmResourcePool(vm mo.VirtualMachine) (string, error) {
+	var resourcePool string
+	if vm.ResourcePool != nil {
+		rp := object.NewResourcePool(v.client.Client, *vm.ResourcePool)
+		rpName, err := rp.ObjectName(v.ctx)
+		if err != nil {
+			v.Logger.Error("failed to get resource pool name", "error", err)
+			return resourcePool, err
+		} else {
+			v.Logger.Debug("Found resource pool name", "rp_name", rpName)
+			resourcePool = rpName
+		}
+	}
+	return resourcePool, nil
 }

 // Helper function to retrieve the full folder path for the VM