keep working towards update workingv0.0.6

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

1 files changed, 12 insertions, 9 deletions

diff --git a/send.go b/send.go
index 99a74ba..045c625 100644
--- a/send.go
+++ b/send.go
@@ -3,6 +3,7 @@
 package main
 
 import (
+	"os"
 	"strings"
 
 	"go.wit.com/lib/gui/shell"
@@ -42,7 +43,7 @@ func pingStatus() error {
 	return nil
 }
 
-func sendMachine() error {
+func sendMachine(s string) error {
 	var url string
 	url = urlbase + "/machine"
 	msg, err := me.machine.Marshal()
@@ -50,7 +51,6 @@ func sendMachine() error {
 		log.Info("proto.Marshal() failed:", err)
 		return err
 	}
-	log.Info("proto Marshal len =", len(msg))
 	body, err := httpPost(url, msg)
 	if err != nil {
 		log.Info("httpPost() failed:", err)
@@ -60,15 +60,18 @@ func sendMachine() error {
 	test := strings.TrimSpace(string(body))
 	// log.Info("virtigo returned body:", test)
 	for _, line := range strings.Split(test, "\n") {
-		switch line {
-		case "upgrade":
-			log.Info("zookeeper is healthy")
-		case "apt update":
-			log.Info("machine upgrade now")
+		if line == "upgrade" {
+			log.Info(s, "zookeeper is healthy", len(msg))
+		} else if line == "kill" {
+			os.Exit(0)
+		} else if strings.HasPrefix(line, "apt update") {
+			log.Info("machine upgrade now", line)
 			shell.Run([]string{"apt", "update"})
 			shell.Run([]string{"apt", "install", "zood"})
-		default:
-			log.Info("zookeeper is maybe not working GOT:", line)
+			log.Sleep(1)
+			os.Exit(0)
+		} else {
+			log.Info(urlbase, "is maybe not working GOT:", line)
 		}
 	}
 	return nil