beszel-ipv6/agent/wifi/wifi.go

// Package wifi collects only currently associated station interfaces. Collection
// failures are empty snapshots, never cached connected state.
package wifi

import (
	"context"
	"log/slog"
	"math"
	"sync"
	"time"
	"unicode/utf8"

	"github.com/henrygd/beszel/agent/utils"
	"github.com/henrygd/beszel/internal/entities/system"
)

// validSSID omits non-UTF-8 SSIDs: 802.11 permits arbitrary octets, but CBOR
// text strings require UTF-8. Metadata must never invalidate the whole response.
func validSSID(ssid string) string {
	if !utf8.ValidString(ssid) {
		return ""
	}
	return ssid
}

// skipFromEnv reports whether SKIP_WIFI disables collection.
func skipFromEnv() bool {
	skip, _ := utils.GetEnv("SKIP_WIFI")
	return skip == "true"
}

// skip caches skipFromEnv, as the environment is fixed for the life of the
// process.
var skip = sync.OnceValue(func() bool {
	if !skipFromEnv() {
		return false
	}
	slog.Info("SKIP_WIFI enabled, skipping Wi-Fi signal monitoring")
	return true
})

// Collect uses a single deadline across interface queries where supported.
// Unsupported platforms and denied association access produce no readings;
// later polls retry. SKIP_WIFI=true disables collection entirely.
func Collect() map[string]system.WiFi {
	if skip() {
		return nil
	}
	ctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)
	defer cancel()
	return collect(ctx)
}

// Signals reduces a snapshot to the RSSI values stored in stats history.
// Interfaces without an available reading are omitted.
func Signals(snapshot map[string]system.WiFi) map[string]int8 {
	var signals map[string]int8
	for id, reading := range snapshot {
		if reading.Signal == nil {
			continue
		}
		if signals == nil {
			signals = make(map[string]int8, len(snapshot))
		}
		signals[id] = int8(max(math.Round(*reading.Signal), math.MinInt8))
	}
	return signals
}