From 13cd6f1d45eaaf9e7d1d61b6053a97e6adb280b0 Mon Sep 17 00:00:00 2001
From: EnumDev <enumdev@enumerated.dev>
Date: Sun, 21 Dec 2025 17:27:39 +0200
Subject: [PATCH] Improve partitions module

---
 src/modules.go    | 17 ++++++++++++++---
 src/partitions.go | 19 ++++++++++++-------
 2 files changed, 26 insertions(+), 10 deletions(-)

diff --git a/src/modules.go b/src/modules.go
index 5883481..1400754 100644
--- a/src/modules.go
+++ b/src/modules.go
@@ -262,9 +262,10 @@ func initializeModuleMap() {
 	RegisterModule(memoryModule)
 
 	// Partitions module
-	partitionsModule := StormfetchModule{stormfetchModuleConfig: stormfetchModuleConfig{Name: "partitions", Format: "%3Partition $PART_AUTONAME: %4$PART_USED / $PART_TOTAL"}, Execute: func(sm StormfetchModule) string {
+	partitionsModule := StormfetchModule{stormfetchModuleConfig: stormfetchModuleConfig{Name: "partitions", Format: "%3Partition ${PART_AUTONAME} (${PART_FS}): %4${PART_USED} / ${PART_TOTAL}"}, Execute: func(sm StormfetchModule) string {
 		hiddenPartitionsInterface, _ := sm.GetData("hidden_partitions", make([]any, 0))
 		hiddenFilesystemsInterface, _ := sm.GetData("hidden_filesystems", make([]any, 0))
+		alternativeNamesInterface, _ := sm.GetData("alternative_names", make(map[string]any, 0))
 
 		// Convert interface slices to string slices
 		hiddenPartitions := make([]string, 0)
@@ -276,12 +277,22 @@ func initializeModuleMap() {
 			hiddenFilesystems = append(hiddenFilesystems, value.(string))
 		}
 
+		// Convert interface map to map[string]string
+		alternativeNames := make(map[string]string)
+		for key, value := range alternativeNamesInterface.(map[string]any) {
+			alternativeNames[key] = value.(string)
+		}
+
 		builder := strings.Builder{}
 		partitions := GetMountedPartitions(hiddenPartitions, hiddenFilesystems)
 
 		for i, partition := range partitions {
-			partitionAutoname := partition.Label
-			if partitionAutoname == "" {
+			partitionAutoname := ""
+			if altName, ok := alternativeNames[partition.Device]; ok {
+				partitionAutoname = altName
+			} else if partition.Label != "" {
+				partitionAutoname = partition.Label
+			} else {
 				partitionAutoname = partition.MountPoint
 			}
 
diff --git a/src/partitions.go b/src/partitions.go
index 047c307..09b5489 100644
--- a/src/partitions.go
+++ b/src/partitions.go
@@ -72,14 +72,19 @@ func GetMountedPartitions(hiddenPartitions, hiddenFilesystems []string) []partit
 			continue
 		}
 
+		device, err := filepath.EvalSymlinks(fields[0])
+		if err != nil {
+			device = fields[0]
+		}
+
 		p := partition{
-			fields[0],
-			fields[1],
-			"",
-			fields[2],
-			0,
-			0,
-			0,
+			Device:        device,
+			MountPoint:    fields[1],
+			Label:         "",
+			FileystemType: fields[2],
+			TotalSize:     0,
+			UsedSize:      0,
+			FreeSize:      0,
 		}
 
 		// Skip already added partitions