Compare commits
6 Commits
v0.3.3
..
492f0787cb
| Author | SHA1 | Date | |
|---|---|---|---|
| 492f0787cb | |||
| 718d3f99e9 | |||
| b2e8d5fc1b | |||
| 2572928195 | |||
| 25ecde96fa | |||
| 16d12e6dce |
Generated
+785
-26
File diff suppressed because it is too large
Load Diff
+4
-2
@@ -1,6 +1,6 @@
|
||||
[package]
|
||||
name = "fluxo-rs"
|
||||
version = "0.3.3"
|
||||
version = "0.4.0"
|
||||
edition = "2024"
|
||||
|
||||
[dependencies]
|
||||
@@ -12,7 +12,7 @@ serde = { version = "1.0.228", features = ["derive"] }
|
||||
serde_json = "1.0.149"
|
||||
sysinfo = "0.38.4"
|
||||
thiserror = "2.0"
|
||||
toml = "1.1.1"
|
||||
toml = "1.1.2"
|
||||
tracing = "0.1.44"
|
||||
tracing-subscriber = { version = "0.3.23", features = ["env-filter"] }
|
||||
maestro = { git = "https://github.com/qzed/pbpctrl", package = "maestro" }
|
||||
@@ -22,6 +22,8 @@ tokio-util = { version = "0.7", features = ["codec", "time"] }
|
||||
futures = "0.3"
|
||||
libpulse-binding = "2.30"
|
||||
nix = { version = "0.31", features = ["net"] }
|
||||
notify = "8.2.0"
|
||||
zbus = "4"
|
||||
|
||||
[dev-dependencies]
|
||||
|
||||
|
||||
@@ -1,18 +1,16 @@
|
||||
# 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 socket.
|
||||
`fluxo-rs` is a high-performance system metrics daemon and client designed specifically for Waybar. It entirely replaces standard shell scripts with a compiled Rust binary that collects data via a background polling loop and serves it over a Unix socket.
|
||||
|
||||
With its **100% Native, Content-Based Event-Driven Architecture**, it consumes effectively 0% CPU while idle and signals Waybar to redraw *only* when the rendered UI text or icons physically change.
|
||||
|
||||
## Key Features
|
||||
|
||||
- **Asynchronous Architecture**: Built on **Tokio**, the daemon handles concurrent IPC requests and background tasks with zero latency and minimal CPU overhead.
|
||||
- **Native Library Integrations**:
|
||||
- **Audio**: Direct `libpulse` integration for event-driven, instant volume and device updates.
|
||||
- **Bluetooth**: Native `bluer` integration for robust device monitoring.
|
||||
- **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.
|
||||
- **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.
|
||||
- **Multi-threaded Polling**: Decoupled subsystem threads ensure that a hang in one system (e.g., a slow GPU probe) never freezes your entire bar.
|
||||
- **100% Native Architecture**: Zero shell-outs or subprocesses. Uses `bluer` for Bluetooth, `libpulse-binding` for audio, `zbus` for MPRIS/DND, and `notify` for backlight.
|
||||
- **Content-Based Event Signaling**: `fluxo-rs` evaluates your custom configuration formats internally. It only sends a `SIGRTMIN+X` signal to Waybar if the resulting string or CSS class has actually changed, eliminating pointless re-renders from raw polling fluctuations.
|
||||
- **Zero-Latency Interactions**: Direct library bindings mean that when you change your volume or connect a Bluetooth device via the CLI, the daemon updates instantly.
|
||||
- **Circuit Breaker (Failsafe)**: Automatically detects failing modules and enters a "Cool down" state, preventing resource waste and log spam. Fallback caching keeps your bar looking clean even during brief failures.
|
||||
- **Multi-threaded Polling**: Decoupled Tokio subsystem threads ensure that a hang in one system (e.g., a slow GPU probe) never freezes your Waybar.
|
||||
|
||||
## Modules
|
||||
|
||||
@@ -24,37 +22,47 @@ fluxo-rs is a high-performance system metrics daemon and client designed specifi
|
||||
| `sys` | System load and uptime | `{uptime}`, `{load1}`, `{load5}`, `{load15}`, `{procs}` |
|
||||
| `disk` | Disk usage | `{mount}`, `{used}`, `{total}` |
|
||||
| `pool` | Btrfs aggregate storage | `{used}`, `{total}` |
|
||||
| `gpu` | GPU usage & thermals | `{usage}`, `{vram_used}`, `{vram_total}`, `{temp}` |
|
||||
| `vol` | Audio output (sink) | `{name}`, `{volume}`, `{icon}` |
|
||||
| `mic` | Audio input (source) | `{name}`, `{volume}`, `{icon}` |
|
||||
| `bt` | Bluetooth status & plugins | `{alias}`, `{mac}`, `{left}`, `{right}`, `{anc}` |
|
||||
| `power` | Battery and AC status | `{percentage}`, `{icon}` |
|
||||
| `game` | Hyprland status | active/inactive icons |
|
||||
| `game` | Hyprland Gamemode status | active/inactive strings |
|
||||
| `mpris` | Media Player status | `{artist}`, `{title}`, `{album}`, `{status_icon}` |
|
||||
| `backlight` | Display Brightness | `{percentage}`, `{icon}` |
|
||||
| `kbd` | Keyboard Layout | `{layout}` |
|
||||
| `dnd` | Do Not Disturb (SwayNC) | active/inactive strings |
|
||||
|
||||
## Setup
|
||||
|
||||
1. **Build**: `cargo build --release`
|
||||
2. **Configure**: Create `~/.config/fluxo/config.toml` (see `example.config.toml`).
|
||||
3. **Daemon**: Start `fluxo-rs daemon`. It's recommended to run this as a systemd user service.
|
||||
2. **Configure**: Create `~/.config/fluxo/config.toml` (see `example.config.toml`). Ensure you map your `[signals]`.
|
||||
3. **Daemon**: Start `fluxo-rs daemon`. It is highly recommended to run this as a systemd user service.
|
||||
|
||||
## Waybar Configuration
|
||||
|
||||
To achieve zero-latency updates, use **Waybar Signals**:
|
||||
To achieve zero-latency updates and zero-polling CPU usage, set `interval: 0` on your modules and rely entirely on **Waybar Signals** mapped in your `config.toml`:
|
||||
|
||||
```jsonc
|
||||
"custom/audio": {
|
||||
"exec": "fluxo vol",
|
||||
"custom/volume": {
|
||||
"exec": "fluxo-rs vol",
|
||||
"return-type": "json",
|
||||
"interval": 5,
|
||||
"signal": 8,
|
||||
"on-click": "fluxo audio cycle sink && pkill -RTMIN+8 waybar"
|
||||
"interval": 0,
|
||||
"signal": 8, // Must match the value in config.toml [signals]
|
||||
"on-click": "fluxo-rs vol mute",
|
||||
"on-scroll-up": "fluxo-rs vol up 1",
|
||||
"on-scroll-down": "fluxo-rs vol down 1",
|
||||
"on-click-right": "fluxo-rs vol cycle"
|
||||
},
|
||||
"custom/bluetooth": {
|
||||
"exec": "fluxo bt",
|
||||
"custom/bluetooth-audio": {
|
||||
"format": "{}",
|
||||
"return-type": "json",
|
||||
"interval": 5,
|
||||
"exec": "fluxo-rs bt",
|
||||
"on-click": "fluxo-rs bt menu",
|
||||
"on-click-right": "fluxo-rs bt cycle_mode",
|
||||
"signal": 9,
|
||||
"on-click": "fluxo bt menu && pkill -RTMIN+9 waybar",
|
||||
"on-click-right": "fluxo bt cycle_mode && pkill -RTMIN+9 waybar"
|
||||
"interval": 0,
|
||||
"tooltip": true
|
||||
}
|
||||
```
|
||||
|
||||
|
||||
+63
-12
@@ -1,46 +1,97 @@
|
||||
# Fluxo configuration example
|
||||
# Place this at ~/.config/fluxo/config.toml
|
||||
|
||||
# Note: All tokens support standard alignment, padding, and precision specifiers dynamically.
|
||||
# For example, you can change {rx:>5.2} to {rx:<8.1} or {used} to {used:^4.0} directly here.
|
||||
|
||||
[general]
|
||||
menu_command = "fuzzel --dmenu --prompt '{prompt}'"
|
||||
# command used for interactive menus (e.g., bluetooth device selection)
|
||||
# $FLUXO_PROMPT is securely injected to prevent shell escaping issues
|
||||
menu_command = "fuzzel --dmenu --prompt \"$FLUXO_PROMPT\""
|
||||
# For Wofi use: menu_command = "wofi --show dmenu --prompt \"$FLUXO_PROMPT\""
|
||||
|
||||
# Map modules to specific Waybar signals for zero-latency, event-driven UI updates.
|
||||
# These MUST match the `signal` configuration in your waybar config.jsonc
|
||||
[signals]
|
||||
network = 1
|
||||
cpu = 2
|
||||
memory = 3
|
||||
gpu = 4
|
||||
sys = 5
|
||||
disk = 6
|
||||
game = 7
|
||||
audio = 8
|
||||
bt = 9
|
||||
power = 10
|
||||
keyboard = 11
|
||||
mpris = 12
|
||||
backlight = 13
|
||||
dnd = 14
|
||||
|
||||
[network]
|
||||
format = "{interface} ({ip}): {rx:>5.2} MB/s {tx:>5.2} MB/s"
|
||||
# tokens: {interface}, {ip}, {rx}, {tx}
|
||||
format = "{interface} ({ip}): {rx:^4.1} MB/s {tx:^4.1} MB/s"
|
||||
|
||||
[cpu]
|
||||
format = "CPU: {usage:>4.1}% {temp:>4.1}C"
|
||||
# tokens: {usage}, {temp}, {model}
|
||||
format = "CPU: {usage:^4.1}% {temp:^4.1}C"
|
||||
|
||||
[memory]
|
||||
format = "{used:>5.2}/{total:>5.2}GB"
|
||||
# tokens: {used}, {total}
|
||||
format = "MEM: {used:^4.1}/{total:^4.1}GB"
|
||||
|
||||
[gpu]
|
||||
# tokens: {usage}, {vram_used}, {vram_total}, {temp}
|
||||
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}"
|
||||
# tokens: {uptime}, {load1}, {load5}, {load15}, {procs}
|
||||
format = "UP: {uptime} LOAD: {load1:^3.1} "
|
||||
|
||||
[disk]
|
||||
format = "{mount} {used:>5.1}/{total:>5.1}G"
|
||||
# tokens: {mount}, {used}, {total}
|
||||
format = "{mount} {used:^3.0}/{total:^3.0}G"
|
||||
|
||||
[pool]
|
||||
# tokens: {used}, {total}
|
||||
format = "{used:>4.0}G / {total:>4.0}G"
|
||||
|
||||
[power]
|
||||
# tokens: {percentage}, {icon}
|
||||
format = "{percentage:>3}% {icon}"
|
||||
|
||||
[bt]
|
||||
format_connected = "{alias} "
|
||||
format_plugin = "{alias} [{left}|{right}] {anc} "
|
||||
format_disconnected = ""
|
||||
format_disabled = " Off"
|
||||
|
||||
[audio]
|
||||
# tokens: {name}, {volume}, {icon}
|
||||
format_sink_unmuted = "{name} {volume:>3}% {icon}"
|
||||
format_sink_muted = "{name} {icon}"
|
||||
format_source_unmuted = "{name} {volume:>3}% {icon}"
|
||||
format_source_muted = "{name} {icon}"
|
||||
|
||||
[bt]
|
||||
# tokens: {alias}, {mac}, {left}, {right}, {anc}
|
||||
format_plugin = "{alias} [{left}|{right}] {anc} "
|
||||
format_connected = " {alias}"
|
||||
format_disconnected = " Disconnected"
|
||||
format_disabled = " Off"
|
||||
|
||||
[game]
|
||||
format_active = "<span size='large'></span>"
|
||||
format_inactive = "<span size='large'></span>"
|
||||
|
||||
[mpris]
|
||||
# tokens: {artist}, {title}, {album}, {status_icon}
|
||||
format = "{status_icon} {artist} - {title}"
|
||||
|
||||
[backlight]
|
||||
# tokens: {percentage}, {icon}
|
||||
format = "{percentage:>3}% {icon}"
|
||||
|
||||
[keyboard]
|
||||
# tokens: {layout}
|
||||
format = "{layout}"
|
||||
|
||||
[dnd]
|
||||
format_dnd = "<span size='large'></span>"
|
||||
format_normal = "<span size='large'></span>"
|
||||
|
||||
+96
-5
@@ -10,6 +10,8 @@ pub struct Config {
|
||||
#[serde(default)]
|
||||
pub general: GeneralConfig,
|
||||
#[serde(default)]
|
||||
pub signals: SignalsConfig,
|
||||
#[serde(default)]
|
||||
pub network: NetworkConfig,
|
||||
#[serde(default)]
|
||||
pub cpu: CpuConfig,
|
||||
@@ -31,6 +33,14 @@ pub struct Config {
|
||||
pub bt: BtConfig,
|
||||
#[serde(default)]
|
||||
pub game: GameConfig,
|
||||
#[serde(default)]
|
||||
pub mpris: MprisConfig,
|
||||
#[serde(default)]
|
||||
pub backlight: BacklightConfig,
|
||||
#[serde(default)]
|
||||
pub keyboard: KeyboardConfig,
|
||||
#[serde(default)]
|
||||
pub dnd: DndConfig,
|
||||
}
|
||||
|
||||
#[derive(Deserialize, Clone)]
|
||||
@@ -41,11 +51,31 @@ pub struct GeneralConfig {
|
||||
impl Default for GeneralConfig {
|
||||
fn default() -> Self {
|
||||
Self {
|
||||
menu_command: "fuzzel --dmenu --prompt '{prompt}'".to_string(),
|
||||
menu_command: "fuzzel --dmenu --prompt \"$FLUXO_PROMPT\"".to_string(),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[allow(dead_code)]
|
||||
#[derive(Deserialize, Default, Clone)]
|
||||
pub struct SignalsConfig {
|
||||
pub network: Option<i32>,
|
||||
pub cpu: Option<i32>,
|
||||
pub memory: Option<i32>,
|
||||
pub gpu: Option<i32>,
|
||||
pub sys: Option<i32>,
|
||||
pub disk: Option<i32>,
|
||||
pub pool: Option<i32>,
|
||||
pub power: Option<i32>,
|
||||
pub audio: Option<i32>,
|
||||
pub bt: Option<i32>,
|
||||
pub game: Option<i32>,
|
||||
pub mpris: Option<i32>,
|
||||
pub backlight: Option<i32>,
|
||||
pub keyboard: Option<i32>,
|
||||
pub dnd: Option<i32>,
|
||||
}
|
||||
|
||||
#[derive(Deserialize, Clone)]
|
||||
pub struct NetworkConfig {
|
||||
pub format: String,
|
||||
@@ -209,6 +239,60 @@ impl Default for GameConfig {
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Deserialize, Clone)]
|
||||
pub struct MprisConfig {
|
||||
pub format: String,
|
||||
}
|
||||
|
||||
impl Default for MprisConfig {
|
||||
fn default() -> Self {
|
||||
Self {
|
||||
format: "{status_icon} {artist} - {title}".to_string(),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Deserialize, Clone)]
|
||||
pub struct BacklightConfig {
|
||||
pub format: String,
|
||||
}
|
||||
|
||||
impl Default for BacklightConfig {
|
||||
fn default() -> Self {
|
||||
Self {
|
||||
format: "{percentage:>3}% {icon}".to_string(),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Deserialize, Clone)]
|
||||
pub struct KeyboardConfig {
|
||||
pub format: String,
|
||||
}
|
||||
|
||||
impl Default for KeyboardConfig {
|
||||
fn default() -> Self {
|
||||
Self {
|
||||
format: "{layout}".to_string(),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Deserialize, Clone)]
|
||||
pub struct DndConfig {
|
||||
pub format_dnd: String,
|
||||
pub format_normal: String,
|
||||
}
|
||||
|
||||
impl Default for DndConfig {
|
||||
fn default() -> Self {
|
||||
Self {
|
||||
format_dnd: "<span size='large'></span>".to_string(),
|
||||
format_normal: "<span size='large'></span>".to_string(),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
static TOKEN_RE: LazyLock<Regex> =
|
||||
LazyLock::new(|| Regex::new(r"\{([a-zA-Z0-9_]+)(?::([<>\^])?(\d+)?(?:\.(\d+))?)?\}").unwrap());
|
||||
|
||||
@@ -284,6 +368,13 @@ impl Config {
|
||||
&self.bt.format_plugin,
|
||||
&["alias", "left", "right", "anc", "mac"],
|
||||
);
|
||||
validate_format(
|
||||
"mpris",
|
||||
&self.mpris.format,
|
||||
&["artist", "title", "album", "status_icon"],
|
||||
);
|
||||
validate_format("backlight", &self.backlight.format, &["percentage", "icon"]);
|
||||
validate_format("keyboard", &self.keyboard.format, &["layout"]);
|
||||
}
|
||||
}
|
||||
|
||||
@@ -330,7 +421,7 @@ mod tests {
|
||||
let config = Config::default();
|
||||
assert_eq!(
|
||||
config.general.menu_command,
|
||||
"fuzzel --dmenu --prompt '{prompt}'"
|
||||
"fuzzel --dmenu --prompt \"$FLUXO_PROMPT\""
|
||||
);
|
||||
assert!(config.cpu.format.contains("usage"));
|
||||
assert!(config.cpu.format.contains("temp"));
|
||||
@@ -344,7 +435,7 @@ mod tests {
|
||||
// Should fallback to defaults without panicking
|
||||
assert_eq!(
|
||||
config.general.menu_command,
|
||||
"fuzzel --dmenu --prompt '{prompt}'"
|
||||
"fuzzel --dmenu --prompt \"$FLUXO_PROMPT\""
|
||||
);
|
||||
}
|
||||
|
||||
@@ -371,7 +462,7 @@ mod tests {
|
||||
// Should fallback to defaults
|
||||
assert_eq!(
|
||||
config.general.menu_command,
|
||||
"fuzzel --dmenu --prompt '{prompt}'"
|
||||
"fuzzel --dmenu --prompt \"$FLUXO_PROMPT\""
|
||||
);
|
||||
}
|
||||
|
||||
@@ -383,7 +474,7 @@ mod tests {
|
||||
let config = load_config(Some(tmpfile.path().to_path_buf()));
|
||||
assert_eq!(
|
||||
config.general.menu_command,
|
||||
"fuzzel --dmenu --prompt '{prompt}'"
|
||||
"fuzzel --dmenu --prompt \"$FLUXO_PROMPT\""
|
||||
);
|
||||
assert!(config.cpu.format.contains("usage"));
|
||||
}
|
||||
|
||||
+249
-35
@@ -3,18 +3,24 @@ use crate::error::FluxoError;
|
||||
use crate::ipc::socket_path;
|
||||
use crate::modules::WaybarModule;
|
||||
use crate::modules::audio::AudioDaemon;
|
||||
use crate::modules::backlight::BacklightDaemon;
|
||||
use crate::modules::bt::BtDaemon;
|
||||
use crate::modules::dnd::DndDaemon;
|
||||
use crate::modules::hardware::HardwareDaemon;
|
||||
use crate::modules::keyboard::KeyboardDaemon;
|
||||
use crate::modules::mpris::MprisDaemon;
|
||||
use crate::modules::network::NetworkDaemon;
|
||||
use crate::signaler::WaybarSignaler;
|
||||
use crate::state::AppReceivers;
|
||||
use anyhow::Result;
|
||||
use notify::{Config as NotifyConfig, Event, RecommendedWatcher, RecursiveMode, Watcher};
|
||||
use std::collections::HashMap;
|
||||
use std::fs;
|
||||
use std::path::PathBuf;
|
||||
use std::sync::Arc;
|
||||
use tokio::io::{AsyncBufReadExt, AsyncWriteExt, BufReader};
|
||||
use tokio::net::UnixListener;
|
||||
use tokio::sync::{RwLock, watch};
|
||||
use tokio::sync::{RwLock, mpsc, watch};
|
||||
use tokio::time::{Duration, Instant, sleep};
|
||||
use tokio_util::sync::CancellationToken;
|
||||
use tracing::{debug, error, info, warn};
|
||||
@@ -30,6 +36,18 @@ impl Drop for SocketGuard {
|
||||
}
|
||||
}
|
||||
|
||||
fn get_config_path(custom_path: Option<PathBuf>) -> PathBuf {
|
||||
custom_path.unwrap_or_else(|| {
|
||||
let config_dir = std::env::var("XDG_CONFIG_HOME")
|
||||
.map(PathBuf::from)
|
||||
.unwrap_or_else(|_| {
|
||||
let home = std::env::var("HOME").unwrap_or_else(|_| String::from("/"));
|
||||
PathBuf::from(home).join(".config")
|
||||
});
|
||||
config_dir.join("fluxo/config.toml")
|
||||
})
|
||||
}
|
||||
|
||||
pub async fn run_daemon(config_path: Option<PathBuf>) -> Result<()> {
|
||||
let sock_path = socket_path();
|
||||
|
||||
@@ -46,7 +64,13 @@ pub async fn run_daemon(config_path: Option<PathBuf>) -> Result<()> {
|
||||
let (disks_tx, disks_rx) = watch::channel(Default::default());
|
||||
let (bt_tx, bt_rx) = watch::channel(Default::default());
|
||||
let (audio_tx, audio_rx) = watch::channel(Default::default());
|
||||
let (mpris_tx, mpris_rx) = watch::channel(Default::default());
|
||||
let (backlight_tx, backlight_rx) = watch::channel(Default::default());
|
||||
let (keyboard_tx, keyboard_rx) = watch::channel(Default::default());
|
||||
let (dnd_tx, dnd_rx) = watch::channel(Default::default());
|
||||
let health = Arc::new(RwLock::new(HashMap::new()));
|
||||
let (bt_force_tx, mut bt_force_rx) = mpsc::channel(1);
|
||||
let (audio_cmd_tx, audio_cmd_rx) = mpsc::channel(8);
|
||||
|
||||
let receivers = AppReceivers {
|
||||
network: net_rx,
|
||||
@@ -57,7 +81,13 @@ pub async fn run_daemon(config_path: Option<PathBuf>) -> Result<()> {
|
||||
disks: disks_rx,
|
||||
bluetooth: bt_rx,
|
||||
audio: audio_rx,
|
||||
mpris: mpris_rx,
|
||||
backlight: backlight_rx,
|
||||
keyboard: keyboard_rx,
|
||||
dnd: dnd_rx,
|
||||
health: Arc::clone(&health),
|
||||
bt_force_poll: bt_force_tx,
|
||||
audio_cmd_tx,
|
||||
};
|
||||
|
||||
let listener = UnixListener::bind(&sock_path)?;
|
||||
@@ -74,8 +104,56 @@ pub async fn run_daemon(config_path: Option<PathBuf>) -> Result<()> {
|
||||
token_clone.cancel();
|
||||
});
|
||||
|
||||
let config_path_clone = config_path.clone();
|
||||
let config = Arc::new(RwLock::new(crate::config::load_config(config_path)));
|
||||
let resolved_config_path = get_config_path(config_path.clone());
|
||||
let config = Arc::new(RwLock::new(crate::config::load_config(config_path.clone())));
|
||||
|
||||
// 0. Config Watcher (Hot Reload)
|
||||
let watcher_config = Arc::clone(&config);
|
||||
let watcher_path = resolved_config_path.clone();
|
||||
tokio::spawn(async move {
|
||||
let (ev_tx, mut ev_rx) = mpsc::channel(1);
|
||||
let mut watcher = RecommendedWatcher::new(
|
||||
move |res: notify::Result<Event>| {
|
||||
if let Ok(event) = res
|
||||
&& (event.kind.is_modify() || event.kind.is_create())
|
||||
{
|
||||
let _ = ev_tx.blocking_send(());
|
||||
}
|
||||
},
|
||||
NotifyConfig::default(),
|
||||
)
|
||||
.unwrap();
|
||||
|
||||
if let Some(parent) = watcher_path.parent() {
|
||||
let _ = watcher.watch(parent, RecursiveMode::NonRecursive);
|
||||
}
|
||||
|
||||
info!("Config watcher started on {:?}", watcher_path);
|
||||
|
||||
loop {
|
||||
tokio::select! {
|
||||
_ = ev_rx.recv() => {
|
||||
// Debounce reloads
|
||||
sleep(Duration::from_millis(100)).await;
|
||||
while ev_rx.try_recv().is_ok() {}
|
||||
reload_config(&watcher_config, Some(watcher_path.clone())).await;
|
||||
}
|
||||
}
|
||||
}
|
||||
});
|
||||
|
||||
// 0.1 SIGHUP Handler
|
||||
let hup_config = Arc::clone(&config);
|
||||
let hup_path = resolved_config_path.clone();
|
||||
tokio::spawn(async move {
|
||||
use tokio::signal::unix::{SignalKind, signal};
|
||||
let mut stream = signal(SignalKind::hangup()).unwrap();
|
||||
loop {
|
||||
stream.recv().await;
|
||||
info!("Received SIGHUP, reloading config...");
|
||||
reload_config(&hup_config, Some(hup_path.clone())).await;
|
||||
}
|
||||
});
|
||||
|
||||
// 1. Network Task
|
||||
let token = cancel_token.clone();
|
||||
@@ -84,14 +162,13 @@ pub async fn run_daemon(config_path: Option<PathBuf>) -> Result<()> {
|
||||
info!("Starting Network polling task");
|
||||
let mut daemon = NetworkDaemon::new();
|
||||
loop {
|
||||
tokio::select! {
|
||||
_ = token.cancelled() => break,
|
||||
_ = sleep(Duration::from_secs(1)) => {
|
||||
if !is_in_backoff("net", &net_health).await {
|
||||
let res = daemon.poll(&net_tx).await;
|
||||
handle_poll_result("net", res, &net_health).await;
|
||||
}
|
||||
}
|
||||
tokio::select! {
|
||||
_ = token.cancelled() => break,
|
||||
_ = sleep(Duration::from_secs(1)) => {}
|
||||
}
|
||||
}
|
||||
info!("Network task shut down.");
|
||||
@@ -104,13 +181,12 @@ pub async fn run_daemon(config_path: Option<PathBuf>) -> Result<()> {
|
||||
info!("Starting Fast Hardware polling task");
|
||||
let mut daemon = HardwareDaemon::new();
|
||||
loop {
|
||||
tokio::select! {
|
||||
_ = token.cancelled() => break,
|
||||
_ = sleep(Duration::from_secs(1)) => {
|
||||
if !is_in_backoff("cpu", &hw_health).await {
|
||||
daemon.poll_fast(&cpu_tx, &mem_tx, &sys_tx).await;
|
||||
}
|
||||
}
|
||||
tokio::select! {
|
||||
_ = token.cancelled() => break,
|
||||
_ = sleep(Duration::from_secs(1)) => {}
|
||||
}
|
||||
}
|
||||
info!("Fast Hardware task shut down.");
|
||||
@@ -123,13 +199,12 @@ pub async fn run_daemon(config_path: Option<PathBuf>) -> Result<()> {
|
||||
info!("Starting Slow Hardware polling task");
|
||||
let mut daemon = HardwareDaemon::new();
|
||||
loop {
|
||||
tokio::select! {
|
||||
_ = token.cancelled() => break,
|
||||
_ = sleep(Duration::from_secs(5)) => {
|
||||
if !is_in_backoff("gpu", &slow_health).await {
|
||||
daemon.poll_slow(&gpu_tx, &disks_tx).await;
|
||||
}
|
||||
}
|
||||
tokio::select! {
|
||||
_ = token.cancelled() => break,
|
||||
_ = sleep(Duration::from_secs(5)) => {}
|
||||
}
|
||||
}
|
||||
info!("Slow Hardware task shut down.");
|
||||
@@ -144,14 +219,14 @@ pub async fn run_daemon(config_path: Option<PathBuf>) -> Result<()> {
|
||||
info!("Starting Bluetooth polling task");
|
||||
let mut daemon = BtDaemon::new();
|
||||
loop {
|
||||
tokio::select! {
|
||||
_ = token.cancelled() => break,
|
||||
_ = sleep(Duration::from_secs(2)) => {
|
||||
if !is_in_backoff("bt", &bt_health).await {
|
||||
let config = poll_config.read().await;
|
||||
daemon.poll(&bt_tx, &poll_receivers, &config).await;
|
||||
}
|
||||
}
|
||||
tokio::select! {
|
||||
_ = token.cancelled() => break,
|
||||
_ = bt_force_rx.recv() => {},
|
||||
_ = sleep(Duration::from_secs(2)) => {}
|
||||
}
|
||||
}
|
||||
info!("Bluetooth task shut down.");
|
||||
@@ -159,7 +234,32 @@ pub async fn run_daemon(config_path: Option<PathBuf>) -> Result<()> {
|
||||
|
||||
// 5. Audio Thread (Event driven)
|
||||
let audio_daemon = AudioDaemon::new();
|
||||
audio_daemon.start(&audio_tx);
|
||||
audio_daemon.start(&audio_tx, audio_cmd_rx);
|
||||
|
||||
// 5.1 Backlight Thread (Event driven)
|
||||
let backlight_daemon = BacklightDaemon::new();
|
||||
backlight_daemon.start(backlight_tx);
|
||||
|
||||
// 5.2 Keyboard Thread (Event driven)
|
||||
let keyboard_daemon = KeyboardDaemon::new();
|
||||
keyboard_daemon.start(keyboard_tx);
|
||||
|
||||
// 5.3 DND Thread (Event driven)
|
||||
let dnd_daemon = DndDaemon::new();
|
||||
dnd_daemon.start(dnd_tx);
|
||||
|
||||
// 5.4 MPRIS Thread
|
||||
let mpris_daemon = MprisDaemon::new();
|
||||
mpris_daemon.start(mpris_tx);
|
||||
|
||||
// 6. Waybar Signaler Task
|
||||
let signaler = WaybarSignaler::new();
|
||||
let sig_config = Arc::clone(&config);
|
||||
let sig_receivers = receivers.clone();
|
||||
tokio::spawn(async move {
|
||||
info!("Starting Waybar Signaler task");
|
||||
signaler.run(sig_config, sig_receivers).await;
|
||||
});
|
||||
|
||||
info!("Fluxo daemon successfully bound to socket: {}", sock_path);
|
||||
|
||||
@@ -173,7 +273,7 @@ pub async fn run_daemon(config_path: Option<PathBuf>) -> Result<()> {
|
||||
Ok((mut stream, _)) => {
|
||||
let state_clone = receivers.clone();
|
||||
let config_clone = Arc::clone(&config);
|
||||
let cp_clone = config_path_clone.clone();
|
||||
let cp_clone = config_path.clone();
|
||||
tokio::spawn(async move {
|
||||
let (reader, mut writer) = stream.split();
|
||||
let mut reader = BufReader::new(reader);
|
||||
@@ -191,12 +291,8 @@ pub async fn run_daemon(config_path: Option<PathBuf>) -> Result<()> {
|
||||
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);
|
||||
let mut config_lock = config_clone.write().await;
|
||||
*config_lock = new_config;
|
||||
reload_config(&config_clone, cp_clone).await;
|
||||
let _ = writer.write_all(b"{\"text\":\"ok\"}").await;
|
||||
info!("Configuration reloaded successfully.");
|
||||
return;
|
||||
}
|
||||
|
||||
@@ -281,6 +377,89 @@ async fn is_in_backoff(
|
||||
false
|
||||
}
|
||||
|
||||
pub async fn reload_config(config_lock: &Arc<RwLock<Config>>, path: Option<PathBuf>) {
|
||||
info!("Reloading configuration...");
|
||||
let new_config = crate::config::load_config(path);
|
||||
let mut lock = config_lock.write().await;
|
||||
*lock = new_config;
|
||||
info!("Configuration reloaded successfully.");
|
||||
}
|
||||
|
||||
pub async fn evaluate_module_for_signaler(
|
||||
module_name: &str,
|
||||
state: &AppReceivers,
|
||||
config: &Config,
|
||||
) -> Option<String> {
|
||||
let result = match module_name {
|
||||
"net" | "network" => {
|
||||
crate::modules::network::NetworkModule
|
||||
.run(config, state, &[])
|
||||
.await
|
||||
}
|
||||
"cpu" => crate::modules::cpu::CpuModule.run(config, state, &[]).await,
|
||||
"mem" | "memory" => {
|
||||
crate::modules::memory::MemoryModule
|
||||
.run(config, state, &[])
|
||||
.await
|
||||
}
|
||||
"disk" => {
|
||||
crate::modules::disk::DiskModule
|
||||
.run(config, state, &["/"])
|
||||
.await
|
||||
}
|
||||
"pool" | "btrfs" => {
|
||||
crate::modules::btrfs::BtrfsModule
|
||||
.run(config, state, &["btrfs"])
|
||||
.await
|
||||
}
|
||||
"vol" | "audio" => {
|
||||
crate::modules::audio::AudioModule
|
||||
.run(config, state, &["sink", "show"])
|
||||
.await
|
||||
}
|
||||
"mic" => {
|
||||
crate::modules::audio::AudioModule
|
||||
.run(config, state, &["source", "show"])
|
||||
.await
|
||||
}
|
||||
"gpu" => crate::modules::gpu::GpuModule.run(config, state, &[]).await,
|
||||
"sys" => crate::modules::sys::SysModule.run(config, state, &[]).await,
|
||||
"bt" | "bluetooth" => {
|
||||
crate::modules::bt::BtModule
|
||||
.run(config, state, &["show"])
|
||||
.await
|
||||
}
|
||||
"power" => {
|
||||
crate::modules::power::PowerModule
|
||||
.run(config, state, &[])
|
||||
.await
|
||||
}
|
||||
"game" => {
|
||||
crate::modules::game::GameModule
|
||||
.run(config, state, &[])
|
||||
.await
|
||||
}
|
||||
"backlight" => {
|
||||
crate::modules::backlight::BacklightModule
|
||||
.run(config, state, &[])
|
||||
.await
|
||||
}
|
||||
"kbd" | "keyboard" => {
|
||||
crate::modules::keyboard::KeyboardModule
|
||||
.run(config, state, &[])
|
||||
.await
|
||||
}
|
||||
"dnd" => crate::modules::dnd::DndModule.run(config, state, &[]).await,
|
||||
"mpris" => {
|
||||
crate::modules::mpris::MprisModule
|
||||
.run(config, state, &[])
|
||||
.await
|
||||
}
|
||||
_ => return None,
|
||||
};
|
||||
result.ok().and_then(|out| serde_json::to_string(&out).ok())
|
||||
}
|
||||
|
||||
async fn handle_request(
|
||||
module_name: &str,
|
||||
args: &[&str],
|
||||
@@ -288,20 +467,27 @@ async fn handle_request(
|
||||
config_lock: &Arc<RwLock<Config>>,
|
||||
) -> String {
|
||||
// 1. Check Circuit Breaker status
|
||||
let is_in_backoff = {
|
||||
let (is_in_backoff, cached_output) = {
|
||||
let lock = state.health.read().await;
|
||||
if let Some(health) = lock.get(module_name) {
|
||||
if let Some(until) = health.backoff_until {
|
||||
let in_backoff = if let Some(until) = health.backoff_until {
|
||||
Instant::now() < until
|
||||
} else {
|
||||
false
|
||||
}
|
||||
};
|
||||
(in_backoff, health.last_successful_output.clone())
|
||||
} else {
|
||||
false
|
||||
(false, None)
|
||||
}
|
||||
};
|
||||
|
||||
if is_in_backoff {
|
||||
if let Some(mut cached) = cached_output {
|
||||
// Add a "warning" class to indicate stale data
|
||||
let class = cached.class.unwrap_or_default();
|
||||
cached.class = Some(format!("{} warning", class).trim().to_string());
|
||||
return serde_json::to_string(&cached).unwrap_or_else(|_| "{}".to_string());
|
||||
}
|
||||
return format!(
|
||||
"{{\"text\":\"\u{200B}Cooling down ({})\u{200B}\",\"class\":\"error\"}}",
|
||||
module_name
|
||||
@@ -336,14 +522,14 @@ async fn handle_request(
|
||||
.run(&config, state, args)
|
||||
.await
|
||||
}
|
||||
"vol" => {
|
||||
"vol" | "audio" => {
|
||||
crate::modules::audio::AudioModule
|
||||
.run(&config, state, &["sink", args.first().unwrap_or(&"show")])
|
||||
.run(&config, state, args)
|
||||
.await
|
||||
}
|
||||
"mic" => {
|
||||
crate::modules::audio::AudioModule
|
||||
.run(&config, state, &["source", args.first().unwrap_or(&"show")])
|
||||
.run(&config, state, args)
|
||||
.await
|
||||
}
|
||||
"gpu" => {
|
||||
@@ -367,20 +553,41 @@ async fn handle_request(
|
||||
.run(&config, state, args)
|
||||
.await
|
||||
}
|
||||
"backlight" => {
|
||||
crate::modules::backlight::BacklightModule
|
||||
.run(&config, state, args)
|
||||
.await
|
||||
}
|
||||
"kbd" | "keyboard" => {
|
||||
crate::modules::keyboard::KeyboardModule
|
||||
.run(&config, state, args)
|
||||
.await
|
||||
}
|
||||
"dnd" => {
|
||||
crate::modules::dnd::DndModule
|
||||
.run(&config, state, args)
|
||||
.await
|
||||
}
|
||||
"mpris" => {
|
||||
crate::modules::mpris::MprisModule
|
||||
.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
|
||||
// 2. Update Health and Cache based on result
|
||||
{
|
||||
let mut lock = state.health.write().await;
|
||||
let health = lock.entry(module_name.to_string()).or_default();
|
||||
match &result {
|
||||
Ok(_) => {
|
||||
Ok(output) => {
|
||||
health.consecutive_failures = 0;
|
||||
health.backoff_until = None;
|
||||
health.last_successful_output = Some(output.clone());
|
||||
}
|
||||
Err(e) => {
|
||||
health.consecutive_failures += 1;
|
||||
@@ -397,6 +604,13 @@ async fn handle_request(
|
||||
match result {
|
||||
Ok(output) => serde_json::to_string(&output).unwrap_or_else(|_| "{}".to_string()),
|
||||
Err(e) => {
|
||||
// If we have a cached output, return it as fallback even on immediate error
|
||||
if let Some(mut cached) = cached_output {
|
||||
let class = cached.class.unwrap_or_default();
|
||||
cached.class = Some(format!("{} warning", class).trim().to_string());
|
||||
return serde_json::to_string(&cached).unwrap_or_else(|_| "{}".to_string());
|
||||
}
|
||||
|
||||
let error_msg = e.to_string();
|
||||
error!(module = module_name, error = %error_msg, "Module execution failed");
|
||||
let err_out = crate::output::WaybarOutput {
|
||||
|
||||
+43
-88
@@ -4,6 +4,7 @@ mod error;
|
||||
mod ipc;
|
||||
mod modules;
|
||||
mod output;
|
||||
mod signaler;
|
||||
mod state;
|
||||
mod utils;
|
||||
|
||||
@@ -18,7 +19,14 @@ use tracing_subscriber::{EnvFilter, fmt, prelude::*};
|
||||
#[command(about = "A high-performance daemon/client for Waybar custom modules", long_about = None)]
|
||||
struct Cli {
|
||||
#[command(subcommand)]
|
||||
command: Commands,
|
||||
command: Option<Commands>,
|
||||
|
||||
/// Module name to query or interact with
|
||||
module: Option<String>,
|
||||
|
||||
/// Arguments to pass to the module
|
||||
#[arg(trailing_var_arg = true)]
|
||||
args: Vec<String>,
|
||||
}
|
||||
|
||||
#[derive(Subcommand)]
|
||||
@@ -31,51 +39,6 @@ enum Commands {
|
||||
},
|
||||
/// Reload the daemon configuration
|
||||
Reload,
|
||||
/// Network speed module
|
||||
#[command(alias = "network")]
|
||||
Net,
|
||||
/// CPU usage and temp module
|
||||
Cpu,
|
||||
/// Memory usage module
|
||||
#[command(alias = "memory")]
|
||||
Mem,
|
||||
/// Disk usage module (path defaults to /)
|
||||
Disk {
|
||||
#[arg(default_value = "/")]
|
||||
path: String,
|
||||
},
|
||||
/// Storage pool aggregate module (e.g., btrfs)
|
||||
#[command(alias = "btrfs")]
|
||||
Pool {
|
||||
#[arg(default_value = "btrfs")]
|
||||
kind: String,
|
||||
},
|
||||
/// Audio volume (sink) control
|
||||
Vol {
|
||||
/// Cycle to the next available output device
|
||||
#[arg(short, long)]
|
||||
cycle: bool,
|
||||
},
|
||||
/// Microphone (source) control
|
||||
Mic {
|
||||
/// Cycle to the next available input device
|
||||
#[arg(short, long)]
|
||||
cycle: bool,
|
||||
},
|
||||
/// GPU usage, VRAM, and temp module
|
||||
Gpu,
|
||||
/// System load average and uptime
|
||||
Sys,
|
||||
/// Bluetooth audio device status
|
||||
#[command(alias = "bluetooth")]
|
||||
Bt {
|
||||
#[arg(default_value = "show")]
|
||||
action: String,
|
||||
},
|
||||
/// System power and battery status
|
||||
Power,
|
||||
/// Hyprland gamemode status
|
||||
Game,
|
||||
}
|
||||
|
||||
fn main() {
|
||||
@@ -86,7 +49,8 @@ fn main() {
|
||||
|
||||
let cli = Cli::parse();
|
||||
|
||||
match &cli.command {
|
||||
if let Some(command) = &cli.command {
|
||||
match command {
|
||||
Commands::Daemon { config } => {
|
||||
info!("Starting Fluxo daemon...");
|
||||
let rt = tokio::runtime::Builder::new_multi_thread()
|
||||
@@ -106,29 +70,16 @@ fn main() {
|
||||
process::exit(1);
|
||||
}
|
||||
},
|
||||
Commands::Net => handle_ipc_response(ipc::request_data("net", &[])),
|
||||
Commands::Cpu => handle_ipc_response(ipc::request_data("cpu", &[])),
|
||||
Commands::Mem => handle_ipc_response(ipc::request_data("mem", &[])),
|
||||
Commands::Disk { path } => handle_ipc_response(ipc::request_data("disk", &[path])),
|
||||
Commands::Pool { kind } => handle_ipc_response(ipc::request_data("pool", &[kind])),
|
||||
Commands::Vol { cycle } => {
|
||||
let action = if *cycle { "cycle" } else { "show" };
|
||||
handle_ipc_response(ipc::request_data("vol", &[action]));
|
||||
}
|
||||
Commands::Mic { cycle } => {
|
||||
let action = if *cycle { "cycle" } else { "show" };
|
||||
handle_ipc_response(ipc::request_data("mic", &[action]));
|
||||
return;
|
||||
}
|
||||
Commands::Gpu => handle_ipc_response(ipc::request_data("gpu", &[])),
|
||||
Commands::Sys => handle_ipc_response(ipc::request_data("sys", &[])),
|
||||
Commands::Bt { action } => {
|
||||
if action == "menu" {
|
||||
// Client-side execution of the menu
|
||||
let config = config::load_config(None);
|
||||
|
||||
if let Some(module) = &cli.module {
|
||||
// Special case for client-side Bluetooth menu which requires UI
|
||||
if module == "bt" && cli.args.first().map(|s| s.as_str()) == Some("menu") {
|
||||
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())
|
||||
@@ -145,23 +96,13 @@ fn main() {
|
||||
|
||||
items.push("--- Connect Device ---".to_string());
|
||||
|
||||
let devices_out = match std::process::Command::new("bluetoothctl")
|
||||
.args(["devices"])
|
||||
.output()
|
||||
if let Ok(json_str) = ipc::request_data("bt", &["menu_data"])
|
||||
&& let Ok(val) = serde_json::from_str::<serde_json::Value>(&json_str)
|
||||
&& let Some(devices_str) = val.get("text").and_then(|t| t.as_str())
|
||||
{
|
||||
Ok(out) => out,
|
||||
Err(e) => {
|
||||
error!("bluetoothctl not found or failed: {}", e);
|
||||
return;
|
||||
}
|
||||
};
|
||||
let stdout = String::from_utf8_lossy(&devices_out.stdout);
|
||||
|
||||
for line in stdout.lines() {
|
||||
if line.starts_with("Device ") {
|
||||
let parts: Vec<&str> = line.splitn(3, ' ').collect();
|
||||
if parts.len() == 3 {
|
||||
items.push(format!("{} ({})", parts[2], parts[1]));
|
||||
for line in devices_str.lines() {
|
||||
if !line.is_empty() {
|
||||
items.push(line.to_string());
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -180,9 +121,7 @@ fn main() {
|
||||
&& let Some(mac_end) = selected.rfind(')')
|
||||
{
|
||||
let mac = &selected[mac_start + 1..mac_end];
|
||||
let _ = std::process::Command::new("bluetoothctl")
|
||||
.args(["connect", mac])
|
||||
.status();
|
||||
handle_ipc_response(ipc::request_data("bt", &["connect", mac]));
|
||||
}
|
||||
}
|
||||
} else {
|
||||
@@ -190,10 +129,26 @@ fn main() {
|
||||
}
|
||||
return;
|
||||
}
|
||||
handle_ipc_response(ipc::request_data("bt", &[action]));
|
||||
}
|
||||
Commands::Power => handle_ipc_response(ipc::request_data("power", &[])),
|
||||
Commands::Game => handle_ipc_response(ipc::request_data("game", &[])),
|
||||
|
||||
// Generic module dispatch
|
||||
// Translate module-specific shorthand targets
|
||||
let (actual_module, actual_args) = if module == "vol" {
|
||||
let mut new_args = vec!["sink".to_string()];
|
||||
new_args.extend(cli.args.clone());
|
||||
("vol".to_string(), new_args)
|
||||
} else if module == "mic" {
|
||||
let mut new_args = vec!["source".to_string()];
|
||||
new_args.extend(cli.args.clone());
|
||||
("vol".to_string(), new_args)
|
||||
} else {
|
||||
(module.clone(), cli.args.clone())
|
||||
};
|
||||
|
||||
let args_ref: Vec<&str> = actual_args.iter().map(|s| s.as_str()).collect();
|
||||
handle_ipc_response(ipc::request_data(&actual_module, &args_ref));
|
||||
} else {
|
||||
println!("Please specify a module or command. See --help.");
|
||||
process::exit(1);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
+211
-82
@@ -9,9 +9,23 @@ 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 tokio::sync::watch;
|
||||
use tokio::sync::{mpsc, watch};
|
||||
use tracing::error;
|
||||
|
||||
pub enum AudioCommand {
|
||||
ChangeVolume {
|
||||
is_sink: bool,
|
||||
step_val: u32,
|
||||
is_up: bool,
|
||||
},
|
||||
ToggleMute {
|
||||
is_sink: bool,
|
||||
},
|
||||
CycleDevice {
|
||||
is_sink: bool,
|
||||
},
|
||||
}
|
||||
|
||||
pub struct AudioDaemon;
|
||||
|
||||
impl AudioDaemon {
|
||||
@@ -19,7 +33,11 @@ impl AudioDaemon {
|
||||
Self
|
||||
}
|
||||
|
||||
pub fn start(&self, state_tx: &watch::Sender<AudioState>) {
|
||||
pub fn start(
|
||||
&self,
|
||||
state_tx: &watch::Sender<AudioState>,
|
||||
mut cmd_rx: mpsc::Receiver<AudioCommand>,
|
||||
) {
|
||||
let state_tx = state_tx.clone();
|
||||
|
||||
std::thread::spawn(move || {
|
||||
@@ -64,11 +82,12 @@ impl AudioDaemon {
|
||||
context.subscribe(interest, |_| {});
|
||||
|
||||
let (tx, rx) = std::sync::mpsc::channel();
|
||||
let tx_cb = tx.clone();
|
||||
|
||||
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(());
|
||||
let _ = tx_cb.send(());
|
||||
}
|
||||
_ => {}
|
||||
}
|
||||
@@ -76,14 +95,110 @@ impl AudioDaemon {
|
||||
|
||||
mainloop.unlock();
|
||||
|
||||
// Background polling loop driven by events or a 2s fallback timeout
|
||||
loop {
|
||||
let _ = rx.recv_timeout(Duration::from_secs(2));
|
||||
{
|
||||
if let Ok(cmd) = cmd_rx.try_recv() {
|
||||
mainloop.lock();
|
||||
let _ = fetch_audio_data_sync(&mut context, &state_tx);
|
||||
mainloop.unlock();
|
||||
match cmd {
|
||||
AudioCommand::ChangeVolume {
|
||||
is_sink,
|
||||
step_val,
|
||||
is_up,
|
||||
} => {
|
||||
let current = state_tx.borrow().clone();
|
||||
let (name, mut vol, channels) = if is_sink {
|
||||
(
|
||||
current.sink.name.clone(),
|
||||
current.sink.volume,
|
||||
current.sink.channels,
|
||||
)
|
||||
} else {
|
||||
(
|
||||
current.source.name.clone(),
|
||||
current.source.volume,
|
||||
current.source.channels,
|
||||
)
|
||||
};
|
||||
|
||||
if is_up {
|
||||
vol = vol.saturating_add(step_val as u8).min(150);
|
||||
} else {
|
||||
vol = vol.saturating_sub(step_val as u8);
|
||||
}
|
||||
|
||||
let pulse_vol = Volume(
|
||||
(vol as f64 / 100.0 * Volume::NORMAL.0 as f64).round() as u32,
|
||||
);
|
||||
let mut cvol = libpulse_binding::volume::ChannelVolumes::default();
|
||||
cvol.set(channels.max(1), pulse_vol);
|
||||
|
||||
if is_sink {
|
||||
context
|
||||
.introspect()
|
||||
.set_sink_volume_by_name(&name, &cvol, None);
|
||||
} else {
|
||||
context
|
||||
.introspect()
|
||||
.set_source_volume_by_name(&name, &cvol, None);
|
||||
}
|
||||
}
|
||||
AudioCommand::ToggleMute { is_sink } => {
|
||||
let current = state_tx.borrow().clone();
|
||||
let (name, muted) = if is_sink {
|
||||
(current.sink.name.clone(), current.sink.muted)
|
||||
} else {
|
||||
(current.source.name.clone(), current.source.muted)
|
||||
};
|
||||
|
||||
if is_sink {
|
||||
context
|
||||
.introspect()
|
||||
.set_sink_mute_by_name(&name, !muted, None);
|
||||
} else {
|
||||
context
|
||||
.introspect()
|
||||
.set_source_mute_by_name(&name, !muted, None);
|
||||
}
|
||||
}
|
||||
AudioCommand::CycleDevice { is_sink } => {
|
||||
let current = state_tx.borrow().clone();
|
||||
let current_name = if is_sink {
|
||||
current.sink.name.clone()
|
||||
} else {
|
||||
current.source.name.clone()
|
||||
};
|
||||
|
||||
let devices = if is_sink {
|
||||
¤t.available_sinks
|
||||
} else {
|
||||
¤t.available_sources
|
||||
};
|
||||
if !devices.is_empty() {
|
||||
let current_idx =
|
||||
devices.iter().position(|d| d == ¤t_name).unwrap_or(0);
|
||||
let next_idx = (current_idx + 1) % devices.len();
|
||||
let next_dev = &devices[next_idx];
|
||||
|
||||
if is_sink {
|
||||
context.set_default_sink(next_dev, |_| {});
|
||||
} else {
|
||||
context.set_default_source(next_dev, |_| {});
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
mainloop.unlock();
|
||||
let _ = tx.send(());
|
||||
}
|
||||
|
||||
let _ = rx.recv_timeout(Duration::from_millis(50));
|
||||
while rx.try_recv().is_ok() {}
|
||||
|
||||
mainloop.lock();
|
||||
|
||||
// Fetch data and update available sinks/sources
|
||||
let _ = fetch_audio_data_sync(&mut context, &state_tx);
|
||||
|
||||
mainloop.unlock();
|
||||
}
|
||||
});
|
||||
}
|
||||
@@ -116,9 +231,16 @@ fn fetch_audio_data_sync(
|
||||
|
||||
let tx_sink = state_tx.clone();
|
||||
context.introspect().get_sink_info_list(move |res| {
|
||||
if let ListResult::Item(item) = res {
|
||||
let mut current = tx_sink.borrow().clone();
|
||||
// If this matches our default sink name, or if we don't have details for any yet
|
||||
match res {
|
||||
ListResult::Item(item) => {
|
||||
if let Some(name) = item.name.as_ref() {
|
||||
let name_str = name.to_string();
|
||||
if !current.available_sinks.contains(&name_str) {
|
||||
current.available_sinks.push(name_str);
|
||||
}
|
||||
}
|
||||
|
||||
let is_default = item
|
||||
.name
|
||||
.as_ref()
|
||||
@@ -130,18 +252,38 @@ fn fetch_audio_data_sync(
|
||||
.as_ref()
|
||||
.map(|s| s.to_string())
|
||||
.unwrap_or_default();
|
||||
current.sink.volume =
|
||||
((item.volume.avg().0 as f64 / Volume::NORMAL.0 as f64) * 100.0).round() as u8;
|
||||
current.sink.volume = ((item.volume.avg().0 as f64 / Volume::NORMAL.0 as f64)
|
||||
* 100.0)
|
||||
.round() as u8;
|
||||
current.sink.muted = item.mute;
|
||||
current.sink.channels = item.volume.len();
|
||||
}
|
||||
let _ = tx_sink.send(current);
|
||||
}
|
||||
ListResult::End => {
|
||||
// Clear the list on End so it rebuilds fresh next time
|
||||
current.available_sinks.clear();
|
||||
let _ = tx_sink.send(current);
|
||||
}
|
||||
ListResult::Error => {}
|
||||
}
|
||||
});
|
||||
|
||||
let tx_source = state_tx.clone();
|
||||
context.introspect().get_source_info_list(move |res| {
|
||||
if let ListResult::Item(item) = res {
|
||||
let mut current = tx_source.borrow().clone();
|
||||
match res {
|
||||
ListResult::Item(item) => {
|
||||
if let Some(name) = item.name.as_ref() {
|
||||
let name_str = name.to_string();
|
||||
// PulseAudio includes monitor sources, ignore them if we want to
|
||||
if !name_str.contains(".monitor")
|
||||
&& !current.available_sources.contains(&name_str)
|
||||
{
|
||||
current.available_sources.push(name_str);
|
||||
}
|
||||
}
|
||||
|
||||
let is_default = item
|
||||
.name
|
||||
.as_ref()
|
||||
@@ -153,11 +295,20 @@ fn fetch_audio_data_sync(
|
||||
.as_ref()
|
||||
.map(|s| s.to_string())
|
||||
.unwrap_or_default();
|
||||
current.source.volume =
|
||||
((item.volume.avg().0 as f64 / Volume::NORMAL.0 as f64) * 100.0).round() as u8;
|
||||
current.source.volume = ((item.volume.avg().0 as f64 / Volume::NORMAL.0 as f64)
|
||||
* 100.0)
|
||||
.round() as u8;
|
||||
current.source.muted = item.mute;
|
||||
current.source.channels = item.volume.len();
|
||||
}
|
||||
let _ = tx_source.send(current);
|
||||
}
|
||||
ListResult::End => {
|
||||
// Clear the list on End so it rebuilds fresh next time
|
||||
current.available_sources.clear();
|
||||
let _ = tx_source.send(current);
|
||||
}
|
||||
ListResult::Error => {}
|
||||
}
|
||||
});
|
||||
|
||||
@@ -175,10 +326,23 @@ impl WaybarModule for AudioModule {
|
||||
) -> Result<WaybarOutput> {
|
||||
let target_type = args.first().unwrap_or(&"sink");
|
||||
let action = args.get(1).unwrap_or(&"show");
|
||||
let step = args.get(2).unwrap_or(&"5");
|
||||
|
||||
match *action {
|
||||
"up" => {
|
||||
self.change_volume(state, target_type, step, true).await?;
|
||||
Ok(WaybarOutput::default())
|
||||
}
|
||||
"down" => {
|
||||
self.change_volume(state, target_type, step, false).await?;
|
||||
Ok(WaybarOutput::default())
|
||||
}
|
||||
"mute" => {
|
||||
self.toggle_mute(state, target_type).await?;
|
||||
Ok(WaybarOutput::default())
|
||||
}
|
||||
"cycle" => {
|
||||
self.cycle_device(target_type).await?;
|
||||
self.cycle_device(state, target_type).await?;
|
||||
Ok(WaybarOutput::default())
|
||||
}
|
||||
"show" => self.get_status(config, state, target_type).await,
|
||||
@@ -280,76 +444,41 @@ impl AudioModule {
|
||||
})
|
||||
}
|
||||
|
||||
async fn cycle_device(&self, target_type: &str) -> Result<()> {
|
||||
// Keep using pactl for cycling for now as it's a rare action
|
||||
// but we could also implement it natively later.
|
||||
let set_cmd = if target_type == "sink" {
|
||||
"set-default-sink"
|
||||
} else {
|
||||
"set-default-source"
|
||||
};
|
||||
|
||||
// We need to find the "next" device.
|
||||
// For simplicity, let's keep the CLI version for now or refactor later.
|
||||
// The user asked for "step by step".
|
||||
|
||||
let list_cmd = if target_type == "sink" {
|
||||
"sinks"
|
||||
} else {
|
||||
"sources"
|
||||
};
|
||||
let output = tokio::process::Command::new("pactl")
|
||||
.args(["list", "short", list_cmd])
|
||||
.output()
|
||||
.await?;
|
||||
let stdout = String::from_utf8_lossy(&output.stdout);
|
||||
|
||||
let devices: Vec<String> = stdout
|
||||
.lines()
|
||||
.filter_map(|l| {
|
||||
let parts: Vec<&str> = l.split_whitespace().collect();
|
||||
if parts.len() >= 2 {
|
||||
let name = parts[1].to_string();
|
||||
if target_type == "source" && name.contains(".monitor") {
|
||||
None
|
||||
} else {
|
||||
Some(name)
|
||||
}
|
||||
} else {
|
||||
None
|
||||
}
|
||||
async fn change_volume(
|
||||
&self,
|
||||
state: &AppReceivers,
|
||||
target_type: &str,
|
||||
step: &str,
|
||||
is_up: bool,
|
||||
) -> Result<()> {
|
||||
let is_sink = target_type == "sink";
|
||||
let step_val: u32 = step.parse().unwrap_or(5);
|
||||
let _ = state
|
||||
.audio_cmd_tx
|
||||
.send(AudioCommand::ChangeVolume {
|
||||
is_sink,
|
||||
step_val,
|
||||
is_up,
|
||||
})
|
||||
.collect();
|
||||
|
||||
if devices.is_empty() {
|
||||
return Ok(());
|
||||
.await;
|
||||
Ok(())
|
||||
}
|
||||
|
||||
let info_output = tokio::process::Command::new("pactl")
|
||||
.args(["info"])
|
||||
.output()
|
||||
.await?;
|
||||
let info_stdout = String::from_utf8_lossy(&info_output.stdout);
|
||||
let search_key = if target_type == "sink" {
|
||||
"Default Sink:"
|
||||
} else {
|
||||
"Default Source:"
|
||||
};
|
||||
let current_dev = info_stdout
|
||||
.lines()
|
||||
.find(|l| l.contains(search_key))
|
||||
.and_then(|l| l.split(':').nth(1))
|
||||
.map(|s| s.trim())
|
||||
.unwrap_or("");
|
||||
async fn toggle_mute(&self, state: &AppReceivers, target_type: &str) -> Result<()> {
|
||||
let is_sink = target_type == "sink";
|
||||
let _ = state
|
||||
.audio_cmd_tx
|
||||
.send(AudioCommand::ToggleMute { is_sink })
|
||||
.await;
|
||||
Ok(())
|
||||
}
|
||||
|
||||
let current_index = devices.iter().position(|d| d == current_dev).unwrap_or(0);
|
||||
let next_index = (current_index + 1) % devices.len();
|
||||
let next_dev = &devices[next_index];
|
||||
|
||||
tokio::process::Command::new("pactl")
|
||||
.args([set_cmd, next_dev])
|
||||
.status()
|
||||
.await?;
|
||||
async fn cycle_device(&self, state: &AppReceivers, target_type: &str) -> Result<()> {
|
||||
let is_sink = target_type == "sink";
|
||||
let _ = state
|
||||
.audio_cmd_tx
|
||||
.send(AudioCommand::CycleDevice { is_sink })
|
||||
.await;
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
|
||||
@@ -0,0 +1,154 @@
|
||||
use crate::config::Config;
|
||||
use crate::error::Result;
|
||||
use crate::modules::WaybarModule;
|
||||
use crate::output::WaybarOutput;
|
||||
use crate::state::{AppReceivers, BacklightState};
|
||||
use crate::utils::{TokenValue, format_template};
|
||||
use notify::{Config as NotifyConfig, Event, RecommendedWatcher, RecursiveMode, Watcher};
|
||||
use std::path::PathBuf;
|
||||
use std::sync::mpsc;
|
||||
use std::time::Duration;
|
||||
use tokio::sync::watch;
|
||||
use tracing::{error, info};
|
||||
|
||||
pub struct BacklightModule;
|
||||
|
||||
impl WaybarModule for BacklightModule {
|
||||
async fn run(
|
||||
&self,
|
||||
config: &Config,
|
||||
state: &AppReceivers,
|
||||
_args: &[&str],
|
||||
) -> Result<WaybarOutput> {
|
||||
let percentage = state.backlight.borrow().percentage;
|
||||
|
||||
let icon = if percentage < 30 {
|
||||
""
|
||||
} else if percentage < 70 {
|
||||
""
|
||||
} else {
|
||||
""
|
||||
};
|
||||
|
||||
let text = format_template(
|
||||
&config.backlight.format,
|
||||
&[
|
||||
("percentage", TokenValue::Int(percentage as i64)),
|
||||
("icon", TokenValue::String(icon.to_string())),
|
||||
],
|
||||
);
|
||||
|
||||
Ok(WaybarOutput {
|
||||
text,
|
||||
tooltip: Some(format!("Brightness: {}%", percentage)),
|
||||
class: Some("normal".to_string()),
|
||||
percentage: Some(percentage),
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
pub struct BacklightDaemon;
|
||||
|
||||
impl BacklightDaemon {
|
||||
pub fn new() -> Self {
|
||||
Self
|
||||
}
|
||||
|
||||
pub fn start(&self, tx: watch::Sender<BacklightState>) {
|
||||
std::thread::spawn(move || {
|
||||
let base_dir = PathBuf::from("/sys/class/backlight");
|
||||
let mut device_dir = None;
|
||||
|
||||
if let Ok(entries) = std::fs::read_dir(&base_dir) {
|
||||
for entry in entries.flatten() {
|
||||
let path = entry.path();
|
||||
if path.is_dir() {
|
||||
device_dir = Some(path);
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
let Some(dir) = device_dir else {
|
||||
error!("No backlight device found in /sys/class/backlight");
|
||||
return;
|
||||
};
|
||||
|
||||
info!("Monitoring backlight device: {:?}", dir);
|
||||
|
||||
let max_brightness_path = dir.join("max_brightness");
|
||||
let brightness_path = dir.join("actual_brightness");
|
||||
let brightness_path_fallback = dir.join("brightness");
|
||||
|
||||
let target_file = if brightness_path.exists() {
|
||||
brightness_path
|
||||
} else {
|
||||
brightness_path_fallback
|
||||
};
|
||||
|
||||
let get_percentage = || -> u8 {
|
||||
let max: f64 = std::fs::read_to_string(&max_brightness_path)
|
||||
.unwrap_or_default()
|
||||
.trim()
|
||||
.parse()
|
||||
.unwrap_or(100.0);
|
||||
let current: f64 = std::fs::read_to_string(&target_file)
|
||||
.unwrap_or_default()
|
||||
.trim()
|
||||
.parse()
|
||||
.unwrap_or(0.0);
|
||||
|
||||
if max > 0.0 {
|
||||
((current / max) * 100.0).round() as u8
|
||||
} else {
|
||||
0
|
||||
}
|
||||
};
|
||||
|
||||
// Initial poll
|
||||
let _ = tx.send(BacklightState {
|
||||
percentage: get_percentage(),
|
||||
});
|
||||
|
||||
// Set up notify watcher
|
||||
let (ev_tx, ev_rx) = mpsc::channel();
|
||||
let mut watcher = RecommendedWatcher::new(
|
||||
move |res: notify::Result<Event>| {
|
||||
if let Ok(event) = res
|
||||
&& event.kind.is_modify()
|
||||
{
|
||||
let _ = ev_tx.send(());
|
||||
}
|
||||
},
|
||||
NotifyConfig::default(),
|
||||
)
|
||||
.unwrap();
|
||||
|
||||
if let Err(e) = watcher.watch(&target_file, RecursiveMode::NonRecursive) {
|
||||
error!("Failed to watch backlight file: {}", e);
|
||||
return;
|
||||
}
|
||||
|
||||
loop {
|
||||
// Block until an event occurs or a timeout to catch missed events
|
||||
if ev_rx.recv_timeout(Duration::from_secs(5)).is_ok() {
|
||||
// Debounce rapid events
|
||||
std::thread::sleep(Duration::from_millis(50));
|
||||
while ev_rx.try_recv().is_ok() {}
|
||||
|
||||
let _ = tx.send(BacklightState {
|
||||
percentage: get_percentage(),
|
||||
});
|
||||
} else {
|
||||
// Timeout hit, poll just in case
|
||||
let current = get_percentage();
|
||||
if tx.borrow().percentage != current {
|
||||
let _ = tx.send(BacklightState {
|
||||
percentage: current,
|
||||
});
|
||||
}
|
||||
}
|
||||
}
|
||||
});
|
||||
}
|
||||
}
|
||||
+55
-5
@@ -109,6 +109,17 @@ impl BtDaemon {
|
||||
static PLUGINS: LazyLock<Vec<Box<dyn BtPlugin>>> =
|
||||
LazyLock::new(|| vec![Box::new(PixelBudsPlugin)]);
|
||||
|
||||
fn trigger_robust_poll(state: AppReceivers) {
|
||||
tokio::spawn(async move {
|
||||
// Poll immediately and then a few times over the next few seconds
|
||||
// to catch slow state changes in bluez or plugins.
|
||||
for delay in [200, 500, 1000, 2000, 3000] {
|
||||
tokio::time::sleep(std::time::Duration::from_millis(delay)).await;
|
||||
let _ = state.bt_force_poll.try_send(());
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
pub struct BtModule;
|
||||
|
||||
impl WaybarModule for BtModule {
|
||||
@@ -124,19 +135,57 @@ impl WaybarModule for BtModule {
|
||||
let bt_state = state.bluetooth.borrow().clone();
|
||||
|
||||
match action.as_str() {
|
||||
"disconnect" if bt_state.connected => {
|
||||
let _ = tokio::process::Command::new("bluetoothctl")
|
||||
.args(["disconnect", &bt_state.device_address])
|
||||
.output()
|
||||
.await;
|
||||
"connect" => {
|
||||
if let Some(mac) = args.get(1) {
|
||||
if let Ok(session) = bluer::Session::new().await
|
||||
&& let Ok(adapter) = session.default_adapter().await
|
||||
&& let Ok(addr) = mac.parse::<bluer::Address>()
|
||||
&& let Ok(device) = adapter.device(addr)
|
||||
{
|
||||
let _ = device.connect().await;
|
||||
}
|
||||
trigger_robust_poll(state.clone());
|
||||
}
|
||||
return Ok(WaybarOutput::default());
|
||||
}
|
||||
"disconnect" if bt_state.connected => {
|
||||
if let Ok(session) = bluer::Session::new().await
|
||||
&& let Ok(adapter) = session.default_adapter().await
|
||||
&& let Ok(addr) = bt_state.device_address.parse::<bluer::Address>()
|
||||
&& let Ok(device) = adapter.device(addr)
|
||||
{
|
||||
let _ = device.disconnect().await;
|
||||
}
|
||||
trigger_robust_poll(state.clone());
|
||||
return Ok(WaybarOutput::default());
|
||||
}
|
||||
"menu_data" => {
|
||||
let mut devs = Vec::new();
|
||||
if let Ok(session) = bluer::Session::new().await
|
||||
&& let Ok(adapter) = session.default_adapter().await
|
||||
&& let Ok(addresses) = adapter.device_addresses().await
|
||||
{
|
||||
for addr in addresses {
|
||||
if let Ok(device) = adapter.device(addr)
|
||||
&& device.is_paired().await.unwrap_or(false)
|
||||
{
|
||||
let alias = device.alias().await.unwrap_or_else(|_| addr.to_string());
|
||||
devs.push(format!("{} ({})", alias, addr));
|
||||
}
|
||||
}
|
||||
}
|
||||
return Ok(WaybarOutput {
|
||||
text: devs.join("\n"),
|
||||
..Default::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?;
|
||||
trigger_robust_poll(state.clone());
|
||||
}
|
||||
return Ok(WaybarOutput::default());
|
||||
}
|
||||
@@ -161,6 +210,7 @@ impl WaybarModule for BtModule {
|
||||
.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?;
|
||||
trigger_robust_poll(state.clone());
|
||||
}
|
||||
}
|
||||
return Ok(WaybarOutput::default());
|
||||
|
||||
@@ -18,6 +18,13 @@ impl WaybarModule for DiskModule {
|
||||
|
||||
let disks = state.disks.borrow().clone();
|
||||
|
||||
if disks.is_empty() {
|
||||
return Ok(WaybarOutput {
|
||||
text: "Disk Loading...".to_string(),
|
||||
..Default::default()
|
||||
});
|
||||
}
|
||||
|
||||
for disk in &disks {
|
||||
if disk.mount_point == *mountpoint {
|
||||
let total = disk.total_bytes as f64;
|
||||
|
||||
@@ -0,0 +1,104 @@
|
||||
use crate::config::Config;
|
||||
use crate::error::Result;
|
||||
use crate::modules::WaybarModule;
|
||||
use crate::output::WaybarOutput;
|
||||
use crate::state::{AppReceivers, DndState};
|
||||
use futures::StreamExt;
|
||||
use tokio::sync::watch;
|
||||
use tracing::{debug, error, info};
|
||||
use zbus::{Connection, fdo::PropertiesProxy, proxy};
|
||||
|
||||
pub struct DndModule;
|
||||
|
||||
impl WaybarModule for DndModule {
|
||||
async fn run(
|
||||
&self,
|
||||
config: &Config,
|
||||
state: &AppReceivers,
|
||||
_args: &[&str],
|
||||
) -> Result<WaybarOutput> {
|
||||
let is_dnd = state.dnd.borrow().is_dnd;
|
||||
|
||||
if is_dnd {
|
||||
Ok(WaybarOutput {
|
||||
text: config.dnd.format_dnd.clone(),
|
||||
tooltip: Some("Do Not Disturb: On".to_string()),
|
||||
class: Some("dnd".to_string()),
|
||||
percentage: None,
|
||||
})
|
||||
} else {
|
||||
Ok(WaybarOutput {
|
||||
text: config.dnd.format_normal.clone(),
|
||||
tooltip: Some("Do Not Disturb: Off".to_string()),
|
||||
class: Some("normal".to_string()),
|
||||
percentage: None,
|
||||
})
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
pub struct DndDaemon;
|
||||
|
||||
#[proxy(
|
||||
interface = "org.erikreider.swaync.control",
|
||||
default_service = "org.erikreider.swaync.control",
|
||||
default_path = "/org/erikreider/swaync/control"
|
||||
)]
|
||||
trait SwayncControl {
|
||||
#[zbus(property)]
|
||||
fn dnd(&self) -> zbus::Result<bool>;
|
||||
}
|
||||
|
||||
impl DndDaemon {
|
||||
pub fn new() -> Self {
|
||||
Self
|
||||
}
|
||||
|
||||
pub fn start(&self, tx: watch::Sender<DndState>) {
|
||||
tokio::spawn(async move {
|
||||
loop {
|
||||
if let Err(e) = Self::listen_loop(&tx).await {
|
||||
error!("DND listener error: {}", e);
|
||||
tokio::time::sleep(tokio::time::Duration::from_secs(5)).await;
|
||||
}
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
async fn listen_loop(tx: &watch::Sender<DndState>) -> anyhow::Result<()> {
|
||||
let connection = Connection::session().await?;
|
||||
|
||||
info!("Connected to D-Bus for DND monitoring");
|
||||
|
||||
// Try SwayNC first
|
||||
if let Ok(proxy) = SwayncControlProxy::new(&connection).await {
|
||||
debug!("Found SwayNC, using it for DND state.");
|
||||
|
||||
// Get initial state
|
||||
if let Ok(is_dnd) = proxy.dnd().await {
|
||||
let _ = tx.send(DndState { is_dnd });
|
||||
}
|
||||
|
||||
// Monitor properties changed
|
||||
if let Ok(props_proxy) = PropertiesProxy::builder(&connection)
|
||||
.destination("org.erikreider.swaync.control")?
|
||||
.path("/org/erikreider/swaync/control")?
|
||||
.build()
|
||||
.await
|
||||
{
|
||||
let mut stream = props_proxy.receive_properties_changed().await?;
|
||||
while let Some(signal) = stream.next().await {
|
||||
let args = signal.args()?;
|
||||
if args.interface_name == "org.erikreider.swaync.control"
|
||||
&& let Some(val) = args.changed_properties.get("dnd")
|
||||
&& let Ok(is_dnd) = bool::try_from(val)
|
||||
{
|
||||
let _ = tx.send(DndState { is_dnd });
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Err(anyhow::anyhow!("DND stream ended or daemon not found"))
|
||||
}
|
||||
}
|
||||
+1
-5
@@ -3,8 +3,6 @@ use crate::error::Result;
|
||||
use crate::modules::WaybarModule;
|
||||
use crate::output::WaybarOutput;
|
||||
use crate::state::AppReceivers;
|
||||
use anyhow::anyhow;
|
||||
use std::env;
|
||||
use tokio::io::{AsyncReadExt, AsyncWriteExt};
|
||||
use tokio::net::UnixStream;
|
||||
|
||||
@@ -41,9 +39,7 @@ impl WaybarModule for GameModule {
|
||||
}
|
||||
|
||||
async fn hyprland_ipc(cmd: &str) -> Result<String> {
|
||||
let signature = env::var("HYPRLAND_INSTANCE_SIGNATURE")
|
||||
.map_err(|_| anyhow!("HYPRLAND_INSTANCE_SIGNATURE not set"))?;
|
||||
let path = format!("/tmp/hypr/{}/.socket.sock", signature);
|
||||
let path = crate::utils::get_hyprland_socket(".socket.sock")?;
|
||||
|
||||
let mut stream = UnixStream::connect(path).await?;
|
||||
stream.write_all(cmd.as_bytes()).await?;
|
||||
|
||||
@@ -0,0 +1,104 @@
|
||||
use crate::config::Config;
|
||||
use crate::error::Result;
|
||||
use crate::modules::WaybarModule;
|
||||
use crate::output::WaybarOutput;
|
||||
use crate::state::{AppReceivers, KeyboardState};
|
||||
use crate::utils::{TokenValue, format_template};
|
||||
use anyhow::anyhow;
|
||||
use tokio::io::{AsyncBufReadExt, BufReader};
|
||||
use tokio::net::UnixStream;
|
||||
use tokio::sync::watch;
|
||||
use tracing::{error, info};
|
||||
|
||||
pub struct KeyboardModule;
|
||||
|
||||
impl WaybarModule for KeyboardModule {
|
||||
async fn run(
|
||||
&self,
|
||||
config: &Config,
|
||||
state: &AppReceivers,
|
||||
_args: &[&str],
|
||||
) -> Result<WaybarOutput> {
|
||||
let layout = state.keyboard.borrow().layout.clone();
|
||||
|
||||
if layout.is_empty() {
|
||||
return Ok(WaybarOutput {
|
||||
text: "Layout Loading...".to_string(),
|
||||
tooltip: None,
|
||||
class: Some("loading".to_string()),
|
||||
percentage: None,
|
||||
});
|
||||
}
|
||||
|
||||
let text = format_template(
|
||||
&config.keyboard.format,
|
||||
&[("layout", TokenValue::String(layout.clone()))],
|
||||
);
|
||||
|
||||
Ok(WaybarOutput {
|
||||
text,
|
||||
tooltip: Some(format!("Keyboard Layout: {}", layout)),
|
||||
class: Some("normal".to_string()),
|
||||
percentage: None,
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
pub struct KeyboardDaemon;
|
||||
|
||||
impl KeyboardDaemon {
|
||||
pub fn new() -> Self {
|
||||
Self
|
||||
}
|
||||
|
||||
pub fn start(&self, tx: watch::Sender<KeyboardState>) {
|
||||
tokio::spawn(async move {
|
||||
loop {
|
||||
if let Err(e) = Self::listen_loop(&tx).await {
|
||||
error!("Keyboard layout listener error: {}", e);
|
||||
// Fallback to waiting before reconnecting to prevent tight loop
|
||||
tokio::time::sleep(tokio::time::Duration::from_secs(5)).await;
|
||||
}
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
async fn listen_loop(tx: &watch::Sender<KeyboardState>) -> anyhow::Result<()> {
|
||||
let path = crate::utils::get_hyprland_socket(".socket2.sock")?;
|
||||
|
||||
info!("Connecting to Hyprland event socket: {:?}", path);
|
||||
let stream = UnixStream::connect(path).await?;
|
||||
let reader = BufReader::new(stream);
|
||||
let mut lines = reader.lines();
|
||||
|
||||
// Fetch initial layout natively via hyprctl
|
||||
if let Ok(output) = tokio::process::Command::new("hyprctl")
|
||||
.args(["devices", "-j"])
|
||||
.output()
|
||||
.await
|
||||
&& let Ok(json) = serde_json::from_slice::<serde_json::Value>(&output.stdout)
|
||||
&& let Some(keyboards) = json.get("keyboards").and_then(|v| v.as_array())
|
||||
&& let Some(main_kb) = keyboards.last()
|
||||
{
|
||||
// The last active one is usually the main one
|
||||
if let Some(layout) = main_kb.get("active_keymap").and_then(|v| v.as_str()) {
|
||||
let _ = tx.send(KeyboardState {
|
||||
layout: layout.to_string(),
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
while let Ok(Some(line)) = lines.next_line().await {
|
||||
if let Some(payload) = line.strip_prefix("activelayout>>") {
|
||||
// payload format: keyboard_name,layout_name
|
||||
let parts: Vec<&str> = payload.splitn(2, ',').collect();
|
||||
if parts.len() == 2 {
|
||||
let layout = parts[1].to_string();
|
||||
let _ = tx.send(KeyboardState { layout });
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Err(anyhow!("Hyprland socket closed or read error"))
|
||||
}
|
||||
}
|
||||
@@ -1,12 +1,16 @@
|
||||
pub mod audio;
|
||||
pub mod backlight;
|
||||
pub mod bt;
|
||||
pub mod btrfs;
|
||||
pub mod cpu;
|
||||
pub mod disk;
|
||||
pub mod dnd;
|
||||
pub mod game;
|
||||
pub mod gpu;
|
||||
pub mod hardware;
|
||||
pub mod keyboard;
|
||||
pub mod memory;
|
||||
pub mod mpris;
|
||||
pub mod network;
|
||||
pub mod power;
|
||||
pub mod sys;
|
||||
|
||||
@@ -0,0 +1,207 @@
|
||||
use crate::config::Config;
|
||||
use crate::error::Result;
|
||||
use crate::modules::WaybarModule;
|
||||
use crate::output::WaybarOutput;
|
||||
use crate::state::{AppReceivers, MprisState};
|
||||
use crate::utils::{TokenValue, format_template};
|
||||
use tokio::sync::watch;
|
||||
use tracing::{debug, info};
|
||||
use zbus::{Connection, proxy};
|
||||
|
||||
pub struct MprisModule;
|
||||
|
||||
impl WaybarModule for MprisModule {
|
||||
async fn run(
|
||||
&self,
|
||||
config: &Config,
|
||||
state: &AppReceivers,
|
||||
_args: &[&str],
|
||||
) -> Result<WaybarOutput> {
|
||||
let mpris = state.mpris.borrow().clone();
|
||||
|
||||
if mpris.is_stopped && mpris.title.is_empty() {
|
||||
return Ok(WaybarOutput {
|
||||
text: String::new(),
|
||||
tooltip: None,
|
||||
class: Some("stopped".to_string()),
|
||||
percentage: None,
|
||||
});
|
||||
}
|
||||
|
||||
let status_icon = if mpris.is_playing {
|
||||
""
|
||||
} else if mpris.is_paused {
|
||||
""
|
||||
} else {
|
||||
""
|
||||
};
|
||||
|
||||
let class = if mpris.is_playing {
|
||||
"playing"
|
||||
} else if mpris.is_paused {
|
||||
"paused"
|
||||
} else {
|
||||
"stopped"
|
||||
};
|
||||
|
||||
let text = format_template(
|
||||
&config.mpris.format,
|
||||
&[
|
||||
("artist", TokenValue::String(mpris.artist.clone())),
|
||||
("title", TokenValue::String(mpris.title.clone())),
|
||||
("album", TokenValue::String(mpris.album.clone())),
|
||||
("status_icon", TokenValue::String(status_icon.to_string())),
|
||||
],
|
||||
);
|
||||
|
||||
Ok(WaybarOutput {
|
||||
text,
|
||||
tooltip: Some(format!("{} - {}", mpris.artist, mpris.title)),
|
||||
class: Some(class.to_string()),
|
||||
percentage: None,
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
pub struct MprisDaemon;
|
||||
|
||||
#[proxy(
|
||||
interface = "org.freedesktop.DBus",
|
||||
default_service = "org.freedesktop.DBus",
|
||||
default_path = "/org/freedesktop/DBus"
|
||||
)]
|
||||
trait DBus {
|
||||
fn list_names(&self) -> zbus::Result<Vec<String>>;
|
||||
}
|
||||
|
||||
#[proxy(
|
||||
interface = "org.mpris.MediaPlayer2.Player",
|
||||
default_path = "/org/mpris/MediaPlayer2"
|
||||
)]
|
||||
trait MprisPlayer {
|
||||
#[zbus(property)]
|
||||
fn playback_status(&self) -> zbus::Result<String>;
|
||||
|
||||
#[zbus(property)]
|
||||
fn metadata(
|
||||
&self,
|
||||
) -> zbus::Result<std::collections::HashMap<String, zbus::zvariant::Value<'_>>>;
|
||||
}
|
||||
|
||||
impl MprisDaemon {
|
||||
pub fn new() -> Self {
|
||||
Self
|
||||
}
|
||||
|
||||
pub fn start(&self, tx: watch::Sender<MprisState>) {
|
||||
tokio::spawn(async move {
|
||||
loop {
|
||||
if let Err(e) = Self::listen_loop(&tx).await {
|
||||
debug!("MPRIS listener ended or error: {}", e);
|
||||
tokio::time::sleep(tokio::time::Duration::from_secs(5)).await;
|
||||
}
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
async fn listen_loop(tx: &watch::Sender<MprisState>) -> anyhow::Result<()> {
|
||||
let connection = Connection::session().await?;
|
||||
|
||||
info!("Connected to D-Bus for MPRIS monitoring");
|
||||
|
||||
// Periodically poll for the active player and update the MPRIS state.
|
||||
// This avoids complex dynamic signal tracking across ephemeral player instances.
|
||||
|
||||
let dbus_proxy = DBusProxy::new(&connection).await?;
|
||||
|
||||
loop {
|
||||
let names = dbus_proxy.list_names().await?;
|
||||
let mut active_player = None;
|
||||
|
||||
for name in names {
|
||||
if name.starts_with("org.mpris.MediaPlayer2.") {
|
||||
active_player = Some(name);
|
||||
break; // Just grab the first active player for now
|
||||
}
|
||||
}
|
||||
|
||||
if let Some(player_name) = active_player {
|
||||
if let Ok(player_proxy) = MprisPlayerProxy::builder(&connection)
|
||||
.destination(player_name.clone())?
|
||||
.build()
|
||||
.await
|
||||
{
|
||||
let status = player_proxy.playback_status().await.unwrap_or_default();
|
||||
let metadata = player_proxy.metadata().await.unwrap_or_default();
|
||||
|
||||
let is_playing = status == "Playing";
|
||||
let is_paused = status == "Paused";
|
||||
let is_stopped = status == "Stopped";
|
||||
|
||||
let mut artist = String::new();
|
||||
let mut title = String::new();
|
||||
let mut album = String::new();
|
||||
|
||||
if let Some(v) = metadata.get("xesam:artist") {
|
||||
if let Ok(arr) = zbus::zvariant::Array::try_from(v) {
|
||||
let mut artists = Vec::new();
|
||||
for i in 0..arr.len() {
|
||||
if let Ok(Some(s)) = arr.get::<&str>(i) {
|
||||
artists.push(s.to_string());
|
||||
}
|
||||
}
|
||||
artist = artists.join(", ");
|
||||
} else if let Ok(a) = <&str>::try_from(v) {
|
||||
artist = a.to_string();
|
||||
}
|
||||
}
|
||||
if let Some(v) = metadata.get("xesam:title")
|
||||
&& let Ok(t) = <&str>::try_from(v)
|
||||
{
|
||||
title = t.to_string();
|
||||
}
|
||||
if let Some(v) = metadata.get("xesam:album")
|
||||
&& let Ok(a) = <&str>::try_from(v)
|
||||
{
|
||||
album = a.to_string();
|
||||
}
|
||||
|
||||
// Only send if changed
|
||||
let current = tx.borrow();
|
||||
if current.is_playing != is_playing
|
||||
|| current.is_paused != is_paused
|
||||
|| current.is_stopped != is_stopped
|
||||
|| current.title != title
|
||||
|| current.artist != artist
|
||||
|| current.album != album
|
||||
{
|
||||
drop(current); // Drop borrow before send
|
||||
let _ = tx.send(MprisState {
|
||||
is_playing,
|
||||
is_paused,
|
||||
is_stopped,
|
||||
artist,
|
||||
title,
|
||||
album,
|
||||
});
|
||||
}
|
||||
}
|
||||
} else {
|
||||
let current = tx.borrow();
|
||||
if !current.is_stopped || !current.title.is_empty() {
|
||||
drop(current);
|
||||
let _ = tx.send(MprisState {
|
||||
is_playing: false,
|
||||
is_paused: false,
|
||||
is_stopped: true,
|
||||
artist: String::new(),
|
||||
title: String::new(),
|
||||
album: String::new(),
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
tokio::time::sleep(tokio::time::Duration::from_secs(2)).await;
|
||||
}
|
||||
}
|
||||
}
|
||||
+2
-2
@@ -1,6 +1,6 @@
|
||||
use serde::Serialize;
|
||||
use serde::{Deserialize, Serialize};
|
||||
|
||||
#[derive(Serialize, Default)]
|
||||
#[derive(Serialize, Deserialize, Clone, Debug, Default)]
|
||||
pub struct WaybarOutput {
|
||||
pub text: String,
|
||||
#[serde(skip_serializing_if = "Option::is_none")]
|
||||
|
||||
+156
@@ -0,0 +1,156 @@
|
||||
use crate::config::Config;
|
||||
use crate::state::AppReceivers;
|
||||
use nix::libc;
|
||||
use nix::sys::signal::{Signal, kill};
|
||||
use nix::unistd::Pid;
|
||||
use std::collections::HashMap;
|
||||
use std::sync::Arc;
|
||||
use sysinfo::{ProcessesToUpdate, System};
|
||||
use tokio::sync::RwLock;
|
||||
use tokio::time::{Duration, Instant, sleep};
|
||||
use tracing::{debug, warn};
|
||||
|
||||
pub struct WaybarSignaler {
|
||||
cached_pid: Option<Pid>,
|
||||
sys: System,
|
||||
last_signal_sent: HashMap<i32, Instant>,
|
||||
}
|
||||
|
||||
impl WaybarSignaler {
|
||||
pub fn new() -> Self {
|
||||
Self {
|
||||
cached_pid: None,
|
||||
sys: System::new(),
|
||||
last_signal_sent: HashMap::new(),
|
||||
}
|
||||
}
|
||||
|
||||
fn find_waybar_pid(&mut self) -> Option<Pid> {
|
||||
self.sys.refresh_processes(ProcessesToUpdate::All, true);
|
||||
for (pid, process) in self.sys.processes() {
|
||||
if process.name() == "waybar" {
|
||||
return Some(Pid::from_raw(pid.as_u32() as i32));
|
||||
}
|
||||
}
|
||||
None
|
||||
}
|
||||
|
||||
fn send_signal(&mut self, signal_num: i32) {
|
||||
if let Some(last) = self.last_signal_sent.get(&signal_num)
|
||||
&& last.elapsed() < Duration::from_millis(50)
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
let mut valid_pid = false;
|
||||
if let Some(pid) = self.cached_pid
|
||||
&& kill(pid, None).is_ok()
|
||||
{
|
||||
valid_pid = true;
|
||||
}
|
||||
|
||||
if !valid_pid {
|
||||
self.cached_pid = self.find_waybar_pid();
|
||||
}
|
||||
|
||||
if let Some(pid) = self.cached_pid {
|
||||
let rt_sig = match Signal::try_from(libc::SIGRTMIN() + signal_num) {
|
||||
Ok(sig) => sig,
|
||||
Err(_) => {
|
||||
unsafe {
|
||||
if libc::kill(pid.as_raw(), libc::SIGRTMIN() + signal_num) == 0 {
|
||||
debug!("Sent raw SIGRTMIN+{} to waybar (PID: {})", signal_num, pid);
|
||||
self.last_signal_sent.insert(signal_num, Instant::now());
|
||||
} else {
|
||||
warn!("Failed to send raw SIGRTMIN+{} to waybar", signal_num);
|
||||
}
|
||||
}
|
||||
return;
|
||||
}
|
||||
};
|
||||
|
||||
if let Err(e) = kill(pid, rt_sig) {
|
||||
warn!("Failed to signal waybar: {}", e);
|
||||
self.cached_pid = None;
|
||||
} else {
|
||||
debug!("Sent SIGRTMIN+{} to waybar (PID: {})", signal_num, pid);
|
||||
self.last_signal_sent.insert(signal_num, Instant::now());
|
||||
}
|
||||
} else {
|
||||
debug!("Waybar process not found, skipping signal.");
|
||||
}
|
||||
}
|
||||
|
||||
pub async fn run(mut self, config_lock: Arc<RwLock<Config>>, mut receivers: AppReceivers) {
|
||||
let mut last_outputs: HashMap<&'static str, String> = HashMap::new();
|
||||
|
||||
loop {
|
||||
let signals = config_lock.read().await.signals.clone();
|
||||
|
||||
macro_rules! check_and_signal {
|
||||
($module_name:expr, $signal_opt:expr) => {
|
||||
if let Some(sig) = $signal_opt {
|
||||
let config = config_lock.read().await;
|
||||
if let Some(out) = crate::daemon::evaluate_module_for_signaler(
|
||||
$module_name,
|
||||
&receivers,
|
||||
&config,
|
||||
)
|
||||
.await
|
||||
{
|
||||
if last_outputs.get($module_name) != Some(&out) {
|
||||
last_outputs.insert($module_name, out);
|
||||
self.send_signal(sig);
|
||||
}
|
||||
}
|
||||
}
|
||||
};
|
||||
}
|
||||
|
||||
tokio::select! {
|
||||
res = receivers.network.changed(), if signals.network.is_some() => {
|
||||
if res.is_ok() { check_and_signal!("net", signals.network); }
|
||||
}
|
||||
res = receivers.cpu.changed(), if signals.cpu.is_some() => {
|
||||
if res.is_ok() { check_and_signal!("cpu", signals.cpu); }
|
||||
}
|
||||
res = receivers.memory.changed(), if signals.memory.is_some() => {
|
||||
if res.is_ok() { check_and_signal!("mem", signals.memory); }
|
||||
}
|
||||
res = receivers.sys.changed(), if signals.sys.is_some() => {
|
||||
if res.is_ok() { check_and_signal!("sys", signals.sys); }
|
||||
}
|
||||
res = receivers.gpu.changed(), if signals.gpu.is_some() => {
|
||||
if res.is_ok() { check_and_signal!("gpu", signals.gpu); }
|
||||
}
|
||||
res = receivers.disks.changed(), if signals.disk.is_some() => {
|
||||
if res.is_ok() { check_and_signal!("disk", signals.disk); }
|
||||
}
|
||||
res = receivers.bluetooth.changed(), if signals.bt.is_some() => {
|
||||
if res.is_ok() { check_and_signal!("bt", signals.bt); }
|
||||
}
|
||||
res = receivers.audio.changed(), if signals.audio.is_some() => {
|
||||
if res.is_ok() {
|
||||
check_and_signal!("vol", signals.audio);
|
||||
check_and_signal!("mic", signals.audio);
|
||||
}
|
||||
}
|
||||
res = receivers.backlight.changed(), if signals.backlight.is_some() => {
|
||||
if res.is_ok() { check_and_signal!("backlight", signals.backlight); }
|
||||
}
|
||||
res = receivers.keyboard.changed(), if signals.keyboard.is_some() => {
|
||||
if res.is_ok() { check_and_signal!("keyboard", signals.keyboard); }
|
||||
}
|
||||
res = receivers.dnd.changed(), if signals.dnd.is_some() => {
|
||||
if res.is_ok() { check_and_signal!("dnd", signals.dnd); }
|
||||
}
|
||||
res = receivers.mpris.changed(), if signals.mpris.is_some() => {
|
||||
if res.is_ok() { check_and_signal!("mpris", signals.mpris); }
|
||||
}
|
||||
_ = sleep(Duration::from_secs(5)) => {
|
||||
// loop and refresh config
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
+62
-1
@@ -1,6 +1,7 @@
|
||||
use crate::output::WaybarOutput;
|
||||
use std::collections::HashMap;
|
||||
use std::sync::Arc;
|
||||
use tokio::sync::{RwLock, watch};
|
||||
use tokio::sync::{RwLock, mpsc, watch};
|
||||
use tokio::time::Instant;
|
||||
|
||||
#[derive(Clone)]
|
||||
@@ -13,7 +14,13 @@ pub struct AppReceivers {
|
||||
pub disks: watch::Receiver<Vec<DiskInfo>>,
|
||||
pub bluetooth: watch::Receiver<BtState>,
|
||||
pub audio: watch::Receiver<AudioState>,
|
||||
pub mpris: watch::Receiver<MprisState>,
|
||||
pub backlight: watch::Receiver<BacklightState>,
|
||||
pub keyboard: watch::Receiver<KeyboardState>,
|
||||
pub dnd: watch::Receiver<DndState>,
|
||||
pub health: Arc<RwLock<HashMap<String, ModuleHealth>>>,
|
||||
pub bt_force_poll: mpsc::Sender<()>,
|
||||
pub audio_cmd_tx: mpsc::Sender<crate::modules::audio::AudioCommand>,
|
||||
}
|
||||
|
||||
#[derive(Clone, Default)]
|
||||
@@ -21,12 +28,15 @@ pub struct ModuleHealth {
|
||||
pub consecutive_failures: u32,
|
||||
pub last_failure: Option<Instant>,
|
||||
pub backoff_until: Option<Instant>,
|
||||
pub last_successful_output: Option<WaybarOutput>,
|
||||
}
|
||||
|
||||
#[derive(Default, Clone)]
|
||||
pub struct AudioState {
|
||||
pub sink: AudioDeviceInfo,
|
||||
pub source: AudioSourceInfo,
|
||||
pub available_sinks: Vec<String>,
|
||||
pub available_sources: Vec<String>,
|
||||
}
|
||||
|
||||
#[derive(Default, Clone)]
|
||||
@@ -35,6 +45,7 @@ pub struct AudioDeviceInfo {
|
||||
pub description: String,
|
||||
pub volume: u8,
|
||||
pub muted: bool,
|
||||
pub channels: u8,
|
||||
}
|
||||
|
||||
#[derive(Default, Clone)]
|
||||
@@ -43,6 +54,7 @@ pub struct AudioSourceInfo {
|
||||
pub description: String,
|
||||
pub volume: u8,
|
||||
pub muted: bool,
|
||||
pub channels: u8,
|
||||
}
|
||||
|
||||
#[derive(Default, Clone)]
|
||||
@@ -128,6 +140,31 @@ impl Default for GpuState {
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Default, Clone)]
|
||||
pub struct DndState {
|
||||
pub is_dnd: bool,
|
||||
}
|
||||
|
||||
#[derive(Default, Clone)]
|
||||
pub struct KeyboardState {
|
||||
pub layout: String,
|
||||
}
|
||||
|
||||
#[derive(Default, Clone)]
|
||||
pub struct BacklightState {
|
||||
pub percentage: u8,
|
||||
}
|
||||
|
||||
#[derive(Default, Clone)]
|
||||
pub struct MprisState {
|
||||
pub is_playing: bool,
|
||||
pub is_paused: bool,
|
||||
pub is_stopped: bool,
|
||||
pub artist: String,
|
||||
pub title: String,
|
||||
pub album: String,
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
pub struct MockState {
|
||||
pub receivers: AppReceivers,
|
||||
@@ -140,6 +177,10 @@ pub struct MockState {
|
||||
_disks_tx: watch::Sender<Vec<DiskInfo>>,
|
||||
_bt_tx: watch::Sender<BtState>,
|
||||
_audio_tx: watch::Sender<AudioState>,
|
||||
_mpris_tx: watch::Sender<MprisState>,
|
||||
_backlight_tx: watch::Sender<BacklightState>,
|
||||
_keyboard_tx: watch::Sender<KeyboardState>,
|
||||
_dnd_tx: watch::Sender<DndState>,
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
@@ -153,6 +194,10 @@ pub struct AppState {
|
||||
pub disks: Vec<DiskInfo>,
|
||||
pub bluetooth: BtState,
|
||||
pub audio: AudioState,
|
||||
pub mpris: MprisState,
|
||||
pub backlight: BacklightState,
|
||||
pub keyboard: KeyboardState,
|
||||
pub dnd: DndState,
|
||||
pub health: HashMap<String, ModuleHealth>,
|
||||
}
|
||||
|
||||
@@ -166,6 +211,12 @@ pub fn mock_state(state: AppState) -> MockState {
|
||||
let (disks_tx, disks_rx) = watch::channel(state.disks);
|
||||
let (bt_tx, bt_rx) = watch::channel(state.bluetooth);
|
||||
let (audio_tx, audio_rx) = watch::channel(state.audio);
|
||||
let (mpris_tx, mpris_rx) = watch::channel(state.mpris);
|
||||
let (backlight_tx, backlight_rx) = watch::channel(state.backlight);
|
||||
let (keyboard_tx, keyboard_rx) = watch::channel(state.keyboard);
|
||||
let (dnd_tx, dnd_rx) = watch::channel(state.dnd);
|
||||
let (bt_force_tx, _) = mpsc::channel(1);
|
||||
let (audio_cmd_tx, _) = mpsc::channel(1);
|
||||
|
||||
MockState {
|
||||
receivers: AppReceivers {
|
||||
@@ -177,7 +228,13 @@ pub fn mock_state(state: AppState) -> MockState {
|
||||
disks: disks_rx,
|
||||
bluetooth: bt_rx,
|
||||
audio: audio_rx,
|
||||
mpris: mpris_rx,
|
||||
backlight: backlight_rx,
|
||||
keyboard: keyboard_rx,
|
||||
dnd: dnd_rx,
|
||||
health: Arc::new(RwLock::new(state.health)),
|
||||
bt_force_poll: bt_force_tx,
|
||||
audio_cmd_tx,
|
||||
},
|
||||
_net_tx: net_tx,
|
||||
_cpu_tx: cpu_tx,
|
||||
@@ -187,5 +244,9 @@ pub fn mock_state(state: AppState) -> MockState {
|
||||
_disks_tx: disks_tx,
|
||||
_bt_tx: bt_tx,
|
||||
_audio_tx: audio_tx,
|
||||
_mpris_tx: mpris_tx,
|
||||
_backlight_tx: backlight_tx,
|
||||
_keyboard_tx: keyboard_tx,
|
||||
_dnd_tx: dnd_tx,
|
||||
}
|
||||
}
|
||||
|
||||
+35
-1
@@ -3,17 +3,21 @@ use std::io::Write;
|
||||
use std::process::{Command, Stdio};
|
||||
|
||||
pub fn show_menu(prompt: &str, items: &[String], menu_cmd: &str) -> Result<String> {
|
||||
// Backward compatibility for {prompt}, but environment variable is safer
|
||||
let cmd_str = menu_cmd.replace("{prompt}", prompt);
|
||||
let mut child = Command::new("sh")
|
||||
.arg("-c")
|
||||
.arg(&cmd_str)
|
||||
.env("FLUXO_PROMPT", prompt) // Safer shell injection
|
||||
.stdin(Stdio::piped())
|
||||
.stdout(Stdio::piped())
|
||||
.stderr(Stdio::null()) // Suppress GTK/Wayland warnings from tools like wofi
|
||||
.spawn()
|
||||
.context("Failed to spawn menu command")?;
|
||||
|
||||
if let Some(mut stdin) = child.stdin.take() {
|
||||
let input = items.join("\n");
|
||||
let mut input = items.join("\n");
|
||||
input.push('\n'); // Ensure trailing newline for wofi/rofi
|
||||
stdin
|
||||
.write_all(input.as_bytes())
|
||||
.context("Failed to write to menu stdin")?;
|
||||
@@ -33,6 +37,36 @@ pub fn show_menu(prompt: &str, items: &[String], menu_cmd: &str) -> Result<Strin
|
||||
Ok(selected)
|
||||
}
|
||||
|
||||
pub fn get_hyprland_socket(socket_name: &str) -> Result<std::path::PathBuf> {
|
||||
let signature = std::env::var("HYPRLAND_INSTANCE_SIGNATURE")
|
||||
.context("HYPRLAND_INSTANCE_SIGNATURE not set")?;
|
||||
|
||||
// Try XDG_RUNTIME_DIR first (usually /run/user/1000)
|
||||
if let Ok(runtime_dir) = std::env::var("XDG_RUNTIME_DIR") {
|
||||
let path = std::path::PathBuf::from(runtime_dir)
|
||||
.join("hypr")
|
||||
.join(&signature)
|
||||
.join(socket_name);
|
||||
if path.exists() {
|
||||
return Ok(path);
|
||||
}
|
||||
}
|
||||
|
||||
// Fallback to /tmp
|
||||
let path = std::path::PathBuf::from("/tmp/hypr")
|
||||
.join(&signature)
|
||||
.join(socket_name);
|
||||
|
||||
if path.exists() {
|
||||
Ok(path)
|
||||
} else {
|
||||
Err(anyhow::anyhow!(
|
||||
"Hyprland socket {} not found in runtime dir or /tmp",
|
||||
socket_name
|
||||
))
|
||||
}
|
||||
}
|
||||
|
||||
use regex::Regex;
|
||||
use std::sync::LazyLock;
|
||||
|
||||
|
||||
Reference in New Issue
Block a user