Makefile

@@ -7,7 +7,7 @@ BUILDTIME = $(shell date +%Y.%m.%d_%H%M)
 # REDOMOD = $(shell if [ -e go.mod ]; then  echo go.mod; else echo no go mod; fi)
 REDOMOD = $(shell if [ -e go.sum ]; then  echo go.sum exists; else GO111MODULE= go mod init; GO111MODULE= go mod tidy; fi)
 
-all: goimports gocui-debugging
+all: goimports gocui
 	# ./zookeeper
 
 vet:
@@ -18,10 +18,7 @@ nogui:
 	./zookeeper --gui nocui
 
 gocui: build
-	./zookeeper --gui gocui
-
-gocui-debugging: build
-	./zookeeper --gui gocui --gui-file ~/go/src/go.wit.com/toolkits/gocui/gocui.so  >/tmp/forge.log 2>&1
+	./zookeeper --gui gocui  >/tmp/forge.log 2>&1
 
 build: goimports vet
 	GO111MODULE=off go build -v -x \
@@ -64,5 +61,14 @@ http-toogle-ZOOD:
 http-list-machines:
 	curl --silent http://localhost:8080/list
 
-http-uptime:
-	curl --silent http://localhost:8080/uptime
+http-ConfigSave:
+	curl --silent http://localhost:8080/save
+
+http-set-zood-target:
+	curl --silent "http://localhost:8080/target?package=zood&version=v0.0.8"
+
+http-upgrade-hpdev2.grid.wit.com:
+	curl --silent "http://localhost:8080/upgrade?hostname=hpdev2.grid.wit.com"
+
+http-upgrade-mirrors.wit.com:
+	curl --silent "http://localhost:8080/upgrade?hostname=mirrors.wit.com"

doGui.go

@@ -11,27 +11,17 @@ import (
 
 	"go.wit.com/gui"
 	"go.wit.com/lib/gadgets"
-	"go.wit.com/lib/protobuf/zoopb"
 	"go.wit.com/log"
 )
 
 // refresh the windows & tables the user has open
 func refresh() {
+	time.Sleep(90 * time.Second)
 	if argv.Verbose {
 		log.Info("zookeeper scan here")
 	}
 	if me.zood != nil {
-		me.zood.doMachinesUpgradeTable(me.machines)
-		all := me.machines.All()
-		for all.Scan() {
-			m := all.Next()
-			if me.hostname == m.Hostname {
-				// this is me! This is the version of zood that should be installed everywhere
-				v := findVersion(m, "zood")
-				me.zood.version = v
-				me.zood.versionL.SetText(v)
-			}
-		}
+		doMachinesUpgradeTable()
 	}
 }
 
@@ -61,7 +51,7 @@ func doGui() {
 			me.zood.Toggle()
 			return
 		}
-		me.zood = makeZoodWin()
+		makeZoodWin()
 	})
 
 	grid.NewButton("Cluster Events", func() {
@@ -71,18 +61,5 @@ func doGui() {
 	// sit here forever refreshing the GUI
 	for {
 		refresh()
-		time.Sleep(90 * time.Second)
 	}
 }
-
-func saveMachineState() {
-	cur := zoopb.NewMachines()
-
-	all := me.machines.SortByHostname()
-	for all.Scan() {
-		m := all.Next()
-		log.Info("have machine:", m.Hostname)
-		cur.Append(m)
-	}
-	cur.ConfigSave()
-}

http.go

@@ -9,6 +9,7 @@ import (
 	"net/http"
 	"strings"
 
+	"go.wit.com/lib/protobuf/zoopb"
 	"go.wit.com/log"
 )
 
@@ -23,7 +24,9 @@ func okHandler(w http.ResponseWriter, r *http.Request) {
 	route := cleanURL(r.URL.Path)
 
 	hostname := r.URL.Query().Get("hostname")
-	// flag := r.URL.Query().Get("flag")
+	flag := r.URL.Query().Get("flag")
+	packname := r.URL.Query().Get("package")
+	version := r.URL.Query().Get("version")
 
 	msg, err := ioutil.ReadAll(r.Body) // Read the body as []byte
 	if err != nil {
@@ -40,7 +43,6 @@ func okHandler(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 
-	/*
 	if route == "/status" {
 		var packs *zoopb.Packages
 		packs = new(zoopb.Packages)
@@ -54,10 +56,8 @@ func okHandler(w http.ResponseWriter, r *http.Request) {
 		fmt.Fprintln(w, "upgrade")
 		return
 	}
-	*/
 
 	// list out the machines and thier version of zood
-	/*
 	if route == "/list" {
 		log.HttpMode(w)
 		defer log.HttpMode(nil)
@@ -74,27 +74,53 @@ func okHandler(w http.ResponseWriter, r *http.Request) {
 		}
 		return
 	}
-	*/
 
-	if route == "/uptime" {
-		if me.zood == nil {
-			fmt.Fprintf(w, "BAD zood == nil\n")
+	// save the config file
+	if route == "/save" {
+		log.HttpMode(w)
+		defer log.HttpMode(nil)
+		if err := me.machines2.ConfigSave(); err == nil {
+			log.Log(NOW, "ConfigSave() ok")
+		} else {
+			log.Log(NOW, "ConfigSave() failed", err)
+		}
 		return
 	}
-		var count int
-		var bad int
-		all := me.machines.All()
-		for all.Scan() {
-			m := all.Next()
-			count += 1
-			if findVersion(m, "zood") != me.zood.version {
-				bad += 1
+
+	// flag a package to attempt to upgrade
+	if route == "/upgrade" {
+		log.HttpMode(w)
+		defer log.HttpMode(nil)
+		me.upgrade[hostname] = true
+		log.Log(NOW, "setting package ", packname, " to upgrade")
+		return
 	}
+
+	// set the target version for a package
+	if route == "/target" {
+		log.HttpMode(w)
+		defer log.HttpMode(nil)
+		// me.targets[packname] = version
+		log.Log(NOW, "setting package/version to ", packname, " ", version)
+		return
 	}
-		if bad == 0 {
-			fmt.Fprintf(w, "GOOD machine count=(%d) all machines are version %s\n", count, me.zood.version)
+
+	// toggle logging flags
+	if route == "/flag" {
+		log.HttpMode(w)
+		defer log.HttpMode(nil)
+		log.Info("going to toggle flag:", flag)
+		switch flag {
+		case "ZOOD":
+			if ZOOD.Enabled() {
+				log.Log(NOW, "toogle ZOOD false")
+				ZOOD.SetBool(false)
 			} else {
-			fmt.Fprintf(w, "BAD machine count=(%d) upgrade=(%d) to %s\n", count, bad, me.zood.version)
+				log.Log(NOW, "toogle ZOOD true")
+				ZOOD.SetBool(true)
+			}
+		default:
+			log.Info("unknown looging flag:", flag)
 		}
 		return
 	}

machine.go

@@ -43,7 +43,7 @@ func handleMachine(r *http.Request, w http.ResponseWriter, hostname string, data
 		am := new(zoopb.Machine)
 		am.Hostname = newm.Hostname
 		am.Memory = newm.Memory
-		// me.machines2.Append(am)
+		me.machines2.Append(am)
 		me.machines.Append(newm)
 		log.Info("new machine", am.Hostname, am.Memory)
 		return
@@ -59,7 +59,7 @@ func handleMachine(r *http.Request, w http.ResponseWriter, hostname string, data
 		fmt.Fprintln(w, "apt update")
 		m.Upgrade = false
 	} else {
-		fmt.Fprintln(w, "good")
+		fmt.Fprintln(w, "upgrade")
 	}
 	// log.Info("update machine protobuf", hostname)
 	updateMachine(newm)
@@ -75,12 +75,8 @@ func updateMachine(u *zoopb.Machine) string {
 	if m == nil {
 		log.Info("adding new machine", u.Hostname)
 		me.machines.Append(u)
-		if me.zood == nil {
-			// do nothing. window has not been opened
-		} else {
-			me.zood.doMachinesUpgradeTable(me.machines)
-		}
-		saveMachineState()
+		log.Info("save machines pb file here...")
+		me.machines2.ConfigSave()
 		return "new"
 	}
 	// log.Info("updating machine", m.Hostname)

main.go

@@ -37,19 +37,17 @@ func main() {
 
 	me = new(zookeep)
 	me.hostname, _ = os.Hostname()
-	me.pollDelay = time.Hour
+	me.pollDelay = 10 * time.Second
 	me.machines = zoopb.NewMachines()
-	// me.machines2 = zoopb.NewMachines()
+	me.machines2 = zoopb.NewMachines()
 	if err := me.machines.ConfigLoad(); err != nil {
 		log.Warn("load config failed", err)
 		os.Exit(-1)
 	}
-	/*
 	if err := me.machines2.ConfigLoad(); err != nil {
 		log.Warn("load config failed", err)
 		os.Exit(-1)
 	}
-	*/
 	// me.targets =  make(map[string]string) // keep track of what versions the machines should be running
 	me.upgrade = make(map[string]bool) // used to trigger upgrade attempts
 

structs.go

@@ -4,6 +4,7 @@
 package main
 
 import (
+	"sync"
 	"time"
 
 	"go.wit.com/gui"
@@ -22,7 +23,7 @@ type zookeep struct {
 	distro      string                 // debian,redhat,gentoo,macos,wincrap
 	packages    *zoopb.Packages        // installed packages and versions
 	machines    *zoopb.Machines        // every machine that has reported itself to the zookeeper
-	// machines2   *zoopb.Machines        // every machine that has reported itself to the zookeeper
+	machines2   *zoopb.Machines        // every machine that has reported itself to the zookeeper
 	targets     map[string]string      // what versions the machines should be running
 	upgrade     map[string]bool        // use this to trigger builds
 	myGui       *gui.Node              // the gui toolkit handle
@@ -31,3 +32,21 @@ type zookeep struct {
 	machinesTB  *zoopb.MachinesTable   // the machines gui table buffer
 	zood        *stdTableWin           // the zood version window
 }
+
+type stdTableWin struct {
+	sync.Mutex
+	win    *gadgets.GenericWindow // the machines gui window
+	box    *gui.Node              // the machines gui parent box widget
+	TB     *zoopb.MachinesTable   // the machines gui table buffer
+	update bool                   // if the window should be updated
+}
+
+func (w *stdTableWin) Toggle() {
+	if w == nil {
+		return
+	}
+	if w.win == nil {
+		return
+	}
+	w.win.Toggle()
+}

watchdog.go

@@ -55,7 +55,7 @@ func NewWatchdog() {
 					// log.Info("know about machine", m.Hostname, "zood version", zood.Version)
 				}
 			}
-			log.Info("hour watchdog:", counter, "machines. Current time:", t)
+			log.Info("zookeeper has", counter, "machines. Current time:", t)
 
 			// h.pollHypervisor()
 			// h.Scan()

windowZood.go

@@ -5,7 +5,6 @@ package main
 
 import (
 	"fmt"
-	"sync"
 	"time"
 
 	"go.wit.com/gui"
@@ -14,87 +13,42 @@ import (
 	"go.wit.com/log"
 )
 
-type stdTableWin struct {
-	sync.Mutex
-	win      *gadgets.GenericWindow // the machines gui window
-	box      *gui.Node              // the machines gui parent box widget
-	TB       *zoopb.MachinesTable   // the machines gui table buffer
-	version  string                 // the current zood version
-	versionL *gui.Node              // label widget to display the current zood version
-	update   bool                   // if the window should be updated
-}
-
-func (w *stdTableWin) Toggle() {
-	if w == nil {
-		return
-	}
-	if w.win == nil {
-		return
-	}
-	w.win.Toggle()
-}
-
-func makeZoodWin() *stdTableWin {
-	zood := new(stdTableWin)
-	zood.win = gadgets.NewGenericWindow("zood daemon versions", "todo: add global controls here")
-	zood.win.Custom = func() {
+func makeZoodWin() {
+	me.zood = new(stdTableWin)
+	me.zood.win = gadgets.NewGenericWindow("zood daemon versions", "todo: add global controls here")
+	me.zood.win.Custom = func() {
 		log.Info("test delete window here")
 	}
-	grid := zood.win.Group.RawGrid()
+	grid := me.zood.win.Group.RawGrid()
 	grid.NewButton("save machines.pb", func() {
 		saveMachineState()
 	})
-	grid.NewButton("show active", func() {
-		zood.doMachinesUpgradeTable(me.machines)
-	})
-	grid.NewButton("refresh", func() {
-		refresh()
-	})
-	zood.versionL = grid.NewLabel("scan")
-	grid.NewButton("show out of date", func() {
-		found := zoopb.NewMachines()
-		all := me.machines.All()
-		for all.Scan() {
-			m := all.Next()
-			if findVersion(m, "zood") != me.zood.version {
-				found.Append(m)
-			}
-		}
-		zood.doMachinesUpgradeTable(found)
+	grid.NewCheckbox("hide active")
+	grid.NewButton("update", func() {
+		doMachinesUpgradeTable()
 	})
 
 	// make a box at the bottom of the window for the protobuf table
-	zood.box = zood.win.Bottom.Box().SetProgName("TBOX")
-	zood.doMachinesUpgradeTable(me.machines)
-
-	return zood
+	me.zood.box = me.zood.win.Bottom.Box().SetProgName("TBOX")
+	doMachinesUpgradeTable()
 }
 
-func (zood *stdTableWin) doMachinesUpgradeTable(pb *zoopb.Machines) {
-	zood.Lock()
-	defer zood.Unlock()
-	if zood.TB != nil {
-		zood.TB.Delete()
-		zood.TB = nil
+func doMachinesUpgradeTable() {
+	me.zood.Lock()
+	defer me.zood.Unlock()
+	if me.zood.TB != nil {
+		me.zood.TB.Delete()
+		me.zood.TB = nil
 	}
 
-	/*
-		found := zoopb.NewMachines()
-		all := pb.SortByHostname()
-		for all.Scan() {
-			m := all.Next()
-			found.Append(m)
-		}
-	*/
-
 	// display the protobuf
-	zood.TB = AddMachinesPB(zood.box, pb)
+	me.zood.TB = AddMachinesPB(me.zood.box, me.machines)
 	f := func(m *zoopb.Machine) {
 		log.Info("Triggering machine", m.Hostname, "to upgrade zood")
 		m.Upgrade = true
 	}
-	zood.TB.Custom(f)
-	log.Info("table has uuid", zood.TB.GetUuid())
+	me.zood.TB.Custom(f)
+	log.Info("table has uuid", me.zood.TB.GetUuid())
 }
 
 func AddMachinesPB(tbox *gui.Node, pb *zoopb.Machines) *zoopb.MachinesTable {
@@ -102,22 +56,11 @@ func AddMachinesPB(tbox *gui.Node, pb *zoopb.Machines) *zoopb.MachinesTable {
 	t.NewUuid()
 	t.SetParent(tbox)
 
-	upbut := t.AddButtonFunc("upgrade", func(m *zoopb.Machine) string {
-		if me.zood != nil {
-			mver := findVersion(m, "zood")
-			if mver == me.zood.version {
-				return ""
-			} else {
-				// log.Info("machine mismatch", m.Hostname, mver, me.zood.version)
-			}
-		}
-		// log.Info("machine =", m.Hostname)
+	f := func(m *zoopb.Machine) string {
+		log.Info("machine =", m.Hostname)
 		return "now"
-	})
-	upbut.Custom = func(m *zoopb.Machine) {
-		log.Info("Triggering machine", m.Hostname, "to upgrade zood")
-		m.Upgrade = true
 	}
+	t.AddButtonFunc("upgrade", f)
 
 	t.AddHostname()
 	t.AddMemory()
@@ -129,10 +72,6 @@ func AddMachinesPB(tbox *gui.Node, pb *zoopb.Machines) *zoopb.MachinesTable {
 	t.AddStringFunc("zood", func(m *zoopb.Machine) string {
 		return findVersion(m, "zood")
 	})
-	delf := func(m *zoopb.Machine) string {
-		return "delete"
-	}
-	t.AddButtonFunc("delete", delf)
 
 	/*
 		// show if the machine needs to be upgraded
@@ -159,3 +98,15 @@ func findVersion(m *zoopb.Machine, pkgname string) string {
 	}
 	return zood.Version
 }
+
+func saveMachineState() {
+	cur := zoopb.NewMachines()
+
+	all := me.machines.SortByHostname()
+	for all.Scan() {
+		m := all.Next()
+		log.Info("have machine:", m.Hostname)
+		cur.Append(m)
+	}
+	cur.ConfigSave()
+}