13 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
nvrl 6fa14c0b84 fixed testcase
Release / Build and Release (push) Successful in 2m1s
2026-04-01 14:27:26 +02:00
nvrl 7b5b2fba6b added libpulse-dev to ci
Release / Build and Release (push) Failing after 1m58s
2026-04-01 14:22:17 +02:00
nvrl 939336fddf WIP
Release / Build and Release (push) Failing after 1m25s
2026-04-01 14:19:03 +02:00
nvrl 7805fd06d0 fixed ci
Release / Build and Release (push) Failing after 58s
2026-04-01 14:17:07 +02:00
nvrl 4407437928 implemented separete hardware threads
Release / Build and Release (push) Failing after 37s
2026-04-01 14:15:18 +02:00
nvrl 7a8f70938e removed legacy code and clippy fixes + fixed ci
Release / Build and Release (push) Failing after 38s
2026-04-01 14:01:16 +02:00
nvrl bd18bbf866 version bump
Release / Build and Release (push) Failing after 39s
2026-04-01 12:23:41 +02:00
nvrl 2ce5aceae0 implemented library calls instead of cli calls
Release / Build and Release (push) Has been cancelled
2026-04-01 12:23:04 +02:00
30 changed files with 2569 additions and 828 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
+1 -1
View File
@@ -22,7 +22,7 @@ jobs:
cache: false cache: false
- name: Install packaging tools - name: Install packaging tools
run: apt-get update && apt-get install -y dpkg-dev run: apt-get update && apt-get install -y dpkg-dev pkg-config libdbus-1-dev protobuf-compiler libpulse-dev
- name: Get Version - name: Get Version
id: get_version id: get_version
Generated
+855 -33
View File
File diff suppressed because it is too large Load Diff
+12 -3
View File
@@ -1,19 +1,28 @@
[package] [package]
name = "fluxo-rs" name = "fluxo-rs"
version = "0.2.0" version = "0.3.1"
edition = "2024" edition = "2024"
[dependencies] [dependencies]
anyhow = "1.0.102" anyhow = "1.0.102"
clap = { version = "4.6.0", features = ["derive"] } clap = { version = "4.6.0", features = ["derive"] }
ctrlc = "3" ctrlc = "3"
regex = "1.10" 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" }
bluer = { version = "0.17", features = ["bluetoothd", "rfcomm", "id"] }
tokio = { version = "1", features = ["rt", "rt-multi-thread", "sync", "time", "macros", "signal"] }
tokio-util = { version = "0.7", features = ["codec"] }
futures = "0.3"
libpulse-binding = "2.30"
nix = { version = "0.31", features = ["net"] }
[dev-dependencies] [dev-dependencies]
tempfile = "3" tempfile = "3"
+44 -74
View File
@@ -1,96 +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 | `{alias}` | | `bt` | Bluetooth status & plugins | `{alias}`, `{mac}`, `{left}`, `{right}`, `{anc}` |
| `buds` | Pixel Buds Pro control | `{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.
+46
View File
@@ -0,0 +1,46 @@
# Fluxo configuration example
[general]
menu_command = "fuzzel --dmenu --prompt '{prompt}'"
[network]
format = "{interface} ({ip}):  {rx:>5.2} MB/s  {tx:>5.2} MB/s"
[cpu]
format = "CPU: {usage:>4.1}% {temp:>4.1}C"
[memory]
format = "{used:>5.2}/{total:>5.2}GB"
[gpu]
format_amd = "AMD: {usage:>3.0}% {vram_used:>4.1}/{vram_total:>4.1}GB {temp:>4.1}C"
format_intel = "iGPU: {usage:>3.0}%"
format_nvidia = "NV: {usage:>3.0}% {vram_used:>4.1}/{vram_total:>4.1}GB {temp:>4.1}C"
[sys]
format = "UP: {uptime} | LOAD: {load1:>4.2} {load5:>4.2} {load15:>4.2}"
[disk]
format = "{mount} {used:>5.1}/{total:>5.1}G"
[pool]
format = "{used:>4.0}G / {total:>4.0}G"
[power]
format = "{percentage:>3}% {icon}"
[bt]
format_connected = "{alias} 󰂰"
format_plugin = "{alias} [{left}|{right}] {anc} 󰂰"
format_disconnected = "󰂯"
format_disabled = "󰂲 Off"
[audio]
format_sink_unmuted = "{name} {volume:>3}% {icon}"
format_sink_muted = "{name} {icon}"
format_source_unmuted = "{name} {volume:>3}% {icon}"
format_source_muted = "{name} {icon}"
[game]
format_active = "<span size='large'>󰊖</span>"
format_inactive = "<span size='large'></span>"
+20 -33
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,
@@ -26,8 +26,6 @@ pub struct Config {
#[serde(default)] #[serde(default)]
pub power: PowerConfig, pub power: PowerConfig,
#[serde(default)] #[serde(default)]
pub buds: BudsConfig,
#[serde(default)]
pub audio: AudioConfig, pub audio: AudioConfig,
#[serde(default)] #[serde(default)]
pub bt: BtConfig, pub bt: BtConfig,
@@ -35,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,
} }
@@ -48,7 +46,7 @@ impl Default for GeneralConfig {
} }
} }
#[derive(Deserialize)] #[derive(Deserialize, Clone)]
pub struct NetworkConfig { pub struct NetworkConfig {
pub format: String, pub format: String,
} }
@@ -61,7 +59,7 @@ impl Default for NetworkConfig {
} }
} }
#[derive(Deserialize)] #[derive(Deserialize, Clone)]
pub struct CpuConfig { pub struct CpuConfig {
pub format: String, pub format: String,
} }
@@ -74,7 +72,7 @@ impl Default for CpuConfig {
} }
} }
#[derive(Deserialize)] #[derive(Deserialize, Clone)]
pub struct MemoryConfig { pub struct MemoryConfig {
pub format: String, pub format: String,
} }
@@ -87,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,
@@ -106,7 +104,7 @@ impl Default for GpuConfig {
} }
} }
#[derive(Deserialize)] #[derive(Deserialize, Clone)]
pub struct SysConfig { pub struct SysConfig {
pub format: String, pub format: String,
} }
@@ -119,7 +117,7 @@ impl Default for SysConfig {
} }
} }
#[derive(Deserialize)] #[derive(Deserialize, Clone)]
pub struct DiskConfig { pub struct DiskConfig {
pub format: String, pub format: String,
} }
@@ -132,7 +130,7 @@ impl Default for DiskConfig {
} }
} }
#[derive(Deserialize)] #[derive(Deserialize, Clone)]
pub struct PoolConfig { pub struct PoolConfig {
pub format: String, pub format: String,
} }
@@ -145,7 +143,7 @@ impl Default for PoolConfig {
} }
} }
#[derive(Deserialize)] #[derive(Deserialize, Clone)]
pub struct PowerConfig { pub struct PowerConfig {
pub format: String, pub format: String,
} }
@@ -158,24 +156,7 @@ impl Default for PowerConfig {
} }
} }
#[derive(Deserialize)] #[derive(Deserialize, Clone)]
pub struct BudsConfig {
pub mac: String,
pub format: String,
pub format_disconnected: String,
}
impl Default for BudsConfig {
fn default() -> Self {
Self {
mac: "B4:23:A2:09:D3:53".to_string(),
format: "{left} | {right} | {anc}".to_string(),
format_disconnected: "<span size='large'></span>".to_string(),
}
}
}
#[derive(Deserialize)]
pub struct AudioConfig { pub struct AudioConfig {
pub format_sink_unmuted: String, pub format_sink_unmuted: String,
pub format_sink_muted: String, pub format_sink_muted: String,
@@ -194,9 +175,10 @@ 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_disconnected: String, pub format_disconnected: String,
pub format_disabled: String, pub format_disabled: String,
} }
@@ -205,13 +187,14 @@ impl Default for BtConfig {
fn default() -> Self { fn default() -> Self {
Self { Self {
format_connected: "{alias} 󰂰".to_string(), format_connected: "{alias} 󰂰".to_string(),
format_plugin: "{alias} [{left}|{right}] {anc} 󰂰".to_string(),
format_disconnected: "󰂯".to_string(), format_disconnected: "󰂯".to_string(),
format_disabled: "󰂲 Off".to_string(), format_disabled: "󰂲 Off".to_string(),
} }
} }
} }
#[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,
@@ -275,7 +258,6 @@ impl Config {
validate_format("disk", &self.disk.format, &["mount", "used", "total"]); validate_format("disk", &self.disk.format, &["mount", "used", "total"]);
validate_format("pool", &self.pool.format, &["used", "total"]); validate_format("pool", &self.pool.format, &["used", "total"]);
validate_format("power", &self.power.format, &["percentage", "icon"]); validate_format("power", &self.power.format, &["percentage", "icon"]);
validate_format("buds", &self.buds.format, &["left", "right", "anc"]);
validate_format( validate_format(
"audio.sink_unmuted", "audio.sink_unmuted",
&self.audio.format_sink_unmuted, &self.audio.format_sink_unmuted,
@@ -297,6 +279,11 @@ impl Config {
&["name", "icon"], &["name", "icon"],
); );
validate_format("bt.connected", &self.bt.format_connected, &["alias"]); validate_format("bt.connected", &self.bt.format_connected, &["alias"]);
validate_format(
"bt.plugin",
&self.bt.format_plugin,
&["alias", "left", "right", "anc", "mac"],
);
} }
} }
+224 -108
View File
@@ -1,19 +1,20 @@
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::bt::BtDaemon;
use crate::modules::hardware::HardwareDaemon; use crate::modules::hardware::HardwareDaemon;
use crate::modules::network::NetworkDaemon; 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 {
@@ -27,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() {
@@ -41,86 +42,128 @@ 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 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 background polling thread"); let mut daemon = NetworkDaemon::new();
let mut network_daemon = NetworkDaemon::new(); loop {
let mut hardware_daemon = HardwareDaemon::new(); daemon.poll(Arc::clone(&poll_state)).await;
while poll_running.load(Ordering::SeqCst) { sleep(Duration::from_secs(1)).await;
network_daemon.poll(Arc::clone(&poll_state));
hardware_daemon.poll(Arc::clone(&poll_state));
thread::sleep(Duration::from_secs(1));
} }
}); });
// 2. Fast Hardware Task (CPU, Mem, Load)
let poll_state = Arc::clone(&state);
tokio::spawn(async move {
info!("Starting Fast Hardware polling task");
let mut daemon = HardwareDaemon::new();
loop {
daemon.poll_fast(Arc::clone(&poll_state)).await;
sleep(Duration::from_secs(1)).await;
}
});
// 3. Slow Hardware Task (GPU, Disks)
let poll_state = Arc::clone(&state);
tokio::spawn(async move {
info!("Starting Slow Hardware polling task");
let mut daemon = HardwareDaemon::new();
loop {
daemon.poll_slow(Arc::clone(&poll_state)).await;
sleep(Duration::from_secs(1)).await;
}
});
// 4. Bluetooth Task
let poll_state = Arc::clone(&state);
let poll_config = Arc::clone(&config);
tokio::spawn(async move {
info!("Starting Bluetooth polling task");
let mut daemon = BtDaemon::new();
loop {
let config = poll_config.read().await;
daemon.poll(Arc::clone(&poll_state), &config).await;
sleep(Duration::from_secs(1)).await;
}
});
// 5. Audio Thread (Event driven - pulse usually needs its own thread)
let audio_daemon = AudioDaemon::new();
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() { }
Ok((mut stream, _)) => { res = listener.accept() => {
let state_clone = Arc::clone(&state); match res {
let config_clone = Arc::clone(&config); Ok((mut stream, _)) => {
let cp_clone = config_path_clone.clone(); let state_clone = Arc::clone(&state);
thread::spawn(move || { let config_clone = Arc::clone(&config);
let mut reader = BufReader::new(&stream); let cp_clone = config_path_clone.clone();
let mut request = String::new(); tokio::spawn(async move {
if let Err(e) = reader.read_line(&mut request) { let (reader, mut writer) = stream.split();
error!("Failed to read from IPC stream: {}", e); let mut reader = BufReader::new(reader);
return; let mut request = String::new();
} if let Err(e) = reader.read_line(&mut request).await {
drop(reader); error!("Failed to read from IPC stream: {}", e);
return;
let request = request.trim();
if request.is_empty() {
return;
}
let parts: Vec<&str> = request.split_whitespace().collect();
if let Some(module_name) = parts.first() {
if *module_name == "reload" {
info!("Reloading configuration...");
let new_config = crate::config::load_config(cp_clone);
if let Ok(mut config_lock) = config_clone.write() {
*config_lock = new_config;
let _ = stream.write_all(b"{\"text\":\"ok\"}");
info!("Configuration reloaded successfully.");
} else {
error!("Failed to acquire write lock for configuration reload.");
} }
let _ = stream.shutdown(Shutdown::Write);
return;
}
debug!(module = module_name, args = ?&parts[1..], "Handling IPC request"); let request = request.trim();
let response = if request.is_empty() {
handle_request(module_name, &parts[1..], &state_clone, &config_clone); return;
if let Err(e) = stream.write_all(response.as_bytes()) { }
error!("Failed to write IPC response: {}", e);
} let parts: Vec<&str> = request.split_whitespace().collect();
let _ = stream.shutdown(Shutdown::Write); if let Some(module_name) = parts.first() {
if *module_name == "reload" {
info!("Reloading configuration...");
let new_config = crate::config::load_config(cp_clone);
let mut config_lock = config_clone.write().await;
*config_lock = new_config;
let _ = writer.write_all(b"{\"text\":\"ok\"}").await;
info!("Configuration reloaded successfully.");
return;
}
debug!(module = module_name, args = ?&parts[1..], "Handling IPC request");
let response =
handle_request(module_name, &parts[1..], &state_clone, &config_clone).await;
if let Err(e) = writer.write_all(response.as_bytes()).await {
if e.kind() == std::io::ErrorKind::BrokenPipe
|| e.kind() == std::io::ErrorKind::ConnectionReset
{
debug!(
"IPC client disconnected before response could be sent: {}",
e
);
} else {
error!("Failed to write IPC response: {}", e);
}
}
}
});
} }
}); Err(e) => error!("Failed to accept incoming connection: {}", e),
}
} }
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),
} }
} }
@@ -128,54 +171,127 @@ pub fn run_daemon(config_path: Option<PathBuf>) -> Result<()> {
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 = {
} else { let lock = state.read().await;
error!("Failed to acquire read lock for configuration."); if let Some(health) = lock.health.get(module_name) {
return "{\"text\":\"error: config lock failed\"}".to_string(); if let Some(until) = health.backoff_until {
}; Instant::now() < until
} else {
let result = match module_name { false
"net" | "network" => crate::modules::network::NetworkModule.run(&config, state, args), }
"cpu" => crate::modules::cpu::CpuModule.run(&config, state, args), } else {
"mem" | "memory" => crate::modules::memory::MemoryModule.run(&config, state, args), false
"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),
"buds" => crate::modules::buds::BudsModule.run(&config, state, args),
"power" => crate::modules::power::PowerModule.run(&config, state, args),
"game" => crate::modules::game::GameModule.run(&config, state, args),
_ => {
warn!("Received request for unknown module: '{}'", module_name);
Err(anyhow::anyhow!("Unknown module: {}", module_name))
} }
}; };
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>;
+42 -16
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;
@@ -71,11 +72,6 @@ enum Commands {
#[arg(default_value = "show")] #[arg(default_value = "show")]
action: String, action: String,
}, },
/// Pixel Buds Pro ANC and Battery
Buds {
#[arg(default_value = "show")]
action: String,
},
/// System power and battery status /// System power and battery status
Power, Power,
/// Hyprland gamemode status /// Hyprland gamemode status
@@ -93,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);
} }
@@ -125,6 +126,25 @@ fn main() {
// Client-side execution of the menu // Client-side execution of the menu
let config = config::load_config(None); let config = config::load_config(None);
let mut items = Vec::new();
// If connected, show plugin modes and disconnect
if let Ok(json_str) = ipc::request_data("bt", &["get_modes"])
&& let Ok(val) = serde_json::from_str::<serde_json::Value>(&json_str)
&& let Some(modes_str) = val.get("text").and_then(|t| t.as_str())
&& !modes_str.is_empty()
{
for mode in modes_str.lines() {
items.push(format!("Mode: {}", mode));
}
}
if !items.is_empty() {
items.push("Disconnect".to_string());
}
items.push("--- Connect Device ---".to_string());
let devices_out = match std::process::Command::new("bluetoothctl") let devices_out = match std::process::Command::new("bluetoothctl")
.args(["devices"]) .args(["devices"])
.output() .output()
@@ -137,7 +157,6 @@ fn main() {
}; };
let stdout = String::from_utf8_lossy(&devices_out.stdout); let stdout = String::from_utf8_lossy(&devices_out.stdout);
let mut items = Vec::new();
for line in stdout.lines() { for line in stdout.lines() {
if line.starts_with("Device ") { if line.starts_with("Device ") {
let parts: Vec<&str> = line.splitn(3, ' ').collect(); let parts: Vec<&str> = line.splitn(3, ' ').collect();
@@ -149,23 +168,30 @@ fn main() {
if !items.is_empty() { if !items.is_empty() {
if let Ok(selected) = if let Ok(selected) =
utils::show_menu("Connect BT: ", &items, &config.general.menu_command) utils::show_menu("BT Menu: ", &items, &config.general.menu_command)
&& let Some(mac_start) = selected.rfind('(')
&& let Some(mac_end) = selected.rfind(')')
{ {
let mac = &selected[mac_start + 1..mac_end]; if let Some(mode) = selected.strip_prefix("Mode: ") {
let _ = std::process::Command::new("bluetoothctl") handle_ipc_response(ipc::request_data("bt", &["set_mode", mode]));
.args(["connect", mac]) } else if selected == "Disconnect" {
.status(); handle_ipc_response(ipc::request_data("bt", &["disconnect"]));
} else if selected == "--- Connect Device ---" {
// Do nothing
} else if let Some(mac_start) = selected.rfind('(')
&& let Some(mac_end) = selected.rfind(')')
{
let mac = &selected[mac_start + 1..mac_end];
let _ = std::process::Command::new("bluetoothctl")
.args(["connect", mac])
.status();
}
} }
} else { } else {
info!("No paired Bluetooth devices found."); info!("No Bluetooth options found.");
} }
return; return;
} }
handle_ipc_response(ipc::request_data("bt", &[action])); handle_ipc_response(ipc::request_data("bt", &[action]));
} }
Commands::Buds { action } => handle_ipc_response(ipc::request_data("buds", &[action])),
Commands::Power => handle_ipc_response(ipc::request_data("power", &[])), Commands::Power => handle_ipc_response(ipc::request_data("power", &[])),
Commands::Game => handle_ipc_response(ipc::request_data("game", &[])), Commands::Game => handle_ipc_response(ipc::request_data("game", &[])),
} }
+233 -79
View File
@@ -1,56 +1,236 @@
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, run_command}; use crate::utils::{TokenValue, format_template};
use anyhow::{Result, anyhow}; use libpulse_binding::callbacks::ListResult;
use libpulse_binding::context::subscribe::{Facility, InterestMaskSet};
use libpulse_binding::context::{Context, FlagSet as ContextFlag};
use libpulse_binding::mainloop::threaded::Mainloop as ThreadedMainloop;
use libpulse_binding::volume::Volume;
use std::process::Command; use std::process::Command;
use std::sync::{Arc, LazyLock};
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;
impl AudioDaemon {
pub fn new() -> Self {
Self
}
pub fn start(&self, state: SharedState) {
let state_arc = Arc::clone(&state);
std::thread::spawn(move || {
let mut mainloop =
ThreadedMainloop::new().expect("Failed to create pulse threaded mainloop");
let mut context =
Context::new(&mainloop, "fluxo-rs").expect("Failed to create pulse context");
context
.connect(None, ContextFlag::NOFLAGS, None)
.expect("Failed to connect pulse context");
mainloop.start().expect("Failed to start pulse mainloop");
mainloop.lock();
// Wait for context to be ready
loop {
match context.get_state() {
libpulse_binding::context::State::Ready => break,
libpulse_binding::context::State::Failed
| libpulse_binding::context::State::Terminated => {
error!("Pulse context failed or terminated");
mainloop.unlock();
return;
}
_ => {
mainloop.unlock();
std::thread::sleep(Duration::from_millis(50));
mainloop.lock();
}
}
}
// Initial fetch
let _ = fetch_audio_data_sync(&mut context, &state_arc);
// Subscribe to events
let interest =
InterestMaskSet::SINK | InterestMaskSet::SOURCE | InterestMaskSet::SERVER;
context.subscribe(interest, |_| {});
let (tx, rx) = std::sync::mpsc::channel();
context.set_subscribe_callback(Some(Box::new(move |facility, _operation, _index| {
match facility {
Some(Facility::Sink) | Some(Facility::Source) | Some(Facility::Server) => {
let _ = tx.send(());
}
_ => {}
}
})));
mainloop.unlock();
// Background polling loop driven by events or a 2s fallback timeout
loop {
let _ = rx.recv_timeout(Duration::from_secs(2));
{
mainloop.lock();
let _ = fetch_audio_data_sync(&mut context, &state_arc);
mainloop.unlock();
}
}
});
}
}
use std::time::Duration;
fn fetch_audio_data_sync(context: &mut Context, state: &SharedState) -> Result<()> {
let state_inner = Arc::clone(state);
// We fetch all sinks and sources, and also server info to know defaults.
// The order doesn't strictly matter as long as we update correctly.
let st_server = Arc::clone(&state_inner);
context.introspect().get_server_info(move |info| {
let mut lock = RUNTIME.block_on(st_server.write());
lock.audio.sink.name = info
.default_sink_name
.as_ref()
.map(|s| s.to_string())
.unwrap_or_default();
lock.audio.source.name = info
.default_source_name
.as_ref()
.map(|s| s.to_string())
.unwrap_or_default();
});
let st_sink = Arc::clone(&state_inner);
context.introspect().get_sink_info_list(move |res| {
if let ListResult::Item(item) = res {
let mut lock = RUNTIME.block_on(st_sink.write());
// If this matches our default sink name, or if we don't have details for any yet
let is_default = item
.name
.as_ref()
.map(|s| s.as_ref() == lock.audio.sink.name)
.unwrap_or(false);
if is_default {
lock.audio.sink.description = item
.description
.as_ref()
.map(|s| s.to_string())
.unwrap_or_default();
lock.audio.sink.volume =
((item.volume.avg().0 as f64 / Volume::NORMAL.0 as f64) * 100.0).round() as u8;
lock.audio.sink.muted = item.mute;
}
}
});
let st_source = Arc::clone(&state_inner);
context.introspect().get_source_info_list(move |res| {
if let ListResult::Item(item) = res {
let mut lock = RUNTIME.block_on(st_source.write());
let is_default = item
.name
.as_ref()
.map(|s| s.as_ref() == lock.audio.source.name)
.unwrap_or(false);
if is_default {
lock.audio.source.description = item
.description
.as_ref()
.map(|s| s.to_string())
.unwrap_or_default();
lock.audio.source.volume =
((item.volume.avg().0 as f64 / Volume::NORMAL.0 as f64) * 100.0).round() as u8;
lock.audio.source.muted = item.mute;
}
}
});
Ok(())
}
pub struct AudioModule; 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");
match *action { match *action {
"cycle" => { "cycle" => {
self.cycle_device(target_type)?; self.cycle_device(target_type)?;
Ok(WaybarOutput { Ok(WaybarOutput::default())
text: String::new(),
tooltip: None,
class: None,
percentage: None,
})
} }
"show" => self.get_status(config, 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(&self, config: &Config, target_type: &str) -> Result<WaybarOutput> { async fn get_status(
let target = if target_type == "sink" { &self,
"@DEFAULT_AUDIO_SINK@" config: &Config,
} else { state: &SharedState,
"@DEFAULT_AUDIO_SOURCE@" target_type: &str,
) -> Result<WaybarOutput> {
let audio_state = {
let lock = state.read().await;
lock.audio.clone()
}; };
let stdout = run_command("wpctl", &["get-volume", target])?; let (name, description, volume, muted) = if target_type == "sink" {
(
audio_state.sink.name,
audio_state.sink.description,
audio_state.sink.volume,
audio_state.sink.muted,
)
} else {
(
audio_state.source.name,
audio_state.source.description,
audio_state.source.volume,
audio_state.source.muted,
)
};
let parts: Vec<&str> = stdout.split_whitespace().collect(); if name.is_empty() {
if parts.len() < 2 { // Fallback if daemon hasn't populated state yet
return Err(anyhow!("Could not parse wpctl output: {}", stdout)); return Ok(WaybarOutput {
text: "Audio Loading...".to_string(),
..Default::default()
});
} }
let vol_val: f64 = parts[1].parse().unwrap_or(0.0); let display_name = if description.len() > 20 {
let vol = (vol_val * 100.0).round() as u8;
let display_vol = std::cmp::min(vol, 100);
let muted = stdout.contains("[MUTED]");
let description = self.get_description(target_type)?;
let name = if description.len() > 20 {
format!("{}...", &description[..17]) format!("{}...", &description[..17])
} else { } else {
description.clone() description.clone()
@@ -66,16 +246,16 @@ impl AudioModule {
let t = format_template( let t = format_template(
format_str, format_str,
&[ &[
("name", TokenValue::String(&name)), ("name", TokenValue::String(display_name)),
("icon", TokenValue::String(icon)), ("icon", TokenValue::String(icon.to_string())),
], ],
); );
(t, "muted") (t, "muted")
} else { } else {
let icon = if target_type == "sink" { let icon = if target_type == "sink" {
if display_vol <= 30 { if volume <= 30 {
"" ""
} else if display_vol <= 60 { } else if volume <= 60 {
"" ""
} else { } else {
"" ""
@@ -91,9 +271,9 @@ impl AudioModule {
let t = format_template( let t = format_template(
format_str, format_str,
&[ &[
("name", TokenValue::String(&name)), ("name", TokenValue::String(display_name)),
("icon", TokenValue::String(icon)), ("icon", TokenValue::String(icon.to_string())),
("volume", TokenValue::Int(display_vol as i64)), ("volume", TokenValue::Int(volume as i64)),
], ],
); );
(t, "unmuted") (t, "unmuted")
@@ -103,52 +283,32 @@ impl AudioModule {
text, text,
tooltip: Some(description), tooltip: Some(description),
class: Some(class.to_string()), class: Some(class.to_string()),
percentage: Some(display_vol), percentage: Some(volume),
}) })
} }
fn get_description(&self, target_type: &str) -> Result<String> {
let info_stdout = run_command("pactl", &["info"])?;
let search_key = if target_type == "sink" {
"Default Sink:"
} else {
"Default Source:"
};
let default_dev = info_stdout
.lines()
.find(|l| l.contains(search_key))
.and_then(|l| l.split(':').nth(1))
.map(|s| s.trim())
.ok_or_else(|| anyhow!("Default {} not found", target_type))?;
let list_cmd = if target_type == "sink" {
"sinks"
} else {
"sources"
};
let list_stdout = run_command("pactl", &["list", list_cmd])?;
let mut current_name = String::new();
for line in list_stdout.lines() {
if line.trim().starts_with("Name: ") {
current_name = line.split(':').nth(1).unwrap_or("").trim().to_string();
}
if current_name == default_dev && line.trim().starts_with("Description: ") {
return Ok(line.split(':').nth(1).unwrap_or("").trim().to_string());
}
}
Ok(default_dev.to_string())
}
fn cycle_device(&self, target_type: &str) -> Result<()> { fn cycle_device(&self, target_type: &str) -> Result<()> {
// Keep using pactl for cycling for now as it's a rare action
// but we could also implement it natively later.
let set_cmd = if target_type == "sink" {
"set-default-sink"
} else {
"set-default-source"
};
// We need to find the "next" device.
// For simplicity, let's keep the CLI version for now or refactor later.
// The user asked for "step by step".
let list_cmd = if target_type == "sink" { let list_cmd = if target_type == "sink" {
"sinks" "sinks"
} else { } else {
"sources" "sources"
}; };
let stdout = run_command("pactl", &["list", "short", list_cmd])?; let output = Command::new("pactl")
.args(["list", "short", list_cmd])
.output()?;
let stdout = String::from_utf8_lossy(&output.stdout);
let devices: Vec<String> = stdout let devices: Vec<String> = stdout
.lines() .lines()
@@ -171,13 +331,13 @@ impl AudioModule {
return Ok(()); return Ok(());
} }
let info_stdout = run_command("pactl", &["info"])?; let info_output = Command::new("pactl").args(["info"]).output()?;
let info_stdout = String::from_utf8_lossy(&info_output.stdout);
let search_key = if target_type == "sink" { let search_key = if target_type == "sink" {
"Default Sink:" "Default Sink:"
} else { } else {
"Default Source:" "Default Source:"
}; };
let current_dev = info_stdout let current_dev = info_stdout
.lines() .lines()
.find(|l| l.contains(search_key)) .find(|l| l.contains(search_key))
@@ -189,13 +349,7 @@ impl AudioModule {
let next_index = (current_index + 1) % devices.len(); let next_index = (current_index + 1) % devices.len();
let next_dev = &devices[next_index]; let next_dev = &devices[next_index];
let set_cmd = if target_type == "sink" {
"set-default-sink"
} else {
"set-default-source"
};
Command::new("pactl").args([set_cmd, next_dev]).status()?; Command::new("pactl").args([set_cmd, next_dev]).status()?;
Ok(()) Ok(())
} }
} }
-119
View File
@@ -1,119 +0,0 @@
use crate::config::Config;
use crate::modules::WaybarModule;
use crate::output::WaybarOutput;
use crate::state::SharedState;
use crate::utils::{TokenValue, format_template, run_command};
use anyhow::Result;
use std::process::Command;
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");
if *action == "disconnect" {
if let Some(mac) = find_audio_device() {
let _ = Command::new("bluetoothctl")
.args(["disconnect", &mac])
.output();
}
return Ok(WaybarOutput {
text: String::new(),
tooltip: None,
class: None,
percentage: None,
});
}
if let Ok(stdout) = run_command("bluetoothctl", &["show"])
&& stdout.contains("Powered: no")
{
return Ok(WaybarOutput {
text: config.bt.format_disabled.clone(),
tooltip: Some("Bluetooth Disabled".to_string()),
class: Some("disabled".to_string()),
percentage: None,
});
}
if let Some(mac) = find_audio_device() {
let info = run_command("bluetoothctl", &["info", &mac])?;
let mut alias = mac.clone();
let mut battery = None;
let mut trusted = "no";
for line in info.lines() {
if line.contains("Alias:") {
alias = line.split("Alias:").nth(1).unwrap_or("").trim().to_string();
} else if line.contains("Battery Percentage:") {
if let Some(bat_str) = line.split('(').nth(1).and_then(|s| s.split(')').next())
{
battery = bat_str.parse::<u8>().ok();
}
} else if line.contains("Trusted: yes") {
trusted = "yes";
}
}
let tooltip = format!(
"{} | MAC: {}\nTrusted: {} | Bat: {}",
alias,
mac,
trusted,
battery
.map(|b| format!("{}%", b))
.unwrap_or_else(|| "N/A".to_string())
);
let text = format_template(
&config.bt.format_connected,
&[("alias", TokenValue::String(&alias))],
);
Ok(WaybarOutput {
text,
tooltip: Some(tooltip),
class: Some("connected".to_string()),
percentage: battery,
})
} else {
Ok(WaybarOutput {
text: config.bt.format_disconnected.clone(),
tooltip: Some("Bluetooth On (Disconnected)".to_string()),
class: Some("disconnected".to_string()),
percentage: None,
})
}
}
}
fn find_audio_device() -> Option<String> {
if let Ok(sink) = run_command("pactl", &["get-default-sink"])
&& sink.starts_with("bluez_output.")
{
let parts: Vec<&str> = sink.split('.').collect();
if parts.len() >= 2 {
return Some(parts[1].replace('_', ":"));
}
}
if let Ok(stdout) = run_command("bluetoothctl", &["devices", "Connected"]) {
for line in stdout.lines() {
if line.starts_with("Device ") {
let parts: Vec<&str> = line.split_whitespace().collect();
if parts.len() >= 2 {
let mac = parts[1];
if let Ok(info_str) = run_command("bluetoothctl", &["info", mac])
&& info_str.contains("0000110b-0000-1000-8000-00805f9b34fb")
{
return Some(mac.to_string());
}
}
}
}
}
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;
-136
View File
@@ -1,136 +0,0 @@
use crate::config::Config;
use crate::modules::WaybarModule;
use crate::output::WaybarOutput;
use crate::state::SharedState;
use crate::utils::{TokenValue, format_template, run_command};
use anyhow::Result;
pub struct BudsModule;
impl WaybarModule for BudsModule {
fn run(&self, config: &Config, _state: &SharedState, args: &[&str]) -> Result<WaybarOutput> {
let action = args.first().unwrap_or(&"show");
let mac = &config.buds.mac;
match *action {
"cycle_anc" => {
let current_mode = run_command("pbpctrl", &["get", "anc"])?;
let next_mode = match current_mode.as_str() {
"active" => "aware",
"aware" => "off",
_ => "active",
};
let _ = run_command("pbpctrl", &["set", "anc", next_mode]);
return Ok(WaybarOutput {
text: String::new(),
tooltip: None,
class: None,
percentage: None,
});
}
"connect" => {
let _ = run_command("bluetoothctl", &["connect", mac]);
return Ok(WaybarOutput {
text: String::new(),
tooltip: None,
class: None,
percentage: None,
});
}
"disconnect" => {
let _ = run_command("bluetoothctl", &["disconnect", mac]);
return Ok(WaybarOutput {
text: String::new(),
tooltip: None,
class: None,
percentage: None,
});
}
"show" => {}
other => {
return Err(anyhow::anyhow!("Unknown buds action: '{}'", other));
}
}
let bt_str = run_command("bluetoothctl", &["info", mac])?;
if !bt_str.contains("Connected: yes") {
return Ok(WaybarOutput {
text: config.buds.format_disconnected.clone(),
tooltip: Some("Pixel Buds Pro 2 not connected".to_string()),
class: Some("disconnected".to_string()),
percentage: None,
});
}
let bat_output = match run_command("pbpctrl", &["show", "battery"]) {
Ok(output) => output,
Err(_) => {
return Ok(WaybarOutput {
text: config.buds.format_disconnected.clone(),
tooltip: Some("Pixel Buds Pro 2 connected (No Data)".to_string()),
class: Some("disconnected".to_string()),
percentage: None,
});
}
};
let mut left_bud = "unknown";
let mut right_bud = "unknown";
for line in bat_output.lines() {
if line.contains("left bud:") {
left_bud = line.split_whitespace().nth(2).unwrap_or("unknown");
} else if line.contains("right bud:") {
right_bud = line.split_whitespace().nth(2).unwrap_or("unknown");
}
}
if left_bud == "unknown" && right_bud == "unknown" {
return Ok(WaybarOutput {
text: "{}".to_string(),
tooltip: None,
class: None,
percentage: None,
});
}
let left_display = if left_bud == "unknown" {
"---".to_string()
} else {
format!("{}%", left_bud)
};
let right_display = if right_bud == "unknown" {
"---".to_string()
} else {
format!("{}%", right_bud)
};
let current_mode = run_command("pbpctrl", &["get", "anc"]).unwrap_or_default();
let (anc_icon, class) = match current_mode.as_str() {
"active" => ("ANC", "anc-active"),
"aware" => ("Aware", "anc-aware"),
"off" => ("Off", "anc-off"),
_ => ("?", "anc-unknown"),
};
let text = format_template(
&config.buds.format,
&[
("left", TokenValue::String(&left_display)),
("right", TokenValue::String(&right_display)),
("anc", TokenValue::String(anc_icon)),
],
);
Ok(WaybarOutput {
text,
tooltip: Some("Pixel Buds Pro 2".to_string()),
class: Some(class.to_string()),
percentage: None,
})
}
}
+22 -20
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"));
} }
} }
+23 -16
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 {
@@ -44,7 +48,7 @@ impl WaybarModule for DiskModule {
let text = format_template( let text = format_template(
&config.disk.format, &config.disk.format,
&[ &[
("mount", TokenValue::String(mountpoint)), ("mount", TokenValue::String(mountpoint.to_string())),
("used", TokenValue::Float(used_gb)), ("used", TokenValue::Float(used_gb)),
("total", TokenValue::Float(total_gb)), ("total", TokenValue::Float(total_gb)),
], ],
@@ -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());
} }
} }
+26 -4
View File
@@ -1,15 +1,23 @@
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::run_command; use anyhow::anyhow;
use anyhow::Result; use std::env;
use std::io::{Read, Write};
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(
let is_gamemode = run_command("hyprctl", &["getoption", "animations:enabled", "-j"]) &self,
config: &Config,
_state: &SharedState,
_args: &[&str],
) -> Result<WaybarOutput> {
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);
@@ -30,3 +38,17 @@ impl WaybarModule for GameModule {
} }
} }
} }
fn hyprland_ipc(cmd: &str) -> Result<String> {
let signature = env::var("HYPRLAND_INSTANCE_SIGNATURE")
.map_err(|_| anyhow!("HYPRLAND_INSTANCE_SIGNATURE not set"))?;
let path = format!("/tmp/hypr/{}/.socket.sock", signature);
let mut stream = UnixStream::connect(path)?;
stream.write_all(cmd.as_bytes())?;
let mut response = String::new();
stream.read_to_string(&mut response)?;
Ok(response)
}
+17 -23
View File
@@ -1,36 +1,30 @@
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(),
state_lock.gpu.usage, state_lock.gpu.usage,
state_lock.gpu.vram_used, state_lock.gpu.vram_used,
state_lock.gpu.vram_total, state_lock.gpu.vram_total,
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 {
+40 -24
View File
@@ -20,11 +20,11 @@ impl HardwareDaemon {
components, components,
gpu_vendor: None, gpu_vendor: None,
gpu_poll_counter: 0, gpu_poll_counter: 0,
disk_poll_counter: 0, disk_poll_counter: 9, // Start at 9 to poll on the first tick
} }
} }
pub fn poll(&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,30 +70,36 @@ 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;
state_lock.memory.total_gb = total_mem; state_lock.memory.total_gb = total_mem;
state_lock.memory.used_gb = used_mem; state_lock.memory.used_gb = used_mem;
state_lock.sys.load_1 = load_avg.one; state_lock.sys.load_1 = load_avg.one;
state_lock.sys.load_5 = load_avg.five; state_lock.sys.load_5 = load_avg.five;
state_lock.sys.load_15 = load_avg.fifteen; state_lock.sys.load_15 = load_avg.fifteen;
state_lock.sys.uptime = uptime; state_lock.sys.uptime = uptime;
state_lock.sys.process_count = process_count; state_lock.sys.process_count = process_count;
}
// Poll GPU every 5 seconds to avoid expensive nvidia-smi calls pub async fn poll_slow(&mut self, state: SharedState) {
self.gpu_poll_counter = (self.gpu_poll_counter + 1) % 5; // 1. Gather GPU data outside of lock
if self.gpu_poll_counter == 0 { let mut gpu_state = crate::state::GpuState::default();
self.poll_gpu(&mut state_lock.gpu); self.gpu_poll_counter = (self.gpu_poll_counter + 1) % 5;
} let should_poll_gpu = self.gpu_poll_counter == 0;
if should_poll_gpu {
self.poll_gpu(&mut gpu_state);
}
// Poll disks every 10 seconds // 2. Gather Disk data outside of lock
self.disk_poll_counter = (self.disk_poll_counter + 1) % 10; let mut disks_data = None;
if self.disk_poll_counter == 0 { self.disk_poll_counter = (self.disk_poll_counter + 1) % 10;
state_lock.disks = Disks::new_with_refreshed_list() if self.disk_poll_counter == 0 {
disks_data = Some(
Disks::new_with_refreshed_list()
.iter() .iter()
.map(|d| DiskInfo { .map(|d| DiskInfo {
mount_point: d.mount_point().to_string_lossy().into_owned(), mount_point: d.mount_point().to_string_lossy().into_owned(),
@@ -101,8 +107,18 @@ impl HardwareDaemon {
total_bytes: d.total_space(), total_bytes: d.total_space(),
available_bytes: d.available_space(), available_bytes: d.available_space(),
}) })
.collect(); .collect::<Vec<DiskInfo>>(),
} );
}
// 3. Apply to state
let mut state_lock = state.write().await;
if should_poll_gpu {
state_lock.gpu = gpu_state;
}
if let Some(d) = disks_data {
state_lock.disks = d;
} }
} }
+18 -16
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 -3
View File
@@ -1,7 +1,6 @@
pub mod audio; pub mod audio;
pub mod bt; pub mod bt;
pub mod btrfs; pub mod btrfs;
pub mod buds;
pub mod cpu; pub mod cpu;
pub mod disk; pub mod disk;
pub mod game; pub mod game;
@@ -13,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;
} }
+64 -72
View File
@@ -1,9 +1,10 @@
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, run_command}; use crate::utils::{TokenValue, format_template};
use anyhow::Result; use nix::ifaddrs::getifaddrs;
use std::fs; use std::fs;
use std::time::{SystemTime, UNIX_EPOCH}; use std::time::{SystemTime, UNIX_EPOCH};
@@ -28,22 +29,19 @@ impl NetworkDaemon {
} }
} }
pub fn poll(&mut self, state: SharedState) { pub async fn poll(&mut self, state: SharedState) {
// Cache invalidation: if the interface directory doesn't exist, clear cache // Re-detect interface on every poll to catch VPNs or route changes immediately
if let Some(ref iface) = self.cached_interface if let Ok(iface) = get_primary_interface()
&& !std::path::Path::new(&format!("/sys/class/net/{}", iface)).exists()
{
self.cached_interface = None;
self.cached_ip = None;
}
// Re-detect interface if needed
if self.cached_interface.is_none()
&& let Ok(iface) = get_primary_interface()
&& !iface.is_empty() && !iface.is_empty()
{ {
self.cached_ip = get_ip_address(&iface); // If the interface changed, or we don't have an IP yet, update the IP
self.cached_interface = Some(iface); if self.cached_interface.as_ref() != Some(&iface) || self.cached_ip.is_none() {
self.cached_ip = get_ip_address(&iface);
self.cached_interface = Some(iface);
}
} else {
self.cached_interface = None;
self.cached_ip = None;
} }
if let Some(ref interface) = self.cached_interface { if let Some(ref interface) = self.cached_interface {
@@ -64,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();
} }
@@ -83,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();
} }
@@ -92,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() {
@@ -123,8 +121,8 @@ impl WaybarModule for NetworkModule {
let mut output_text = format_template( let mut output_text = format_template(
&config.network.format, &config.network.format,
&[ &[
("interface", TokenValue::String(&interface)), ("interface", TokenValue::String(interface.clone())),
("ip", TokenValue::String(ip_display)), ("ip", TokenValue::String(ip_display.to_string())),
("rx", TokenValue::Float(rx_mbps)), ("rx", TokenValue::Float(rx_mbps)),
("tx", TokenValue::Float(tx_mbps)), ("tx", TokenValue::Float(tx_mbps)),
], ],
@@ -134,6 +132,8 @@ impl WaybarModule for NetworkModule {
|| interface.starts_with("wg") || interface.starts_with("wg")
|| interface.starts_with("ppp") || interface.starts_with("ppp")
|| interface.starts_with("pvpn") || interface.starts_with("pvpn")
|| interface.starts_with("proton")
|| interface.starts_with("ipsec")
{ {
output_text = format!(" {}", output_text); output_text = format!(" {}", output_text);
} }
@@ -148,24 +148,18 @@ impl WaybarModule for NetworkModule {
} }
fn get_primary_interface() -> Result<String> { fn get_primary_interface() -> Result<String> {
let stdout = run_command("ip", &["route", "list"])?; let content = fs::read_to_string("/proc/net/route")?;
let mut defaults = Vec::new(); let mut defaults = Vec::new();
for line in stdout.lines() { for line in content.lines().skip(1) {
if line.starts_with("default") { let parts: Vec<&str> = line.split_whitespace().collect();
let parts: Vec<&str> = line.split_whitespace().collect(); if parts.len() >= 7 {
let mut dev = ""; let iface = parts[0];
let mut metric = 0; let dest = parts[1];
for i in 0..parts.len() { let metric = parts[6].parse::<i32>().unwrap_or(0);
if parts[i] == "dev" && i + 1 < parts.len() {
dev = parts[i + 1]; if dest == "00000000" {
} defaults.push((metric, iface.to_string()));
if parts[i] == "metric" && i + 1 < parts.len() {
metric = parts[i + 1].parse::<i32>().unwrap_or(0);
}
}
if !dev.is_empty() {
defaults.push((metric, dev.to_string()));
} }
} }
} }
@@ -179,15 +173,13 @@ fn get_primary_interface() -> Result<String> {
} }
fn get_ip_address(interface: &str) -> Option<String> { fn get_ip_address(interface: &str) -> Option<String> {
let stdout = run_command("ip", &["-4", "addr", "show", interface]).ok()?; let addrs = getifaddrs().ok()?;
for line in stdout.lines() { for ifaddr in addrs {
if line.trim().starts_with("inet ") { if ifaddr.interface_name == interface
let parts: Vec<&str> = line.split_whitespace().collect(); && let Some(address) = ifaddr.address
if parts.len() > 1 { && let Some(sockaddr) = address.as_sockaddr_in()
let ip_cidr = parts[1]; {
let ip = ip_cidr.split('/').next().unwrap_or(ip_cidr); return Some(sockaddr.ip().to_string());
return Some(ip.to_string());
}
} }
} }
None None
@@ -216,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,
@@ -236,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,
@@ -255,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,
@@ -271,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"));
} }
} }
+8 -3
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;
@@ -96,7 +101,7 @@ impl WaybarModule for PowerModule {
&config.power.format, &config.power.format,
&[ &[
("percentage", TokenValue::Int(percentage as i64)), ("percentage", TokenValue::Int(percentage as i64)),
("icon", TokenValue::String(icon)), ("icon", TokenValue::String(icon.to_string())),
], ],
); );
+16 -14
View File
@@ -1,26 +1,28 @@
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,
state_lock.sys.load_15, state_lock.sys.load_15,
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;
@@ -34,7 +36,7 @@ impl WaybarModule for SysModule {
let text = format_template( let text = format_template(
&config.sys.format, &config.sys.format,
&[ &[
("uptime", TokenValue::String(&uptime_str)), ("uptime", TokenValue::String(uptime_str.clone())),
("load1", TokenValue::Float(load1)), ("load1", TokenValue::Float(load1)),
("load5", TokenValue::Float(load5)), ("load5", TokenValue::Float(load5)),
("load15", TokenValue::Float(load15)), ("load15", TokenValue::Float(load15)),
+1 -1
View File
@@ -1,6 +1,6 @@
use serde::Serialize; use serde::Serialize;
#[derive(Serialize)] #[derive(Serialize, Default)]
pub struct WaybarOutput { pub struct WaybarOutput {
pub text: String, pub text: String,
#[serde(skip_serializing_if = "Option::is_none")] #[serde(skip_serializing_if = "Option::is_none")]
+46 -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 {
@@ -8,6 +11,48 @@ pub struct AppState {
pub sys: SysState, pub sys: SysState,
pub gpu: GpuState, pub gpu: GpuState,
pub disks: Vec<DiskInfo>, pub disks: Vec<DiskInfo>,
pub bluetooth: BtState,
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)]
pub struct AudioState {
pub sink: AudioDeviceInfo,
pub source: AudioSourceInfo,
}
#[derive(Default, Clone)]
pub struct AudioDeviceInfo {
pub name: String,
pub description: String,
pub volume: u8,
pub muted: bool,
}
#[derive(Default, Clone)]
pub struct AudioSourceInfo {
pub name: String,
pub description: String,
pub volume: u8,
pub muted: bool,
}
#[derive(Default, Clone)]
pub struct BtState {
pub connected: bool,
pub adapter_powered: bool,
pub device_alias: String,
pub device_address: String,
pub battery_percentage: Option<u8>,
pub plugin_data: Vec<(String, String)>,
} }
#[derive(Default, Clone)] #[derive(Default, Clone)]
+22 -24
View File
@@ -2,20 +2,6 @@ use anyhow::{Context, Result};
use std::io::Write; use std::io::Write;
use std::process::{Command, Stdio}; use std::process::{Command, Stdio};
/// Run an external command and return its stdout as a trimmed String.
/// Provides clear error messages when the command is not found or fails.
pub fn run_command(cmd: &str, args: &[&str]) -> Result<String> {
let output = Command::new(cmd)
.args(args)
.output()
.with_context(|| format!("'{}' not found or failed to execute. Is it installed?", cmd))?;
if !output.status.success() {
let stderr = String::from_utf8_lossy(&output.stderr);
anyhow::bail!("'{}' exited with {}: {}", cmd, output.status, stderr.trim());
}
Ok(String::from_utf8_lossy(&output.stdout).trim().to_string())
}
pub fn show_menu(prompt: &str, items: &[String], menu_cmd: &str) -> Result<String> { pub fn show_menu(prompt: &str, items: &[String], menu_cmd: &str) -> Result<String> {
let cmd_str = menu_cmd.replace("{prompt}", prompt); let cmd_str = menu_cmd.replace("{prompt}", prompt);
let mut child = Command::new("sh") let mut child = Command::new("sh")
@@ -50,20 +36,23 @@ pub fn show_menu(prompt: &str, items: &[String], menu_cmd: &str) -> Result<Strin
use regex::Regex; use regex::Regex;
use std::sync::LazyLock; use std::sync::LazyLock;
pub enum TokenValue<'a> { pub enum TokenValue {
Float(f64), Float(f64),
Int(i64), Int(i64),
String(&'a str), String(String),
} }
pub fn format_template(template: &str, values: &[(&str, TokenValue)]) -> String { pub fn format_template<K>(template: &str, values: &[(K, TokenValue)]) -> String
where
K: AsRef<str>,
{
static RE: LazyLock<Regex> = LazyLock::new(|| { static RE: LazyLock<Regex> = LazyLock::new(|| {
Regex::new(r"\{([a-zA-Z0-9_]+)(?::([<>\^])?(\d+)?(?:\.(\d+))?)?\}").unwrap() Regex::new(r"\{([a-zA-Z0-9_]+)(?::([<>\^])?(\d+)?(?:\.(\d+))?)?\}").unwrap()
}); });
RE.replace_all(template, |caps: &regex::Captures| { RE.replace_all(template, |caps: &regex::Captures| {
let name = &caps[1]; let name = &caps[1];
if let Some((_, val)) = values.iter().find(|(k, _)| *k == name) { if let Some((_, val)) = values.iter().find(|(k, _)| k.as_ref() == name) {
let align = caps.get(2).map(|m| m.as_str()).unwrap_or(">"); let align = caps.get(2).map(|m| m.as_str()).unwrap_or(">");
let width = caps let width = caps
.get(3) .get(3)
@@ -121,7 +110,10 @@ mod tests {
#[test] #[test]
fn test_simple_string_token() { fn test_simple_string_token() {
let result = format_template("{name}", &[("name", TokenValue::String("hello"))]); let result = format_template(
"{name}",
&[("name", TokenValue::String("hello".to_string()))],
);
assert_eq!(result, "hello"); assert_eq!(result, "hello");
} }
@@ -166,14 +158,20 @@ mod tests {
#[test] #[test]
fn test_string_left_align() { fn test_string_left_align() {
let result = format_template("{val:<10}", &[("val", TokenValue::String("hi"))]); let result = format_template(
"{val:<10}",
&[("val", TokenValue::String("hi".to_string()))],
);
assert_eq!(result, "hi "); assert_eq!(result, "hi ");
assert_eq!(result.len(), 10); assert_eq!(result.len(), 10);
} }
#[test] #[test]
fn test_unknown_token_preserved() { fn test_unknown_token_preserved() {
let result = format_template("{unknown}", &[("name", TokenValue::String("test"))]); let result = format_template(
"{unknown}",
&[("name", TokenValue::String("test".to_string()))],
);
assert_eq!(result, "{unknown}"); assert_eq!(result, "{unknown}");
} }
@@ -191,7 +189,7 @@ mod tests {
#[test] #[test]
fn test_no_tokens() { fn test_no_tokens() {
let result = format_template("plain text", &[]); let result = format_template::<&'static str>("plain text", &[]);
assert_eq!(result, "plain text"); assert_eq!(result, "plain text");
} }
@@ -206,8 +204,8 @@ mod tests {
let result = format_template( let result = format_template(
"{name} ({ip}): {rx:>5.2} MB/s", "{name} ({ip}): {rx:>5.2} MB/s",
&[ &[
("name", TokenValue::String("eth0")), ("name", TokenValue::String("eth0".to_string())),
("ip", TokenValue::String("10.0.0.1")), ("ip", TokenValue::String("10.0.0.1".to_string())),
("rx", TokenValue::Float(1.5)), ("rx", TokenValue::Float(1.5)),
], ],
); );