fixed dnd

Cargo.lock

@@ -621,7 +621,7 @@ checksum = "1d674e81391d1e1ab681a28d99df07927c6d4aa5b027d7da16ba32d1d21ecd99"
 
 [[package]]
 name = "fluxo-rs"
-version = "0.3.3"
+version = "0.4.0"
 dependencies = [
  "anyhow",
  "bluer",

src/modules/disk.rs

@@ -121,7 +121,7 @@ mod tests {
 
     #[tokio::test]
     async fn test_disk_not_found() {
-        let state = mock_state(AppState::default());
+        let state = state_with_disk("/", 100, 50);
         let config = Config::default();
         let result = DiskModule
             .run(&config, &state.receivers, &["/nonexistent"])

src/modules/dnd.rs

@@ -15,8 +15,38 @@ impl WaybarModule for DndModule {
         &self,
         config: &Config,
         state: &AppReceivers,
-        _args: &[&str],
+        args: &[&str],
     ) -> Result<WaybarOutput> {
+        let action = args.first().unwrap_or(&"show");
+
+        if *action == "toggle" {
+            let connection = Connection::session().await.map_err(|e| crate::error::FluxoError::Module {
+                module: "dnd",
+                message: format!("DBus connection failed: {}", e),
+            })?;
+
+            // Try toggling SwayNC
+            if let Ok(proxy) = SwayncControlProxy::new(&connection).await {
+                if let Ok(is_dnd) = proxy.dnd().await {
+                    let _ = proxy.set_dnd(!is_dnd).await;
+                    return Ok(WaybarOutput::default());
+                }
+            }
+
+            // Try toggling Dunst
+            if let Ok(proxy) = DunstControlProxy::new(&connection).await {
+                if let Ok(is_paused) = proxy.paused().await {
+                    let _ = proxy.set_paused(!is_paused).await;
+                    return Ok(WaybarOutput::default());
+                }
+            }
+
+            return Err(crate::error::FluxoError::Module {
+                module: "dnd",
+                message: "No supported notification daemon found to toggle".to_string(),
+            });
+        }
+
         let is_dnd = state.dnd.borrow().is_dnd;
 
         if is_dnd {
@@ -47,6 +77,20 @@ pub struct DndDaemon;
 trait SwayncControl {
     #[zbus(property)]
     fn dnd(&self) -> zbus::Result<bool>;
+    #[zbus(property)]
+    fn set_dnd(&self, value: bool) -> zbus::Result<()>;
+}
+
+#[proxy(
+    interface = "org.dunstproject.cmd0",
+    default_service = "org.freedesktop.Notifications",
+    default_path = "/org/freedesktop/Notifications"
+)]
+trait DunstControl {
+    #[zbus(property)]
+    fn paused(&self) -> zbus::Result<bool>;
+    #[zbus(property)]
+    fn set_paused(&self, value: bool) -> zbus::Result<()>;
 }
 
 impl DndDaemon {
@@ -70,16 +114,12 @@ impl DndDaemon {
 
         info!("Connected to D-Bus for DND monitoring");
 
-        // Try SwayNC first
+        // Try SwayNC
         if let Ok(proxy) = SwayncControlProxy::new(&connection).await {
-            debug!("Found SwayNC, using it for DND state.");
-
-            // Get initial state
             if let Ok(is_dnd) = proxy.dnd().await {
+                debug!("Found SwayNC, using it for DND state.");
                 let _ = tx.send(DndState { is_dnd });
-            }
 
-            // Monitor properties changed
                 if let Ok(props_proxy) = PropertiesProxy::builder(&connection)
                     .destination("org.erikreider.swaync.control")?
                     .path("/org/erikreider/swaync/control")?
@@ -98,6 +138,33 @@ impl DndDaemon {
                     }
                 }
             }
+        }
+
+        // Try Dunst
+        if let Ok(proxy) = DunstControlProxy::new(&connection).await {
+            if let Ok(is_dnd) = proxy.paused().await {
+                debug!("Found Dunst, using it for DND state.");
+                let _ = tx.send(DndState { is_dnd });
+
+                if let Ok(props_proxy) = PropertiesProxy::builder(&connection)
+                    .destination("org.freedesktop.Notifications")?
+                    .path("/org/freedesktop/Notifications")?
+                    .build()
+                    .await
+                {
+                    let mut stream = props_proxy.receive_properties_changed().await?;
+                    while let Some(signal) = stream.next().await {
+                        let args = signal.args()?;
+                        if args.interface_name == "org.dunstproject.cmd0"
+                            && let Some(val) = args.changed_properties.get("paused")
+                            && let Ok(is_dnd) = bool::try_from(val)
+                        {
+                            let _ = tx.send(DndState { is_dnd });
+                        }
+                    }
+                }
+            }
+        }
 
         Err(anyhow::anyhow!("DND stream ended or daemon not found"))
     }