5 Commits

Author SHA1 Message Date
nvrl a932e6b422 updated documentation + readme and seperated bt modules
CI / Lint and Test (pull_request) Successful in 2m11s
CI / Version Check (pull_request) Failing after 3s
2026-04-01 18:16:22 +02:00
nvrl 81c9b78cb3 improved robustness of pixel buds plugin 2026-04-01 17:54:16 +02:00
nvrl c8a50d7b67 fixed ci with release fixes
CI / Lint and Test (pull_request) Successful in 2m8s
CI / Version Check (pull_request) Successful in 3s
2026-04-01 16:51:44 +02:00
nvrl e57b69a735 version bump
CI / Lint and Test (pull_request) Failing after 28s
CI / Version Check (pull_request) Successful in 3s
2026-04-01 16:49:33 +02:00
nvrl ed0051f2c9 migrated daemon to tokio with seperate hardware threads + thiserror 2026-04-01 16:49:03 +02:00
25 changed files with 1306 additions and 1013 deletions
+2
View File
@@ -19,6 +19,8 @@ jobs:
toolchain: stable toolchain: stable
components: rustfmt, clippy components: rustfmt, clippy
cache: false cache: false
- name: Install packaging tools
run: apt-get update && apt-get install -y dpkg-dev pkg-config libdbus-1-dev protobuf-compiler libpulse-dev
- name: Check formatting - name: Check formatting
run: cargo fmt --check run: cargo fmt --check
Generated
+47 -15
View File
@@ -353,7 +353,7 @@ checksum = "1d674e81391d1e1ab681a28d99df07927c6d4aa5b027d7da16ba32d1d21ecd99"
[[package]] [[package]]
name = "fluxo-rs" name = "fluxo-rs"
version = "0.3.0" version = "0.3.1"
dependencies = [ dependencies = [
"anyhow", "anyhow",
"bluer", "bluer",
@@ -362,12 +362,13 @@ dependencies = [
"futures", "futures",
"libpulse-binding", "libpulse-binding",
"maestro", "maestro",
"nix 0.29.0", "nix 0.31.2",
"regex", "regex",
"serde", "serde",
"serde_json", "serde_json",
"sysinfo", "sysinfo",
"tempfile", "tempfile",
"thiserror",
"tokio", "tokio",
"tokio-util", "tokio-util",
"toml", "toml",
@@ -584,9 +585,9 @@ checksum = "09edd9e8b54e49e587e4f6295a7d29c3ea94d469cb40ab8ca70b288248a81db2"
[[package]] [[package]]
name = "libc" name = "libc"
version = "0.2.183" version = "0.2.184"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "b5b646652bf6661599e1da8901b3b9522896f01e736bad5f723fe7a3a27f899d" checksum = "48f5d2a454e16a5ea0f4ced81bd44e4cfc7bd3a507b61887c99fd3538b28e4af"
[[package]] [[package]]
name = "libdbus-sys" name = "libdbus-sys"
@@ -710,7 +711,6 @@ dependencies = [
"cfg-if", "cfg-if",
"cfg_aliases", "cfg_aliases",
"libc", "libc",
"memoffset",
] ]
[[package]] [[package]]
@@ -723,6 +723,7 @@ dependencies = [
"cfg-if", "cfg-if",
"cfg_aliases", "cfg_aliases",
"libc", "libc",
"memoffset",
] ]
[[package]] [[package]]
@@ -1067,9 +1068,9 @@ dependencies = [
[[package]] [[package]]
name = "serde_spanned" name = "serde_spanned"
version = "1.1.0" version = "1.1.1"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "876ac351060d4f882bb1032b6369eb0aef79ad9df1ea8bc404874d8cc3d0cd98" checksum = "6662b5879511e06e8999a8a235d848113e942c9124f211511b16466ee2995f26"
dependencies = [ dependencies = [
"serde_core", "serde_core",
] ]
@@ -1083,6 +1084,16 @@ dependencies = [
"lazy_static", "lazy_static",
] ]
[[package]]
name = "signal-hook-registry"
version = "1.4.8"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "c4db69cba1110affc0e9f7bcd48bbf87b3f4fc7c61fc9155afd4c469eb3d6c1b"
dependencies = [
"errno",
"libc",
]
[[package]] [[package]]
name = "slab" name = "slab"
version = "0.4.12" version = "0.4.12"
@@ -1182,6 +1193,26 @@ dependencies = [
"windows-sys 0.61.2", "windows-sys 0.61.2",
] ]
[[package]]
name = "thiserror"
version = "2.0.18"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "4288b5bcbc7920c07a1149a35cf9590a2aa808e0bc1eafaade0b80947865fbc4"
dependencies = [
"thiserror-impl",
]
[[package]]
name = "thiserror-impl"
version = "2.0.18"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "ebc4ee7f67670e9b64d05fa4253e753e016c6c95ff35b89b7941d6b856dec1d5"
dependencies = [
"proc-macro2",
"quote",
"syn",
]
[[package]] [[package]]
name = "thread_local" name = "thread_local"
version = "1.1.9" version = "1.1.9"
@@ -1201,6 +1232,7 @@ dependencies = [
"libc", "libc",
"mio", "mio",
"pin-project-lite", "pin-project-lite",
"signal-hook-registry",
"socket2", "socket2",
"tokio-macros", "tokio-macros",
"windows-sys 0.61.2", "windows-sys 0.61.2",
@@ -1243,9 +1275,9 @@ dependencies = [
[[package]] [[package]]
name = "toml" name = "toml"
version = "1.1.0+spec-1.1.0" version = "1.1.1+spec-1.1.0"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "f8195ca05e4eb728f4ba94f3e3291661320af739c4e43779cbdfae82ab239fcc" checksum = "994b95d9e7bae62b34bab0e2a4510b801fa466066a6a8b2b57361fa1eba068ee"
dependencies = [ dependencies = [
"indexmap", "indexmap",
"serde_core", "serde_core",
@@ -1258,9 +1290,9 @@ dependencies = [
[[package]] [[package]]
name = "toml_datetime" name = "toml_datetime"
version = "1.1.0+spec-1.1.0" version = "1.1.1+spec-1.1.0"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "97251a7c317e03ad83774a8752a7e81fb6067740609f75ea2b585b569a59198f" checksum = "3165f65f62e28e0115a00b2ebdd37eb6f3b641855f9d636d3cd4103767159ad7"
dependencies = [ dependencies = [
"serde_core", "serde_core",
] ]
@@ -1279,18 +1311,18 @@ dependencies = [
[[package]] [[package]]
name = "toml_parser" name = "toml_parser"
version = "1.1.0+spec-1.1.0" version = "1.1.1+spec-1.1.0"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "2334f11ee363607eb04df9b8fc8a13ca1715a72ba8662a26ac285c98aabb4011" checksum = "39ca317ebc49f06bd748bfba29533eac9485569dc9bf80b849024b025e814fb9"
dependencies = [ dependencies = [
"winnow", "winnow",
] ]
[[package]] [[package]]
name = "toml_writer" name = "toml_writer"
version = "1.1.0+spec-1.1.0" version = "1.1.1+spec-1.1.0"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "d282ade6016312faf3e41e57ebbba0c073e4056dab1232ab1cb624199648f8ed" checksum = "756daf9b1013ebe47a8776667b466417e2d4c5679d441c26230efd9ef78692db"
[[package]] [[package]]
name = "tracing" name = "tracing"
+6 -5
View File
@@ -1,6 +1,6 @@
[package] [package]
name = "fluxo-rs" name = "fluxo-rs"
version = "0.3.0" version = "0.3.1"
edition = "2024" edition = "2024"
[dependencies] [dependencies]
@@ -11,16 +11,17 @@ regex = "1.12"
serde = { version = "1.0.228", features = ["derive"] } serde = { version = "1.0.228", features = ["derive"] }
serde_json = "1.0.149" serde_json = "1.0.149"
sysinfo = "0.38.4" sysinfo = "0.38.4"
toml = "1.0.6" thiserror = "2.0"
toml = "1.1.1"
tracing = "0.1.44" tracing = "0.1.44"
tracing-subscriber = { version = "0.3.23", features = ["env-filter"] } tracing-subscriber = { version = "0.3.23", features = ["env-filter"] }
maestro = { git = "https://github.com/qzed/pbpctrl", package = "maestro" } maestro = { git = "https://github.com/qzed/pbpctrl", package = "maestro" }
bluer = { version = "0.17", features = ["bluetoothd", "rfcomm", "id"] } bluer = { version = "0.17", features = ["bluetoothd", "rfcomm", "id"] }
tokio = { version = "1", features = ["rt", "rt-multi-thread", "sync", "time", "macros"] } tokio = { version = "1", features = ["rt", "rt-multi-thread", "sync", "time", "macros", "signal"] }
tokio-util = { version = "0.7", features = ["codec"] } tokio-util = { version = "0.7", features = ["codec"] }
futures = "0.3" futures = "0.3"
libpulse-binding = "2.26" libpulse-binding = "2.30"
nix = { version = "0.29", features = ["net"] } nix = { version = "0.31", features = ["net"] }
[dev-dependencies] [dev-dependencies]
+43 -72
View File
@@ -1,95 +1,66 @@
# fluxo-rs # fluxo-rs
fluxo-rs is a high-performance system metrics daemon and client designed specifically for waybar. It replaces standard shell scripts with a compiled rust binary that collects data via a background polling loop and serves it over a unix domain socket (`/tmp/fluxo.sock`). fluxo-rs is a high-performance system metrics daemon and client designed specifically for Waybar. It replaces standard shell scripts with a compiled Rust binary that collects data via a background polling loop and serves it over a Unix socket.
## Description ## Key Features
The project follows a client-server architecture: - **Asynchronous Architecture**: Built on **Tokio**, the daemon handles concurrent IPC requests and background tasks with zero latency and minimal CPU overhead.
- **Daemon**: Handles heavy lifting (polling cpu, memory, network, gpu) and stores state in memory. - **Native Library Integrations**:
- **Client**: A thin cli wrapper that connects to the daemon's socket to retrieve formatted json for waybar. - **Audio**: Direct `libpulse` integration for event-driven, instant volume and device updates.
- **Bluetooth**: Native `bluer` integration for robust device monitoring.
This approach eliminates process spawning overhead and temporary file locking, resulting in near-zero cpu usage for custom modules. - **Pixel Buds Pro**: Custom native RPC implementation for real-time battery and ANC control.
- **Network**: Native `nix` and `/proc` inspection for high-speed interface monitoring.
## Features - **Hyprland**: Direct IPC Unix socket communication for gamemode and animation status.
- **Circuit Breaker (Failsafe)**: Automatically detects failing modules and enters a "Cool down" state to prevent resource waste and log spam.
- **Ultra-lightweight**: Background polling is highly optimized (e.g., O(1) process counting). - **Multi-threaded Polling**: Decoupled subsystem threads ensure that a hang in one system (e.g., a slow GPU probe) never freezes your entire bar.
- **Jitter-free**: Uses zero-width sentinels and figure spaces to prevent waybar from trimming padding.
- **Configurable**: Fully customizable output formats via toml config.
- **Interactive Menus**: Integrated support for selecting items (like Bluetooth devices) via external menus (e.g., Rofi, Wofi, Fuzzel).
- **Live Reload**: Configuration can be reloaded without restarting the daemon.
- **Multi-vendor GPU**: Native support for intel (igpu), amd, and nvidia.
## Modules ## Modules
| Command | Description | Tokens | | Command | Description | Tokens |
| :--- | :--- | :--- | | :--- | :--- | :--- |
| `net` | Network speed (rx/tx) | `{interface}`, `{ip}`, `{rx}`, `{tx}` | | `cpu` | CPU usage and temperature | `{usage}`, `{temp}`, `{model}` |
| `cpu` | CPU usage and temp | `{usage}`, `{temp}` |
| `mem` | Memory usage | `{used}`, `{total}` | | `mem` | Memory usage | `{used}`, `{total}` |
| `gpu` | GPU utilization | `{usage}`, `{vram_used}`, `{vram_total}`, `{temp}` | | `net` | Network status & speeds | `{interface}`, `{ip}`, `{rx}`, `{tx}` |
| `sys` | System load and uptime | `{uptime}`, `{load1}`, `{load5}`, `{load15}` | | `sys` | System load and uptime | `{uptime}`, `{load1}`, `{load5}`, `{load15}`, `{procs}` |
| `disk` | Disk usage (default: /) | `{mount}`, `{used}`, `{total}` | | `disk` | Disk usage | `{mount}`, `{used}`, `{total}` |
| `pool` | Aggregate storage (btrfs) | `{used}`, `{total}` | | `pool` | Btrfs aggregate storage | `{used}`, `{total}` |
| `vol` | Audio output volume | `{name}`, `{volume}`, `{icon}` | | `vol` | Audio output (sink) | `{name}`, `{volume}`, `{icon}` |
| `mic` | Audio input volume | `{name}`, `{volume}`, `{icon}` | | `mic` | Audio input (source) | `{name}`, `{volume}`, `{icon}` |
| `bt` | Bluetooth status & plugins | `{alias}`, `{mac}`, `{left}`, `{right}`, `{anc}` | | `bt` | Bluetooth status & plugins | `{alias}`, `{mac}`, `{left}`, `{right}`, `{anc}` |
| `power` | Battery and AC status | `{percentage}`, `{icon}` | | `power` | Battery and AC status | `{percentage}`, `{icon}` |
| `game` | Hyprland gamemode status | active/inactive icon strings | | `game` | Hyprland status | active/inactive icons |
## Setup ## Setup
1. Build the project: 1. **Build**: `cargo build --release`
```bash 2. **Configure**: Create `~/.config/fluxo/config.toml` (see `example.config.toml`).
cd fluxo-rs 3. **Daemon**: Start `fluxo-rs daemon`. It's recommended to run this as a systemd user service.
cargo build --release
```
2. Start the daemon: ## Waybar Configuration
```bash
# Starts the daemon using the default config path (~/.config/fluxo/config.toml)
./target/release/fluxo-rs daemon &
# Or provide a custom path To achieve zero-latency updates, use **Waybar Signals**:
./target/release/fluxo-rs daemon --config /path/to/your/config.toml &
```
3. Configuration: ```jsonc
Create `~/.config/fluxo/config.toml` (see `example.config.toml` for all default options). "custom/audio": {
"exec": "fluxo vol",
4. Waybar configuration (`config.jsonc`): "return-type": "json",
```json "interval": 5,
"custom/cpu": { "signal": 8,
"exec": "~/path/to/fluxo-rs cpu", "on-click": "fluxo audio cycle sink && pkill -RTMIN+8 waybar"
"return-type": "json" },
} "custom/bluetooth": {
``` "exec": "fluxo bt",
"return-type": "json",
## Development "interval": 5,
"signal": 9,
### Architecture "on-click": "fluxo bt menu && pkill -RTMIN+9 waybar",
- `src/main.rs`: Entry point, CLI parsing, interactive GUI spawns (menus), and client-side formatting logic. "on-click-right": "fluxo bt cycle_mode && pkill -RTMIN+9 waybar"
- `src/daemon.rs`: UDS listener, configuration management, and polling orchestration. }
- `src/ipc.rs`: Unix domain socket communication logic.
- `src/utils.rs`: Generic GUI utilities (like the menu spawner).
- `src/modules/`: Individual metric implementations.
- `src/state.rs`: Shared thread-safe data structures.
### Adding a Module
1. Add the required config block to `src/config.rs`.
2. Add the required state fields to `src/state.rs`.
3. Implement the `WaybarModule` trait in a new file in `src/modules/`.
4. Add polling logic to `src/modules/hardware.rs` or `src/daemon.rs`.
5. Register the new subcommand in `src/main.rs` and the router in `src/daemon.rs`.
### Configuration Reload
The daemon can reload its configuration live:
```bash
fluxo-rs reload
``` ```
### Logging & Debugging ## Debugging
`fluxo-rs` uses the `tracing` ecosystem. If a module isn't behaving properly or your configuration isn't applying, start the daemon with debug logging enabled in the foreground:
Start the daemon with `RUST_LOG=debug` to see detailed logs of library interactions and circuit breaker status:
```bash ```bash
RUST_LOG=debug fluxo-rs daemon RUST_LOG=debug fluxo-rs daemon
``` ```
This will print verbose information regarding config parsing errors, subprocess failures, and IPC requests.
+13 -13
View File
@@ -5,7 +5,7 @@ use std::path::PathBuf;
use std::sync::LazyLock; use std::sync::LazyLock;
use tracing::{debug, info, warn}; use tracing::{debug, info, warn};
#[derive(Deserialize, Default)] #[derive(Deserialize, Default, Clone)]
pub struct Config { pub struct Config {
#[serde(default)] #[serde(default)]
pub general: GeneralConfig, pub general: GeneralConfig,
@@ -33,7 +33,7 @@ pub struct Config {
pub game: GameConfig, pub game: GameConfig,
} }
#[derive(Deserialize)] #[derive(Deserialize, Clone)]
pub struct GeneralConfig { pub struct GeneralConfig {
pub menu_command: String, pub menu_command: String,
} }
@@ -46,7 +46,7 @@ impl Default for GeneralConfig {
} }
} }
#[derive(Deserialize)] #[derive(Deserialize, Clone)]
pub struct NetworkConfig { pub struct NetworkConfig {
pub format: String, pub format: String,
} }
@@ -59,7 +59,7 @@ impl Default for NetworkConfig {
} }
} }
#[derive(Deserialize)] #[derive(Deserialize, Clone)]
pub struct CpuConfig { pub struct CpuConfig {
pub format: String, pub format: String,
} }
@@ -72,7 +72,7 @@ impl Default for CpuConfig {
} }
} }
#[derive(Deserialize)] #[derive(Deserialize, Clone)]
pub struct MemoryConfig { pub struct MemoryConfig {
pub format: String, pub format: String,
} }
@@ -85,7 +85,7 @@ impl Default for MemoryConfig {
} }
} }
#[derive(Deserialize)] #[derive(Deserialize, Clone)]
pub struct GpuConfig { pub struct GpuConfig {
pub format_amd: String, pub format_amd: String,
pub format_intel: String, pub format_intel: String,
@@ -104,7 +104,7 @@ impl Default for GpuConfig {
} }
} }
#[derive(Deserialize)] #[derive(Deserialize, Clone)]
pub struct SysConfig { pub struct SysConfig {
pub format: String, pub format: String,
} }
@@ -117,7 +117,7 @@ impl Default for SysConfig {
} }
} }
#[derive(Deserialize)] #[derive(Deserialize, Clone)]
pub struct DiskConfig { pub struct DiskConfig {
pub format: String, pub format: String,
} }
@@ -130,7 +130,7 @@ impl Default for DiskConfig {
} }
} }
#[derive(Deserialize)] #[derive(Deserialize, Clone)]
pub struct PoolConfig { pub struct PoolConfig {
pub format: String, pub format: String,
} }
@@ -143,7 +143,7 @@ impl Default for PoolConfig {
} }
} }
#[derive(Deserialize)] #[derive(Deserialize, Clone)]
pub struct PowerConfig { pub struct PowerConfig {
pub format: String, pub format: String,
} }
@@ -156,7 +156,7 @@ impl Default for PowerConfig {
} }
} }
#[derive(Deserialize)] #[derive(Deserialize, Clone)]
pub struct AudioConfig { pub struct AudioConfig {
pub format_sink_unmuted: String, pub format_sink_unmuted: String,
pub format_sink_muted: String, pub format_sink_muted: String,
@@ -175,7 +175,7 @@ impl Default for AudioConfig {
} }
} }
#[derive(Deserialize)] #[derive(Deserialize, Clone)]
pub struct BtConfig { pub struct BtConfig {
pub format_connected: String, pub format_connected: String,
pub format_plugin: String, pub format_plugin: String,
@@ -194,7 +194,7 @@ impl Default for BtConfig {
} }
} }
#[derive(Deserialize)] #[derive(Deserialize, Clone)]
pub struct GameConfig { pub struct GameConfig {
pub format_active: String, pub format_active: String,
pub format_inactive: String, pub format_inactive: String,
+165 -98
View File
@@ -1,4 +1,5 @@
use crate::config::Config; use crate::config::Config;
use crate::error::FluxoError;
use crate::ipc::socket_path; use crate::ipc::socket_path;
use crate::modules::WaybarModule; use crate::modules::WaybarModule;
use crate::modules::audio::AudioDaemon; use crate::modules::audio::AudioDaemon;
@@ -8,14 +9,12 @@ use crate::modules::network::NetworkDaemon;
use crate::state::{AppState, SharedState}; use crate::state::{AppState, SharedState};
use anyhow::Result; use anyhow::Result;
use std::fs; 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::path::PathBuf;
use std::sync::atomic::{AtomicBool, Ordering}; use std::sync::Arc;
use std::sync::{Arc, RwLock}; use tokio::io::{AsyncBufReadExt, AsyncWriteExt, BufReader};
use std::thread; use tokio::net::UnixListener;
use std::time::Duration; use tokio::sync::RwLock;
use tokio::time::{Duration, Instant, sleep};
use tracing::{debug, error, info, warn}; use tracing::{debug, error, info, warn};
struct SocketGuard { struct SocketGuard {
@@ -29,7 +28,7 @@ impl Drop for SocketGuard {
} }
} }
pub fn run_daemon(config_path: Option<PathBuf>) -> Result<()> { pub async fn run_daemon(config_path: Option<PathBuf>) -> Result<()> {
let sock_path = socket_path(); let sock_path = socket_path();
if fs::metadata(&sock_path).is_ok() { if fs::metadata(&sock_path).is_ok() {
@@ -43,89 +42,89 @@ pub fn run_daemon(config_path: Option<PathBuf>) -> Result<()> {
path: sock_path.clone(), path: sock_path.clone(),
}; };
// Signal handling: set flag so main loop exits cleanly // Signal handling for graceful shutdown in async context
let running = Arc::new(AtomicBool::new(true)); let running = Arc::new(tokio::sync::Notify::new());
let running_clone = Arc::clone(&running); let r_clone = Arc::clone(&running);
ctrlc::set_handler(move || { tokio::spawn(async move {
tokio::signal::ctrl_c().await.unwrap();
info!("Received shutdown signal, exiting..."); info!("Received shutdown signal, exiting...");
running_clone.store(false, Ordering::SeqCst); r_clone.notify_waiters();
})?; });
// We store the original config_path to allow proper reloading later
let config_path_clone = config_path.clone(); let config_path_clone = config_path.clone();
let config = Arc::new(RwLock::new(crate::config::load_config(config_path))); let config = Arc::new(RwLock::new(crate::config::load_config(config_path)));
// 1. Network Thread // 1. Network Task
let poll_state = Arc::clone(&state); let poll_state = Arc::clone(&state);
let poll_running = Arc::clone(&running); tokio::spawn(async move {
thread::spawn(move || { info!("Starting Network polling task");
info!("Starting Network polling thread");
let mut daemon = NetworkDaemon::new(); let mut daemon = NetworkDaemon::new();
while poll_running.load(Ordering::SeqCst) { loop {
daemon.poll(Arc::clone(&poll_state)); daemon.poll(Arc::clone(&poll_state)).await;
thread::sleep(Duration::from_secs(1)); sleep(Duration::from_secs(1)).await;
} }
}); });
// 2. Fast Hardware Thread (CPU, Mem, Load) // 2. Fast Hardware Task (CPU, Mem, Load)
let poll_state = Arc::clone(&state); let poll_state = Arc::clone(&state);
let poll_running = Arc::clone(&running); tokio::spawn(async move {
thread::spawn(move || { info!("Starting Fast Hardware polling task");
info!("Starting Fast Hardware polling thread");
let mut daemon = HardwareDaemon::new(); let mut daemon = HardwareDaemon::new();
while poll_running.load(Ordering::SeqCst) { loop {
daemon.poll_fast(Arc::clone(&poll_state)); daemon.poll_fast(Arc::clone(&poll_state)).await;
thread::sleep(Duration::from_secs(1)); sleep(Duration::from_secs(1)).await;
} }
}); });
// 3. Slow Hardware Thread (GPU, Disks) // 3. Slow Hardware Task (GPU, Disks)
let poll_state = Arc::clone(&state); let poll_state = Arc::clone(&state);
let poll_running = Arc::clone(&running); tokio::spawn(async move {
thread::spawn(move || { info!("Starting Slow Hardware polling task");
info!("Starting Slow Hardware polling thread");
let mut daemon = HardwareDaemon::new(); let mut daemon = HardwareDaemon::new();
while poll_running.load(Ordering::SeqCst) { loop {
daemon.poll_slow(Arc::clone(&poll_state)); daemon.poll_slow(Arc::clone(&poll_state)).await;
thread::sleep(Duration::from_secs(1)); sleep(Duration::from_secs(1)).await;
} }
}); });
// 4. Bluetooth Thread // 4. Bluetooth Task
let poll_state = Arc::clone(&state); let poll_state = Arc::clone(&state);
let poll_running = Arc::clone(&running); let poll_config = Arc::clone(&config);
thread::spawn(move || { tokio::spawn(async move {
info!("Starting Bluetooth polling thread"); info!("Starting Bluetooth polling task");
let mut daemon = BtDaemon::new(); let mut daemon = BtDaemon::new();
while poll_running.load(Ordering::SeqCst) { loop {
daemon.poll(Arc::clone(&poll_state)); let config = poll_config.read().await;
thread::sleep(Duration::from_secs(1)); daemon.poll(Arc::clone(&poll_state), &config).await;
sleep(Duration::from_secs(1)).await;
} }
}); });
// 5. Audio Thread (Event driven) // 5. Audio Thread (Event driven - pulse usually needs its own thread)
let audio_daemon = AudioDaemon::new(); let audio_daemon = AudioDaemon::new();
audio_daemon.start(Arc::clone(&state)); audio_daemon.start(Arc::clone(&state));
info!("Fluxo daemon successfully bound to socket: {}", sock_path); info!("Fluxo daemon successfully bound to socket: {}", sock_path);
// Use non-blocking accept so we can check the running flag loop {
listener.set_nonblocking(true)?; tokio::select! {
_ = running.notified() => {
while running.load(Ordering::SeqCst) { break;
match listener.accept() { }
res = listener.accept() => {
match res {
Ok((mut stream, _)) => { Ok((mut stream, _)) => {
let state_clone = Arc::clone(&state); let state_clone = Arc::clone(&state);
let config_clone = Arc::clone(&config); let config_clone = Arc::clone(&config);
let cp_clone = config_path_clone.clone(); let cp_clone = config_path_clone.clone();
thread::spawn(move || { tokio::spawn(async move {
let mut reader = BufReader::new(&stream); let (reader, mut writer) = stream.split();
let mut reader = BufReader::new(reader);
let mut request = String::new(); let mut request = String::new();
if let Err(e) = reader.read_line(&mut request) { if let Err(e) = reader.read_line(&mut request).await {
error!("Failed to read from IPC stream: {}", e); error!("Failed to read from IPC stream: {}", e);
return; return;
} }
drop(reader);
let request = request.trim(); let request = request.trim();
if request.is_empty() { if request.is_empty() {
@@ -137,21 +136,17 @@ pub fn run_daemon(config_path: Option<PathBuf>) -> Result<()> {
if *module_name == "reload" { if *module_name == "reload" {
info!("Reloading configuration..."); info!("Reloading configuration...");
let new_config = crate::config::load_config(cp_clone); let new_config = crate::config::load_config(cp_clone);
if let Ok(mut config_lock) = config_clone.write() { let mut config_lock = config_clone.write().await;
*config_lock = new_config; *config_lock = new_config;
let _ = stream.write_all(b"{\"text\":\"ok\"}"); let _ = writer.write_all(b"{\"text\":\"ok\"}").await;
info!("Configuration reloaded successfully."); info!("Configuration reloaded successfully.");
} else {
error!("Failed to acquire write lock for configuration reload.");
}
let _ = stream.shutdown(Shutdown::Write);
return; return;
} }
debug!(module = module_name, args = ?&parts[1..], "Handling IPC request"); debug!(module = module_name, args = ?&parts[1..], "Handling IPC request");
let response = let response =
handle_request(module_name, &parts[1..], &state_clone, &config_clone); handle_request(module_name, &parts[1..], &state_clone, &config_clone).await;
if let Err(e) = stream.write_all(response.as_bytes()) { if let Err(e) = writer.write_all(response.as_bytes()).await {
if e.kind() == std::io::ErrorKind::BrokenPipe if e.kind() == std::io::ErrorKind::BrokenPipe
|| e.kind() == std::io::ErrorKind::ConnectionReset || e.kind() == std::io::ErrorKind::ConnectionReset
{ {
@@ -163,68 +158,140 @@ pub fn run_daemon(config_path: Option<PathBuf>) -> Result<()> {
error!("Failed to write IPC response: {}", e); 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), Err(e) => error!("Failed to accept incoming connection: {}", e),
} }
} }
}
}
info!("Daemon shutting down gracefully."); info!("Daemon shutting down gracefully.");
Ok(()) Ok(())
} }
fn handle_request( async fn handle_request(
module_name: &str, module_name: &str,
args: &[&str], args: &[&str],
state: &SharedState, state: &SharedState,
config_lock: &Arc<RwLock<Config>>, config_lock: &Arc<RwLock<Config>>,
) -> String { ) -> String {
let config = if let Ok(c) = config_lock.read() { // 1. Check Circuit Breaker status
c let is_in_backoff = {
let lock = state.read().await;
if let Some(health) = lock.health.get(module_name) {
if let Some(until) = health.backoff_until {
Instant::now() < until
} else { } else {
error!("Failed to acquire read lock for configuration."); false
return "{\"text\":\"error: config lock failed\"}".to_string(); }
}; } else {
false
let result = match module_name {
"net" | "network" => crate::modules::network::NetworkModule.run(&config, state, args),
"cpu" => crate::modules::cpu::CpuModule.run(&config, state, args),
"mem" | "memory" => crate::modules::memory::MemoryModule.run(&config, state, args),
"disk" => crate::modules::disk::DiskModule.run(&config, state, args),
"pool" | "btrfs" => crate::modules::btrfs::BtrfsModule.run(&config, state, args),
"vol" => crate::modules::audio::AudioModule.run(
&config,
state,
&["sink", args.first().unwrap_or(&"show")],
),
"mic" => crate::modules::audio::AudioModule.run(
&config,
state,
&["source", args.first().unwrap_or(&"show")],
),
"gpu" => crate::modules::gpu::GpuModule.run(&config, state, args),
"sys" => crate::modules::sys::SysModule.run(&config, state, args),
"bt" | "bluetooth" => crate::modules::bt::BtModule.run(&config, state, args),
"power" => crate::modules::power::PowerModule.run(&config, state, args),
"game" => crate::modules::game::GameModule.run(&config, state, args),
_ => {
warn!("Received request for unknown module: '{}'", module_name);
Err(anyhow::anyhow!("Unknown module: {}", module_name))
} }
}; };
if is_in_backoff {
return format!(
"{{\"text\":\"\u{200B}Cooling down ({})\u{200B}\",\"class\":\"error\"}}",
module_name
);
}
let config = config_lock.read().await;
let result = match module_name {
"net" | "network" => {
crate::modules::network::NetworkModule
.run(&config, state, args)
.await
}
"cpu" => {
crate::modules::cpu::CpuModule
.run(&config, state, args)
.await
}
"mem" | "memory" => {
crate::modules::memory::MemoryModule
.run(&config, state, args)
.await
}
"disk" => {
crate::modules::disk::DiskModule
.run(&config, state, args)
.await
}
"pool" | "btrfs" => {
crate::modules::btrfs::BtrfsModule
.run(&config, state, args)
.await
}
"vol" => {
crate::modules::audio::AudioModule
.run(&config, state, &["sink", args.first().unwrap_or(&"show")])
.await
}
"mic" => {
crate::modules::audio::AudioModule
.run(&config, state, &["source", args.first().unwrap_or(&"show")])
.await
}
"gpu" => {
crate::modules::gpu::GpuModule
.run(&config, state, args)
.await
}
"sys" => {
crate::modules::sys::SysModule
.run(&config, state, args)
.await
}
"bt" | "bluetooth" => crate::modules::bt::BtModule.run(&config, state, args).await,
"power" => {
crate::modules::power::PowerModule
.run(&config, state, args)
.await
}
"game" => {
crate::modules::game::GameModule
.run(&config, state, args)
.await
}
_ => {
warn!("Received request for unknown module: '{}'", module_name);
Err(FluxoError::Ipc(format!("Unknown module: {}", module_name)))
}
};
// 2. Update Health based on result
{
let mut lock = state.write().await;
let health = lock.health.entry(module_name.to_string()).or_default();
match &result {
Ok(_) => {
health.consecutive_failures = 0;
health.backoff_until = None;
}
Err(e) => {
health.consecutive_failures += 1;
health.last_failure = Some(Instant::now());
if health.consecutive_failures >= 3 {
// Backoff for 30 seconds after 3 failures
health.backoff_until = Some(Instant::now() + Duration::from_secs(30));
warn!(module = module_name, error = %e, "Module entered backoff state due to repeated failures");
}
}
}
}
match result { match result {
Ok(output) => serde_json::to_string(&output).unwrap_or_else(|_| "{}".to_string()), Ok(output) => serde_json::to_string(&output).unwrap_or_else(|_| "{}".to_string()),
Err(e) => { Err(e) => {
error!(module = module_name, error = %e, "Module execution failed"); let error_msg = e.to_string();
error!(module = module_name, error = %error_msg, "Module execution failed");
let err_out = crate::output::WaybarOutput { let err_out = crate::output::WaybarOutput {
text: "\u{200B}Error\u{200B}".to_string(), text: "\u{200B}Error\u{200B}".to_string(),
tooltip: Some(e.to_string()), tooltip: Some(error_msg),
class: Some("error".to_string()), class: Some("error".to_string()),
percentage: None, percentage: None,
}; };
+31
View File
@@ -0,0 +1,31 @@
use thiserror::Error;
#[derive(Error, Debug)]
#[allow(dead_code)]
pub enum FluxoError {
#[error("Configuration error: {0}")]
Config(String),
#[error("Module error ({module}): {message}")]
Module {
module: &'static str,
message: String,
},
#[error("Daemon IPC error: {0}")]
Ipc(String),
#[error("External system error: {0}")]
System(String),
#[error("IO error: {0}")]
Io(#[from] std::io::Error),
#[error("Serialization error: {0}")]
Serialization(#[from] serde_json::Error),
#[error("Other error: {0}")]
Other(#[from] anyhow::Error),
}
pub type Result<T> = std::result::Result<T, FluxoError>;
+7 -1
View File
@@ -1,5 +1,6 @@
mod config; mod config;
mod daemon; mod daemon;
mod error;
mod ipc; mod ipc;
mod modules; mod modules;
mod output; mod output;
@@ -88,7 +89,12 @@ fn main() {
match &cli.command { match &cli.command {
Commands::Daemon { config } => { Commands::Daemon { config } => {
info!("Starting Fluxo daemon..."); info!("Starting Fluxo daemon...");
if let Err(e) = daemon::run_daemon(config.clone()) { let rt = tokio::runtime::Builder::new_multi_thread()
.enable_all()
.build()
.unwrap();
if let Err(e) = rt.block_on(daemon::run_daemon(config.clone())) {
error!("Daemon failed: {}", e); error!("Daemon failed: {}", e);
process::exit(1); process::exit(1);
} }
+25 -10
View File
@@ -1,18 +1,25 @@
use crate::config::Config; use crate::config::Config;
use crate::error::{FluxoError, Result};
use crate::modules::WaybarModule; use crate::modules::WaybarModule;
use crate::output::WaybarOutput; use crate::output::WaybarOutput;
use crate::state::SharedState; use crate::state::SharedState;
use crate::utils::{TokenValue, format_template}; use crate::utils::{TokenValue, format_template};
use anyhow::{Result, anyhow};
use libpulse_binding::callbacks::ListResult; use libpulse_binding::callbacks::ListResult;
use libpulse_binding::context::subscribe::{Facility, InterestMaskSet}; use libpulse_binding::context::subscribe::{Facility, InterestMaskSet};
use libpulse_binding::context::{Context, FlagSet as ContextFlag}; use libpulse_binding::context::{Context, FlagSet as ContextFlag};
use libpulse_binding::mainloop::threaded::Mainloop as ThreadedMainloop; use libpulse_binding::mainloop::threaded::Mainloop as ThreadedMainloop;
use libpulse_binding::volume::Volume; use libpulse_binding::volume::Volume;
use std::process::Command; use std::process::Command;
use std::sync::Arc; use std::sync::{Arc, LazyLock};
use tracing::error; use tracing::error;
static RUNTIME: LazyLock<tokio::runtime::Runtime> = LazyLock::new(|| {
tokio::runtime::Builder::new_current_thread()
.enable_all()
.build()
.expect("Failed to create Audio tokio runtime")
});
pub struct AudioDaemon; pub struct AudioDaemon;
impl AudioDaemon { impl AudioDaemon {
@@ -100,7 +107,7 @@ fn fetch_audio_data_sync(context: &mut Context, state: &SharedState) -> Result<(
let st_server = Arc::clone(&state_inner); let st_server = Arc::clone(&state_inner);
context.introspect().get_server_info(move |info| { context.introspect().get_server_info(move |info| {
let mut lock = st_server.write().unwrap(); let mut lock = RUNTIME.block_on(st_server.write());
lock.audio.sink.name = info lock.audio.sink.name = info
.default_sink_name .default_sink_name
.as_ref() .as_ref()
@@ -116,7 +123,7 @@ fn fetch_audio_data_sync(context: &mut Context, state: &SharedState) -> Result<(
let st_sink = Arc::clone(&state_inner); let st_sink = Arc::clone(&state_inner);
context.introspect().get_sink_info_list(move |res| { context.introspect().get_sink_info_list(move |res| {
if let ListResult::Item(item) = res { if let ListResult::Item(item) = res {
let mut lock = st_sink.write().unwrap(); let mut lock = RUNTIME.block_on(st_sink.write());
// If this matches our default sink name, or if we don't have details for any yet // If this matches our default sink name, or if we don't have details for any yet
let is_default = item let is_default = item
.name .name
@@ -139,7 +146,7 @@ fn fetch_audio_data_sync(context: &mut Context, state: &SharedState) -> Result<(
let st_source = Arc::clone(&state_inner); let st_source = Arc::clone(&state_inner);
context.introspect().get_source_info_list(move |res| { context.introspect().get_source_info_list(move |res| {
if let ListResult::Item(item) = res { if let ListResult::Item(item) = res {
let mut lock = st_source.write().unwrap(); let mut lock = RUNTIME.block_on(st_source.write());
let is_default = item let is_default = item
.name .name
.as_ref() .as_ref()
@@ -164,7 +171,12 @@ fn fetch_audio_data_sync(context: &mut Context, state: &SharedState) -> Result<(
pub struct AudioModule; pub struct AudioModule;
impl WaybarModule for AudioModule { impl WaybarModule for AudioModule {
fn run(&self, config: &Config, state: &SharedState, args: &[&str]) -> Result<WaybarOutput> { async fn run(
&self,
config: &Config,
state: &SharedState,
args: &[&str],
) -> Result<WaybarOutput> {
let target_type = args.first().unwrap_or(&"sink"); let target_type = args.first().unwrap_or(&"sink");
let action = args.get(1).unwrap_or(&"show"); let action = args.get(1).unwrap_or(&"show");
@@ -173,21 +185,24 @@ impl WaybarModule for AudioModule {
self.cycle_device(target_type)?; self.cycle_device(target_type)?;
Ok(WaybarOutput::default()) Ok(WaybarOutput::default())
} }
"show" => self.get_status(config, state, target_type), "show" => self.get_status(config, state, target_type).await,
other => Err(anyhow!("Unknown audio action: '{}'", other)), other => Err(FluxoError::Module {
module: "audio",
message: format!("Unknown audio action: '{}'", other),
}),
} }
} }
} }
impl AudioModule { impl AudioModule {
fn get_status( async fn get_status(
&self, &self,
config: &Config, config: &Config,
state: &SharedState, state: &SharedState,
target_type: &str, target_type: &str,
) -> Result<WaybarOutput> { ) -> Result<WaybarOutput> {
let audio_state = { let audio_state = {
let lock = state.read().unwrap(); let lock = state.read().await;
lock.audio.clone() lock.audio.clone()
}; };
-608
View File
@@ -1,608 +0,0 @@
use crate::config::Config;
use crate::modules::WaybarModule;
use crate::output::WaybarOutput;
use crate::state::{BtState, SharedState};
use crate::utils::{TokenValue, format_template};
use anyhow::{Context, Result};
use futures::StreamExt;
use std::collections::HashMap;
use std::process::Command;
use std::sync::{Arc, LazyLock, Mutex};
use std::time::{Duration, Instant};
use tokio::sync::mpsc;
use tracing::{debug, error, info, warn};
// Maestro imports
#[allow(unused_imports)]
use maestro::protocol::codec::Codec;
#[allow(unused_imports)]
use maestro::pwrpc::client::{Client, ClientHandle};
#[allow(unused_imports)]
use maestro::service::MaestroService;
#[allow(unused_imports)]
use maestro::service::settings::{self, Setting, SettingValue};
static RUNTIME: LazyLock<tokio::runtime::Runtime> = LazyLock::new(|| {
tokio::runtime::Builder::new_current_thread()
.enable_all()
.build()
.expect("Failed to create BT tokio runtime")
});
#[derive(Clone, Default)]
struct BudsStatus {
left_battery: Option<u8>,
right_battery: Option<u8>,
case_battery: Option<u8>,
anc_state: String,
#[allow(dead_code)]
last_update: Option<Instant>,
error: Option<String>,
}
enum BudsCommand {
SetAnc(String),
}
enum ManagerCommand {
EnsureTask(String),
SendCommand(String, BudsCommand),
}
struct MaestroManager {
statuses: Arc<Mutex<HashMap<String, BudsStatus>>>,
management_tx: mpsc::UnboundedSender<ManagerCommand>,
}
impl MaestroManager {
fn new() -> Self {
let (tx, mut rx) = mpsc::unbounded_channel::<ManagerCommand>();
let statuses = Arc::new(Mutex::new(HashMap::new()));
let statuses_clone = Arc::clone(&statuses);
// Start dedicated BT management thread
std::thread::spawn(move || {
let rt = tokio::runtime::Builder::new_current_thread()
.enable_all()
.build()
.unwrap();
let local = tokio::task::LocalSet::new();
local.block_on(&rt, async move {
let mut command_txs: HashMap<String, mpsc::Sender<BudsCommand>> = HashMap::new();
loop {
tokio::select! {
Some(cmd) = rx.recv() => {
match cmd {
ManagerCommand::EnsureTask(mac) => {
if !command_txs.contains_key(&mac) {
let (tx, buds_rx) = mpsc::channel::<BudsCommand>(10);
command_txs.insert(mac.clone(), tx);
let mac_clone = mac.clone();
let st_clone = Arc::clone(&statuses_clone);
tokio::task::spawn_local(async move {
if let Err(e) = buds_task(&mac_clone, st_clone, buds_rx).await {
error!("Buds task for {} failed: {}", mac_clone, e);
}
});
}
}
ManagerCommand::SendCommand(mac, buds_cmd) => {
if let Some(tx) = command_txs.get(&mac) {
let _ = tx.try_send(buds_cmd);
}
}
}
}
_ = tokio::time::sleep(Duration::from_millis(100)) => {
// Cleanup dropped tasks if needed
}
}
}
});
});
Self {
statuses,
management_tx: tx,
}
}
fn get_status(&self, mac: &str) -> BudsStatus {
let statuses = self.statuses.lock().unwrap();
statuses.get(mac).cloned().unwrap_or_default()
}
fn ensure_task(&self, mac: &str) {
let _ = self
.management_tx
.send(ManagerCommand::EnsureTask(mac.to_string()));
}
fn send_command(&self, mac: &str, cmd: BudsCommand) -> Result<()> {
self.ensure_task(mac);
let _ = self
.management_tx
.send(ManagerCommand::SendCommand(mac.to_string(), cmd));
Ok(())
}
}
async fn buds_task(
mac: &str,
statuses: Arc<Mutex<HashMap<String, BudsStatus>>>,
mut rx: mpsc::Receiver<BudsCommand>,
) -> Result<()> {
info!("Starting native Maestro connection task for {}", mac);
loop {
let addr: bluer::Address = mac.parse().context("Failed to parse MAC address")?;
let session = bluer::Session::new()
.await
.context("Failed to create bluer session")?;
let adapter = session
.default_adapter()
.await
.context("Failed to get default adapter")?;
let device = adapter
.device(addr)
.context("Failed to get device handle")?;
if !device.is_connected().await.unwrap_or(false) {
debug!("Device {} not connected to BT, stopping maestro task", mac);
break;
}
// Connect to Maestro RFCOMM service
// We try channel 1 then 2, which covers most Pixel Buds variants.
let mut stream = None;
for channel in [1, 2] {
let socket = match bluer::rfcomm::Socket::new() {
Ok(s) => s,
Err(e) => {
error!("Failed to create RFCOMM socket: {}", e);
return Err(e.into());
}
};
let target = bluer::rfcomm::SocketAddr::new(addr, channel);
debug!(
"Trying to connect RFCOMM to {} on channel {}...",
mac, channel
);
match socket.connect(target).await {
Ok(s) => {
stream = Some(s);
break;
}
Err(e) => {
debug!("Failed to connect to channel {}: {}", channel, e);
}
}
}
let stream = match stream {
Some(s) => s,
None => {
warn!(
"Failed to connect RFCOMM to {} on any common channel. Retrying in 15s...",
mac
);
tokio::time::sleep(Duration::from_secs(15)).await;
continue;
}
};
info!("Connected Maestro RFCOMM to {} on channel", mac);
// Initialize Maestro communication stack
let codec = Codec::new();
let stream = codec.wrap(stream);
let mut client = Client::new(stream);
let handle = client.handle();
// Resolve Maestro channel (typically 1 or 2)
let channel = match maestro::protocol::utils::resolve_channel(&mut client).await {
Ok(c) => c,
Err(e) => {
error!("Failed to resolve Maestro channel for {}: {}", mac, e);
continue;
}
};
// Run client in background to handle RPC packets
tokio::spawn(async move {
if let Err(e) = client.run().await {
error!("Maestro client loop error: {}", e);
}
});
let mut service = MaestroService::new(handle, channel);
// Query initial ANC state
if let Ok(val) = service
.read_setting_var(settings::SettingId::CurrentAncrState)
.await
&& let SettingValue::CurrentAncrState(anc_state) = val
{
let mut status = MAESTRO.get_status(mac);
status.anc_state = anc_state_to_string(&anc_state);
statuses.lock().unwrap().insert(mac.to_string(), status);
}
// Subscribe to real-time status updates (battery, ANC, wear)
let mut call = match service.subscribe_to_runtime_info() {
Ok(c) => c,
Err(e) => {
error!("Failed to subscribe to runtime info for {}: {}", mac, e);
continue;
}
};
let mut runtime_info = call.stream();
debug!("Subscribed to runtime info for {}", mac);
loop {
tokio::select! {
cmd = rx.recv() => {
match cmd {
Some(BudsCommand::SetAnc(mode)) => {
debug!("Setting ANC mode to {} for {}", mode, mac);
let state = mode_to_anc_state(&mode);
let val = SettingValue::CurrentAncrState(state);
if let Err(e) = service.write_setting(val).await {
error!("Failed to write ANC setting for {}: {}", mac, e);
}
}
None => return Ok(()),
}
}
Some(Ok(info)) = runtime_info.next() => {
let mut status = MAESTRO.get_status(mac);
status.last_update = Some(Instant::now());
if let Some(bat) = info.battery_info {
status.left_battery = bat.left.map(|b| b.level as u8);
status.right_battery = bat.right.map(|b| b.level as u8);
status.case_battery = bat.case.map(|b| b.level as u8);
}
statuses.lock().unwrap().insert(mac.to_string(), status);
}
_ = tokio::time::sleep(Duration::from_secs(30)) => {
// Check if still connected to BT
if !device.is_connected().await.unwrap_or(false) {
break;
}
}
}
}
if !device.is_connected().await.unwrap_or(false) {
break;
}
}
Ok(())
}
fn mode_to_anc_state(mode: &str) -> settings::AncState {
match mode {
"active" => settings::AncState::Active,
"aware" => settings::AncState::Aware,
"off" => settings::AncState::Off,
_ => settings::AncState::Off,
}
}
fn anc_state_to_string(state: &settings::AncState) -> String {
match state {
settings::AncState::Active => "active".to_string(),
settings::AncState::Aware => "aware".to_string(),
settings::AncState::Off => "off".to_string(),
_ => "unknown".to_string(),
}
}
static MAESTRO: LazyLock<MaestroManager> = LazyLock::new(MaestroManager::new);
pub struct BtDaemon {
session: Option<bluer::Session>,
}
impl BtDaemon {
pub fn new() -> Self {
Self { session: None }
}
pub fn poll(&mut self, state: SharedState) {
if let Err(e) = RUNTIME.block_on(self.poll_async(state)) {
error!("BT daemon error: {}", e);
}
}
async fn poll_async(&mut self, state: SharedState) -> Result<()> {
if self.session.is_none() {
self.session = Some(bluer::Session::new().await?);
}
let session = self.session.as_ref().unwrap();
let adapter = session.default_adapter().await?;
let adapter_powered = adapter.is_powered().await.unwrap_or(false);
let mut bt_state = BtState {
adapter_powered,
..Default::default()
};
if adapter_powered {
let devices = adapter.device_addresses().await?;
for addr in devices {
let device = adapter.device(addr)?;
if device.is_connected().await.unwrap_or(false) {
let uuids = device.uuids().await?.unwrap_or_default();
// Audio sink UUID (0x110b)
let audio_sink_uuid =
bluer::Uuid::from_u128(0x0000110b_0000_1000_8000_00805f9b34fb);
if uuids.contains(&audio_sink_uuid) {
bt_state.connected = true;
bt_state.device_address = addr.to_string();
bt_state.device_alias =
device.alias().await.unwrap_or_else(|_| addr.to_string());
bt_state.battery_percentage =
device.battery_percentage().await.unwrap_or(None);
// Plugin detection
for p in PLUGINS.iter() {
if p.can_handle(&bt_state.device_alias, &bt_state.device_address) {
match p.get_data(&bt_state.device_address) {
Ok(data) => {
bt_state.plugin_data = data
.into_iter()
.map(|(k, v)| {
let val_str = match v {
TokenValue::String(s) => s,
TokenValue::Int(i) => i.to_string(),
TokenValue::Float(f) => format!("{:.1}", f),
};
(k, val_str)
})
.collect();
}
Err(e) => {
warn!("Plugin {} failed for {}: {}", p.name(), addr, e);
bt_state
.plugin_data
.push(("plugin_error".to_string(), e.to_string()));
}
}
break;
}
}
break;
}
}
}
}
if let Ok(mut lock) = state.write() {
lock.bluetooth = bt_state;
}
Ok(())
}
}
pub trait BtPlugin: Send + Sync {
fn name(&self) -> &str;
fn can_handle(&self, alias: &str, mac: &str) -> bool;
fn get_data(&self, mac: &str) -> Result<Vec<(String, TokenValue)>>;
fn get_modes(&self, _mac: &str) -> Result<Vec<String>> {
Ok(vec![])
}
fn set_mode(&self, _mode: &str, _mac: &str) -> Result<()> {
Ok(())
}
fn cycle_mode(&self, _mac: &str) -> Result<()> {
Ok(())
}
}
pub struct PixelBudsPlugin;
impl BtPlugin for PixelBudsPlugin {
fn name(&self) -> &str {
"Pixel Buds Pro"
}
fn can_handle(&self, alias: &str, _mac: &str) -> bool {
alias.contains("Pixel Buds Pro")
}
fn get_data(&self, mac: &str) -> Result<Vec<(String, TokenValue)>> {
MAESTRO.ensure_task(mac);
let status = MAESTRO.get_status(mac);
if let Some(err) = status.error {
return Err(anyhow::anyhow!(err));
}
let left_display = status
.left_battery
.map(|b| format!("{}%", b))
.unwrap_or_else(|| "---".to_string());
let right_display = status
.right_battery
.map(|b| format!("{}%", b))
.unwrap_or_else(|| "---".to_string());
let (anc_icon, class) = match status.anc_state.as_str() {
"active" => ("ANC", "anc-active"),
"aware" => ("Aware", "anc-aware"),
"off" => ("Off", "anc-off"),
_ => ("?", "anc-unknown"),
};
Ok(vec![
("left".to_string(), TokenValue::String(left_display)),
("right".to_string(), TokenValue::String(right_display)),
("anc".to_string(), TokenValue::String(anc_icon.to_string())),
(
"plugin_class".to_string(),
TokenValue::String(class.to_string()),
),
])
}
fn get_modes(&self, _mac: &str) -> Result<Vec<String>> {
Ok(vec![
"active".to_string(),
"aware".to_string(),
"off".to_string(),
])
}
fn set_mode(&self, mode: &str, mac: &str) -> Result<()> {
MAESTRO.send_command(mac, BudsCommand::SetAnc(mode.to_string()))
}
fn cycle_mode(&self, mac: &str) -> Result<()> {
let status = MAESTRO.get_status(mac);
let next_mode = match status.anc_state.as_str() {
"active" => "aware",
"aware" => "off",
_ => "active",
};
self.set_mode(next_mode, mac)
}
}
static PLUGINS: LazyLock<Vec<Box<dyn BtPlugin>>> =
LazyLock::new(|| vec![Box::new(PixelBudsPlugin)]);
pub struct BtModule;
impl WaybarModule for BtModule {
fn run(&self, config: &Config, state: &SharedState, args: &[&str]) -> Result<WaybarOutput> {
let action = args.first().unwrap_or(&"show");
let bt_state = {
let lock = state.read().unwrap();
lock.bluetooth.clone()
};
match *action {
"disconnect" if bt_state.connected => {
let _ = Command::new("bluetoothctl")
.args(["disconnect", &bt_state.device_address])
.output();
return Ok(WaybarOutput::default());
}
"cycle_mode" if bt_state.connected => {
let plugin = PLUGINS
.iter()
.find(|p| p.can_handle(&bt_state.device_alias, &bt_state.device_address));
if let Some(p) = plugin {
p.cycle_mode(&bt_state.device_address)?;
}
return Ok(WaybarOutput::default());
}
"get_modes" if bt_state.connected => {
let plugin = PLUGINS
.iter()
.find(|p| p.can_handle(&bt_state.device_alias, &bt_state.device_address));
let modes = if let Some(p) = plugin {
p.get_modes(&bt_state.device_address)?
} else {
vec![]
};
return Ok(WaybarOutput {
text: modes.join("\n"),
..Default::default()
});
}
"set_mode" if bt_state.connected => {
if let Some(mode) = args.get(1) {
let plugin = PLUGINS
.iter()
.find(|p| p.can_handle(&bt_state.device_alias, &bt_state.device_address));
if let Some(p) = plugin {
p.set_mode(mode, &bt_state.device_address)?;
}
}
return Ok(WaybarOutput::default());
}
"show" => {}
_ => {}
}
if !bt_state.adapter_powered {
return Ok(WaybarOutput {
text: config.bt.format_disabled.clone(),
tooltip: Some("Bluetooth Disabled".to_string()),
class: Some("disabled".to_string()),
percentage: None,
});
}
if bt_state.connected {
let mut tokens: Vec<(String, TokenValue)> = vec![
(
"alias".to_string(),
TokenValue::String(bt_state.device_alias.clone()),
),
(
"mac".to_string(),
TokenValue::String(bt_state.device_address.clone()),
),
];
let mut class = vec!["connected".to_string()];
let mut has_plugin = false;
for (k, v) in &bt_state.plugin_data {
if k == "plugin_class" {
class.push(v.clone());
has_plugin = true;
} else if k == "plugin_error" {
class.push("plugin-error".to_string());
} else {
tokens.push((k.clone(), TokenValue::String(v.clone())));
}
}
let format = if has_plugin {
&config.bt.format_plugin
} else {
&config.bt.format_connected
};
let text = format_template(format, &tokens);
let tooltip = format!(
"{} | MAC: {}\nBattery: {}",
bt_state.device_alias,
bt_state.device_address,
bt_state
.battery_percentage
.map(|b| format!("{}%", b))
.unwrap_or_else(|| "N/A".to_string())
);
Ok(WaybarOutput {
text,
tooltip: Some(tooltip),
class: Some(class.join(" ")),
percentage: bt_state.battery_percentage,
})
} else {
Ok(WaybarOutput {
text: config.bt.format_disconnected.clone(),
tooltip: Some("Bluetooth On (Disconnected)".to_string()),
class: Some("disconnected".to_string()),
percentage: None,
})
}
}
}
+142
View File
@@ -0,0 +1,142 @@
use crate::config::Config;
use crate::error::{FluxoError, Result as FluxoResult};
use crate::modules::bt::maestro::{BudsCommand, get_maestro};
use crate::state::SharedState;
use crate::utils::TokenValue;
use futures::future::BoxFuture;
use std::sync::Arc;
pub trait BtPlugin: Send + Sync {
fn name(&self) -> &str;
fn can_handle(&self, alias: &str, mac: &str) -> bool;
fn get_data(
&self,
config: &Config,
state: &SharedState,
mac: &str,
) -> BoxFuture<'static, FluxoResult<Vec<(String, TokenValue)>>>;
fn get_modes(
&self,
mac: &str,
state: &SharedState,
) -> BoxFuture<'static, FluxoResult<Vec<String>>>;
fn set_mode(
&self,
mode: &str,
mac: &str,
state: &SharedState,
) -> BoxFuture<'static, FluxoResult<()>>;
fn cycle_mode(&self, mac: &str, state: &SharedState) -> BoxFuture<'static, FluxoResult<()>>;
}
pub struct PixelBudsPlugin;
impl BtPlugin for PixelBudsPlugin {
fn name(&self) -> &str {
"Pixel Buds Pro"
}
fn can_handle(&self, alias: &str, _mac: &str) -> bool {
alias.contains("Pixel Buds Pro")
}
fn get_data(
&self,
_config: &Config,
state: &SharedState,
mac: &str,
) -> BoxFuture<'static, FluxoResult<Vec<(String, TokenValue)>>> {
let mac = mac.to_string();
let state = Arc::clone(state);
Box::pin(async move {
let maestro = get_maestro(&state);
maestro.ensure_task(&mac);
let status = maestro.get_status(&mac);
if let Some(err) = status.error {
return Err(FluxoError::Module {
module: "bt.buds",
message: err,
});
}
let left_display = status
.left_battery
.map(|b| format!("{}%", b))
.unwrap_or_else(|| "---".to_string());
let right_display = status
.right_battery
.map(|b| format!("{}%", b))
.unwrap_or_else(|| "---".to_string());
let (anc_icon, class) = match status.anc_state.as_str() {
"active" => ("ANC", "anc-active"),
"aware" => ("Aware", "anc-aware"),
"off" => ("Off", "anc-off"),
_ => ("?", "anc-unknown"),
};
Ok(vec![
("left".to_string(), TokenValue::String(left_display)),
("right".to_string(), TokenValue::String(right_display)),
("anc".to_string(), TokenValue::String(anc_icon.to_string())),
(
"plugin_class".to_string(),
TokenValue::String(class.to_string()),
),
])
})
}
fn get_modes(
&self,
_mac: &str,
_state: &SharedState,
) -> BoxFuture<'static, FluxoResult<Vec<String>>> {
Box::pin(async move {
Ok(vec![
"active".to_string(),
"aware".to_string(),
"off".to_string(),
])
})
}
fn set_mode(
&self,
mode: &str,
mac: &str,
state: &SharedState,
) -> BoxFuture<'static, FluxoResult<()>> {
let mode = mode.to_string();
let mac = mac.to_string();
let state = Arc::clone(state);
Box::pin(async move {
get_maestro(&state)
.send_command(&mac, BudsCommand::SetAnc(mode))
.map_err(|e: anyhow::Error| FluxoError::Module {
module: "bt.buds",
message: e.to_string(),
})
})
}
fn cycle_mode(&self, mac: &str, state: &SharedState) -> BoxFuture<'static, FluxoResult<()>> {
let mac = mac.to_string();
let state = Arc::clone(state);
Box::pin(async move {
let status = get_maestro(&state).get_status(&mac);
let next_mode = match status.anc_state.as_str() {
"active" => "aware",
"aware" => "off",
_ => "active",
};
get_maestro(&state)
.send_command(&mac, BudsCommand::SetAnc(next_mode.to_string()))
.map_err(|e: anyhow::Error| FluxoError::Module {
module: "bt.buds",
message: e.to_string(),
})
})
}
}
+362
View File
@@ -0,0 +1,362 @@
use crate::state::SharedState;
use anyhow::{Context, Result};
use futures::StreamExt;
use std::collections::HashMap;
use std::sync::{Arc, Mutex, OnceLock};
use std::time::{Duration, Instant};
use tokio::sync::mpsc;
use tracing::{debug, error, info, warn};
// Maestro imports
use maestro::protocol::codec::Codec;
use maestro::pwrpc::client::Client;
use maestro::service::MaestroService;
use maestro::service::settings::{self, SettingValue};
#[derive(Clone, Default)]
pub struct BudsStatus {
pub left_battery: Option<u8>,
pub right_battery: Option<u8>,
pub case_battery: Option<u8>,
pub anc_state: String,
#[allow(dead_code)]
pub last_update: Option<Instant>,
pub error: Option<String>,
}
pub enum BudsCommand {
SetAnc(String),
}
pub enum ManagerCommand {
EnsureTask(String),
SendCommand(String, BudsCommand),
}
pub struct MaestroManager {
statuses: Arc<Mutex<HashMap<String, BudsStatus>>>,
management_tx: mpsc::UnboundedSender<ManagerCommand>,
}
impl MaestroManager {
pub fn new(state: SharedState) -> Self {
let (tx, mut rx) = mpsc::unbounded_channel::<ManagerCommand>();
let statuses = Arc::new(Mutex::new(HashMap::new()));
let statuses_clone = Arc::clone(&statuses);
let state_clone = Arc::clone(&state);
// Start dedicated BT management thread
std::thread::spawn(move || {
let rt = tokio::runtime::Builder::new_current_thread()
.enable_all()
.build()
.unwrap();
let local = tokio::task::LocalSet::new();
local.block_on(&rt, async move {
let mut command_txs: HashMap<String, mpsc::Sender<BudsCommand>> = HashMap::new();
loop {
tokio::select! {
Some(cmd) = rx.recv() => {
match cmd {
ManagerCommand::EnsureTask(mac) => {
if !command_txs.contains_key(&mac) {
let (tx, buds_rx) = mpsc::channel::<BudsCommand>(10);
command_txs.insert(mac.clone(), tx);
let mac_clone = mac.clone();
let st_clone = Arc::clone(&statuses_clone);
let state_inner = Arc::clone(&state_clone);
tokio::task::spawn_local(async move {
if let Err(e) = buds_task(&mac_clone, st_clone, buds_rx, state_inner).await {
error!("Buds task for {} failed: {}", mac_clone, e);
}
});
}
}
ManagerCommand::SendCommand(mac, buds_cmd) => {
if let Some(tx) = command_txs.get(&mac) {
let _ = tx.try_send(buds_cmd);
}
}
}
}
_ = tokio::time::sleep(Duration::from_millis(100)) => {
// Cleanup dropped tasks if needed
}
}
}
});
});
Self {
statuses,
management_tx: tx,
}
}
pub fn get_status(&self, mac: &str) -> BudsStatus {
let statuses = self.statuses.lock().unwrap();
statuses.get(mac).cloned().unwrap_or_default()
}
pub fn ensure_task(&self, mac: &str) {
let _ = self
.management_tx
.send(ManagerCommand::EnsureTask(mac.to_string()));
}
pub fn send_command(&self, mac: &str, cmd: BudsCommand) -> Result<()> {
self.ensure_task(mac);
let _ = self
.management_tx
.send(ManagerCommand::SendCommand(mac.to_string(), cmd));
Ok(())
}
}
async fn buds_task(
mac: &str,
statuses: Arc<Mutex<HashMap<String, BudsStatus>>>,
mut rx: mpsc::Receiver<BudsCommand>,
state: SharedState,
) -> Result<()> {
info!("Starting native Maestro connection task for {}", mac);
loop {
let addr: bluer::Address = match mac.parse() {
Ok(a) => a,
Err(e) => {
error!("Failed to parse MAC address {}: {}", mac, e);
return Err(e.into());
}
};
let session = bluer::Session::new()
.await
.context("Failed to create bluer session")?;
let adapter = session
.default_adapter()
.await
.context("Failed to get default adapter")?;
let device = adapter
.device(addr)
.context("Failed to get device handle")?;
if !device.is_connected().await.unwrap_or(false) {
debug!("Device {} not connected to BT, stopping maestro task", mac);
break;
}
// Connect to Maestro RFCOMM service
let mut stream = None;
for channel in [1, 2] {
let socket = match bluer::rfcomm::Socket::new() {
Ok(s) => s,
Err(e) => {
error!("Failed to create RFCOMM socket: {}", e);
return Err(e.into());
}
};
let target = bluer::rfcomm::SocketAddr::new(addr, channel);
debug!(
"Trying to connect RFCOMM to {} on channel {}...",
mac, channel
);
match socket.connect(target).await {
Ok(s) => {
stream = Some(s);
break;
}
Err(e) => {
debug!("Failed to connect to channel {}: {}", channel, e);
}
}
}
let stream = match stream {
Some(s) => s,
None => {
warn!(
"Failed to connect RFCOMM to {} on any common channel. Retrying in 15s...",
mac
);
tokio::time::sleep(Duration::from_secs(15)).await;
continue;
}
};
info!("Connected Maestro RFCOMM to {} on channel", mac);
// Initialize Maestro communication stack
let codec = Codec::new();
let stream = codec.wrap(stream);
let mut client = Client::new(stream);
let handle = client.handle();
// Resolve Maestro channel
let channel = match maestro::protocol::utils::resolve_channel(&mut client).await {
Ok(c) => c,
Err(e) => {
error!("Failed to resolve Maestro channel for {}: {}", mac, e);
continue;
}
};
tokio::spawn(async move {
if let Err(e) = client.run().await {
error!("Maestro client loop error: {}", e);
}
});
let mut service = MaestroService::new(handle, channel);
// Update health
{
let mut lock = state.write().await;
let health = lock.health.entry("bt.buds".to_string()).or_default();
health.consecutive_failures = 0;
health.backoff_until = None;
}
// Query initial ANC state
if let Ok(val) = service
.read_setting_var(settings::SettingId::CurrentAncrState)
.await
&& let SettingValue::CurrentAncrState(anc_state) = val
{
let mut lock = statuses.lock().unwrap();
let status = lock.entry(mac.to_string()).or_default();
status.anc_state = anc_state_to_string(&anc_state);
status.last_update = Some(Instant::now());
}
let mut runtime_info_call = match service.subscribe_to_runtime_info() {
Ok(c) => c,
Err(e) => {
error!("Failed to subscribe to runtime info for {}: {}", mac, e);
continue;
}
};
let mut runtime_info = runtime_info_call.stream();
let mut settings_changes_call = match service.subscribe_to_settings_changes() {
Ok(s) => s,
Err(e) => {
error!("Failed to subscribe to settings changes for {}: {}", mac, e);
continue;
}
};
let mut settings_changes = settings_changes_call.stream();
debug!("Subscribed to status and settings for {}", mac);
loop {
tokio::select! {
cmd = rx.recv() => {
match cmd {
Some(BudsCommand::SetAnc(mode)) => {
debug!("Setting ANC mode to {} for {}", mode, mac);
let state = mode_to_anc_state(&mode);
let val = SettingValue::CurrentAncrState(state);
{
let mut lock = statuses.lock().unwrap();
let status = lock.entry(mac.to_string()).or_default();
status.anc_state = mode.clone();
status.last_update = Some(Instant::now());
}
if let Err(e) = service.write_setting(val).await {
error!("Failed to write ANC setting for {}: {}", mac, e);
}
}
None => return Ok(()),
}
}
Some(res) = runtime_info.next() => {
match res {
Ok(info) => {
let mut lock = statuses.lock().unwrap();
let status = lock.entry(mac.to_string()).or_default();
status.last_update = Some(Instant::now());
if let Some(bat) = info.battery_info {
status.left_battery = bat.left.map(|b| b.level as u8);
status.right_battery = bat.right.map(|b| b.level as u8);
status.case_battery = bat.case.map(|b| b.level as u8);
}
}
Err(e) => {
warn!("Runtime info stream error for {}: {}", mac, e);
break;
}
}
}
Some(res) = settings_changes.next() => {
if let Ok(change) = res {
use maestro::protocol::types::settings_rsp::ValueOneof as RspOneof;
use maestro::protocol::types::setting_value::ValueOneof as ValOneof;
if let Some(RspOneof::Value(setting_val)) = change.value_oneof
&& let Some(ValOneof::CurrentAncrState(anc_state_raw)) = setting_val.value_oneof {
let mut lock = statuses.lock().unwrap();
let status = lock.entry(mac.to_string()).or_default();
let anc_state = match anc_state_raw {
1 => settings::AncState::Off,
2 => settings::AncState::Active,
3 => settings::AncState::Aware,
4 => settings::AncState::Adaptive,
_ => settings::AncState::Unknown(anc_state_raw),
};
status.anc_state = anc_state_to_string(&anc_state);
status.last_update = Some(Instant::now());
debug!(mode = %status.anc_state, "Caught physical ANC toggle");
}
}
}
_ = tokio::time::sleep(Duration::from_secs(30)) => {
if !device.is_connected().await.unwrap_or(false) {
break;
}
}
}
}
if !device.is_connected().await.unwrap_or(false) {
break;
}
}
Ok(())
}
fn mode_to_anc_state(mode: &str) -> settings::AncState {
match mode {
"active" => settings::AncState::Active,
"aware" => settings::AncState::Aware,
"off" => settings::AncState::Off,
_ => settings::AncState::Off,
}
}
pub fn anc_state_to_string(state: &settings::AncState) -> String {
match state {
settings::AncState::Active => "active".to_string(),
settings::AncState::Aware => "aware".to_string(),
settings::AncState::Off => "off".to_string(),
_ => "unknown".to_string(),
}
}
static MAESTRO: OnceLock<MaestroManager> = OnceLock::new();
pub fn get_maestro(state: &SharedState) -> &MaestroManager {
MAESTRO.get_or_init(|| MaestroManager::new(Arc::clone(state)))
}
+236
View File
@@ -0,0 +1,236 @@
pub mod buds;
pub mod maestro;
use crate::config::Config;
use crate::error::Result as FluxoResult;
use crate::modules::WaybarModule;
use crate::output::WaybarOutput;
use crate::state::{BtState, SharedState};
use crate::utils::{TokenValue, format_template};
use anyhow::Result;
use std::process::Command;
use std::sync::{Arc, LazyLock};
use tracing::{error, warn};
use self::buds::{BtPlugin, PixelBudsPlugin};
pub struct BtDaemon {
session: Option<bluer::Session>,
}
impl BtDaemon {
pub fn new() -> Self {
Self { session: None }
}
pub async fn poll(&mut self, state: SharedState, config: &Config) {
if let Err(e) = self.poll_async(state, config).await {
error!("BT daemon error: {}", e);
}
}
async fn poll_async(&mut self, state: SharedState, config: &Config) -> Result<()> {
if self.session.is_none() {
self.session = Some(bluer::Session::new().await?);
}
let session = self.session.as_ref().unwrap();
let adapter = session.default_adapter().await?;
let adapter_powered = adapter.is_powered().await.unwrap_or(false);
let mut bt_state = BtState {
adapter_powered,
..Default::default()
};
if adapter_powered {
let devices = adapter.device_addresses().await?;
for addr in devices {
let device = adapter.device(addr)?;
if device.is_connected().await.unwrap_or(false) {
let uuids = device.uuids().await?.unwrap_or_default();
let audio_sink_uuid =
bluer::Uuid::from_u128(0x0000110b_0000_1000_8000_00805f9b34fb);
if uuids.contains(&audio_sink_uuid) {
bt_state.connected = true;
bt_state.device_address = addr.to_string();
bt_state.device_alias =
device.alias().await.unwrap_or_else(|_| addr.to_string());
bt_state.battery_percentage =
device.battery_percentage().await.unwrap_or(None);
for p in PLUGINS.iter() {
if p.can_handle(&bt_state.device_alias, &bt_state.device_address) {
match p.get_data(config, &state, &bt_state.device_address).await {
Ok(data) => {
bt_state.plugin_data = data
.into_iter()
.map(|(k, v)| {
let val_str = match v {
TokenValue::String(s) => s,
TokenValue::Int(i) => i.to_string(),
TokenValue::Float(f) => format!("{:.1}", f),
};
(k, val_str)
})
.collect();
}
Err(e) => {
warn!("Plugin {} failed for {}: {}", p.name(), addr, e);
bt_state
.plugin_data
.push(("plugin_error".to_string(), e.to_string()));
}
}
break;
}
}
break;
}
}
}
}
let mut lock = state.write().await;
lock.bluetooth = bt_state;
Ok(())
}
}
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],
) -> impl std::future::Future<Output = FluxoResult<WaybarOutput>> + Send {
let action = args.first().cloned().unwrap_or("show").to_string();
let args = args.iter().map(|s| s.to_string()).collect::<Vec<_>>();
let state = Arc::clone(state);
let config = config.clone();
async move {
let bt_state = {
let lock = state.read().await;
lock.bluetooth.clone()
};
match action.as_str() {
"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, &state).await?;
}
return Ok(WaybarOutput::default());
}
"get_modes" if bt_state.connected => {
let plugin = PLUGINS
.iter()
.find(|p| p.can_handle(&bt_state.device_alias, &bt_state.device_address));
let modes = if let Some(p) = plugin {
p.get_modes(&bt_state.device_address, &state).await?
} else {
vec![]
};
return Ok(WaybarOutput {
text: modes.join("\n"),
..Default::default()
});
}
"set_mode" if bt_state.connected => {
if let Some(mode) = args.get(1) {
let plugin = PLUGINS.iter().find(|p| {
p.can_handle(&bt_state.device_alias, &bt_state.device_address)
});
if let Some(p) = plugin {
p.set_mode(mode, &bt_state.device_address, &state).await?;
}
}
return Ok(WaybarOutput::default());
}
"show" => {}
_ => {}
}
if !bt_state.adapter_powered {
return Ok(WaybarOutput {
text: config.bt.format_disabled.clone(),
tooltip: Some("Bluetooth Disabled".to_string()),
class: Some("disabled".to_string()),
percentage: None,
});
}
if bt_state.connected {
let mut tokens: Vec<(String, TokenValue)> = vec![
(
"alias".to_string(),
TokenValue::String(bt_state.device_alias.clone()),
),
(
"mac".to_string(),
TokenValue::String(bt_state.device_address.clone()),
),
];
let mut class = vec!["connected".to_string()];
let mut has_plugin = false;
for (k, v) in &bt_state.plugin_data {
if k == "plugin_class" {
class.push(v.clone());
has_plugin = true;
} else if k == "plugin_error" {
class.push("plugin-error".to_string());
} else {
tokens.push((k.clone(), TokenValue::String(v.clone())));
}
}
let format = if has_plugin {
&config.bt.format_plugin
} else {
&config.bt.format_connected
};
let text = format_template(format, &tokens);
let tooltip = format!(
"{} | MAC: {}\nBattery: {}",
bt_state.device_alias,
bt_state.device_address,
bt_state
.battery_percentage
.map(|b| format!("{}%", b))
.unwrap_or_else(|| "N/A".to_string())
);
Ok(WaybarOutput {
text,
tooltip: Some(tooltip),
class: Some(class.join(" ")),
percentage: bt_state.battery_percentage,
})
} else {
Ok(WaybarOutput {
text: config.bt.format_disconnected.clone(),
tooltip: Some("Bluetooth On (Disconnected)".to_string()),
class: Some("disconnected".to_string()),
percentage: None,
})
}
}
}
}
+9 -5
View File
@@ -1,18 +1,22 @@
use crate::config::Config; use crate::config::Config;
use crate::error::Result;
use crate::modules::WaybarModule; use crate::modules::WaybarModule;
use crate::output::WaybarOutput; use crate::output::WaybarOutput;
use crate::state::SharedState; use crate::state::SharedState;
use crate::utils::{TokenValue, format_template}; use crate::utils::{TokenValue, format_template};
use anyhow::Result;
pub struct BtrfsModule; pub struct BtrfsModule;
impl WaybarModule for BtrfsModule { impl WaybarModule for BtrfsModule {
fn run(&self, config: &Config, state: &SharedState, _args: &[&str]) -> Result<WaybarOutput> { async fn run(
let disks = if let Ok(s) = state.read() { &self,
config: &Config,
state: &SharedState,
_args: &[&str],
) -> Result<WaybarOutput> {
let disks = {
let s = state.read().await;
s.disks.clone() s.disks.clone()
} else {
return Err(anyhow::anyhow!("Failed to read state"));
}; };
let mut total_used: f64 = 0.0; let mut total_used: f64 = 0.0;
+17 -15
View File
@@ -1,24 +1,26 @@
use crate::config::Config; use crate::config::Config;
use crate::error::Result;
use crate::modules::WaybarModule; use crate::modules::WaybarModule;
use crate::output::WaybarOutput; use crate::output::WaybarOutput;
use crate::state::SharedState; use crate::state::SharedState;
use crate::utils::{TokenValue, format_template}; use crate::utils::{TokenValue, format_template};
use anyhow::Result;
pub struct CpuModule; pub struct CpuModule;
impl WaybarModule for CpuModule { impl WaybarModule for CpuModule {
fn run(&self, config: &Config, state: &SharedState, _args: &[&str]) -> Result<WaybarOutput> { async fn run(
&self,
config: &Config,
state: &SharedState,
_args: &[&str],
) -> Result<WaybarOutput> {
let (usage, temp, model) = { let (usage, temp, model) = {
if let Ok(state_lock) = state.read() { let state_lock = state.read().await;
( (
state_lock.cpu.usage, state_lock.cpu.usage,
state_lock.cpu.temp, state_lock.cpu.temp,
state_lock.cpu.model.clone(), state_lock.cpu.model.clone(),
) )
} else {
(0.0, 0.0, String::from("Unknown"))
}
}; };
let text = format_template( let text = format_template(
@@ -51,8 +53,8 @@ mod tests {
use super::*; use super::*;
use crate::state::{AppState, CpuState, mock_state}; use crate::state::{AppState, CpuState, mock_state};
#[test] #[tokio::test]
fn test_cpu_normal() { async fn test_cpu_normal() {
let state = mock_state(AppState { let state = mock_state(AppState {
cpu: CpuState { cpu: CpuState {
usage: 25.0, usage: 25.0,
@@ -62,7 +64,7 @@ mod tests {
..Default::default() ..Default::default()
}); });
let config = Config::default(); let config = Config::default();
let output = CpuModule.run(&config, &state, &[]).unwrap(); let output = CpuModule.run(&config, &state, &[]).await.unwrap();
assert!(output.text.contains("25.0")); assert!(output.text.contains("25.0"));
assert!(output.text.contains("45.0")); assert!(output.text.contains("45.0"));
assert_eq!(output.class.as_deref(), Some("normal")); assert_eq!(output.class.as_deref(), Some("normal"));
@@ -70,8 +72,8 @@ mod tests {
assert_eq!(output.tooltip.as_deref(), Some("Test CPU")); assert_eq!(output.tooltip.as_deref(), Some("Test CPU"));
} }
#[test] #[tokio::test]
fn test_cpu_high() { async fn test_cpu_high() {
let state = mock_state(AppState { let state = mock_state(AppState {
cpu: CpuState { cpu: CpuState {
usage: 80.0, usage: 80.0,
@@ -81,12 +83,12 @@ mod tests {
..Default::default() ..Default::default()
}); });
let config = Config::default(); let config = Config::default();
let output = CpuModule.run(&config, &state, &[]).unwrap(); let output = CpuModule.run(&config, &state, &[]).await.unwrap();
assert_eq!(output.class.as_deref(), Some("high")); assert_eq!(output.class.as_deref(), Some("high"));
} }
#[test] #[tokio::test]
fn test_cpu_max() { async fn test_cpu_max() {
let state = mock_state(AppState { let state = mock_state(AppState {
cpu: CpuState { cpu: CpuState {
usage: 99.0, usage: 99.0,
@@ -96,7 +98,7 @@ mod tests {
..Default::default() ..Default::default()
}); });
let config = Config::default(); let config = Config::default();
let output = CpuModule.run(&config, &state, &[]).unwrap(); let output = CpuModule.run(&config, &state, &[]).await.unwrap();
assert_eq!(output.class.as_deref(), Some("max")); assert_eq!(output.class.as_deref(), Some("max"));
} }
} }
+22 -15
View File
@@ -1,20 +1,24 @@
use crate::config::Config; use crate::config::Config;
use crate::error::{FluxoError, Result};
use crate::modules::WaybarModule; use crate::modules::WaybarModule;
use crate::output::WaybarOutput; use crate::output::WaybarOutput;
use crate::state::SharedState; use crate::state::SharedState;
use crate::utils::{TokenValue, format_template}; use crate::utils::{TokenValue, format_template};
use anyhow::Result;
pub struct DiskModule; pub struct DiskModule;
impl WaybarModule for DiskModule { impl WaybarModule for DiskModule {
fn run(&self, config: &Config, state: &SharedState, args: &[&str]) -> Result<WaybarOutput> { async fn run(
&self,
config: &Config,
state: &SharedState,
args: &[&str],
) -> Result<WaybarOutput> {
let mountpoint = args.first().unwrap_or(&"/"); let mountpoint = args.first().unwrap_or(&"/");
let disks = if let Ok(s) = state.read() { let disks = {
let s = state.read().await;
s.disks.clone() s.disks.clone()
} else {
return Err(anyhow::anyhow!("Failed to read state"));
}; };
for disk in &disks { for disk in &disks {
@@ -62,7 +66,10 @@ impl WaybarModule for DiskModule {
} }
} }
Err(anyhow::anyhow!("Mountpoint {} not found", mountpoint)) Err(FluxoError::Module {
module: "disk",
message: format!("Mountpoint {} not found", mountpoint),
})
} }
} }
@@ -83,30 +90,30 @@ mod tests {
}) })
} }
#[test] #[tokio::test]
fn test_disk_found() { async fn test_disk_found() {
let gb = 1024 * 1024 * 1024; let gb = 1024 * 1024 * 1024;
let state = state_with_disk("/", 100 * gb, 60 * gb); let state = state_with_disk("/", 100 * gb, 60 * gb);
let config = Config::default(); let config = Config::default();
let output = DiskModule.run(&config, &state, &["/"]).unwrap(); let output = DiskModule.run(&config, &state, &["/"]).await.unwrap();
assert_eq!(output.class.as_deref(), Some("normal")); assert_eq!(output.class.as_deref(), Some("normal"));
assert_eq!(output.percentage, Some(40)); // 40% used assert_eq!(output.percentage, Some(40)); // 40% used
} }
#[test] #[tokio::test]
fn test_disk_high() { async fn test_disk_high() {
let gb = 1024 * 1024 * 1024; let gb = 1024 * 1024 * 1024;
let state = state_with_disk("/", 100 * gb, 15 * gb); let state = state_with_disk("/", 100 * gb, 15 * gb);
let config = Config::default(); let config = Config::default();
let output = DiskModule.run(&config, &state, &["/"]).unwrap(); let output = DiskModule.run(&config, &state, &["/"]).await.unwrap();
assert_eq!(output.class.as_deref(), Some("high")); // 85% used assert_eq!(output.class.as_deref(), Some("high")); // 85% used
} }
#[test] #[tokio::test]
fn test_disk_not_found() { async fn test_disk_not_found() {
let state = mock_state(AppState::default()); let state = mock_state(AppState::default());
let config = Config::default(); let config = Config::default();
let result = DiskModule.run(&config, &state, &["/nonexistent"]); let result = DiskModule.run(&config, &state, &["/nonexistent"]).await;
assert!(result.is_err()); assert!(result.is_err());
} }
} }
+8 -2
View File
@@ -1,8 +1,9 @@
use crate::config::Config; use crate::config::Config;
use crate::error::Result;
use crate::modules::WaybarModule; use crate::modules::WaybarModule;
use crate::output::WaybarOutput; use crate::output::WaybarOutput;
use crate::state::SharedState; use crate::state::SharedState;
use anyhow::{Result, anyhow}; use anyhow::anyhow;
use std::env; use std::env;
use std::io::{Read, Write}; use std::io::{Read, Write};
use std::os::unix::net::UnixStream; use std::os::unix::net::UnixStream;
@@ -10,7 +11,12 @@ use std::os::unix::net::UnixStream;
pub struct GameModule; pub struct GameModule;
impl WaybarModule for GameModule { impl WaybarModule for GameModule {
fn run(&self, config: &Config, _state: &SharedState, _args: &[&str]) -> Result<WaybarOutput> { async fn run(
&self,
config: &Config,
_state: &SharedState,
_args: &[&str],
) -> Result<WaybarOutput> {
let is_gamemode = hyprland_ipc("j/getoption animations:enabled") let is_gamemode = hyprland_ipc("j/getoption animations:enabled")
.map(|stdout| stdout.contains("\"int\": 0")) .map(|stdout| stdout.contains("\"int\": 0"))
.unwrap_or(false); .unwrap_or(false);
+8 -14
View File
@@ -1,16 +1,21 @@
use crate::config::Config; use crate::config::Config;
use crate::error::Result;
use crate::modules::WaybarModule; use crate::modules::WaybarModule;
use crate::output::WaybarOutput; use crate::output::WaybarOutput;
use crate::state::SharedState; use crate::state::SharedState;
use crate::utils::{TokenValue, format_template}; use crate::utils::{TokenValue, format_template};
use anyhow::Result;
pub struct GpuModule; pub struct GpuModule;
impl WaybarModule for GpuModule { impl WaybarModule for GpuModule {
fn run(&self, config: &Config, state: &SharedState, _args: &[&str]) -> Result<WaybarOutput> { async fn run(
&self,
config: &Config,
state: &SharedState,
_args: &[&str],
) -> Result<WaybarOutput> {
let (active, vendor, usage, vram_used, vram_total, temp, model) = { let (active, vendor, usage, vram_used, vram_total, temp, model) = {
if let Ok(state_lock) = state.read() { let state_lock = state.read().await;
( (
state_lock.gpu.active, state_lock.gpu.active,
state_lock.gpu.vendor.clone(), state_lock.gpu.vendor.clone(),
@@ -20,17 +25,6 @@ impl WaybarModule for GpuModule {
state_lock.gpu.temp, state_lock.gpu.temp,
state_lock.gpu.model.clone(), state_lock.gpu.model.clone(),
) )
} else {
(
false,
String::from("Unknown"),
0.0,
0.0,
0.0,
0.0,
String::from("Unknown"),
)
}
}; };
if !active { if !active {
+4 -6
View File
@@ -24,7 +24,7 @@ impl HardwareDaemon {
} }
} }
pub fn poll_fast(&mut self, state: SharedState) { pub async fn poll_fast(&mut self, state: SharedState) {
self.sys.refresh_cpu_usage(); self.sys.refresh_cpu_usage();
self.sys.refresh_memory(); self.sys.refresh_memory();
self.components.refresh(true); self.components.refresh(true);
@@ -70,7 +70,7 @@ impl HardwareDaemon {
} }
} }
if let Ok(mut state_lock) = state.write() { let mut state_lock = state.write().await;
state_lock.cpu.usage = cpu_usage as f64; state_lock.cpu.usage = cpu_usage as f64;
state_lock.cpu.temp = cpu_temp; state_lock.cpu.temp = cpu_temp;
state_lock.cpu.model = cpu_model; state_lock.cpu.model = cpu_model;
@@ -84,9 +84,8 @@ impl HardwareDaemon {
state_lock.sys.uptime = uptime; state_lock.sys.uptime = uptime;
state_lock.sys.process_count = process_count; state_lock.sys.process_count = process_count;
} }
}
pub fn poll_slow(&mut self, state: SharedState) { pub async fn poll_slow(&mut self, state: SharedState) {
// 1. Gather GPU data outside of lock // 1. Gather GPU data outside of lock
let mut gpu_state = crate::state::GpuState::default(); let mut gpu_state = crate::state::GpuState::default();
self.gpu_poll_counter = (self.gpu_poll_counter + 1) % 5; self.gpu_poll_counter = (self.gpu_poll_counter + 1) % 5;
@@ -113,7 +112,7 @@ impl HardwareDaemon {
} }
// 3. Apply to state // 3. Apply to state
if let Ok(mut state_lock) = state.write() { let mut state_lock = state.write().await;
if should_poll_gpu { if should_poll_gpu {
state_lock.gpu = gpu_state; state_lock.gpu = gpu_state;
} }
@@ -122,7 +121,6 @@ impl HardwareDaemon {
state_lock.disks = d; state_lock.disks = d;
} }
} }
}
fn poll_gpu(&mut self, gpu: &mut crate::state::GpuState) { fn poll_gpu(&mut self, gpu: &mut crate::state::GpuState) {
gpu.active = false; gpu.active = false;
+17 -15
View File
@@ -1,20 +1,22 @@
use crate::config::Config; use crate::config::Config;
use crate::error::Result;
use crate::modules::WaybarModule; use crate::modules::WaybarModule;
use crate::output::WaybarOutput; use crate::output::WaybarOutput;
use crate::state::SharedState; use crate::state::SharedState;
use crate::utils::{TokenValue, format_template}; use crate::utils::{TokenValue, format_template};
use anyhow::Result;
pub struct MemoryModule; pub struct MemoryModule;
impl WaybarModule for MemoryModule { impl WaybarModule for MemoryModule {
fn run(&self, config: &Config, state: &SharedState, _args: &[&str]) -> Result<WaybarOutput> { async fn run(
&self,
config: &Config,
state: &SharedState,
_args: &[&str],
) -> Result<WaybarOutput> {
let (used_gb, total_gb) = { let (used_gb, total_gb) = {
if let Ok(state_lock) = state.read() { let state_lock = state.read().await;
(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 { let ratio = if total_gb > 0.0 {
@@ -53,8 +55,8 @@ mod tests {
use super::*; use super::*;
use crate::state::{AppState, MemoryState, mock_state}; use crate::state::{AppState, MemoryState, mock_state};
#[test] #[tokio::test]
fn test_memory_normal() { async fn test_memory_normal() {
let state = mock_state(AppState { let state = mock_state(AppState {
memory: MemoryState { memory: MemoryState {
used_gb: 8.0, used_gb: 8.0,
@@ -63,15 +65,15 @@ mod tests {
..Default::default() ..Default::default()
}); });
let config = Config::default(); let config = Config::default();
let output = MemoryModule.run(&config, &state, &[]).unwrap(); let output = MemoryModule.run(&config, &state, &[]).await.unwrap();
assert!(output.text.contains("8.00")); assert!(output.text.contains("8.00"));
assert!(output.text.contains("32.00")); assert!(output.text.contains("32.00"));
assert_eq!(output.class.as_deref(), Some("normal")); assert_eq!(output.class.as_deref(), Some("normal"));
assert_eq!(output.percentage, Some(25)); // 8/32 = 25% assert_eq!(output.percentage, Some(25)); // 8/32 = 25%
} }
#[test] #[tokio::test]
fn test_memory_high() { async fn test_memory_high() {
let state = mock_state(AppState { let state = mock_state(AppState {
memory: MemoryState { memory: MemoryState {
used_gb: 26.0, used_gb: 26.0,
@@ -80,12 +82,12 @@ mod tests {
..Default::default() ..Default::default()
}); });
let config = Config::default(); let config = Config::default();
let output = MemoryModule.run(&config, &state, &[]).unwrap(); let output = MemoryModule.run(&config, &state, &[]).await.unwrap();
assert_eq!(output.class.as_deref(), Some("high")); // 81% assert_eq!(output.class.as_deref(), Some("high")); // 81%
} }
#[test] #[tokio::test]
fn test_memory_zero_total() { async fn test_memory_zero_total() {
let state = mock_state(AppState { let state = mock_state(AppState {
memory: MemoryState { memory: MemoryState {
used_gb: 0.0, used_gb: 0.0,
@@ -94,7 +96,7 @@ mod tests {
..Default::default() ..Default::default()
}); });
let config = Config::default(); let config = Config::default();
let output = MemoryModule.run(&config, &state, &[]).unwrap(); let output = MemoryModule.run(&config, &state, &[]).await.unwrap();
assert_eq!(output.class.as_deref(), Some("normal")); assert_eq!(output.class.as_deref(), Some("normal"));
assert_eq!(output.percentage, Some(0)); assert_eq!(output.percentage, Some(0));
} }
+7 -2
View File
@@ -12,10 +12,15 @@ pub mod power;
pub mod sys; pub mod sys;
use crate::config::Config; use crate::config::Config;
use crate::error::Result as FluxoResult;
use crate::output::WaybarOutput; use crate::output::WaybarOutput;
use crate::state::SharedState; use crate::state::SharedState;
use anyhow::Result;
pub trait WaybarModule { pub trait WaybarModule {
fn run(&self, config: &Config, state: &SharedState, args: &[&str]) -> Result<WaybarOutput>; fn run(
&self,
config: &Config,
state: &SharedState,
args: &[&str],
) -> impl std::future::Future<Output = FluxoResult<WaybarOutput>> + Send;
} }
+27 -27
View File
@@ -1,9 +1,9 @@
use crate::config::Config; use crate::config::Config;
use crate::error::Result;
use crate::modules::WaybarModule; use crate::modules::WaybarModule;
use crate::output::WaybarOutput; use crate::output::WaybarOutput;
use crate::state::SharedState; use crate::state::SharedState;
use crate::utils::{TokenValue, format_template}; use crate::utils::{TokenValue, format_template};
use anyhow::Result;
use nix::ifaddrs::getifaddrs; use nix::ifaddrs::getifaddrs;
use std::fs; use std::fs;
use std::time::{SystemTime, UNIX_EPOCH}; use std::time::{SystemTime, UNIX_EPOCH};
@@ -29,7 +29,7 @@ impl NetworkDaemon {
} }
} }
pub fn poll(&mut self, state: SharedState) { pub async fn poll(&mut self, state: SharedState) {
// Re-detect interface on every poll to catch VPNs or route changes immediately // Re-detect interface on every poll to catch VPNs or route changes immediately
if let Ok(iface) = get_primary_interface() if let Ok(iface) = get_primary_interface()
&& !iface.is_empty() && !iface.is_empty()
@@ -62,14 +62,14 @@ impl NetworkDaemon {
/ 1024.0 / 1024.0
/ 1024.0; / 1024.0;
if let Ok(mut state_lock) = state.write() { let mut state_lock = state.write().await;
state_lock.network.rx_mbps = rx_mbps; state_lock.network.rx_mbps = rx_mbps;
state_lock.network.tx_mbps = tx_mbps; state_lock.network.tx_mbps = tx_mbps;
state_lock.network.interface = interface.clone(); state_lock.network.interface = interface.clone();
state_lock.network.ip = self.cached_ip.clone().unwrap_or_default(); state_lock.network.ip = self.cached_ip.clone().unwrap_or_default();
} } else {
} else if let Ok(mut state_lock) = state.write() {
// First poll: no speed data yet, but update interface/ip // First poll: no speed data yet, but update interface/ip
let mut state_lock = state.write().await;
state_lock.network.interface = interface.clone(); state_lock.network.interface = interface.clone();
state_lock.network.ip = self.cached_ip.clone().unwrap_or_default(); state_lock.network.ip = self.cached_ip.clone().unwrap_or_default();
} }
@@ -81,8 +81,9 @@ impl NetworkDaemon {
// Read failed, might be down // Read failed, might be down
self.cached_interface = None; self.cached_interface = None;
} }
} else if let Ok(mut state_lock) = state.write() { } else {
// No interface detected // No interface detected
let mut state_lock = state.write().await;
state_lock.network.interface.clear(); state_lock.network.interface.clear();
state_lock.network.ip.clear(); state_lock.network.ip.clear();
} }
@@ -90,21 +91,20 @@ impl NetworkDaemon {
} }
impl WaybarModule for NetworkModule { impl WaybarModule for NetworkModule {
fn run(&self, config: &Config, state: &SharedState, _args: &[&str]) -> Result<WaybarOutput> { async fn run(
let (interface, ip, rx_mbps, tx_mbps) = if let Ok(s) = state.read() { &self,
config: &Config,
state: &SharedState,
_args: &[&str],
) -> Result<WaybarOutput> {
let (interface, ip, rx_mbps, tx_mbps) = {
let s = state.read().await;
( (
s.network.interface.clone(), s.network.interface.clone(),
s.network.ip.clone(), s.network.ip.clone(),
s.network.rx_mbps, s.network.rx_mbps,
s.network.tx_mbps, s.network.tx_mbps,
) )
} else {
return Ok(WaybarOutput {
text: "No connection".to_string(),
tooltip: None,
class: None,
percentage: None,
});
}; };
if interface.is_empty() { if interface.is_empty() {
@@ -208,16 +208,16 @@ mod tests {
use super::*; use super::*;
use crate::state::{AppState, NetworkState, mock_state}; use crate::state::{AppState, NetworkState, mock_state};
#[test] #[tokio::test]
fn test_network_no_connection() { async fn test_network_no_connection() {
let state = mock_state(AppState::default()); let state = mock_state(AppState::default());
let config = Config::default(); let config = Config::default();
let output = NetworkModule.run(&config, &state, &[]).unwrap(); let output = NetworkModule.run(&config, &state, &[]).await.unwrap();
assert_eq!(output.text, "No connection"); assert_eq!(output.text, "No connection");
} }
#[test] #[tokio::test]
fn test_network_connected() { async fn test_network_connected() {
let state = mock_state(AppState { let state = mock_state(AppState {
network: NetworkState { network: NetworkState {
rx_mbps: 1.5, rx_mbps: 1.5,
@@ -228,15 +228,15 @@ mod tests {
..Default::default() ..Default::default()
}); });
let config = Config::default(); let config = Config::default();
let output = NetworkModule.run(&config, &state, &[]).unwrap(); let output = NetworkModule.run(&config, &state, &[]).await.unwrap();
assert!(output.text.contains("eth0")); assert!(output.text.contains("eth0"));
assert!(output.text.contains("192.168.1.100")); assert!(output.text.contains("192.168.1.100"));
assert!(output.text.contains("1.50")); assert!(output.text.contains("1.50"));
assert_eq!(output.class.as_deref(), Some("eth0")); assert_eq!(output.class.as_deref(), Some("eth0"));
} }
#[test] #[tokio::test]
fn test_network_vpn_prefix() { async fn test_network_vpn_prefix() {
let state = mock_state(AppState { let state = mock_state(AppState {
network: NetworkState { network: NetworkState {
rx_mbps: 0.0, rx_mbps: 0.0,
@@ -247,12 +247,12 @@ mod tests {
..Default::default() ..Default::default()
}); });
let config = Config::default(); let config = Config::default();
let output = NetworkModule.run(&config, &state, &[]).unwrap(); let output = NetworkModule.run(&config, &state, &[]).await.unwrap();
assert!(output.text.starts_with(" ")); assert!(output.text.starts_with(" "));
} }
#[test] #[tokio::test]
fn test_network_no_ip() { async fn test_network_no_ip() {
let state = mock_state(AppState { let state = mock_state(AppState {
network: NetworkState { network: NetworkState {
rx_mbps: 0.0, rx_mbps: 0.0,
@@ -263,7 +263,7 @@ mod tests {
..Default::default() ..Default::default()
}); });
let config = Config::default(); let config = Config::default();
let output = NetworkModule.run(&config, &state, &[]).unwrap(); let output = NetworkModule.run(&config, &state, &[]).await.unwrap();
assert!(output.text.contains("No IP")); assert!(output.text.contains("No IP"));
} }
} }
+7 -2
View File
@@ -1,15 +1,20 @@
use crate::config::Config; use crate::config::Config;
use crate::error::Result;
use crate::modules::WaybarModule; use crate::modules::WaybarModule;
use crate::output::WaybarOutput; use crate::output::WaybarOutput;
use crate::state::SharedState; use crate::state::SharedState;
use crate::utils::{TokenValue, format_template}; use crate::utils::{TokenValue, format_template};
use anyhow::Result;
use std::fs; use std::fs;
pub struct PowerModule; pub struct PowerModule;
impl WaybarModule for PowerModule { impl WaybarModule for PowerModule {
fn run(&self, config: &Config, _state: &SharedState, _args: &[&str]) -> Result<WaybarOutput> { async fn run(
&self,
config: &Config,
_state: &SharedState,
_args: &[&str],
) -> Result<WaybarOutput> {
let critical_threshold = 15; let critical_threshold = 15;
let warning_threshold = 50; let warning_threshold = 50;
+8 -6
View File
@@ -1,16 +1,21 @@
use crate::config::Config; use crate::config::Config;
use crate::error::Result;
use crate::modules::WaybarModule; use crate::modules::WaybarModule;
use crate::output::WaybarOutput; use crate::output::WaybarOutput;
use crate::state::SharedState; use crate::state::SharedState;
use crate::utils::{TokenValue, format_template}; use crate::utils::{TokenValue, format_template};
use anyhow::Result;
pub struct SysModule; pub struct SysModule;
impl WaybarModule for SysModule { impl WaybarModule for SysModule {
fn run(&self, config: &Config, state: &SharedState, _args: &[&str]) -> Result<WaybarOutput> { async fn run(
&self,
config: &Config,
state: &SharedState,
_args: &[&str],
) -> Result<WaybarOutput> {
let (load1, load5, load15, uptime_secs, process_count) = { let (load1, load5, load15, uptime_secs, process_count) = {
if let Ok(state_lock) = state.read() { let state_lock = state.read().await;
( (
state_lock.sys.load_1, state_lock.sys.load_1,
state_lock.sys.load_5, state_lock.sys.load_5,
@@ -18,9 +23,6 @@ impl WaybarModule for SysModule {
state_lock.sys.uptime, state_lock.sys.uptime,
state_lock.sys.process_count, state_lock.sys.process_count,
) )
} else {
(0.0, 0.0, 0.0, 0, 0)
}
}; };
let hours = uptime_secs / 3600; let hours = uptime_secs / 3600;
+12 -1
View File
@@ -1,4 +1,7 @@
use std::sync::{Arc, RwLock}; use std::collections::HashMap;
use std::sync::Arc;
use tokio::sync::RwLock;
use tokio::time::Instant;
#[derive(Default, Clone)] #[derive(Default, Clone)]
pub struct AppState { pub struct AppState {
@@ -10,6 +13,14 @@ pub struct AppState {
pub disks: Vec<DiskInfo>, pub disks: Vec<DiskInfo>,
pub bluetooth: BtState, pub bluetooth: BtState,
pub audio: AudioState, pub audio: AudioState,
pub health: HashMap<String, ModuleHealth>,
}
#[derive(Clone, Default)]
pub struct ModuleHealth {
pub consecutive_failures: u32,
pub last_failure: Option<Instant>,
pub backoff_until: Option<Instant>,
} }
#[derive(Default, Clone)] #[derive(Default, Clone)]