Add network control feature

Author: facchinm

handlers_stats.go

@@ -24,9 +24,26 @@ import (
 
 	"github.com/arduino/go-system-stats/disk"
 	"github.com/arduino/go-system-stats/mem"
+	"github.com/arduino/go-system-stats/network"
 	mqtt "github.com/eclipse/paho.mqtt.golang"
+	"github.com/pkg/errors"
 )
 
+// WiFiEvent tries to connect to the specified wifi network
+func (s *Status) WiFiEvent(client mqtt.Client, msg mqtt.Message) {
+	// try registering a new wifi network
+	var info struct {
+		SSID     string `json:"ssid"`
+		Password string `json:"password"`
+	}
+	err := json.Unmarshal(msg.Payload(), &info)
+	if err != nil {
+		s.Error("/wifi", errors.Wrapf(err, "unmarshal %s", msg.Payload()))
+		return
+	}
+	net.AddWirelessConnection(info.SSID, info.Password)
+}
+
 // StatsEvent sends statistics about resource used in the system (RAM, Disk, Network, etc...)
 func (s *Status) StatsEvent(client mqtt.Client, msg mqtt.Message) {
 	// Gather all system data metrics
@@ -42,14 +59,22 @@ func (s *Status) StatsEvent(client mqtt.Client, msg mqtt.Message) {
 		return
 	}
 
+	netStats, err := net.GetNetworkStats()
+	if err != nil {
+		s.Error("/stats/error", fmt.Errorf("Retrieving network stats: %s", err))
+		return
+	}
+
 	type StatsPayload struct {
-		Memory *mem.Stats      `json:"memory"`
-		Disk   []*disk.FSStats `json:"disk"`
+		Memory  *mem.Stats      `json:"memory"`
+		Disk    []*disk.FSStats `json:"disk"`
+		Network *net.Stats      `json:"network"`
 	}
 
 	info := StatsPayload{
-		Memory: memStats,
-		Disk:   diskStats,
+		Memory:  memStats,
+		Disk:    diskStats,
+		Network: netStats,
 	}
 
 	// Send result

main.go

@@ -212,6 +212,7 @@ func subscribeTopics(mqttClient mqtt.Client, id string, status *Status) {
 	mqttClient.Subscribe("$aws/things/"+id+"/sketch/post", 1, status.SketchEvent)
 	mqttClient.Subscribe("$aws/things/"+id+"/update/post", 1, status.UpdateEvent)
 	mqttClient.Subscribe("$aws/things/"+id+"/stats/post", 1, status.StatsEvent)
+	mqttClient.Subscribe("$aws/things/"+id+"/wifi/post", 1, status.WiFiEvent)
 
 	mqttClient.Subscribe("$aws/things/"+id+"/apt/list/post", 1, status.AptListEvent)
 	mqttClient.Subscribe("$aws/things/"+id+"/apt/install/post", 1, status.AptInstallEvent)