136 lines
4.1 KiB
Go
136 lines
4.1 KiB
Go
package poller
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/davidnewhall/unifi-poller/pkg/metrics"
|
|
"golift.io/unifi"
|
|
)
|
|
|
|
// GetUnifi returns a UniFi controller interface.
|
|
func (u *UnifiPoller) GetUnifi() (err error) {
|
|
// Create an authenticated session to the Unifi Controller.
|
|
u.Unifi, err = unifi.NewUnifi(&unifi.Config{
|
|
User: u.Config.UnifiUser,
|
|
Pass: u.Config.UnifiPass,
|
|
URL: u.Config.UnifiBase,
|
|
VerifySSL: u.Config.VerifySSL,
|
|
ErrorLog: u.LogErrorf, // Log all errors.
|
|
DebugLog: u.LogDebugf, // Log debug messages.
|
|
})
|
|
if err != nil {
|
|
return fmt.Errorf("unifi controller: %v", err)
|
|
}
|
|
u.LogDebugf("Authenticated with controller successfully")
|
|
|
|
return u.CheckSites()
|
|
}
|
|
|
|
// CheckSites makes sure the list of provided sites exists on the controller.
|
|
// This does not run in Lambda (run-once) mode.
|
|
func (u *UnifiPoller) CheckSites() error {
|
|
if strings.Contains(strings.ToLower(u.Config.Mode), "lambda") {
|
|
return nil // Skip this in lambda mode.
|
|
}
|
|
u.LogDebugf("Checking Controller Sites List")
|
|
sites, err := u.Unifi.GetSites()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
msg := []string{}
|
|
for _, site := range sites {
|
|
msg = append(msg, site.Name+" ("+site.Desc+")")
|
|
}
|
|
u.Logf("Found %d site(s) on controller: %v", len(msg), strings.Join(msg, ", "))
|
|
if StringInSlice("all", u.Config.Sites) {
|
|
u.Config.Sites = []string{"all"}
|
|
return nil
|
|
}
|
|
FIRST:
|
|
for _, s := range u.Config.Sites {
|
|
for _, site := range sites {
|
|
if s == site.Name {
|
|
continue FIRST
|
|
}
|
|
}
|
|
// This is fine, it may get added later.
|
|
u.LogErrorf("configured site not found on controller: %v", s)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// CollectMetrics grabs all the measurements from a UniFi controller and returns them.
|
|
func (u *UnifiPoller) CollectMetrics() (*metrics.Metrics, error) {
|
|
m := &metrics.Metrics{TS: u.LastCheck} // At this point, it's the Current Check.
|
|
var err error
|
|
// Get the sites we care about.
|
|
m.Sites, err = u.GetFilteredSites()
|
|
u.LogError(err, "unifi.GetSites()")
|
|
if u.Config.CollectIDS {
|
|
m.IDSList, err = u.Unifi.GetIDS(m.Sites, time.Now().Add(u.Config.Interval.Duration), time.Now())
|
|
u.LogError(err, "unifi.GetIDS()")
|
|
}
|
|
// Get all the points.
|
|
m.Clients, err = u.Unifi.GetClients(m.Sites)
|
|
u.LogError(err, "unifi.GetClients()")
|
|
m.Devices, err = u.Unifi.GetDevices(m.Sites)
|
|
u.LogError(err, "unifi.GetDevices()")
|
|
return m, err
|
|
}
|
|
|
|
// AugmentMetrics is our middleware layer between collecting metrics and writing them.
|
|
// This is where we can manipuate the returned data or make arbitrary decisions.
|
|
// This function currently adds parent device names to client metrics.
|
|
func (u *UnifiPoller) AugmentMetrics(metrics *metrics.Metrics) {
|
|
if metrics == nil || metrics.Devices == nil || metrics.Clients == nil {
|
|
return
|
|
}
|
|
devices := make(map[string]string)
|
|
bssdIDs := make(map[string]string)
|
|
for _, r := range metrics.UAPs {
|
|
devices[r.Mac] = r.Name
|
|
for _, v := range r.VapTable {
|
|
bssdIDs[v.Bssid] = fmt.Sprintf("%s %s %s:", r.Name, v.Radio, v.RadioName)
|
|
}
|
|
}
|
|
for _, r := range metrics.USGs {
|
|
devices[r.Mac] = r.Name
|
|
}
|
|
for _, r := range metrics.USWs {
|
|
devices[r.Mac] = r.Name
|
|
}
|
|
for _, r := range metrics.UDMs {
|
|
devices[r.Mac] = r.Name
|
|
}
|
|
// These come blank, so set them here.
|
|
for i, c := range metrics.Clients {
|
|
metrics.Clients[i].SwName = devices[c.SwMac]
|
|
metrics.Clients[i].ApName = devices[c.ApMac]
|
|
metrics.Clients[i].GwName = devices[c.GwMac]
|
|
metrics.Clients[i].RadioDescription = bssdIDs[metrics.Clients[i].Bssid] + metrics.Clients[i].RadioProto
|
|
}
|
|
}
|
|
|
|
// GetFilteredSites returns a list of sites to fetch data for.
|
|
// Omits requested but unconfigured sites. Grabs the full list from the
|
|
// controller and returns the sites provided in the config file.
|
|
func (u *UnifiPoller) GetFilteredSites() (unifi.Sites, error) {
|
|
sites, err := u.Unifi.GetSites()
|
|
if err != nil {
|
|
return nil, err
|
|
} else if len(u.Config.Sites) < 1 || StringInSlice("all", u.Config.Sites) {
|
|
return sites, nil
|
|
}
|
|
var i int
|
|
for _, s := range sites {
|
|
// Only include valid sites in the request filter.
|
|
if StringInSlice(s.Name, u.Config.Sites) {
|
|
sites[i] = s
|
|
i++
|
|
}
|
|
}
|
|
return sites[:i], nil
|
|
}
|