91 lines
1.9 KiB
Go
91 lines
1.9 KiB
Go
package ups
|
|
|
|
import (
|
|
"git.openpdu.org/OpenPDU/openpdu/events"
|
|
"git.openpdu.org/OpenPDU/openpdu/syslog"
|
|
nut "github.com/robbiet480/go.nut"
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
var upsClient nut.Client
|
|
var ups nut.UPS
|
|
var Vars map[string]nut.Variable
|
|
var Connected bool
|
|
|
|
var defaults = map[string]interface{}{
|
|
"Ups.Host": "localhost",
|
|
"Ups.Port": 3493, // not yet implemented in github.com/robbiet480/go.nut, see this PR https://github.com/robbiet480/go.nut/pull/5
|
|
"Ups.Name": "ups",
|
|
"Ups.Username": "",
|
|
"Ups.Password": "",
|
|
}
|
|
|
|
func init() {
|
|
for k, v := range defaults {
|
|
viper.SetDefault(k, v)
|
|
}
|
|
Connected = false
|
|
events.AddListener("config_changed", Reconfigure)
|
|
}
|
|
|
|
func UpsConnect() {
|
|
var username, password string
|
|
var connectErr, authenticationError error
|
|
var u nut.UPS
|
|
|
|
upsClient, connectErr = nut.Connect(viper.GetString("Ups.Host"), viper.GetInt("Ups.Port"))
|
|
if connectErr != nil {
|
|
syslog.Err(connectErr.Error())
|
|
return
|
|
}
|
|
|
|
if username = viper.GetString("Ups.Username"); username == "" {
|
|
return
|
|
}
|
|
if password = viper.GetString("Ups.Password"); password == "" {
|
|
return
|
|
}
|
|
|
|
_, authenticationError = upsClient.Authenticate(username, password)
|
|
if authenticationError != nil {
|
|
syslog.Err(authenticationError.Error())
|
|
return
|
|
}
|
|
|
|
upsList, listErr := upsClient.GetUPSList()
|
|
if listErr != nil {
|
|
syslog.Err(listErr.Error())
|
|
return
|
|
}
|
|
|
|
for _, u = range upsList {
|
|
if u.Name == viper.GetString("Ups.Name") {
|
|
ups = u
|
|
Connected = true
|
|
syslog.Notice("UPS connected")
|
|
events.FireEvent("ups_connected")
|
|
UpsReadVars()
|
|
}
|
|
}
|
|
}
|
|
|
|
func UpsReadVars() {
|
|
Vars = map[string]nut.Variable{}
|
|
for _, v := range ups.Variables {
|
|
Vars[v.Name] = v
|
|
}
|
|
}
|
|
|
|
func Reconfigure() {
|
|
ups = nut.UPS{}
|
|
upsClient = nut.Client{}
|
|
syslog.Notice("UPS disconnected")
|
|
events.FireEvent("ups_disconnected")
|
|
Connected = false
|
|
for key, _ := range Vars {
|
|
delete(Vars, key)
|
|
}
|
|
// Vars = make(map[string]nut.Variable)
|
|
go UpsConnect()
|
|
}
|