starting prometheus alerter

cmd/tpapi/config.go

@@ -35,6 +35,29 @@ type configPrometheus struct {
 	NumGroupDeleted prometheus.Gauge
 }
 
+type alertmanagerStruct struct {
+	Version            string                     `json:"version"`
+	GroupKey           string                     `json:"groupKey"`
+	TruncatedAlerts    int                        `json:"truncatedAlerts"`
+	Status             string                     `json:"status"`
+	Reciver            string                     `json:"receiver"`
+	GroupLabels        string                     `json:"groupLabels"`
+	ComminLabels       string                     `json:"commonLabels"`
+	CommonAnnotaations string                     `json:"commonAnnotations"`
+	ExternalURL        string                     `json:"externalURL"`
+	Alerts             []alertmanagerAlertsStruct `json:"alerts"`
+}
+
+type alertmanagerAlertsStruct struct {
+	Status       string `json:"status"`
+	Labels       string `json:"labels"`
+	Annotations  string `json:"annotations"`
+	StartsAt     string `json:"startsAt"`
+	EndsAt       string `json:"endsAt"`
+	GeneratorURL string `json:"generatorURL"`
+	Fingerprint  string `json:"fingerprint"`
+}
+
 var config = configStructure{
 	TimeFormat: "2006-01-02 15:04:05",
 	Log: &logutils.LevelFilter{

cmd/tpapi/httpServer.go

@@ -1,18 +1,13 @@
 package main
 
 import (
-	"encoding/json"
-	"fmt"
 	"log"
-	"net"
 	"strconv"
 	"strings"
 	"time"
 
 	"net/http"
 
-	"tplink/internal/tplink"
-
 	"github.com/prometheus/client_golang/prometheus/promhttp"
 )
 
@@ -83,133 +78,13 @@ func webTPLink(w http.ResponseWriter, r *http.Request) {
 	crossSiteOrigin(w)
 
 	if strings.ToLower(r.Method) == "get" {
-		keys := r.URL.Query()
+		tmpltTPlinkGet(w, r)
-		if len(keys) == 0 {
+	} else if strings.ToLower(r.Method) == "post" {
-			log.Printf("[INFO] Required parameters missing: no host, state, or deviceid was provided.\n")
+		tmpltTPlinkPost(w, r)
-			tmpltError(w, http.StatusBadRequest, "Required parameters missing: no parameter for host, state, or deviceid was provided.")
-			return
-		}
-
-		var (
-			host     string
-			hostSet  bool
-			state    bool
-			stateSet bool
-			id       int
-		)
-
-		for k, v := range keys {
-			log.Printf("%s: %v\n", k, v)
-			switch strings.ToLower(k) {
-			case "host":
-				if len(v) != 1 {
-					log.Printf("[INFO] Host was defined multiple times.\n")
-					tmpltError(w, http.StatusBadRequest, "Required parameter host was defined multiple times.")
-					return
-				}
-				_, err := net.LookupHost(v[0])
-				if err != nil {
-					log.Printf("[INFO] Unable to resolve provided hostname: %s\n", v[0])
-					tmpltError(w, http.StatusBadRequest, "Unable to resolve provided hostname.")
-					return
-				}
-				hostSet = true
-				host = v[0]
-
-			case "state":
-				if len(v) != 1 {
-					log.Printf("[INFO] State was defined multiple times.\n")
-					tmpltError(w, http.StatusBadRequest, "Required parameter state was defined multiple times.")
-					return
-				}
-				s, err := strconv.ParseBool(v[0])
-				if err != nil {
-					log.Printf("[INFO] State is not boolean: %s\n", v[0])
-					tmpltError(w, http.StatusBadRequest, "Required parameter state is not boolean.")
-					return
-				}
-				stateSet = true
-				state = s
-
-			case "deviceid":
-				if len(v) != 1 {
-					log.Printf("[INFO] Deviceid was defined multiple times.\n")
-					tmpltError(w, http.StatusBadRequest, "Required parameter device was defined multiple times.")
-					return
-				}
-				d, err := strconv.Atoi(v[0])
-				if err != nil {
-					log.Printf("[INFO] Deviceid is not an integer: %s\n", v[0])
-					tmpltError(w, http.StatusBadRequest, "Required parameter deviceid is not an integer.")
-					return
-				}
-				id = d
-			}
-		}
-
-		if !hostSet {
-			log.Printf("[INFO] Required parameter missing: no host was provided.\n")
-			tmpltError(w, http.StatusBadRequest, "Required parameters missing: no parameter for host provided.")
-			return
-		}
-		if !stateSet {
-			log.Printf("[INFO] Required parameter missing: no state was provided.\n")
-			tmpltError(w, http.StatusBadRequest, "Required parameters missing: no parameter for state provided.")
-			return
-		}
-
-		t := tplink.Tplink{
-			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)
-			tmpltError(w, http.StatusInternalServerError, fmt.Sprintf("Unable to read info from device: %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"`
-			DeviceID int    `json:"device_id" yaml:"deviceID"`
-		}{
-			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
-
 	} else if strings.ToLower(r.Method) == "options" {
 		return
 	} else {
-		log.Printf("[DEBUG] Request to '%s' was made using the wrong method: expected %s, got %s", "GET|OPTIONS", r.URL.Path, strings.ToUpper(r.Method))
+		log.Printf("[DEBUG] Request to '%s' was made using the wrong method: expected %s, got %s", "GET|POST|OPTIONS", r.URL.Path, strings.ToUpper(r.Method))
 		tmpltError(w, http.StatusBadRequest, "Invalid http method.")
 	}
 }

cmd/tpapi/httpTemplates.go

@@ -2,8 +2,13 @@ package main
 
 import (
 	"encoding/json"
+	"fmt"
 	"log"
+	"net"
 	"net/http"
+	"strconv"
+	"strings"
+	"tplink/internal/tplink"
 )
 
 func tmpltError(w http.ResponseWriter, s int, m string) {
@@ -65,3 +70,139 @@ func tmpltHealthCheck(w http.ResponseWriter) {
 	w.Header().Add("Content-Type", "application/json")
 	w.Write(output) //nolint:errcheck
 }
+
+func tmpltTPlinkGet(w http.ResponseWriter, r *http.Request) {
+	keys := r.URL.Query()
+	if len(keys) == 0 {
+		log.Printf("[INFO] Required parameters missing: no host, state, or deviceid was provided.\n")
+		tmpltError(w, http.StatusBadRequest, "Required parameters missing: no parameter for host, state, or deviceid was provided.")
+		return
+	}
+
+	var (
+		host     string
+		hostSet  bool
+		state    bool
+		stateSet bool
+		id       int
+	)
+
+	for k, v := range keys {
+		switch strings.ToLower(k) {
+		case "host":
+			if len(v) != 1 {
+				log.Printf("[INFO] Host was defined multiple times.\n")
+				tmpltError(w, http.StatusBadRequest, "Required parameter host was defined multiple times.")
+				return
+			}
+			_, err := net.LookupHost(v[0])
+			if err != nil {
+				log.Printf("[INFO] Unable to resolve provided hostname: %s\n", v[0])
+				tmpltError(w, http.StatusBadRequest, "Unable to resolve provided hostname.")
+				return
+			}
+			hostSet = true
+			host = v[0]
+
+		case "state":
+			if len(v) != 1 {
+				log.Printf("[INFO] State was defined multiple times.\n")
+				tmpltError(w, http.StatusBadRequest, "Required parameter state was defined multiple times.")
+				return
+			}
+			s, err := strconv.ParseBool(v[0])
+			if err != nil {
+				log.Printf("[INFO] State is not boolean: %s\n", v[0])
+				tmpltError(w, http.StatusBadRequest, "Required parameter state is not boolean.")
+				return
+			}
+			stateSet = true
+			state = s
+
+		case "deviceid":
+			if len(v) != 1 {
+				log.Printf("[INFO] Deviceid was defined multiple times.\n")
+				tmpltError(w, http.StatusBadRequest, "Required parameter device was defined multiple times.")
+				return
+			}
+			d, err := strconv.Atoi(v[0])
+			if err != nil {
+				log.Printf("[INFO] Deviceid is not an integer: %s\n", v[0])
+				tmpltError(w, http.StatusBadRequest, "Required parameter deviceid is not an integer.")
+				return
+			}
+			id = d
+		}
+	}
+
+	if !hostSet {
+		log.Printf("[INFO] Required parameter missing: no host was provided.\n")
+		tmpltError(w, http.StatusBadRequest, "Required parameters missing: no parameter for host provided.")
+		return
+	}
+	if !stateSet {
+		log.Printf("[INFO] Required parameter missing: no state was provided.\n")
+		tmpltError(w, http.StatusBadRequest, "Required parameters missing: no parameter for state provided.")
+		return
+	}
+
+	t := tplink.Tplink{
+		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)
+		tmpltError(w, http.StatusInternalServerError, fmt.Sprintf("Unable to read info from device: %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"`
+		DeviceID int    `json:"device_id" yaml:"deviceID"`
+	}{
+		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
+}
+
+func tmpltTPlinkPost(w http.ResponseWriter, r *http.Request) {
+	var payload alertmanagerStruct
+
+	err := json.NewDecoder(r.Body).Decode(&payload)
+	if err != nil {
+		log.Printf("[INFO] Unable to parse payload from API call: %v\n", err)
+		log.Printf("[INFO] Data received: %#v\n", payload)
+		tmpltError(w, http.StatusInternalServerError, fmt.Sprintf("Unable to read info from device: %v\n", err))
+		return
+	}
+}