fixed testcase

.gitea/workflows/ci.yml

@@ -0,0 +1,60 @@
+name: CI
+
+on:
+  pull_request:
+    branches:
+      - main
+
+jobs:
+  check:
+    name: Lint and Test
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout code
+        uses: actions/checkout@v4
+
+      - name: Install Rust
+        uses: actions-rust-lang/setup-rust-toolchain@v1
+        with:
+          toolchain: stable
+          components: rustfmt, clippy
+          cache: false
+
+      - name: Check formatting
+        run: cargo fmt --check
+
+      - name: Clippy
+        run: cargo clippy --tests -- -D warnings
+
+      - name: Test
+        run: cargo test
+
+      - name: Build
+        run: cargo build --release
+
+  version-check:
+    name: Version Check
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout repository
+        uses: actions/checkout@v4
+        with:
+          fetch-depth: 0
+
+      - name: Compare versions
+        shell: bash
+        run: |
+          NEW_VERSION=$(grep -m1 '^version =' Cargo.toml | cut -d '"' -f 2)
+
+          git fetch origin ${{ github.base_ref }}
+          OLD_VERSION=$(git show origin/${{ github.base_ref }}:Cargo.toml | grep -m1 '^version =' | cut -d '"' -f 2)
+
+          echo "Old version (main): $OLD_VERSION"
+          echo "New version (PR): $NEW_VERSION"
+
+          if [ "$NEW_VERSION" = "$OLD_VERSION" ]; then
+            echo "Error: Cargo.toml version has not been updated in this PR!"
+            exit 1
+          else
+            echo "Success: Version updated from $OLD_VERSION to $NEW_VERSION"
+          fi

.gitea/workflows/release.yml

@@ -0,0 +1,109 @@
+name: Release
+
+on:
+  push:
+    branches:
+      - main
+    tags:
+      - 'v*'
+
+jobs:
+  build:
+    name: Build and Release
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout code
+        uses: actions/checkout@v4
+
+      - name: Install Rust
+        uses: actions-rust-lang/setup-rust-toolchain@v1
+        with:
+          toolchain: stable
+          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: Get Version
+        id: get_version
+        run: |
+          VERSION=$(grep -m1 '^version =' Cargo.toml | cut -d '"' -f 2)
+          echo "VERSION=$VERSION" >> $GITHUB_OUTPUT
+          if [[ "${{ github.ref }}" == refs/tags/v* ]]; then
+            echo "TAG=${{ github.ref_name }}" >> $GITHUB_OUTPUT
+          else
+            echo "TAG=v$VERSION" >> $GITHUB_OUTPUT
+          fi
+
+      - name: Check if Release Exists
+        id: check_release
+        shell: bash
+        run: |
+          HTTP_STATUS=$(curl -s -o /dev/null -w "%{http_code}" \
+            -H "Authorization: token ${{ secrets.GITHUB_TOKEN }}" \
+            "${{ github.server_url }}/api/v1/repos/${{ github.repository }}/releases/tags/${{ steps.get_version.outputs.TAG }}")
+
+          if [ "$HTTP_STATUS" = "200" ]; then
+            echo "EXISTS=true" >> $GITHUB_OUTPUT
+            echo "Release already exists for tag ${{ steps.get_version.outputs.TAG }}. Skipping."
+          else
+            echo "EXISTS=false" >> $GITHUB_OUTPUT
+          fi
+
+      - name: Build
+        if: steps.check_release.outputs.EXISTS == 'false'
+        run: cargo build --release
+
+      - name: Test
+        if: steps.check_release.outputs.EXISTS == 'false'
+        run: cargo test
+
+      - name: Package .deb
+        if: steps.check_release.outputs.EXISTS == 'false'
+        env:
+          VERSION: ${{ steps.get_version.outputs.VERSION }}
+          TAG: ${{ steps.get_version.outputs.TAG }}
+        run: |
+          PKG="fluxo-rs_${VERSION}_amd64"
+
+          mkdir -p "${PKG}/DEBIAN"
+          mkdir -p "${PKG}/usr/bin"
+
+          cp target/release/fluxo-rs "${PKG}/usr/bin/"
+          strip "${PKG}/usr/bin/fluxo-rs"
+
+          printf '%s\n' \
+            "Package: fluxo-rs" \
+            "Version: ${VERSION}" \
+            "Section: utils" \
+            "Priority: optional" \
+            "Architecture: amd64" \
+            "Maintainer: fluxo-rs contributors" \
+            "Description: High-performance daemon/client for Waybar custom modules" \
+            " fluxo-rs is a compiled Rust daemon that polls system metrics and" \
+            " serves formatted JSON output to Waybar custom modules over a Unix" \
+            " domain socket. Replaces shell scripts with a single binary." \
+            > "${PKG}/DEBIAN/control"
+
+          dpkg-deb --build "${PKG}"
+
+          mv "${PKG}.deb" "fluxo-rs-${TAG}-amd64.deb"
+          echo "Built: fluxo-rs-${TAG}-amd64.deb"
+
+      - name: Create Release and Upload Assets
+        if: steps.check_release.outputs.EXISTS == 'false'
+        uses: https://github.com/softprops/action-gh-release@v1
+        with:
+          tag_name: ${{ steps.get_version.outputs.TAG }}
+          name: Release ${{ steps.get_version.outputs.TAG }}
+          body: |
+            Automated release for version ${{ steps.get_version.outputs.VERSION }}
+            Commit: ${{ github.sha }}
+            Branch: ${{ github.ref_name }}
+          files: |
+            fluxo-rs-${{ steps.get_version.outputs.TAG }}-amd64.deb
+            target/release/fluxo-rs
+          draft: false
+          prerelease: false
+        env:
+          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}

Cargo.toml

@@ -1,16 +1,27 @@
 [package]
 name = "fluxo-rs"
-version = "0.1.0"
+version = "0.3.0"
 edition = "2024"
 
 [dependencies]
 anyhow = "1.0.102"
 clap = { version = "4.6.0", features = ["derive"] }
-fs4 = "0.13.1"
-regex = "1.10"
+ctrlc = "3"
+regex = "1.12"
 serde = { version = "1.0.228", features = ["derive"] }
 serde_json = "1.0.149"
 sysinfo = "0.38.4"
 toml = "1.0.6"
 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-util = { version = "0.7", features = ["codec"] }
+futures = "0.3"
+libpulse-binding = "2.26"
+nix = { version = "0.29", features = ["net"] }
+
+[dev-dependencies]
+
+tempfile = "3"

README.md

@@ -32,8 +32,7 @@ This approach eliminates process spawning overhead and temporary file locking, r
 | `pool` | Aggregate storage (btrfs) | `{used}`, `{total}` |
 | `vol` | Audio output volume | `{name}`, `{volume}`, `{icon}` |
 | `mic` | Audio input volume | `{name}`, `{volume}`, `{icon}` |
-| `bt` | Bluetooth status | `{alias}` |
-| `buds` | Pixel Buds Pro control | `{left}`, `{right}`, `{anc}` |
+| `bt` | Bluetooth status & plugins | `{alias}`, `{mac}`, `{left}`, `{right}`, `{anc}` |
 | `power` | Battery and AC status | `{percentage}`, `{icon}` |
 | `game` | Hyprland gamemode status | active/inactive icon strings |
 

example.config.toml

@@ -0,0 +1,46 @@
+# Fluxo configuration example
+
+[general]
+menu_command = "fuzzel --dmenu --prompt '{prompt}'"
+
+[network]
+format = "{interface} ({ip}):  {rx:>5.2} MB/s   {tx:>5.2} MB/s"
+
+[cpu]
+format = "CPU: {usage:>4.1}% {temp:>4.1}C"
+
+[memory]
+format = "{used:>5.2}/{total:>5.2}GB"
+
+[gpu]
+format_amd = "AMD: {usage:>3.0}% {vram_used:>4.1}/{vram_total:>4.1}GB {temp:>4.1}C"
+format_intel = "iGPU: {usage:>3.0}%"
+format_nvidia = "NV: {usage:>3.0}% {vram_used:>4.1}/{vram_total:>4.1}GB {temp:>4.1}C"
+
+[sys]
+format = "UP: {uptime} | LOAD: {load1:>4.2} {load5:>4.2} {load15:>4.2}"
+
+[disk]
+format = "{mount} {used:>5.1}/{total:>5.1}G"
+
+[pool]
+format = "{used:>4.0}G / {total:>4.0}G"
+
+[power]
+format = "{percentage:>3}%  {icon}"
+
+[bt]
+format_connected = "{alias} 󰂰"
+format_plugin = "{alias} [{left}|{right}] {anc} 󰂰"
+format_disconnected = "󰂯"
+format_disabled = "󰂲 Off"
+
+[audio]
+format_sink_unmuted = "{name} {volume:>3}% {icon}"
+format_sink_muted = "{name} {icon}"
+format_source_unmuted = "{name} {volume:>3}% {icon}"
+format_source_muted = "{name} {icon}"
+
+[game]
+format_active = "<span size='large'>󰊖</span>"
+format_inactive = "<span size='large'></span>"

src/config.rs

@@ -1,6 +1,8 @@
+use regex::Regex;
 use serde::Deserialize;
 use std::fs;
 use std::path::PathBuf;
+use std::sync::LazyLock;
 use tracing::{debug, info, warn};
 
 #[derive(Deserialize, Default)]
@@ -24,8 +26,6 @@ pub struct Config {
     #[serde(default)]
     pub power: PowerConfig,
     #[serde(default)]
-    pub buds: BudsConfig,
-    #[serde(default)]
     pub audio: AudioConfig,
     #[serde(default)]
     pub bt: BtConfig,
@@ -95,9 +95,11 @@ pub struct GpuConfig {
 impl Default for GpuConfig {
     fn default() -> Self {
         Self {
-            format_amd: "AMD: {usage:>3.0}% {vram_used:>4.1}/{vram_total:>4.1}GB {temp:>4.1}C".to_string(),
+            format_amd: "AMD: {usage:>3.0}% {vram_used:>4.1}/{vram_total:>4.1}GB {temp:>4.1}C"
+                .to_string(),
             format_intel: "iGPU: {usage:>3.0}%".to_string(),
-            format_nvidia: "NV: {usage:>3.0}% {vram_used:>4.1}/{vram_total:>4.1}GB {temp:>4.1}C".to_string(),
+            format_nvidia: "NV: {usage:>3.0}% {vram_used:>4.1}/{vram_total:>4.1}GB {temp:>4.1}C"
+                .to_string(),
         }
     }
 }
@@ -154,23 +156,6 @@ impl Default for PowerConfig {
     }
 }
 
-#[derive(Deserialize)]
-pub struct BudsConfig {
-    pub mac: String,
-    pub format: String,
-    pub format_disconnected: String,
-}
-
-impl Default for BudsConfig {
-    fn default() -> Self {
-        Self {
-            mac: "B4:23:A2:09:D3:53".to_string(),
-            format: "{left} | {right} | {anc}".to_string(),
-            format_disconnected: "<span size='large'></span>".to_string(),
-        }
-    }
-}
-
 #[derive(Deserialize)]
 pub struct AudioConfig {
     pub format_sink_unmuted: String,
@@ -193,6 +178,7 @@ impl Default for AudioConfig {
 #[derive(Deserialize)]
 pub struct BtConfig {
     pub format_connected: String,
+    pub format_plugin: String,
     pub format_disconnected: String,
     pub format_disabled: String,
 }
@@ -201,6 +187,7 @@ impl Default for BtConfig {
     fn default() -> Self {
         Self {
             format_connected: "{alias} 󰂰".to_string(),
+            format_plugin: "{alias} [{left}|{right}] {anc} 󰂰".to_string(),
             format_disconnected: "󰂯".to_string(),
             format_disabled: "󰂲 Off".to_string(),
         }
@@ -222,6 +209,84 @@ impl Default for GameConfig {
     }
 }
 
+static TOKEN_RE: LazyLock<Regex> =
+    LazyLock::new(|| Regex::new(r"\{([a-zA-Z0-9_]+)(?::([<>\^])?(\d+)?(?:\.(\d+))?)?\}").unwrap());
+
+fn extract_tokens(format_str: &str) -> Vec<String> {
+    TOKEN_RE
+        .captures_iter(format_str)
+        .map(|cap| cap[1].to_string())
+        .collect()
+}
+
+fn validate_format(label: &str, format_str: &str, known_tokens: &[&str]) {
+    for token in extract_tokens(format_str) {
+        if !known_tokens.contains(&token.as_str()) {
+            warn!(
+                "Config [{}]: unknown token '{{{}}}' in format string. Known tokens: {:?}",
+                label, token, known_tokens
+            );
+        }
+    }
+}
+
+impl Config {
+    pub fn validate(&self) {
+        validate_format(
+            "network",
+            &self.network.format,
+            &["interface", "ip", "rx", "tx"],
+        );
+        validate_format("cpu", &self.cpu.format, &["usage", "temp"]);
+        validate_format("memory", &self.memory.format, &["used", "total"]);
+        validate_format(
+            "gpu.amd",
+            &self.gpu.format_amd,
+            &["usage", "vram_used", "vram_total", "temp"],
+        );
+        validate_format("gpu.intel", &self.gpu.format_intel, &["usage", "freq"]);
+        validate_format(
+            "gpu.nvidia",
+            &self.gpu.format_nvidia,
+            &["usage", "vram_used", "vram_total", "temp"],
+        );
+        validate_format(
+            "sys",
+            &self.sys.format,
+            &["uptime", "load1", "load5", "load15", "procs"],
+        );
+        validate_format("disk", &self.disk.format, &["mount", "used", "total"]);
+        validate_format("pool", &self.pool.format, &["used", "total"]);
+        validate_format("power", &self.power.format, &["percentage", "icon"]);
+        validate_format(
+            "audio.sink_unmuted",
+            &self.audio.format_sink_unmuted,
+            &["name", "icon", "volume"],
+        );
+        validate_format(
+            "audio.sink_muted",
+            &self.audio.format_sink_muted,
+            &["name", "icon"],
+        );
+        validate_format(
+            "audio.source_unmuted",
+            &self.audio.format_source_unmuted,
+            &["name", "icon", "volume"],
+        );
+        validate_format(
+            "audio.source_muted",
+            &self.audio.format_source_muted,
+            &["name", "icon"],
+        );
+        validate_format("bt.connected", &self.bt.format_connected, &["alias"]);
+        validate_format(
+            "bt.plugin",
+            &self.bt.format_plugin,
+            &["alias", "left", "right", "anc", "mac"],
+        );
+    }
+}
+
 pub fn load_config(custom_path: Option<PathBuf>) -> Config {
     let config_path = custom_path.unwrap_or_else(|| {
         let config_dir = std::env::var("XDG_CONFIG_HOME")
@@ -234,9 +299,10 @@ pub fn load_config(custom_path: Option<PathBuf>) -> Config {
     });
 
     if let Ok(content) = fs::read_to_string(&config_path) {
-        match toml::from_str(&content) {
+        match toml::from_str::<Config>(&content) {
             Ok(cfg) => {
                 info!("Successfully loaded configuration from {:?}", config_path);
+                cfg.validate();
                 cfg
             }
             Err(e) => {
@@ -246,7 +312,79 @@ pub fn load_config(custom_path: Option<PathBuf>) -> Config {
             }
         }
     } else {
-        debug!("No config file found at {:?}, using default settings.", config_path);
+        debug!(
+            "No config file found at {:?}, using default settings.",
+            config_path
+        );
         Config::default()
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    use std::io::Write;
+
+    #[test]
+    fn test_default_config() {
+        let config = Config::default();
+        assert_eq!(
+            config.general.menu_command,
+            "fuzzel --dmenu --prompt '{prompt}'"
+        );
+        assert!(config.cpu.format.contains("usage"));
+        assert!(config.cpu.format.contains("temp"));
+        assert!(config.memory.format.contains("used"));
+        assert!(config.memory.format.contains("total"));
+    }
+
+    #[test]
+    fn test_load_missing_config() {
+        let config = load_config(Some(PathBuf::from("/nonexistent/config.toml")));
+        // Should fallback to defaults without panicking
+        assert_eq!(
+            config.general.menu_command,
+            "fuzzel --dmenu --prompt '{prompt}'"
+        );
+    }
+
+    #[test]
+    fn test_load_valid_partial_config() {
+        let mut tmpfile = tempfile::NamedTempFile::new().unwrap();
+        // In TOML, braces have no special meaning in strings
+        writeln!(tmpfile, "[cpu]").unwrap();
+        writeln!(tmpfile, "format = \"custom: {{usage}}\"").unwrap();
+
+        let config = load_config(Some(tmpfile.path().to_path_buf()));
+        // TOML treats {{ as literal {{ (no escape), so the value is "custom: {{usage}}"
+        assert!(config.cpu.format.contains("usage"));
+        // Other sections still have defaults
+        assert!(config.memory.format.contains("used"));
+    }
+
+    #[test]
+    fn test_load_invalid_toml() {
+        let mut tmpfile = tempfile::NamedTempFile::new().unwrap();
+        write!(tmpfile, "this is not valid toml {{{{").unwrap();
+
+        let config = load_config(Some(tmpfile.path().to_path_buf()));
+        // Should fallback to defaults
+        assert_eq!(
+            config.general.menu_command,
+            "fuzzel --dmenu --prompt '{prompt}'"
+        );
+    }
+
+    #[test]
+    fn test_load_empty_config() {
+        let tmpfile = tempfile::NamedTempFile::new().unwrap();
+        // Empty file is valid TOML, all sections default
+
+        let config = load_config(Some(tmpfile.path().to_path_buf()));
+        assert_eq!(
+            config.general.menu_command,
+            "fuzzel --dmenu --prompt '{prompt}'"
+        );
+        assert!(config.cpu.format.contains("usage"));
+    }
+}

src/daemon.rs

@@ -1,62 +1,136 @@
 use crate::config::Config;
-use crate::ipc::SOCKET_PATH;
-use crate::modules::network::NetworkDaemon;
-use crate::modules::hardware::HardwareDaemon;
+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 anyhow::Result;
 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, RwLock};
 use std::thread;
 use std::time::Duration;
-use tracing::{info, warn, error, debug};
-use std::path::PathBuf;
+use tracing::{debug, error, info, warn};
+
+struct SocketGuard {
+    path: String,
+}
+
+impl Drop for SocketGuard {
+    fn drop(&mut self) {
+        debug!("Cleaning up socket file: {}", self.path);
+        let _ = fs::remove_file(&self.path);
+    }
+}
 
 pub fn run_daemon(config_path: Option<PathBuf>) -> Result<()> {
-    if fs::metadata(SOCKET_PATH).is_ok() {
-        debug!("Removing stale socket file: {}", SOCKET_PATH);
-        fs::remove_file(SOCKET_PATH)?;
+    let sock_path = socket_path();
+
+    if fs::metadata(&sock_path).is_ok() {
+        debug!("Removing stale socket file: {}", sock_path);
+        fs::remove_file(&sock_path)?;
     }
 
     let state: SharedState = Arc::new(RwLock::new(AppState::default()));
-    let listener = UnixListener::bind(SOCKET_PATH)?;
+    let listener = UnixListener::bind(&sock_path)?;
+    let _guard = SocketGuard {
+        path: sock_path.clone(),
+    };
+
+    // Signal handling: set flag so main loop exits cleanly
+    let running = Arc::new(AtomicBool::new(true));
+    let running_clone = Arc::clone(&running);
+    ctrlc::set_handler(move || {
+        info!("Received shutdown signal, exiting...");
+        running_clone.store(false, Ordering::SeqCst);
+    })?;
 
     // 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
     let poll_state = Arc::clone(&state);
+    let poll_running = Arc::clone(&running);
     thread::spawn(move || {
-        info!("Starting background polling thread");
-        let mut network_daemon = NetworkDaemon::new();
-        let mut hardware_daemon = HardwareDaemon::new();
-        loop {
-            network_daemon.poll(Arc::clone(&poll_state));
-            hardware_daemon.poll(Arc::clone(&poll_state));
+        info!("Starting Network polling thread");
+        let mut daemon = NetworkDaemon::new();
+        while poll_running.load(Ordering::SeqCst) {
+            daemon.poll(Arc::clone(&poll_state));
             thread::sleep(Duration::from_secs(1));
         }
     });
 
-    info!("Fluxo daemon successfully bound to socket: {}", SOCKET_PATH);
+    // 2. Fast Hardware Thread (CPU, Mem, Load)
+    let poll_state = Arc::clone(&state);
+    let poll_running = Arc::clone(&running);
+    thread::spawn(move || {
+        info!("Starting Fast Hardware polling thread");
+        let mut daemon = HardwareDaemon::new();
+        while poll_running.load(Ordering::SeqCst) {
+            daemon.poll_fast(Arc::clone(&poll_state));
+            thread::sleep(Duration::from_secs(1));
+        }
+    });
 
-    for stream in listener.incoming() {
-        match stream {
-            Ok(mut stream) => {
+    // 3. Slow Hardware Thread (GPU, Disks)
+    let poll_state = Arc::clone(&state);
+    let poll_running = Arc::clone(&running);
+    thread::spawn(move || {
+        info!("Starting Slow Hardware polling thread");
+        let mut daemon = HardwareDaemon::new();
+        while poll_running.load(Ordering::SeqCst) {
+            daemon.poll_slow(Arc::clone(&poll_state));
+            thread::sleep(Duration::from_secs(1));
+        }
+    });
+
+    // 4. Bluetooth Thread
+    let poll_state = Arc::clone(&state);
+    let poll_running = Arc::clone(&running);
+    thread::spawn(move || {
+        info!("Starting Bluetooth polling thread");
+        let mut daemon = BtDaemon::new();
+        while poll_running.load(Ordering::SeqCst) {
+            daemon.poll(Arc::clone(&poll_state));
+            thread::sleep(Duration::from_secs(1));
+        }
+    });
+
+    // 5. Audio Thread (Event driven)
+    let audio_daemon = AudioDaemon::new();
+    audio_daemon.start(Arc::clone(&state));
+
+    info!("Fluxo daemon successfully bound to socket: {}", sock_path);
+
+    // Use non-blocking accept so we can check the running flag
+    listener.set_nonblocking(true)?;
+
+    while running.load(Ordering::SeqCst) {
+        match listener.accept() {
+            Ok((mut stream, _)) => {
                 let state_clone = Arc::clone(&state);
                 let config_clone = Arc::clone(&config);
                 let cp_clone = config_path_clone.clone();
                 thread::spawn(move || {
-                    let mut reader = BufReader::new(stream.try_clone().unwrap());
+                    let mut reader = BufReader::new(&stream);
                     let mut request = String::new();
                     if let Err(e) = reader.read_line(&mut request) {
                         error!("Failed to read from IPC stream: {}", e);
                         return;
                     }
+                    drop(reader);
 
                     let request = request.trim();
-                    if request.is_empty() { return; }
+                    if request.is_empty() {
+                        return;
+                    }
 
                     let parts: Vec<&str> = request.split_whitespace().collect();
                     if let Some(module_name) = parts.first() {
@@ -70,25 +144,46 @@ pub fn run_daemon(config_path: Option<PathBuf>) -> Result<()> {
                             } 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);
+                        let response =
+                            handle_request(module_name, &parts[1..], &state_clone, &config_clone);
                         if let Err(e) = stream.write_all(response.as_bytes()) {
-                            error!("Failed to write IPC response: {}", e);
+                            if e.kind() == std::io::ErrorKind::BrokenPipe
+                                || e.kind() == std::io::ErrorKind::ConnectionReset
+                            {
+                                debug!(
+                                    "IPC client disconnected before response could be sent: {}",
+                                    e
+                                );
+                            } else {
+                                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(module_name: &str, args: &[&str], state: &SharedState, config_lock: &Arc<RwLock<Config>>) -> String {
+fn handle_request(
+    module_name: &str,
+    args: &[&str],
+    state: &SharedState,
+    config_lock: &Arc<RwLock<Config>>,
+) -> String {
     let config = if let Ok(c) = config_lock.read() {
         c
     } else {
@@ -102,18 +197,25 @@ fn handle_request(module_name: &str, args: &[&str], state: &SharedState, config_
         "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.get(0).unwrap_or(&"show")]),
-        "mic" => crate::modules::audio::AudioModule.run(&config, state, &["source", args.get(0).unwrap_or(&"show")]),
+        "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),
-        "buds" => crate::modules::buds::BudsModule.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))
-        },
+        }
     };
 
     match result {
@@ -121,7 +223,7 @@ fn handle_request(module_name: &str, args: &[&str], state: &SharedState, config_
         Err(e) => {
             error!(module = module_name, error = %e, "Module execution failed");
             let err_out = crate::output::WaybarOutput {
-                text: format!("\u{200B}Error\u{200B}"),
+                text: "\u{200B}Error\u{200B}".to_string(),
                 tooltip: Some(e.to_string()),
                 class: Some("error".to_string()),
                 percentage: None,

src/ipc.rs

@@ -1,12 +1,21 @@
 use std::io::{Read, Write};
 use std::os::unix::net::UnixStream;
+use std::time::Duration;
 use tracing::debug;
 
-pub const SOCKET_PATH: &str = "/tmp/fluxo.sock";
+pub fn socket_path() -> String {
+    if let Ok(dir) = std::env::var("XDG_RUNTIME_DIR") {
+        format!("{}/fluxo.sock", dir)
+    } else {
+        "/tmp/fluxo.sock".to_string()
+    }
+}
 
-pub fn request_data(module: &str, args: &[String]) -> anyhow::Result<String> {
-    debug!(module, ?args, "Connecting to daemon socket: {}", SOCKET_PATH);
-    let mut stream = UnixStream::connect(SOCKET_PATH)?;
+pub fn request_data(module: &str, args: &[&str]) -> anyhow::Result<String> {
+    let sock = socket_path();
+    debug!(module, ?args, "Connecting to daemon socket: {}", sock);
+    let mut stream = UnixStream::connect(&sock)?;
+    stream.set_read_timeout(Some(Duration::from_secs(5)))?;
 
     // Send module and args
     let mut request = module.to_string();

src/main.rs

@@ -10,7 +10,7 @@ use clap::{Parser, Subcommand};
 use std::path::PathBuf;
 use std::process;
 use tracing::{error, info};
-use tracing_subscriber::{fmt, prelude::*, EnvFilter};
+use tracing_subscriber::{EnvFilter, fmt, prelude::*};
 
 #[derive(Parser)]
 #[command(name = "fluxo")]
@@ -71,11 +71,6 @@ enum Commands {
         #[arg(default_value = "show")]
         action: String,
     },
-    /// Pixel Buds Pro ANC and Battery
-    Buds {
-        #[arg(default_value = "show")]
-        action: String,
-    },
     /// System power and battery status
     Power,
     /// Hyprland gamemode status
@@ -98,27 +93,25 @@ fn main() {
                 process::exit(1);
             }
         }
-        Commands::Reload => {
-            match ipc::request_data("reload", &[]) {
-                Ok(_) => info!("Reload signal sent to daemon"),
-                Err(e) => {
-                    error!("Failed to send reload signal: {}", e);
-                    process::exit(1);
-                }
+        Commands::Reload => match ipc::request_data("reload", &[]) {
+            Ok(_) => info!("Reload signal sent to daemon"),
+            Err(e) => {
+                error!("Failed to send reload signal: {}", e);
+                process::exit(1);
             }
-        }
+        },
         Commands::Net => handle_ipc_response(ipc::request_data("net", &[])),
         Commands::Cpu => handle_ipc_response(ipc::request_data("cpu", &[])),
         Commands::Mem => handle_ipc_response(ipc::request_data("mem", &[])),
-        Commands::Disk { path } => handle_ipc_response(ipc::request_data("disk", &[path.clone()])),
-        Commands::Pool { kind } => handle_ipc_response(ipc::request_data("pool", &[kind.clone()])),
+        Commands::Disk { path } => handle_ipc_response(ipc::request_data("disk", &[path])),
+        Commands::Pool { kind } => handle_ipc_response(ipc::request_data("pool", &[kind])),
         Commands::Vol { cycle } => {
             let action = if *cycle { "cycle" } else { "show" };
-            handle_ipc_response(ipc::request_data("vol", &[action.to_string()]));
+            handle_ipc_response(ipc::request_data("vol", &[action]));
         }
         Commands::Mic { cycle } => {
             let action = if *cycle { "cycle" } else { "show" };
-            handle_ipc_response(ipc::request_data("mic", &[action.to_string()]));
+            handle_ipc_response(ipc::request_data("mic", &[action]));
         }
         Commands::Gpu => handle_ipc_response(ipc::request_data("gpu", &[])),
         Commands::Sys => handle_ipc_response(ipc::request_data("sys", &[])),
@@ -127,13 +120,37 @@ fn main() {
                 // Client-side execution of the menu
                 let config = config::load_config(None);
 
-                let devices_out = std::process::Command::new("bluetoothctl")
+                let mut items = Vec::new();
+
+                // If connected, show plugin modes and disconnect
+                if let Ok(json_str) = ipc::request_data("bt", &["get_modes"])
+                    && let Ok(val) = serde_json::from_str::<serde_json::Value>(&json_str)
+                    && let Some(modes_str) = val.get("text").and_then(|t| t.as_str())
+                    && !modes_str.is_empty()
+                {
+                    for mode in modes_str.lines() {
+                        items.push(format!("Mode: {}", mode));
+                    }
+                }
+
+                if !items.is_empty() {
+                    items.push("Disconnect".to_string());
+                }
+
+                items.push("--- Connect Device ---".to_string());
+
+                let devices_out = match std::process::Command::new("bluetoothctl")
                     .args(["devices"])
                     .output()
-                    .expect("Failed to run bluetoothctl");
+                {
+                    Ok(out) => out,
+                    Err(e) => {
+                        error!("bluetoothctl not found or failed: {}", e);
+                        return;
+                    }
+                };
                 let stdout = String::from_utf8_lossy(&devices_out.stdout);
 
-                let mut items = Vec::new();
                 for line in stdout.lines() {
                     if line.starts_with("Device ") {
                         let parts: Vec<&str> = line.splitn(3, ' ').collect();
@@ -144,24 +161,31 @@ fn main() {
                 }
 
                 if !items.is_empty() {
-                    if let Ok(selected) = utils::show_menu("Connect BT: ", &items, &config.general.menu_command) {
-                        if let Some(mac_start) = selected.rfind('(') {
-                            if let Some(mac_end) = selected.rfind(')') {
-                                let mac = &selected[mac_start + 1..mac_end];
-                                let _ = std::process::Command::new("bluetoothctl")
-                                    .args(["connect", mac])
-                                    .status();
-                            }
+                    if let Ok(selected) =
+                        utils::show_menu("BT Menu: ", &items, &config.general.menu_command)
+                    {
+                        if let Some(mode) = selected.strip_prefix("Mode: ") {
+                            handle_ipc_response(ipc::request_data("bt", &["set_mode", mode]));
+                        } else if selected == "Disconnect" {
+                            handle_ipc_response(ipc::request_data("bt", &["disconnect"]));
+                        } else if selected == "--- Connect Device ---" {
+                            // Do nothing
+                        } else if let Some(mac_start) = selected.rfind('(')
+                            && let Some(mac_end) = selected.rfind(')')
+                        {
+                            let mac = &selected[mac_start + 1..mac_end];
+                            let _ = std::process::Command::new("bluetoothctl")
+                                .args(["connect", mac])
+                                .status();
                         }
                     }
                 } else {
-                    info!("No paired Bluetooth devices found.");
+                    info!("No Bluetooth options found.");
                 }
                 return;
             }
-            handle_ipc_response(ipc::request_data("bt", &[action.clone()]));
+            handle_ipc_response(ipc::request_data("bt", &[action]));
         }
-        Commands::Buds { action } => handle_ipc_response(ipc::request_data("buds", &[action.clone()])),
         Commands::Power => handle_ipc_response(ipc::request_data("power", &[])),
         Commands::Game => handle_ipc_response(ipc::request_data("game", &[])),
     }
@@ -169,24 +193,22 @@ fn main() {
 
 fn handle_ipc_response(response: anyhow::Result<String>) {
     match response {
-        Ok(json_str) => {
-            match serde_json::from_str::<serde_json::Value>(&json_str) {
-                Ok(mut val) => {
-                    if let Some(text) = val.get_mut("text").and_then(|t| t.as_str()) {
-                        let processed_text = if text.contains('<') {
-                            text.to_string()
-                        } else {
-                            text.replace(' ', "\u{2007}")
-                        };
+        Ok(json_str) => match serde_json::from_str::<serde_json::Value>(&json_str) {
+            Ok(mut val) => {
+                if let Some(text) = val.get_mut("text").and_then(|t| t.as_str()) {
+                    let processed_text = if text.contains('<') {
+                        text.to_string()
+                    } else {
+                        text.replace(' ', "\u{2007}")
+                    };
 
-                        let fixed_text = format!("\u{200B}{}\u{200B}", processed_text);
-                        val["text"] = serde_json::Value::String(fixed_text);
-                    }
-                    println!("{}", serde_json::to_string(&val).unwrap());
+                    let fixed_text = format!("\u{200B}{}\u{200B}", processed_text);
+                    val["text"] = serde_json::Value::String(fixed_text);
                 }
-                Err(_) => println!("{}", json_str),
+                println!("{}", serde_json::to_string(&val).unwrap());
             }
-        }
+            Err(_) => println!("{}", json_str),
+        },
         Err(e) => {
             let err_out = output::WaybarOutput {
                 text: format!("\u{200B}Daemon offline ({})\u{200B}", e),

src/modules/audio.rs

@@ -2,55 +2,220 @@ use crate::config::Config;
 use crate::modules::WaybarModule;
 use crate::output::WaybarOutput;
 use crate::state::SharedState;
-use crate::utils::{format_template, TokenValue};
+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 std::sync::Arc;
+use tracing::error;
+
+pub struct AudioDaemon;
+
+impl AudioDaemon {
+    pub fn new() -> Self {
+        Self
+    }
+
+    pub fn start(&self, state: SharedState) {
+        let state_arc = Arc::clone(&state);
+
+        std::thread::spawn(move || {
+            let mut mainloop =
+                ThreadedMainloop::new().expect("Failed to create pulse threaded mainloop");
+
+            let mut context =
+                Context::new(&mainloop, "fluxo-rs").expect("Failed to create pulse context");
+
+            context
+                .connect(None, ContextFlag::NOFLAGS, None)
+                .expect("Failed to connect pulse context");
+
+            mainloop.start().expect("Failed to start pulse mainloop");
+
+            mainloop.lock();
+
+            // Wait for context to be ready
+            loop {
+                match context.get_state() {
+                    libpulse_binding::context::State::Ready => break,
+                    libpulse_binding::context::State::Failed
+                    | libpulse_binding::context::State::Terminated => {
+                        error!("Pulse context failed or terminated");
+                        mainloop.unlock();
+                        return;
+                    }
+                    _ => {
+                        mainloop.unlock();
+                        std::thread::sleep(Duration::from_millis(50));
+                        mainloop.lock();
+                    }
+                }
+            }
+
+            // Initial fetch
+            let _ = fetch_audio_data_sync(&mut context, &state_arc);
+
+            // Subscribe to events
+            let interest =
+                InterestMaskSet::SINK | InterestMaskSet::SOURCE | InterestMaskSet::SERVER;
+            context.subscribe(interest, |_| {});
+
+            let (tx, rx) = std::sync::mpsc::channel();
+
+            context.set_subscribe_callback(Some(Box::new(move |facility, _operation, _index| {
+                match facility {
+                    Some(Facility::Sink) | Some(Facility::Source) | Some(Facility::Server) => {
+                        let _ = tx.send(());
+                    }
+                    _ => {}
+                }
+            })));
+
+            mainloop.unlock();
+
+            // Background polling loop driven by events or a 2s fallback timeout
+            loop {
+                let _ = rx.recv_timeout(Duration::from_secs(2));
+                {
+                    mainloop.lock();
+                    let _ = fetch_audio_data_sync(&mut context, &state_arc);
+                    mainloop.unlock();
+                }
+            }
+        });
+    }
+}
+
+use std::time::Duration;
+
+fn fetch_audio_data_sync(context: &mut Context, state: &SharedState) -> Result<()> {
+    let state_inner = Arc::clone(state);
+
+    // 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);
+    context.introspect().get_server_info(move |info| {
+        let mut lock = st_server.write().unwrap();
+        lock.audio.sink.name = info
+            .default_sink_name
+            .as_ref()
+            .map(|s| s.to_string())
+            .unwrap_or_default();
+        lock.audio.source.name = info
+            .default_source_name
+            .as_ref()
+            .map(|s| s.to_string())
+            .unwrap_or_default();
+    });
+
+    let st_sink = Arc::clone(&state_inner);
+    context.introspect().get_sink_info_list(move |res| {
+        if let ListResult::Item(item) = res {
+            let mut lock = st_sink.write().unwrap();
+            // 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)
+                .unwrap_or(false);
+            if is_default {
+                lock.audio.sink.description = item
+                    .description
+                    .as_ref()
+                    .map(|s| s.to_string())
+                    .unwrap_or_default();
+                lock.audio.sink.volume =
+                    ((item.volume.avg().0 as f64 / Volume::NORMAL.0 as f64) * 100.0).round() as u8;
+                lock.audio.sink.muted = item.mute;
+            }
+        }
+    });
+
+    let st_source = Arc::clone(&state_inner);
+    context.introspect().get_source_info_list(move |res| {
+        if let ListResult::Item(item) = res {
+            let mut lock = st_source.write().unwrap();
+            let is_default = item
+                .name
+                .as_ref()
+                .map(|s| s.as_ref() == lock.audio.source.name)
+                .unwrap_or(false);
+            if is_default {
+                lock.audio.source.description = item
+                    .description
+                    .as_ref()
+                    .map(|s| s.to_string())
+                    .unwrap_or_default();
+                lock.audio.source.volume =
+                    ((item.volume.avg().0 as f64 / Volume::NORMAL.0 as f64) * 100.0).round() as u8;
+                lock.audio.source.muted = item.mute;
+            }
+        }
+    });
+
+    Ok(())
+}
 
 pub struct AudioModule;
 
 impl WaybarModule for AudioModule {
-    fn run(&self, config: &Config, _state: &SharedState, args: &[&str]) -> Result<WaybarOutput> {
+    fn run(&self, config: &Config, state: &SharedState, 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)?;
-                return Ok(WaybarOutput {
-                    text: String::new(),
-                    tooltip: None,
-                    class: None,
-                    percentage: None,
-                });
-            }
-            "show" | _ => {
-                self.get_status(config, target_type)
+                Ok(WaybarOutput::default())
             }
+            "show" => self.get_status(config, state, target_type),
+            other => Err(anyhow!("Unknown audio action: '{}'", other)),
         }
     }
 }
 
 impl AudioModule {
-    fn get_status(&self, config: &Config, target_type: &str) -> Result<WaybarOutput> {
-        let target = if target_type == "sink" { "@DEFAULT_AUDIO_SINK@" } else { "@DEFAULT_AUDIO_SOURCE@" };
+    fn get_status(
+        &self,
+        config: &Config,
+        state: &SharedState,
+        target_type: &str,
+    ) -> Result<WaybarOutput> {
+        let audio_state = {
+            let lock = state.read().unwrap();
+            lock.audio.clone()
+        };
 
-        let output = Command::new("wpctl")
-            .args(["get-volume", target])
-            .output()?;
-        let stdout = String::from_utf8_lossy(&output.stdout);
+        let (name, description, volume, muted) = if target_type == "sink" {
+            (
+                audio_state.sink.name,
+                audio_state.sink.description,
+                audio_state.sink.volume,
+                audio_state.sink.muted,
+            )
+        } else {
+            (
+                audio_state.source.name,
+                audio_state.source.description,
+                audio_state.source.volume,
+                audio_state.source.muted,
+            )
+        };
 
-        let parts: Vec<&str> = stdout.trim().split_whitespace().collect();
-        if parts.len() < 2 {
-            return Err(anyhow!("Could not parse wpctl output: {}", stdout));
+        if name.is_empty() {
+            // Fallback if daemon hasn't populated state yet
+            return Ok(WaybarOutput {
+                text: "Audio Loading...".to_string(),
+                ..Default::default()
+            });
         }
 
-        let vol_val: f64 = parts[1].parse().unwrap_or(0.0);
-        let vol = (vol_val * 100.0).round() as u8;
-        let display_vol = std::cmp::min(vol, 100);
-        let muted = stdout.contains("[MUTED]");
-
-        let description = self.get_description(target_type)?;
-        let name = if description.len() > 20 {
+        let display_name = if description.len() > 20 {
             format!("{}...", &description[..17])
         } else {
             description.clone()
@@ -58,31 +223,43 @@ impl AudioModule {
 
         let (text, class) = if muted {
             let icon = if target_type == "sink" { "" } else { "" };
-            let format_str = if target_type == "sink" { &config.audio.format_sink_muted } else { &config.audio.format_source_muted };
+            let format_str = if target_type == "sink" {
+                &config.audio.format_sink_muted
+            } else {
+                &config.audio.format_source_muted
+            };
             let t = format_template(
                 format_str,
                 &[
-                    ("name", TokenValue::String(&name)),
-                    ("icon", TokenValue::String(icon)),
-                ]
+                    ("name", TokenValue::String(display_name)),
+                    ("icon", TokenValue::String(icon.to_string())),
+                ],
             );
             (t, "muted")
         } else {
             let icon = if target_type == "sink" {
-                if display_vol <= 30 { "" }
-                else if display_vol <= 60 { "" }
-                else { "" }
+                if volume <= 30 {
+                    ""
+                } else if volume <= 60 {
+                    ""
+                } else {
+                    ""
+                }
             } else {
                 ""
             };
-            let format_str = if target_type == "sink" { &config.audio.format_sink_unmuted } else { &config.audio.format_source_unmuted };
+            let format_str = if target_type == "sink" {
+                &config.audio.format_sink_unmuted
+            } else {
+                &config.audio.format_source_unmuted
+            };
             let t = format_template(
                 format_str,
                 &[
-                    ("name", TokenValue::String(&name)),
-                    ("icon", TokenValue::String(icon)),
-                    ("volume", TokenValue::Int(display_vol as i64)),
-                ]
+                    ("name", TokenValue::String(display_name)),
+                    ("icon", TokenValue::String(icon.to_string())),
+                    ("volume", TokenValue::Int(volume as i64)),
+                ],
             );
             (t, "unmuted")
         };
@@ -91,44 +268,35 @@ impl AudioModule {
             text,
             tooltip: Some(description),
             class: Some(class.to_string()),
-            percentage: Some(display_vol),
+            percentage: Some(volume),
         })
     }
 
-    fn get_description(&self, target_type: &str) -> Result<String> {
-        let info_output = Command::new("pactl").arg("info").output()?;
-        let info_stdout = String::from_utf8_lossy(&info_output.stdout);
-        let search_key = if target_type == "sink" { "Default Sink:" } else { "Default Source:" };
-        
-        let default_dev = info_stdout.lines()
-            .find(|l| l.contains(search_key))
-            .and_then(|l| l.split(':').nth(1))
-            .map(|s| s.trim())
-            .ok_or_else(|| anyhow!("Default {} not found", target_type))?;
-
-        let list_cmd = if target_type == "sink" { "sinks" } else { "sources" };
-        let list_output = Command::new("pactl").args(["list", list_cmd]).output()?;
-        let list_stdout = String::from_utf8_lossy(&list_output.stdout);
-
-        let mut current_name = String::new();
-        for line in list_stdout.lines() {
-            if line.trim().starts_with("Name: ") {
-                current_name = line.split(':').nth(1).unwrap_or("").trim().to_string();
-            }
-            if current_name == default_dev && line.trim().starts_with("Description: ") {
-                return Ok(line.split(':').nth(1).unwrap_or("").trim().to_string());
-            }
-        }
-
-        Ok(default_dev.to_string())
-    }
-
     fn cycle_device(&self, target_type: &str) -> Result<()> {
-        let list_cmd = if target_type == "sink" { "sinks" } else { "sources" };
-        let output = Command::new("pactl").args(["list", "short", list_cmd]).output()?;
+        // 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" {
+            "set-default-sink"
+        } else {
+            "set-default-source"
+        };
+
+        // We need to find the "next" device.
+        // For simplicity, let's keep the CLI version for now or refactor later.
+        // The user asked for "step by step".
+
+        let list_cmd = if target_type == "sink" {
+            "sinks"
+        } else {
+            "sources"
+        };
+        let output = Command::new("pactl")
+            .args(["list", "short", list_cmd])
+            .output()?;
         let stdout = String::from_utf8_lossy(&output.stdout);
 
-        let devices: Vec<String> = stdout.lines()
+        let devices: Vec<String> = stdout
+            .lines()
             .filter_map(|l| {
                 let parts: Vec<&str> = l.split_whitespace().collect();
                 if parts.len() >= 2 {
@@ -144,13 +312,19 @@ impl AudioModule {
             })
             .collect();
 
-        if devices.is_empty() { return Ok(()); }
+        if devices.is_empty() {
+            return Ok(());
+        }
 
-        let info_output = Command::new("pactl").arg("info").output()?;
+        let info_output = Command::new("pactl").args(["info"]).output()?;
         let info_stdout = String::from_utf8_lossy(&info_output.stdout);
-        let search_key = if target_type == "sink" { "Default Sink:" } else { "Default Source:" };
-        
-        let current_dev = info_stdout.lines()
+        let search_key = if target_type == "sink" {
+            "Default Sink:"
+        } else {
+            "Default Source:"
+        };
+        let current_dev = info_stdout
+            .lines()
             .find(|l| l.contains(search_key))
             .and_then(|l| l.split(':').nth(1))
             .map(|s| s.trim())
@@ -160,9 +334,7 @@ impl AudioModule {
         let next_index = (current_index + 1) % devices.len();
         let next_dev = &devices[next_index];
 
-        let set_cmd = if target_type == "sink" { "set-default-sink" } else { "set-default-source" };
         Command::new("pactl").args([set_cmd, next_dev]).status()?;
-
         Ok(())
     }
 }

src/modules/bt.rs

@@ -1,79 +1,600 @@
 use crate::config::Config;
 use crate::modules::WaybarModule;
 use crate::output::WaybarOutput;
-use crate::state::SharedState;
-use crate::utils::{format_template, TokenValue};
-use anyhow::Result;
+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> {
+    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()
+        };
 
-        if *action == "disconnect" {
-            if let Some(mac) = find_audio_device() {
-                let _ = Command::new("bluetoothctl").args(["disconnect", &mac]).output();
+        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: String::new(),
-                tooltip: None,
-                class: None,
+                text: config.bt.format_disabled.clone(),
+                tooltip: Some("Bluetooth Disabled".to_string()),
+                class: Some("disabled".to_string()),
                 percentage: None,
             });
         }
 
-        if let Ok(output) = Command::new("bluetoothctl").arg("show").output() {
-            let stdout = String::from_utf8_lossy(&output.stdout);
-            if stdout.contains("Powered: no") {
-                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()),
+                ),
+            ];
 
-        if let Some(mac) = find_audio_device() {
-            let info_output = Command::new("bluetoothctl").args(["info", &mac]).output()?;
-            let info = String::from_utf8_lossy(&info_output.stdout);
+            let mut class = vec!["connected".to_string()];
+            let mut has_plugin = false;
 
-            let mut alias = mac.clone();
-            let mut battery = None;
-            let mut trusted = "no";
-
-            for line in info.lines() {
-                if line.contains("Alias:") {
-                    alias = line.split("Alias:").nth(1).unwrap_or("").trim().to_string();
-                } else if line.contains("Battery Percentage:") {
-                    if let Some(bat_str) = line.split('(').nth(1).and_then(|s| s.split(')').next()) {
-                        battery = bat_str.parse::<u8>().ok();
-                    }
-                } else if line.contains("Trusted: yes") {
-                    trusted = "yes";
+            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 tooltip = format!(
-                "{} | MAC: {}\nTrusted: {} | Bat: {}",
-                alias,
-                mac,
-                trusted,
-                battery.map(|b| format!("{}%", b)).unwrap_or_else(|| "N/A".to_string())
-            );
+            let format = if has_plugin {
+                &config.bt.format_plugin
+            } else {
+                &config.bt.format_connected
+            };
 
-            let text = format_template(
-                &config.bt.format_connected,
-                &[("alias", TokenValue::String(&alias))]
+            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("connected".to_string()),
-                percentage: battery,
+                class: Some(class.join(" ")),
+                percentage: bt_state.battery_percentage,
             })
         } else {
             Ok(WaybarOutput {
@@ -85,35 +606,3 @@ impl WaybarModule for BtModule {
         }
     }
 }
-
-fn find_audio_device() -> Option<String> {
-    if let Ok(output) = Command::new("pactl").arg("get-default-sink").output() {
-        let sink = String::from_utf8_lossy(&output.stdout).trim().to_string();
-        if sink.starts_with("bluez_output.") {
-            let parts: Vec<&str> = sink.split('.').collect();
-            if parts.len() >= 2 {
-                return Some(parts[1].replace('_', ":"));
-            }
-        }
-    }
-
-    if let Ok(output) = Command::new("bluetoothctl").args(["devices", "Connected"]).output() {
-        let stdout = String::from_utf8_lossy(&output.stdout);
-        for line in stdout.lines() {
-            if line.starts_with("Device ") {
-                let parts: Vec<&str> = line.split_whitespace().collect();
-                if parts.len() >= 2 {
-                    let mac = parts[1];
-                    if let Ok(info) = Command::new("bluetoothctl").args(["info", mac]).output() {
-                        let info_str = String::from_utf8_lossy(&info.stdout);
-                        if info_str.contains("0000110b-0000-1000-8000-00805f9b34fb") {
-                            return Some(mac.to_string());
-                        }
-                    }
-                }
-            }
-        }
-    }
-
-    None
-}

src/modules/btrfs.rs

@@ -2,22 +2,26 @@ use crate::config::Config;
 use crate::modules::WaybarModule;
 use crate::output::WaybarOutput;
 use crate::state::SharedState;
-use crate::utils::{format_template, TokenValue};
+use crate::utils::{TokenValue, format_template};
 use anyhow::Result;
-use sysinfo::Disks;
 
 pub struct BtrfsModule;
 
 impl WaybarModule for BtrfsModule {
-    fn run(&self, config: &Config, _state: &SharedState, _args: &[&str]) -> Result<WaybarOutput> {
-        let disks = Disks::new_with_refreshed_list();
+    fn run(&self, config: &Config, state: &SharedState, _args: &[&str]) -> Result<WaybarOutput> {
+        let disks = if let Ok(s) = state.read() {
+            s.disks.clone()
+        } else {
+            return Err(anyhow::anyhow!("Failed to read state"));
+        };
+
         let mut total_used: f64 = 0.0;
         let mut total_size: f64 = 0.0;
 
         for disk in &disks {
-            if disk.file_system().to_string_lossy().to_lowercase().contains("btrfs") {
-                let size = disk.total_space() as f64;
-                let available = disk.available_space() as f64;
+            if disk.filesystem.contains("btrfs") {
+                let size = disk.total_bytes as f64;
+                let available = disk.available_bytes as f64;
                 total_size += size;
                 total_used += size - available;
             }
@@ -49,7 +53,7 @@ impl WaybarModule for BtrfsModule {
             &[
                 ("used", TokenValue::Float(used_gb)),
                 ("total", TokenValue::Float(size_gb)),
-            ]
+            ],
         );
 
         Ok(WaybarOutput {

src/modules/buds.rs

@@ -1,121 +0,0 @@
-use crate::config::Config;
-use crate::modules::WaybarModule;
-use crate::output::WaybarOutput;
-use crate::state::SharedState;
-use crate::utils::{format_template, TokenValue};
-use anyhow::Result;
-use std::process::Command;
-
-pub struct BudsModule;
-
-impl WaybarModule for BudsModule {
-    fn run(&self, config: &Config, _state: &SharedState, args: &[&str]) -> Result<WaybarOutput> {
-        let action = args.first().unwrap_or(&"show");
-        let mac = &config.buds.mac;
-
-        match *action {
-            "cycle_anc" => {
-                let output = Command::new("pbpctrl").args(["get", "anc"]).output()?;
-                let current_mode = String::from_utf8_lossy(&output.stdout).trim().to_string();
-                
-                let next_mode = match current_mode.as_str() {
-                    "active" => "aware",
-                    "aware" => "off",
-                    _ => "active",
-                };
-                
-                Command::new("pbpctrl").args(["set", "anc", next_mode]).status()?;
-                return Ok(WaybarOutput {
-                    text: String::new(),
-                    tooltip: None, class: None, percentage: None,
-                });
-            }
-            "connect" => {
-                Command::new("bluetoothctl").args(["connect", mac]).status()?;
-                return Ok(WaybarOutput {
-                    text: String::new(),
-                    tooltip: None, class: None, percentage: None,
-                });
-            }
-            "disconnect" => {
-                Command::new("bluetoothctl").args(["disconnect", mac]).status()?;
-                return Ok(WaybarOutput {
-                    text: String::new(),
-                    tooltip: None, class: None, percentage: None,
-                });
-            }
-            "show" | _ => {}
-        }
-
-        let bt_info = Command::new("bluetoothctl").args(["info", mac]).output()?;
-        let bt_str = String::from_utf8_lossy(&bt_info.stdout);
-        
-        if !bt_str.contains("Connected: yes") {
-            return Ok(WaybarOutput {
-                text: config.buds.format_disconnected.clone(),
-                tooltip: Some("Pixel Buds Pro 2 not connected".to_string()),
-                class: Some("disconnected".to_string()),
-                percentage: None,
-            });
-        }
-
-        let bat_cmd = Command::new("pbpctrl").args(["show", "battery"]).output();
-        if bat_cmd.is_err() || !bat_cmd.as_ref().unwrap().status.success() {
-            return Ok(WaybarOutput {
-                text: config.buds.format_disconnected.clone(),
-                tooltip: Some("Pixel Buds Pro 2 connected (No Data)".to_string()),
-                class: Some("disconnected".to_string()),
-                percentage: None,
-            });
-        }
-        
-        let bat_result = bat_cmd.unwrap();
-        let bat_output = String::from_utf8_lossy(&bat_result.stdout);
-        let mut left_bud = "unknown";
-        let mut right_bud = "unknown";
-
-        for line in bat_output.lines() {
-            if line.contains("left bud:") {
-                left_bud = line.split_whitespace().nth(2).unwrap_or("unknown");
-            } else if line.contains("right bud:") {
-                right_bud = line.split_whitespace().nth(2).unwrap_or("unknown");
-            }
-        }
-
-        if left_bud == "unknown" && right_bud == "unknown" {
-            return Ok(WaybarOutput {
-                text: "{}".to_string(),
-                tooltip: None, class: None, percentage: None,
-            });
-        }
-
-        let left_display = if left_bud == "unknown" { "---".to_string() } else { format!("{}%", left_bud) };
-        let right_display = if right_bud == "unknown" { "---".to_string() } else { format!("{}%", right_bud) };
-
-        let anc_cmd = Command::new("pbpctrl").args(["get", "anc"]).output()?;
-        let current_mode = String::from_utf8_lossy(&anc_cmd.stdout).trim().to_string();
-
-        let (anc_icon, class) = match current_mode.as_str() {
-            "active" => ("ANC", "anc-active"),
-            "aware" => ("Aware", "anc-aware"),
-            "off" => ("Off", "anc-off"),
-            _ => ("?", "anc-unknown"),
-        };
-
-        let text = format_template(
-            &config.buds.format,
-            &[
-                ("left", TokenValue::String(&left_display)),
-                ("right", TokenValue::String(&right_display)),
-                ("anc", TokenValue::String(anc_icon)),
-            ]
-        );
-
-        Ok(WaybarOutput {
-            text,
-            tooltip: Some("Pixel Buds Pro 2".to_string()),
-            class: Some(class.to_string()),
-            percentage: None,
-        })
-    }
-}

src/modules/cpu.rs

@@ -2,7 +2,7 @@ use crate::config::Config;
 use crate::modules::WaybarModule;
 use crate::output::WaybarOutput;
 use crate::state::SharedState;
-use crate::utils::{format_template, TokenValue};
+use crate::utils::{TokenValue, format_template};
 use anyhow::Result;
 
 pub struct CpuModule;
@@ -26,7 +26,7 @@ impl WaybarModule for CpuModule {
             &[
                 ("usage", TokenValue::Float(usage)),
                 ("temp", TokenValue::Float(temp)),
-            ]
+            ],
         );
 
         let class = if usage > 95.0 {
@@ -45,3 +45,58 @@ impl WaybarModule for CpuModule {
         })
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    use crate::state::{AppState, CpuState, mock_state};
+
+    #[test]
+    fn test_cpu_normal() {
+        let state = mock_state(AppState {
+            cpu: CpuState {
+                usage: 25.0,
+                temp: 45.0,
+                model: "Test CPU".into(),
+            },
+            ..Default::default()
+        });
+        let config = Config::default();
+        let output = CpuModule.run(&config, &state, &[]).unwrap();
+        assert!(output.text.contains("25.0"));
+        assert!(output.text.contains("45.0"));
+        assert_eq!(output.class.as_deref(), Some("normal"));
+        assert_eq!(output.percentage, Some(25));
+        assert_eq!(output.tooltip.as_deref(), Some("Test CPU"));
+    }
+
+    #[test]
+    fn test_cpu_high() {
+        let state = mock_state(AppState {
+            cpu: CpuState {
+                usage: 80.0,
+                temp: 70.0,
+                model: "Test".into(),
+            },
+            ..Default::default()
+        });
+        let config = Config::default();
+        let output = CpuModule.run(&config, &state, &[]).unwrap();
+        assert_eq!(output.class.as_deref(), Some("high"));
+    }
+
+    #[test]
+    fn test_cpu_max() {
+        let state = mock_state(AppState {
+            cpu: CpuState {
+                usage: 99.0,
+                temp: 95.0,
+                model: "Test".into(),
+            },
+            ..Default::default()
+        });
+        let config = Config::default();
+        let output = CpuModule.run(&config, &state, &[]).unwrap();
+        assert_eq!(output.class.as_deref(), Some("max"));
+    }
+}

src/modules/disk.rs

@@ -2,28 +2,36 @@ use crate::config::Config;
 use crate::modules::WaybarModule;
 use crate::output::WaybarOutput;
 use crate::state::SharedState;
-use crate::utils::{format_template, TokenValue};
+use crate::utils::{TokenValue, format_template};
 use anyhow::Result;
-use sysinfo::Disks;
 
 pub struct DiskModule;
 
 impl WaybarModule for DiskModule {
-    fn run(&self, config: &Config, _state: &SharedState, args: &[&str]) -> Result<WaybarOutput> {
+    fn run(&self, config: &Config, state: &SharedState, args: &[&str]) -> Result<WaybarOutput> {
         let mountpoint = args.first().unwrap_or(&"/");
 
-        let disks = Disks::new_with_refreshed_list();
+        let disks = if let Ok(s) = state.read() {
+            s.disks.clone()
+        } else {
+            return Err(anyhow::anyhow!("Failed to read state"));
+        };
+
         for disk in &disks {
-            if disk.mount_point().to_string_lossy() == *mountpoint {
-                let total = disk.total_space() as f64;
-                let available = disk.available_space() as f64;
+            if disk.mount_point == *mountpoint {
+                let total = disk.total_bytes as f64;
+                let available = disk.available_bytes as f64;
                 let used = total - available;
 
                 let used_gb = used / 1024.0 / 1024.0 / 1024.0;
                 let total_gb = total / 1024.0 / 1024.0 / 1024.0;
                 let free_gb = available / 1024.0 / 1024.0 / 1024.0;
 
-                let percentage = if total > 0.0 { (used / total) * 100.0 } else { 0.0 };
+                let percentage = if total > 0.0 {
+                    (used / total) * 100.0
+                } else {
+                    0.0
+                };
 
                 let class = if percentage > 95.0 {
                     "max"
@@ -36,15 +44,18 @@ impl WaybarModule for DiskModule {
                 let text = format_template(
                     &config.disk.format,
                     &[
-                        ("mount", TokenValue::String(mountpoint)),
+                        ("mount", TokenValue::String(mountpoint.to_string())),
                         ("used", TokenValue::Float(used_gb)),
                         ("total", TokenValue::Float(total_gb)),
-                    ]
+                    ],
                 );
 
                 return Ok(WaybarOutput {
                     text,
-                    tooltip: Some(format!("Used: {:.1}G\nTotal: {:.1}G\nFree: {:.1}G", used_gb, total_gb, free_gb)),
+                    tooltip: Some(format!(
+                        "Used: {:.1}G\nTotal: {:.1}G\nFree: {:.1}G",
+                        used_gb, total_gb, free_gb
+                    )),
                     class: Some(class.to_string()),
                     percentage: Some(percentage as u8),
                 });
@@ -54,3 +65,48 @@ impl WaybarModule for DiskModule {
         Err(anyhow::anyhow!("Mountpoint {} not found", mountpoint))
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    use crate::state::{AppState, DiskInfo, mock_state};
+
+    fn state_with_disk(mount: &str, total: u64, available: u64) -> crate::state::SharedState {
+        mock_state(AppState {
+            disks: vec![DiskInfo {
+                mount_point: mount.to_string(),
+                filesystem: "ext4".to_string(),
+                total_bytes: total,
+                available_bytes: available,
+            }],
+            ..Default::default()
+        })
+    }
+
+    #[test]
+    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();
+        assert_eq!(output.class.as_deref(), Some("normal"));
+        assert_eq!(output.percentage, Some(40)); // 40% used
+    }
+
+    #[test]
+    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();
+        assert_eq!(output.class.as_deref(), Some("high")); // 85% used
+    }
+
+    #[test]
+    fn test_disk_not_found() {
+        let state = mock_state(AppState::default());
+        let config = Config::default();
+        let result = DiskModule.run(&config, &state, &["/nonexistent"]);
+        assert!(result.is_err());
+    }
+}

src/modules/game.rs

@@ -2,25 +2,18 @@ use crate::config::Config;
 use crate::modules::WaybarModule;
 use crate::output::WaybarOutput;
 use crate::state::SharedState;
-use anyhow::Result;
-use std::process::Command;
+use anyhow::{Result, anyhow};
+use std::env;
+use std::io::{Read, Write};
+use std::os::unix::net::UnixStream;
 
 pub struct GameModule;
 
 impl WaybarModule for GameModule {
     fn run(&self, config: &Config, _state: &SharedState, _args: &[&str]) -> Result<WaybarOutput> {
-        let output = Command::new("hyprctl")
-            .args(["getoption", "animations:enabled", "-j"])
-            .output();
-
-        let mut is_gamemode = false;
-
-        if let Ok(out) = output {
-            let stdout = String::from_utf8_lossy(&out.stdout);
-            if stdout.contains("\"int\": 0") {
-                is_gamemode = true;
-            }
-        }
+        let is_gamemode = hyprland_ipc("j/getoption animations:enabled")
+            .map(|stdout| stdout.contains("\"int\": 0"))
+            .unwrap_or(false);
 
         if is_gamemode {
             Ok(WaybarOutput {
@@ -39,3 +32,17 @@ impl WaybarModule for GameModule {
         }
     }
 }
+
+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)?;
+    stream.write_all(cmd.as_bytes())?;
+
+    let mut response = String::new();
+    stream.read_to_string(&mut response)?;
+
+    Ok(response)
+}

src/modules/gpu.rs

@@ -2,7 +2,7 @@ use crate::config::Config;
 use crate::modules::WaybarModule;
 use crate::output::WaybarOutput;
 use crate::state::SharedState;
-use crate::utils::{format_template, TokenValue};
+use crate::utils::{TokenValue, format_template};
 use anyhow::Result;
 
 pub struct GpuModule;
@@ -21,7 +21,15 @@ impl WaybarModule for GpuModule {
                     state_lock.gpu.model.clone(),
                 )
             } else {
-                (false, String::from("Unknown"), 0.0, 0.0, 0.0, 0.0, String::from("Unknown"))
+                (
+                    false,
+                    String::from("Unknown"),
+                    0.0,
+                    0.0,
+                    0.0,
+                    0.0,
+                    String::from("Unknown"),
+                )
             }
         };
 
@@ -55,13 +63,16 @@ impl WaybarModule for GpuModule {
                 ("vram_used", TokenValue::Float(vram_used)),
                 ("vram_total", TokenValue::Float(vram_total)),
                 ("temp", TokenValue::Float(temp)),
-            ]
+            ],
         );
 
         let tooltip = if vendor == "Intel" {
             format!("Model: {}\nApprox Usage: {:.0}%", model, usage)
         } else {
-            format!("Model: {}\nUsage: {:.0}%\nVRAM: {:.1}/{:.1}GB\nTemp: {:.1}°C", model, usage, vram_used, vram_total, temp)
+            format!(
+                "Model: {}\nUsage: {:.0}%\nVRAM: {:.1}/{:.1}GB\nTemp: {:.1}°C",
+                model, usage, vram_used, vram_total, temp
+            )
         };
 
         Ok(WaybarOutput {

src/modules/hardware.rs

@@ -1,35 +1,54 @@
-use crate::state::SharedState;
-use sysinfo::{Components, System};
+use crate::state::{DiskInfo, SharedState};
+use sysinfo::{Components, Disks, System};
 
 pub struct HardwareDaemon {
     sys: System,
     components: Components,
     gpu_vendor: Option<String>,
+    gpu_poll_counter: u8,
+    disk_poll_counter: u8,
 }
 
 impl HardwareDaemon {
     pub fn new() -> Self {
-        let mut sys = System::new_all();
-        sys.refresh_all();
+        let mut sys = System::new();
+        sys.refresh_cpu_usage();
+        sys.refresh_memory();
         let components = Components::new_with_refreshed_list();
-        Self { sys, components, gpu_vendor: None }
+        Self {
+            sys,
+            components,
+            gpu_vendor: None,
+            gpu_poll_counter: 0,
+            disk_poll_counter: 9, // Start at 9 to poll on the first tick
+        }
     }
 
-    pub fn poll(&mut self, state: SharedState) {
+    pub fn poll_fast(&mut self, state: SharedState) {
         self.sys.refresh_cpu_usage();
         self.sys.refresh_memory();
         self.components.refresh(true);
 
         let cpu_usage = self.sys.global_cpu_usage();
-        let cpu_model = self.sys.cpus().first().map(|c| c.brand().to_string()).unwrap_or_else(|| "Unknown".to_string());
+        let cpu_model = self
+            .sys
+            .cpus()
+            .first()
+            .map(|c| c.brand().to_string())
+            .unwrap_or_else(|| "Unknown".to_string());
 
         let mut cpu_temp = 0.0;
         for component in &self.components {
             let label = component.label().to_lowercase();
-            if label.contains("tctl") || label.contains("cpu") || label.contains("package") || label.contains("temp1") {
-                if let Some(temp) = component.temperature() {
-                    cpu_temp = temp as f64;
-                    if cpu_temp > 0.0 { break; }
+            if (label.contains("tctl")
+                || label.contains("cpu")
+                || label.contains("package")
+                || label.contains("temp1"))
+                && let Some(temp) = component.temperature()
+            {
+                cpu_temp = temp as f64;
+                if cpu_temp > 0.0 {
+                    break;
                 }
             }
         }
@@ -44,16 +63,16 @@ impl HardwareDaemon {
         let mut process_count = 0;
         if let Ok(loadavg_str) = std::fs::read_to_string("/proc/loadavg") {
             let parts: Vec<&str> = loadavg_str.split_whitespace().collect();
-            if parts.len() >= 4 {
-                if let Some(total_procs) = parts[3].split('/').nth(1) {
-                    process_count = total_procs.parse().unwrap_or(0);
-                }
+            if parts.len() >= 4
+                && let Some(total_procs) = parts[3].split('/').nth(1)
+            {
+                process_count = total_procs.parse().unwrap_or(0);
             }
         }
 
         if let Ok(mut state_lock) = state.write() {
             state_lock.cpu.usage = cpu_usage as f64;
-            state_lock.cpu.temp = cpu_temp as f64;
+            state_lock.cpu.temp = cpu_temp;
             state_lock.cpu.model = cpu_model;
 
             state_lock.memory.total_gb = total_mem;
@@ -64,101 +83,162 @@ impl HardwareDaemon {
             state_lock.sys.load_15 = load_avg.fifteen;
             state_lock.sys.uptime = uptime;
             state_lock.sys.process_count = process_count;
+        }
+    }
 
-            self.poll_gpu(&mut state_lock.gpu);
+    pub fn poll_slow(&mut self, state: SharedState) {
+        // 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);
+        }
+
+        // 2. Gather Disk data outside of lock
+        let mut disks_data = None;
+        self.disk_poll_counter = (self.disk_poll_counter + 1) % 10;
+        if self.disk_poll_counter == 0 {
+            disks_data = Some(
+                Disks::new_with_refreshed_list()
+                    .iter()
+                    .map(|d| DiskInfo {
+                        mount_point: d.mount_point().to_string_lossy().into_owned(),
+                        filesystem: d.file_system().to_string_lossy().to_lowercase(),
+                        total_bytes: d.total_space(),
+                        available_bytes: d.available_space(),
+                    })
+                    .collect::<Vec<DiskInfo>>(),
+            );
+        }
+
+        // 3. Apply to state
+        if let Ok(mut state_lock) = state.write() {
+            if should_poll_gpu {
+                state_lock.gpu = gpu_state;
+            }
+
+            if let Some(d) = disks_data {
+                state_lock.disks = d;
+            }
         }
     }
 
     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() {
-            if let Ok(output) = std::process::Command::new("nvidia-smi")
-                .args(["--query-gpu=utilization.gpu,memory.used,memory.total,temperature.gpu,name", "--format=csv,noheader,nounits"])
+        if (self.gpu_vendor.as_deref() == Some("NVIDIA") || self.gpu_vendor.is_none())
+            && let Ok(output) = std::process::Command::new("nvidia-smi")
+                .args([
+                    "--query-gpu=utilization.gpu,memory.used,memory.total,temperature.gpu,name",
+                    "--format=csv,noheader,nounits",
+                ])
                 .output()
-            {
-                if output.status.success() {
-                    let stdout = String::from_utf8_lossy(&output.stdout);
-                    if let Some(line) = stdout.lines().next() {
-                        let parts: Vec<&str> = line.split(',').collect();
-                        if parts.len() >= 5 {
-                            gpu.active = true;
-                            gpu.vendor = "NVIDIA".to_string();
-                            gpu.usage = parts[0].trim().parse().unwrap_or(0.0);
-                            gpu.vram_used = parts[1].trim().parse::<f64>().unwrap_or(0.0) / 1024.0;
-                            gpu.vram_total = parts[2].trim().parse::<f64>().unwrap_or(0.0) / 1024.0;
-                            gpu.temp = parts[3].trim().parse().unwrap_or(0.0);
-                            gpu.model = parts[4].trim().to_string();
-                            self.gpu_vendor = Some("NVIDIA".to_string());
-                            return;
-                        }
-                    }
+            && output.status.success()
+        {
+            let stdout = String::from_utf8_lossy(&output.stdout);
+            if let Some(line) = stdout.lines().next() {
+                let parts: Vec<&str> = line.split(',').collect();
+                if parts.len() >= 5 {
+                    gpu.active = true;
+                    gpu.vendor = "NVIDIA".to_string();
+                    gpu.usage = parts[0].trim().parse().unwrap_or(0.0);
+                    gpu.vram_used = parts[1].trim().parse::<f64>().unwrap_or(0.0) / 1024.0;
+                    gpu.vram_total = parts[2].trim().parse::<f64>().unwrap_or(0.0) / 1024.0;
+                    gpu.temp = parts[3].trim().parse().unwrap_or(0.0);
+                    gpu.model = parts[4].trim().to_string();
+                    self.gpu_vendor = Some("NVIDIA".to_string());
+                    return;
                 }
             }
         }
 
-        if self.gpu_vendor.as_deref() == Some("AMD") || self.gpu_vendor.as_deref() == Some("Intel") || self.gpu_vendor.is_none() {
+        if self.gpu_vendor.as_deref() == Some("AMD")
+            || self.gpu_vendor.as_deref() == Some("Intel")
+            || self.gpu_vendor.is_none()
+        {
             for i in 0..=3 {
                 let base = format!("/sys/class/drm/card{}/device", i);
 
-                if self.gpu_vendor.as_deref() == Some("AMD") || self.gpu_vendor.is_none() {
-                    if let Ok(usage_str) = std::fs::read_to_string(format!("{}/gpu_busy_percent", base)) {
-                        gpu.active = true;
-                        gpu.vendor = "AMD".to_string();
-                        gpu.usage = usage_str.trim().parse().unwrap_or(0.0);
+                if (self.gpu_vendor.as_deref() == Some("AMD") || self.gpu_vendor.is_none())
+                    && let Ok(usage_str) =
+                        std::fs::read_to_string(format!("{}/gpu_busy_percent", base))
+                {
+                    gpu.active = true;
+                    gpu.vendor = "AMD".to_string();
+                    gpu.usage = usage_str.trim().parse().unwrap_or(0.0);
 
-                        if let Ok(mem_used) = std::fs::read_to_string(format!("{}/mem_info_vram_used", base)) {
-                            gpu.vram_used = mem_used.trim().parse::<f64>().unwrap_or(0.0) / 1024.0 / 1024.0 / 1024.0;
-                        }
-                        if let Ok(mem_total) = std::fs::read_to_string(format!("{}/mem_info_vram_total", base)) {
-                            gpu.vram_total = mem_total.trim().parse::<f64>().unwrap_or(0.0) / 1024.0 / 1024.0 / 1024.0;
-                        }
+                    if let Ok(mem_used) =
+                        std::fs::read_to_string(format!("{}/mem_info_vram_used", base))
+                    {
+                        gpu.vram_used = mem_used.trim().parse::<f64>().unwrap_or(0.0)
+                            / 1024.0
+                            / 1024.0
+                            / 1024.0;
+                    }
+                    if let Ok(mem_total) =
+                        std::fs::read_to_string(format!("{}/mem_info_vram_total", base))
+                    {
+                        gpu.vram_total = mem_total.trim().parse::<f64>().unwrap_or(0.0)
+                            / 1024.0
+                            / 1024.0
+                            / 1024.0;
+                    }
 
-                        if let Ok(entries) = std::fs::read_dir(format!("{}/hwmon", base)) {
-                            for entry in entries.flatten() {
-                                let temp_path = entry.path().join("temp1_input");
-                                if let Ok(temp_str) = std::fs::read_to_string(temp_path) {
-                                    gpu.temp = temp_str.trim().parse::<f64>().unwrap_or(0.0) / 1000.0;
-                                    break;
-                                }
+                    if let Ok(entries) = std::fs::read_dir(format!("{}/hwmon", base)) {
+                        for entry in entries.flatten() {
+                            let temp_path = entry.path().join("temp1_input");
+                            if let Ok(temp_str) = std::fs::read_to_string(temp_path) {
+                                gpu.temp = temp_str.trim().parse::<f64>().unwrap_or(0.0) / 1000.0;
+                                break;
                             }
                         }
-                        gpu.model = "AMD GPU".to_string();
-                        self.gpu_vendor = Some("AMD".to_string());
-                        return;
                     }
+                    gpu.model = "AMD GPU".to_string();
+                    self.gpu_vendor = Some("AMD".to_string());
+                    return;
                 }
 
                 if self.gpu_vendor.as_deref() == Some("Intel") || self.gpu_vendor.is_none() {
-                    let freq_path = if std::path::Path::new(&format!("{}/gt_cur_freq_mhz", base)).exists() {
-                        Some(format!("{}/gt_cur_freq_mhz", base))
-                    } else if std::path::Path::new(&format!("/sys/class/drm/card{}/gt_cur_freq_mhz", i)).exists() {
-                        Some(format!("/sys/class/drm/card{}/gt_cur_freq_mhz", i))
-                    } else {
-                        None
-                    };
+                    let freq_path =
+                        if std::path::Path::new(&format!("{}/gt_cur_freq_mhz", base)).exists() {
+                            Some(format!("{}/gt_cur_freq_mhz", base))
+                        } else if std::path::Path::new(&format!(
+                            "/sys/class/drm/card{}/gt_cur_freq_mhz",
+                            i
+                        ))
+                        .exists()
+                        {
+                            Some(format!("/sys/class/drm/card{}/gt_cur_freq_mhz", i))
+                        } else {
+                            None
+                        };
 
-                    if let Some(path) = freq_path {
-                        if let Ok(freq_str) = std::fs::read_to_string(&path) {
-                            gpu.active = true;
-                            gpu.vendor = "Intel".to_string();
+                    if let Some(path) = freq_path
+                        && let Ok(freq_str) = std::fs::read_to_string(&path)
+                    {
+                        gpu.active = true;
+                        gpu.vendor = "Intel".to_string();
 
-                            let cur_freq = freq_str.trim().parse::<f64>().unwrap_or(0.0);
-                            let mut max_freq = 0.0;
+                        let cur_freq = freq_str.trim().parse::<f64>().unwrap_or(0.0);
+                        let mut max_freq = 0.0;
 
-                            let max_path = path.replace("gt_cur_freq_mhz", "gt_max_freq_mhz");
-                            if let Ok(max_str) = std::fs::read_to_string(max_path) {
-                                max_freq = max_str.trim().parse::<f64>().unwrap_or(0.0);
-                            }
-
-                            gpu.usage = if max_freq > 0.0 { (cur_freq / max_freq) * 100.0 } else { 0.0 };
-                            gpu.temp = 0.0;
-                            gpu.vram_used = 0.0;
-                            gpu.vram_total = 0.0;
-                            gpu.model = format!("Intel iGPU ({}MHz)", cur_freq);
-                            self.gpu_vendor = Some("Intel".to_string());
-                            return;
+                        let max_path = path.replace("gt_cur_freq_mhz", "gt_max_freq_mhz");
+                        if let Ok(max_str) = std::fs::read_to_string(max_path) {
+                            max_freq = max_str.trim().parse::<f64>().unwrap_or(0.0);
                         }
+
+                        gpu.usage = if max_freq > 0.0 {
+                            (cur_freq / max_freq) * 100.0
+                        } else {
+                            0.0
+                        };
+                        gpu.temp = 0.0;
+                        gpu.vram_used = 0.0;
+                        gpu.vram_total = 0.0;
+                        gpu.model = format!("Intel iGPU ({}MHz)", cur_freq);
+                        self.gpu_vendor = Some("Intel".to_string());
+                        return;
                     }
                 }
             }

src/modules/memory.rs

@@ -2,7 +2,7 @@ use crate::config::Config;
 use crate::modules::WaybarModule;
 use crate::output::WaybarOutput;
 use crate::state::SharedState;
-use crate::utils::{format_template, TokenValue};
+use crate::utils::{TokenValue, format_template};
 use anyhow::Result;
 
 pub struct MemoryModule;
@@ -11,23 +11,24 @@ impl WaybarModule for MemoryModule {
     fn run(&self, config: &Config, state: &SharedState, _args: &[&str]) -> Result<WaybarOutput> {
         let (used_gb, total_gb) = {
             if let Ok(state_lock) = state.read() {
-                (
-                    state_lock.memory.used_gb,
-                    state_lock.memory.total_gb,
-                )
+                (state_lock.memory.used_gb, state_lock.memory.total_gb)
             } else {
                 (0.0, 0.0)
             }
         };
 
-        let ratio = if total_gb > 0.0 { (used_gb / total_gb) * 100.0 } else { 0.0 };
+        let ratio = if total_gb > 0.0 {
+            (used_gb / total_gb) * 100.0
+        } else {
+            0.0
+        };
 
         let text = format_template(
             &config.memory.format,
             &[
                 ("used", TokenValue::Float(used_gb)),
                 ("total", TokenValue::Float(total_gb)),
-            ]
+            ],
         );
 
         let class = if ratio > 95.0 {
@@ -46,3 +47,55 @@ impl WaybarModule for MemoryModule {
         })
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    use crate::state::{AppState, MemoryState, mock_state};
+
+    #[test]
+    fn test_memory_normal() {
+        let state = mock_state(AppState {
+            memory: MemoryState {
+                used_gb: 8.0,
+                total_gb: 32.0,
+            },
+            ..Default::default()
+        });
+        let config = Config::default();
+        let output = MemoryModule.run(&config, &state, &[]).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]
+    fn test_memory_high() {
+        let state = mock_state(AppState {
+            memory: MemoryState {
+                used_gb: 26.0,
+                total_gb: 32.0,
+            },
+            ..Default::default()
+        });
+        let config = Config::default();
+        let output = MemoryModule.run(&config, &state, &[]).unwrap();
+        assert_eq!(output.class.as_deref(), Some("high")); // 81%
+    }
+
+    #[test]
+    fn test_memory_zero_total() {
+        let state = mock_state(AppState {
+            memory: MemoryState {
+                used_gb: 0.0,
+                total_gb: 0.0,
+            },
+            ..Default::default()
+        });
+        let config = Config::default();
+        let output = MemoryModule.run(&config, &state, &[]).unwrap();
+        assert_eq!(output.class.as_deref(), Some("normal"));
+        assert_eq!(output.percentage, Some(0));
+    }
+}

src/modules/mod.rs

@@ -1,16 +1,15 @@
-pub mod network;
-pub mod cpu;
-pub mod memory;
-pub mod hardware;
-pub mod disk;
-pub mod btrfs;
 pub mod audio;
-pub mod gpu;
-pub mod sys;
 pub mod bt;
-pub mod buds;
-pub mod power;
+pub mod btrfs;
+pub mod cpu;
+pub mod disk;
 pub mod game;
+pub mod gpu;
+pub mod hardware;
+pub mod memory;
+pub mod network;
+pub mod power;
+pub mod sys;
 
 use crate::config::Config;
 use crate::output::WaybarOutput;
@@ -20,4 +19,3 @@ use anyhow::Result;
 pub trait WaybarModule {
     fn run(&self, config: &Config, state: &SharedState, args: &[&str]) -> Result<WaybarOutput>;
 }
-

src/modules/network.rs

@@ -2,8 +2,9 @@ use crate::config::Config;
 use crate::modules::WaybarModule;
 use crate::output::WaybarOutput;
 use crate::state::SharedState;
-use crate::utils::{format_template, TokenValue};
+use crate::utils::{TokenValue, format_template};
 use anyhow::Result;
+use nix::ifaddrs::getifaddrs;
 use std::fs;
 use std::time::{SystemTime, UNIX_EPOCH};
 
@@ -29,22 +30,18 @@ impl NetworkDaemon {
     }
 
     pub fn poll(&mut self, state: SharedState) {
-        // Cache invalidation: if the interface directory doesn't exist, clear cache
-        if let Some(ref iface) = self.cached_interface {
-            if !std::path::Path::new(&format!("/sys/class/net/{}", iface)).exists() {
-                self.cached_interface = None;
-                self.cached_ip = None;
-            }
-        }
-
-        // Re-detect interface if needed
-        if self.cached_interface.is_none() {
-            if let Ok(iface) = get_primary_interface() {
-                if !iface.is_empty() {
-                    self.cached_interface = Some(iface.clone());
-                    self.cached_ip = get_ip_address(&iface);
-                }
+        // Re-detect interface on every poll to catch VPNs or route changes immediately
+        if let Ok(iface) = get_primary_interface()
+            && !iface.is_empty()
+        {
+            // If the interface changed, or we don't have an IP yet, update the IP
+            if self.cached_interface.as_ref() != Some(&iface) || self.cached_ip.is_none() {
+                self.cached_ip = get_ip_address(&iface);
+                self.cached_interface = Some(iface);
             }
+        } else {
+            self.cached_interface = None;
+            self.cached_ip = None;
         }
 
         if let Some(ref interface) = self.cached_interface {
@@ -55,17 +52,26 @@ impl NetworkDaemon {
 
             if let Ok((rx_bytes_now, tx_bytes_now)) = get_bytes(interface) {
                 if self.last_time > 0 && time_now > self.last_time {
-                    let time_diff = time_now - self.last_time;
-                    let rx_bps = (rx_bytes_now.saturating_sub(self.last_rx_bytes)) / time_diff;
-                    let tx_bps = (tx_bytes_now.saturating_sub(self.last_tx_bytes)) / time_diff;
-
-                    let rx_mbps = (rx_bps as f64) / 1024.0 / 1024.0;
-                    let tx_mbps = (tx_bps as f64) / 1024.0 / 1024.0;
+                    let time_diff = (time_now - self.last_time) as f64;
+                    let rx_mbps = (rx_bytes_now.saturating_sub(self.last_rx_bytes)) as f64
+                        / time_diff
+                        / 1024.0
+                        / 1024.0;
+                    let tx_mbps = (tx_bytes_now.saturating_sub(self.last_tx_bytes)) as f64
+                        / time_diff
+                        / 1024.0
+                        / 1024.0;
 
                     if let Ok(mut state_lock) = state.write() {
                         state_lock.network.rx_mbps = rx_mbps;
                         state_lock.network.tx_mbps = tx_mbps;
+                        state_lock.network.interface = interface.clone();
+                        state_lock.network.ip = self.cached_ip.clone().unwrap_or_default();
                     }
+                } else if let Ok(mut state_lock) = state.write() {
+                    // First poll: no speed data yet, but update interface/ip
+                    state_lock.network.interface = interface.clone();
+                    state_lock.network.ip = self.cached_ip.clone().unwrap_or_default();
                 }
 
                 self.last_time = time_now;
@@ -75,13 +81,32 @@ impl NetworkDaemon {
                 // Read failed, might be down
                 self.cached_interface = None;
             }
+        } else if let Ok(mut state_lock) = state.write() {
+            // No interface detected
+            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> {
-        let interface = get_primary_interface()?;
+        let (interface, ip, rx_mbps, tx_mbps) = if let Ok(s) = state.read() {
+            (
+                s.network.interface.clone(),
+                s.network.ip.clone(),
+                s.network.rx_mbps,
+                s.network.tx_mbps,
+            )
+        } else {
+            return Ok(WaybarOutput {
+                text: "No connection".to_string(),
+                tooltip: None,
+                class: None,
+                percentage: None,
+            });
+        };
+
         if interface.is_empty() {
             return Ok(WaybarOutput {
                 text: "No connection".to_string(),
@@ -91,37 +116,31 @@ impl WaybarModule for NetworkModule {
             });
         }
 
-        let ip = get_ip_address(&interface).unwrap_or_else(|| String::from("No IP"));
-        
-        let (rx_mbps, tx_mbps) = {
-            if let Ok(state_lock) = state.read() {
-                (state_lock.network.rx_mbps, state_lock.network.tx_mbps)
-            } else {
-                (0.0, 0.0)
-            }
-        };
+        let ip_display = if ip.is_empty() { "No IP" } else { &ip };
 
         let mut output_text = format_template(
             &config.network.format,
             &[
-                ("interface", TokenValue::String(&interface)),
-                ("ip", TokenValue::String(&ip)),
+                ("interface", TokenValue::String(interface.clone())),
+                ("ip", TokenValue::String(ip_display.to_string())),
                 ("rx", TokenValue::Float(rx_mbps)),
                 ("tx", TokenValue::Float(tx_mbps)),
-            ]
+            ],
         );
 
         if interface.starts_with("tun")
             || interface.starts_with("wg")
             || interface.starts_with("ppp")
             || interface.starts_with("pvpn")
+            || interface.starts_with("proton")
+            || interface.starts_with("ipsec")
         {
-            output_text = format!("  {}", output_text);
+            output_text = format!("  {}", output_text);
         }
 
         Ok(WaybarOutput {
             text: output_text,
-            tooltip: Some(format!("Interface: {}\nIP: {}", interface, ip)),
+            tooltip: Some(format!("Interface: {}\nIP: {}", interface, ip_display)),
             class: Some(interface),
             percentage: None,
         })
@@ -129,27 +148,18 @@ impl WaybarModule for NetworkModule {
 }
 
 fn get_primary_interface() -> Result<String> {
-    let output = std::process::Command::new("ip")
-        .args(["route", "list"])
-        .output()?;
-    let stdout = String::from_utf8_lossy(&output.stdout);
+    let content = fs::read_to_string("/proc/net/route")?;
 
     let mut defaults = Vec::new();
-    for line in stdout.lines() {
-        if line.starts_with("default") {
-            let parts: Vec<&str> = line.split_whitespace().collect();
-            let mut dev = "";
-            let mut metric = 0;
-            for i in 0..parts.len() {
-                if parts[i] == "dev" && i + 1 < parts.len() {
-                    dev = parts[i + 1];
-                }
-                if parts[i] == "metric" && i + 1 < parts.len() {
-                    metric = parts[i + 1].parse::<i32>().unwrap_or(0);
-                }
-            }
-            if !dev.is_empty() {
-                defaults.push((metric, dev.to_string()));
+    for line in content.lines().skip(1) {
+        let parts: Vec<&str> = line.split_whitespace().collect();
+        if parts.len() >= 7 {
+            let iface = parts[0];
+            let dest = parts[1];
+            let metric = parts[6].parse::<i32>().unwrap_or(0);
+
+            if dest == "00000000" {
+                defaults.push((metric, iface.to_string()));
             }
         }
     }
@@ -163,19 +173,13 @@ fn get_primary_interface() -> Result<String> {
 }
 
 fn get_ip_address(interface: &str) -> Option<String> {
-    let output = std::process::Command::new("ip")
-        .args(["-4", "addr", "show", interface])
-        .output()
-        .ok()?;
-    let stdout = String::from_utf8_lossy(&output.stdout);
-    for line in stdout.lines() {
-        if line.trim().starts_with("inet ") {
-            let parts: Vec<&str> = line.trim().split_whitespace().collect();
-            if parts.len() > 1 {
-                let ip_cidr = parts[1];
-                let ip = ip_cidr.split('/').next().unwrap_or(ip_cidr);
-                return Some(ip.to_string());
-            }
+    let addrs = getifaddrs().ok()?;
+    for ifaddr in addrs {
+        if ifaddr.interface_name == interface
+            && let Some(address) = ifaddr.address
+            && let Some(sockaddr) = address.as_sockaddr_in()
+        {
+            return Some(sockaddr.ip().to_string());
         }
     }
     None
@@ -198,3 +202,68 @@ fn get_bytes(interface: &str) -> Result<(u64, u64)> {
 
     Ok((rx, tx))
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    use crate::state::{AppState, NetworkState, mock_state};
+
+    #[test]
+    fn test_network_no_connection() {
+        let state = mock_state(AppState::default());
+        let config = Config::default();
+        let output = NetworkModule.run(&config, &state, &[]).unwrap();
+        assert_eq!(output.text, "No connection");
+    }
+
+    #[test]
+    fn test_network_connected() {
+        let state = mock_state(AppState {
+            network: NetworkState {
+                rx_mbps: 1.5,
+                tx_mbps: 0.3,
+                interface: "eth0".to_string(),
+                ip: "192.168.1.100".to_string(),
+            },
+            ..Default::default()
+        });
+        let config = Config::default();
+        let output = NetworkModule.run(&config, &state, &[]).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]
+    fn test_network_vpn_prefix() {
+        let state = mock_state(AppState {
+            network: NetworkState {
+                rx_mbps: 0.0,
+                tx_mbps: 0.0,
+                interface: "wg0".to_string(),
+                ip: "10.0.0.1".to_string(),
+            },
+            ..Default::default()
+        });
+        let config = Config::default();
+        let output = NetworkModule.run(&config, &state, &[]).unwrap();
+        assert!(output.text.starts_with("  "));
+    }
+
+    #[test]
+    fn test_network_no_ip() {
+        let state = mock_state(AppState {
+            network: NetworkState {
+                rx_mbps: 0.0,
+                tx_mbps: 0.0,
+                interface: "eth0".to_string(),
+                ip: String::new(),
+            },
+            ..Default::default()
+        });
+        let config = Config::default();
+        let output = NetworkModule.run(&config, &state, &[]).unwrap();
+        assert!(output.text.contains("No IP"));
+    }
+}

src/modules/power.rs

@@ -2,7 +2,7 @@ use crate::config::Config;
 use crate::modules::WaybarModule;
 use crate::output::WaybarOutput;
 use crate::state::SharedState;
-use crate::utils::{format_template, TokenValue};
+use crate::utils::{TokenValue, format_template};
 use anyhow::Result;
 use std::fs;
 
@@ -32,11 +32,11 @@ impl WaybarModule for PowerModule {
                 let name = entry.file_name().to_string_lossy().to_string();
                 if name.starts_with("AC") || name.starts_with("ADP") {
                     let online_path = entry.path().join("online");
-                    if let Ok(online_str) = fs::read_to_string(online_path) {
-                        if online_str.trim() == "1" {
-                            ac_online = true;
-                            break;
-                        }
+                    if let Ok(online_str) = fs::read_to_string(online_path)
+                        && online_str.trim() == "1"
+                    {
+                        ac_online = true;
+                        break;
                     }
                 }
             }
@@ -60,9 +60,11 @@ impl WaybarModule for PowerModule {
             }
         };
 
-        let capacity_str = fs::read_to_string(bat_path.join("capacity")).unwrap_or_else(|_| "0".to_string());
+        let capacity_str =
+            fs::read_to_string(bat_path.join("capacity")).unwrap_or_else(|_| "0".to_string());
         let percentage: u8 = capacity_str.trim().parse().unwrap_or(0);
-        let status_str = fs::read_to_string(bat_path.join("status")).unwrap_or_else(|_| "Unknown".to_string());
+        let status_str =
+            fs::read_to_string(bat_path.join("status")).unwrap_or_else(|_| "Unknown".to_string());
         let state = status_str.trim().to_lowercase();
 
         let (icon, class, tooltip) = if state == "charging" || ac_online {
@@ -94,8 +96,8 @@ impl WaybarModule for PowerModule {
             &config.power.format,
             &[
                 ("percentage", TokenValue::Int(percentage as i64)),
-                ("icon", TokenValue::String(icon)),
-            ]
+                ("icon", TokenValue::String(icon.to_string())),
+            ],
         );
 
         Ok(WaybarOutput {

src/modules/sys.rs

@@ -2,7 +2,7 @@ use crate::config::Config;
 use crate::modules::WaybarModule;
 use crate::output::WaybarOutput;
 use crate::state::SharedState;
-use crate::utils::{format_template, TokenValue};
+use crate::utils::{TokenValue, format_template};
 use anyhow::Result;
 
 pub struct SysModule;
@@ -34,11 +34,11 @@ impl WaybarModule for SysModule {
         let text = format_template(
             &config.sys.format,
             &[
-                ("uptime", TokenValue::String(&uptime_str)),
+                ("uptime", TokenValue::String(uptime_str.clone())),
                 ("load1", TokenValue::Float(load1)),
                 ("load5", TokenValue::Float(load5)),
                 ("load15", TokenValue::Float(load15)),
-            ]
+            ],
         );
 
         Ok(WaybarOutput {

src/output.rs

@@ -1,6 +1,6 @@
 use serde::Serialize;
 
-#[derive(Serialize)]
+#[derive(Serialize, Default)]
 pub struct WaybarOutput {
     pub text: String,
     #[serde(skip_serializing_if = "Option::is_none")]
@@ -10,3 +10,55 @@ pub struct WaybarOutput {
     #[serde(skip_serializing_if = "Option::is_none")]
     pub percentage: Option<u8>,
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn test_full_output_serialization() {
+        let output = WaybarOutput {
+            text: "CPU: 50%".to_string(),
+            tooltip: Some("Details".to_string()),
+            class: Some("normal".to_string()),
+            percentage: Some(50),
+        };
+        let json = serde_json::to_string(&output).unwrap();
+        let val: serde_json::Value = serde_json::from_str(&json).unwrap();
+        assert_eq!(val["text"], "CPU: 50%");
+        assert_eq!(val["tooltip"], "Details");
+        assert_eq!(val["class"], "normal");
+        assert_eq!(val["percentage"], 50);
+    }
+
+    #[test]
+    fn test_optional_fields_omitted() {
+        let output = WaybarOutput {
+            text: "test".to_string(),
+            tooltip: None,
+            class: None,
+            percentage: None,
+        };
+        let json = serde_json::to_string(&output).unwrap();
+        let val: serde_json::Value = serde_json::from_str(&json).unwrap();
+        assert_eq!(val["text"], "test");
+        assert!(val.get("tooltip").is_none());
+        assert!(val.get("class").is_none());
+        assert!(val.get("percentage").is_none());
+    }
+
+    #[test]
+    fn test_partial_optional_fields() {
+        let output = WaybarOutput {
+            text: "test".to_string(),
+            tooltip: Some("tip".to_string()),
+            class: None,
+            percentage: Some(75),
+        };
+        let json = serde_json::to_string(&output).unwrap();
+        let val: serde_json::Value = serde_json::from_str(&json).unwrap();
+        assert_eq!(val["tooltip"], "tip");
+        assert!(val.get("class").is_none());
+        assert_eq!(val["percentage"], 75);
+    }
+}

src/state.rs

@@ -7,12 +7,57 @@ pub struct AppState {
     pub memory: MemoryState,
     pub sys: SysState,
     pub gpu: GpuState,
+    pub disks: Vec<DiskInfo>,
+    pub bluetooth: BtState,
+    pub audio: AudioState,
+}
+
+#[derive(Default, Clone)]
+pub struct AudioState {
+    pub sink: AudioDeviceInfo,
+    pub source: AudioSourceInfo,
+}
+
+#[derive(Default, Clone)]
+pub struct AudioDeviceInfo {
+    pub name: String,
+    pub description: String,
+    pub volume: u8,
+    pub muted: bool,
+}
+
+#[derive(Default, Clone)]
+pub struct AudioSourceInfo {
+    pub name: String,
+    pub description: String,
+    pub volume: u8,
+    pub muted: bool,
+}
+
+#[derive(Default, Clone)]
+pub struct BtState {
+    pub connected: bool,
+    pub adapter_powered: bool,
+    pub device_alias: String,
+    pub device_address: String,
+    pub battery_percentage: Option<u8>,
+    pub plugin_data: Vec<(String, String)>,
+}
+
+#[derive(Default, Clone)]
+pub struct DiskInfo {
+    pub mount_point: String,
+    pub filesystem: String,
+    pub total_bytes: u64,
+    pub available_bytes: u64,
 }
 
 #[derive(Default, Clone)]
 pub struct NetworkState {
     pub rx_mbps: f64,
     pub tx_mbps: f64,
+    pub interface: String,
+    pub ip: String,
 }
 
 #[derive(Clone)]
@@ -73,3 +118,8 @@ impl Default for GpuState {
 }
 
 pub type SharedState = Arc<RwLock<AppState>>;
+
+#[cfg(test)]
+pub fn mock_state(state: AppState) -> SharedState {
+    Arc::new(RwLock::new(state))
+}

src/utils.rs

@@ -14,7 +14,9 @@ pub fn show_menu(prompt: &str, items: &[String], menu_cmd: &str) -> Result<Strin
 
     if let Some(mut stdin) = child.stdin.take() {
         let input = items.join("\n");
-        stdin.write_all(input.as_bytes()).context("Failed to write to menu stdin")?;
+        stdin
+            .write_all(input.as_bytes())
+            .context("Failed to write to menu stdin")?;
     }
 
     let output = child.wait_with_output().context("Failed to wait on menu")?;
@@ -34,23 +36,31 @@ pub fn show_menu(prompt: &str, items: &[String], menu_cmd: &str) -> Result<Strin
 use regex::Regex;
 use std::sync::LazyLock;
 
-pub enum TokenValue<'a> {
+pub enum TokenValue {
     Float(f64),
     Int(i64),
-    String(&'a str),
+    String(String),
 }
 
-pub fn format_template(template: &str, values: &[(&str, TokenValue)]) -> String {
+pub fn format_template<K>(template: &str, values: &[(K, TokenValue)]) -> String
+where
+    K: AsRef<str>,
+{
     static RE: LazyLock<Regex> = LazyLock::new(|| {
         Regex::new(r"\{([a-zA-Z0-9_]+)(?::([<>\^])?(\d+)?(?:\.(\d+))?)?\}").unwrap()
     });
 
     RE.replace_all(template, |caps: &regex::Captures| {
         let name = &caps[1];
-        if let Some((_, val)) = values.iter().find(|(k, _)| *k == name) {
+        if let Some((_, val)) = values.iter().find(|(k, _)| k.as_ref() == name) {
             let align = caps.get(2).map(|m| m.as_str()).unwrap_or(">");
-            let width = caps.get(3).map(|m| m.as_str().parse::<usize>().unwrap_or(0)).unwrap_or(0);
-            let precision = caps.get(4).map(|m| m.as_str().parse::<usize>().unwrap_or(0));
+            let width = caps
+                .get(3)
+                .map(|m| m.as_str().parse::<usize>().unwrap_or(0))
+                .unwrap_or(0);
+            let precision = caps
+                .get(4)
+                .map(|m| m.as_str().parse::<usize>().unwrap_or(0));
 
             match val {
                 TokenValue::Float(f) => format_float(*f, align, width, precision),
@@ -60,35 +70,151 @@ pub fn format_template(template: &str, values: &[(&str, TokenValue)]) -> String
         } else {
             caps[0].to_string()
         }
-    }).into_owned()
+    })
+    .into_owned()
 }
 
 fn format_float(f: f64, align: &str, width: usize, precision: Option<usize>) -> String {
     match (align, precision) {
-        ("<", Some(p)) => format!("{:<width$.p$}", f, width=width, p=p),
-        ("^", Some(p)) => format!("{:^width$.p$}", f, width=width, p=p),
-        (">", Some(p)) => format!("{:>width$.p$}", f, width=width, p=p),
-        ("<", None) => format!("{:<width$}", f, width=width),
-        ("^", None) => format!("{:^width$}", f, width=width),
-        (">", None) => format!("{:>width$}", f, width=width),
+        ("<", Some(p)) => format!("{:<width$.p$}", f, width = width, p = p),
+        ("^", Some(p)) => format!("{:^width$.p$}", f, width = width, p = p),
+        (">", Some(p)) => format!("{:>width$.p$}", f, width = width, p = p),
+        ("<", None) => format!("{:<width$}", f, width = width),
+        ("^", None) => format!("{:^width$}", f, width = width),
+        (">", None) => format!("{:>width$}", f, width = width),
         _ => format!("{}", f),
     }
 }
 
 fn format_int(i: i64, align: &str, width: usize) -> String {
     match align {
-        "<" => format!("{:<width$}", i, width=width),
-        "^" => format!("{:^width$}", i, width=width),
-        ">" => format!("{:>width$}", i, width=width),
+        "<" => format!("{:<width$}", i, width = width),
+        "^" => format!("{:^width$}", i, width = width),
+        ">" => format!("{:>width$}", i, width = width),
         _ => format!("{}", i),
     }
 }
 
 fn format_str(s: &str, align: &str, width: usize) -> String {
     match align {
-        "<" => format!("{:<width$}", s, width=width),
-        "^" => format!("{:^width$}", s, width=width),
-        ">" => format!("{:>width$}", s, width=width),
-        _ => format!("{}", s),
+        "<" => format!("{:<width$}", s, width = width),
+        "^" => format!("{:^width$}", s, width = width),
+        ">" => format!("{:>width$}", s, width = width),
+        _ => s.to_string(),
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn test_simple_string_token() {
+        let result = format_template(
+            "{name}",
+            &[("name", TokenValue::String("hello".to_string()))],
+        );
+        assert_eq!(result, "hello");
+    }
+
+    #[test]
+    fn test_simple_float_token() {
+        let result = format_template("{val}", &[("val", TokenValue::Float(3.15))]);
+        assert_eq!(result, "3.15");
+    }
+
+    #[test]
+    fn test_simple_int_token() {
+        let result = format_template("{count}", &[("count", TokenValue::Int(42))]);
+        assert_eq!(result, "42");
+    }
+
+    #[test]
+    fn test_float_right_align_with_precision() {
+        let result = format_template("{val:>8.2}", &[("val", TokenValue::Float(3.15))]);
+        assert_eq!(result, "    3.15");
+        assert_eq!(result.len(), 8);
+    }
+
+    #[test]
+    fn test_float_left_align_with_precision() {
+        let result = format_template("{val:<8.2}", &[("val", TokenValue::Float(3.15))]);
+        assert_eq!(result, "3.15    ");
+        assert_eq!(result.len(), 8);
+    }
+
+    #[test]
+    fn test_float_center_align_with_precision() {
+        let result = format_template("{val:^8.2}", &[("val", TokenValue::Float(3.15))]);
+        assert_eq!(result, "  3.15  ");
+        assert_eq!(result.len(), 8);
+    }
+
+    #[test]
+    fn test_int_right_align() {
+        let result = format_template("{val:>5}", &[("val", TokenValue::Int(42))]);
+        assert_eq!(result, "   42");
+    }
+
+    #[test]
+    fn test_string_left_align() {
+        let result = format_template(
+            "{val:<10}",
+            &[("val", TokenValue::String("hi".to_string()))],
+        );
+        assert_eq!(result, "hi        ");
+        assert_eq!(result.len(), 10);
+    }
+
+    #[test]
+    fn test_unknown_token_preserved() {
+        let result = format_template(
+            "{unknown}",
+            &[("name", TokenValue::String("test".to_string()))],
+        );
+        assert_eq!(result, "{unknown}");
+    }
+
+    #[test]
+    fn test_multiple_tokens() {
+        let result = format_template(
+            "CPU: {usage:>4.1}% {temp:>4.1}C",
+            &[
+                ("usage", TokenValue::Float(55.3)),
+                ("temp", TokenValue::Float(65.0)),
+            ],
+        );
+        assert_eq!(result, "CPU: 55.3% 65.0C");
+    }
+
+    #[test]
+    fn test_no_tokens() {
+        let result = format_template::<&'static str>("plain text", &[]);
+        assert_eq!(result, "plain text");
+    }
+
+    #[test]
+    fn test_empty_template() {
+        let result = format_template("", &[("x", TokenValue::Int(1))]);
+        assert_eq!(result, "");
+    }
+
+    #[test]
+    fn test_mixed_token_types() {
+        let result = format_template(
+            "{name} ({ip}): {rx:>5.2} MB/s",
+            &[
+                ("name", TokenValue::String("eth0".to_string())),
+                ("ip", TokenValue::String("10.0.0.1".to_string())),
+                ("rx", TokenValue::Float(1.5)),
+            ],
+        );
+        assert_eq!(result, "eth0 (10.0.0.1):  1.50 MB/s");
+    }
+
+    #[test]
+    fn test_float_precision_zero() {
+        let result = format_template("{val:>3.0}", &[("val", TokenValue::Float(99.7))]);
+        assert_eq!(result, "100");
     }
 }