refact: using global wifi objects in order to expose them from the api.rest module

network/wifi.go

@@ -0,0 +1,61 @@
+package network
+
+import (
+	"sync"
+	"time"
+)
+
+type WiFi struct {
+	sync.Mutex
+	Interface *Endpoint
+	Stations  map[string]*WiFiStation
+}
+
+func NewWiFi(iface *Endpoint) *WiFi {
+	return &WiFi{
+		Interface: iface,
+		Stations:  make(map[string]*WiFiStation),
+	}
+}
+
+func (w *WiFi) List() (list []*WiFiStation) {
+	w.Lock()
+	defer w.Unlock()
+
+	list = make([]*WiFiStation, 0)
+	for _, t := range w.Stations {
+		list = append(list, t)
+	}
+	return
+}
+
+func (w *WiFi) Remove(mac string) {
+	w.Lock()
+	defer w.Unlock()
+
+	if _, found := w.Stations[mac]; found {
+		delete(w.Stations, mac)
+	}
+}
+
+func (w *WiFi) AddIfNew(ssid, mac string, isAp bool, channel int, rssi int8) *WiFiStation {
+	w.Lock()
+	defer w.Unlock()
+
+	mac = NormalizeMac(mac)
+	if station, found := w.Stations[mac]; found {
+		station.LastSeen = time.Now()
+		station.RSSI = rssi
+		return station
+	}
+
+	newStation := NewWiFiStation(ssid, mac, isAp, channel, rssi)
+	w.Stations[mac] = newStation
+
+	return nil
+}
+
+func (w *WiFi) Clear() error {
+	w.Stations = make(map[string]*WiFiStation)
+	return nil
+}