finds disks and puts dirs in the protobuf cluster

Signed-off-by: Jeff Carr <[email protected]>

1 files changed, 43 insertions, 8 deletions

diff --git a/start.go b/start.go
index 3cd319b..b682926 100644
--- a/start.go
+++ b/start.go
@@ -5,14 +5,16 @@ package main
 import (
 	"fmt"
 	"os"
+	"path/filepath"
 
 	"go.wit.com/log"
 	"libvirt.org/go/libvirtxml"
 )
 
-func makeDroplet(start string) {
-	tmp := findDroplet(start)
-	d := tmp.pb
+// generate the XML for 'virsh create'
+func startDropletXml(start string) {
+	meDrop := findDroplet(start)
+	d := meDrop.pb
 	if d == nil {
 		log.Info("droplet is unknown:", start)
 		os.Exit(0)
@@ -21,8 +23,8 @@ func makeDroplet(start string) {
 	domcfg := &libvirtxml.Domain{}
 
 	addDefaultXml(domcfg, "standard.x86")
-	addDefaultXml(domcfg, "memory")
-	addDefaultXml(domcfg, "network")
+	// addDefaultXml(domcfg, "memory")
+	// addDefaultXml(domcfg, "network")
 	addDefaultXml(domcfg, "spice")
 	addDefaultXml(domcfg, "qcow")
 
@@ -33,6 +35,12 @@ func makeDroplet(start string) {
 	var i uint
 	i = uint(d.Memory / (1024 * 1024))
 
+	// var tmp string
+	// tmp = domcfg.VCPU
+	domcfg.VCPU = new(libvirtxml.DomainVCPU)
+	domcfg.VCPU.Value = uint(d.Cpus)
+
+	domcfg.Memory = new(libvirtxml.DomainMemory)
 	domcfg.Memory.Value = i
 	domcfg.Memory.Unit = "MiB"
 
@@ -48,8 +56,35 @@ func makeDroplet(start string) {
 	// add a check here to make these unique
 	// setRandomMacs(domcfg)
 
-	qcow := "/home/nfs/" + d.Hostname + ".qcow2"
-	setSimpleDisk(domcfg, qcow)
+	for _, disk := range d.Disks {
+		fullname := findDisk(disk.Filename)
+		if fullname == "" {
+			log.Info("can not find disk", d.Hostname, "dir", disk.Filepath, "filename", disk.Filename)
+		} else {
+			// qcow := "/home/nfs/" + d.Hostname + ".qcow2"
+			setSimpleDisk(domcfg, fullname)
+		}
+	}
 
-	writeoutXml(domcfg, "blahcarr")
+	writeoutXml(domcfg, d.Hostname)
+	os.Exit(-1)
+}
+
+func findDisk(filename string) string {
+	for _, dirname := range me.cluster.Dirs {
+		// log.Info("look in dir", dirname)
+		var count int
+		newdir, _ := os.ReadDir(dirname)
+		for _, file := range newdir {
+			count += 1
+			if file.Name() == filename {
+				log.Info("Found file", filename, "in", dirname)
+				return filepath.Join(dirname, file.Name())
+			}
+		}
+		if count == 0 {
+			log.Info("Warning? dirname", dirname, "was empty. Not mounted?")
+		}
+	}
+	return ""
 }