init

src/modules/disk.rs

@@ -0,0 +1,46 @@
+use crate::config::Config;
+use crate::modules::WaybarModule;
+use crate::output::WaybarOutput;
+use crate::state::SharedState;
+use anyhow::Result;
+use sysinfo::Disks;
+
+pub struct DiskModule;
+
+impl WaybarModule for DiskModule {
+    fn run(&self, _config: &Config, _state: &SharedState, args: &[&str]) -> Result<WaybarOutput> {
+        let mountpoint = args.first().unwrap_or(&"/");
+        
+        let disks = Disks::new_with_refreshed_list();
+        for disk in &disks {
+            if disk.mount_point().to_string_lossy() == *mountpoint {
+                let total = disk.total_space() as f64;
+                let available = disk.available_space() as f64;
+                let used = total - available;
+                
+                let used_gb = used / 1024.0 / 1024.0 / 1024.0;
+                let total_gb = total / 1024.0 / 1024.0 / 1024.0;
+                let free_gb = available / 1024.0 / 1024.0 / 1024.0;
+                
+                let percentage = if total > 0.0 { (used / total) * 100.0 } else { 0.0 };
+
+                let class = if percentage > 95.0 {
+                    "max"
+                } else if percentage > 80.0 {
+                    "high"
+                } else {
+                    "normal"
+                };
+
+                return Ok(WaybarOutput {
+                    text: format!("{} {:.1}G/{:.1}G", mountpoint, used_gb, total_gb),
+                    tooltip: Some(format!("Used: {:.1}G\nTotal: {:.1}G\nFree: {:.1}G", used_gb, total_gb, free_gb)),
+                    class: Some(class.to_string()),
+                    percentage: Some(percentage as u8),
+                });
+            }
+        }
+
+        Err(anyhow::anyhow!("Mountpoint {} not found", mountpoint))
+    }
+}