completes prometheus api functionality

cmd/tpapi/httpTemplates.go

@@ -10,8 +10,6 @@ import (
 
 	"encoding/json"
 	"net/http"
-
-	"tplink/internal/tplink"
 )
 
 func tmpltError(w http.ResponseWriter, s int, m string) {
@@ -149,37 +147,13 @@ func tmpltTPlinkGet(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 
-	t := tplink.Tplink{
+	fs, err := switchDevice(host, id, state)
-		Host:     host,
-		SwitchID: id,
-	}
-	if err := t.ChangeStateMultiSwitch(state); err != nil {
-		log.Printf("[INFO] Unable to change device state: %v\n", err)
-		tmpltError(w, http.StatusInternalServerError, fmt.Sprintf("Unable to change device state: %v\n", err))
-		return
-	}
-
-	sysInfo, err := t.SystemInfo()
 	if err != nil {
-		log.Printf("[INFO] Unable to get info from target device: %v\n", err)
+		log.Printf("[INFO] %v\n", err)
-		tmpltError(w, http.StatusInternalServerError, fmt.Sprintf("Unable to read info from device: %v\n", err))
+		tmpltError(w, http.StatusBadRequest, fmt.Sprintf("[INFO] %v\n", err))
 		return
 	}
 
-	var ds int
-	if len(sysInfo.System.GetSysinfo.Children) == 0 {
-		ds = sysInfo.System.GetSysinfo.RelayState
-	} else {
-		ds = sysInfo.System.GetSysinfo.Children[id].State
-	}
-
-	var fs string
-	if ds == 0 {
-		fs = "OFF"
-	} else {
-		fs = "ON"
-	}
-
 	o := struct {
 		Status   string `json:"status" yaml:"status"`
 		Host     string `json:"host" yaml:"host"`
@@ -199,7 +173,16 @@ func tmpltTPlinkGet(w http.ResponseWriter, r *http.Request) {
 }
 
 func tmpltTPlinkPost(w http.ResponseWriter, r *http.Request) {
-	var payload alertmanagerStruct
+	type outputStruct struct {
+		Status   string `json:"status" yaml:"status"`
+		Host     string `json:"host" yaml:"host"`
+		DeviceID int    `json:"device_id" yaml:"deviceID"`
+	}
+
+	var (
+		payload alertmanagerStruct
+		o       []outputStruct
+	)
 
 	// read body
 	defer r.Body.Close()
@@ -219,5 +202,60 @@ func tmpltTPlinkPost(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 
-	log.Printf("[TRACE] Data received: %#v\n", payload)
+	for _, alert := range payload.Alerts {
+		var (
+			host     string
+			id       int
+			validKey bool
+			fs       string
+		)
+
+		// validate keys exist
+		if host, validKey = alert.Annotations["host"]; !validKey {
+			log.Printf("[DEBUG] Unable to find host key annotation.")
+			tmpltError(w, http.StatusInternalServerError, "Unable to find host key annotation.")
+			return
+		}
+		if _, validKey = alert.Annotations["id"]; !validKey {
+			log.Printf("[DEBUG] Unable to find device id key annotation.")
+			tmpltError(w, http.StatusInternalServerError, "Unable to find device id key annotation.")
+			return
+		}
+
+		// convert annotation string to integer
+		id, err := strconv.Atoi(alert.Annotations["id"])
+		if err != nil {
+			log.Printf("[DEBUG] Unable to convert string to integer for provided device id: %v\n", err)
+			tmpltError(w, http.StatusInternalServerError, fmt.Sprintf("Unable to convert string to integer for provided device id: %v\n", err))
+			return
+		}
+
+		switch strings.ToLower(alert.Status) {
+		case "firing":
+			fs, err = switchDevice(host, id, true)
+		case "resolved":
+			fs, err = switchDevice(host, id, false)
+		default:
+		}
+
+		if err != nil {
+			log.Printf("[INFO] %v\n", err)
+			tmpltError(w, http.StatusBadRequest, fmt.Sprintf("[INFO] %v\n", err))
+			return
+		}
+
+		o = append(o, outputStruct{
+			Status:   fs,
+			Host:     host,
+			DeviceID: id,
+		})
+	}
+
+	w.Header().Add("Content-Type", "application/json")
+
+	output, err := json.MarshalIndent(o, "", "  ")
+	if err != nil {
+		log.Printf("[TRACE] Unable to marshal error message: %v.", err)
+	}
+	w.Write(output) //nolint:errcheck
 }

cmd/tpapi/supportingFunctions.go

@@ -0,0 +1,45 @@
+package main
+
+import (
+	"fmt"
+	"time"
+	"tplink/internal/tplink"
+)
+
+func switchDevice(host string, id int, state bool) (string, error) {
+	t := tplink.Tplink{
+		Host:     host,
+		SwitchID: id,
+	}
+
+	// send request to device
+	if err := t.ChangeStateMultiSwitch(state); err != nil {
+		return "", fmt.Errorf("Unable to change device state: %v\n", err)
+	}
+
+	// give device time to execute command
+	time.Sleep(100 * time.Millisecond)
+
+	// get device state
+	time.Sleep(100 * time.Millisecond)
+	sysInfo, err := t.SystemInfo()
+	if err != nil {
+		return "", fmt.Errorf("Unable to get info from target device: %v\n", err)
+	}
+
+	var ds int
+	if len(sysInfo.System.GetSysinfo.Children) == 0 {
+		ds = sysInfo.System.GetSysinfo.RelayState
+	} else {
+		ds = sysInfo.System.GetSysinfo.Children[id].State
+	}
+
+	var fs string
+	if ds == 0 {
+		fs = "OFF"
+	} else {
+		fs = "ON"
+	}
+
+	return fs, nil
+}