diff --git a/internal/storage/storage.go b/internal/storage/storage.go index 58628db..23f9ada 100644 --- a/internal/storage/storage.go +++ b/internal/storage/storage.go @@ -66,8 +66,6 @@ type Partition struct { UsageStat UsageStat `json:"usage_stat"` } -const mainDiskID = "mmcblk0" - // GetDevices : Returns a list of all available sotrage devices in structs filled with information func GetDevices() []Device { @@ -82,6 +80,14 @@ func GetDevices() []Device { firstDevice := true currentDeviceInUseFlag := false + mainDiskID := "sda" + + if diagnostics.Version == "prod" || diagnostics.Version == "dev" { + mainDiskID = "mmcblk0" + } else if diagnostics.Version == "cloud" { + mainDiskID = "vda" + } + for scanner.Scan() { // 1 Device is represented here. Extract words in order for filling a Device struct // Example deviceRawInfo: "mmcblk0 179:0 0 29.7G 0 disk"