Add power usage support via corsair rmi sensors in linux (#3539)

Author: timmo001

data/module/system.go

@@ -180,14 +180,17 @@ func (sm SystemModule) Update(ctx context.Context) (any, error) {
 	systemData.RunMode = types.RunModeStandalone // Always set RunMode to standalone
 
 	// Populate optional fields when available per-OS
-	// Linux: detect camera usage processes and pending reboot via subpackage helpers
+	// Linux: detect camera usage processes, pending reboot, and PSU power usage via subpackage helpers
 	if infoStat.OS == "linux" || infoStat.Platform == "linux" || strings.Contains(strings.ToLower(infoStat.Platform), "arch") || strings.Contains(strings.ToLower(infoStat.Platform), "ubuntu") || strings.Contains(strings.ToLower(infoStat.Platform), "debian") {
 		if cu := system.GetCameraUsage(); len(cu) > 0 {
 			systemData.CameraUsage = cu
 		}
 		if pr := system.GetPendingReboot(); pr != nil {
 			systemData.PendingReboot = pr
 		}
+		if pu := system.GetPSUPowerUsage(); pu != nil {
+			systemData.PowerUsage = pu
+		}
 	}
 
 	// Get version information

data/module/system/system_linux.go

@@ -6,6 +6,7 @@ package system
 import (
 	"os"
 	"path/filepath"
+	"strconv"
 	"strings"
 
 	"log/slog"
@@ -96,3 +97,67 @@ func GetPendingReboot() *bool {
 	}
 	return nil
 }
+
+// GetPSUPowerUsage attempts to read PSU power usage from hwmon interfaces on Linux.
+// This function specifically looks for Corsair PSU sensors via the corsair_psu driver.
+// Returns power usage in watts, or nil if not available.
+func GetPSUPowerUsage() *float64 {
+	// Look for hwmon directories that might contain PSU power sensors
+	hwmonDirs, err := filepath.Glob("/sys/class/hwmon/hwmon*")
+	if err != nil {
+		slog.Debug("Failed to glob hwmon directories", "error", err)
+		return nil
+	}
+
+	for _, hwmonDir := range hwmonDirs {
+		// Check if this is a Corsair PSU by looking at the name file
+		nameFile := filepath.Join(hwmonDir, "name")
+		nameData, err := os.ReadFile(nameFile)
+		if err != nil {
+			continue
+		}
+		name := strings.TrimSpace(string(nameData))
+
+		// Look for corsair_psu driver or similar PSU-related names
+		if strings.Contains(strings.ToLower(name), "corsair") ||
+			strings.Contains(strings.ToLower(name), "psu") ||
+			strings.Contains(strings.ToLower(name), "rmi") {
+
+			// Try to read power1_input (total power consumption)
+			powerFile := filepath.Join(hwmonDir, "power1_input")
+			if powerData, err := os.ReadFile(powerFile); err == nil {
+				valueStr := strings.TrimSpace(string(powerData))
+				if value, err := strconv.ParseFloat(valueStr, 64); err == nil {
+					// Convert from microwatts to watts
+					powerWatts := value / 1_000_000
+					slog.Info("Found PSU power usage", "hwmon", hwmonDir, "name", name, "power_watts", powerWatts)
+					return &powerWatts
+				}
+			}
+		}
+	}
+
+	// Fallback: try to find any power1_input file in hwmon directories
+	// This covers cases where the PSU might not be properly identified by name
+	for _, hwmonDir := range hwmonDirs {
+		powerFile := filepath.Join(hwmonDir, "power1_input")
+		if powerData, err := os.ReadFile(powerFile); err == nil {
+			// Check if there's a corresponding power1_label to see if it's PSU-related
+			labelFile := filepath.Join(hwmonDir, "power1_label")
+			if labelData, err := os.ReadFile(labelFile); err == nil {
+				label := strings.ToLower(strings.TrimSpace(string(labelData)))
+				if strings.Contains(label, "psu") || strings.Contains(label, "power") {
+					valueStr := strings.TrimSpace(string(powerData))
+					if value, err := strconv.ParseFloat(valueStr, 64); err == nil {
+						powerWatts := value / 1_000_000
+						slog.Debug("Found PSU power usage via label", "hwmon", hwmonDir, "label", label, "power_watts", powerWatts)
+						return &powerWatts
+					}
+				}
+			}
+		}
+	}
+
+	slog.Debug("No PSU power usage sensors found")
+	return nil
+}

data/module/system/system_other.go

@@ -10,3 +10,7 @@ func GetCameraUsage() []string {
 func GetPendingReboot() *bool {
 	return nil
 }
+
+func GetPSUPowerUsage() *float64 {
+	return nil
+}

types/system.go

@@ -26,6 +26,7 @@ type SystemData struct {
 	MACAddress            string       `json:"mac_address" mapstructure:"mac_address"`
 	PlatformVersion       string       `json:"platform_version" mapstructure:"platform_version"`
 	Platform              string       `json:"platform" mapstructure:"platform"`
+	PowerUsage            *float64     `json:"power_usage" mapstructure:"power_usage"`
 	Uptime                uint64       `json:"uptime" mapstructure:"uptime"`
 	Users                 []SystemUser `json:"users" mapstructure:"users"`
 	UUID                  string       `json:"uuid" mapstructure:"uuid"`