add networks

core/unifi/.gitignore

@@ -1 +1,2 @@
 /vendor
+*.swp

core/unifi/networks.go

@@ -0,0 +1,74 @@
+package unifi
+
+import (
+	"encoding/json"
+	"fmt"
+)
+
+// GetNetworks returns a response full of network data from the UniFi Controller.
+func (u *Unifi) GetNetworks(sites []*Site) ([]Network, error) {
+	networks := make([]Network, 0)
+
+	for _, site := range sites {
+		var response struct {
+			Data []json.RawMessage `json:"data"`
+		}
+
+		networkPath := fmt.Sprintf(APINetworkPath, site.Name)
+		if err := u.GetData(networkPath, &response); err != nil {
+			return nil, err
+		}
+
+		for _, data := range response.Data {
+			network := u.parseNetwork(data, site.SiteName)
+			networks = append(networks, *network)
+		}
+	}
+
+	return networks, nil
+}
+
+// parseNetwork parses the raw JSON from the Unifi Controller into network structures.
+func (u *Unifi) parseNetwork(data json.RawMessage, siteName string) *Network {
+	network := new(Network)
+	u.unmarshalNetwork(data, &network)
+	return network
+}
+
+// unmarshalNetwork handles logging for the unmarshal operations in parseNetwork().
+func (u *Unifi) unmarshalNetwork(data json.RawMessage, v interface{}) (err error) {
+	if err = json.Unmarshal(data, v); err != nil {
+		u.ErrorLog("json.Unmarshal(): %v", err)
+		u.ErrorLog("Enable Debug Logging to output the failed payload.")
+
+		json, err := data.MarshalJSON()
+		u.DebugLog("Failed Payload: %s (marshal err: %v)", json, err)
+		u.DebugLog("The above payload can prove useful during torubleshooting when you open an Issue:")
+		u.DebugLog("==- https://github.com/unifi-poller/unifi/issues/new -==")
+	}
+
+	return err
+}
+
+// Network is metadata about a network managed by a UniFi controller
+type Network struct {
+	DhcpdDNSEnabled        FlexBool `json:"dhcpd_dns_enabled"`
+	DhcpdEnabled           FlexBool `json:"dhcpd_enabled"`
+	DhcpdGatewayEnabled    FlexBool `json:"dhcpd_gateway_enabled"`
+	DhcpdIP1               string   `json:"dhcpd_ip_1"`
+	DhcpdLeasetime         FlexInt  `json:"dhcpd_leasetime"`
+	DhcpRelayEnabled       FlexBool `json:"dhcp_relay_enabled"`
+	DhcpdTimeOffsetEnabled FlexBool `json:"dhcpd_time_offset_enabled"`
+	DhcpGuardEnabled       FlexBool `json:"dhcpguard_enabled"`
+	DomainName             string   `json:"domain_name"`
+	Enabled                FlexBool `json:"enabled"`
+	ID                     string   `json:"_id"`
+	IPSubnet               string   `json:"ip_subnet"`
+	IsNat                  FlexBool `json:"is_nat"`
+	Name                   string   `json:"name"`
+	Networkgroup           string   `json:"networkgroup"`
+	Purpose                string   `json:"purpose"`
+	SiteID                 string   `json:"site_id"`
+	Vlan                   FlexInt  `json:"vlan"`
+	VlanEnabled            FlexBool `json:"vlan_enabled"`
+}

core/unifi/types.go

@@ -29,6 +29,8 @@ const (
 	APIClientDPI string = "/api/s/%s/stat/stadpi"
 	// APIClientPath is Unifi Clients API Path
 	APIClientPath string = "/api/s/%s/stat/sta"
+	// APINetworkPath is where we get data about Unifi networks.
+	APINetworkPath string = "/api/s/%s/rest/networkconf"
 	// APIDevicePath is where we get data about Unifi devices.
 	APIDevicePath string = "/api/s/%s/stat/device"
 	// APILoginPath is Unifi Controller Login API Path