add log flags

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

1 files changed, 23 insertions, 2 deletions

diff --git a/http.go b/http.go
index 7c19a3a..0460896 100644
--- a/http.go
+++ b/http.go
@@ -17,10 +17,11 @@ func cleanURL(url string) string {
 }
 
 func okHandler(w http.ResponseWriter, r *http.Request) {
-	log.Info("Got URL Path: ", r.URL.Path)
+	// log.Info("Got URL Path: ", r.URL.Path)
 	route := cleanURL(r.URL.Path)
 
 	hostname := r.URL.Query().Get("hostname")
+	flag := r.URL.Query().Get("flag")
 
 	msg, err := ioutil.ReadAll(r.Body) // Read the body as []byte
 	if err != nil {
@@ -40,7 +41,7 @@ func okHandler(w http.ResponseWriter, r *http.Request) {
 			log.Info("error =", err)
 			return
 		}
-		log.Info("proto.Unmarshal() worked on wire message len", len(msg), "from", m.Hostname)
+		log.Log(ZOOD, "proto.Unmarshal() worked on wire message len", len(msg), "from", m.Hostname)
 		switch updateMachine(m) {
 		case "upgrade":
 			fmt.Fprintln(w, "upgrade")
@@ -69,6 +70,26 @@ func okHandler(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 
+	// 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.Bool() {
+				log.Log(NOW, "toogle ZOOD false")
+				ZOOD.SetBool(false)
+			} else {
+				log.Log(NOW, "toogle ZOOD true")
+				ZOOD.SetBool(true)
+			}
+		default:
+			log.Info("unknown looging flag:", flag)
+		}
+		return
+	}
+
 	log.Warn("BAD URL =", route)
 }