nhyatt/tplinkcmd
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

completes prometheus api functionality

Browse Source
This commit is contained in:
Hyatt 2022-01-10 11:27:35 -06:00
parent 6414587580
commit d307a6a2a5
Signed by: nhyatt
GPG Key ID: C50D0BBB5BC40BEA
2 changed files with 114 additions and 31 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

100
cmd/tpapi/httpTemplates.go
View File

@ -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{
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()
fs, err := switchDevice(host, id, state)
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))
log.Printf("[INFO] %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
}

45
cmd/tpapi/supportingFunctions.go Normal file
View File

@ -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
}