From 8cab0857fda04a5e9b06860aec15d05342b747f1 Mon Sep 17 00:00:00 2001
From: Jeff Carr <jcarr@wit.com>
Date: Tue, 22 Oct 2024 03:50:01 -0500
Subject: [PATCH] next step, write out yaml or json

Signed-off-by: Jeff Carr <jcarr@wit.com>
---
 configfile/main.go | 14 ++++++++++----
 droplet.proto      |  2 ++
 hypervisor.proto   | 10 ++++++----
 sampleData.go      | 22 ++++++++++++++++++++++
 4 files changed, 40 insertions(+), 8 deletions(-)

diff --git a/configfile/main.go b/configfile/main.go
index 627649f..4e873fa 100644
--- a/configfile/main.go
+++ b/configfile/main.go
@@ -26,9 +26,15 @@ func main() {
 		log.Fatalln("Failed to parse droplet:", err)
 	}
 
-	log.Println(aCluster)
-	for i, d := range aCluster.Droplets {
-		log.Println("\tdrop =", i, d)
+	// log.Println(aCluster)
+	// show the droplets to STDOUT
+	for _, d := range aCluster.Droplets {
+		log.Println("\tdroplet =", d.Hostname, "preffered host:", d.PreferredHypervisor)
+	}
+
+	// show the hypervisors to STDOUT
+	for _, h := range aCluster.Hypervisors {
+		log.Println("\thypervisor =", h.Hostname)
 	}
 }
 
@@ -50,7 +56,7 @@ func marshalWriteToFile(myWriter *bufio.Writer, c *pb.Cluster) {
 func TestWriteCluster() {
 	buf := new(bytes.Buffer)
 
-	c := pb.CreateSampleCluster(3)
+	c := pb.CreateSampleCluster(7)
 
 	got := buf.String()
 	log.Println(got)
diff --git a/droplet.proto b/droplet.proto
index 7cf14de..7f54c63 100644
--- a/droplet.proto
+++ b/droplet.proto
@@ -14,6 +14,8 @@ message Droplet {
 	repeated Disk disks = 9;
 
 	string	comment = 10;
+	string	default_state = 11;
+	string	preferred_hypervisor = 12;
 
 	message Network {
 		string	mac     = 1;
diff --git a/hypervisor.proto b/hypervisor.proto
index aa2a9da..c76612d 100644
--- a/hypervisor.proto
+++ b/hypervisor.proto
@@ -2,8 +2,10 @@ syntax = "proto3";
 package virtbuf;
 
 message Hypervisor {
-	string	hostname = 1;
-	bool	active = 2;
-	int64	cpus	= 3;
-	int64	memory	= 4;
+	string	uuid = 1;
+	string	hostname = 2;
+	bool	active = 3;
+	int64	cpus	= 4;
+	int64	memory	= 5;
+	string	comment	= 6;
 }
diff --git a/sampleData.go b/sampleData.go
index d0932d4..eaf4cb7 100644
--- a/sampleData.go
+++ b/sampleData.go
@@ -25,6 +25,19 @@ func CreateSampleDroplet(hostname string) *Droplet {
 	return d
 }
 
+func CreateSampleHypervisor(hostname string) *Hypervisor {
+	// Generate a new UUID
+	id := uuid.New()
+	h := &Hypervisor{
+		Uuid:     id.String(),
+		Hostname: hostname,
+		Cpus: 16,
+		Memory: 256,
+		Comment:  "this is a fake hypervisor",
+	}
+	return h
+}
+
 func CreateSampleCluster(total int) *Cluster {
 	var c *Cluster
 	c = new(Cluster)
@@ -33,9 +46,18 @@ func CreateSampleCluster(total int) *Cluster {
 		hostname := fmt.Sprintf("bmath%d.wit.com", i)
 		d := CreateSampleDroplet(hostname)
 		d.Comment = fmt.Sprintf("Sample Droplet %d", i)
+		d.PreferredHypervisor = fmt.Sprintf("farm%d", i)
 
 		c.Droplets = append(c.Droplets, d)
 	}
 
+	for i := 0; i < 3; i++ {
+		hostname := fmt.Sprintf("farm%d", i)
+		h := CreateSampleHypervisor(hostname)
+		h.Comment = fmt.Sprintf("Sample hypervisor %d", i)
+
+		c.Hypervisors = append(c.Hypervisors, h)
+	}
+
 	return c
 }