feature: multiple storage devices (#170)

Implemented changes to INFO response visualization and to DiskWriter node to accommodate for the multiple storage devices change to Synapse API.

Author: calvinleng-science

synapse-api

@@ -91,6 +91,22 @@ def visualize_peripherals(info_dict):
         tree.add("No peripherals found")
     return tree
 
+def visualize_storage_devices(status):
+    tree = Tree("Storage Devices")
+    storage_devices = status.get("storage", {}).get("storage_devices", [])
+    if storage_devices:
+        for storage_device in storage_devices:
+            storage_devices_tree = tree.add(
+                f"{storage_device.get('name', 'Unknown')}"
+            )
+            total = float(storage_device.get("total_gb", 0))
+            used = float(storage_device.get("used_gb", 0))
+            used_percent = (used / total * 100) if total > 0 else 0
+            storage_devices_tree.add(f"ID: {storage_device.get('storage_device_id', 'Unknown')}")
+            storage_devices_tree.add(f"Storage: {used_percent:.1f}% used ({used:.1f}GB / {total:.1f}GB)")
+    else:
+        tree.add("No storage devices found")
+    return tree
 
 class DeviceInfoDisplay:
     """A class for displaying device information."""
@@ -135,15 +151,6 @@ def summary(self, device: Device):
                 battery = status["power"].get("battery_level_percent", "N/A")
                 self.console.print(f"Battery: {battery}%", highlight=False)
 
-            if "storage" in status:
-                storage = status["storage"]
-                total = float(storage.get("total_gb", 0))
-                used = float(storage.get("used_gb", 0))
-                used_percent = (used / total * 100) if total > 0 else 0
-                self.console.print(
-                    f"Storage: {used_percent:.1f}% used ({used:.1f}GB / {total:.1f}GB)",
-                    highlight=False,
-                )
-
+            self.console.print(visualize_storage_devices(status))
             self.console.print(visualize_peripherals(info_dict))
             self.console.print(visualize_configuration(info_dict, status))

synapse/cli/device_info_display.py

@@ -6,12 +6,13 @@
 class DiskWriter(Node):
     type = NodeType.kDiskWriter
 
-    def __init__(self, filename: str):
+    def __init__(self, filename: str, storage_device_id: int):
         self.filename = filename
+        self.storage_device_id = storage_device_id
 
     def _to_proto(self):
         n = NodeConfig()
-        p = DiskWriterConfig(filename=self.filename)
+        p = DiskWriterConfig(filename=self.filename, storage_device_id=self.storage_device_id)
         n.disk_writer.CopyFrom(p)
         return n
 
@@ -22,4 +23,4 @@ def _from_proto(proto: DiskWriterConfig):
         if not isinstance(proto, DiskWriterConfig):
             raise ValueError("proto is not of type DiskWriterConfig")
 
-        return DiskWriter(filename=proto.filename)
+        return DiskWriter(filename=proto.filename, storage_device_id=proto.storage_device_id)