protobuf file for block devices

1 files changed, 8 insertions, 3 deletions

diff --git a/doDrives.go b/doDrives.go
index a1e0b91..6ded27a 100644
--- a/doDrives.go
+++ b/doDrives.go
@@ -17,7 +17,11 @@ func doDrives() {
 	}
 
 	for _, p := range parts {
-		log.Printf("Device: /dev/%s\tSize: %d KiB\n", p.Name, p.Blocks)
+		log.Printf("RAW Device: /dev/%s\tSize: %d KiB\n", p.Name, p.Blocks)
+		devname := "/dev/" + p.Name
+		if blkpb := me.pb.FindByName(devname); blkpb != nil {
+			blkpb.SizeBytes = uint64(p.Blocks) * 1024
+		}
 	}
 
 	blockDir := "/sys/block"
@@ -44,11 +48,12 @@ func doDrives() {
 			used := false
 			for _, part := range partitions {
 				partName := filepath.Base(part)
+				partDev := "/dev/" + partName
 				if partName == dev {
+					log.Printf("  SKIP MAIN Partition: %s\n", partDev)
 					continue // skip the main device
 				}
 				hasPartition = true
-				partDev := "/dev/" + partName
 				isMounted := mounts[partDev]
 				log.Printf("  Partition: %s [%v]\n", partDev, isMounted)
 				if isMounted {
@@ -61,7 +66,7 @@ func doDrives() {
 			}
 
 			if !used {
-				log.Println("  → Drive appears to be unused.")
+				log.Println("  * Drive appears to be unused.")
 			}
 		}
 		return nil