Add sites support. This changes the interface.

core/unifi/unidev.go

@@ -13,14 +13,16 @@ import (
 )
 
 const (
+	// SiteList is the path to the api site list.
+	SiteList string = "/api/self/sites"
 	// ClientPath is Unifi Clients API Path
-	ClientPath string = "/api/s/default/stat/sta"
+	ClientPath string = "/api/s/%s/stat/sta"
 	// DevicePath is where we get data about Unifi devices.
-	DevicePath string = "/api/s/default/stat/device"
+	DevicePath string = "/api/s/%s/stat/device"
 	// NetworkPath contains network-configuration data. Not really graphable.
-	NetworkPath string = "/api/s/default/rest/networkconf"
+	NetworkPath string = "/api/s/%s/rest/networkconf"
 	// UserGroupPath contains usergroup configurations.
-	UserGroupPath string = "/api/s/default/rest/usergroup"
+	UserGroupPath string = "/api/s/%s/rest/usergroup"
 	// LoginPath is Unifi Controller Login API Path
 	LoginPath string = "/api/login"
 )

core/unifi/unifi.go

@@ -2,57 +2,83 @@ package unifi
 
 import (
 	"encoding/json"
+	"fmt"
 	"io/ioutil"
+	"strings"
 
 	"github.com/pkg/errors"
 )
 
 // GetClients returns a response full of clients' data from the Unifi Controller.
-func (u *Unifi) GetClients() (*Clients, error) {
+func (u *Unifi) GetClients(sites []string) (*Clients, error) {
+	var data []UCL
+	for _, site := range sites {
 		var response struct {
-		Clients []UCL `json:"data"`
+			Data []UCL `json:"data"`
 		}
-	req, err := u.UniReq(ClientPath, "")
+		clientPath := fmt.Sprintf(ClientPath, site)
-	if err != nil {
+		if err := u.GetData(clientPath, &response); err != nil {
-		return nil, errors.Wrap(err, "c.UniReq(ClientPath)")
+			return nil, err
 		}
-	resp, err := u.Do(req)
+		data = append(data, response.Data...)
-	if err != nil {
-		return nil, errors.Wrap(err, "c.Do(req)")
 	}
-	defer func() {
+	return &Clients{UCLs: data}, nil
-		_ = resp.Body.Close()
-	}()
-	if body, err := ioutil.ReadAll(resp.Body); err != nil {
-		return nil, errors.Wrap(err, "ioutil.ReadAll(resp.Body)")
-	} else if err = json.Unmarshal(body, &response); err != nil {
-		return nil, errors.Wrap(err, "json.Unmarshal([]UCL)")
-	}
-	return &Clients{UCLs: response.Clients}, nil
 }
 
 // GetDevices returns a response full of devices' data from the Unifi Controller.
-func (u *Unifi) GetDevices() (*Devices, error) {
+func (u *Unifi) GetDevices(sites []string) (*Devices, error) {
+	var data []json.RawMessage
+	for _, site := range sites {
 		var response struct {
 			Data []json.RawMessage `json:"data"`
 		}
-	req, err := u.UniReq(DevicePath, "")
+		devicePath := fmt.Sprintf(DevicePath, site)
+		if err := u.GetData(devicePath, &response); err != nil {
+			return nil, err
+		}
+		data = append(data, response.Data...)
+	}
+	return u.parseDevices(data), nil
+}
+
+// GetSites returns a list of configured sites on the Unifi controller.
+func (u *Unifi) GetSites() ([]string, error) {
+	var response struct {
+		Data []struct {
+			// This is the only field we need. There are others.
+			Name string `json:"name"`
+		} `json:"data"`
+	}
+	if err := u.GetData(SiteList, &response); err != nil {
+		return nil, err
+	}
+	var output []string
+	for i := range response.Data {
+		output = append(output, response.Data[i].Name)
+	}
+	u.dLogf("Found %d sites: %v", len(output), strings.Join(output, ","))
+	return output, nil
+}
+
+// GetData makes a unifi request and unmarshal the response into a provided pointer.
+func (u *Unifi) GetData(methodPath string, v interface{}) error {
+	req, err := u.UniReq(methodPath, "")
 	if err != nil {
-		return nil, errors.Wrap(err, "c.UniReq(DevicePath)")
+		return errors.Wrapf(err, "c.UniReq(%v)", methodPath)
 	}
 	resp, err := u.Do(req)
 	if err != nil {
-		return nil, errors.Wrap(err, "c.Do(req)")
+		return errors.Wrapf(err, "c.Do(%v)", methodPath)
 	}
 	defer func() {
 		_ = resp.Body.Close()
 	}()
 	if body, err := ioutil.ReadAll(resp.Body); err != nil {
-		return nil, errors.Wrap(err, "ioutil.ReadAll(resp.Body)")
+		return errors.Wrapf(err, "ioutil.ReadAll(%v)", methodPath)
-	} else if err = json.Unmarshal(body, &response); err != nil {
+	} else if err = json.Unmarshal(body, v); err != nil {
-		return nil, errors.Wrap(err, "json.Unmarshal([]json.RawMessage)")
+		return errors.Wrapf(err, "json.Unmarshal(%v)", methodPath)
 	}
-	return u.parseDevices(response.Data), nil
+	return nil
 }
 
 // parseDevices parses the raw JSON from the Unifi Controller into device structures.