1 files changed, 4 insertions, 73 deletions
diff --git a/main.go b/main.go
index b3057fb..5525b36 100644
--- a/main.go
+++ b/main.go
@@ -17,13 +17,10 @@ package main
 import (
 	"embed"
 	"flag"
-	"fmt"
-	"log"
 	"net"
 	"time"
 
 	hypervisor "github.com/digitalocean/go-qemu/hypervisor"
-	"github.com/digitalocean/go-qemu/qemu"
 )
 
 var Version string
@@ -31,6 +28,8 @@ var Version string
 //go:embed resources/*
 var resources embed.FS
 
+var hv *hypervisor.Hypervisor
+
 func main() {
 	flag.Parse()
 
@@ -42,77 +41,9 @@ func main() {
 	}
 
 	driver := hypervisor.NewRPCDriver(newConn)
-	hv := hypervisor.New(driver)
+	hv = hypervisor.New(driver)
 
-	// fmt.Printf("\n**********Domains**********\n")
-	domains, err := hv.Domains()
-	if err != nil {
-		log.Fatalf("Unable to get domains from hypervisor: %v", err)
-	}
-	for _, dom := range domains {
-		var name string
-		var err error
-		var drives []string
-		var status qemu.Status
-		name = dom.Name
-		// fmt.Printf("%s\n", dom.Name)
-		status, err = dom.Status()
-		if err != nil {
-			fmt.Println(name, status, err)
-			continue
-		}
-		// fmt.Printf("\nStatus: %s\n", status)
-		/*
-			if status == qemu.StatusRunning {
-				// getBlockDevices(dom)
-				// fmt.Println(name, status, err)
-			}
-		*/
-		blockDevices, err := dom.BlockDevices()
-		for _, blockDevice := range blockDevices {
-			drives = append(drives, blockDevice.Device)
-			drives = append(drives, blockDevice.Inserted.Driver)
-			drives = append(drives, blockDevice.Inserted.File)
-			// fmt.Printf("%20s %8s %30s\n",
-			// 	blockDevice.Device, blockDevice.Inserted.Driver, blockDevice.Inserted.File)
-		}
-		if status == qemu.StatusRunning {
-			fmt.Println("ON ", name, drives)
-		} else {
-			fmt.Println("OFF", status, name, drives)
-		}
-	}
-	// fmt.Printf("\n***************************\n")
+	poolHypervisor(hv)
 
 	startHTTP()
 }
-
-func displayBlockDevices(domain *qemu.Domain) {
-	// var []blks string
-	blockDevices, err := domain.BlockDevices()
-	if err != nil {
-		log.Fatalf("Error getting blockDevices: %v\n", blockDevices)
-	}
-	fmt.Printf("\n[ BlockDevices ]\n")
-	fmt.Printf("========================================================================\n")
-	fmt.Printf("%20s %8s %30s\n", "Device", "Driver", "File")
-	fmt.Printf("========================================================================\n")
-	for _, blockDevice := range blockDevices {
-		fmt.Printf("%20s %8s %30s\n",
-			blockDevice.Device, blockDevice.Inserted.Driver, blockDevice.Inserted.File)
-	}
-}
-
-func displayPCIDevices(domain *qemu.Domain) {
-	pciDevices, err := domain.PCIDevices()
-	if err != nil {
-		log.Fatalf("Error getting PCIDevices: %v\n", pciDevices)
-	}
-	fmt.Printf("\n[ PCIDevices ]\n")
-	fmt.Printf("======================================\n")
-	fmt.Printf("%10s %20s\n", "[ID]", "[Description]")
-	fmt.Printf("======================================\n")
-	for _, pciDevice := range pciDevices {
-		fmt.Printf("[%10s] [%20s]\n", pciDevice.QdevID, pciDevice.ClassInfo.Desc)
-	}
-}