generate GUI table code

1 files changed, 213 insertions, 0 deletions

diff --git a/GuiGenerate b/GuiGenerate
new file mode 100644
index 0000000..aac8c37
--- /dev/null
+++ b/GuiGenerate
@@ -0,0 +1,213 @@
+package zoopb
+
+import (
+	"time"
+
+	"go.wit.com/gui"
+	"go.wit.com/lib/protobuf/guipb"
+	"go.wit.com/log"
+	timestamppb "google.golang.org/protobuf/types/known/timestamppb"
+)
+
+func (x *Machines) NewTable(title string) *MachinesTable {
+	t := new(MachinesTable)
+	t.x = x
+	pb := new(guipb.Table)
+	pb.Title = title
+	t.pb = pb
+	return t
+}
+
+func (t *MachinesTable) AddStringFunc(title string, f func(*Machine) string) {
+	t.pb.Order = append(t.pb.Order, title)
+
+	sf := new(MachineStringFunc)
+	sf.title = title
+	sf.f = f
+	t.stringFuncs = append(t.stringFuncs, sf)
+}
+
+func (t *MachinesTable) AddIntFunc(title string, f func(*Machine) int) {
+	t.pb.Order = append(t.pb.Order, title)
+
+	sf := new(MachineIntFunc)
+	sf.title = title
+	sf.f = f
+	t.intFuncs = append(t.intFuncs, sf)
+}
+
+func (t *MachinesTable) AddTimeFunc(title string, f func(*Machine) time.Time) {
+	t.pb.Order = append(t.pb.Order, title)
+
+	sf := new(MachineTimeFunc)
+	sf.title = title
+	sf.f = f
+	t.timeFuncs = append(t.timeFuncs, sf)
+}
+
+func (mt *MachinesTable) ShowTable() {
+	log.Info("zoopb.ShowTable() SENDING TO GUI")
+	mt.MakeTable()
+	gui.ShowTable(mt.pb)
+}
+
+type MachineStringFunc struct {
+	title string
+	f     func(*Machine) string
+}
+
+type MachineIntFunc struct {
+	title string
+	f     func(*Machine) int
+}
+
+type MachineTimeFunc struct {
+	title string
+	f     func(*Machine) time.Time
+}
+
+type MachinesTable struct {
+	// gt        *gui.NodeTable
+	pb          *guipb.Table
+	x           *Machines
+	hostnames   []string
+	stringFuncs []*MachineStringFunc
+	intFuncs    []*MachineIntFunc
+	timeFuncs   []*MachineTimeFunc
+}
+
+func (mt *MachinesTable) doStringFunc(name string) bool {
+	for _, sf := range mt.stringFuncs {
+		if sf.title != name {
+			continue
+		}
+		log.Info("zoopb: found stringfunc name:", name)
+		r := new(guipb.StringRow)
+		r.Header = new(guipb.Widget)
+		r.Header.Name = name
+		all := mt.x.All()
+		for all.Scan() {
+			m := all.Next()
+			r.Vals = append(r.Vals, sf.f(m))
+			log.Info("zoopb: adding", name, r.Vals)
+		}
+		mt.pb.StringRows = append(mt.pb.StringRows, r)
+		return true
+	}
+	return false
+}
+
+func (mt *MachinesTable) doIntFunc(name string) bool {
+	for _, sf := range mt.intFuncs {
+		if sf.title != name {
+			continue
+		}
+		log.Info("zoopb: found intfunc name:", name)
+		r := new(guipb.IntRow)
+		r.Header = new(guipb.Widget)
+		r.Header.Name = name
+		all := mt.x.All()
+		for all.Scan() {
+			m := all.Next()
+			r.Vals = append(r.Vals, int64(sf.f(m)))
+			log.Info("zoopb: adding", name, r.Vals)
+		}
+		mt.pb.IntRows = append(mt.pb.IntRows, r)
+		return true
+	}
+	return false
+}
+
+func (mt *MachinesTable) doTimeFunc(name string) bool {
+	for _, sf := range mt.timeFuncs {
+		if sf.title != name {
+			continue
+		}
+		log.Info("zoopb: found timefunc name:", name)
+		r := new(guipb.TimeRow)
+		r.Header = new(guipb.Widget)
+		r.Header.Name = name
+		all := mt.x.All()
+		for all.Scan() {
+			m := all.Next()
+			t := sf.f(m)
+			r.Vals = append(r.Vals, timestamppb.New(t)) // convert to protobuf time
+			log.Info("zoopb: adding", name, r.Vals)
+		}
+		mt.pb.TimeRows = append(mt.pb.TimeRows, r)
+		return true
+	}
+	return false
+}
+
+func (t *MachinesTable) AddHostname() {
+	// t.pb.Order = append(t.pb.Order, "Hostname")
+
+	t.AddStringFunc("Hostname", func(m *zoopb.Machine) string {
+		return m.Hostname
+	})
+}
+
+func (t *MachinesTable) AddMemory() {
+	t.pb.Order = append(t.pb.Order, "Memory")
+}
+
+func (t *MachinesTable) AddCpus() {
+	t.pb.Order = append(t.pb.Order, "Cpus")
+}
+
+func (mt *MachinesTable) MakeTable() {
+	for _, name := range mt.pb.Order {
+		log.Info("zoopb: looking for row name()", name)
+		switch name {
+		case "Hostname":
+			r := new(guipb.StringRow)
+			r.Header = new(guipb.Widget)
+			r.Header.Name = name
+			all := mt.x.All()
+			for all.Scan() {
+				m := all.Next()
+				r.Vals = append(r.Vals, m.Hostname)
+				log.Info("zoopb: adding", name, r.Vals)
+			}
+			mt.pb.StringRows = append(mt.pb.StringRows, r)
+			continue
+		case "Cpus":
+			i := new(guipb.IntRow)
+			i.Header = new(guipb.Widget)
+			i.Header.Name = name
+			all := mt.x.All()
+			for all.Scan() {
+				m := all.Next()
+				i.Vals = append(i.Vals, m.Cpus)
+				log.Info("zoopb: adding", name, i.Vals)
+			}
+			mt.pb.IntRows = append(mt.pb.IntRows, i)
+			continue
+		case "Memory":
+			i := new(guipb.IntRow)
+			i.Header = new(guipb.Widget)
+			i.Header.Name = name
+			all := mt.x.All()
+			for all.Scan() {
+				m := all.Next()
+				i.Vals = append(i.Vals, m.Memory)
+				log.Info("zoopb: adding", name, i.Vals)
+			}
+			mt.pb.IntRows = append(mt.pb.IntRows, i)
+			continue
+		default:
+			// mt.addFuncRow(name)
+		}
+		log.Info("zoopb: didn't find name. trying StringFuncs", name)
+		if mt.doStringFunc(name) {
+			continue
+		}
+		if mt.doIntFunc(name) {
+			continue
+		}
+		if mt.doTimeFunc(name) {
+			continue
+		}
+	}
+}