version bupm

Reviewed-on: #1

.gitea/workflows/ci.yml

@@ -19,6 +19,8 @@ jobs:
           toolchain: stable
           components: rustfmt, clippy
           cache: false
+      - name: Install packaging tools
+        run: apt-get update && apt-get install -y dpkg-dev pkg-config libdbus-1-dev protobuf-compiler libpulse-dev
 
       - name: Check formatting
         run: cargo fmt --check

Cargo.lock

@@ -353,7 +353,7 @@ checksum = "1d674e81391d1e1ab681a28d99df07927c6d4aa5b027d7da16ba32d1d21ecd99"
 
 [[package]]
 name = "fluxo-rs"
-version = "0.3.0"
+version = "0.3.2"
 dependencies = [
  "anyhow",
  "bluer",
@@ -362,12 +362,13 @@ dependencies = [
  "futures",
  "libpulse-binding",
  "maestro",
- "nix 0.29.0",
+ "nix 0.31.2",
  "regex",
  "serde",
  "serde_json",
  "sysinfo",
  "tempfile",
+ "thiserror",
  "tokio",
  "tokio-util",
  "toml",
@@ -584,9 +585,9 @@ checksum = "09edd9e8b54e49e587e4f6295a7d29c3ea94d469cb40ab8ca70b288248a81db2"
 
 [[package]]
 name = "libc"
-version = "0.2.183"
+version = "0.2.184"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "b5b646652bf6661599e1da8901b3b9522896f01e736bad5f723fe7a3a27f899d"
+checksum = "48f5d2a454e16a5ea0f4ced81bd44e4cfc7bd3a507b61887c99fd3538b28e4af"
 
 [[package]]
 name = "libdbus-sys"
@@ -710,7 +711,6 @@ dependencies = [
  "cfg-if",
  "cfg_aliases",
  "libc",
- "memoffset",
 ]
 
 [[package]]
@@ -723,6 +723,7 @@ dependencies = [
  "cfg-if",
  "cfg_aliases",
  "libc",
+ "memoffset",
 ]
 
 [[package]]
@@ -1067,9 +1068,9 @@ dependencies = [
 
 [[package]]
 name = "serde_spanned"
-version = "1.1.0"
+version = "1.1.1"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "876ac351060d4f882bb1032b6369eb0aef79ad9df1ea8bc404874d8cc3d0cd98"
+checksum = "6662b5879511e06e8999a8a235d848113e942c9124f211511b16466ee2995f26"
 dependencies = [
  "serde_core",
 ]
@@ -1083,6 +1084,16 @@ dependencies = [
  "lazy_static",
 ]
 
+[[package]]
+name = "signal-hook-registry"
+version = "1.4.8"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "c4db69cba1110affc0e9f7bcd48bbf87b3f4fc7c61fc9155afd4c469eb3d6c1b"
+dependencies = [
+ "errno",
+ "libc",
+]
+
 [[package]]
 name = "slab"
 version = "0.4.12"
@@ -1182,6 +1193,26 @@ dependencies = [
  "windows-sys 0.61.2",
 ]
 
+[[package]]
+name = "thiserror"
+version = "2.0.18"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "4288b5bcbc7920c07a1149a35cf9590a2aa808e0bc1eafaade0b80947865fbc4"
+dependencies = [
+ "thiserror-impl",
+]
+
+[[package]]
+name = "thiserror-impl"
+version = "2.0.18"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "ebc4ee7f67670e9b64d05fa4253e753e016c6c95ff35b89b7941d6b856dec1d5"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "syn",
+]
+
 [[package]]
 name = "thread_local"
 version = "1.1.9"
@@ -1201,6 +1232,7 @@ dependencies = [
  "libc",
  "mio",
  "pin-project-lite",
+ "signal-hook-registry",
  "socket2",
  "tokio-macros",
  "windows-sys 0.61.2",
@@ -1238,14 +1270,15 @@ dependencies = [
  "futures-core",
  "futures-sink",
  "pin-project-lite",
+ "slab",
  "tokio",
 ]
 
 [[package]]
 name = "toml"
-version = "1.1.0+spec-1.1.0"
+version = "1.1.1+spec-1.1.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "f8195ca05e4eb728f4ba94f3e3291661320af739c4e43779cbdfae82ab239fcc"
+checksum = "994b95d9e7bae62b34bab0e2a4510b801fa466066a6a8b2b57361fa1eba068ee"
 dependencies = [
  "indexmap",
  "serde_core",
@@ -1258,9 +1291,9 @@ dependencies = [
 
 [[package]]
 name = "toml_datetime"
-version = "1.1.0+spec-1.1.0"
+version = "1.1.1+spec-1.1.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "97251a7c317e03ad83774a8752a7e81fb6067740609f75ea2b585b569a59198f"
+checksum = "3165f65f62e28e0115a00b2ebdd37eb6f3b641855f9d636d3cd4103767159ad7"
 dependencies = [
  "serde_core",
 ]
@@ -1279,18 +1312,18 @@ dependencies = [
 
 [[package]]
 name = "toml_parser"
-version = "1.1.0+spec-1.1.0"
+version = "1.1.1+spec-1.1.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "2334f11ee363607eb04df9b8fc8a13ca1715a72ba8662a26ac285c98aabb4011"
+checksum = "39ca317ebc49f06bd748bfba29533eac9485569dc9bf80b849024b025e814fb9"
 dependencies = [
  "winnow",
 ]
 
 [[package]]
 name = "toml_writer"
-version = "1.1.0+spec-1.1.0"
+version = "1.1.1+spec-1.1.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "d282ade6016312faf3e41e57ebbba0c073e4056dab1232ab1cb624199648f8ed"
+checksum = "756daf9b1013ebe47a8776667b466417e2d4c5679d441c26230efd9ef78692db"
 
 [[package]]
 name = "tracing"

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "fluxo-rs"
-version = "0.3.0"
+version = "0.3.3"
 edition = "2024"
 
 [dependencies]
@@ -11,16 +11,17 @@ regex = "1.12"
 serde = { version = "1.0.228", features = ["derive"] }
 serde_json = "1.0.149"
 sysinfo = "0.38.4"
-toml = "1.0.6"
+thiserror = "2.0"
+toml = "1.1.1"
 tracing = "0.1.44"
 tracing-subscriber = { version = "0.3.23", features = ["env-filter"] }
 maestro = { git = "https://github.com/qzed/pbpctrl", package = "maestro" }
 bluer = { version = "0.17", features = ["bluetoothd", "rfcomm", "id"] }
-tokio = { version = "1", features = ["rt", "rt-multi-thread", "sync", "time", "macros"] }
+tokio = { version = "1", features = ["rt", "rt-multi-thread", "sync", "time", "macros", "signal", "process"] }
-tokio-util = { version = "0.7", features = ["codec"] }
+tokio-util = { version = "0.7", features = ["codec", "time"] }
 futures = "0.3"
-libpulse-binding = "2.26"
+libpulse-binding = "2.30"
-nix = { version = "0.29", features = ["net"] }
+nix = { version = "0.31", features = ["net"] }
 
 [dev-dependencies]
 

README.md

@@ -1,95 +1,66 @@
 # fluxo-rs
 
-fluxo-rs is a high-performance system metrics daemon and client designed specifically for waybar. It replaces standard shell scripts with a compiled rust binary that collects data via a background polling loop and serves it over a unix domain socket (`/tmp/fluxo.sock`).
+fluxo-rs is a high-performance system metrics daemon and client designed specifically for Waybar. It replaces standard shell scripts with a compiled Rust binary that collects data via a background polling loop and serves it over a Unix socket.
 
-## Description
+## Key Features
 
-The project follows a client-server architecture:
+- **Asynchronous Architecture**: Built on **Tokio**, the daemon handles concurrent IPC requests and background tasks with zero latency and minimal CPU overhead.
-- **Daemon**: Handles heavy lifting (polling cpu, memory, network, gpu) and stores state in memory.
+- **Native Library Integrations**: 
-- **Client**: A thin cli wrapper that connects to the daemon's socket to retrieve formatted json for waybar.
+    - **Audio**: Direct `libpulse` integration for event-driven, instant volume and device updates.
-
+    - **Bluetooth**: Native `bluer` integration for robust device monitoring.
-This approach eliminates process spawning overhead and temporary file locking, resulting in near-zero cpu usage for custom modules.
+    - **Pixel Buds Pro**: Custom native RPC implementation for real-time battery and ANC control.
-
+    - **Network**: Native `nix` and `/proc` inspection for high-speed interface monitoring.
-## Features
+    - **Hyprland**: Direct IPC Unix socket communication for gamemode and animation status.
-
+- **Circuit Breaker (Failsafe)**: Automatically detects failing modules and enters a "Cool down" state to prevent resource waste and log spam.
-- **Ultra-lightweight**: Background polling is highly optimized (e.g., O(1) process counting).
+- **Multi-threaded Polling**: Decoupled subsystem threads ensure that a hang in one system (e.g., a slow GPU probe) never freezes your entire bar.
-- **Jitter-free**: Uses zero-width sentinels and figure spaces to prevent waybar from trimming padding.
-- **Configurable**: Fully customizable output formats via toml config.
-- **Interactive Menus**: Integrated support for selecting items (like Bluetooth devices) via external menus (e.g., Rofi, Wofi, Fuzzel).
-- **Live Reload**: Configuration can be reloaded without restarting the daemon.
-- **Multi-vendor GPU**: Native support for intel (igpu), amd, and nvidia.
 
 ## Modules
 
 | Command | Description | Tokens |
 | :--- | :--- | :--- |
-| `net` | Network speed (rx/tx) | `{interface}`, `{ip}`, `{rx}`, `{tx}` |
+| `cpu` | CPU usage and temperature | `{usage}`, `{temp}`, `{model}` |
-| `cpu` | CPU usage and temp | `{usage}`, `{temp}` |
 | `mem` | Memory usage | `{used}`, `{total}` |
-| `gpu` | GPU utilization | `{usage}`, `{vram_used}`, `{vram_total}`, `{temp}` |
+| `net` | Network status & speeds | `{interface}`, `{ip}`, `{rx}`, `{tx}` |
-| `sys` | System load and uptime | `{uptime}`, `{load1}`, `{load5}`, `{load15}` |
+| `sys` | System load and uptime | `{uptime}`, `{load1}`, `{load5}`, `{load15}`, `{procs}` |
-| `disk` | Disk usage (default: /) | `{mount}`, `{used}`, `{total}` |
+| `disk` | Disk usage | `{mount}`, `{used}`, `{total}` |
-| `pool` | Aggregate storage (btrfs) | `{used}`, `{total}` |
+| `pool` | Btrfs aggregate storage | `{used}`, `{total}` |
-| `vol` | Audio output volume | `{name}`, `{volume}`, `{icon}` |
+| `vol` | Audio output (sink) | `{name}`, `{volume}`, `{icon}` |
-| `mic` | Audio input volume | `{name}`, `{volume}`, `{icon}` |
+| `mic` | Audio input (source) | `{name}`, `{volume}`, `{icon}` |
 | `bt` | Bluetooth status & plugins | `{alias}`, `{mac}`, `{left}`, `{right}`, `{anc}` |
 | `power` | Battery and AC status | `{percentage}`, `{icon}` |
-| `game` | Hyprland gamemode status | active/inactive icon strings |
+| `game` | Hyprland status | active/inactive icons |
 
 ## Setup
 
-1. Build the project:
+1. **Build**: `cargo build --release`
-   ```bash
+2. **Configure**: Create `~/.config/fluxo/config.toml` (see `example.config.toml`).
-   cd fluxo-rs
+3. **Daemon**: Start `fluxo-rs daemon`. It's recommended to run this as a systemd user service.
-   cargo build --release
-   ```
 
-2. Start the daemon:
+## Waybar Configuration
-   ```bash
-   # Starts the daemon using the default config path (~/.config/fluxo/config.toml)
-   ./target/release/fluxo-rs daemon &
 
-   # Or provide a custom path
+To achieve zero-latency updates, use **Waybar Signals**:
-   ./target/release/fluxo-rs daemon --config /path/to/your/config.toml &
-   ```
 
-3. Configuration:
+```jsonc
-   Create `~/.config/fluxo/config.toml` (see `example.config.toml` for all default options).
+"custom/audio": {
-
+    "exec": "fluxo vol",
-4. Waybar configuration (`config.jsonc`):
+    "return-type": "json",
-   ```json
+    "interval": 5,
-   "custom/cpu": {
+    "signal": 8,
-       "exec": "~/path/to/fluxo-rs cpu",
+    "on-click": "fluxo audio cycle sink && pkill -RTMIN+8 waybar"
-       "return-type": "json"
+},
-   }
+"custom/bluetooth": {
-   ```
+    "exec": "fluxo bt",
-
+    "return-type": "json",
-## Development
+    "interval": 5,
-
+    "signal": 9,
-### Architecture
+    "on-click": "fluxo bt menu && pkill -RTMIN+9 waybar",
-- `src/main.rs`: Entry point, CLI parsing, interactive GUI spawns (menus), and client-side formatting logic.
+    "on-click-right": "fluxo bt cycle_mode && pkill -RTMIN+9 waybar"
-- `src/daemon.rs`: UDS listener, configuration management, and polling orchestration.
+}
-- `src/ipc.rs`: Unix domain socket communication logic.
-- `src/utils.rs`: Generic GUI utilities (like the menu spawner).
-- `src/modules/`: Individual metric implementations.
-- `src/state.rs`: Shared thread-safe data structures.
-
-### Adding a Module
-1. Add the required config block to `src/config.rs`.
-2. Add the required state fields to `src/state.rs`.
-3. Implement the `WaybarModule` trait in a new file in `src/modules/`.
-4. Add polling logic to `src/modules/hardware.rs` or `src/daemon.rs`.
-5. Register the new subcommand in `src/main.rs` and the router in `src/daemon.rs`.
-
-### Configuration Reload
-The daemon can reload its configuration live:
-```bash
-fluxo-rs reload
 ```
 
-### Logging & Debugging
+## Debugging
-`fluxo-rs` uses the `tracing` ecosystem. If a module isn't behaving properly or your configuration isn't applying, start the daemon with debug logging enabled in the foreground:
+
+Start the daemon with `RUST_LOG=debug` to see detailed logs of library interactions and circuit breaker status:
 ```bash
 RUST_LOG=debug fluxo-rs daemon
 ```
-This will print verbose information regarding config parsing errors, subprocess failures, and IPC requests.

src/config.rs

@@ -5,7 +5,7 @@ use std::path::PathBuf;
 use std::sync::LazyLock;
 use tracing::{debug, info, warn};
 
-#[derive(Deserialize, Default)]
+#[derive(Deserialize, Default, Clone)]
 pub struct Config {
     #[serde(default)]
     pub general: GeneralConfig,
@@ -33,7 +33,7 @@ pub struct Config {
     pub game: GameConfig,
 }
 
-#[derive(Deserialize)]
+#[derive(Deserialize, Clone)]
 pub struct GeneralConfig {
     pub menu_command: String,
 }
@@ -46,7 +46,7 @@ impl Default for GeneralConfig {
     }
 }
 
-#[derive(Deserialize)]
+#[derive(Deserialize, Clone)]
 pub struct NetworkConfig {
     pub format: String,
 }
@@ -59,7 +59,7 @@ impl Default for NetworkConfig {
     }
 }
 
-#[derive(Deserialize)]
+#[derive(Deserialize, Clone)]
 pub struct CpuConfig {
     pub format: String,
 }
@@ -72,7 +72,7 @@ impl Default for CpuConfig {
     }
 }
 
-#[derive(Deserialize)]
+#[derive(Deserialize, Clone)]
 pub struct MemoryConfig {
     pub format: String,
 }
@@ -85,7 +85,7 @@ impl Default for MemoryConfig {
     }
 }
 
-#[derive(Deserialize)]
+#[derive(Deserialize, Clone)]
 pub struct GpuConfig {
     pub format_amd: String,
     pub format_intel: String,
@@ -104,7 +104,7 @@ impl Default for GpuConfig {
     }
 }
 
-#[derive(Deserialize)]
+#[derive(Deserialize, Clone)]
 pub struct SysConfig {
     pub format: String,
 }
@@ -117,7 +117,7 @@ impl Default for SysConfig {
     }
 }
 
-#[derive(Deserialize)]
+#[derive(Deserialize, Clone)]
 pub struct DiskConfig {
     pub format: String,
 }
@@ -130,7 +130,7 @@ impl Default for DiskConfig {
     }
 }
 
-#[derive(Deserialize)]
+#[derive(Deserialize, Clone)]
 pub struct PoolConfig {
     pub format: String,
 }
@@ -143,7 +143,7 @@ impl Default for PoolConfig {
     }
 }
 
-#[derive(Deserialize)]
+#[derive(Deserialize, Clone)]
 pub struct PowerConfig {
     pub format: String,
 }
@@ -156,7 +156,7 @@ impl Default for PowerConfig {
     }
 }
 
-#[derive(Deserialize)]
+#[derive(Deserialize, Clone)]
 pub struct AudioConfig {
     pub format_sink_unmuted: String,
     pub format_sink_muted: String,
@@ -175,7 +175,7 @@ impl Default for AudioConfig {
     }
 }
 
-#[derive(Deserialize)]
+#[derive(Deserialize, Clone)]
 pub struct BtConfig {
     pub format_connected: String,
     pub format_plugin: String,
@@ -194,7 +194,7 @@ impl Default for BtConfig {
     }
 }
 
-#[derive(Deserialize)]
+#[derive(Deserialize, Clone)]
 pub struct GameConfig {
     pub format_active: String,
     pub format_inactive: String,

src/daemon.rs

@@ -1,21 +1,22 @@
 use crate::config::Config;
+use crate::error::FluxoError;
 use crate::ipc::socket_path;
 use crate::modules::WaybarModule;
 use crate::modules::audio::AudioDaemon;
 use crate::modules::bt::BtDaemon;
 use crate::modules::hardware::HardwareDaemon;
 use crate::modules::network::NetworkDaemon;
-use crate::state::{AppState, SharedState};
+use crate::state::AppReceivers;
 use anyhow::Result;
+use std::collections::HashMap;
 use std::fs;
-use std::io::{BufRead, BufReader, Write};
-use std::net::Shutdown;
-use std::os::unix::net::UnixListener;
 use std::path::PathBuf;
-use std::sync::atomic::{AtomicBool, Ordering};
+use std::sync::Arc;
-use std::sync::{Arc, RwLock};
+use tokio::io::{AsyncBufReadExt, AsyncWriteExt, BufReader};
-use std::thread;
+use tokio::net::UnixListener;
-use std::time::Duration;
+use tokio::sync::{RwLock, watch};
+use tokio::time::{Duration, Instant, sleep};
+use tokio_util::sync::CancellationToken;
 use tracing::{debug, error, info, warn};
 
 struct SocketGuard {
@@ -29,7 +30,7 @@ impl Drop for SocketGuard {
     }
 }
 
-pub fn run_daemon(config_path: Option<PathBuf>) -> Result<()> {
+pub async fn run_daemon(config_path: Option<PathBuf>) -> Result<()> {
     let sock_path = socket_path();
 
     if fs::metadata(&sock_path).is_ok() {
@@ -37,95 +38,150 @@ pub fn run_daemon(config_path: Option<PathBuf>) -> Result<()> {
         fs::remove_file(&sock_path)?;
     }
 
-    let state: SharedState = Arc::new(RwLock::new(AppState::default()));
+    let (net_tx, net_rx) = watch::channel(Default::default());
+    let (cpu_tx, cpu_rx) = watch::channel(Default::default());
+    let (mem_tx, mem_rx) = watch::channel(Default::default());
+    let (sys_tx, sys_rx) = watch::channel(Default::default());
+    let (gpu_tx, gpu_rx) = watch::channel(Default::default());
+    let (disks_tx, disks_rx) = watch::channel(Default::default());
+    let (bt_tx, bt_rx) = watch::channel(Default::default());
+    let (audio_tx, audio_rx) = watch::channel(Default::default());
+    let health = Arc::new(RwLock::new(HashMap::new()));
+
+    let receivers = AppReceivers {
+        network: net_rx,
+        cpu: cpu_rx,
+        memory: mem_rx,
+        sys: sys_rx,
+        gpu: gpu_rx,
+        disks: disks_rx,
+        bluetooth: bt_rx,
+        audio: audio_rx,
+        health: Arc::clone(&health),
+    };
+
     let listener = UnixListener::bind(&sock_path)?;
     let _guard = SocketGuard {
         path: sock_path.clone(),
     };
 
-    // Signal handling: set flag so main loop exits cleanly
+    // Signal handling for graceful shutdown
-    let running = Arc::new(AtomicBool::new(true));
+    let cancel_token = CancellationToken::new();
-    let running_clone = Arc::clone(&running);
+    let token_clone = cancel_token.clone();
-    ctrlc::set_handler(move || {
+    tokio::spawn(async move {
+        tokio::signal::ctrl_c().await.unwrap();
         info!("Received shutdown signal, exiting...");
-        running_clone.store(false, Ordering::SeqCst);
+        token_clone.cancel();
-    })?;
+    });
 
-    // We store the original config_path to allow proper reloading later
     let config_path_clone = config_path.clone();
     let config = Arc::new(RwLock::new(crate::config::load_config(config_path)));
 
-    // 1. Network Thread
+    // 1. Network Task
-    let poll_state = Arc::clone(&state);
+    let token = cancel_token.clone();
-    let poll_running = Arc::clone(&running);
+    let net_health = Arc::clone(&health);
-    thread::spawn(move || {
+    tokio::spawn(async move {
-        info!("Starting Network polling thread");
+        info!("Starting Network polling task");
         let mut daemon = NetworkDaemon::new();
-        while poll_running.load(Ordering::SeqCst) {
+        loop {
-            daemon.poll(Arc::clone(&poll_state));
+            tokio::select! {
-            thread::sleep(Duration::from_secs(1));
+                _ = token.cancelled() => break,
+                _ = sleep(Duration::from_secs(1)) => {
+                    if !is_in_backoff("net", &net_health).await {
+                        let res = daemon.poll(&net_tx).await;
+                        handle_poll_result("net", res, &net_health).await;
                     }
+                }
+            }
+        }
+        info!("Network task shut down.");
     });
 
-    // 2. Fast Hardware Thread (CPU, Mem, Load)
+    // 2. Fast Hardware Task (CPU, Mem, Load)
-    let poll_state = Arc::clone(&state);
+    let token = cancel_token.clone();
-    let poll_running = Arc::clone(&running);
+    let hw_health = Arc::clone(&health);
-    thread::spawn(move || {
+    tokio::spawn(async move {
-        info!("Starting Fast Hardware polling thread");
+        info!("Starting Fast Hardware polling task");
         let mut daemon = HardwareDaemon::new();
-        while poll_running.load(Ordering::SeqCst) {
+        loop {
-            daemon.poll_fast(Arc::clone(&poll_state));
+            tokio::select! {
-            thread::sleep(Duration::from_secs(1));
+                _ = token.cancelled() => break,
+                _ = sleep(Duration::from_secs(1)) => {
+                    if !is_in_backoff("cpu", &hw_health).await {
+                        daemon.poll_fast(&cpu_tx, &mem_tx, &sys_tx).await;
                     }
+                }
+            }
+        }
+        info!("Fast Hardware task shut down.");
     });
 
-    // 3. Slow Hardware Thread (GPU, Disks)
+    // 3. Slow Hardware Task (GPU, Disks)
-    let poll_state = Arc::clone(&state);
+    let token = cancel_token.clone();
-    let poll_running = Arc::clone(&running);
+    let slow_health = Arc::clone(&health);
-    thread::spawn(move || {
+    tokio::spawn(async move {
-        info!("Starting Slow Hardware polling thread");
+        info!("Starting Slow Hardware polling task");
         let mut daemon = HardwareDaemon::new();
-        while poll_running.load(Ordering::SeqCst) {
+        loop {
-            daemon.poll_slow(Arc::clone(&poll_state));
+            tokio::select! {
-            thread::sleep(Duration::from_secs(1));
+                _ = token.cancelled() => break,
+                _ = sleep(Duration::from_secs(5)) => {
+                    if !is_in_backoff("gpu", &slow_health).await {
+                        daemon.poll_slow(&gpu_tx, &disks_tx).await;
                     }
+                }
+            }
+        }
+        info!("Slow Hardware task shut down.");
     });
 
-    // 4. Bluetooth Thread
+    // 4. Bluetooth Task
-    let poll_state = Arc::clone(&state);
+    let token = cancel_token.clone();
-    let poll_running = Arc::clone(&running);
+    let bt_health = Arc::clone(&health);
-    thread::spawn(move || {
+    let poll_config = Arc::clone(&config);
-        info!("Starting Bluetooth polling thread");
+    let poll_receivers = receivers.clone();
+    tokio::spawn(async move {
+        info!("Starting Bluetooth polling task");
         let mut daemon = BtDaemon::new();
-        while poll_running.load(Ordering::SeqCst) {
+        loop {
-            daemon.poll(Arc::clone(&poll_state));
+            tokio::select! {
-            thread::sleep(Duration::from_secs(1));
+                _ = token.cancelled() => break,
+                _ = sleep(Duration::from_secs(2)) => {
+                    if !is_in_backoff("bt", &bt_health).await {
+                        let config = poll_config.read().await;
+                        daemon.poll(&bt_tx, &poll_receivers, &config).await;
                     }
+                }
+            }
+        }
+        info!("Bluetooth task shut down.");
     });
 
     // 5. Audio Thread (Event driven)
     let audio_daemon = AudioDaemon::new();
-    audio_daemon.start(Arc::clone(&state));
+    audio_daemon.start(&audio_tx);
 
     info!("Fluxo daemon successfully bound to socket: {}", sock_path);
 
-    // Use non-blocking accept so we can check the running flag
+    loop {
-    listener.set_nonblocking(true)?;
+        tokio::select! {
-
+            _ = cancel_token.cancelled() => {
-    while running.load(Ordering::SeqCst) {
+                break;
-        match listener.accept() {
+            }
+            res = listener.accept() => {
+                match res {
                     Ok((mut stream, _)) => {
-                let state_clone = Arc::clone(&state);
+                        let state_clone = receivers.clone();
                         let config_clone = Arc::clone(&config);
                         let cp_clone = config_path_clone.clone();
-                thread::spawn(move || {
+                        tokio::spawn(async move {
-                    let mut reader = BufReader::new(&stream);
+                            let (reader, mut writer) = stream.split();
+                            let mut reader = BufReader::new(reader);
                             let mut request = String::new();
-                    if let Err(e) = reader.read_line(&mut request) {
+                            if let Err(e) = reader.read_line(&mut request).await {
                                 error!("Failed to read from IPC stream: {}", e);
                                 return;
                             }
-                    drop(reader);
 
                             let request = request.trim();
                             if request.is_empty() {
@@ -137,21 +193,17 @@ pub fn run_daemon(config_path: Option<PathBuf>) -> Result<()> {
                                 if *module_name == "reload" {
                                     info!("Reloading configuration...");
                                     let new_config = crate::config::load_config(cp_clone);
-                            if let Ok(mut config_lock) = config_clone.write() {
+                                    let mut config_lock = config_clone.write().await;
                                     *config_lock = new_config;
-                                let _ = stream.write_all(b"{\"text\":\"ok\"}");
+                                    let _ = writer.write_all(b"{\"text\":\"ok\"}").await;
                                     info!("Configuration reloaded successfully.");
-                            } else {
-                                error!("Failed to acquire write lock for configuration reload.");
-                            }
-                            let _ = stream.shutdown(Shutdown::Write);
                                     return;
                                 }
 
                                 debug!(module = module_name, args = ?&parts[1..], "Handling IPC request");
                                 let response =
-                            handle_request(module_name, &parts[1..], &state_clone, &config_clone);
+                                    handle_request(module_name, &parts[1..], &state_clone, &config_clone).await;
-                        if let Err(e) = stream.write_all(response.as_bytes()) {
+                                if let Err(e) = writer.write_all(response.as_bytes()).await {
                                     if e.kind() == std::io::ErrorKind::BrokenPipe
                                         || e.kind() == std::io::ErrorKind::ConnectionReset
                                     {
@@ -163,68 +215,193 @@ pub fn run_daemon(config_path: Option<PathBuf>) -> Result<()> {
                                         error!("Failed to write IPC response: {}", e);
                                     }
                                 }
-                        let _ = stream.shutdown(Shutdown::Write);
                             }
                         });
                     }
-            Err(ref e) if e.kind() == std::io::ErrorKind::WouldBlock => {
-                thread::sleep(Duration::from_millis(50));
-            }
                     Err(e) => error!("Failed to accept incoming connection: {}", e),
                 }
             }
+        }
+    }
 
     info!("Daemon shutting down gracefully.");
     Ok(())
 }
 
-fn handle_request(
+async fn handle_poll_result(
+    module_name: &str,
+    result: crate::error::Result<()>,
+    health_lock: &Arc<RwLock<HashMap<String, crate::state::ModuleHealth>>>,
+) {
+    let mut lock = health_lock.write().await;
+    let health = lock.entry(module_name.to_string()).or_default();
+
+    match result {
+        Ok(_) => {
+            if health.consecutive_failures > 0 {
+                info!(
+                    module = module_name,
+                    "Module recovered after {} failures", health.consecutive_failures
+                );
+            }
+            health.consecutive_failures = 0;
+            health.backoff_until = None;
+        }
+        Err(e) => {
+            health.consecutive_failures += 1;
+            health.last_failure = Some(Instant::now());
+
+            if !e.is_transient() {
+                // Fatal errors trigger immediate long backoff
+                health.backoff_until = Some(Instant::now() + Duration::from_secs(60));
+                error!(module = module_name, error = %e, "Fatal module error, entering long cooldown");
+            } else if health.consecutive_failures >= 3 {
+                // Exponential backoff for transient errors: 30s, 60s, 120s...
+                let backoff_secs = 30 * (2u64.pow(health.consecutive_failures.saturating_sub(3)));
+                let backoff_secs = backoff_secs.min(3600); // Cap at 1 hour
+                health.backoff_until = Some(Instant::now() + Duration::from_secs(backoff_secs));
+                warn!(module = module_name, error = %e, backoff = backoff_secs, "Repeated transient failures, entering backoff");
+            } else {
+                debug!(module = module_name, error = %e, "Transient module failure (attempt {})", health.consecutive_failures);
+            }
+        }
+    }
+}
+
+async fn is_in_backoff(
+    module_name: &str,
+    health_lock: &Arc<RwLock<HashMap<String, crate::state::ModuleHealth>>>,
+) -> bool {
+    let lock = health_lock.read().await;
+    if let Some(health) = lock.get(module_name)
+        && let Some(until) = health.backoff_until
+    {
+        return Instant::now() < until;
+    }
+    false
+}
+
+async fn handle_request(
     module_name: &str,
     args: &[&str],
-    state: &SharedState,
+    state: &AppReceivers,
     config_lock: &Arc<RwLock<Config>>,
 ) -> String {
-    let config = if let Ok(c) = config_lock.read() {
+    // 1. Check Circuit Breaker status
-        c
+    let is_in_backoff = {
+        let lock = state.health.read().await;
+        if let Some(health) = lock.get(module_name) {
+            if let Some(until) = health.backoff_until {
+                Instant::now() < until
             } else {
-        error!("Failed to acquire read lock for configuration.");
+                false
-        return "{\"text\":\"error: config lock failed\"}".to_string();
+            }
-    };
+        } else {
-
+            false
-    let result = match module_name {
-        "net" | "network" => crate::modules::network::NetworkModule.run(&config, state, args),
-        "cpu" => crate::modules::cpu::CpuModule.run(&config, state, args),
-        "mem" | "memory" => crate::modules::memory::MemoryModule.run(&config, state, args),
-        "disk" => crate::modules::disk::DiskModule.run(&config, state, args),
-        "pool" | "btrfs" => crate::modules::btrfs::BtrfsModule.run(&config, state, args),
-        "vol" => crate::modules::audio::AudioModule.run(
-            &config,
-            state,
-            &["sink", args.first().unwrap_or(&"show")],
-        ),
-        "mic" => crate::modules::audio::AudioModule.run(
-            &config,
-            state,
-            &["source", args.first().unwrap_or(&"show")],
-        ),
-        "gpu" => crate::modules::gpu::GpuModule.run(&config, state, args),
-        "sys" => crate::modules::sys::SysModule.run(&config, state, args),
-        "bt" | "bluetooth" => crate::modules::bt::BtModule.run(&config, state, args),
-        "power" => crate::modules::power::PowerModule.run(&config, state, args),
-        "game" => crate::modules::game::GameModule.run(&config, state, args),
-        _ => {
-            warn!("Received request for unknown module: '{}'", module_name);
-            Err(anyhow::anyhow!("Unknown module: {}", module_name))
         }
     };
 
+    if is_in_backoff {
+        return format!(
+            "{{\"text\":\"\u{200B}Cooling down ({})\u{200B}\",\"class\":\"error\"}}",
+            module_name
+        );
+    }
+
+    let config = config_lock.read().await;
+
+    let result = match module_name {
+        "net" | "network" => {
+            crate::modules::network::NetworkModule
+                .run(&config, state, args)
+                .await
+        }
+        "cpu" => {
+            crate::modules::cpu::CpuModule
+                .run(&config, state, args)
+                .await
+        }
+        "mem" | "memory" => {
+            crate::modules::memory::MemoryModule
+                .run(&config, state, args)
+                .await
+        }
+        "disk" => {
+            crate::modules::disk::DiskModule
+                .run(&config, state, args)
+                .await
+        }
+        "pool" | "btrfs" => {
+            crate::modules::btrfs::BtrfsModule
+                .run(&config, state, args)
+                .await
+        }
+        "vol" => {
+            crate::modules::audio::AudioModule
+                .run(&config, state, &["sink", args.first().unwrap_or(&"show")])
+                .await
+        }
+        "mic" => {
+            crate::modules::audio::AudioModule
+                .run(&config, state, &["source", args.first().unwrap_or(&"show")])
+                .await
+        }
+        "gpu" => {
+            crate::modules::gpu::GpuModule
+                .run(&config, state, args)
+                .await
+        }
+        "sys" => {
+            crate::modules::sys::SysModule
+                .run(&config, state, args)
+                .await
+        }
+        "bt" | "bluetooth" => crate::modules::bt::BtModule.run(&config, state, args).await,
+        "power" => {
+            crate::modules::power::PowerModule
+                .run(&config, state, args)
+                .await
+        }
+        "game" => {
+            crate::modules::game::GameModule
+                .run(&config, state, args)
+                .await
+        }
+        _ => {
+            warn!("Received request for unknown module: '{}'", module_name);
+            Err(FluxoError::Ipc(format!("Unknown module: {}", module_name)))
+        }
+    };
+
+    // 2. Update Health based on result
+    {
+        let mut lock = state.health.write().await;
+        let health = lock.entry(module_name.to_string()).or_default();
+        match &result {
+            Ok(_) => {
+                health.consecutive_failures = 0;
+                health.backoff_until = None;
+            }
+            Err(e) => {
+                health.consecutive_failures += 1;
+                health.last_failure = Some(Instant::now());
+                if health.consecutive_failures >= 3 {
+                    // Backoff for 30 seconds after 3 failures
+                    health.backoff_until = Some(Instant::now() + Duration::from_secs(30));
+                    warn!(module = module_name, error = %e, "Module entered backoff state due to repeated failures");
+                }
+            }
+        }
+    }
+
     match result {
         Ok(output) => serde_json::to_string(&output).unwrap_or_else(|_| "{}".to_string()),
         Err(e) => {
-            error!(module = module_name, error = %e, "Module execution failed");
+            let error_msg = e.to_string();
+            error!(module = module_name, error = %error_msg, "Module execution failed");
             let err_out = crate::output::WaybarOutput {
                 text: "\u{200B}Error\u{200B}".to_string(),
-                tooltip: Some(e.to_string()),
+                tooltip: Some(error_msg),
                 class: Some("error".to_string()),
                 percentage: None,
             };

src/error.rs

@@ -0,0 +1,54 @@
+use thiserror::Error;
+
+#[derive(Error, Debug)]
+#[allow(dead_code)]
+pub enum FluxoError {
+    #[error("Configuration error: {0}")]
+    Config(String),
+
+    #[error("Module error ({module}): {message}")]
+    Module {
+        module: &'static str,
+        message: String,
+    },
+
+    #[error("Daemon IPC error: {0}")]
+    Ipc(String),
+
+    #[error("External system error: {0}")]
+    System(String),
+
+    #[error("Bluetooth error: {0}")]
+    Bluetooth(String),
+
+    #[error("Network error: {0}")]
+    Network(String),
+
+    #[error("Hardware error: {0}")]
+    Hardware(String),
+
+    #[error("IO error: {0}")]
+    Io(#[from] std::io::Error),
+
+    #[error("Serialization error: {0}")]
+    Serialization(#[from] serde_json::Error),
+
+    #[error("Other error: {0}")]
+    Other(#[from] anyhow::Error),
+}
+
+impl FluxoError {
+    pub fn is_transient(&self) -> bool {
+        matches!(
+            self,
+            Self::Io(_)
+                | Self::System(_)
+                | Self::Bluetooth(_)
+                | Self::Network(_)
+                | Self::Hardware(_)
+                | Self::Module { .. }
+        )
+    }
+}
+
+pub type Result<T> = std::result::Result<T, FluxoError>;

src/main.rs

@@ -1,5 +1,6 @@
 mod config;
 mod daemon;
+mod error;
 mod ipc;
 mod modules;
 mod output;
@@ -88,7 +89,12 @@ fn main() {
     match &cli.command {
         Commands::Daemon { config } => {
             info!("Starting Fluxo daemon...");
-            if let Err(e) = daemon::run_daemon(config.clone()) {
+            let rt = tokio::runtime::Builder::new_multi_thread()
+                .enable_all()
+                .build()
+                .unwrap();
+
+            if let Err(e) = rt.block_on(daemon::run_daemon(config.clone())) {
                 error!("Daemon failed: {}", e);
                 process::exit(1);
             }

src/modules/audio.rs

@@ -1,16 +1,15 @@
 use crate::config::Config;
+use crate::error::{FluxoError, Result};
 use crate::modules::WaybarModule;
 use crate::output::WaybarOutput;
-use crate::state::SharedState;
+use crate::state::{AppReceivers, AudioState};
 use crate::utils::{TokenValue, format_template};
-use anyhow::{Result, anyhow};
 use libpulse_binding::callbacks::ListResult;
 use libpulse_binding::context::subscribe::{Facility, InterestMaskSet};
 use libpulse_binding::context::{Context, FlagSet as ContextFlag};
 use libpulse_binding::mainloop::threaded::Mainloop as ThreadedMainloop;
 use libpulse_binding::volume::Volume;
-use std::process::Command;
+use tokio::sync::watch;
-use std::sync::Arc;
 use tracing::error;
 
 pub struct AudioDaemon;
@@ -20,8 +19,8 @@ impl AudioDaemon {
         Self
     }
 
-    pub fn start(&self, state: SharedState) {
+    pub fn start(&self, state_tx: &watch::Sender<AudioState>) {
-        let state_arc = Arc::clone(&state);
+        let state_tx = state_tx.clone();
 
         std::thread::spawn(move || {
             let mut mainloop =
@@ -57,7 +56,7 @@ impl AudioDaemon {
             }
 
             // Initial fetch
-            let _ = fetch_audio_data_sync(&mut context, &state_arc);
+            let _ = fetch_audio_data_sync(&mut context, &state_tx);
 
             // Subscribe to events
             let interest =
@@ -82,7 +81,7 @@ impl AudioDaemon {
                 let _ = rx.recv_timeout(Duration::from_secs(2));
                 {
                     mainloop.lock();
-                    let _ = fetch_audio_data_sync(&mut context, &state_arc);
+                    let _ = fetch_audio_data_sync(&mut context, &state_tx);
                     mainloop.unlock();
                 }
             }
@@ -92,68 +91,72 @@ impl AudioDaemon {
 
 use std::time::Duration;
 
-fn fetch_audio_data_sync(context: &mut Context, state: &SharedState) -> Result<()> {
+fn fetch_audio_data_sync(
-    let state_inner = Arc::clone(state);
+    context: &mut Context,
-
+    state_tx: &watch::Sender<AudioState>,
+) -> Result<()> {
     // We fetch all sinks and sources, and also server info to know defaults.
     // The order doesn't strictly matter as long as we update correctly.
 
-    let st_server = Arc::clone(&state_inner);
+    let tx_server = state_tx.clone();
     context.introspect().get_server_info(move |info| {
-        let mut lock = st_server.write().unwrap();
+        let mut current = tx_server.borrow().clone();
-        lock.audio.sink.name = info
+        current.sink.name = info
             .default_sink_name
             .as_ref()
             .map(|s| s.to_string())
             .unwrap_or_default();
-        lock.audio.source.name = info
+        current.source.name = info
             .default_source_name
             .as_ref()
             .map(|s| s.to_string())
             .unwrap_or_default();
+        let _ = tx_server.send(current);
     });
 
-    let st_sink = Arc::clone(&state_inner);
+    let tx_sink = state_tx.clone();
     context.introspect().get_sink_info_list(move |res| {
         if let ListResult::Item(item) = res {
-            let mut lock = st_sink.write().unwrap();
+            let mut current = tx_sink.borrow().clone();
             // If this matches our default sink name, or if we don't have details for any yet
             let is_default = item
                 .name
                 .as_ref()
-                .map(|s| s.as_ref() == lock.audio.sink.name)
+                .map(|s| s.as_ref() == current.sink.name)
                 .unwrap_or(false);
             if is_default {
-                lock.audio.sink.description = item
+                current.sink.description = item
                     .description
                     .as_ref()
                     .map(|s| s.to_string())
                     .unwrap_or_default();
-                lock.audio.sink.volume =
+                current.sink.volume =
                     ((item.volume.avg().0 as f64 / Volume::NORMAL.0 as f64) * 100.0).round() as u8;
-                lock.audio.sink.muted = item.mute;
+                current.sink.muted = item.mute;
+                let _ = tx_sink.send(current);
             }
         }
     });
 
-    let st_source = Arc::clone(&state_inner);
+    let tx_source = state_tx.clone();
     context.introspect().get_source_info_list(move |res| {
         if let ListResult::Item(item) = res {
-            let mut lock = st_source.write().unwrap();
+            let mut current = tx_source.borrow().clone();
             let is_default = item
                 .name
                 .as_ref()
-                .map(|s| s.as_ref() == lock.audio.source.name)
+                .map(|s| s.as_ref() == current.source.name)
                 .unwrap_or(false);
             if is_default {
-                lock.audio.source.description = item
+                current.source.description = item
                     .description
                     .as_ref()
                     .map(|s| s.to_string())
                     .unwrap_or_default();
-                lock.audio.source.volume =
+                current.source.volume =
                     ((item.volume.avg().0 as f64 / Volume::NORMAL.0 as f64) * 100.0).round() as u8;
-                lock.audio.source.muted = item.mute;
+                current.source.muted = item.mute;
+                let _ = tx_source.send(current);
             }
         }
     });
@@ -164,32 +167,37 @@ fn fetch_audio_data_sync(context: &mut Context, state: &SharedState) -> Result<(
 pub struct AudioModule;
 
 impl WaybarModule for AudioModule {
-    fn run(&self, config: &Config, state: &SharedState, args: &[&str]) -> Result<WaybarOutput> {
+    async fn run(
+        &self,
+        config: &Config,
+        state: &AppReceivers,
+        args: &[&str],
+    ) -> Result<WaybarOutput> {
         let target_type = args.first().unwrap_or(&"sink");
         let action = args.get(1).unwrap_or(&"show");
 
         match *action {
             "cycle" => {
-                self.cycle_device(target_type)?;
+                self.cycle_device(target_type).await?;
                 Ok(WaybarOutput::default())
             }
-            "show" => self.get_status(config, state, target_type),
+            "show" => self.get_status(config, state, target_type).await,
-            other => Err(anyhow!("Unknown audio action: '{}'", other)),
+            other => Err(FluxoError::Module {
+                module: "audio",
+                message: format!("Unknown audio action: '{}'", other),
+            }),
         }
     }
 }
 
 impl AudioModule {
-    fn get_status(
+    async fn get_status(
         &self,
         config: &Config,
-        state: &SharedState,
+        state: &AppReceivers,
         target_type: &str,
     ) -> Result<WaybarOutput> {
-        let audio_state = {
+        let audio_state = state.audio.borrow().clone();
-            let lock = state.read().unwrap();
-            lock.audio.clone()
-        };
 
         let (name, description, volume, muted) = if target_type == "sink" {
             (
@@ -272,7 +280,7 @@ impl AudioModule {
         })
     }
 
-    fn cycle_device(&self, target_type: &str) -> Result<()> {
+    async fn cycle_device(&self, target_type: &str) -> Result<()> {
         // Keep using pactl for cycling for now as it's a rare action
         // but we could also implement it natively later.
         let set_cmd = if target_type == "sink" {
@@ -290,9 +298,10 @@ impl AudioModule {
         } else {
             "sources"
         };
-        let output = Command::new("pactl")
+        let output = tokio::process::Command::new("pactl")
             .args(["list", "short", list_cmd])
-            .output()?;
+            .output()
+            .await?;
         let stdout = String::from_utf8_lossy(&output.stdout);
 
         let devices: Vec<String> = stdout
@@ -316,7 +325,10 @@ impl AudioModule {
             return Ok(());
         }
 
-        let info_output = Command::new("pactl").args(["info"]).output()?;
+        let info_output = tokio::process::Command::new("pactl")
+            .args(["info"])
+            .output()
+            .await?;
         let info_stdout = String::from_utf8_lossy(&info_output.stdout);
         let search_key = if target_type == "sink" {
             "Default Sink:"
@@ -334,7 +346,10 @@ impl AudioModule {
         let next_index = (current_index + 1) % devices.len();
         let next_dev = &devices[next_index];
 
-        Command::new("pactl").args([set_cmd, next_dev]).status()?;
+        tokio::process::Command::new("pactl")
+            .args([set_cmd, next_dev])
+            .status()
+            .await?;
         Ok(())
     }
 }

src/modules/bt.rs

@@ -1,608 +0,0 @@
-use crate::config::Config;
-use crate::modules::WaybarModule;
-use crate::output::WaybarOutput;
-use crate::state::{BtState, SharedState};
-use crate::utils::{TokenValue, format_template};
-use anyhow::{Context, Result};
-use futures::StreamExt;
-use std::collections::HashMap;
-use std::process::Command;
-use std::sync::{Arc, LazyLock, Mutex};
-use std::time::{Duration, Instant};
-use tokio::sync::mpsc;
-use tracing::{debug, error, info, warn};
-
-// Maestro imports
-#[allow(unused_imports)]
-use maestro::protocol::codec::Codec;
-#[allow(unused_imports)]
-use maestro::pwrpc::client::{Client, ClientHandle};
-#[allow(unused_imports)]
-use maestro::service::MaestroService;
-#[allow(unused_imports)]
-use maestro::service::settings::{self, Setting, SettingValue};
-
-static RUNTIME: LazyLock<tokio::runtime::Runtime> = LazyLock::new(|| {
-    tokio::runtime::Builder::new_current_thread()
-        .enable_all()
-        .build()
-        .expect("Failed to create BT tokio runtime")
-});
-
-#[derive(Clone, Default)]
-struct BudsStatus {
-    left_battery: Option<u8>,
-    right_battery: Option<u8>,
-    case_battery: Option<u8>,
-    anc_state: String,
-    #[allow(dead_code)]
-    last_update: Option<Instant>,
-    error: Option<String>,
-}
-
-enum BudsCommand {
-    SetAnc(String),
-}
-
-enum ManagerCommand {
-    EnsureTask(String),
-    SendCommand(String, BudsCommand),
-}
-
-struct MaestroManager {
-    statuses: Arc<Mutex<HashMap<String, BudsStatus>>>,
-    management_tx: mpsc::UnboundedSender<ManagerCommand>,
-}
-
-impl MaestroManager {
-    fn new() -> Self {
-        let (tx, mut rx) = mpsc::unbounded_channel::<ManagerCommand>();
-        let statuses = Arc::new(Mutex::new(HashMap::new()));
-        let statuses_clone = Arc::clone(&statuses);
-
-        // Start dedicated BT management thread
-        std::thread::spawn(move || {
-            let rt = tokio::runtime::Builder::new_current_thread()
-                .enable_all()
-                .build()
-                .unwrap();
-
-            let local = tokio::task::LocalSet::new();
-
-            local.block_on(&rt, async move {
-                let mut command_txs: HashMap<String, mpsc::Sender<BudsCommand>> = HashMap::new();
-
-                loop {
-                    tokio::select! {
-                        Some(cmd) = rx.recv() => {
-                            match cmd {
-                                ManagerCommand::EnsureTask(mac) => {
-                                    if !command_txs.contains_key(&mac) {
-                                        let (tx, buds_rx) = mpsc::channel::<BudsCommand>(10);
-                                        command_txs.insert(mac.clone(), tx);
-
-                                        let mac_clone = mac.clone();
-                                        let st_clone = Arc::clone(&statuses_clone);
-
-                                        tokio::task::spawn_local(async move {
-                                            if let Err(e) = buds_task(&mac_clone, st_clone, buds_rx).await {
-                                                error!("Buds task for {} failed: {}", mac_clone, e);
-                                            }
-                                        });
-                                    }
-                                }
-                                ManagerCommand::SendCommand(mac, buds_cmd) => {
-                                    if let Some(tx) = command_txs.get(&mac) {
-                                        let _ = tx.try_send(buds_cmd);
-                                    }
-                                }
-                            }
-                        }
-                        _ = tokio::time::sleep(Duration::from_millis(100)) => {
-                            // Cleanup dropped tasks if needed
-                        }
-                    }
-                }
-            });
-        });
-
-        Self {
-            statuses,
-            management_tx: tx,
-        }
-    }
-
-    fn get_status(&self, mac: &str) -> BudsStatus {
-        let statuses = self.statuses.lock().unwrap();
-        statuses.get(mac).cloned().unwrap_or_default()
-    }
-
-    fn ensure_task(&self, mac: &str) {
-        let _ = self
-            .management_tx
-            .send(ManagerCommand::EnsureTask(mac.to_string()));
-    }
-
-    fn send_command(&self, mac: &str, cmd: BudsCommand) -> Result<()> {
-        self.ensure_task(mac);
-        let _ = self
-            .management_tx
-            .send(ManagerCommand::SendCommand(mac.to_string(), cmd));
-        Ok(())
-    }
-}
-
-async fn buds_task(
-    mac: &str,
-    statuses: Arc<Mutex<HashMap<String, BudsStatus>>>,
-    mut rx: mpsc::Receiver<BudsCommand>,
-) -> Result<()> {
-    info!("Starting native Maestro connection task for {}", mac);
-
-    loop {
-        let addr: bluer::Address = mac.parse().context("Failed to parse MAC address")?;
-        let session = bluer::Session::new()
-            .await
-            .context("Failed to create bluer session")?;
-        let adapter = session
-            .default_adapter()
-            .await
-            .context("Failed to get default adapter")?;
-        let device = adapter
-            .device(addr)
-            .context("Failed to get device handle")?;
-
-        if !device.is_connected().await.unwrap_or(false) {
-            debug!("Device {} not connected to BT, stopping maestro task", mac);
-            break;
-        }
-
-        // Connect to Maestro RFCOMM service
-        // We try channel 1 then 2, which covers most Pixel Buds variants.
-        let mut stream = None;
-        for channel in [1, 2] {
-            let socket = match bluer::rfcomm::Socket::new() {
-                Ok(s) => s,
-                Err(e) => {
-                    error!("Failed to create RFCOMM socket: {}", e);
-                    return Err(e.into());
-                }
-            };
-            let target = bluer::rfcomm::SocketAddr::new(addr, channel);
-            debug!(
-                "Trying to connect RFCOMM to {} on channel {}...",
-                mac, channel
-            );
-            match socket.connect(target).await {
-                Ok(s) => {
-                    stream = Some(s);
-                    break;
-                }
-                Err(e) => {
-                    debug!("Failed to connect to channel {}: {}", channel, e);
-                }
-            }
-        }
-
-        let stream = match stream {
-            Some(s) => s,
-            None => {
-                warn!(
-                    "Failed to connect RFCOMM to {} on any common channel. Retrying in 15s...",
-                    mac
-                );
-                tokio::time::sleep(Duration::from_secs(15)).await;
-                continue;
-            }
-        };
-
-        info!("Connected Maestro RFCOMM to {} on channel", mac);
-
-        // Initialize Maestro communication stack
-        let codec = Codec::new();
-        let stream = codec.wrap(stream);
-        let mut client = Client::new(stream);
-        let handle = client.handle();
-
-        // Resolve Maestro channel (typically 1 or 2)
-        let channel = match maestro::protocol::utils::resolve_channel(&mut client).await {
-            Ok(c) => c,
-            Err(e) => {
-                error!("Failed to resolve Maestro channel for {}: {}", mac, e);
-                continue;
-            }
-        };
-
-        // Run client in background to handle RPC packets
-        tokio::spawn(async move {
-            if let Err(e) = client.run().await {
-                error!("Maestro client loop error: {}", e);
-            }
-        });
-
-        let mut service = MaestroService::new(handle, channel);
-
-        // Query initial ANC state
-        if let Ok(val) = service
-            .read_setting_var(settings::SettingId::CurrentAncrState)
-            .await
-            && let SettingValue::CurrentAncrState(anc_state) = val
-        {
-            let mut status = MAESTRO.get_status(mac);
-            status.anc_state = anc_state_to_string(&anc_state);
-            statuses.lock().unwrap().insert(mac.to_string(), status);
-        }
-
-        // Subscribe to real-time status updates (battery, ANC, wear)
-        let mut call = match service.subscribe_to_runtime_info() {
-            Ok(c) => c,
-            Err(e) => {
-                error!("Failed to subscribe to runtime info for {}: {}", mac, e);
-                continue;
-            }
-        };
-
-        let mut runtime_info = call.stream();
-
-        debug!("Subscribed to runtime info for {}", mac);
-
-        loop {
-            tokio::select! {
-                cmd = rx.recv() => {
-                    match cmd {
-                        Some(BudsCommand::SetAnc(mode)) => {
-                            debug!("Setting ANC mode to {} for {}", mode, mac);
-                            let state = mode_to_anc_state(&mode);
-                            let val = SettingValue::CurrentAncrState(state);
-                            if let Err(e) = service.write_setting(val).await {
-                                error!("Failed to write ANC setting for {}: {}", mac, e);
-                            }
-                        }
-                        None => return Ok(()),
-                    }
-                }
-                Some(Ok(info)) = runtime_info.next() => {
-                    let mut status = MAESTRO.get_status(mac);
-                    status.last_update = Some(Instant::now());
-
-                    if let Some(bat) = info.battery_info {
-                        status.left_battery = bat.left.map(|b| b.level as u8);
-                        status.right_battery = bat.right.map(|b| b.level as u8);
-                        status.case_battery = bat.case.map(|b| b.level as u8);
-                    }
-
-                    statuses.lock().unwrap().insert(mac.to_string(), status);
-                }
-                _ = tokio::time::sleep(Duration::from_secs(30)) => {
-                    // Check if still connected to BT
-                    if !device.is_connected().await.unwrap_or(false) {
-                        break;
-                    }
-                }
-            }
-        }
-
-        if !device.is_connected().await.unwrap_or(false) {
-            break;
-        }
-    }
-
-    Ok(())
-}
-
-fn mode_to_anc_state(mode: &str) -> settings::AncState {
-    match mode {
-        "active" => settings::AncState::Active,
-        "aware" => settings::AncState::Aware,
-        "off" => settings::AncState::Off,
-        _ => settings::AncState::Off,
-    }
-}
-
-fn anc_state_to_string(state: &settings::AncState) -> String {
-    match state {
-        settings::AncState::Active => "active".to_string(),
-        settings::AncState::Aware => "aware".to_string(),
-        settings::AncState::Off => "off".to_string(),
-        _ => "unknown".to_string(),
-    }
-}
-
-static MAESTRO: LazyLock<MaestroManager> = LazyLock::new(MaestroManager::new);
-
-pub struct BtDaemon {
-    session: Option<bluer::Session>,
-}
-
-impl BtDaemon {
-    pub fn new() -> Self {
-        Self { session: None }
-    }
-
-    pub fn poll(&mut self, state: SharedState) {
-        if let Err(e) = RUNTIME.block_on(self.poll_async(state)) {
-            error!("BT daemon error: {}", e);
-        }
-    }
-
-    async fn poll_async(&mut self, state: SharedState) -> Result<()> {
-        if self.session.is_none() {
-            self.session = Some(bluer::Session::new().await?);
-        }
-        let session = self.session.as_ref().unwrap();
-        let adapter = session.default_adapter().await?;
-        let adapter_powered = adapter.is_powered().await.unwrap_or(false);
-
-        let mut bt_state = BtState {
-            adapter_powered,
-            ..Default::default()
-        };
-
-        if adapter_powered {
-            let devices = adapter.device_addresses().await?;
-            for addr in devices {
-                let device = adapter.device(addr)?;
-                if device.is_connected().await.unwrap_or(false) {
-                    let uuids = device.uuids().await?.unwrap_or_default();
-                    // Audio sink UUID (0x110b)
-                    let audio_sink_uuid =
-                        bluer::Uuid::from_u128(0x0000110b_0000_1000_8000_00805f9b34fb);
-                    if uuids.contains(&audio_sink_uuid) {
-                        bt_state.connected = true;
-                        bt_state.device_address = addr.to_string();
-                        bt_state.device_alias =
-                            device.alias().await.unwrap_or_else(|_| addr.to_string());
-                        bt_state.battery_percentage =
-                            device.battery_percentage().await.unwrap_or(None);
-
-                        // Plugin detection
-                        for p in PLUGINS.iter() {
-                            if p.can_handle(&bt_state.device_alias, &bt_state.device_address) {
-                                match p.get_data(&bt_state.device_address) {
-                                    Ok(data) => {
-                                        bt_state.plugin_data = data
-                                            .into_iter()
-                                            .map(|(k, v)| {
-                                                let val_str = match v {
-                                                    TokenValue::String(s) => s,
-                                                    TokenValue::Int(i) => i.to_string(),
-                                                    TokenValue::Float(f) => format!("{:.1}", f),
-                                                };
-                                                (k, val_str)
-                                            })
-                                            .collect();
-                                    }
-                                    Err(e) => {
-                                        warn!("Plugin {} failed for {}: {}", p.name(), addr, e);
-                                        bt_state
-                                            .plugin_data
-                                            .push(("plugin_error".to_string(), e.to_string()));
-                                    }
-                                }
-                                break;
-                            }
-                        }
-                        break;
-                    }
-                }
-            }
-        }
-
-        if let Ok(mut lock) = state.write() {
-            lock.bluetooth = bt_state;
-        }
-
-        Ok(())
-    }
-}
-
-pub trait BtPlugin: Send + Sync {
-    fn name(&self) -> &str;
-    fn can_handle(&self, alias: &str, mac: &str) -> bool;
-    fn get_data(&self, mac: &str) -> Result<Vec<(String, TokenValue)>>;
-    fn get_modes(&self, _mac: &str) -> Result<Vec<String>> {
-        Ok(vec![])
-    }
-    fn set_mode(&self, _mode: &str, _mac: &str) -> Result<()> {
-        Ok(())
-    }
-    fn cycle_mode(&self, _mac: &str) -> Result<()> {
-        Ok(())
-    }
-}
-
-pub struct PixelBudsPlugin;
-
-impl BtPlugin for PixelBudsPlugin {
-    fn name(&self) -> &str {
-        "Pixel Buds Pro"
-    }
-
-    fn can_handle(&self, alias: &str, _mac: &str) -> bool {
-        alias.contains("Pixel Buds Pro")
-    }
-
-    fn get_data(&self, mac: &str) -> Result<Vec<(String, TokenValue)>> {
-        MAESTRO.ensure_task(mac);
-        let status = MAESTRO.get_status(mac);
-
-        if let Some(err) = status.error {
-            return Err(anyhow::anyhow!(err));
-        }
-
-        let left_display = status
-            .left_battery
-            .map(|b| format!("{}%", b))
-            .unwrap_or_else(|| "---".to_string());
-        let right_display = status
-            .right_battery
-            .map(|b| format!("{}%", b))
-            .unwrap_or_else(|| "---".to_string());
-
-        let (anc_icon, class) = match status.anc_state.as_str() {
-            "active" => ("ANC", "anc-active"),
-            "aware" => ("Aware", "anc-aware"),
-            "off" => ("Off", "anc-off"),
-            _ => ("?", "anc-unknown"),
-        };
-
-        Ok(vec![
-            ("left".to_string(), TokenValue::String(left_display)),
-            ("right".to_string(), TokenValue::String(right_display)),
-            ("anc".to_string(), TokenValue::String(anc_icon.to_string())),
-            (
-                "plugin_class".to_string(),
-                TokenValue::String(class.to_string()),
-            ),
-        ])
-    }
-
-    fn get_modes(&self, _mac: &str) -> Result<Vec<String>> {
-        Ok(vec![
-            "active".to_string(),
-            "aware".to_string(),
-            "off".to_string(),
-        ])
-    }
-
-    fn set_mode(&self, mode: &str, mac: &str) -> Result<()> {
-        MAESTRO.send_command(mac, BudsCommand::SetAnc(mode.to_string()))
-    }
-
-    fn cycle_mode(&self, mac: &str) -> Result<()> {
-        let status = MAESTRO.get_status(mac);
-        let next_mode = match status.anc_state.as_str() {
-            "active" => "aware",
-            "aware" => "off",
-            _ => "active",
-        };
-        self.set_mode(next_mode, mac)
-    }
-}
-
-static PLUGINS: LazyLock<Vec<Box<dyn BtPlugin>>> =
-    LazyLock::new(|| vec![Box::new(PixelBudsPlugin)]);
-
-pub struct BtModule;
-
-impl WaybarModule for BtModule {
-    fn run(&self, config: &Config, state: &SharedState, args: &[&str]) -> Result<WaybarOutput> {
-        let action = args.first().unwrap_or(&"show");
-        let bt_state = {
-            let lock = state.read().unwrap();
-            lock.bluetooth.clone()
-        };
-
-        match *action {
-            "disconnect" if bt_state.connected => {
-                let _ = Command::new("bluetoothctl")
-                    .args(["disconnect", &bt_state.device_address])
-                    .output();
-                return Ok(WaybarOutput::default());
-            }
-            "cycle_mode" if bt_state.connected => {
-                let plugin = PLUGINS
-                    .iter()
-                    .find(|p| p.can_handle(&bt_state.device_alias, &bt_state.device_address));
-                if let Some(p) = plugin {
-                    p.cycle_mode(&bt_state.device_address)?;
-                }
-                return Ok(WaybarOutput::default());
-            }
-            "get_modes" if bt_state.connected => {
-                let plugin = PLUGINS
-                    .iter()
-                    .find(|p| p.can_handle(&bt_state.device_alias, &bt_state.device_address));
-                let modes = if let Some(p) = plugin {
-                    p.get_modes(&bt_state.device_address)?
-                } else {
-                    vec![]
-                };
-                return Ok(WaybarOutput {
-                    text: modes.join("\n"),
-                    ..Default::default()
-                });
-            }
-            "set_mode" if bt_state.connected => {
-                if let Some(mode) = args.get(1) {
-                    let plugin = PLUGINS
-                        .iter()
-                        .find(|p| p.can_handle(&bt_state.device_alias, &bt_state.device_address));
-                    if let Some(p) = plugin {
-                        p.set_mode(mode, &bt_state.device_address)?;
-                    }
-                }
-                return Ok(WaybarOutput::default());
-            }
-            "show" => {}
-            _ => {}
-        }
-
-        if !bt_state.adapter_powered {
-            return Ok(WaybarOutput {
-                text: config.bt.format_disabled.clone(),
-                tooltip: Some("Bluetooth Disabled".to_string()),
-                class: Some("disabled".to_string()),
-                percentage: None,
-            });
-        }
-
-        if bt_state.connected {
-            let mut tokens: Vec<(String, TokenValue)> = vec![
-                (
-                    "alias".to_string(),
-                    TokenValue::String(bt_state.device_alias.clone()),
-                ),
-                (
-                    "mac".to_string(),
-                    TokenValue::String(bt_state.device_address.clone()),
-                ),
-            ];
-
-            let mut class = vec!["connected".to_string()];
-            let mut has_plugin = false;
-
-            for (k, v) in &bt_state.plugin_data {
-                if k == "plugin_class" {
-                    class.push(v.clone());
-                    has_plugin = true;
-                } else if k == "plugin_error" {
-                    class.push("plugin-error".to_string());
-                } else {
-                    tokens.push((k.clone(), TokenValue::String(v.clone())));
-                }
-            }
-
-            let format = if has_plugin {
-                &config.bt.format_plugin
-            } else {
-                &config.bt.format_connected
-            };
-
-            let text = format_template(format, &tokens);
-            let tooltip = format!(
-                "{} | MAC: {}\nBattery: {}",
-                bt_state.device_alias,
-                bt_state.device_address,
-                bt_state
-                    .battery_percentage
-                    .map(|b| format!("{}%", b))
-                    .unwrap_or_else(|| "N/A".to_string())
-            );
-
-            Ok(WaybarOutput {
-                text,
-                tooltip: Some(tooltip),
-                class: Some(class.join(" ")),
-                percentage: bt_state.battery_percentage,
-            })
-        } else {
-            Ok(WaybarOutput {
-                text: config.bt.format_disconnected.clone(),
-                tooltip: Some("Bluetooth On (Disconnected)".to_string()),
-                class: Some("disconnected".to_string()),
-                percentage: None,
-            })
-        }
-    }
-}

src/modules/bt/buds.rs

@@ -0,0 +1,141 @@
+use crate::config::Config;
+use crate::error::{FluxoError, Result as FluxoResult};
+use crate::modules::bt::maestro::BudsCommand;
+use crate::state::AppReceivers;
+use crate::utils::TokenValue;
+use futures::future::BoxFuture;
+
+pub trait BtPlugin: Send + Sync {
+    fn name(&self) -> &str;
+    fn can_handle(&self, alias: &str, mac: &str) -> bool;
+    fn get_data(
+        &self,
+        config: &Config,
+        state: &AppReceivers,
+        mac: &str,
+    ) -> BoxFuture<'static, FluxoResult<Vec<(String, TokenValue)>>>;
+    fn get_modes(
+        &self,
+        mac: &str,
+        state: &AppReceivers,
+    ) -> BoxFuture<'static, FluxoResult<Vec<String>>>;
+    fn set_mode(
+        &self,
+        mode: &str,
+        mac: &str,
+        state: &AppReceivers,
+    ) -> BoxFuture<'static, FluxoResult<()>>;
+    fn cycle_mode(&self, mac: &str, state: &AppReceivers) -> BoxFuture<'static, FluxoResult<()>>;
+}
+
+pub struct PixelBudsPlugin;
+
+impl BtPlugin for PixelBudsPlugin {
+    fn name(&self) -> &str {
+        "Pixel Buds Pro"
+    }
+
+    fn can_handle(&self, alias: &str, _mac: &str) -> bool {
+        alias.contains("Pixel Buds Pro")
+    }
+
+    fn get_data(
+        &self,
+        _config: &Config,
+        state: &AppReceivers,
+        mac: &str,
+    ) -> BoxFuture<'static, FluxoResult<Vec<(String, TokenValue)>>> {
+        let mac = mac.to_string();
+        let state = state.clone();
+        Box::pin(async move {
+            let maestro = crate::modules::bt::maestro::get_maestro(&state);
+            maestro.ensure_task(&mac);
+            let status = maestro.get_status(&mac);
+
+            if let Some(err) = status.error {
+                return Err(FluxoError::Module {
+                    module: "bt.buds",
+                    message: err,
+                });
+            }
+
+            let left_display = status
+                .left_battery
+                .map(|b| format!("{}%", b))
+                .unwrap_or_else(|| "---".to_string());
+            let right_display = status
+                .right_battery
+                .map(|b| format!("{}%", b))
+                .unwrap_or_else(|| "---".to_string());
+
+            let (anc_icon, class) = match status.anc_state.as_str() {
+                "active" => ("ANC", "anc-active"),
+                "aware" => ("Aware", "anc-aware"),
+                "off" => ("Off", "anc-off"),
+                _ => ("?", "anc-unknown"),
+            };
+
+            Ok(vec![
+                ("left".to_string(), TokenValue::String(left_display)),
+                ("right".to_string(), TokenValue::String(right_display)),
+                ("anc".to_string(), TokenValue::String(anc_icon.to_string())),
+                (
+                    "plugin_class".to_string(),
+                    TokenValue::String(class.to_string()),
+                ),
+            ])
+        })
+    }
+
+    fn get_modes(
+        &self,
+        _mac: &str,
+        _state: &AppReceivers,
+    ) -> BoxFuture<'static, FluxoResult<Vec<String>>> {
+        Box::pin(async move {
+            Ok(vec![
+                "active".to_string(),
+                "aware".to_string(),
+                "off".to_string(),
+            ])
+        })
+    }
+
+    fn set_mode(
+        &self,
+        mode: &str,
+        mac: &str,
+        state: &AppReceivers,
+    ) -> BoxFuture<'static, FluxoResult<()>> {
+        let mode = mode.to_string();
+        let mac = mac.to_string();
+        let state = state.clone();
+        Box::pin(async move {
+            crate::modules::bt::maestro::get_maestro(&state)
+                .send_command(&mac, BudsCommand::SetAnc(mode))
+                .map_err(|e: anyhow::Error| FluxoError::Module {
+                    module: "bt.buds",
+                    message: e.to_string(),
+                })
+        })
+    }
+
+    fn cycle_mode(&self, mac: &str, state: &AppReceivers) -> BoxFuture<'static, FluxoResult<()>> {
+        let mac = mac.to_string();
+        let state = state.clone();
+        Box::pin(async move {
+            let status = crate::modules::bt::maestro::get_maestro(&state).get_status(&mac);
+            let next_mode = match status.anc_state.as_str() {
+                "active" => "aware",
+                "aware" => "off",
+                _ => "active",
+            };
+            crate::modules::bt::maestro::get_maestro(&state)
+                .send_command(&mac, BudsCommand::SetAnc(next_mode.to_string()))
+                .map_err(|e: anyhow::Error| FluxoError::Module {
+                    module: "bt.buds",
+                    message: e.to_string(),
+                })
+        })
+    }
+}

src/modules/bt/maestro.rs

@@ -0,0 +1,362 @@
+use crate::state::AppReceivers;
+use anyhow::{Context, Result};
+use futures::StreamExt;
+use std::collections::HashMap;
+use std::sync::{Arc, Mutex, OnceLock};
+use std::time::{Duration, Instant};
+use tokio::sync::mpsc;
+use tracing::{debug, error, info, warn};
+
+// Maestro imports
+use maestro::protocol::codec::Codec;
+use maestro::pwrpc::client::Client;
+use maestro::service::MaestroService;
+use maestro::service::settings::{self, SettingValue};
+
+#[derive(Clone, Default)]
+pub struct BudsStatus {
+    pub left_battery: Option<u8>,
+    pub right_battery: Option<u8>,
+    pub case_battery: Option<u8>,
+    pub anc_state: String,
+    #[allow(dead_code)]
+    pub last_update: Option<Instant>,
+    pub error: Option<String>,
+}
+
+pub enum BudsCommand {
+    SetAnc(String),
+}
+
+pub enum ManagerCommand {
+    EnsureTask(String),
+    SendCommand(String, BudsCommand),
+}
+
+pub struct MaestroManager {
+    statuses: Arc<Mutex<HashMap<String, BudsStatus>>>,
+    management_tx: mpsc::UnboundedSender<ManagerCommand>,
+}
+
+impl MaestroManager {
+    pub fn new(state: AppReceivers) -> Self {
+        let (tx, mut rx) = mpsc::unbounded_channel::<ManagerCommand>();
+        let statuses = Arc::new(Mutex::new(HashMap::new()));
+        let statuses_clone = Arc::clone(&statuses);
+        let state_clone = state.clone();
+
+        // Start dedicated BT management thread
+        std::thread::spawn(move || {
+            let rt = tokio::runtime::Builder::new_current_thread()
+                .enable_all()
+                .build()
+                .unwrap();
+
+            let local = tokio::task::LocalSet::new();
+
+            local.block_on(&rt, async move {
+                let mut command_txs: HashMap<String, mpsc::Sender<BudsCommand>> = HashMap::new();
+
+                loop {
+                    tokio::select! {
+                        Some(cmd) = rx.recv() => {
+                            match cmd {
+                                ManagerCommand::EnsureTask(mac) => {
+                                    if !command_txs.contains_key(&mac) {
+                                        let (tx, buds_rx) = mpsc::channel::<BudsCommand>(10);
+                                        command_txs.insert(mac.clone(), tx);
+
+                                        let mac_clone = mac.clone();
+                                        let st_clone = Arc::clone(&statuses_clone);
+                                        let state_inner = state_clone.clone();
+
+                                        tokio::task::spawn_local(async move {
+                                            if let Err(e) = buds_task(&mac_clone, st_clone, buds_rx, state_inner).await {
+                                                error!("Buds task for {} failed: {}", mac_clone, e);
+                                            }
+                                        });
+                                    }
+                                }
+                                ManagerCommand::SendCommand(mac, buds_cmd) => {
+                                    if let Some(tx) = command_txs.get(&mac) {
+                                        let _ = tx.try_send(buds_cmd);
+                                    }
+                                }
+                            }
+                        }
+                        _ = tokio::time::sleep(Duration::from_millis(100)) => {
+                            // Cleanup dropped tasks if needed
+                        }
+                    }
+                }
+            });
+        });
+
+        Self {
+            statuses,
+            management_tx: tx,
+        }
+    }
+
+    pub fn get_status(&self, mac: &str) -> BudsStatus {
+        let statuses = self.statuses.lock().unwrap();
+        statuses.get(mac).cloned().unwrap_or_default()
+    }
+
+    pub fn ensure_task(&self, mac: &str) {
+        let _ = self
+            .management_tx
+            .send(ManagerCommand::EnsureTask(mac.to_string()));
+    }
+
+    pub fn send_command(&self, mac: &str, cmd: BudsCommand) -> Result<()> {
+        self.ensure_task(mac);
+        let _ = self
+            .management_tx
+            .send(ManagerCommand::SendCommand(mac.to_string(), cmd));
+        Ok(())
+    }
+}
+
+async fn buds_task(
+    mac: &str,
+    statuses: Arc<Mutex<HashMap<String, BudsStatus>>>,
+    mut rx: mpsc::Receiver<BudsCommand>,
+    state: AppReceivers,
+) -> Result<()> {
+    info!("Starting native Maestro connection task for {}", mac);
+
+    loop {
+        let addr: bluer::Address = match mac.parse() {
+            Ok(a) => a,
+            Err(e) => {
+                error!("Failed to parse MAC address {}: {}", mac, e);
+                return Err(e.into());
+            }
+        };
+        let session = bluer::Session::new()
+            .await
+            .context("Failed to create bluer session")?;
+        let adapter = session
+            .default_adapter()
+            .await
+            .context("Failed to get default adapter")?;
+        let device = adapter
+            .device(addr)
+            .context("Failed to get device handle")?;
+
+        if !device.is_connected().await.unwrap_or(false) {
+            debug!("Device {} not connected to BT, stopping maestro task", mac);
+            break;
+        }
+
+        // Connect to Maestro RFCOMM service
+        let mut stream = None;
+        for channel in [1, 2] {
+            let socket = match bluer::rfcomm::Socket::new() {
+                Ok(s) => s,
+                Err(e) => {
+                    error!("Failed to create RFCOMM socket: {}", e);
+                    return Err(e.into());
+                }
+            };
+            let target = bluer::rfcomm::SocketAddr::new(addr, channel);
+            debug!(
+                "Trying to connect RFCOMM to {} on channel {}...",
+                mac, channel
+            );
+            match socket.connect(target).await {
+                Ok(s) => {
+                    stream = Some(s);
+                    break;
+                }
+                Err(e) => {
+                    debug!("Failed to connect to channel {}: {}", channel, e);
+                }
+            }
+        }
+
+        let stream = match stream {
+            Some(s) => s,
+            None => {
+                warn!(
+                    "Failed to connect RFCOMM to {} on any common channel. Retrying in 15s...",
+                    mac
+                );
+                tokio::time::sleep(Duration::from_secs(15)).await;
+                continue;
+            }
+        };
+
+        info!("Connected Maestro RFCOMM to {} on channel", mac);
+
+        // Initialize Maestro communication stack
+        let codec = Codec::new();
+        let stream = codec.wrap(stream);
+        let mut client = Client::new(stream);
+        let handle = client.handle();
+
+        // Resolve Maestro channel
+        let channel = match maestro::protocol::utils::resolve_channel(&mut client).await {
+            Ok(c) => c,
+            Err(e) => {
+                error!("Failed to resolve Maestro channel for {}: {}", mac, e);
+                continue;
+            }
+        };
+
+        tokio::spawn(async move {
+            if let Err(e) = client.run().await {
+                error!("Maestro client loop error: {}", e);
+            }
+        });
+
+        let mut service = MaestroService::new(handle, channel);
+
+        // Update health
+        {
+            let mut lock = state.health.write().await;
+            let health = lock.entry("bt.buds".to_string()).or_default();
+            health.consecutive_failures = 0;
+            health.backoff_until = None;
+        }
+
+        // Query initial ANC state
+        if let Ok(val) = service
+            .read_setting_var(settings::SettingId::CurrentAncrState)
+            .await
+            && let SettingValue::CurrentAncrState(anc_state) = val
+        {
+            let mut lock = statuses.lock().unwrap();
+            let status = lock.entry(mac.to_string()).or_default();
+            status.anc_state = anc_state_to_string(&anc_state);
+            status.last_update = Some(Instant::now());
+        }
+
+        let mut runtime_info_call = match service.subscribe_to_runtime_info() {
+            Ok(c) => c,
+            Err(e) => {
+                error!("Failed to subscribe to runtime info for {}: {}", mac, e);
+                continue;
+            }
+        };
+
+        let mut runtime_info = runtime_info_call.stream();
+
+        let mut settings_changes_call = match service.subscribe_to_settings_changes() {
+            Ok(s) => s,
+            Err(e) => {
+                error!("Failed to subscribe to settings changes for {}: {}", mac, e);
+                continue;
+            }
+        };
+
+        let mut settings_changes = settings_changes_call.stream();
+
+        debug!("Subscribed to status and settings for {}", mac);
+
+        loop {
+            tokio::select! {
+                cmd = rx.recv() => {
+                    match cmd {
+                        Some(BudsCommand::SetAnc(mode)) => {
+                            debug!("Setting ANC mode to {} for {}", mode, mac);
+                            let state = mode_to_anc_state(&mode);
+                            let val = SettingValue::CurrentAncrState(state);
+
+                            {
+                                let mut lock = statuses.lock().unwrap();
+                                let status = lock.entry(mac.to_string()).or_default();
+                                status.anc_state = mode.clone();
+                                status.last_update = Some(Instant::now());
+                            }
+
+                            if let Err(e) = service.write_setting(val).await {
+                                error!("Failed to write ANC setting for {}: {}", mac, e);
+                            }
+                        }
+                        None => return Ok(()),
+                    }
+                }
+                Some(res) = runtime_info.next() => {
+                    match res {
+                        Ok(info) => {
+                            let mut lock = statuses.lock().unwrap();
+                            let status = lock.entry(mac.to_string()).or_default();
+                            status.last_update = Some(Instant::now());
+
+                            if let Some(bat) = info.battery_info {
+                                status.left_battery = bat.left.map(|b| b.level as u8);
+                                status.right_battery = bat.right.map(|b| b.level as u8);
+                                status.case_battery = bat.case.map(|b| b.level as u8);
+                            }
+                        }
+                        Err(e) => {
+                            warn!("Runtime info stream error for {}: {}", mac, e);
+                            break;
+                        }
+                    }
+                }
+                Some(res) = settings_changes.next() => {
+                    if let Ok(change) = res {
+                        use maestro::protocol::types::settings_rsp::ValueOneof as RspOneof;
+                        use maestro::protocol::types::setting_value::ValueOneof as ValOneof;
+
+                        if let Some(RspOneof::Value(setting_val)) = change.value_oneof
+                            && let Some(ValOneof::CurrentAncrState(anc_state_raw)) = setting_val.value_oneof {
+                                let mut lock = statuses.lock().unwrap();
+                                let status = lock.entry(mac.to_string()).or_default();
+
+                                let anc_state = match anc_state_raw {
+                                    1 => settings::AncState::Off,
+                                    2 => settings::AncState::Active,
+                                    3 => settings::AncState::Aware,
+                                    4 => settings::AncState::Adaptive,
+                                    _ => settings::AncState::Unknown(anc_state_raw),
+                                };
+
+                                status.anc_state = anc_state_to_string(&anc_state);
+                                status.last_update = Some(Instant::now());
+                                debug!(mode = %status.anc_state, "Caught physical ANC toggle");
+                        }
+                    }
+                }
+                _ = tokio::time::sleep(Duration::from_secs(30)) => {
+                    if !device.is_connected().await.unwrap_or(false) {
+                        break;
+                    }
+                }
+            }
+        }
+
+        if !device.is_connected().await.unwrap_or(false) {
+            break;
+        }
+    }
+
+    Ok(())
+}
+
+fn mode_to_anc_state(mode: &str) -> settings::AncState {
+    match mode {
+        "active" => settings::AncState::Active,
+        "aware" => settings::AncState::Aware,
+        "off" => settings::AncState::Off,
+        _ => settings::AncState::Off,
+    }
+}
+
+pub fn anc_state_to_string(state: &settings::AncState) -> String {
+    match state {
+        settings::AncState::Active => "active".to_string(),
+        settings::AncState::Aware => "aware".to_string(),
+        settings::AncState::Off => "off".to_string(),
+        _ => "unknown".to_string(),
+    }
+}
+
+static MAESTRO: OnceLock<MaestroManager> = OnceLock::new();
+
+pub fn get_maestro(state: &AppReceivers) -> &MaestroManager {
+    MAESTRO.get_or_init(|| MaestroManager::new(state.clone()))
+}

src/modules/bt/mod.rs

@@ -0,0 +1,239 @@
+pub mod buds;
+pub mod maestro;
+
+use crate::config::Config;
+use crate::error::Result as FluxoResult;
+use crate::modules::WaybarModule;
+use crate::output::WaybarOutput;
+use crate::state::{AppReceivers, BtState};
+use crate::utils::{TokenValue, format_template};
+use anyhow::Result;
+use std::sync::LazyLock;
+use tokio::sync::watch;
+use tracing::{error, warn};
+
+use self::buds::{BtPlugin, PixelBudsPlugin};
+
+pub struct BtDaemon {
+    session: Option<bluer::Session>,
+}
+
+impl BtDaemon {
+    pub fn new() -> Self {
+        Self { session: None }
+    }
+
+    pub async fn poll(
+        &mut self,
+        tx: &watch::Sender<BtState>,
+        state: &AppReceivers,
+        config: &Config,
+    ) {
+        if let Err(e) = self.poll_async(tx, state, config).await {
+            error!("BT daemon error: {}", e);
+        }
+    }
+
+    async fn poll_async(
+        &mut self,
+        tx: &watch::Sender<BtState>,
+        state: &AppReceivers,
+        config: &Config,
+    ) -> Result<()> {
+        if self.session.is_none() {
+            self.session = Some(bluer::Session::new().await?);
+        }
+        let session = self.session.as_ref().unwrap();
+        let adapter = session.default_adapter().await?;
+        let adapter_powered = adapter.is_powered().await.unwrap_or(false);
+
+        let mut bt_state = BtState {
+            adapter_powered,
+            ..Default::default()
+        };
+
+        if adapter_powered {
+            let devices = adapter.device_addresses().await?;
+            for addr in devices {
+                let device = adapter.device(addr)?;
+                if device.is_connected().await.unwrap_or(false) {
+                    let uuids = device.uuids().await?.unwrap_or_default();
+                    let audio_sink_uuid =
+                        bluer::Uuid::from_u128(0x0000110b_0000_1000_8000_00805f9b34fb);
+                    if uuids.contains(&audio_sink_uuid) {
+                        bt_state.connected = true;
+                        bt_state.device_address = addr.to_string();
+                        bt_state.device_alias =
+                            device.alias().await.unwrap_or_else(|_| addr.to_string());
+                        bt_state.battery_percentage =
+                            device.battery_percentage().await.unwrap_or(None);
+
+                        for p in PLUGINS.iter() {
+                            if p.can_handle(&bt_state.device_alias, &bt_state.device_address) {
+                                match p.get_data(config, state, &bt_state.device_address).await {
+                                    Ok(data) => {
+                                        bt_state.plugin_data = data
+                                            .into_iter()
+                                            .map(|(k, v)| {
+                                                let val_str = match v {
+                                                    TokenValue::String(s) => s,
+                                                    TokenValue::Int(i) => i.to_string(),
+                                                    TokenValue::Float(f) => format!("{:.1}", f),
+                                                };
+                                                (k, val_str)
+                                            })
+                                            .collect();
+                                    }
+                                    Err(e) => {
+                                        warn!("Plugin {} failed for {}: {}", p.name(), addr, e);
+                                        bt_state
+                                            .plugin_data
+                                            .push(("plugin_error".to_string(), e.to_string()));
+                                    }
+                                }
+                                break;
+                            }
+                        }
+                        break;
+                    }
+                }
+            }
+        }
+
+        let _ = tx.send(bt_state);
+
+        Ok(())
+    }
+}
+
+static PLUGINS: LazyLock<Vec<Box<dyn BtPlugin>>> =
+    LazyLock::new(|| vec![Box::new(PixelBudsPlugin)]);
+
+pub struct BtModule;
+
+impl WaybarModule for BtModule {
+    async fn run(
+        &self,
+        config: &Config,
+        state: &AppReceivers,
+        args: &[&str],
+    ) -> FluxoResult<WaybarOutput> {
+        let action = args.first().cloned().unwrap_or("show").to_string();
+        let args = args.iter().map(|s| s.to_string()).collect::<Vec<_>>();
+
+        let bt_state = state.bluetooth.borrow().clone();
+
+        match action.as_str() {
+            "disconnect" if bt_state.connected => {
+                let _ = tokio::process::Command::new("bluetoothctl")
+                    .args(["disconnect", &bt_state.device_address])
+                    .output()
+                    .await;
+                return Ok(WaybarOutput::default());
+            }
+            "cycle_mode" if bt_state.connected => {
+                let plugin = PLUGINS
+                    .iter()
+                    .find(|p| p.can_handle(&bt_state.device_alias, &bt_state.device_address));
+                if let Some(p) = plugin {
+                    p.cycle_mode(&bt_state.device_address, state).await?;
+                }
+                return Ok(WaybarOutput::default());
+            }
+            "get_modes" if bt_state.connected => {
+                let plugin = PLUGINS
+                    .iter()
+                    .find(|p| p.can_handle(&bt_state.device_alias, &bt_state.device_address));
+                let modes = if let Some(p) = plugin {
+                    p.get_modes(&bt_state.device_address, state).await?
+                } else {
+                    vec![]
+                };
+                return Ok(WaybarOutput {
+                    text: modes.join("\n"),
+                    ..Default::default()
+                });
+            }
+            "set_mode" if bt_state.connected => {
+                if let Some(mode) = args.get(1) {
+                    let plugin = PLUGINS
+                        .iter()
+                        .find(|p| p.can_handle(&bt_state.device_alias, &bt_state.device_address));
+                    if let Some(p) = plugin {
+                        p.set_mode(mode, &bt_state.device_address, state).await?;
+                    }
+                }
+                return Ok(WaybarOutput::default());
+            }
+            "show" => {}
+            _ => {}
+        }
+
+        if !bt_state.adapter_powered {
+            return Ok(WaybarOutput {
+                text: config.bt.format_disabled.clone(),
+                tooltip: Some("Bluetooth Disabled".to_string()),
+                class: Some("disabled".to_string()),
+                percentage: None,
+            });
+        }
+
+        if bt_state.connected {
+            let mut tokens: Vec<(String, TokenValue)> = vec![
+                (
+                    "alias".to_string(),
+                    TokenValue::String(bt_state.device_alias.clone()),
+                ),
+                (
+                    "mac".to_string(),
+                    TokenValue::String(bt_state.device_address.clone()),
+                ),
+            ];
+
+            let mut class = vec!["connected".to_string()];
+            let mut has_plugin = false;
+
+            for (k, v) in &bt_state.plugin_data {
+                if k == "plugin_class" {
+                    class.push(v.clone());
+                    has_plugin = true;
+                } else if k == "plugin_error" {
+                    class.push("plugin-error".to_string());
+                } else {
+                    tokens.push((k.clone(), TokenValue::String(v.clone())));
+                }
+            }
+
+            let format = if has_plugin {
+                &config.bt.format_plugin
+            } else {
+                &config.bt.format_connected
+            };
+
+            let text = format_template(format, &tokens);
+            let tooltip = format!(
+                "{} | MAC: {}\nBattery: {}",
+                bt_state.device_alias,
+                bt_state.device_address,
+                bt_state
+                    .battery_percentage
+                    .map(|b| format!("{}%", b))
+                    .unwrap_or_else(|| "N/A".to_string())
+            );
+
+            Ok(WaybarOutput {
+                text,
+                tooltip: Some(tooltip),
+                class: Some(class.join(" ")),
+                percentage: bt_state.battery_percentage,
+            })
+        } else {
+            Ok(WaybarOutput {
+                text: config.bt.format_disconnected.clone(),
+                tooltip: Some("Bluetooth On (Disconnected)".to_string()),
+                class: Some("disconnected".to_string()),
+                percentage: None,
+            })
+        }
+    }
+}

src/modules/btrfs.rs

@@ -1,18 +1,22 @@
 use crate::config::Config;
+use crate::error::Result;
 use crate::modules::WaybarModule;
 use crate::output::WaybarOutput;
-use crate::state::SharedState;
+use crate::state::AppReceivers;
 use crate::utils::{TokenValue, format_template};
-use anyhow::Result;
 
 pub struct BtrfsModule;
 
 impl WaybarModule for BtrfsModule {
-    fn run(&self, config: &Config, state: &SharedState, _args: &[&str]) -> Result<WaybarOutput> {
+    async fn run(
-        let disks = if let Ok(s) = state.read() {
+        &self,
-            s.disks.clone()
+        config: &Config,
-        } else {
+        state: &AppReceivers,
-            return Err(anyhow::anyhow!("Failed to read state"));
+        _args: &[&str],
+    ) -> Result<WaybarOutput> {
+        let disks = {
+            let s = state.disks.borrow();
+            s.clone()
         };
 
         let mut total_used: f64 = 0.0;

src/modules/cpu.rs

@@ -1,24 +1,22 @@
 use crate::config::Config;
+use crate::error::Result;
 use crate::modules::WaybarModule;
 use crate::output::WaybarOutput;
-use crate::state::SharedState;
+use crate::state::AppReceivers;
 use crate::utils::{TokenValue, format_template};
-use anyhow::Result;
 
 pub struct CpuModule;
 
 impl WaybarModule for CpuModule {
-    fn run(&self, config: &Config, state: &SharedState, _args: &[&str]) -> Result<WaybarOutput> {
+    async fn run(
+        &self,
+        config: &Config,
+        state: &AppReceivers,
+        _args: &[&str],
+    ) -> Result<WaybarOutput> {
         let (usage, temp, model) = {
-            if let Ok(state_lock) = state.read() {
+            let s = state.cpu.borrow();
-                (
+            (s.usage, s.temp, s.model.clone())
-                    state_lock.cpu.usage,
-                    state_lock.cpu.temp,
-                    state_lock.cpu.model.clone(),
-                )
-            } else {
-                (0.0, 0.0, String::from("Unknown"))
-            }
         };
 
         let text = format_template(
@@ -51,8 +49,8 @@ mod tests {
     use super::*;
     use crate::state::{AppState, CpuState, mock_state};
 
-    #[test]
+    #[tokio::test]
-    fn test_cpu_normal() {
+    async fn test_cpu_normal() {
         let state = mock_state(AppState {
             cpu: CpuState {
                 usage: 25.0,
@@ -62,7 +60,7 @@ mod tests {
             ..Default::default()
         });
         let config = Config::default();
-        let output = CpuModule.run(&config, &state, &[]).unwrap();
+        let output = CpuModule.run(&config, &state.receivers, &[]).await.unwrap();
         assert!(output.text.contains("25.0"));
         assert!(output.text.contains("45.0"));
         assert_eq!(output.class.as_deref(), Some("normal"));
@@ -70,8 +68,8 @@ mod tests {
         assert_eq!(output.tooltip.as_deref(), Some("Test CPU"));
     }
 
-    #[test]
+    #[tokio::test]
-    fn test_cpu_high() {
+    async fn test_cpu_high() {
         let state = mock_state(AppState {
             cpu: CpuState {
                 usage: 80.0,
@@ -81,12 +79,12 @@ mod tests {
             ..Default::default()
         });
         let config = Config::default();
-        let output = CpuModule.run(&config, &state, &[]).unwrap();
+        let output = CpuModule.run(&config, &state.receivers, &[]).await.unwrap();
         assert_eq!(output.class.as_deref(), Some("high"));
     }
 
-    #[test]
+    #[tokio::test]
-    fn test_cpu_max() {
+    async fn test_cpu_max() {
         let state = mock_state(AppState {
             cpu: CpuState {
                 usage: 99.0,
@@ -96,7 +94,7 @@ mod tests {
             ..Default::default()
         });
         let config = Config::default();
-        let output = CpuModule.run(&config, &state, &[]).unwrap();
+        let output = CpuModule.run(&config, &state.receivers, &[]).await.unwrap();
         assert_eq!(output.class.as_deref(), Some("max"));
     }
 }

src/modules/disk.rs

@@ -1,21 +1,22 @@
 use crate::config::Config;
+use crate::error::{FluxoError, Result};
 use crate::modules::WaybarModule;
 use crate::output::WaybarOutput;
-use crate::state::SharedState;
+use crate::state::AppReceivers;
 use crate::utils::{TokenValue, format_template};
-use anyhow::Result;
 
 pub struct DiskModule;
 
 impl WaybarModule for DiskModule {
-    fn run(&self, config: &Config, state: &SharedState, args: &[&str]) -> Result<WaybarOutput> {
+    async fn run(
+        &self,
+        config: &Config,
+        state: &AppReceivers,
+        args: &[&str],
+    ) -> Result<WaybarOutput> {
         let mountpoint = args.first().unwrap_or(&"/");
 
-        let disks = if let Ok(s) = state.read() {
+        let disks = state.disks.borrow().clone();
-            s.disks.clone()
-        } else {
-            return Err(anyhow::anyhow!("Failed to read state"));
-        };
 
         for disk in &disks {
             if disk.mount_point == *mountpoint {
@@ -62,16 +63,19 @@ impl WaybarModule for DiskModule {
             }
         }
 
-        Err(anyhow::anyhow!("Mountpoint {} not found", mountpoint))
+        Err(FluxoError::Module {
+            module: "disk",
+            message: format!("Mountpoint {} not found", mountpoint),
+        })
     }
 }
 
 #[cfg(test)]
 mod tests {
     use super::*;
-    use crate::state::{AppState, DiskInfo, mock_state};
+    use crate::state::{AppState, DiskInfo, MockState, mock_state};
 
-    fn state_with_disk(mount: &str, total: u64, available: u64) -> crate::state::SharedState {
+    fn state_with_disk(mount: &str, total: u64, available: u64) -> MockState {
         mock_state(AppState {
             disks: vec![DiskInfo {
                 mount_point: mount.to_string(),
@@ -83,30 +87,38 @@ mod tests {
         })
     }
 
-    #[test]
+    #[tokio::test]
-    fn test_disk_found() {
+    async fn test_disk_found() {
         let gb = 1024 * 1024 * 1024;
         let state = state_with_disk("/", 100 * gb, 60 * gb);
         let config = Config::default();
-        let output = DiskModule.run(&config, &state, &["/"]).unwrap();
+        let output = DiskModule
+            .run(&config, &state.receivers, &["/"])
+            .await
+            .unwrap();
         assert_eq!(output.class.as_deref(), Some("normal"));
         assert_eq!(output.percentage, Some(40)); // 40% used
     }
 
-    #[test]
+    #[tokio::test]
-    fn test_disk_high() {
+    async fn test_disk_high() {
         let gb = 1024 * 1024 * 1024;
         let state = state_with_disk("/", 100 * gb, 15 * gb);
         let config = Config::default();
-        let output = DiskModule.run(&config, &state, &["/"]).unwrap();
+        let output = DiskModule
+            .run(&config, &state.receivers, &["/"])
+            .await
+            .unwrap();
         assert_eq!(output.class.as_deref(), Some("high")); // 85% used
     }
 
-    #[test]
+    #[tokio::test]
-    fn test_disk_not_found() {
+    async fn test_disk_not_found() {
         let state = mock_state(AppState::default());
         let config = Config::default();
-        let result = DiskModule.run(&config, &state, &["/nonexistent"]);
+        let result = DiskModule
+            .run(&config, &state.receivers, &["/nonexistent"])
+            .await;
         assert!(result.is_err());
     }
 }

src/modules/game.rs

@@ -1,17 +1,24 @@
 use crate::config::Config;
+use crate::error::Result;
 use crate::modules::WaybarModule;
 use crate::output::WaybarOutput;
-use crate::state::SharedState;
+use crate::state::AppReceivers;
-use anyhow::{Result, anyhow};
+use anyhow::anyhow;
 use std::env;
-use std::io::{Read, Write};
+use tokio::io::{AsyncReadExt, AsyncWriteExt};
-use std::os::unix::net::UnixStream;
+use tokio::net::UnixStream;
 
 pub struct GameModule;
 
 impl WaybarModule for GameModule {
-    fn run(&self, config: &Config, _state: &SharedState, _args: &[&str]) -> Result<WaybarOutput> {
+    async fn run(
+        &self,
+        config: &Config,
+        _state: &AppReceivers,
+        _args: &[&str],
+    ) -> Result<WaybarOutput> {
         let is_gamemode = hyprland_ipc("j/getoption animations:enabled")
+            .await
             .map(|stdout| stdout.contains("\"int\": 0"))
             .unwrap_or(false);
 
@@ -33,16 +40,16 @@ impl WaybarModule for GameModule {
     }
 }
 
-fn hyprland_ipc(cmd: &str) -> Result<String> {
+async fn hyprland_ipc(cmd: &str) -> Result<String> {
     let signature = env::var("HYPRLAND_INSTANCE_SIGNATURE")
         .map_err(|_| anyhow!("HYPRLAND_INSTANCE_SIGNATURE not set"))?;
     let path = format!("/tmp/hypr/{}/.socket.sock", signature);
 
-    let mut stream = UnixStream::connect(path)?;
+    let mut stream = UnixStream::connect(path).await?;
-    stream.write_all(cmd.as_bytes())?;
+    stream.write_all(cmd.as_bytes()).await?;
 
     let mut response = String::new();
-    stream.read_to_string(&mut response)?;
+    stream.read_to_string(&mut response).await?;
 
     Ok(response)
 }

src/modules/gpu.rs

@@ -1,36 +1,30 @@
 use crate::config::Config;
+use crate::error::Result;
 use crate::modules::WaybarModule;
 use crate::output::WaybarOutput;
-use crate::state::SharedState;
+use crate::state::AppReceivers;
 use crate::utils::{TokenValue, format_template};
-use anyhow::Result;
 
 pub struct GpuModule;
 
 impl WaybarModule for GpuModule {
-    fn run(&self, config: &Config, state: &SharedState, _args: &[&str]) -> Result<WaybarOutput> {
+    async fn run(
+        &self,
+        config: &Config,
+        state: &AppReceivers,
+        _args: &[&str],
+    ) -> Result<WaybarOutput> {
         let (active, vendor, usage, vram_used, vram_total, temp, model) = {
-            if let Ok(state_lock) = state.read() {
+            let s = state.gpu.borrow();
             (
-                    state_lock.gpu.active,
+                s.active,
-                    state_lock.gpu.vendor.clone(),
+                s.vendor.clone(),
-                    state_lock.gpu.usage,
+                s.usage,
-                    state_lock.gpu.vram_used,
+                s.vram_used,
-                    state_lock.gpu.vram_total,
+                s.vram_total,
-                    state_lock.gpu.temp,
+                s.temp,
-                    state_lock.gpu.model.clone(),
+                s.model.clone(),
             )
-            } else {
-                (
-                    false,
-                    String::from("Unknown"),
-                    0.0,
-                    0.0,
-                    0.0,
-                    0.0,
-                    String::from("Unknown"),
-                )
-            }
         };
 
         if !active {

src/modules/hardware.rs

@@ -1,5 +1,6 @@
-use crate::state::{DiskInfo, SharedState};
+use crate::state::{CpuState, DiskInfo, GpuState, MemoryState, SysState};
 use sysinfo::{Components, Disks, System};
+use tokio::sync::watch;
 
 pub struct HardwareDaemon {
     sys: System,
@@ -24,7 +25,12 @@ impl HardwareDaemon {
         }
     }
 
-    pub fn poll_fast(&mut self, state: SharedState) {
+    pub async fn poll_fast(
+        &mut self,
+        cpu_tx: &watch::Sender<CpuState>,
+        mem_tx: &watch::Sender<MemoryState>,
+        sys_tx: &watch::Sender<SysState>,
+    ) {
         self.sys.refresh_cpu_usage();
         self.sys.refresh_memory();
         self.components.refresh(true);
@@ -70,29 +76,37 @@ impl HardwareDaemon {
             }
         }
 
-        if let Ok(mut state_lock) = state.write() {
+        let mut cpu = cpu_tx.borrow().clone();
-            state_lock.cpu.usage = cpu_usage as f64;
+        cpu.usage = cpu_usage as f64;
-            state_lock.cpu.temp = cpu_temp;
+        cpu.temp = cpu_temp;
-            state_lock.cpu.model = cpu_model;
+        cpu.model = cpu_model;
+        let _ = cpu_tx.send(cpu);
 
-            state_lock.memory.total_gb = total_mem;
+        let mut mem = mem_tx.borrow().clone();
-            state_lock.memory.used_gb = used_mem;
+        mem.total_gb = total_mem;
+        mem.used_gb = used_mem;
+        let _ = mem_tx.send(mem);
 
-            state_lock.sys.load_1 = load_avg.one;
+        let mut sys = sys_tx.borrow().clone();
-            state_lock.sys.load_5 = load_avg.five;
+        sys.load_1 = load_avg.one;
-            state_lock.sys.load_15 = load_avg.fifteen;
+        sys.load_5 = load_avg.five;
-            state_lock.sys.uptime = uptime;
+        sys.load_15 = load_avg.fifteen;
-            state_lock.sys.process_count = process_count;
+        sys.uptime = uptime;
-        }
+        sys.process_count = process_count;
+        let _ = sys_tx.send(sys);
     }
 
-    pub fn poll_slow(&mut self, state: SharedState) {
+    pub async fn poll_slow(
+        &mut self,
+        gpu_tx: &watch::Sender<GpuState>,
+        disks_tx: &watch::Sender<Vec<DiskInfo>>,
+    ) {
         // 1. Gather GPU data outside of lock
         let mut gpu_state = crate::state::GpuState::default();
         self.gpu_poll_counter = (self.gpu_poll_counter + 1) % 5;
         let should_poll_gpu = self.gpu_poll_counter == 0;
         if should_poll_gpu {
-            self.poll_gpu(&mut gpu_state);
+            self.poll_gpu(&mut gpu_state).await;
         }
 
         // 2. Gather Disk data outside of lock
@@ -100,6 +114,7 @@ impl HardwareDaemon {
         self.disk_poll_counter = (self.disk_poll_counter + 1) % 10;
         if self.disk_poll_counter == 0 {
             disks_data = Some(
+                tokio::task::spawn_blocking(|| {
                     Disks::new_with_refreshed_list()
                         .iter()
                         .map(|d| DiskInfo {
@@ -108,32 +123,34 @@ impl HardwareDaemon {
                             total_bytes: d.total_space(),
                             available_bytes: d.available_space(),
                         })
-                    .collect::<Vec<DiskInfo>>(),
+                        .collect::<Vec<DiskInfo>>()
+                })
+                .await
+                .unwrap_or_default(),
             );
         }
 
         // 3. Apply to state
-        if let Ok(mut state_lock) = state.write() {
         if should_poll_gpu {
-                state_lock.gpu = gpu_state;
+            let _ = gpu_tx.send(gpu_state);
         }
 
         if let Some(d) = disks_data {
-                state_lock.disks = d;
+            let _ = disks_tx.send(d);
-            }
         }
     }
 
-    fn poll_gpu(&mut self, gpu: &mut crate::state::GpuState) {
+    async fn poll_gpu(&mut self, gpu: &mut crate::state::GpuState) {
         gpu.active = false;
 
         if (self.gpu_vendor.as_deref() == Some("NVIDIA") || self.gpu_vendor.is_none())
-            && let Ok(output) = std::process::Command::new("nvidia-smi")
+            && let Ok(output) = tokio::process::Command::new("nvidia-smi")
                 .args([
                     "--query-gpu=utilization.gpu,memory.used,memory.total,temperature.gpu,name",
                     "--format=csv,noheader,nounits",
                 ])
                 .output()
+                .await
             && output.status.success()
         {
             let stdout = String::from_utf8_lossy(&output.stdout);

src/modules/memory.rs

@@ -1,20 +1,22 @@
 use crate::config::Config;
+use crate::error::Result;
 use crate::modules::WaybarModule;
 use crate::output::WaybarOutput;
-use crate::state::SharedState;
+use crate::state::AppReceivers;
 use crate::utils::{TokenValue, format_template};
-use anyhow::Result;
 
 pub struct MemoryModule;
 
 impl WaybarModule for MemoryModule {
-    fn run(&self, config: &Config, state: &SharedState, _args: &[&str]) -> Result<WaybarOutput> {
+    async fn run(
+        &self,
+        config: &Config,
+        state: &AppReceivers,
+        _args: &[&str],
+    ) -> Result<WaybarOutput> {
         let (used_gb, total_gb) = {
-            if let Ok(state_lock) = state.read() {
+            let s = state.memory.borrow();
-                (state_lock.memory.used_gb, state_lock.memory.total_gb)
+            (s.used_gb, s.total_gb)
-            } else {
-                (0.0, 0.0)
-            }
         };
 
         let ratio = if total_gb > 0.0 {
@@ -53,8 +55,8 @@ mod tests {
     use super::*;
     use crate::state::{AppState, MemoryState, mock_state};
 
-    #[test]
+    #[tokio::test]
-    fn test_memory_normal() {
+    async fn test_memory_normal() {
         let state = mock_state(AppState {
             memory: MemoryState {
                 used_gb: 8.0,
@@ -63,15 +65,18 @@ mod tests {
             ..Default::default()
         });
         let config = Config::default();
-        let output = MemoryModule.run(&config, &state, &[]).unwrap();
+        let output = MemoryModule
+            .run(&config, &state.receivers, &[])
+            .await
+            .unwrap();
         assert!(output.text.contains("8.00"));
         assert!(output.text.contains("32.00"));
         assert_eq!(output.class.as_deref(), Some("normal"));
         assert_eq!(output.percentage, Some(25)); // 8/32 = 25%
     }
 
-    #[test]
+    #[tokio::test]
-    fn test_memory_high() {
+    async fn test_memory_high() {
         let state = mock_state(AppState {
             memory: MemoryState {
                 used_gb: 26.0,
@@ -80,12 +85,15 @@ mod tests {
             ..Default::default()
         });
         let config = Config::default();
-        let output = MemoryModule.run(&config, &state, &[]).unwrap();
+        let output = MemoryModule
+            .run(&config, &state.receivers, &[])
+            .await
+            .unwrap();
         assert_eq!(output.class.as_deref(), Some("high")); // 81%
     }
 
-    #[test]
+    #[tokio::test]
-    fn test_memory_zero_total() {
+    async fn test_memory_zero_total() {
         let state = mock_state(AppState {
             memory: MemoryState {
                 used_gb: 0.0,
@@ -94,7 +102,10 @@ mod tests {
             ..Default::default()
         });
         let config = Config::default();
-        let output = MemoryModule.run(&config, &state, &[]).unwrap();
+        let output = MemoryModule
+            .run(&config, &state.receivers, &[])
+            .await
+            .unwrap();
         assert_eq!(output.class.as_deref(), Some("normal"));
         assert_eq!(output.percentage, Some(0));
     }

src/modules/mod.rs

@@ -12,10 +12,15 @@ pub mod power;
 pub mod sys;
 
 use crate::config::Config;
+use crate::error::Result as FluxoResult;
 use crate::output::WaybarOutput;
-use crate::state::SharedState;
+use crate::state::AppReceivers;
-use anyhow::Result;
 
 pub trait WaybarModule {
-    fn run(&self, config: &Config, state: &SharedState, args: &[&str]) -> Result<WaybarOutput>;
+    fn run(
+        &self,
+        config: &Config,
+        state: &AppReceivers,
+        args: &[&str],
+    ) -> impl std::future::Future<Output = FluxoResult<WaybarOutput>> + Send;
 }

src/modules/network.rs

@@ -1,12 +1,13 @@
 use crate::config::Config;
+use crate::error::Result;
 use crate::modules::WaybarModule;
 use crate::output::WaybarOutput;
-use crate::state::SharedState;
+use crate::state::{AppReceivers, NetworkState};
 use crate::utils::{TokenValue, format_template};
-use anyhow::Result;
 use nix::ifaddrs::getifaddrs;
 use std::fs;
 use std::time::{SystemTime, UNIX_EPOCH};
+use tokio::sync::watch;
 
 pub struct NetworkModule;
 
@@ -18,6 +19,8 @@ pub struct NetworkDaemon {
     cached_ip: Option<String>,
 }
 
+type PollResult = crate::error::Result<(String, Option<String>, Option<(u64, u64)>)>;
+
 impl NetworkDaemon {
     pub fn new() -> Self {
         Self {
@@ -29,28 +32,63 @@ impl NetworkDaemon {
         }
     }
 
-    pub fn poll(&mut self, state: SharedState) {
+    pub async fn poll(
-        // Re-detect interface on every poll to catch VPNs or route changes immediately
+        &mut self,
-        if let Ok(iface) = get_primary_interface()
+        state_tx: &watch::Sender<NetworkState>,
-            && !iface.is_empty()
+    ) -> crate::error::Result<()> {
-        {
+        let (iface, ip_opt, bytes_opt) = tokio::task::spawn_blocking(|| -> PollResult {
-            // If the interface changed, or we don't have an IP yet, update the IP
+            let iface = get_primary_interface()?;
+            if iface.is_empty() {
+                return Ok((String::new(), None, None));
+            }
+            let ip = get_ip_address(&iface);
+            let bytes = get_bytes(&iface).ok();
+            Ok((iface, ip, bytes))
+        })
+        .await
+        .map_err(|e| crate::error::FluxoError::System(e.to_string()))??;
+
+        if !iface.is_empty() {
             if self.cached_interface.as_ref() != Some(&iface) || self.cached_ip.is_none() {
-                self.cached_ip = get_ip_address(&iface);
+                self.cached_ip = ip_opt;
-                self.cached_interface = Some(iface);
+                self.cached_interface = Some(iface.clone());
             }
         } else {
             self.cached_interface = None;
             self.cached_ip = None;
+            // Provide a default state for "No connection"
+            let mut network = state_tx.borrow().clone();
+            network.interface.clear();
+            network.ip.clear();
+            network.rx_mbps = 0.0;
+            network.tx_mbps = 0.0;
+            let _ = state_tx.send(network);
+            return Err(crate::error::FluxoError::Network(
+                "No primary interface found".into(),
+            ));
         }
 
-        if let Some(ref interface) = self.cached_interface {
+        let interface = if let Some(ref interface) = self.cached_interface {
+            interface.clone()
+        } else {
+            // No interface detected
+            let mut network = state_tx.borrow().clone();
+            network.interface.clear();
+            network.ip.clear();
+            network.rx_mbps = 0.0;
+            network.tx_mbps = 0.0;
+            let _ = state_tx.send(network);
+            return Err(crate::error::FluxoError::Network(
+                "Interface disappeared during poll".into(),
+            ));
+        };
+
         let time_now = SystemTime::now()
             .duration_since(UNIX_EPOCH)
             .unwrap_or_default()
             .as_secs();
 
-            if let Ok((rx_bytes_now, tx_bytes_now)) = get_bytes(interface) {
+        if let Some((rx_bytes_now, tx_bytes_now)) = bytes_opt {
             if self.last_time > 0 && time_now > self.last_time {
                 let time_diff = (time_now - self.last_time) as f64;
                 let rx_mbps = (rx_bytes_now.saturating_sub(self.last_rx_bytes)) as f64
@@ -62,16 +100,18 @@ impl NetworkDaemon {
                     / 1024.0
                     / 1024.0;
 
-                    if let Ok(mut state_lock) = state.write() {
+                let mut network = state_tx.borrow().clone();
-                        state_lock.network.rx_mbps = rx_mbps;
+                network.rx_mbps = rx_mbps;
-                        state_lock.network.tx_mbps = tx_mbps;
+                network.tx_mbps = tx_mbps;
-                        state_lock.network.interface = interface.clone();
+                network.interface = interface.clone();
-                        state_lock.network.ip = self.cached_ip.clone().unwrap_or_default();
+                network.ip = self.cached_ip.clone().unwrap_or_default();
-                    }
+                let _ = state_tx.send(network);
-                } else if let Ok(mut state_lock) = state.write() {
+            } else {
                 // First poll: no speed data yet, but update interface/ip
-                    state_lock.network.interface = interface.clone();
+                let mut network = state_tx.borrow().clone();
-                    state_lock.network.ip = self.cached_ip.clone().unwrap_or_default();
+                network.interface = interface.clone();
+                network.ip = self.cached_ip.clone().unwrap_or_default();
+                let _ = state_tx.send(network);
             }
 
             self.last_time = time_now;
@@ -80,31 +120,26 @@ impl NetworkDaemon {
         } else {
             // Read failed, might be down
             self.cached_interface = None;
+            return Err(crate::error::FluxoError::Network(format!(
+                "Failed to read bytes for {}",
+                interface
+            )));
         }
-        } else if let Ok(mut state_lock) = state.write() {
+
-            // No interface detected
+        Ok(())
-            state_lock.network.interface.clear();
-            state_lock.network.ip.clear();
-        }
     }
 }
 
 impl WaybarModule for NetworkModule {
-    fn run(&self, config: &Config, state: &SharedState, _args: &[&str]) -> Result<WaybarOutput> {
+    async fn run(
-        let (interface, ip, rx_mbps, tx_mbps) = if let Ok(s) = state.read() {
+        &self,
-            (
+        config: &Config,
-                s.network.interface.clone(),
+        state: &AppReceivers,
-                s.network.ip.clone(),
+        _args: &[&str],
-                s.network.rx_mbps,
+    ) -> Result<WaybarOutput> {
-                s.network.tx_mbps,
+        let (interface, ip, rx_mbps, tx_mbps) = {
-            )
+            let s = state.network.borrow();
-        } else {
+            (s.interface.clone(), s.ip.clone(), s.rx_mbps, s.tx_mbps)
-            return Ok(WaybarOutput {
-                text: "No connection".to_string(),
-                tooltip: None,
-                class: None,
-                percentage: None,
-            });
         };
 
         if interface.is_empty() {
@@ -148,24 +183,26 @@ impl WaybarModule for NetworkModule {
 }
 
 fn get_primary_interface() -> Result<String> {
-    let content = fs::read_to_string("/proc/net/route")?;
+    let content = std::fs::read_to_string("/proc/net/route")?;
 
     let mut defaults = Vec::new();
     for line in content.lines().skip(1) {
         let parts: Vec<&str> = line.split_whitespace().collect();
-        if parts.len() >= 7 {
+        if parts.len() >= 8 {
             let iface = parts[0];
             let dest = parts[1];
             let metric = parts[6].parse::<i32>().unwrap_or(0);
+            let mask = u32::from_str_radix(parts[7], 16).unwrap_or(0);
 
             if dest == "00000000" {
-                defaults.push((metric, iface.to_string()));
+                defaults.push((mask, metric, iface.to_string()));
             }
         }
     }
 
-    defaults.sort_by_key(|k| k.0);
+    // Sort by mask descending (longest prefix match first), then by metric ascending
-    if let Some((_, dev)) = defaults.first() {
+    defaults.sort_by(|a, b| b.0.cmp(&a.0).then(a.1.cmp(&b.1)));
+    if let Some((_, _, dev)) = defaults.first() {
         Ok(dev.clone())
     } else {
         Ok(String::new())
@@ -208,16 +245,19 @@ mod tests {
     use super::*;
     use crate::state::{AppState, NetworkState, mock_state};
 
-    #[test]
+    #[tokio::test]
-    fn test_network_no_connection() {
+    async fn test_network_no_connection() {
         let state = mock_state(AppState::default());
         let config = Config::default();
-        let output = NetworkModule.run(&config, &state, &[]).unwrap();
+        let output = NetworkModule
+            .run(&config, &state.receivers, &[])
+            .await
+            .unwrap();
         assert_eq!(output.text, "No connection");
     }
 
-    #[test]
+    #[tokio::test]
-    fn test_network_connected() {
+    async fn test_network_connected() {
         let state = mock_state(AppState {
             network: NetworkState {
                 rx_mbps: 1.5,
@@ -228,15 +268,18 @@ mod tests {
             ..Default::default()
         });
         let config = Config::default();
-        let output = NetworkModule.run(&config, &state, &[]).unwrap();
+        let output = NetworkModule
+            .run(&config, &state.receivers, &[])
+            .await
+            .unwrap();
         assert!(output.text.contains("eth0"));
         assert!(output.text.contains("192.168.1.100"));
         assert!(output.text.contains("1.50"));
         assert_eq!(output.class.as_deref(), Some("eth0"));
     }
 
-    #[test]
+    #[tokio::test]
-    fn test_network_vpn_prefix() {
+    async fn test_network_vpn_prefix() {
         let state = mock_state(AppState {
             network: NetworkState {
                 rx_mbps: 0.0,
@@ -247,12 +290,15 @@ mod tests {
             ..Default::default()
         });
         let config = Config::default();
-        let output = NetworkModule.run(&config, &state, &[]).unwrap();
+        let output = NetworkModule
+            .run(&config, &state.receivers, &[])
+            .await
+            .unwrap();
         assert!(output.text.starts_with("  "));
     }
 
-    #[test]
+    #[tokio::test]
-    fn test_network_no_ip() {
+    async fn test_network_no_ip() {
         let state = mock_state(AppState {
             network: NetworkState {
                 rx_mbps: 0.0,
@@ -263,7 +309,10 @@ mod tests {
             ..Default::default()
         });
         let config = Config::default();
-        let output = NetworkModule.run(&config, &state, &[]).unwrap();
+        let output = NetworkModule
+            .run(&config, &state.receivers, &[])
+            .await
+            .unwrap();
         assert!(output.text.contains("No IP"));
     }
 }

src/modules/power.rs

@@ -1,15 +1,20 @@
 use crate::config::Config;
+use crate::error::Result;
 use crate::modules::WaybarModule;
 use crate::output::WaybarOutput;
-use crate::state::SharedState;
+use crate::state::AppReceivers;
 use crate::utils::{TokenValue, format_template};
-use anyhow::Result;
 use std::fs;
 
 pub struct PowerModule;
 
 impl WaybarModule for PowerModule {
-    fn run(&self, config: &Config, _state: &SharedState, _args: &[&str]) -> Result<WaybarOutput> {
+    async fn run(
+        &self,
+        config: &Config,
+        _state: &AppReceivers,
+        _args: &[&str],
+    ) -> Result<WaybarOutput> {
         let critical_threshold = 15;
         let warning_threshold = 50;
 

src/modules/sys.rs

@@ -1,26 +1,28 @@
 use crate::config::Config;
+use crate::error::Result;
 use crate::modules::WaybarModule;
 use crate::output::WaybarOutput;
-use crate::state::SharedState;
+use crate::state::AppReceivers;
 use crate::utils::{TokenValue, format_template};
-use anyhow::Result;
 
 pub struct SysModule;
 
 impl WaybarModule for SysModule {
-    fn run(&self, config: &Config, state: &SharedState, _args: &[&str]) -> Result<WaybarOutput> {
+    async fn run(
+        &self,
+        config: &Config,
+        state: &AppReceivers,
+        _args: &[&str],
+    ) -> Result<WaybarOutput> {
         let (load1, load5, load15, uptime_secs, process_count) = {
-            if let Ok(state_lock) = state.read() {
+            let state_lock = state.sys.borrow();
             (
-                    state_lock.sys.load_1,
+                state_lock.load_1,
-                    state_lock.sys.load_5,
+                state_lock.load_5,
-                    state_lock.sys.load_15,
+                state_lock.load_15,
-                    state_lock.sys.uptime,
+                state_lock.uptime,
-                    state_lock.sys.process_count,
+                state_lock.process_count,
             )
-            } else {
-                (0.0, 0.0, 0.0, 0, 0)
-            }
         };
 
         let hours = uptime_secs / 3600;

src/state.rs

@@ -1,15 +1,26 @@
-use std::sync::{Arc, RwLock};
+use std::collections::HashMap;
+use std::sync::Arc;
+use tokio::sync::{RwLock, watch};
+use tokio::time::Instant;
 
-#[derive(Default, Clone)]
+#[derive(Clone)]
-pub struct AppState {
+pub struct AppReceivers {
-    pub network: NetworkState,
+    pub network: watch::Receiver<NetworkState>,
-    pub cpu: CpuState,
+    pub cpu: watch::Receiver<CpuState>,
-    pub memory: MemoryState,
+    pub memory: watch::Receiver<MemoryState>,
-    pub sys: SysState,
+    pub sys: watch::Receiver<SysState>,
-    pub gpu: GpuState,
+    pub gpu: watch::Receiver<GpuState>,
-    pub disks: Vec<DiskInfo>,
+    pub disks: watch::Receiver<Vec<DiskInfo>>,
-    pub bluetooth: BtState,
+    pub bluetooth: watch::Receiver<BtState>,
-    pub audio: AudioState,
+    pub audio: watch::Receiver<AudioState>,
+    pub health: Arc<RwLock<HashMap<String, ModuleHealth>>>,
+}
+
+#[derive(Clone, Default)]
+pub struct ModuleHealth {
+    pub consecutive_failures: u32,
+    pub last_failure: Option<Instant>,
+    pub backoff_until: Option<Instant>,
 }
 
 #[derive(Default, Clone)]
@@ -117,9 +128,64 @@ impl Default for GpuState {
     }
 }
 
-pub type SharedState = Arc<RwLock<AppState>>;
+#[cfg(test)]
+pub struct MockState {
+    pub receivers: AppReceivers,
+    // Keep senders alive so receivers don't return Closed errors
+    _net_tx: watch::Sender<NetworkState>,
+    _cpu_tx: watch::Sender<CpuState>,
+    _mem_tx: watch::Sender<MemoryState>,
+    _sys_tx: watch::Sender<SysState>,
+    _gpu_tx: watch::Sender<GpuState>,
+    _disks_tx: watch::Sender<Vec<DiskInfo>>,
+    _bt_tx: watch::Sender<BtState>,
+    _audio_tx: watch::Sender<AudioState>,
+}
 
 #[cfg(test)]
-pub fn mock_state(state: AppState) -> SharedState {
+#[derive(Default, Clone)]
-    Arc::new(RwLock::new(state))
+pub struct AppState {
+    pub network: NetworkState,
+    pub cpu: CpuState,
+    pub memory: MemoryState,
+    pub sys: SysState,
+    pub gpu: GpuState,
+    pub disks: Vec<DiskInfo>,
+    pub bluetooth: BtState,
+    pub audio: AudioState,
+    pub health: HashMap<String, ModuleHealth>,
+}
+
+#[cfg(test)]
+pub fn mock_state(state: AppState) -> MockState {
+    let (net_tx, net_rx) = watch::channel(state.network);
+    let (cpu_tx, cpu_rx) = watch::channel(state.cpu);
+    let (mem_tx, mem_rx) = watch::channel(state.memory);
+    let (sys_tx, sys_rx) = watch::channel(state.sys);
+    let (gpu_tx, gpu_rx) = watch::channel(state.gpu);
+    let (disks_tx, disks_rx) = watch::channel(state.disks);
+    let (bt_tx, bt_rx) = watch::channel(state.bluetooth);
+    let (audio_tx, audio_rx) = watch::channel(state.audio);
+
+    MockState {
+        receivers: AppReceivers {
+            network: net_rx,
+            cpu: cpu_rx,
+            memory: mem_rx,
+            sys: sys_rx,
+            gpu: gpu_rx,
+            disks: disks_rx,
+            bluetooth: bt_rx,
+            audio: audio_rx,
+            health: Arc::new(RwLock::new(state.health)),
+        },
+        _net_tx: net_tx,
+        _cpu_tx: cpu_tx,
+        _mem_tx: mem_tx,
+        _sys_tx: sys_tx,
+        _gpu_tx: gpu_tx,
+        _disks_tx: disks_tx,
+        _bt_tx: bt_tx,
+        _audio_tx: audio_tx,
+    }
 }