47 Commits

Author SHA1 Message Date
nvrl c1152ce1b9 added .deb build target + systemd unit + updated readme
Release / Build and Release (push) Successful in 2m57s
2026-04-07 16:24:44 +02:00
nvrl 230604dae3 changed bin name to fluxo
Release / Build and Release (push) Successful in 2m53s
2026-04-07 12:01:54 +02:00
nvrl 2d35313d75 version bump
Release / Build and Release (push) Successful in 2m49s
2026-04-07 11:43:20 +02:00
nvrl eaeba84092 refactored cli, extracted menu to cli, removed deprecated
Release / Build and Release (push) Has been cancelled
2026-04-07 11:42:57 +02:00
nvrl ffdb689ef9 added missing docs
Release / Build and Release (push) Successful in 23s
2026-04-05 21:12:11 +02:00
nvrl f89833a62e missing docs
Release / Build and Release (push) Successful in 2m54s
2026-04-05 11:56:58 +02:00
nvrl 198986efac cleanup of gpu and config gen 2026-04-05 11:42:08 +02:00
nvrl d1178ff8ba implemented signals for mpris 2026-04-05 11:40:31 +02:00
nvrl 5c5d0148f8 removed callback patterns
Release / Build and Release (push) Successful in 22s
2026-04-05 11:38:14 +02:00
nvrl fdfd54b518 central registry 2026-04-05 11:35:57 +02:00
nvrl 708317a10b daemon decomposition 2026-04-04 18:39:15 +02:00
nvrl f1e601f9ed poll abstraction 2026-04-04 18:37:27 +02:00
nvrl 57906de920 redesign bt module/menu
Release / Build and Release (push) Successful in 2m55s
2026-04-04 14:43:35 +02:00
nvrl 2050c345f1 feature/scroll-animation
Release / Build and Release (push) Successful in 2m50s
2026-04-04 05:11:04 +02:00
nvrl 75601305e2 version bump
Release / Build and Release (push) Successful in 2m47s
2026-04-04 04:25:04 +02:00
nvrl 52cc041c42 clippy 2026-04-04 04:24:35 +02:00
nvrl cb8b641447 implemented 'enabled' 2026-04-04 04:24:09 +02:00
nvrl 071d973fea version bump
Release / Build and Release (push) Successful in 2m53s
2026-04-04 00:18:50 +02:00
nvrl 31072bc645 refactor + feature flags
Release / Build and Release (push) Has been cancelled
2026-04-04 00:18:22 +02:00
nvrl 941942d550 fixed dnd 2026-04-03 23:45:58 +02:00
nvrl 492f0787cb WIP: version bump
Release / Build and Release (push) Failing after 2m43s
2026-04-03 15:16:27 +02:00
nvrl 718d3f99e9 version bump
Release / Build and Release (push) Failing after 22s
2026-04-03 15:15:13 +02:00
nvrl b2e8d5fc1b updated docu 2026-04-03 15:14:49 +02:00
nvrl 2572928195 Merge pull request 'feature/events' (#2) from feature/events into main
Release / Build and Release (push) Successful in 21s
Reviewed-on: #2
2026-04-03 15:02:09 +02:00
nvrl 25ecde96fa clippy
CI / Lint and Test (pull_request) Failing after 1m31s
CI / Version Check (pull_request) Failing after 3s
2026-04-03 15:01:23 +02:00
nvrl 16d12e6dce implemented signal based daemon and dbus events + mpris/dnd/keyboard module 2026-04-03 15:00:58 +02:00
nvrl d28e51de6d version bupm
Release / Build and Release (push) Successful in 2m10s
2026-04-02 18:46:33 +02:00
nvrl b18af49ac8 graceful shutdown
Release / Build and Release (push) Has been cancelled
2026-04-02 18:46:17 +02:00
nvrl 0617e53412 version bump
Release / Build and Release (push) Successful in 2m8s
2026-04-02 18:11:47 +02:00
nvrl bdbd6a8a40 added tokio shared states instead of monolithic state
Release / Build and Release (push) Has been cancelled
2026-04-02 18:11:21 +02:00
nvrl bb3f6e565d Merge pull request 'migrated daemon to tokio with seperate hardware threads + thiserror' (#1) from release/0.3.1 into main
Release / Build and Release (push) Successful in 2m7s
Reviewed-on: #1
2026-04-01 18:16:24 +02:00
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
nvrl f640f116ec added more helper func, cfg validation and testing
Release / Build and Release (push) Successful in 1m4s
2026-03-31 07:54:21 +02:00
nvrl c1b3d9134e clippy fixes 2026-03-30 15:53:03 +02:00
nvrl 78c004bcb4 improved performance 2026-03-30 15:52:23 +02:00
42 changed files with 8184 additions and 1210 deletions
+62
View File
@@ -0,0 +1,62 @@
name: CI
on:
pull_request:
branches:
- main
jobs:
check:
name: Lint and Test
runs-on: ubuntu-latest
steps:
- name: Checkout code
uses: actions/checkout@v4
- name: Install Rust
uses: actions-rust-lang/setup-rust-toolchain@v1
with:
toolchain: stable
components: rustfmt, clippy
cache: false
- name: Install packaging tools
run: apt-get update && apt-get install -y dpkg-dev pkg-config libdbus-1-dev protobuf-compiler libpulse-dev
- name: Check formatting
run: cargo fmt --check
- name: Clippy
run: cargo clippy --tests -- -D warnings
- name: Test
run: cargo test
- name: Build
run: cargo build --release
version-check:
name: Version Check
runs-on: ubuntu-latest
steps:
- name: Checkout repository
uses: actions/checkout@v4
with:
fetch-depth: 0
- name: Compare versions
shell: bash
run: |
NEW_VERSION=$(grep -m1 '^version =' Cargo.toml | cut -d '"' -f 2)
git fetch origin ${{ github.base_ref }}
OLD_VERSION=$(git show origin/${{ github.base_ref }}:Cargo.toml | grep -m1 '^version =' | cut -d '"' -f 2)
echo "Old version (main): $OLD_VERSION"
echo "New version (PR): $NEW_VERSION"
if [ "$NEW_VERSION" = "$OLD_VERSION" ]; then
echo "Error: Cargo.toml version has not been updated in this PR!"
exit 1
else
echo "Success: Version updated from $OLD_VERSION to $NEW_VERSION"
fi
+109
View File
@@ -0,0 +1,109 @@
name: Release
on:
push:
branches:
- main
tags:
- 'v*'
jobs:
build:
name: Build and Release
runs-on: ubuntu-latest
steps:
- name: Checkout code
uses: actions/checkout@v4
- name: Install Rust
uses: actions-rust-lang/setup-rust-toolchain@v1
with:
toolchain: stable
cache: false
- name: Install packaging tools
run: apt-get update && apt-get install -y dpkg-dev pkg-config libdbus-1-dev protobuf-compiler libpulse-dev
- name: Get Version
id: get_version
run: |
VERSION=$(grep -m1 '^version =' Cargo.toml | cut -d '"' -f 2)
echo "VERSION=$VERSION" >> $GITHUB_OUTPUT
if [[ "${{ github.ref }}" == refs/tags/v* ]]; then
echo "TAG=${{ github.ref_name }}" >> $GITHUB_OUTPUT
else
echo "TAG=v$VERSION" >> $GITHUB_OUTPUT
fi
- name: Check if Release Exists
id: check_release
shell: bash
run: |
HTTP_STATUS=$(curl -s -o /dev/null -w "%{http_code}" \
-H "Authorization: token ${{ secrets.GITHUB_TOKEN }}" \
"${{ github.server_url }}/api/v1/repos/${{ github.repository }}/releases/tags/${{ steps.get_version.outputs.TAG }}")
if [ "$HTTP_STATUS" = "200" ]; then
echo "EXISTS=true" >> $GITHUB_OUTPUT
echo "Release already exists for tag ${{ steps.get_version.outputs.TAG }}. Skipping."
else
echo "EXISTS=false" >> $GITHUB_OUTPUT
fi
- name: Build
if: steps.check_release.outputs.EXISTS == 'false'
run: cargo build --release
- name: Test
if: steps.check_release.outputs.EXISTS == 'false'
run: cargo test
- name: Package .deb
if: steps.check_release.outputs.EXISTS == 'false'
env:
VERSION: ${{ steps.get_version.outputs.VERSION }}
TAG: ${{ steps.get_version.outputs.TAG }}
run: |
PKG="fluxo_${VERSION}_amd64"
mkdir -p "${PKG}/DEBIAN"
mkdir -p "${PKG}/usr/bin"
cp target/release/fluxo "${PKG}/usr/bin/"
strip "${PKG}/usr/bin/fluxo"
printf '%s\n' \
"Package: fluxo" \
"Version: ${VERSION}" \
"Section: utils" \
"Priority: optional" \
"Architecture: amd64" \
"Maintainer: fluxo contributors" \
"Description: High-performance daemon/client for Waybar custom modules" \
" fluxo is a compiled Rust daemon that polls system metrics and" \
" serves formatted JSON output to Waybar custom modules over a Unix" \
" domain socket. Replaces shell scripts with a single binary." \
> "${PKG}/DEBIAN/control"
dpkg-deb --build "${PKG}"
mv "${PKG}.deb" "fluxo-${TAG}-amd64.deb"
echo "Built: fluxo-${TAG}-amd64.deb"
- name: Create Release and Upload Assets
if: steps.check_release.outputs.EXISTS == 'false'
uses: https://github.com/softprops/action-gh-release@v1
with:
tag_name: ${{ steps.get_version.outputs.TAG }}
name: Release ${{ steps.get_version.outputs.TAG }}
body: |
Automated release for version ${{ steps.get_version.outputs.VERSION }}
Commit: ${{ github.sha }}
Branch: ${{ github.ref_name }}
files: |
fluxo-${{ steps.get_version.outputs.TAG }}-amd64.deb
target/release/fluxo
draft: false
prerelease: false
env:
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
Generated
+1676 -47
View File
File diff suppressed because it is too large Load Diff
+45 -4
View File
@@ -1,16 +1,57 @@
[package]
name = "fluxo-rs"
version = "0.1.0"
version = "0.5.4"
edition = "2024"
[[bin]]
name = "fluxo"
path = "src/main.rs"
[features]
default = ["mod-audio", "mod-bt", "mod-network", "mod-hardware", "mod-dbus"]
mod-audio = ["dep:libpulse-binding"]
mod-bt = ["dep:bluer", "dep:maestro"]
mod-network = ["dep:nix"]
mod-hardware = []
mod-dbus = ["dep:zbus", "dep:futures"]
[dependencies]
anyhow = "1.0.102"
clap = { version = "4.6.0", features = ["derive"] }
fs4 = "0.13.1"
regex = "1.10"
ctrlc = "3"
libc = "0.2"
regex = "1.12"
serde = { version = "1.0.228", features = ["derive"] }
serde_json = "1.0.149"
sysinfo = "0.38.4"
toml = "1.0.6"
thiserror = "2.0"
toml = "1.1.2"
tracing = "0.1.44"
tracing-subscriber = { version = "0.3.23", features = ["env-filter"] }
tokio = { version = "1", features = ["rt", "rt-multi-thread", "sync", "time", "macros", "signal", "process", "io-util", "net"] }
tokio-util = { version = "0.7", features = ["codec", "time"] }
notify = "8.2.0"
# Optional module dependencies
maestro = { git = "https://github.com/qzed/pbpctrl", package = "maestro", optional = true }
bluer = { version = "0.17", features = ["bluetoothd", "rfcomm", "id"], optional = true }
futures = { version = "0.3", optional = true }
libpulse-binding = { version = "2.30", optional = true }
nix = { version = "0.31", features = ["net"], optional = true }
zbus = { version = "5", optional = true }
[dev-dependencies]
tempfile = "3"
[package.metadata.deb]
maintainer = "Nils Pukropp"
copyright = "2024-2026 Nils Pukropp"
depends = "$auto"
section = "utils"
priority = "optional"
assets = [
["target/release/fluxo", "usr/bin/", "755"],
["dist/fluxo.service", "usr/lib/systemd/user/", "644"],
["README.md", "usr/share/doc/fluxo/", "644"],
["example.config.toml", "usr/share/doc/fluxo/", "644"],
]
+115 -74
View File
@@ -1,96 +1,137 @@
# fluxo-rs
# fluxo
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` 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.
## Description
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.
The project follows a client-server architecture:
- **Daemon**: Handles heavy lifting (polling cpu, memory, network, gpu) and stores state in memory.
- **Client**: A thin cli wrapper that connects to the daemon's socket to retrieve formatted json for waybar.
## Key Features
This approach eliminates process spawning overhead and temporary file locking, resulting in near-zero cpu usage for custom modules.
## Features
- **Ultra-lightweight**: Background polling is highly optimized (e.g., O(1) process counting).
- **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.
- **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` 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
| Command | Description | Tokens |
| :--- | :--- | :--- |
| `net` | Network speed (rx/tx) | `{interface}`, `{ip}`, `{rx}`, `{tx}` |
| `cpu` | CPU usage and temp | `{usage}`, `{temp}` |
| `cpu` | CPU usage and temperature | `{usage}`, `{temp}`, `{model}` |
| `mem` | Memory usage | `{used}`, `{total}` |
| `gpu` | GPU utilization | `{usage}`, `{vram_used}`, `{vram_total}`, `{temp}` |
| `sys` | System load and uptime | `{uptime}`, `{load1}`, `{load5}`, `{load15}` |
| `disk` | Disk usage (default: /) | `{mount}`, `{used}`, `{total}` |
| `pool` | Aggregate storage (btrfs) | `{used}`, `{total}` |
| `vol` | Audio output volume | `{name}`, `{volume}`, `{icon}` |
| `mic` | Audio input volume | `{name}`, `{volume}`, `{icon}` |
| `bt` | Bluetooth status | `{alias}` |
| `buds` | Pixel Buds Pro control | `{left}`, `{right}`, `{anc}` |
| `net` | Network status & speeds | `{interface}`, `{ip}`, `{rx}`, `{tx}` |
| `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 gamemode status | active/inactive icon strings |
| `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 |
## Installation
### From Source
```bash
cargo build --release
cp target/release/fluxo ~/.cargo/bin/
```
### Debian/Ubuntu (.deb)
```bash
cargo install cargo-deb
cargo deb
sudo dpkg -i target/debian/fluxo-rs_*.deb
```
The `.deb` package installs the binary to `/usr/bin/fluxo`, the systemd user service to `/usr/lib/systemd/user/fluxo.service`, and documentation to `/usr/share/doc/fluxo/`.
## Setup
1. Build the project:
```bash
cd fluxo-rs
cargo build --release
```
1. **Configure**: Create `~/.config/fluxo/config.toml` (see `example.config.toml`). Ensure you map your `[signals]`.
2. **Start the daemon** via systemd (recommended) or manually:
2. Start the daemon:
```bash
# Starts the daemon using the default config path (~/.config/fluxo/config.toml)
./target/release/fluxo-rs daemon &
# Or provide a custom path
./target/release/fluxo-rs daemon --config /path/to/your/config.toml &
```
### systemd (recommended)
3. Configuration:
Create `~/.config/fluxo/config.toml` (see `example.config.toml` for all default options).
If installed from the `.deb`, the service file is already in place. For manual installs:
4. Waybar configuration (`config.jsonc`):
```json
"custom/cpu": {
"exec": "~/path/to/fluxo-rs cpu",
"return-type": "json"
}
```
## Development
### Architecture
- `src/main.rs`: Entry point, CLI parsing, interactive GUI spawns (menus), and client-side formatting logic.
- `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
mkdir -p ~/.config/systemd/user
cp dist/fluxo.service ~/.config/systemd/user/
```
### Logging & 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:
If your binary is not at `~/.cargo/bin/fluxo`, edit the `ExecStart=` path in the service file.
Then enable and start:
```bash
RUST_LOG=debug fluxo-rs daemon
systemctl --user daemon-reload
systemctl --user enable --now fluxo
```
Check status:
```bash
systemctl --user status fluxo
journalctl --user -u fluxo -f
```
### Manual
```bash
fluxo daemon
```
## Waybar Configuration
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/volume": {
"exec": "fluxo vol",
"return-type": "json",
"interval": 0,
"signal": 8, // Must match the value in config.toml [signals]
"on-click": "fluxo vol mute",
"on-scroll-up": "fluxo vol up 1",
"on-scroll-down": "fluxo vol down 1",
"on-click-right": "fluxo vol cycle"
},
"custom/bluetooth-audio": {
"format": "{}",
"return-type": "json",
"exec": "fluxo bt",
"on-click": "fluxo bt menu",
"on-click-right": "fluxo bt cycle_mode",
"signal": 9,
"interval": 0,
"tooltip": true
}
```
## Debugging
Use `--loglevel` to control log verbosity (trace, debug, info, warn, error):
```bash
fluxo daemon --loglevel debug
```
Or via the `RUST_LOG` environment variable:
```bash
RUST_LOG=debug fluxo daemon
```
For module help and available arguments:
```bash
fluxo help # overview of all modules
fluxo help vol # detailed help for a specific module
```
This will print verbose information regarding config parsing errors, subprocess failures, and IPC requests.
+116
View File
@@ -0,0 +1,116 @@
# 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]
# 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]
# enabled = false # set to false to disable this module at runtime
# tokens: {interface}, {ip}, {rx}, {tx}
format = "{interface} ({ip}):  {rx:^4.1} MB/s  {tx:^4.1} MB/s"
[cpu]
# enabled = false # set to false to disable this module at runtime
# tokens: {usage}, {temp}, {model}
format = "CPU: {usage:^4.1}% {temp:^4.1}C"
[memory]
# enabled = false # set to false to disable this module at runtime
# tokens: {used}, {total}
format = "MEM: {used:^4.1}/{total:^4.1}GB"
[gpu]
# enabled = false # set to false to disable this module at runtime
# 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]
# enabled = false # set to false to disable this module at runtime
# tokens: {uptime}, {load1}, {load5}, {load15}, {procs}
format = "UP: {uptime} LOAD: {load1:^3.1} "
[disk]
# enabled = false # set to false to disable this module at runtime
# tokens: {mount}, {used}, {total}
format = "{mount} {used:^3.0}/{total:^3.0}G"
[pool]
# enabled = false # set to false to disable this module at runtime
# tokens: {used}, {total}
format = "{used:>4.0}G / {total:>4.0}G"
[power]
# enabled = false # set to false to disable this module at runtime
# tokens: {percentage}, {icon}
format = "{percentage:>3}% {icon}"
[audio]
# enabled = false # set to false to disable this module at runtime
# 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]
# enabled = false # set to false to disable this module at runtime
# tokens: {alias}, {mac}, {left}, {right}, {anc}
format_plugin = "{alias} [{left}|{right}] {anc} 󰂰"
format_connected = "󰂰 {alias}"
format_disconnected = "󰂯 Disconnected"
format_disabled = "󰂲 Off"
[game]
# enabled = false # set to false to disable this module at runtime
format_active = "<span size='large'>󰊖</span>"
format_inactive = "<span size='large'></span>"
[mpris]
# enabled = false # set to false to disable this module at runtime
# max_length = 30 # truncate text beyond this character length (adds '...')
# scroll = true # enable marquee scroll animation (requires max_length)
# scroll_speed = 500 # ms between scroll steps (only while playing)
# scroll_separator = " /// " # separator shown between loops when scrolling
# tokens: {artist}, {title}, {album}, {status_icon}
format = "{status_icon} {artist} - {title}"
[backlight]
# enabled = false # set to false to disable this module at runtime
# tokens: {percentage}, {icon}
format = "{percentage:>3}% {icon}"
[keyboard]
# enabled = false # set to false to disable this module at runtime
# tokens: {layout}
format = "{layout}"
[dnd]
# enabled = false # set to false to disable this module at runtime
format_dnd = "<span size='large'>󰂛</span>"
format_normal = "<span size='large'>󰂚</span>"
+135
View File
@@ -0,0 +1,135 @@
//! Bluetooth interactive menu (client-side).
//!
//! Runs entirely in the client process because it needs to spawn the user's
//! menu command (rofi/dmenu/wofi) — the daemon has no business opening GUI
//! windows. Communicates with the daemon via IPC to fetch device lists and
//! dispatch connect/disconnect/mode commands.
/// Format strings used both when *building* menu items and when *parsing*
/// the user's selection back. Keeping them together prevents drift.
mod fmt {
/// Connected device with a plugin mode: `"<alias>: Mode: <mode> [<mac>]"`.
pub const MODE_INFIX: &str = ": Mode: ";
/// Disconnect action: `"Disconnect <alias> [<mac>]"`.
pub const DISCONNECT_PREFIX: &str = "Disconnect ";
/// Visual separator before paired-but-not-connected devices.
pub const CONNECT_HEADER: &str = "--- Connect Device ---";
}
/// Extract a MAC address enclosed in `[…]` at the end of a string.
fn parse_mac_from_brackets(s: &str) -> Option<&str> {
let start = s.rfind('[')?;
let end = s.rfind(']')?;
if end > start + 1 {
Some(&s[start + 1..end])
} else {
None
}
}
/// Extract a MAC address enclosed in `(…)` at the end of a string.
fn parse_mac_from_parens(s: &str) -> Option<&str> {
let start = s.rfind('(')?;
let end = s.rfind(')')?;
if end > start + 1 {
Some(&s[start + 1..end])
} else {
None
}
}
/// Parse a mode selection line: `"<alias>: Mode: <mode> [<mac>]"`.
/// Returns `(mode, mac)`.
fn parse_mode_selection(s: &str) -> Option<(&str, &str)> {
let mac = parse_mac_from_brackets(s)?;
let mode_start = s.find(fmt::MODE_INFIX)?;
let mode_begin = mode_start + fmt::MODE_INFIX.len();
let bracket_start = s.rfind('[')?;
if bracket_start > mode_begin {
let mode = s[mode_begin..bracket_start].trim_end();
Some((mode, mac))
} else {
None
}
}
/// Run the interactive Bluetooth device menu.
///
/// Fetches connected/paired devices from the daemon, presents them in the
/// user's configured menu command, and dispatches the selected action back
/// to the daemon.
pub fn run_bt_menu() {
let config = crate::config::load_config(None);
let mut items = Vec::new();
let mut connected: Vec<(String, String)> = Vec::new();
let mut paired: Vec<(String, String)> = Vec::new();
// Fetch the device list from the daemon.
if let Ok(json_str) = crate::ipc::request_data("bt", &["menu_data"])
&& let Ok(val) = serde_json::from_str::<serde_json::Value>(&json_str)
&& let Some(text) = val.get("text").and_then(|t| t.as_str())
{
for line in text.lines() {
if let Some(rest) = line.strip_prefix("CONNECTED:")
&& let Some((alias, mac)) = rest.split_once('|')
{
connected.push((alias.to_string(), mac.to_string()));
} else if let Some(rest) = line.strip_prefix("PAIRED:")
&& let Some((alias, mac)) = rest.split_once('|')
{
paired.push((alias.to_string(), mac.to_string()));
}
}
}
// Build menu items for connected devices (modes + disconnect).
for (alias, mac) in &connected {
if let Ok(json_str) = crate::ipc::request_data("bt", &["get_modes", mac])
&& 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!("{}{}{} [{}]", alias, fmt::MODE_INFIX, mode, mac));
}
}
items.push(format!("{}{} [{}]", fmt::DISCONNECT_PREFIX, alias, mac));
}
// Paired-but-not-connected devices go below a separator.
if !paired.is_empty() {
items.push(fmt::CONNECT_HEADER.to_string());
for (alias, mac) in &paired {
items.push(format!("{} ({})", alias, mac));
}
}
if items.is_empty() {
tracing::info!("No Bluetooth options found.");
return;
}
let Ok(selected) = crate::utils::show_menu("BT Menu: ", &items, &config.general.menu_command)
else {
return;
};
if let Some((mode, mac)) = parse_mode_selection(&selected) {
crate::output::print_waybar_response(crate::ipc::request_data(
"bt",
&["set_mode", mode, mac],
));
} else if selected.starts_with(fmt::DISCONNECT_PREFIX) {
if let Some(mac) = parse_mac_from_brackets(&selected) {
crate::output::print_waybar_response(crate::ipc::request_data(
"bt",
&["disconnect", mac],
));
}
} else if selected == fmt::CONNECT_HEADER {
// Section header — no action.
} else if let Some(mac) = parse_mac_from_parens(&selected) {
crate::output::print_waybar_response(crate::ipc::request_data("bt", &["connect", mac]));
}
}
+44
View File
@@ -0,0 +1,44 @@
//! Client-side module command dispatch.
//!
//! Resolves CLI aliases (e.g. `mic` → audio source), delegates to
//! special-case handlers (BT menu), and falls through to the standard
//! IPC → daemon → Waybar JSON path for everything else.
/// Resolve client-side module aliases that prepend implicit arguments.
///
/// `vol` maps to the audio sink, `mic` maps to the audio source — both
/// dispatch to the `"vol"` module on the daemon with a `"sink"` or
/// `"source"` prefix argument.
fn resolve_alias(module: &str, args: &[String]) -> (String, Vec<String>) {
match module {
"vol" => {
let mut a = vec!["sink".to_string()];
a.extend(args.iter().cloned());
("vol".to_string(), a)
}
"mic" => {
let mut a = vec!["source".to_string()];
a.extend(args.iter().cloned());
("vol".to_string(), a)
}
_ => (module.to_string(), args.to_vec()),
}
}
/// Entry point for all `fluxo <module> [args...]` invocations.
///
/// Handles the BT menu special case client-side, resolves aliases, and
/// sends the request to the daemon via IPC.
pub fn run_module_command(module: &str, args: &[String]) {
// Bluetooth menu runs client-side because it spawns the user's menu
// command (rofi/dmenu/wofi) — the daemon must not open GUI windows.
#[cfg(feature = "mod-bt")]
if module == "bt" && args.first().map(|s| s.as_str()) == Some("menu") {
crate::bt_menu::run_bt_menu();
return;
}
let (actual_module, actual_args) = resolve_alias(module, args);
let args_ref: Vec<&str> = actual_args.iter().map(|s| s.as_str()).collect();
crate::output::print_waybar_response(crate::ipc::request_data(&actual_module, &args_ref));
}
+431 -46
View File
@@ -1,92 +1,163 @@
//! Configuration loading + top-level `Config` struct.
//!
//! Every module section `#[derive(Deserialize)]`s its own struct with `serde`
//! defaults, so missing TOML sections/fields simply fall back to baked-in
//! values. The live `Config` is held behind an `Arc<RwLock<_>>` by the daemon
//! and replaced atomically on reload (see [`crate::daemon::reload_config`]).
use serde::Deserialize;
use std::fs;
use std::path::PathBuf;
use tracing::{debug, info, warn};
#[derive(Deserialize, Default)]
/// Top-level parsed `config.toml`. Each field corresponds to a `[section]`.
#[derive(Deserialize, Default, Clone)]
pub struct Config {
#[serde(default)]
pub general: GeneralConfig,
#[serde(default)]
pub signals: SignalsConfig,
#[cfg(feature = "mod-network")]
#[serde(default)]
pub network: NetworkConfig,
#[cfg(feature = "mod-hardware")]
#[serde(default)]
pub cpu: CpuConfig,
#[cfg(feature = "mod-hardware")]
#[serde(default)]
pub memory: MemoryConfig,
#[cfg(feature = "mod-hardware")]
#[serde(default)]
pub gpu: GpuConfig,
#[cfg(feature = "mod-hardware")]
#[serde(default)]
pub sys: SysConfig,
#[cfg(feature = "mod-hardware")]
#[serde(default)]
pub disk: DiskConfig,
#[cfg(feature = "mod-hardware")]
#[serde(default)]
pub pool: PoolConfig,
#[cfg(feature = "mod-hardware")]
#[serde(default)]
pub power: PowerConfig,
#[serde(default)]
pub buds: BudsConfig,
#[cfg(feature = "mod-audio")]
#[serde(default)]
pub audio: AudioConfig,
#[cfg(feature = "mod-bt")]
#[serde(default)]
pub bt: BtConfig,
#[cfg(feature = "mod-hardware")]
#[serde(default)]
pub game: GameConfig,
#[cfg(feature = "mod-dbus")]
#[serde(default)]
pub mpris: MprisConfig,
#[cfg(feature = "mod-dbus")]
#[serde(default)]
pub backlight: BacklightConfig,
#[cfg(feature = "mod-dbus")]
#[serde(default)]
pub keyboard: KeyboardConfig,
#[cfg(feature = "mod-dbus")]
#[serde(default)]
pub dnd: DndConfig,
}
#[derive(Deserialize)]
/// Process-wide settings that aren't tied to a single module.
#[derive(Deserialize, Clone)]
pub struct GeneralConfig {
/// Shell command spawned by [`crate::utils::show_menu`] for interactive pickers.
pub menu_command: String,
}
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(),
}
}
}
#[derive(Deserialize)]
/// Which Waybar `SIGRTMIN+N` signal each module should trigger, if any.
///
/// `None` disables signalling for that module entirely.
#[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>,
}
/// Network module config. `format` tokens: `interface`, `ip`, `rx`, `tx`.
#[derive(Deserialize, Clone)]
pub struct NetworkConfig {
#[serde(default = "default_true")]
pub enabled: bool,
pub format: String,
}
impl Default for NetworkConfig {
fn default() -> Self {
Self {
enabled: true,
format: "{interface} ({ip}):  {rx:>5.2} MB/s  {tx:>5.2} MB/s".to_string(),
}
}
}
#[derive(Deserialize)]
/// CPU module config. `format` tokens: `usage`, `temp`.
#[derive(Deserialize, Clone)]
pub struct CpuConfig {
#[serde(default = "default_true")]
pub enabled: bool,
pub format: String,
}
impl Default for CpuConfig {
fn default() -> Self {
Self {
enabled: true,
format: "CPU: {usage:>4.1}% {temp:>4.1}C".to_string(),
}
}
}
#[derive(Deserialize)]
/// Memory module config. `format` tokens: `used`, `total` (gigabytes).
#[derive(Deserialize, Clone)]
pub struct MemoryConfig {
#[serde(default = "default_true")]
pub enabled: bool,
pub format: String,
}
impl Default for MemoryConfig {
fn default() -> Self {
Self {
enabled: true,
format: "{used:>5.2}/{total:>5.2}GB".to_string(),
}
}
}
#[derive(Deserialize)]
/// GPU module config with per-vendor format strings.
#[derive(Deserialize, Clone)]
pub struct GpuConfig {
#[serde(default = "default_true")]
pub enabled: bool,
pub format_amd: String,
pub format_intel: String,
pub format_nvidia: String,
@@ -95,84 +166,89 @@ pub struct GpuConfig {
impl Default for GpuConfig {
fn default() -> Self {
Self {
format_amd: "AMD: {usage:>3.0}% {vram_used:>4.1}/{vram_total:>4.1}GB {temp:>4.1}C".to_string(),
enabled: true,
format_amd: "AMD: {usage:>3.0}% {vram_used:>4.1}/{vram_total:>4.1}GB {temp:>4.1}C"
.to_string(),
format_intel: "iGPU: {usage:>3.0}%".to_string(),
format_nvidia: "NV: {usage:>3.0}% {vram_used:>4.1}/{vram_total:>4.1}GB {temp:>4.1}C".to_string(),
format_nvidia: "NV: {usage:>3.0}% {vram_used:>4.1}/{vram_total:>4.1}GB {temp:>4.1}C"
.to_string(),
}
}
}
#[derive(Deserialize)]
/// Sys module config. `format` tokens: `uptime`, `load1`, `load5`, `load15`, `procs`.
#[derive(Deserialize, Clone)]
pub struct SysConfig {
#[serde(default = "default_true")]
pub enabled: bool,
pub format: String,
}
impl Default for SysConfig {
fn default() -> Self {
Self {
enabled: true,
format: "UP: {uptime} | LOAD: {load1:>4.2} {load5:>4.2} {load15:>4.2}".to_string(),
}
}
}
#[derive(Deserialize)]
/// Disk module config. `format` tokens: `mount`, `used`, `total`.
#[derive(Deserialize, Clone)]
pub struct DiskConfig {
#[serde(default = "default_true")]
pub enabled: bool,
pub format: String,
}
impl Default for DiskConfig {
fn default() -> Self {
Self {
enabled: true,
format: "{mount} {used:>5.1}/{total:>5.1}G".to_string(),
}
}
}
#[derive(Deserialize)]
/// Btrfs pool module config. `format` tokens: `used`, `total`.
#[derive(Deserialize, Clone)]
pub struct PoolConfig {
#[serde(default = "default_true")]
pub enabled: bool,
pub format: String,
}
impl Default for PoolConfig {
fn default() -> Self {
Self {
enabled: true,
format: "{used:>4.0}G / {total:>4.0}G".to_string(),
}
}
}
#[derive(Deserialize)]
/// Battery/power module config. `format` tokens: `percentage`, `icon`.
#[derive(Deserialize, Clone)]
pub struct PowerConfig {
#[serde(default = "default_true")]
pub enabled: bool,
pub format: String,
}
impl Default for PowerConfig {
fn default() -> Self {
Self {
enabled: true,
format: "{percentage:>3}% {icon}".to_string(),
}
}
}
#[derive(Deserialize)]
pub struct BudsConfig {
pub mac: String,
pub format: String,
pub format_disconnected: String,
}
impl Default for BudsConfig {
fn default() -> Self {
Self {
mac: "B4:23:A2:09:D3:53".to_string(),
format: "{left} | {right} | {anc}".to_string(),
format_disconnected: "<span size='large'></span>".to_string(),
}
}
}
#[derive(Deserialize)]
/// Audio module config, one format per (sink|source) × (muted|unmuted) state.
#[derive(Deserialize, Clone)]
pub struct AudioConfig {
#[serde(default = "default_true")]
pub enabled: bool,
pub format_sink_unmuted: String,
pub format_sink_muted: String,
pub format_source_unmuted: String,
@@ -182,6 +258,7 @@ pub struct AudioConfig {
impl Default for AudioConfig {
fn default() -> Self {
Self {
enabled: true,
format_sink_unmuted: "{name} {volume:>3}% {icon}".to_string(),
format_sink_muted: "{name} {icon}".to_string(),
format_source_unmuted: "{name} {volume:>3}% {icon}".to_string(),
@@ -190,9 +267,14 @@ impl Default for AudioConfig {
}
}
#[derive(Deserialize)]
/// Bluetooth module config. Plugin line tokens: `alias`, `left`, `right`,
/// `anc`, `mac`. Plain connect line tokens: `alias`.
#[derive(Deserialize, Clone)]
pub struct BtConfig {
#[serde(default = "default_true")]
pub enabled: bool,
pub format_connected: String,
pub format_plugin: String,
pub format_disconnected: String,
pub format_disabled: String,
}
@@ -200,15 +282,20 @@ pub struct BtConfig {
impl Default for BtConfig {
fn default() -> Self {
Self {
enabled: true,
format_connected: "{alias} 󰂰".to_string(),
format_plugin: "{alias} [{left}|{right}] {anc} 󰂰".to_string(),
format_disconnected: "󰂯".to_string(),
format_disabled: "󰂲 Off".to_string(),
}
}
}
#[derive(Deserialize)]
/// Gamemode indicator config (active/inactive glyphs).
#[derive(Deserialize, Clone)]
pub struct GameConfig {
#[serde(default = "default_true")]
pub enabled: bool,
pub format_active: String,
pub format_inactive: String,
}
@@ -216,27 +303,253 @@ pub struct GameConfig {
impl Default for GameConfig {
fn default() -> Self {
Self {
enabled: true,
format_active: "<span size='large'>󰊖</span>".to_string(),
format_inactive: "<span size='large'></span>".to_string(),
}
}
}
/// MPRIS module config. `format` tokens: `artist`, `title`, `album`,
/// `status_icon`. Optional marquee scrolling when `scroll = true`.
#[derive(Deserialize, Clone)]
pub struct MprisConfig {
#[serde(default = "default_true")]
pub enabled: bool,
pub format: String,
#[serde(default)]
pub max_length: Option<usize>,
#[serde(default)]
pub scroll: bool,
#[serde(default = "default_scroll_speed")]
pub scroll_speed: u64,
#[serde(default = "default_scroll_separator")]
pub scroll_separator: String,
}
fn default_scroll_speed() -> u64 {
500
}
fn default_scroll_separator() -> String {
" /// ".to_string()
}
impl Default for MprisConfig {
fn default() -> Self {
Self {
enabled: true,
format: "{status_icon} {artist} - {title}".to_string(),
max_length: None,
scroll: false,
scroll_speed: 500,
scroll_separator: " /// ".to_string(),
}
}
}
/// Backlight module config. `format` tokens: `percentage`, `icon`.
#[derive(Deserialize, Clone)]
pub struct BacklightConfig {
#[serde(default = "default_true")]
pub enabled: bool,
pub format: String,
}
impl Default for BacklightConfig {
fn default() -> Self {
Self {
enabled: true,
format: "{percentage:>3}% {icon}".to_string(),
}
}
}
/// Keyboard layout module config. `format` tokens: `layout`.
#[derive(Deserialize, Clone)]
pub struct KeyboardConfig {
#[serde(default = "default_true")]
pub enabled: bool,
pub format: String,
}
impl Default for KeyboardConfig {
fn default() -> Self {
Self {
enabled: true,
format: "{layout}".to_string(),
}
}
}
/// Do-Not-Disturb indicator config (dnd/normal glyphs).
#[derive(Deserialize, Clone)]
pub struct DndConfig {
#[serde(default = "default_true")]
pub enabled: bool,
pub format_dnd: String,
pub format_normal: String,
}
fn default_true() -> bool {
true
}
impl Default for DndConfig {
fn default() -> Self {
Self {
enabled: true,
format_dnd: "<span size='large'>󰂛</span>".to_string(),
format_normal: "<span size='large'>󰂚</span>".to_string(),
}
}
}
fn extract_tokens(format_str: &str) -> Vec<String> {
crate::utils::TOKEN_RE
.captures_iter(format_str)
.map(|cap| cap[1].to_string())
.collect()
}
fn validate_format(label: &str, format_str: &str, known_tokens: &[&str]) {
for token in extract_tokens(format_str) {
if !known_tokens.contains(&token.as_str()) {
warn!(
"Config [{}]: unknown token '{{{}}}' in format string. Known tokens: {:?}",
label, token, known_tokens
);
}
}
}
impl Config {
/// Check if a module is enabled in the configuration.
/// Returns false if the module is explicitly disabled; true if enabled or unknown.
pub fn is_module_enabled(&self, module_name: &str) -> bool {
macro_rules! gen_enabled_match {
($( { $feature:literal, $field:ident, $state:ty, [$($name:literal),+], [$($sig_name:literal),+], $module:path, $signal:ident, [$($default_arg:literal),*], $config:ident } )*) => {
match module_name {
$(
#[cfg(feature = $feature)]
$($name)|+ => self.$config.enabled,
)*
// Dispatch-only modules (no watch channel)
#[cfg(feature = "mod-audio")]
"mic" => self.audio.enabled,
#[cfg(feature = "mod-hardware")]
"pool" | "btrfs" => self.pool.enabled,
#[cfg(feature = "mod-hardware")]
"power" => self.power.enabled,
#[cfg(feature = "mod-hardware")]
"game" => self.game.enabled,
_ => true,
}
};
}
for_each_watched_module!(gen_enabled_match)
}
/// Warn-log any `{token}` placeholders used in format strings that the
/// corresponding module does not know how to fill in.
pub fn validate(&self) {
#[cfg(feature = "mod-network")]
validate_format(
"network",
&self.network.format,
&["interface", "ip", "rx", "tx"],
);
#[cfg(feature = "mod-hardware")]
{
validate_format("cpu", &self.cpu.format, &["usage", "temp"]);
validate_format("memory", &self.memory.format, &["used", "total"]);
validate_format(
"gpu.amd",
&self.gpu.format_amd,
&["usage", "vram_used", "vram_total", "temp"],
);
validate_format("gpu.intel", &self.gpu.format_intel, &["usage", "freq"]);
validate_format(
"gpu.nvidia",
&self.gpu.format_nvidia,
&["usage", "vram_used", "vram_total", "temp"],
);
validate_format(
"sys",
&self.sys.format,
&["uptime", "load1", "load5", "load15", "procs"],
);
validate_format("disk", &self.disk.format, &["mount", "used", "total"]);
validate_format("pool", &self.pool.format, &["used", "total"]);
validate_format("power", &self.power.format, &["percentage", "icon"]);
}
#[cfg(feature = "mod-audio")]
{
validate_format(
"audio.sink_unmuted",
&self.audio.format_sink_unmuted,
&["name", "icon", "volume"],
);
validate_format(
"audio.sink_muted",
&self.audio.format_sink_muted,
&["name", "icon"],
);
validate_format(
"audio.source_unmuted",
&self.audio.format_source_unmuted,
&["name", "icon", "volume"],
);
validate_format(
"audio.source_muted",
&self.audio.format_source_muted,
&["name", "icon"],
);
}
#[cfg(feature = "mod-bt")]
{
validate_format("bt.connected", &self.bt.format_connected, &["alias"]);
validate_format(
"bt.plugin",
&self.bt.format_plugin,
&["alias", "left", "right", "anc", "mac"],
);
}
#[cfg(feature = "mod-dbus")]
{
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"]);
}
}
}
/// Resolve the default config path: `$XDG_CONFIG_HOME/fluxo/config.toml`
/// (or `~/.config/fluxo/config.toml`).
pub fn default_config_path() -> PathBuf {
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")
}
/// Load and validate the config, falling back to [`Config::default`] when
/// the file is missing or malformed. Never panics.
pub fn load_config(custom_path: Option<PathBuf>) -> Config {
let config_path = custom_path.unwrap_or_else(|| {
let config_dir = std::env::var("XDG_CONFIG_HOME")
.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")
});
let config_path = custom_path.unwrap_or_else(default_config_path);
if let Ok(content) = fs::read_to_string(&config_path) {
match toml::from_str(&content) {
match toml::from_str::<Config>(&content) {
Ok(cfg) => {
info!("Successfully loaded configuration from {:?}", config_path);
cfg.validate();
cfg
}
Err(e) => {
@@ -246,7 +559,79 @@ pub fn load_config(custom_path: Option<PathBuf>) -> Config {
}
}
} else {
debug!("No config file found at {:?}, using default settings.", config_path);
debug!(
"No config file found at {:?}, using default settings.",
config_path
);
Config::default()
}
}
#[cfg(test)]
mod tests {
use super::*;
use std::io::Write;
#[test]
fn test_default_config() {
let config = Config::default();
assert_eq!(
config.general.menu_command,
"fuzzel --dmenu --prompt \"$FLUXO_PROMPT\""
);
assert!(config.cpu.format.contains("usage"));
assert!(config.cpu.format.contains("temp"));
assert!(config.memory.format.contains("used"));
assert!(config.memory.format.contains("total"));
}
#[test]
fn test_load_missing_config() {
let config = load_config(Some(PathBuf::from("/nonexistent/config.toml")));
// Should fallback to defaults without panicking
assert_eq!(
config.general.menu_command,
"fuzzel --dmenu --prompt \"$FLUXO_PROMPT\""
);
}
#[test]
fn test_load_valid_partial_config() {
let mut tmpfile = tempfile::NamedTempFile::new().unwrap();
// In TOML, braces have no special meaning in strings
writeln!(tmpfile, "[cpu]").unwrap();
writeln!(tmpfile, "format = \"custom: {{usage}}\"").unwrap();
let config = load_config(Some(tmpfile.path().to_path_buf()));
// TOML treats {{ as literal {{ (no escape), so the value is "custom: {{usage}}"
assert!(config.cpu.format.contains("usage"));
// Other sections still have defaults
assert!(config.memory.format.contains("used"));
}
#[test]
fn test_load_invalid_toml() {
let mut tmpfile = tempfile::NamedTempFile::new().unwrap();
write!(tmpfile, "this is not valid toml {{{{").unwrap();
let config = load_config(Some(tmpfile.path().to_path_buf()));
// Should fallback to defaults
assert_eq!(
config.general.menu_command,
"fuzzel --dmenu --prompt \"$FLUXO_PROMPT\""
);
}
#[test]
fn test_load_empty_config() {
let tmpfile = tempfile::NamedTempFile::new().unwrap();
// Empty file is valid TOML, all sections default
let config = load_config(Some(tmpfile.path().to_path_buf()));
assert_eq!(
config.general.menu_command,
"fuzzel --dmenu --prompt \"$FLUXO_PROMPT\""
);
assert!(config.cpu.format.contains("usage"));
}
}
+511 -106
View File
@@ -1,132 +1,537 @@
use crate::config::Config;
use crate::ipc::SOCKET_PATH;
use crate::modules::network::NetworkDaemon;
use crate::modules::hardware::HardwareDaemon;
use crate::modules::WaybarModule;
use crate::state::{AppState, SharedState};
use anyhow::Result;
use std::fs;
use std::io::{BufRead, BufReader, Write};
use std::os::unix::net::UnixListener;
use std::sync::{Arc, RwLock};
use std::thread;
use std::time::Duration;
use tracing::{info, warn, error, debug};
use std::path::PathBuf;
//! Daemon entry point: orchestrates polling tasks, signal handling, config
//! hot-reloading, and the IPC server.
//!
//! Layout of [`run_daemon`]:
//!
//! 1. **Channels** — `watch::channel()` pairs for every module that pushes
//! state from a background task.
//! 2. **Polling / event tasks** — one per module; each writes into its sender,
//! the signaler and request handlers read from the matching receiver.
//! 3. **Config watchers** — filesystem notifier + `SIGHUP` handler refresh the
//! [`Config`] in place so modules see updates immediately.
//! 4. **Signaler** — watches all state receivers and pokes Waybar.
//! 5. **IPC loop** — `UnixListener` accepting client requests; each connection
//! dispatches to [`crate::registry::dispatch`] and returns JSON.
pub fn run_daemon(config_path: Option<PathBuf>) -> Result<()> {
if fs::metadata(SOCKET_PATH).is_ok() {
debug!("Removing stale socket file: {}", SOCKET_PATH);
fs::remove_file(SOCKET_PATH)?;
use crate::config::Config;
use crate::ipc::socket_path;
#[cfg(feature = "mod-audio")]
use crate::modules::audio::AudioDaemon;
#[cfg(feature = "mod-dbus")]
use crate::modules::backlight::BacklightDaemon;
#[cfg(feature = "mod-bt")]
use crate::modules::bt::BtDaemon;
#[cfg(feature = "mod-dbus")]
use crate::modules::dnd::DndDaemon;
#[cfg(feature = "mod-hardware")]
use crate::modules::hardware::HardwareDaemon;
#[cfg(feature = "mod-dbus")]
use crate::modules::keyboard::KeyboardDaemon;
#[cfg(feature = "mod-dbus")]
use crate::modules::mpris::MprisDaemon;
#[cfg(feature = "mod-network")]
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, mpsc, watch};
use tokio::time::{Duration, sleep};
use tokio_util::sync::CancellationToken;
use tracing::{debug, error, info};
/// Spawn a health-tracked polling loop.
///
/// Each iteration: skip if in backoff, else await `$poll_expr` and feed the
/// `Result` to [`crate::health::handle_poll_result`]. The loop breaks when
/// `$token` is cancelled.
macro_rules! spawn_poll_loop {
($name:expr, $interval:expr, $health:expr, $token:expr, $poll_expr:expr) => {
tokio::spawn(async move {
info!(concat!("Starting ", $name, " polling task"));
loop {
if !crate::health::is_poll_in_backoff($name, &$health).await {
let res: crate::error::Result<()> = $poll_expr.await;
crate::health::handle_poll_result($name, res, &$health).await;
}
tokio::select! {
_ = $token.cancelled() => break,
_ = sleep($interval) => {}
}
}
info!(concat!($name, " task shut down."));
})
};
}
/// Spawn a health-tracked polling loop with an extra trigger channel.
///
/// Identical to [`spawn_poll_loop`] but `$trigger` can wake the loop early —
/// used by the Bluetooth daemon when a client forces an immediate refresh.
macro_rules! spawn_poll_loop_triggered {
($name:expr, $interval:expr, $health:expr, $token:expr, $trigger:expr, $poll_expr:expr) => {
tokio::spawn(async move {
info!(concat!("Starting ", $name, " polling task"));
loop {
if !crate::health::is_poll_in_backoff($name, &$health).await {
let res: crate::error::Result<()> = $poll_expr.await;
crate::health::handle_poll_result($name, res, &$health).await;
}
tokio::select! {
_ = $token.cancelled() => break,
_ = $trigger.recv() => {},
_ = sleep($interval) => {}
}
}
info!(concat!($name, " task shut down."));
})
};
}
/// Spawn a polling loop with no health tracking.
///
/// Used for internal daemons (hardware fast/slow) whose poll functions are
/// infallible and whose failures don't drive client-visible backoff.
macro_rules! spawn_poll_loop_simple {
($name:expr, $interval:expr, $token:expr, $poll_expr:expr) => {
tokio::spawn(async move {
info!(concat!("Starting ", $name, " polling task"));
loop {
$poll_expr.await;
tokio::select! {
_ = $token.cancelled() => break,
_ = sleep($interval) => {}
}
}
info!(concat!($name, " task shut down."));
})
};
}
struct SocketGuard {
path: String,
}
impl Drop for SocketGuard {
fn drop(&mut self) {
debug!("Cleaning up socket file: {}", self.path);
let _ = fs::remove_file(&self.path);
}
}
fn get_config_path(custom_path: Option<PathBuf>) -> PathBuf {
custom_path.unwrap_or_else(crate::config::default_config_path)
}
/// Run the daemon to completion.
///
/// Sets up the socket, spawns all enabled module tasks, hooks up config
/// hot-reloading, and finally enters the IPC accept loop. Returns only on
/// a fatal error or `Ctrl+C`.
pub async fn run_daemon(config_path: Option<PathBuf>) -> Result<()> {
let sock_path = socket_path();
if fs::metadata(&sock_path).is_ok() {
debug!("Removing stale socket file: {}", sock_path);
fs::remove_file(&sock_path)?;
}
let state: SharedState = Arc::new(RwLock::new(AppState::default()));
let listener = UnixListener::bind(SOCKET_PATH)?;
// We store the original config_path to allow proper reloading later
let config_path_clone = config_path.clone();
let config = Arc::new(RwLock::new(crate::config::load_config(config_path)));
#[cfg(feature = "mod-network")]
let (net_tx, net_rx) = watch::channel(Default::default());
#[cfg(feature = "mod-hardware")]
let (cpu_tx, cpu_rx) = watch::channel(Default::default());
#[cfg(feature = "mod-hardware")]
let (mem_tx, mem_rx) = watch::channel(Default::default());
#[cfg(feature = "mod-hardware")]
let (sys_tx, sys_rx) = watch::channel(Default::default());
#[cfg(feature = "mod-hardware")]
let (gpu_tx, gpu_rx) = watch::channel(Default::default());
#[cfg(feature = "mod-hardware")]
let (disks_tx, disks_rx) = watch::channel(Default::default());
#[cfg(feature = "mod-bt")]
let (bt_tx, bt_rx) = watch::channel(Default::default());
#[cfg(feature = "mod-audio")]
let (audio_tx, audio_rx) = watch::channel(Default::default());
#[cfg(feature = "mod-dbus")]
let (mpris_tx, mpris_rx) = watch::channel(Default::default());
#[cfg(feature = "mod-dbus")]
let (backlight_tx, backlight_rx) = watch::channel(Default::default());
#[cfg(feature = "mod-dbus")]
let (keyboard_tx, keyboard_rx) = watch::channel(Default::default());
#[cfg(feature = "mod-dbus")]
let (dnd_tx, dnd_rx) = watch::channel(Default::default());
#[cfg(feature = "mod-dbus")]
let mpris_scroll = Arc::new(RwLock::new(crate::state::MprisScrollState::default()));
#[cfg(feature = "mod-dbus")]
let (mpris_scroll_tick_tx, mpris_scroll_tick_rx) = watch::channel(0u64);
let health = Arc::new(RwLock::new(HashMap::new()));
#[cfg(feature = "mod-bt")]
let (bt_force_tx, mut bt_force_rx) = mpsc::channel(1);
#[cfg(feature = "mod-bt")]
let bt_cycle = Arc::new(RwLock::new(0usize));
#[cfg(feature = "mod-audio")]
let (audio_cmd_tx, audio_cmd_rx) = mpsc::channel(8);
let receivers = AppReceivers {
#[cfg(feature = "mod-network")]
network: net_rx,
#[cfg(feature = "mod-hardware")]
cpu: cpu_rx,
#[cfg(feature = "mod-hardware")]
memory: mem_rx,
#[cfg(feature = "mod-hardware")]
sys: sys_rx,
#[cfg(feature = "mod-hardware")]
gpu: gpu_rx,
#[cfg(feature = "mod-hardware")]
disks: disks_rx,
#[cfg(feature = "mod-bt")]
bluetooth: bt_rx,
#[cfg(feature = "mod-bt")]
bt_cycle,
#[cfg(feature = "mod-audio")]
audio: audio_rx,
#[cfg(feature = "mod-dbus")]
mpris: mpris_rx,
#[cfg(feature = "mod-dbus")]
backlight: backlight_rx,
#[cfg(feature = "mod-dbus")]
keyboard: keyboard_rx,
#[cfg(feature = "mod-dbus")]
dnd: dnd_rx,
#[cfg(feature = "mod-dbus")]
mpris_scroll: Arc::clone(&mpris_scroll),
#[cfg(feature = "mod-dbus")]
mpris_scroll_tick: mpris_scroll_tick_rx,
health: Arc::clone(&health),
#[cfg(feature = "mod-bt")]
bt_force_poll: bt_force_tx,
#[cfg(feature = "mod-audio")]
audio_cmd_tx,
};
let listener = UnixListener::bind(&sock_path)?;
let _guard = SocketGuard {
path: sock_path.clone(),
};
// Ctrl+C triggers a graceful shutdown by cancelling this token; every
// spawned polling task checks it in its `select!`.
let cancel_token = CancellationToken::new();
let token_clone = cancel_token.clone();
tokio::spawn(async move {
tokio::signal::ctrl_c().await.unwrap();
info!("Received shutdown signal, exiting...");
token_clone.cancel();
});
let resolved_config_path = get_config_path(config_path.clone());
let config = Arc::new(RwLock::new(crate::config::load_config(config_path.clone())));
spawn_config_watchers(&config, &resolved_config_path);
#[cfg(feature = "mod-network")]
if config.read().await.network.enabled {
let mut daemon = NetworkDaemon::new();
let token = cancel_token.clone();
let h = Arc::clone(&health);
spawn_poll_loop!(
"net",
Duration::from_secs(1),
h,
token,
daemon.poll(&net_tx)
);
}
// Fast-cycle hardware (cpu/mem/load) polled at 1 Hz.
#[cfg(feature = "mod-hardware")]
{
let cfg = config.read().await;
let fast_enabled = cfg.cpu.enabled || cfg.memory.enabled || cfg.sys.enabled;
drop(cfg);
if fast_enabled {
let mut daemon = HardwareDaemon::new();
let token = cancel_token.clone();
spawn_poll_loop_simple!(
"fast_hw",
Duration::from_secs(1),
token,
daemon.poll_fast(&cpu_tx, &mem_tx, &sys_tx)
);
}
}
// Slow-cycle hardware (gpu/disks) polled every 5 s — expensive to sample.
#[cfg(feature = "mod-hardware")]
{
let cfg = config.read().await;
let slow_enabled = cfg.gpu.enabled || cfg.disk.enabled;
drop(cfg);
if slow_enabled {
let mut daemon = HardwareDaemon::new();
let token = cancel_token.clone();
spawn_poll_loop_simple!(
"slow_hw",
Duration::from_secs(5),
token,
daemon.poll_slow(&gpu_tx, &disks_tx)
);
}
}
#[cfg(feature = "mod-bt")]
if config.read().await.bt.enabled {
let mut daemon = BtDaemon::new();
let token = cancel_token.clone();
let h = Arc::clone(&health);
let poll_config = Arc::clone(&config);
let poll_receivers = receivers.clone();
spawn_poll_loop_triggered!("bt", Duration::from_secs(2), h, token, bt_force_rx, async {
let config = poll_config.read().await;
daemon.poll(&bt_tx, &poll_receivers, &config).await;
Ok(())
});
}
// Event-driven subsystems — these spawn their own threads internally and
// push into their watch sender as events arrive (no polling loop).
#[cfg(feature = "mod-audio")]
if config.read().await.audio.enabled {
let audio_daemon = AudioDaemon::new();
audio_daemon.start(&audio_tx, audio_cmd_rx);
}
#[cfg(feature = "mod-dbus")]
if config.read().await.backlight.enabled {
let backlight_daemon = BacklightDaemon::new();
backlight_daemon.start(backlight_tx);
}
#[cfg(feature = "mod-dbus")]
if config.read().await.keyboard.enabled {
let keyboard_daemon = KeyboardDaemon::new();
keyboard_daemon.start(keyboard_tx);
}
#[cfg(feature = "mod-dbus")]
if config.read().await.dnd.enabled {
let dnd_daemon = DndDaemon::new();
dnd_daemon.start(dnd_tx);
}
#[cfg(feature = "mod-dbus")]
if config.read().await.mpris.enabled {
let mpris_daemon = MprisDaemon::new();
mpris_daemon.start(mpris_tx);
// Ticks the scroll offset forward for the marquee animation.
let scroll_config = Arc::clone(&config);
let scroll_rx = receivers.mpris.clone();
let scroll_state = Arc::clone(&mpris_scroll);
tokio::spawn(async move {
crate::modules::mpris::mpris_scroll_ticker(
scroll_config,
scroll_rx,
scroll_state,
mpris_scroll_tick_tx,
)
.await;
});
}
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);
run_ipc_loop(listener, receivers, config, config_path, cancel_token).await
}
/// Spawn background tasks that hot-reload the daemon's [`Config`].
///
/// Installs a `notify`-based filesystem watcher on the config file's parent
/// directory, plus a `SIGHUP` handler — either triggers a reload of the
/// shared `Arc<RwLock<Config>>`.
fn spawn_config_watchers(config: &Arc<RwLock<Config>>, resolved_path: &std::path::Path) {
// `notify` recursively tracks the parent dir so atomic-write editors
// (which rename a new file into place) still get picked up.
let watcher_config = Arc::clone(config);
let watcher_path = resolved_path.to_path_buf();
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);
let poll_state = Arc::clone(&state);
thread::spawn(move || {
info!("Starting background polling thread");
let mut network_daemon = NetworkDaemon::new();
let mut hardware_daemon = HardwareDaemon::new();
loop {
network_daemon.poll(Arc::clone(&poll_state));
hardware_daemon.poll(Arc::clone(&poll_state));
thread::sleep(Duration::from_secs(1));
tokio::select! {
_ = ev_rx.recv() => {
// Coalesce rapid editor writes into one reload.
sleep(Duration::from_millis(100)).await;
while ev_rx.try_recv().is_ok() {}
reload_config(&watcher_config, Some(watcher_path.clone())).await;
}
}
}
});
info!("Fluxo daemon successfully bound to socket: {}", SOCKET_PATH);
let hup_config = Arc::clone(config);
let hup_path = resolved_path.to_path_buf();
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;
}
});
}
for stream in listener.incoming() {
match stream {
Ok(mut stream) => {
let state_clone = Arc::clone(&state);
let config_clone = Arc::clone(&config);
let cp_clone = config_path_clone.clone();
thread::spawn(move || {
let mut reader = BufReader::new(stream.try_clone().unwrap());
let mut request = String::new();
if let Err(e) = reader.read_line(&mut request) {
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.");
/// Accept loop for the client Unix socket.
///
/// Each client request spawns a short-lived task that reads one line, looks
/// up the module via [`crate::registry::dispatch`], and writes the JSON
/// response back. Broken-pipe errors are logged at `debug` — they just mean
/// the client timed out before we responded.
async fn run_ipc_loop(
listener: UnixListener,
receivers: AppReceivers,
config: Arc<RwLock<Config>>,
config_path: Option<PathBuf>,
cancel_token: CancellationToken,
) -> Result<()> {
loop {
tokio::select! {
_ = cancel_token.cancelled() => break,
res = listener.accept() => {
match res {
Ok((mut stream, _)) => {
let state_clone = receivers.clone();
let config_clone = Arc::clone(&config);
let cp_clone = config_path.clone();
tokio::spawn(async move {
let (reader, mut writer) = stream.split();
let mut reader = BufReader::new(reader);
let mut request = String::new();
if let Err(e) = reader.read_line(&mut request).await {
error!("Failed to read from IPC stream: {}", e);
return;
}
return;
}
debug!(module = module_name, args = ?&parts[1..], "Handling IPC request");
let response = handle_request(*module_name, &parts[1..], &state_clone, &config_clone);
if let Err(e) = stream.write_all(response.as_bytes()) {
error!("Failed to write IPC response: {}", e);
}
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" {
reload_config(&config_clone, cp_clone).await;
let _ = writer.write_all(b"{\"text\":\"ok\"}").await;
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(e) => error!("Failed to accept incoming connection: {}", e),
}
}
info!("Daemon shutting down gracefully.");
Ok(())
}
fn handle_request(module_name: &str, args: &[&str], state: &SharedState, config_lock: &Arc<RwLock<Config>>) -> String {
let config = if let Ok(c) = config_lock.read() {
c
} else {
error!("Failed to acquire read lock for configuration.");
return "{\"text\":\"error: config lock failed\"}".to_string();
};
/// Re-read the configuration file and swap it into the shared lock.
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.");
}
let result = match module_name {
"net" | "network" => crate::modules::network::NetworkModule.run(&config, state, args),
"cpu" => crate::modules::cpu::CpuModule.run(&config, state, args),
"mem" | "memory" => crate::modules::memory::MemoryModule.run(&config, state, args),
"disk" => crate::modules::disk::DiskModule.run(&config, state, args),
"pool" | "btrfs" => crate::modules::btrfs::BtrfsModule.run(&config, state, args),
"vol" => crate::modules::audio::AudioModule.run(&config, state, &["sink", args.get(0).unwrap_or(&"show")]),
"mic" => crate::modules::audio::AudioModule.run(&config, state, &["source", args.get(0).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))
},
};
/// Evaluate a module with its signaler-default args and return the JSON body.
///
/// Used by the [`crate::signaler`] to decide whether the module's output has
/// actually changed before sending Waybar a signal.
pub async fn evaluate_module_for_signaler(
module_name: &str,
state: &AppReceivers,
config: &Config,
) -> Option<String> {
let args = crate::registry::signaler_default_args(module_name);
crate::registry::dispatch(module_name, config, state, args)
.await
.ok()
.and_then(|out| serde_json::to_string(&out).ok())
}
async fn handle_request(
module_name: &str,
args: &[&str],
state: &AppReceivers,
config_lock: &Arc<RwLock<Config>>,
) -> String {
let (is_in_backoff, cached_output) = crate::health::check_backoff(module_name, state).await;
if is_in_backoff {
return crate::health::backoff_response(module_name, cached_output);
}
let config = config_lock.read().await;
let result = crate::registry::dispatch(module_name, &config, state, args).await;
crate::health::update_health(module_name, &result, state).await;
match result {
Ok(output) => serde_json::to_string(&output).unwrap_or_else(|_| "{}".to_string()),
Err(e) => {
error!(module = module_name, error = %e, "Module execution failed");
let err_out = crate::output::WaybarOutput {
text: format!("\u{200B}Error\u{200B}"),
tooltip: Some(e.to_string()),
class: Some("error".to_string()),
percentage: None,
};
serde_json::to_string(&err_out).unwrap_or_else(|_| "{}".to_string())
Err(crate::error::FluxoError::Disabled(_)) => {
serde_json::to_string(&crate::output::WaybarOutput::disabled())
.unwrap_or_else(|_| "{}".to_string())
}
Err(e) => crate::health::error_response(module_name, &e, cached_output),
}
}
+67
View File
@@ -0,0 +1,67 @@
//! Error types used across the crate.
use thiserror::Error;
/// Canonical error type for all fluxo subsystems.
///
/// Errors are categorised so that [`FluxoError::is_transient`] can distinguish
/// temporary runtime failures (retried with backoff) from permanent ones.
#[derive(Error, Debug)]
#[allow(dead_code)]
pub enum FluxoError {
#[error("Configuration error: {0}")]
Config(String),
#[error("Module error ({module}): {message}")]
Module {
module: &'static str,
message: String,
},
#[error("Daemon IPC error: {0}")]
Ipc(String),
#[error("External system error: {0}")]
System(String),
#[error("Bluetooth error: {0}")]
Bluetooth(String),
#[error("Network error: {0}")]
Network(String),
#[error("Hardware error: {0}")]
Hardware(String),
#[error("IO error: {0}")]
Io(#[from] std::io::Error),
#[error("Serialization error: {0}")]
Serialization(#[from] serde_json::Error),
#[error("Module disabled: {0}")]
Disabled(String),
#[error("Other error: {0}")]
Other(#[from] anyhow::Error),
}
impl FluxoError {
/// Returns `true` for errors that represent likely-transient failures
/// (IO, external systems, hardware) and should trigger exponential backoff
/// rather than permanent cooldown.
pub fn is_transient(&self) -> bool {
matches!(
self,
Self::Io(_)
| Self::System(_)
| Self::Bluetooth(_)
| Self::Network(_)
| Self::Hardware(_)
| Self::Module { .. }
)
}
}
/// Crate-wide `Result` alias using [`FluxoError`].
pub type Result<T> = std::result::Result<T, FluxoError>;
+144
View File
@@ -0,0 +1,144 @@
//! Per-module health tracking and exponential backoff.
//!
//! Both the IPC request handler (for on-demand module evaluation) and the
//! background polling loops consult this module. Transient errors increment a
//! failure counter that grows the cooldown window; permanent errors trigger an
//! immediate long cooldown.
use crate::output::WaybarOutput;
use crate::state::{AppReceivers, ModuleHealth};
use std::collections::HashMap;
use std::sync::Arc;
use tokio::sync::RwLock;
use tokio::time::{Duration, Instant};
use tracing::{debug, error, info, warn};
/// Check if a module is in backoff (used by request handler).
pub async fn check_backoff(
module_name: &str,
state: &AppReceivers,
) -> (bool, Option<WaybarOutput>) {
let lock = state.health.read().await;
if let Some(health) = lock.get(module_name) {
let in_backoff = health
.backoff_until
.is_some_and(|until| Instant::now() < until);
(in_backoff, health.last_successful_output.clone())
} else {
(false, None)
}
}
/// Update health after a request dispatch (used by request handler).
pub async fn update_health(
module_name: &str,
result: &Result<WaybarOutput, crate::error::FluxoError>,
state: &AppReceivers,
) {
let mut lock = state.health.write().await;
let health = lock.entry(module_name.to_string()).or_default();
match result {
Ok(output) => {
health.consecutive_failures = 0;
health.backoff_until = None;
health.last_successful_output = Some(output.clone());
}
Err(crate::error::FluxoError::Disabled(_)) => {}
Err(e) => {
health.consecutive_failures += 1;
health.last_failure = Some(Instant::now());
if health.consecutive_failures >= 3 {
health.backoff_until = Some(Instant::now() + Duration::from_secs(30));
warn!(module = module_name, error = %e, "Module entered backoff state due to repeated failures");
}
}
}
}
/// Check if a polling daemon module is in backoff.
pub async fn is_poll_in_backoff(
module_name: &str,
health_lock: &Arc<RwLock<HashMap<String, ModuleHealth>>>,
) -> bool {
let lock = health_lock.read().await;
if let Some(health) = lock.get(module_name)
&& let Some(until) = health.backoff_until
{
return Instant::now() < until;
}
false
}
/// Update health after a polling daemon result.
pub async fn handle_poll_result(
module_name: &str,
result: crate::error::Result<()>,
health_lock: &Arc<RwLock<HashMap<String, ModuleHealth>>>,
) {
let mut lock = health_lock.write().await;
let health = lock.entry(module_name.to_string()).or_default();
match result {
Ok(_) => {
if health.consecutive_failures > 0 {
info!(
module = module_name,
"Module recovered after {} failures", health.consecutive_failures
);
}
health.consecutive_failures = 0;
health.backoff_until = None;
}
Err(e) => {
health.consecutive_failures += 1;
health.last_failure = Some(Instant::now());
if !e.is_transient() {
health.backoff_until = Some(Instant::now() + Duration::from_secs(60));
error!(module = module_name, error = %e, "Fatal module error, entering long cooldown");
} else if health.consecutive_failures >= 3 {
let backoff_secs = 30 * (2u64.pow(health.consecutive_failures.saturating_sub(3)));
let backoff_secs = backoff_secs.min(3600);
health.backoff_until = Some(Instant::now() + Duration::from_secs(backoff_secs));
warn!(module = module_name, error = %e, backoff = backoff_secs, "Repeated transient failures, entering backoff");
} else {
debug!(module = module_name, error = %e, "Transient module failure (attempt {})", health.consecutive_failures);
}
}
}
}
/// Serialise a response to return while a module is cooling down.
///
/// If a cached successful output exists, it is returned with a `warning` CSS
/// class appended; otherwise a generic "Cooling down" placeholder is emitted.
pub fn backoff_response(module_name: &str, cached: Option<WaybarOutput>) -> String {
if let Some(mut cached) = cached {
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 zws = crate::output::ZERO_WIDTH_SPACE;
format!("{{\"text\":\"{zws}Cooling down ({module_name}){zws}\",\"class\":\"error\"}}")
}
/// Serialise a fallback response for a module that errored this request.
///
/// Prefers showing the last successful cached output (with a `warning` class)
/// over a bare error text, to keep the bar visually stable.
pub fn error_response(
module_name: &str,
e: &crate::error::FluxoError,
cached: Option<WaybarOutput>,
) -> String {
if let Some(mut cached) = cached {
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 = WaybarOutput::error(&error_msg);
serde_json::to_string(&err_out).unwrap_or_else(|_| "{}".to_string())
}
+499
View File
@@ -0,0 +1,499 @@
//! Human-readable help output for all available modules.
//!
//! `fluxo help` prints an overview of every module with its aliases, arguments,
//! and format tokens. `fluxo help <module>` shows the detailed page for a
//! single module.
/// Module help descriptor used to build the help output.
struct ModuleHelp {
/// Primary display name.
name: &'static str,
/// CLI aliases that dispatch to this module.
aliases: &'static [&'static str],
/// Cargo feature gate required at compile time.
feature: &'static str,
/// One-line summary of what the module does.
summary: &'static str,
/// Argument synopsis in `[arg]` notation.
args_synopsis: &'static str,
/// Detailed argument descriptions.
args_detail: &'static [(&'static str, &'static str)],
/// Format tokens available in `config.toml`.
tokens: &'static [(&'static str, &'static str)],
/// Concrete usage examples.
examples: &'static [(&'static str, &'static str)],
}
/// All module descriptors, ordered by category.
const MODULES: &[ModuleHelp] = &[
// ── Hardware ─────────────────────────────────────────────────────
ModuleHelp {
name: "cpu",
aliases: &["cpu"],
feature: "mod-hardware",
summary: "CPU usage percentage and temperature.",
args_synopsis: "",
args_detail: &[],
tokens: &[
("usage", "CPU usage as a percentage (0.0 - 100.0)"),
("temp", "CPU temperature in degrees Celsius"),
],
examples: &[("fluxo cpu", "Show current CPU usage and temperature")],
},
ModuleHelp {
name: "memory",
aliases: &["mem", "memory"],
feature: "mod-hardware",
summary: "RAM usage in gigabytes with usage classification.",
args_synopsis: "",
args_detail: &[],
tokens: &[
("used", "Used memory in GB"),
("total", "Total memory in GB"),
],
examples: &[("fluxo mem", "Show current RAM usage")],
},
ModuleHelp {
name: "sys",
aliases: &["sys"],
feature: "mod-hardware",
summary: "Uptime, load averages, and process count.",
args_synopsis: "",
args_detail: &[],
tokens: &[
("uptime", "Human-readable uptime (e.g. \"2d 5h\")"),
("load1", "1-minute load average"),
("load5", "5-minute load average"),
("load15", "15-minute load average"),
("procs", "Number of running processes"),
],
examples: &[("fluxo sys", "Show system uptime and load")],
},
ModuleHelp {
name: "gpu",
aliases: &["gpu"],
feature: "mod-hardware",
summary: "GPU usage, VRAM, and temperature (AMD/NVIDIA/Intel).",
args_synopsis: "",
args_detail: &[],
tokens: &[
("usage", "GPU utilisation percentage"),
("vram_used", "Used VRAM in GB (AMD/NVIDIA)"),
("vram_total", "Total VRAM in GB (AMD/NVIDIA)"),
("temp", "GPU temperature in Celsius (AMD/NVIDIA)"),
("freq", "GPU frequency in MHz (Intel)"),
],
examples: &[("fluxo gpu", "Show GPU stats for the detected vendor")],
},
ModuleHelp {
name: "disk",
aliases: &["disk"],
feature: "mod-hardware",
summary: "Filesystem usage for a given mount point.",
args_synopsis: "[mountpoint]",
args_detail: &[(
"mountpoint",
"Path to the mount point to display (default: \"/\")",
)],
tokens: &[
("mount", "The mount point path"),
("used", "Used space in GB"),
("total", "Total space in GB"),
],
examples: &[
("fluxo disk", "Show usage of the root filesystem (/)"),
("fluxo disk /home", "Show usage of /home"),
],
},
ModuleHelp {
name: "pool",
aliases: &["pool", "btrfs"],
feature: "mod-hardware",
summary: "Aggregated Btrfs pool usage across all btrfs mounts.",
args_synopsis: "",
args_detail: &[],
tokens: &[
("used", "Total used space in GB across all btrfs mounts"),
("total", "Total capacity in GB across all btrfs mounts"),
],
examples: &[
("fluxo pool", "Show combined Btrfs pool usage"),
("fluxo btrfs", "Same as above (alias)"),
],
},
ModuleHelp {
name: "power",
aliases: &["power"],
feature: "mod-hardware",
summary: "Battery percentage and charge state from sysfs.",
args_synopsis: "",
args_detail: &[],
tokens: &[
("percentage", "Battery level (0 - 100)"),
("icon", "State icon (varies by charge level and AC status)"),
],
examples: &[("fluxo power", "Show battery status")],
},
ModuleHelp {
name: "game",
aliases: &["game"],
feature: "mod-hardware",
summary: "Gamemode indicator (Hyprland animation state).",
args_synopsis: "",
args_detail: &[],
tokens: &[],
examples: &[("fluxo game", "Show whether gamemode is active")],
},
// ── Network ──────────────────────────────────────────────────────
ModuleHelp {
name: "network",
aliases: &["net", "network"],
feature: "mod-network",
summary: "Primary network interface, IP, and transfer rates.",
args_synopsis: "",
args_detail: &[],
tokens: &[
("interface", "Active interface name (e.g. \"wlan0\")"),
("ip", "IPv4 address of the active interface"),
("rx", "Receive rate in MB/s"),
("tx", "Transmit rate in MB/s"),
],
examples: &[("fluxo net", "Show network status and throughput")],
},
// ── Audio ────────────────────────────────────────────────────────
ModuleHelp {
name: "vol (sink)",
aliases: &["vol"],
feature: "mod-audio",
summary: "PulseAudio/PipeWire output (sink) volume and controls.",
args_synopsis: "[show|up|down|mute|cycle] [step]",
args_detail: &[
(
"show",
"Display current sink volume and mute state (default)",
),
("up", "Increase volume by <step> percent (default: 5)"),
("down", "Decrease volume by <step> percent (default: 5)"),
("mute", "Toggle mute on the default sink"),
("cycle", "Switch to the next available output device"),
("step", "Volume change increment in percent (default: 5)"),
],
tokens: &[
("name", "Device description (truncated to 20 chars)"),
("icon", "Volume-level icon (changes with volume/mute)"),
("volume", "Current volume percentage (0 - 150)"),
],
examples: &[
("fluxo vol", "Show current sink volume"),
("fluxo vol up", "Increase volume by 5%"),
("fluxo vol up 10", "Increase volume by 10%"),
("fluxo vol down 2", "Decrease volume by 2%"),
("fluxo vol mute", "Toggle sink mute"),
("fluxo vol cycle", "Switch to next output device"),
],
},
ModuleHelp {
name: "mic (source)",
aliases: &["mic"],
feature: "mod-audio",
summary: "PulseAudio/PipeWire input (source/microphone) controls.",
args_synopsis: "[show|up|down|mute|cycle] [step]",
args_detail: &[
(
"show",
"Display current source volume and mute state (default)",
),
("up", "Increase mic volume by <step> percent (default: 5)"),
("down", "Decrease mic volume by <step> percent (default: 5)"),
("mute", "Toggle mute on the default source"),
("cycle", "Switch to the next available input device"),
("step", "Volume change increment in percent (default: 5)"),
],
tokens: &[
("name", "Device description (truncated to 20 chars)"),
("icon", "Microphone icon (changes with mute state)"),
("volume", "Current volume percentage (0 - 150)"),
],
examples: &[
("fluxo mic", "Show current microphone volume"),
("fluxo mic mute", "Toggle microphone mute"),
("fluxo mic up 10", "Increase mic volume by 10%"),
],
},
// ── Bluetooth ────────────────────────────────────────────────────
ModuleHelp {
name: "bluetooth",
aliases: &["bt", "bluetooth"],
feature: "mod-bt",
summary: "Bluetooth device status, connection management, and plugin modes.",
args_synopsis: "[show|connect|disconnect|cycle|menu|get_modes|set_mode|cycle_mode] [args...]",
args_detail: &[
("show", "Display the active device's status (default)"),
(
"connect <mac>",
"Connect to the device with the given MAC address",
),
(
"disconnect [mac]",
"Disconnect the active device, or a specific MAC",
),
(
"cycle",
"Cycle through connected devices (multi-device setups)",
),
(
"menu",
"Open an interactive device picker (client-side, uses menu_command)",
),
(
"get_modes [mac]",
"List available plugin modes (e.g. ANC modes for Pixel Buds)",
),
(
"set_mode <mode> [mac]",
"Set a plugin mode on the active or specified device",
),
("cycle_mode [mac]", "Advance to the next plugin mode"),
],
tokens: &[
("alias", "Device display name (e.g. \"Pixel Buds Pro\")"),
("mac", "Device MAC address"),
("left", "Left earbud battery (plugin, e.g. \"85%\")"),
("right", "Right earbud battery (plugin, e.g. \"90%\")"),
(
"anc",
"ANC mode label (plugin, e.g. \"ANC\", \"Aware\", \"Off\")",
),
],
examples: &[
("fluxo bt", "Show the active BT device"),
(
"fluxo bt connect AA:BB:CC:DD:EE:FF",
"Connect to a specific device",
),
("fluxo bt disconnect", "Disconnect the active device"),
("fluxo bt menu", "Open the interactive BT device menu"),
("fluxo bt cycle_mode", "Toggle ANC mode on Pixel Buds"),
("fluxo bt set_mode aware", "Set ANC to aware mode"),
],
},
// ── D-Bus ────────────────────────────────────────────────────────
ModuleHelp {
name: "mpris",
aliases: &["mpris"],
feature: "mod-dbus",
summary: "MPRIS media player status (artist, title, playback state).",
args_synopsis: "",
args_detail: &[],
tokens: &[
("artist", "Current track artist"),
("title", "Current track title"),
("album", "Current track album"),
("status_icon", "Playback icon (play/pause/stop glyph)"),
],
examples: &[("fluxo mpris", "Show current media player status")],
},
ModuleHelp {
name: "backlight",
aliases: &["backlight"],
feature: "mod-dbus",
summary: "Screen brightness percentage (inotify-driven).",
args_synopsis: "",
args_detail: &[],
tokens: &[
("percentage", "Current brightness level (0 - 100)"),
("icon", "Brightness bucket icon"),
],
examples: &[("fluxo backlight", "Show current screen brightness")],
},
ModuleHelp {
name: "keyboard",
aliases: &["kbd", "keyboard"],
feature: "mod-dbus",
summary: "Active keyboard layout (Hyprland event-driven).",
args_synopsis: "",
args_detail: &[],
tokens: &[(
"layout",
"Active keyboard layout name (e.g. \"English (US)\")",
)],
examples: &[("fluxo kbd", "Show the current keyboard layout")],
},
ModuleHelp {
name: "dnd",
aliases: &["dnd"],
feature: "mod-dbus",
summary: "Do-Not-Disturb toggle (SwayNC signal-driven / Dunst polling).",
args_synopsis: "[show|toggle]",
args_detail: &[
("show", "Display the current DND state (default)"),
("toggle", "Toggle DND on/off via the notification daemon"),
],
tokens: &[],
examples: &[
("fluxo dnd", "Show current DND state"),
("fluxo dnd toggle", "Toggle Do-Not-Disturb"),
],
},
];
/// Print help for all modules or a single module by name.
pub fn print_help(module: Option<&str>) {
if let Some(name) = module {
let found = MODULES.iter().find(|m| {
m.aliases.iter().any(|a| a.eq_ignore_ascii_case(name))
|| m.name.eq_ignore_ascii_case(name)
});
match found {
Some(m) => print_module_detail(m),
None => {
eprintln!("Unknown module: \"{}\"\n", name);
eprintln!("Run `fluxo help` to see all available modules.");
std::process::exit(1);
}
}
} else {
print_overview();
}
}
fn print_overview() {
println!("\x1b[1;36mfluxo\x1b[0m — high-performance daemon/client for Waybar custom modules\n");
println!("\x1b[1mUSAGE:\x1b[0m");
println!(" fluxo daemon [--config <path>] Start the background daemon");
println!(" fluxo reload Hot-reload the daemon config");
println!(" fluxo <module> [args...] Query or control a module");
println!(" fluxo help [module] Show this help or module details\n");
println!("\x1b[1mCONFIGURATION:\x1b[0m");
println!(" Config file: $XDG_CONFIG_HOME/fluxo/config.toml");
println!(" Format tokens in config strings use {{token}} syntax.");
println!(" Run `fluxo help <module>` to see available tokens.\n");
let categories: &[(&str, &[&str])] = &[
(
"Hardware",
&[
"cpu", "memory", "sys", "gpu", "disk", "pool", "power", "game",
],
),
("Network", &["network"]),
("Audio", &["vol (sink)", "mic (source)"]),
("Bluetooth", &["bluetooth"]),
("D-Bus", &["mpris", "backlight", "keyboard", "dnd"]),
];
println!("\x1b[1mMODULES:\x1b[0m\n");
for (category, names) in categories {
println!(
" \x1b[1;33m{}\x1b[0m ({})",
category,
feature_for_category(category)
);
for module_name in *names {
if let Some(m) = MODULES.iter().find(|m| m.name == *module_name) {
let aliases = m.aliases.join(", ");
println!(" \x1b[1;32m{:<18}\x1b[0m {}", aliases, m.summary,);
if !m.args_synopsis.is_empty() {
println!(" {:<18} args: {}", "", m.args_synopsis,);
}
}
}
println!();
}
println!("\x1b[1mEXAMPLES:\x1b[0m\n");
println!(" fluxo daemon Start the daemon");
println!(" fluxo cpu Show CPU usage and temperature");
println!(" fluxo vol up 10 Increase volume by 10%");
println!(" fluxo bt menu Open Bluetooth device picker");
println!(" fluxo dnd toggle Toggle Do-Not-Disturb");
println!(" fluxo help vol Show detailed help for the volume module");
println!();
println!("For detailed module info: \x1b[1mfluxo help <module>\x1b[0m");
}
fn print_module_detail(m: &ModuleHelp) {
println!("\x1b[1;36mfluxo {}\x1b[0m — {}\n", m.name, m.summary);
// Aliases
if m.aliases.len() > 1
|| m.aliases.first() != Some(&m.name.split_whitespace().next().unwrap_or(m.name))
{
println!("\x1b[1mALIASES:\x1b[0m {}", m.aliases.join(", "));
println!();
}
// Feature gate
println!("\x1b[1mFEATURE:\x1b[0m {}", m.feature);
println!();
// Usage
println!("\x1b[1mUSAGE:\x1b[0m");
let primary = m.aliases.first().unwrap_or(&m.name);
if m.args_synopsis.is_empty() {
println!(" fluxo {}", primary);
} else {
println!(" fluxo {} {}", primary, m.args_synopsis);
}
println!();
// Arguments
if !m.args_detail.is_empty() {
println!("\x1b[1mARGUMENTS:\x1b[0m\n");
let max_name = m
.args_detail
.iter()
.map(|(n, _)| n.len())
.max()
.unwrap_or(0);
for (name, desc) in m.args_detail {
println!(
" \x1b[32m{:<width$}\x1b[0m {}",
name,
desc,
width = max_name
);
}
println!();
}
// Format tokens
if !m.tokens.is_empty() {
println!("\x1b[1mFORMAT TOKENS:\x1b[0m (for use in config.toml format strings)\n");
let max_token = m.tokens.iter().map(|(t, _)| t.len()).max().unwrap_or(0);
for (token, desc) in m.tokens {
let padded = format!("{{{}}}", token);
println!(
" \x1b[33m{:<width$}\x1b[0m {}",
padded,
desc,
width = max_token + 2 // +2 for the braces
);
}
println!();
}
// Examples
if !m.examples.is_empty() {
println!("\x1b[1mEXAMPLES:\x1b[0m\n");
for (cmd, desc) in m.examples {
println!(" \x1b[1m$\x1b[0m {:<34} # {}", cmd, desc);
}
println!();
}
}
fn feature_for_category(category: &str) -> &'static str {
match category {
"Hardware" => "mod-hardware",
"Network" => "mod-network",
"Audio" => "mod-audio",
"Bluetooth" => "mod-bt",
"D-Bus" => "mod-dbus",
_ => "default",
}
}
+28 -9
View File
@@ -1,27 +1,46 @@
//! Client/daemon Unix socket IPC.
//!
//! Requests are newline-terminated `module arg1 arg2...` lines; responses are
//! the daemon's JSON payload written until EOF.
use std::io::{Read, Write};
use std::os::unix::net::UnixStream;
use std::time::Duration;
use tracing::debug;
pub const SOCKET_PATH: &str = "/tmp/fluxo.sock";
/// Returns the path to the daemon's Unix socket.
///
/// Prefers `$XDG_RUNTIME_DIR/fluxo.sock`, falling back to `/tmp/fluxo.sock`.
pub fn socket_path() -> String {
if let Ok(dir) = std::env::var("XDG_RUNTIME_DIR") {
format!("{}/fluxo.sock", dir)
} else {
"/tmp/fluxo.sock".to_string()
}
}
/// Send a module invocation to the daemon and return its response body.
///
/// Blocks for up to 5 seconds waiting for the daemon to reply.
pub fn request_data(module: &str, args: &[&str]) -> anyhow::Result<String> {
let sock = socket_path();
debug!(module, ?args, "Connecting to daemon socket: {}", sock);
let mut stream = UnixStream::connect(&sock)?;
stream.set_read_timeout(Some(Duration::from_secs(5)))?;
pub fn request_data(module: &str, args: &[String]) -> anyhow::Result<String> {
debug!(module, ?args, "Connecting to daemon socket: {}", SOCKET_PATH);
let mut stream = UnixStream::connect(SOCKET_PATH)?;
// Send module and args
let mut request = module.to_string();
for arg in args {
request.push(' ');
request.push_str(arg);
}
request.push('\n');
debug!("Sending IPC request: {}", request.trim());
stream.write_all(request.as_bytes())?;
let mut response = String::new();
stream.read_to_string(&mut response)?;
debug!("Received IPC response: {}", response);
Ok(response)
}
+43
View File
@@ -0,0 +1,43 @@
//! Central declarative macro that registers every watched module.
//!
//! Every piece of per-module boilerplate (AppReceivers field, IPC dispatch arm,
//! signaler future binding, signaler select arm, config enabled-lookup, default
//! signaler args) is generated from this single table. Adding a new module is
//! a one-line edit here plus writing the module file itself.
/// Central module registry. Defines all modules with watch channels in one place.
///
/// Invoke with a callback macro name. The callback receives repeated entries of the form:
/// { $feature:literal, $field:ident, $state:ty, [$($name:literal),+], [$($sig_name:literal),+], $module:path, $signal:ident, [$($default_arg:literal),*], $config:ident }
///
/// Fields:
/// - feature: Cargo feature gate (e.g., "mod-network")
/// - field: AppReceivers field name (e.g., network)
/// - state: State type for the watch channel (e.g., NetworkState)
/// - names: CLI name aliases for dispatch (e.g., ["net", "network"])
/// - signaler_names: Waybar module names to signal when the channel fires
/// (usually [first dispatch name], but audio signals ["vol", "mic"])
/// - module: Module struct implementing WaybarModule (e.g., network::NetworkModule)
/// - signal: SignalsConfig field name (e.g., network)
/// - default_args: Default args for signaler evaluation
/// - config: Config section field name (e.g., network)
///
/// Modules without watch channels (power, game, pool/btrfs) are handled manually.
macro_rules! for_each_watched_module {
($m:ident) => {
$m! {
{ "mod-network", network, crate::state::NetworkState, ["net", "network"], ["net"], crate::modules::network::NetworkModule, network, [], network }
{ "mod-hardware", cpu, crate::state::CpuState, ["cpu"], ["cpu"], crate::modules::cpu::CpuModule, cpu, [], cpu }
{ "mod-hardware", memory, crate::state::MemoryState, ["mem", "memory"], ["mem"], crate::modules::memory::MemoryModule, memory, [], memory }
{ "mod-hardware", sys, crate::state::SysState, ["sys"], ["sys"], crate::modules::sys::SysModule, sys, [], sys }
{ "mod-hardware", gpu, crate::state::GpuState, ["gpu"], ["gpu"], crate::modules::gpu::GpuModule, gpu, [], gpu }
{ "mod-hardware", disks, Vec<crate::state::DiskInfo>, ["disk"], ["disk"], crate::modules::disk::DiskModule, disk, ["/"], disk }
{ "mod-bt", bluetooth, crate::state::BtState, ["bt", "bluetooth"], ["bt"], crate::modules::bt::BtModule, bt, ["show"], bt }
{ "mod-audio", audio, crate::state::AudioState, ["vol", "audio"], ["vol", "mic"], crate::modules::audio::AudioModule, audio, ["sink", "show"], audio }
{ "mod-dbus", mpris, crate::state::MprisState, ["mpris"], ["mpris"], crate::modules::mpris::MprisModule, mpris, [], mpris }
{ "mod-dbus", backlight, crate::state::BacklightState, ["backlight"], ["backlight"], crate::modules::backlight::BacklightModule, backlight, [], backlight }
{ "mod-dbus", keyboard, crate::state::KeyboardState, ["kbd", "keyboard"], ["kbd"], crate::modules::keyboard::KeyboardModule, keyboard, [], keyboard }
{ "mod-dbus", dnd, crate::state::DndState, ["dnd"], ["dnd"], crate::modules::dnd::DndModule, dnd, [], dnd }
}
};
}
+111 -155
View File
@@ -1,23 +1,84 @@
//! `fluxo` — high-performance daemon/client for Waybar custom modules.
//!
//! The binary has two faces:
//! * `fluxo daemon` — starts a long-lived process that polls system state
//! (network, cpu, audio, bluetooth, …) on background tasks and exposes the
//! results over a Unix socket. It also sends `SIGRTMIN+N` signals to Waybar
//! when module output changes, so the bar refreshes instantly.
//! * `fluxo <module> [args]` — a tiny client that asks the daemon to evaluate
//! a single module and prints the Waybar-compatible JSON to stdout.
//!
//! Modules are feature-gated at compile time (`mod-audio`, `mod-bt`, `mod-dbus`,
//! `mod-hardware`, `mod-network`) and registered centrally via the
//! [`for_each_watched_module!`] macro in [`mod@macros`].
#[macro_use]
mod macros;
#[cfg(feature = "mod-bt")]
mod bt_menu;
mod client;
mod config;
mod daemon;
mod error;
mod health;
mod help;
mod ipc;
mod modules;
mod output;
mod registry;
mod signaler;
mod state;
mod utils;
use clap::{Parser, Subcommand};
use clap::{Parser, Subcommand, ValueEnum};
use std::path::PathBuf;
use std::process;
use tracing::{error, info};
use tracing_subscriber::{fmt, prelude::*, EnvFilter};
use tracing_subscriber::{EnvFilter, fmt, prelude::*};
#[derive(Clone, ValueEnum)]
enum LogLevel {
Trace,
Debug,
Info,
Warn,
Error,
}
impl From<LogLevel> for tracing::Level {
fn from(level: LogLevel) -> Self {
match level {
LogLevel::Trace => tracing::Level::TRACE,
LogLevel::Debug => tracing::Level::DEBUG,
LogLevel::Info => tracing::Level::INFO,
LogLevel::Warn => tracing::Level::WARN,
LogLevel::Error => tracing::Level::ERROR,
}
}
}
#[derive(Parser)]
#[command(name = "fluxo")]
#[command(about = "A high-performance daemon/client for Waybar custom modules", long_about = None)]
#[command(disable_help_subcommand = true, disable_help_flag = true)]
struct Cli {
#[command(subcommand)]
command: Commands,
command: Option<Commands>,
/// Print help information
#[arg(short, long, global = true)]
help: bool,
/// Set the log level (trace, debug, info, warn, error)
#[arg(long, global = true, value_enum)]
loglevel: Option<LogLevel>,
/// 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)]
@@ -30,172 +91,67 @@ 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,
/// Show detailed help for all modules or a specific module
Help {
/// Optional module name to show detailed help for
module: Option<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,
},
/// Pixel Buds Pro ANC and Battery
Buds {
#[arg(default_value = "show")]
action: String,
},
/// System power and battery status
Power,
/// Hyprland gamemode status
Game,
}
fn main() {
tracing_subscriber::registry()
.with(fmt::layer().with_target(false).pretty())
.with(EnvFilter::from_default_env().add_directive(tracing::Level::INFO.into()))
.init();
let cli = Cli::parse();
match &cli.command {
Commands::Daemon { config } => {
info!("Starting Fluxo daemon...");
if let Err(e) = daemon::run_daemon(config.clone()) {
error!("Daemon failed: {}", e);
process::exit(1);
// Explicit --loglevel takes priority, then RUST_LOG env var, then a
// sensible default: INFO for the daemon, WARN for client commands.
let default_level = if let Some(level) = &cli.loglevel {
tracing::Level::from(level.clone())
} else if matches!(&cli.command, Some(Commands::Daemon { .. })) {
tracing::Level::INFO
} else {
tracing::Level::WARN
};
tracing_subscriber::registry()
.with(fmt::layer().with_target(false).pretty())
.with(EnvFilter::from_default_env().add_directive(default_level.into()))
.init();
if cli.help {
help::print_help(None);
return;
}
if let Some(command) = &cli.command {
match command {
Commands::Daemon { config } => {
info!("Starting Fluxo daemon...");
let rt = tokio::runtime::Builder::new_multi_thread()
.enable_all()
.build()
.unwrap();
if let Err(e) = rt.block_on(daemon::run_daemon(config.clone())) {
error!("Daemon failed: {}", e);
process::exit(1);
}
}
}
Commands::Reload => {
match ipc::request_data("reload", &[]) {
Commands::Reload => match ipc::request_data("reload", &[]) {
Ok(_) => info!("Reload signal sent to daemon"),
Err(e) => {
error!("Failed to send reload signal: {}", e);
process::exit(1);
}
},
Commands::Help { module } => {
help::print_help(module.as_deref());
}
}
Commands::Net => handle_ipc_response(ipc::request_data("net", &[])),
Commands::Cpu => handle_ipc_response(ipc::request_data("cpu", &[])),
Commands::Mem => handle_ipc_response(ipc::request_data("mem", &[])),
Commands::Disk { path } => handle_ipc_response(ipc::request_data("disk", &[path.clone()])),
Commands::Pool { kind } => handle_ipc_response(ipc::request_data("pool", &[kind.clone()])),
Commands::Vol { cycle } => {
let action = if *cycle { "cycle" } else { "show" };
handle_ipc_response(ipc::request_data("vol", &[action.to_string()]));
}
Commands::Mic { cycle } => {
let action = if *cycle { "cycle" } else { "show" };
handle_ipc_response(ipc::request_data("mic", &[action.to_string()]));
}
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);
let devices_out = std::process::Command::new("bluetoothctl")
.args(["devices"])
.output()
.expect("Failed to run bluetoothctl");
let stdout = String::from_utf8_lossy(&devices_out.stdout);
let mut items = Vec::new();
for line in stdout.lines() {
if line.starts_with("Device ") {
let parts: Vec<&str> = line.splitn(3, ' ').collect();
if parts.len() == 3 {
items.push(format!("{} ({})", parts[2], parts[1]));
}
}
}
return;
}
if !items.is_empty() {
if let Ok(selected) = utils::show_menu("Connect BT: ", &items, &config.general.menu_command) {
if let Some(mac_start) = selected.rfind('(') {
if let Some(mac_end) = selected.rfind(')') {
let mac = &selected[mac_start + 1..mac_end];
let _ = std::process::Command::new("bluetoothctl")
.args(["connect", mac])
.status();
}
}
}
} else {
info!("No paired Bluetooth devices found.");
}
return;
}
handle_ipc_response(ipc::request_data("bt", &[action.clone()]));
}
Commands::Buds { action } => handle_ipc_response(ipc::request_data("buds", &[action.clone()])),
Commands::Power => handle_ipc_response(ipc::request_data("power", &[])),
Commands::Game => handle_ipc_response(ipc::request_data("game", &[])),
}
}
fn handle_ipc_response(response: anyhow::Result<String>) {
match response {
Ok(json_str) => {
match serde_json::from_str::<serde_json::Value>(&json_str) {
Ok(mut val) => {
if let Some(text) = val.get_mut("text").and_then(|t| t.as_str()) {
let processed_text = if text.contains('<') {
text.to_string()
} else {
text.replace(' ', "\u{2007}")
};
let fixed_text = format!("\u{200B}{}\u{200B}", processed_text);
val["text"] = serde_json::Value::String(fixed_text);
}
println!("{}", serde_json::to_string(&val).unwrap());
}
Err(_) => println!("{}", json_str),
}
}
Err(e) => {
let err_out = output::WaybarOutput {
text: format!("\u{200B}Daemon offline ({})\u{200B}", e),
tooltip: Some(e.to_string()),
class: Some("error".to_string()),
percentage: None,
};
println!("{}", serde_json::to_string(&err_out).unwrap());
process::exit(1);
}
if let Some(module) = &cli.module {
client::run_module_command(module, &cli.args);
} else {
help::print_help(None);
}
}
+472 -110
View File
@@ -1,56 +1,439 @@
//! PulseAudio/PipeWire sink + source indicator with live event subscription.
//!
//! The daemon runs on its own OS thread because libpulse's threaded mainloop
//! must drive callbacks inside its own lock scope. Volume/mute changes are
//! routed back via an async [`mpsc`] channel — the module handlers [`run`]s
//! only push commands; the thread performs the actual libpulse calls.
use crate::config::Config;
use crate::error::{FluxoError, Result};
use crate::modules::WaybarModule;
use crate::output::WaybarOutput;
use crate::state::SharedState;
use crate::utils::{format_template, TokenValue};
use anyhow::{Result, anyhow};
use std::process::Command;
use crate::state::{AppReceivers, AudioDeviceInfo, AudioState};
use crate::utils::{TokenValue, format_template};
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::sync::Arc;
use tokio::sync::{mpsc, watch};
use tracing::error;
/// Commands the module handler sends to the audio daemon thread.
pub enum AudioCommand {
ChangeVolume {
is_sink: bool,
step_val: u32,
is_up: bool,
},
ToggleMute {
is_sink: bool,
},
CycleDevice {
is_sink: bool,
},
}
/// Long-lived daemon driving libpulse's threaded mainloop.
pub struct AudioDaemon;
impl AudioDaemon {
/// Construct a new (stateless) daemon.
pub fn new() -> Self {
Self
}
/// Spawn the audio thread, subscribe to sink/source/server events, and
/// start consuming [`AudioCommand`]s.
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 || {
let mut mainloop =
ThreadedMainloop::new().expect("Failed to create pulse threaded mainloop");
let mut context =
Context::new(&mainloop, "fluxo").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();
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();
}
}
}
let _ = fetch_audio_data_sync(&mut context, &state_tx);
let interest =
InterestMaskSet::SINK | InterestMaskSet::SOURCE | InterestMaskSet::SERVER;
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_cb.send(());
}
_ => {}
}
})));
mainloop.unlock();
loop {
if let Ok(cmd) = cmd_rx.try_recv() {
mainloop.lock();
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 {
&current.available_sinks
} else {
&current.available_sources
};
if !devices.is_empty() {
let current_idx =
devices.iter().position(|d| d == &current_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();
let _ = fetch_audio_data_sync(&mut context, &state_tx);
mainloop.unlock();
}
});
}
}
use std::time::Duration;
/// Trigger async libpulse introspection: server defaults + sink/source lists.
/// Callbacks publish onto `state_tx` as results land.
fn fetch_audio_data_sync(
context: &mut Context,
state_tx: &watch::Sender<AudioState>,
) -> Result<()> {
let tx_server = state_tx.clone();
context.introspect().get_server_info(move |info| {
let mut current = tx_server.borrow().clone();
current.sink.name = info
.default_sink_name
.as_ref()
.map(|s| s.to_string())
.unwrap_or_default();
current.source.name = info
.default_source_name
.as_ref()
.map(|s| s.to_string())
.unwrap_or_default();
let _ = tx_server.send(current);
});
fetch_sinks(context, state_tx);
fetch_sources(context, state_tx);
Ok(())
}
/// Shared bookkeeping for a device list fetch.
struct PendingList {
names: Arc<std::sync::Mutex<Vec<String>>>,
}
impl PendingList {
fn new() -> Self {
Self {
names: Arc::new(std::sync::Mutex::new(Vec::new())),
}
}
fn push(&self, name: String) {
self.names.lock().unwrap().push(name);
}
fn drain(&self) -> Vec<String> {
self.names.lock().unwrap().drain(..).collect()
}
}
/// Extract common device info from a pulse item's volume/mute/description fields.
fn device_info_from(
description: Option<&str>,
volume: &libpulse_binding::volume::ChannelVolumes,
muted: bool,
) -> (String, u8, bool, u8) {
let desc = description.unwrap_or_default().to_string();
let vol = ((volume.avg().0 as f64 / Volume::NORMAL.0 as f64) * 100.0).round() as u8;
let channels = volume.len();
(desc, vol, muted, channels)
}
/// Write `info` into `target` only when `item_name` matches the currently
/// selected default device — other sinks/sources are ignored here.
fn apply_device_info(target: &mut AudioDeviceInfo, item_name: &str, info: (String, u8, bool, u8)) {
if item_name == target.name {
target.description = info.0;
target.volume = info.1;
target.muted = info.2;
target.channels = info.3;
}
}
/// Dispatch `get_sink_info_list` and collect names into `available_sinks`.
fn fetch_sinks(context: &mut Context, state_tx: &watch::Sender<AudioState>) {
let tx = state_tx.clone();
let pending = PendingList::new();
let pending_cb = PendingList {
names: Arc::clone(&pending.names),
};
context.introspect().get_sink_info_list(move |res| {
let mut current = tx.borrow().clone();
match res {
ListResult::Item(item) => {
let name_str = item
.name
.as_ref()
.map(|s| s.to_string())
.unwrap_or_default();
if !name_str.is_empty() {
pending_cb.push(name_str.clone());
}
let info = device_info_from(
item.description.as_ref().map(|s| s.as_ref()),
&item.volume,
item.mute,
);
apply_device_info(&mut current.sink, &name_str, info);
let _ = tx.send(current);
}
ListResult::End => {
current.available_sinks = pending_cb.drain();
let _ = tx.send(current);
}
ListResult::Error => {}
}
});
}
/// Dispatch `get_source_info_list` and collect names (skipping `.monitor`
/// virtual sources) into `available_sources`.
fn fetch_sources(context: &mut Context, state_tx: &watch::Sender<AudioState>) {
let tx = state_tx.clone();
let pending = PendingList::new();
let pending_cb = PendingList {
names: Arc::clone(&pending.names),
};
context.introspect().get_source_info_list(move |res| {
let mut current = tx.borrow().clone();
match res {
ListResult::Item(item) => {
let name_str = item
.name
.as_ref()
.map(|s| s.to_string())
.unwrap_or_default();
if !name_str.is_empty() && !name_str.contains(".monitor") {
pending_cb.push(name_str.clone());
}
let info = device_info_from(
item.description.as_ref().map(|s| s.as_ref()),
&item.volume,
item.mute,
);
apply_device_info(&mut current.source, &name_str, info);
let _ = tx.send(current);
}
ListResult::End => {
current.available_sources = pending_cb.drain();
let _ = tx.send(current);
}
ListResult::Error => {}
}
});
}
/// Renders sink/source + dispatches volume/mute/cycle commands.
/// Args: `[sink|source] [show|up|down|mute|cycle] [step]`.
pub struct AudioModule;
impl WaybarModule for AudioModule {
fn run(&self, config: &Config, _state: &SharedState, args: &[&str]) -> Result<WaybarOutput> {
async fn run(
&self,
config: &Config,
state: &AppReceivers,
args: &[&str],
) -> Result<WaybarOutput> {
let target_type = args.first().unwrap_or(&"sink");
let action = args.get(1).unwrap_or(&"show");
let step = args.get(2).unwrap_or(&"5");
match *action {
"up" => {
self.change_volume(state, target_type, step, true).await?;
self.get_status(config, state, target_type).await
}
"down" => {
self.change_volume(state, target_type, step, false).await?;
self.get_status(config, state, target_type).await
}
"mute" => {
self.toggle_mute(state, target_type).await?;
self.get_status(config, state, target_type).await
}
"cycle" => {
self.cycle_device(target_type)?;
return Ok(WaybarOutput {
text: String::new(),
tooltip: None,
class: None,
percentage: None,
});
}
"show" | _ => {
self.get_status(config, target_type)
self.cycle_device(state, target_type).await?;
self.get_status(config, state, target_type).await
}
"show" => self.get_status(config, state, target_type).await,
other => Err(FluxoError::Module {
module: "audio",
message: format!("Unknown audio action: '{}'", other),
}),
}
}
}
impl AudioModule {
fn get_status(&self, config: &Config, target_type: &str) -> Result<WaybarOutput> {
let target = if target_type == "sink" { "@DEFAULT_AUDIO_SINK@" } else { "@DEFAULT_AUDIO_SOURCE@" };
async fn get_status(
&self,
config: &Config,
state: &AppReceivers,
target_type: &str,
) -> Result<WaybarOutput> {
let audio_state = state.audio.borrow().clone();
let output = Command::new("wpctl")
.args(["get-volume", target])
.output()?;
let stdout = String::from_utf8_lossy(&output.stdout);
let parts: Vec<&str> = stdout.trim().split_whitespace().collect();
if parts.len() < 2 {
return Err(anyhow!("Could not parse wpctl output: {}", stdout));
let (name, description, volume, muted) = if target_type == "sink" {
(
audio_state.sink.name,
audio_state.sink.description,
audio_state.sink.volume,
audio_state.sink.muted,
)
} else {
(
audio_state.source.name,
audio_state.source.description,
audio_state.source.volume,
audio_state.source.muted,
)
};
if name.is_empty() {
return Ok(WaybarOutput {
text: "Audio Loading...".to_string(),
..Default::default()
});
}
let vol_val: f64 = parts[1].parse().unwrap_or(0.0);
let vol = (vol_val * 100.0).round() as u8;
let display_vol = std::cmp::min(vol, 100);
let muted = stdout.contains("[MUTED]");
let description = self.get_description(target_type)?;
let name = if description.len() > 20 {
let display_name = if description.len() > 20 {
format!("{}...", &description[..17])
} else {
description.clone()
@@ -58,31 +441,43 @@ impl AudioModule {
let (text, class) = if muted {
let icon = if target_type == "sink" { "" } else { "" };
let format_str = if target_type == "sink" { &config.audio.format_sink_muted } else { &config.audio.format_source_muted };
let format_str = if target_type == "sink" {
&config.audio.format_sink_muted
} else {
&config.audio.format_source_muted
};
let t = format_template(
format_str,
&[
("name", TokenValue::String(&name)),
("icon", TokenValue::String(icon)),
]
("name", TokenValue::String(display_name)),
("icon", TokenValue::String(icon.to_string())),
],
);
(t, "muted")
} else {
let icon = if target_type == "sink" {
if display_vol <= 30 { "" }
else if display_vol <= 60 { "" }
else { "" }
if volume <= 30 {
""
} else if volume <= 60 {
""
} else {
""
}
} else {
""
};
let format_str = if target_type == "sink" { &config.audio.format_sink_unmuted } else { &config.audio.format_source_unmuted };
let format_str = if target_type == "sink" {
&config.audio.format_sink_unmuted
} else {
&config.audio.format_source_unmuted
};
let t = format_template(
format_str,
&[
("name", TokenValue::String(&name)),
("icon", TokenValue::String(icon)),
("volume", TokenValue::Int(display_vol as i64)),
]
("name", TokenValue::String(display_name)),
("icon", TokenValue::String(icon.to_string())),
("volume", TokenValue::Int(volume as i64)),
],
);
(t, "unmuted")
};
@@ -91,78 +486,45 @@ impl AudioModule {
text,
tooltip: Some(description),
class: Some(class.to_string()),
percentage: Some(display_vol),
percentage: Some(volume),
})
}
fn get_description(&self, target_type: &str) -> Result<String> {
let info_output = Command::new("pactl").arg("info").output()?;
let info_stdout = String::from_utf8_lossy(&info_output.stdout);
let search_key = if target_type == "sink" { "Default Sink:" } else { "Default Source:" };
let default_dev = info_stdout.lines()
.find(|l| l.contains(search_key))
.and_then(|l| l.split(':').nth(1))
.map(|s| s.trim())
.ok_or_else(|| anyhow!("Default {} not found", target_type))?;
let list_cmd = if target_type == "sink" { "sinks" } else { "sources" };
let list_output = Command::new("pactl").args(["list", list_cmd]).output()?;
let list_stdout = String::from_utf8_lossy(&list_output.stdout);
let mut current_name = String::new();
for line in list_stdout.lines() {
if line.trim().starts_with("Name: ") {
current_name = line.split(':').nth(1).unwrap_or("").trim().to_string();
}
if current_name == default_dev && line.trim().starts_with("Description: ") {
return Ok(line.split(':').nth(1).unwrap_or("").trim().to_string());
}
}
Ok(default_dev.to_string())
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,
})
.await;
Ok(())
}
fn cycle_device(&self, target_type: &str) -> Result<()> {
let list_cmd = if target_type == "sink" { "sinks" } else { "sources" };
let output = Command::new("pactl").args(["list", "short", list_cmd]).output()?;
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
}
})
.collect();
if devices.is_empty() { return Ok(()); }
let info_output = Command::new("pactl").arg("info").output()?;
let info_stdout = String::from_utf8_lossy(&info_output.stdout);
let search_key = if target_type == "sink" { "Default Sink:" } else { "Default Source:" };
let current_dev = info_stdout.lines()
.find(|l| l.contains(search_key))
.and_then(|l| l.split(':').nth(1))
.map(|s| s.trim())
.unwrap_or("");
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];
let set_cmd = if target_type == "sink" { "set-default-sink" } else { "set-default-source" };
Command::new("pactl").args([set_cmd, next_dev]).status()?;
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(())
}
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(())
}
}
+159
View File
@@ -0,0 +1,159 @@
//! Screen backlight indicator, driven by `inotify` on
//! `/sys/class/backlight/*/actual_brightness`. Falls back to a 5 s poll loop
//! to catch any missed events.
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};
/// Renders the brightness percentage with a vendor-agnostic icon bucket.
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),
})
}
}
/// Background `inotify` watcher thread for the sysfs backlight file.
pub struct BacklightDaemon;
impl BacklightDaemon {
/// Construct a new (stateless) daemon.
pub fn new() -> Self {
Self
}
/// Spawn an OS thread that publishes brightness changes onto `tx`.
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
}
};
let _ = tx.send(BacklightState {
percentage: get_percentage(),
});
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 {
if ev_rx.recv_timeout(Duration::from_secs(5)).is_ok() {
// Debounce bursts from scroll-driven brightness changes.
std::thread::sleep(Duration::from_millis(50));
while ev_rx.try_recv().is_ok() {}
let _ = tx.send(BacklightState {
percentage: get_percentage(),
});
} else {
// Timeout reached — resync in case an event was missed.
let current = get_percentage();
if tx.borrow().percentage != current {
let _ = tx.send(BacklightState {
percentage: current,
});
}
}
}
});
}
}
-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::{format_template, TokenValue};
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(output) = Command::new("bluetoothctl").arg("show").output() {
let stdout = String::from_utf8_lossy(&output.stdout);
if stdout.contains("Powered: no") {
return Ok(WaybarOutput {
text: config.bt.format_disabled.clone(),
tooltip: Some("Bluetooth Disabled".to_string()),
class: Some("disabled".to_string()),
percentage: None,
});
}
}
if let Some(mac) = find_audio_device() {
let info_output = Command::new("bluetoothctl").args(["info", &mac]).output()?;
let info = String::from_utf8_lossy(&info_output.stdout);
let mut 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(output) = Command::new("pactl").arg("get-default-sink").output() {
let sink = String::from_utf8_lossy(&output.stdout).trim().to_string();
if sink.starts_with("bluez_output.") {
let parts: Vec<&str> = sink.split('.').collect();
if parts.len() >= 2 {
return Some(parts[1].replace('_', ":"));
}
}
}
if let Ok(output) = Command::new("bluetoothctl").args(["devices", "Connected"]).output() {
let stdout = String::from_utf8_lossy(&output.stdout);
for line in stdout.lines() {
if line.starts_with("Device ") {
let parts: Vec<&str> = line.split_whitespace().collect();
if parts.len() >= 2 {
let mac = parts[1];
if let Ok(info) = Command::new("bluetoothctl").args(["info", mac]).output() {
let info_str = String::from_utf8_lossy(&info.stdout);
if info_str.contains("0000110b-0000-1000-8000-00805f9b34fb") {
return Some(mac.to_string());
}
}
}
}
}
}
None
}
+153
View File
@@ -0,0 +1,153 @@
//! Per-device BT plugin trait + PixelBuds Pro implementation on top of the
//! Maestro GATT connection.
use crate::config::Config;
use crate::error::{FluxoError, Result as FluxoResult};
use crate::modules::bt::maestro::BudsCommand;
use crate::state::AppReceivers;
use crate::utils::TokenValue;
use futures::future::BoxFuture;
/// A device-specific adapter that can enrich [`BtState`](crate::state::BtState)
/// with extra metadata and expose control actions (modes).
pub trait BtPlugin: Send + Sync {
/// Plugin identifier used for logging.
fn name(&self) -> &str;
/// Return true if this plugin handles a device with `alias`/`mac`.
fn can_handle(&self, alias: &str, mac: &str) -> bool;
/// Return `(token_name, value)` pairs merged into the rendered template.
fn get_data(
&self,
config: &Config,
state: &AppReceivers,
mac: &str,
) -> BoxFuture<'static, FluxoResult<Vec<(String, TokenValue)>>>;
/// List of mode identifiers the plugin can switch between.
fn get_modes(
&self,
mac: &str,
state: &AppReceivers,
) -> BoxFuture<'static, FluxoResult<Vec<String>>>;
/// Switch device to `mode` (must be one returned by `get_modes`).
fn set_mode(
&self,
mode: &str,
mac: &str,
state: &AppReceivers,
) -> BoxFuture<'static, FluxoResult<()>>;
/// Advance to the next mode in the list (wraps around).
fn cycle_mode(&self, mac: &str, state: &AppReceivers) -> BoxFuture<'static, FluxoResult<()>>;
}
/// Google Pixel Buds Pro plugin. Reads battery + ANC state via Maestro GATT.
pub struct PixelBudsPlugin;
impl BtPlugin for PixelBudsPlugin {
fn name(&self) -> &str {
"Pixel Buds Pro"
}
fn can_handle(&self, alias: &str, _mac: &str) -> bool {
alias.contains("Pixel Buds Pro")
}
fn get_data(
&self,
_config: &Config,
state: &AppReceivers,
mac: &str,
) -> BoxFuture<'static, FluxoResult<Vec<(String, TokenValue)>>> {
let mac = mac.to_string();
let state = state.clone();
Box::pin(async move {
let maestro = crate::modules::bt::maestro::get_maestro(&state);
maestro.ensure_task(&mac);
let status = maestro.get_status(&mac);
if let Some(err) = status.error {
return Err(FluxoError::Module {
module: "bt.buds",
message: err,
});
}
let left_display = status
.left_battery
.map(|b| format!("{}%", b))
.unwrap_or_else(|| "---".to_string());
let right_display = status
.right_battery
.map(|b| format!("{}%", b))
.unwrap_or_else(|| "---".to_string());
let (anc_icon, class) = match status.anc_state.as_str() {
"active" => ("ANC", "anc-active"),
"aware" => ("Aware", "anc-aware"),
"off" => ("Off", "anc-off"),
_ => ("?", "anc-unknown"),
};
Ok(vec![
("left".to_string(), TokenValue::String(left_display)),
("right".to_string(), TokenValue::String(right_display)),
("anc".to_string(), TokenValue::String(anc_icon.to_string())),
(
"plugin_class".to_string(),
TokenValue::String(class.to_string()),
),
])
})
}
fn get_modes(
&self,
_mac: &str,
_state: &AppReceivers,
) -> BoxFuture<'static, FluxoResult<Vec<String>>> {
Box::pin(async move {
Ok(vec![
"active".to_string(),
"aware".to_string(),
"off".to_string(),
])
})
}
fn set_mode(
&self,
mode: &str,
mac: &str,
state: &AppReceivers,
) -> BoxFuture<'static, FluxoResult<()>> {
let mode = mode.to_string();
let mac = mac.to_string();
let state = state.clone();
Box::pin(async move {
crate::modules::bt::maestro::get_maestro(&state)
.send_command(&mac, BudsCommand::SetAnc(mode))
.map_err(|e: anyhow::Error| FluxoError::Module {
module: "bt.buds",
message: e.to_string(),
})
})
}
fn cycle_mode(&self, mac: &str, state: &AppReceivers) -> BoxFuture<'static, FluxoResult<()>> {
let mac = mac.to_string();
let state = state.clone();
Box::pin(async move {
let status = crate::modules::bt::maestro::get_maestro(&state).get_status(&mac);
let next_mode = match status.anc_state.as_str() {
"active" => "aware",
"aware" => "off",
_ => "active",
};
crate::modules::bt::maestro::get_maestro(&state)
.send_command(&mac, BudsCommand::SetAnc(next_mode.to_string()))
.map_err(|e: anyhow::Error| FluxoError::Module {
module: "bt.buds",
message: e.to_string(),
})
})
}
}
+382
View File
@@ -0,0 +1,382 @@
//! Google Maestro (PixelBuds GATT) integration.
//!
//! Each connected device gets its own [`buds_task`] running on a dedicated
//! single-threaded runtime. The task opens an RFCOMM channel, speaks the
//! Maestro protocol to read battery + ANC state, and listens for settings
//! changes. External callers interact via [`MaestroManager::send_command`]
//! and [`MaestroManager::get_status`].
use crate::state::AppReceivers;
use anyhow::{Context, Result};
use futures::StreamExt;
use std::collections::HashMap;
use std::sync::{Arc, Mutex, OnceLock};
use std::time::{Duration, Instant};
use tokio::sync::mpsc;
use tracing::{debug, error, info, warn};
use maestro::protocol::codec::Codec;
use maestro::pwrpc::client::Client;
use maestro::service::MaestroService;
use maestro::service::settings::{self, SettingValue};
/// Cached per-device snapshot returned to BT plugin consumers.
#[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>,
}
/// Command that can be issued against a connected buds device.
pub enum BudsCommand {
/// Set the ANC mode: `active`, `aware`, or `off`.
SetAnc(String),
}
/// Messages sent to the [`MaestroManager`] control thread.
pub enum ManagerCommand {
/// Ensure a [`buds_task`] is running for `mac`; spawn if absent.
EnsureTask(String),
/// Forward a [`BudsCommand`] to the task for `mac`.
SendCommand(String, BudsCommand),
}
/// Owns all buds-task lifetimes and a shared status cache.
pub struct MaestroManager {
statuses: Arc<Mutex<HashMap<String, BudsStatus>>>,
management_tx: mpsc::UnboundedSender<ManagerCommand>,
}
impl MaestroManager {
/// Spawn the management thread + runtime and return a handle.
pub fn new(state: AppReceivers) -> Self {
let (tx, mut rx) = mpsc::unbounded_channel::<ManagerCommand>();
let statuses = Arc::new(Mutex::new(HashMap::new()));
let statuses_clone = Arc::clone(&statuses);
let state_clone = state.clone();
// Dedicated thread — bluer uses per-thread local tasks.
std::thread::spawn(move || {
let rt = tokio::runtime::Builder::new_current_thread()
.enable_all()
.build()
.unwrap();
let local = tokio::task::LocalSet::new();
local.block_on(&rt, async move {
let mut command_txs: HashMap<String, mpsc::Sender<BudsCommand>> = HashMap::new();
loop {
tokio::select! {
Some(cmd) = rx.recv() => {
match cmd {
ManagerCommand::EnsureTask(mac) => {
if !command_txs.contains_key(&mac) {
let (tx, buds_rx) = mpsc::channel::<BudsCommand>(10);
command_txs.insert(mac.clone(), tx);
let mac_clone = mac.clone();
let st_clone = Arc::clone(&statuses_clone);
let state_inner = state_clone.clone();
tokio::task::spawn_local(async move {
if let Err(e) = buds_task(&mac_clone, st_clone, buds_rx, state_inner).await {
error!("Buds task for {} failed: {}", mac_clone, e);
}
});
}
}
ManagerCommand::SendCommand(mac, buds_cmd) => {
if let Some(tx) = command_txs.get(&mac) {
let _ = tx.try_send(buds_cmd);
}
}
}
}
_ = tokio::time::sleep(Duration::from_millis(100)) => {
// Wake tick: future hook for task-lifecycle cleanup.
}
}
}
});
});
Self {
statuses,
management_tx: tx,
}
}
/// Return the cached [`BudsStatus`] for `mac` (default if absent).
pub fn get_status(&self, mac: &str) -> BudsStatus {
let statuses = self.statuses.lock().unwrap();
statuses.get(mac).cloned().unwrap_or_default()
}
/// Request that a buds task be running for `mac`. Idempotent.
pub fn ensure_task(&self, mac: &str) {
let _ = self
.management_tx
.send(ManagerCommand::EnsureTask(mac.to_string()));
}
/// Ensure a task exists and forward `cmd` to it.
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(())
}
}
/// Per-device async task: opens RFCOMM, runs the Maestro codec, mirrors
/// battery/ANC state into the shared status map, and consumes commands.
async fn buds_task(
mac: &str,
statuses: Arc<Mutex<HashMap<String, BudsStatus>>>,
mut rx: mpsc::Receiver<BudsCommand>,
state: AppReceivers,
) -> Result<()> {
info!("Starting native Maestro connection task for {}", mac);
loop {
let addr: bluer::Address = match mac.parse() {
Ok(a) => a,
Err(e) => {
error!("Failed to parse MAC address {}: {}", mac, e);
return Err(e.into());
}
};
let session = bluer::Session::new()
.await
.context("Failed to create bluer session")?;
let adapter = session
.default_adapter()
.await
.context("Failed to get default adapter")?;
let device = adapter
.device(addr)
.context("Failed to get device handle")?;
if !device.is_connected().await.unwrap_or(false) {
debug!("Device {} not connected to BT, stopping maestro task", mac);
break;
}
// Maestro historically listens on channel 1 or 2 — probe both.
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);
let codec = Codec::new();
let stream = codec.wrap(stream);
let mut client = Client::new(stream);
let handle = client.handle();
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);
// Successful connect — clear health backoff for bt.buds.
{
let mut lock = state.health.write().await;
let health = lock.entry("bt.buds".to_string()).or_default();
health.consecutive_failures = 0;
health.backoff_until = None;
}
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(())
}
/// String ("active"/"aware"/"off") → Maestro enum; unknown falls back to `Off`.
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,
}
}
/// Inverse of [`mode_to_anc_state`] for status readout.
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();
/// Lazily initialise the process-wide [`MaestroManager`] and return a reference.
pub fn get_maestro(state: &AppReceivers) -> &MaestroManager {
MAESTRO.get_or_init(|| MaestroManager::new(state.clone()))
}
+369
View File
@@ -0,0 +1,369 @@
//! Bluetooth indicator + control (BlueZ via `bluer`).
//!
//! Core loop: filter paired+connected audio-sink devices, enrich them via
//! per-device [`BtPlugin`]s (currently PixelBuds via the Maestro GATT
//! protocol), and publish the result as [`BtState`]. The module handler
//! exposes `connect`, `disconnect`, `cycle`, `menu_data`, `get_modes`,
//! `set_mode`, `cycle_mode` actions for the Waybar menu.
pub mod buds;
pub mod maestro;
use crate::config::Config;
use crate::error::Result as FluxoResult;
use crate::modules::WaybarModule;
use crate::output::WaybarOutput;
use crate::state::{AppReceivers, BtDeviceInfo, BtState};
use crate::utils::{TokenValue, format_template};
use anyhow::Result;
use std::sync::LazyLock;
use tokio::sync::watch;
use tracing::{error, warn};
use self::buds::{BtPlugin, PixelBudsPlugin};
/// Background poller that syncs connected BlueZ devices into [`BtState`].
pub struct BtDaemon {
session: Option<bluer::Session>,
}
impl BtDaemon {
/// Construct a new daemon. The BlueZ session is lazily created on first poll.
pub fn new() -> Self {
Self { session: None }
}
/// Poll wrapper that logs + swallows errors so the outer loop keeps running.
pub async fn poll(
&mut self,
tx: &watch::Sender<BtState>,
state: &AppReceivers,
config: &Config,
) {
if let Err(e) = self.poll_async(tx, state, config).await {
error!("BT daemon error: {}", e);
}
}
async fn poll_async(
&mut self,
tx: &watch::Sender<BtState>,
state: &AppReceivers,
config: &Config,
) -> Result<()> {
if self.session.is_none() {
self.session = Some(bluer::Session::new().await?);
}
let session = self.session.as_ref().unwrap();
let adapter = session.default_adapter().await?;
let adapter_powered = adapter.is_powered().await.unwrap_or(false);
let mut connected_devices = Vec::new();
if adapter_powered {
let mut addresses = adapter.device_addresses().await?;
addresses.sort();
let audio_sink_uuid = bluer::Uuid::from_u128(0x0000110b_0000_1000_8000_00805f9b34fb);
for addr in addresses {
let device = adapter.device(addr)?;
if !device.is_connected().await.unwrap_or(false) {
continue;
}
let uuids = device.uuids().await?.unwrap_or_default();
if !uuids.contains(&audio_sink_uuid) {
continue;
}
let mut dev_info = BtDeviceInfo {
device_address: addr.to_string(),
device_alias: device.alias().await.unwrap_or_else(|_| addr.to_string()),
battery_percentage: device.battery_percentage().await.unwrap_or(None),
plugin_data: vec![],
};
for p in PLUGINS.iter() {
if p.can_handle(&dev_info.device_alias, &dev_info.device_address) {
match p.get_data(config, state, &dev_info.device_address).await {
Ok(data) => {
dev_info.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);
dev_info
.plugin_data
.push(("plugin_error".to_string(), e.to_string()));
}
}
break;
}
}
connected_devices.push(dev_info);
}
}
let _ = tx.send(BtState {
adapter_powered,
devices: connected_devices,
});
Ok(())
}
}
static PLUGINS: LazyLock<Vec<Box<dyn BtPlugin>>> =
LazyLock::new(|| vec![Box::new(PixelBudsPlugin)]);
/// After a user-initiated connect/disconnect, schedule a staircase of
/// forced polls so the UI catches up even if BlueZ is slow to settle.
fn trigger_robust_poll(state: AppReceivers) {
tokio::spawn(async move {
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(());
}
});
}
/// Resolve a target MAC: use explicit arg if given, otherwise fall back to the active device.
async fn resolve_target_mac(
bt_state: &BtState,
state: &AppReceivers,
explicit_mac: Option<&str>,
) -> Option<String> {
if let Some(mac) = explicit_mac {
return Some(mac.to_string());
}
let idx = *state.bt_cycle.read().await;
bt_state
.active_device(idx)
.map(|d| d.device_address.clone())
}
/// Find a device in the current state by MAC.
fn find_device<'a>(bt_state: &'a BtState, mac: &str) -> Option<&'a BtDeviceInfo> {
bt_state.devices.iter().find(|d| d.device_address == mac)
}
/// Renders the current BT status + handles control actions.
pub struct BtModule;
impl WaybarModule for BtModule {
async fn run(
&self,
config: &Config,
state: &AppReceivers,
args: &[&str],
) -> FluxoResult<WaybarOutput> {
let action = args.first().cloned().unwrap_or("show");
let bt_state = state.bluetooth.borrow().clone();
match action {
"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" => {
let target_mac = resolve_target_mac(&bt_state, state, args.get(1).copied()).await;
if let Some(mac) = target_mac {
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.disconnect().await;
}
trigger_robust_poll(state.clone());
}
return Ok(WaybarOutput::default());
}
"cycle" => {
let count = bt_state.devices.len();
if count > 1 {
let mut idx = state.bt_cycle.write().await;
*idx = (*idx + 1) % count;
}
let _ = state.bt_force_poll.try_send(());
return Ok(WaybarOutput::default());
}
"menu_data" => {
let mut lines = Vec::new();
for dev in &bt_state.devices {
lines.push(format!(
"CONNECTED:{}|{}",
dev.device_alias, dev.device_address
));
}
// Also surface paired-but-not-connected devices for the menu.
if let Ok(session) = bluer::Session::new().await
&& let Ok(adapter) = session.default_adapter().await
&& let Ok(addresses) = adapter.device_addresses().await
{
let connected_macs: std::collections::HashSet<&str> = bt_state
.devices
.iter()
.map(|d| d.device_address.as_str())
.collect();
for addr in addresses {
let addr_str = addr.to_string();
if connected_macs.contains(addr_str.as_str()) {
continue;
}
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());
lines.push(format!("PAIRED:{}|{}", alias, addr_str));
}
}
}
return Ok(WaybarOutput {
text: lines.join("\n"),
..Default::default()
});
}
"get_modes" => {
let target_mac = resolve_target_mac(&bt_state, state, args.get(1).copied()).await;
if let Some(mac) = target_mac
&& let Some(dev) = find_device(&bt_state, &mac)
{
let plugin = PLUGINS
.iter()
.find(|p| p.can_handle(&dev.device_alias, &dev.device_address));
if let Some(p) = plugin {
let modes = p.get_modes(&mac, state).await?;
return Ok(WaybarOutput {
text: modes.join("\n"),
..Default::default()
});
}
}
return Ok(WaybarOutput::default());
}
"set_mode" => {
if let Some(mode) = args.get(1) {
let target_mac =
resolve_target_mac(&bt_state, state, args.get(2).copied()).await;
if let Some(mac) = target_mac
&& let Some(dev) = find_device(&bt_state, &mac)
{
let plugin = PLUGINS
.iter()
.find(|p| p.can_handle(&dev.device_alias, &dev.device_address));
if let Some(p) = plugin {
p.set_mode(mode, &mac, state).await?;
trigger_robust_poll(state.clone());
}
}
}
return Ok(WaybarOutput::default());
}
"cycle_mode" => {
let target_mac = resolve_target_mac(&bt_state, state, args.get(1).copied()).await;
if let Some(mac) = target_mac
&& let Some(dev) = find_device(&bt_state, &mac)
{
let plugin = PLUGINS
.iter()
.find(|p| p.can_handle(&dev.device_alias, &dev.device_address));
if let Some(p) = plugin {
p.cycle_mode(&mac, state).await?;
trigger_robust_poll(state.clone());
}
}
return Ok(WaybarOutput::default());
}
_ => {}
}
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,
});
}
let cycle_idx = *state.bt_cycle.read().await;
if let Some(dev) = bt_state.active_device(cycle_idx) {
let mut tokens: Vec<(String, TokenValue)> = vec![
(
"alias".to_string(),
TokenValue::String(dev.device_alias.clone()),
),
(
"mac".to_string(),
TokenValue::String(dev.device_address.clone()),
),
];
let mut class = vec!["connected".to_string()];
let mut has_plugin = false;
for (k, v) in &dev.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: {}",
dev.device_alias,
dev.device_address,
dev.battery_percentage
.map(|b| format!("{}%", b))
.unwrap_or_else(|| "N/A".to_string())
);
Ok(WaybarOutput {
text,
tooltip: Some(tooltip),
class: Some(class.join(" ")),
percentage: dev.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,
})
}
}
}
+24 -17
View File
@@ -1,23 +1,36 @@
//! Btrfs pool renderer: sums usage across all btrfs-typed mounts seen in the
//! `disks` watch channel. Dispatch-only (no dedicated poll task).
use crate::config::Config;
use crate::error::Result;
use crate::modules::WaybarModule;
use crate::output::WaybarOutput;
use crate::state::SharedState;
use crate::utils::{format_template, TokenValue};
use anyhow::Result;
use sysinfo::Disks;
use crate::state::AppReceivers;
use crate::utils::{TokenValue, classify_usage, format_template};
/// Aggregates used/total across every mount whose filesystem name contains
/// `btrfs`. Emits `No BTRFS` when none are present.
pub struct BtrfsModule;
impl WaybarModule for BtrfsModule {
fn run(&self, config: &Config, _state: &SharedState, _args: &[&str]) -> Result<WaybarOutput> {
let disks = Disks::new_with_refreshed_list();
async fn run(
&self,
config: &Config,
state: &AppReceivers,
_args: &[&str],
) -> Result<WaybarOutput> {
let disks = {
let s = state.disks.borrow();
s.clone()
};
let mut total_used: f64 = 0.0;
let mut total_size: f64 = 0.0;
for disk in &disks {
if disk.file_system().to_string_lossy().to_lowercase().contains("btrfs") {
let size = disk.total_space() as f64;
let available = disk.available_space() as f64;
if disk.filesystem.contains("btrfs") {
let size = disk.total_bytes as f64;
let available = disk.available_bytes as f64;
total_size += size;
total_used += size - available;
}
@@ -36,20 +49,14 @@ impl WaybarModule for BtrfsModule {
let size_gb = total_size / 1024.0 / 1024.0 / 1024.0;
let percentage = (total_used / total_size) * 100.0;
let class = if percentage > 95.0 {
"max"
} else if percentage > 80.0 {
"high"
} else {
"normal"
};
let class = classify_usage(percentage, 80.0, 95.0);
let text = format_template(
&config.pool.format,
&[
("used", TokenValue::Float(used_gb)),
("total", TokenValue::Float(size_gb)),
]
],
);
Ok(WaybarOutput {
-121
View File
@@ -1,121 +0,0 @@
use crate::config::Config;
use crate::modules::WaybarModule;
use crate::output::WaybarOutput;
use crate::state::SharedState;
use crate::utils::{format_template, TokenValue};
use anyhow::Result;
use std::process::Command;
pub struct BudsModule;
impl WaybarModule for BudsModule {
fn run(&self, config: &Config, _state: &SharedState, args: &[&str]) -> Result<WaybarOutput> {
let action = args.first().unwrap_or(&"show");
let mac = &config.buds.mac;
match *action {
"cycle_anc" => {
let output = Command::new("pbpctrl").args(["get", "anc"]).output()?;
let current_mode = String::from_utf8_lossy(&output.stdout).trim().to_string();
let next_mode = match current_mode.as_str() {
"active" => "aware",
"aware" => "off",
_ => "active",
};
Command::new("pbpctrl").args(["set", "anc", next_mode]).status()?;
return Ok(WaybarOutput {
text: String::new(),
tooltip: None, class: None, percentage: None,
});
}
"connect" => {
Command::new("bluetoothctl").args(["connect", mac]).status()?;
return Ok(WaybarOutput {
text: String::new(),
tooltip: None, class: None, percentage: None,
});
}
"disconnect" => {
Command::new("bluetoothctl").args(["disconnect", mac]).status()?;
return Ok(WaybarOutput {
text: String::new(),
tooltip: None, class: None, percentage: None,
});
}
"show" | _ => {}
}
let bt_info = Command::new("bluetoothctl").args(["info", mac]).output()?;
let bt_str = String::from_utf8_lossy(&bt_info.stdout);
if !bt_str.contains("Connected: yes") {
return Ok(WaybarOutput {
text: config.buds.format_disconnected.clone(),
tooltip: Some("Pixel Buds Pro 2 not connected".to_string()),
class: Some("disconnected".to_string()),
percentage: None,
});
}
let bat_cmd = Command::new("pbpctrl").args(["show", "battery"]).output();
if bat_cmd.is_err() || !bat_cmd.as_ref().unwrap().status.success() {
return Ok(WaybarOutput {
text: config.buds.format_disconnected.clone(),
tooltip: Some("Pixel Buds Pro 2 connected (No Data)".to_string()),
class: Some("disconnected".to_string()),
percentage: None,
});
}
let bat_result = bat_cmd.unwrap();
let bat_output = String::from_utf8_lossy(&bat_result.stdout);
let mut left_bud = "unknown";
let mut right_bud = "unknown";
for line in bat_output.lines() {
if line.contains("left bud:") {
left_bud = line.split_whitespace().nth(2).unwrap_or("unknown");
} else if line.contains("right bud:") {
right_bud = line.split_whitespace().nth(2).unwrap_or("unknown");
}
}
if left_bud == "unknown" && right_bud == "unknown" {
return Ok(WaybarOutput {
text: "{}".to_string(),
tooltip: None, class: None, percentage: None,
});
}
let left_display = if left_bud == "unknown" { "---".to_string() } else { format!("{}%", left_bud) };
let right_display = if right_bud == "unknown" { "---".to_string() } else { format!("{}%", right_bud) };
let anc_cmd = Command::new("pbpctrl").args(["get", "anc"]).output()?;
let current_mode = String::from_utf8_lossy(&anc_cmd.stdout).trim().to_string();
let (anc_icon, class) = match current_mode.as_str() {
"active" => ("ANC", "anc-active"),
"aware" => ("Aware", "anc-aware"),
"off" => ("Off", "anc-off"),
_ => ("?", "anc-unknown"),
};
let text = format_template(
&config.buds.format,
&[
("left", TokenValue::String(&left_display)),
("right", TokenValue::String(&right_display)),
("anc", TokenValue::String(anc_icon)),
]
);
Ok(WaybarOutput {
text,
tooltip: Some("Pixel Buds Pro 2".to_string()),
class: Some(class.to_string()),
percentage: None,
})
}
}
+72 -22
View File
@@ -1,24 +1,25 @@
//! CPU usage + temperature renderer. Reads from the `cpu` watch channel.
use crate::config::Config;
use crate::error::Result;
use crate::modules::WaybarModule;
use crate::output::WaybarOutput;
use crate::state::SharedState;
use crate::utils::{format_template, TokenValue};
use anyhow::Result;
use crate::state::AppReceivers;
use crate::utils::{TokenValue, classify_usage, format_template};
/// Renders CPU usage/temp using [`CpuConfig::format`](crate::config::CpuConfig).
pub struct CpuModule;
impl WaybarModule for CpuModule {
fn run(&self, config: &Config, state: &SharedState, _args: &[&str]) -> Result<WaybarOutput> {
async fn run(
&self,
config: &Config,
state: &AppReceivers,
_args: &[&str],
) -> Result<WaybarOutput> {
let (usage, temp, model) = {
if let Ok(state_lock) = state.read() {
(
state_lock.cpu.usage,
state_lock.cpu.temp,
state_lock.cpu.model.clone(),
)
} else {
(0.0, 0.0, String::from("Unknown"))
}
let s = state.cpu.borrow();
(s.usage, s.temp, s.model.clone())
};
let text = format_template(
@@ -26,16 +27,10 @@ impl WaybarModule for CpuModule {
&[
("usage", TokenValue::Float(usage)),
("temp", TokenValue::Float(temp)),
]
],
);
let class = if usage > 95.0 {
"max"
} else if usage > 75.0 {
"high"
} else {
"normal"
};
let class = classify_usage(usage, 75.0, 95.0);
Ok(WaybarOutput {
text,
@@ -45,3 +40,58 @@ impl WaybarModule for CpuModule {
})
}
}
#[cfg(test)]
mod tests {
use super::*;
use crate::state::{AppState, CpuState, mock_state};
#[tokio::test]
async fn test_cpu_normal() {
let state = mock_state(AppState {
cpu: CpuState {
usage: 25.0,
temp: 45.0,
model: "Test CPU".into(),
},
..Default::default()
});
let config = Config::default();
let output = CpuModule.run(&config, &state.receivers, &[]).await.unwrap();
assert!(output.text.contains("25.0"));
assert!(output.text.contains("45.0"));
assert_eq!(output.class.as_deref(), Some("normal"));
assert_eq!(output.percentage, Some(25));
assert_eq!(output.tooltip.as_deref(), Some("Test CPU"));
}
#[tokio::test]
async fn test_cpu_high() {
let state = mock_state(AppState {
cpu: CpuState {
usage: 80.0,
temp: 70.0,
model: "Test".into(),
},
..Default::default()
});
let config = Config::default();
let output = CpuModule.run(&config, &state.receivers, &[]).await.unwrap();
assert_eq!(output.class.as_deref(), Some("high"));
}
#[tokio::test]
async fn test_cpu_max() {
let state = mock_state(AppState {
cpu: CpuState {
usage: 99.0,
temp: 95.0,
model: "Test".into(),
},
..Default::default()
});
let config = Config::default();
let output = CpuModule.run(&config, &state.receivers, &[]).await.unwrap();
assert_eq!(output.class.as_deref(), Some("max"));
}
}
+95 -22
View File
@@ -1,56 +1,129 @@
//! Filesystem usage renderer. Args: `[mountpoint]` (default `/`).
use crate::config::Config;
use crate::error::{FluxoError, Result};
use crate::modules::WaybarModule;
use crate::output::WaybarOutput;
use crate::state::SharedState;
use crate::utils::{format_template, TokenValue};
use anyhow::Result;
use sysinfo::Disks;
use crate::state::AppReceivers;
use crate::utils::{TokenValue, classify_usage, format_template};
/// Renders used/total for a given mount point. Returns [`FluxoError::Module`]
/// if the mount point isn't present in the current disk snapshot.
pub struct DiskModule;
impl WaybarModule for DiskModule {
fn run(&self, config: &Config, _state: &SharedState, args: &[&str]) -> Result<WaybarOutput> {
async fn run(
&self,
config: &Config,
state: &AppReceivers,
args: &[&str],
) -> Result<WaybarOutput> {
let mountpoint = args.first().unwrap_or(&"/");
let disks = Disks::new_with_refreshed_list();
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().to_string_lossy() == *mountpoint {
let total = disk.total_space() as f64;
let available = disk.available_space() as f64;
if disk.mount_point == *mountpoint {
let total = disk.total_bytes as f64;
let available = disk.available_bytes as f64;
let used = total - available;
let used_gb = used / 1024.0 / 1024.0 / 1024.0;
let total_gb = total / 1024.0 / 1024.0 / 1024.0;
let free_gb = available / 1024.0 / 1024.0 / 1024.0;
let percentage = if total > 0.0 { (used / total) * 100.0 } else { 0.0 };
let class = if percentage > 95.0 {
"max"
} else if percentage > 80.0 {
"high"
let percentage = if total > 0.0 {
(used / total) * 100.0
} else {
"normal"
0.0
};
let class = classify_usage(percentage, 80.0, 95.0);
let text = format_template(
&config.disk.format,
&[
("mount", TokenValue::String(mountpoint)),
("mount", TokenValue::String(mountpoint.to_string())),
("used", TokenValue::Float(used_gb)),
("total", TokenValue::Float(total_gb)),
]
],
);
return Ok(WaybarOutput {
text,
tooltip: Some(format!("Used: {:.1}G\nTotal: {:.1}G\nFree: {:.1}G", used_gb, total_gb, free_gb)),
tooltip: Some(format!(
"Used: {:.1}G\nTotal: {:.1}G\nFree: {:.1}G",
used_gb, total_gb, free_gb
)),
class: Some(class.to_string()),
percentage: Some(percentage as u8),
});
}
}
Err(anyhow::anyhow!("Mountpoint {} not found", mountpoint))
Err(FluxoError::Module {
module: "disk",
message: format!("Mountpoint {} not found", mountpoint),
})
}
}
#[cfg(test)]
mod tests {
use super::*;
use crate::state::{AppState, DiskInfo, MockState, mock_state};
fn state_with_disk(mount: &str, total: u64, available: u64) -> MockState {
mock_state(AppState {
disks: vec![DiskInfo {
mount_point: mount.to_string(),
filesystem: "ext4".to_string(),
total_bytes: total,
available_bytes: available,
}],
..Default::default()
})
}
#[tokio::test]
async fn test_disk_found() {
let gb = 1024 * 1024 * 1024;
let state = state_with_disk("/", 100 * gb, 60 * gb);
let config = Config::default();
let output = DiskModule
.run(&config, &state.receivers, &["/"])
.await
.unwrap();
assert_eq!(output.class.as_deref(), Some("normal"));
assert_eq!(output.percentage, Some(40)); // 40% used
}
#[tokio::test]
async fn test_disk_high() {
let gb = 1024 * 1024 * 1024;
let state = state_with_disk("/", 100 * gb, 15 * gb);
let config = Config::default();
let output = DiskModule
.run(&config, &state.receivers, &["/"])
.await
.unwrap();
assert_eq!(output.class.as_deref(), Some("high")); // 85% used
}
#[tokio::test]
async fn test_disk_not_found() {
let state = state_with_disk("/", 100, 50);
let config = Config::default();
let result = DiskModule
.run(&config, &state.receivers, &["/nonexistent"])
.await;
assert!(result.is_err());
}
}
+209
View File
@@ -0,0 +1,209 @@
//! Do-Not-Disturb toggle + status for SwayNC or Dunst.
//!
//! SwayNC exposes a `dnd` property on its `org.erikreider.swaync.control`
//! interface that fires PropertiesChanged signals, so we subscribe. Dunst has
//! no change signal for its `paused` property, so we fall back to a 2 s poll.
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 tokio::time::Duration;
use tracing::{debug, error, info};
use zbus::proxy;
use zbus::zvariant::OwnedValue;
use zbus::{Connection, fdo::PropertiesProxy};
/// Renders + toggles DND state. Args: `["show"]` (default) or `["toggle"]`.
pub struct DndModule;
/// Read dunst's `paused` property via raw D-Bus call.
async fn dunst_get_paused(connection: &Connection) -> anyhow::Result<bool> {
let reply = connection
.call_method(
Some("org.freedesktop.Notifications"),
"/org/freedesktop/Notifications",
Some("org.freedesktop.DBus.Properties"),
"Get",
&("org.dunstproject.cmd0", "paused"),
)
.await?;
let value: OwnedValue = reply.body().deserialize()?;
Ok(bool::try_from(&*value)?)
}
/// Set dunst's `paused` property via raw D-Bus call.
async fn dunst_set_paused(connection: &Connection, paused: bool) -> anyhow::Result<()> {
let value = zbus::zvariant::Value::from(paused);
connection
.call_method(
Some("org.freedesktop.Notifications"),
"/org/freedesktop/Notifications",
Some("org.freedesktop.DBus.Properties"),
"Set",
&("org.dunstproject.cmd0", "paused", value),
)
.await?;
Ok(())
}
impl WaybarModule for DndModule {
async fn run(
&self,
config: &Config,
state: &AppReceivers,
args: &[&str],
) -> Result<WaybarOutput> {
let action = args.first().unwrap_or(&"show");
if *action == "toggle" {
let connection =
Connection::session()
.await
.map_err(|e| crate::error::FluxoError::Module {
module: "dnd",
message: format!("DBus connection failed: {}", e),
})?;
if let Ok(proxy) = SwayncControlProxy::new(&connection).await
&& let Ok(is_dnd) = proxy.dnd().await
{
let _ = proxy.set_dnd(!is_dnd).await;
return Ok(WaybarOutput::default());
}
if let Ok(is_paused) = dunst_get_paused(&connection).await {
let _ = dunst_set_paused(&connection, !is_paused).await;
return Ok(WaybarOutput::default());
}
return Err(crate::error::FluxoError::Module {
module: "dnd",
message: "No supported notification daemon found to toggle".to_string(),
});
}
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,
})
}
}
}
/// Background watcher that keeps [`DndState`] in sync with the active
/// notification daemon (SwayNC via signals, Dunst via polling).
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>;
#[zbus(property)]
fn set_dnd(&self, value: bool) -> zbus::Result<()>;
}
impl DndDaemon {
/// Construct a new (stateless) daemon.
pub fn new() -> Self {
Self
}
/// Spawn a supervised listen loop that reconnects with a 5 s backoff.
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");
if let Ok(proxy) = SwayncControlProxy::new(&connection).await
&& let Ok(is_dnd) = proxy.dnd().await
{
debug!("Found SwayNC, using signal-based DND monitoring");
let _ = tx.send(DndState { is_dnd });
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 });
}
}
}
return Err(anyhow::anyhow!("SwayNC DND stream ended"));
}
// Dunst: raw D-Bus call avoids zbus proxy typing quirks with its
// non-standard `org.dunstproject.cmd0` interface.
match dunst_get_paused(&connection).await {
Ok(is_paused) => {
info!("Found Dunst, using polling-based DND monitoring");
let _ = tx.send(DndState { is_dnd: is_paused });
loop {
tokio::time::sleep(Duration::from_secs(2)).await;
match dunst_get_paused(&connection).await {
Ok(is_paused) => {
let current = tx.borrow().is_dnd;
if current != is_paused {
let _ = tx.send(DndState { is_dnd: is_paused });
}
}
Err(e) => {
debug!("Dunst paused() poll failed: {}", e);
break;
}
}
}
return Err(anyhow::anyhow!("Dunst connection lost"));
}
Err(e) => {
info!("Dunst not available: {}", e);
}
}
Err(anyhow::anyhow!(
"No supported notification daemon found (tried SwayNC, Dunst)"
))
}
}
+31 -16
View File
@@ -1,26 +1,28 @@
//! Gamemode indicator. Queries Hyprland's animation setting over its IPC
//! socket; animations disabled => gamemode active. Dispatch-only.
use crate::config::Config;
use crate::error::Result;
use crate::modules::WaybarModule;
use crate::output::WaybarOutput;
use crate::state::SharedState;
use anyhow::Result;
use std::process::Command;
use crate::state::AppReceivers;
use tokio::io::{AsyncReadExt, AsyncWriteExt};
use tokio::net::UnixStream;
/// Renders a glyph depending on whether Hyprland animations are disabled.
pub struct GameModule;
impl WaybarModule for GameModule {
fn run(&self, config: &Config, _state: &SharedState, _args: &[&str]) -> Result<WaybarOutput> {
let output = Command::new("hyprctl")
.args(["getoption", "animations:enabled", "-j"])
.output();
let mut is_gamemode = false;
if let Ok(out) = output {
let stdout = String::from_utf8_lossy(&out.stdout);
if stdout.contains("\"int\": 0") {
is_gamemode = true;
}
}
async fn run(
&self,
config: &Config,
_state: &AppReceivers,
_args: &[&str],
) -> Result<WaybarOutput> {
let is_gamemode = hyprland_ipc("j/getoption animations:enabled")
.await
.map(|stdout| stdout.contains("\"int\": 0"))
.unwrap_or(false);
if is_gamemode {
Ok(WaybarOutput {
@@ -39,3 +41,16 @@ impl WaybarModule for GameModule {
}
}
}
/// Send `cmd` to Hyprland's `.socket.sock` and return the response body.
async fn hyprland_ipc(cmd: &str) -> Result<String> {
let path = crate::utils::get_hyprland_socket(".socket.sock")?;
let mut stream = UnixStream::connect(path).await?;
stream.write_all(cmd.as_bytes()).await?;
let mut response = String::new();
stream.read_to_string(&mut response).await?;
Ok(response)
}
+29 -26
View File
@@ -1,28 +1,34 @@
//! GPU renderer. Picks a vendor-specific format string (AMD/Intel/NVIDIA) and
//! reads from the `gpu` watch channel.
use crate::config::Config;
use crate::error::Result;
use crate::modules::WaybarModule;
use crate::output::WaybarOutput;
use crate::state::SharedState;
use crate::utils::{format_template, TokenValue};
use anyhow::Result;
use crate::state::AppReceivers;
use crate::utils::{TokenValue, classify_usage, format_template};
/// Renders GPU usage / VRAM / temp using the per-vendor format from config.
pub struct GpuModule;
impl WaybarModule for GpuModule {
fn run(&self, config: &Config, state: &SharedState, _args: &[&str]) -> Result<WaybarOutput> {
async fn run(
&self,
config: &Config,
state: &AppReceivers,
_args: &[&str],
) -> Result<WaybarOutput> {
let (active, vendor, usage, vram_used, vram_total, temp, model) = {
if let Ok(state_lock) = state.read() {
(
state_lock.gpu.active,
state_lock.gpu.vendor.clone(),
state_lock.gpu.usage,
state_lock.gpu.vram_used,
state_lock.gpu.vram_total,
state_lock.gpu.temp,
state_lock.gpu.model.clone(),
)
} else {
(false, String::from("Unknown"), 0.0, 0.0, 0.0, 0.0, String::from("Unknown"))
}
let s = state.gpu.borrow();
(
s.active,
s.vendor.clone(),
s.usage,
s.vram_used,
s.vram_total,
s.temp,
s.model.clone(),
)
};
if !active {
@@ -34,13 +40,7 @@ impl WaybarModule for GpuModule {
});
}
let class = if usage > 95.0 {
"max"
} else if usage > 75.0 {
"high"
} else {
"normal"
};
let class = classify_usage(usage, 75.0, 95.0);
let format_str = match vendor.as_str() {
"Intel" => &config.gpu.format_intel,
@@ -55,13 +55,16 @@ impl WaybarModule for GpuModule {
("vram_used", TokenValue::Float(vram_used)),
("vram_total", TokenValue::Float(vram_total)),
("temp", TokenValue::Float(temp)),
]
],
);
let tooltip = if vendor == "Intel" {
format!("Model: {}\nApprox Usage: {:.0}%", model, usage)
} else {
format!("Model: {}\nUsage: {:.0}%\nVRAM: {:.1}/{:.1}GB\nTemp: {:.1}°C", model, usage, vram_used, vram_total, temp)
format!(
"Model: {}\nUsage: {:.0}%\nVRAM: {:.1}/{:.1}GB\nTemp: {:.1}°C",
model, usage, vram_used, vram_total, temp
)
};
Ok(WaybarOutput {
+239 -108
View File
@@ -1,35 +1,73 @@
use crate::state::SharedState;
use sysinfo::{Components, System};
//! Unified CPU/memory/sys/GPU/disk poller.
//!
//! CPU/memory/sys are sampled every fast tick (1 s). GPU polls every 5th fast
//! tick via [`poll_slow`], and disks every 10th (they rarely change). GPU
//! vendor is detected once by probing nvidia-smi / `/sys/class/drm/*`, then
//! cached so subsequent polls take the fast path.
use crate::state::{CpuState, DiskInfo, GpuState, MemoryState, SysState};
use sysinfo::{Components, Disks, System};
use tokio::sync::watch;
/// Long-lived hardware sampler. Holds the `sysinfo::System` handle so
/// successive refreshes can diff against prior samples.
pub struct HardwareDaemon {
sys: System,
components: Components,
gpu_vendor: Option<String>,
gpu_poll_counter: u8,
disk_poll_counter: u8,
}
impl HardwareDaemon {
/// Build a new daemon with an initial `sysinfo` snapshot.
pub fn new() -> Self {
let mut sys = System::new_all();
sys.refresh_all();
let mut sys = System::new();
sys.refresh_cpu_usage();
sys.refresh_memory();
let components = Components::new_with_refreshed_list();
Self { sys, components, gpu_vendor: None }
Self {
sys,
components,
gpu_vendor: None,
gpu_poll_counter: 0,
// Start at 9 so (counter + 1) % 10 == 0 on the first tick.
disk_poll_counter: 9,
}
}
pub fn poll(&mut self, state: SharedState) {
/// Fast path: refresh CPU usage, memory, temperatures, load avg, uptime.
/// Called every daemon tick.
pub async fn poll_fast(
&mut self,
cpu_tx: &watch::Sender<CpuState>,
mem_tx: &watch::Sender<MemoryState>,
sys_tx: &watch::Sender<SysState>,
) {
self.sys.refresh_cpu_usage();
self.sys.refresh_memory();
self.components.refresh(true);
let cpu_usage = self.sys.global_cpu_usage();
let cpu_model = self.sys.cpus().first().map(|c| c.brand().to_string()).unwrap_or_else(|| "Unknown".to_string());
let cpu_model = self
.sys
.cpus()
.first()
.map(|c| c.brand().to_string())
.unwrap_or_else(|| "Unknown".to_string());
let mut cpu_temp = 0.0;
for component in &self.components {
let label = component.label().to_lowercase();
if label.contains("tctl") || label.contains("cpu") || label.contains("package") || label.contains("temp1") {
if let Some(temp) = component.temperature() {
cpu_temp = temp as f64;
if cpu_temp > 0.0 { break; }
if (label.contains("tctl")
|| label.contains("cpu")
|| label.contains("package")
|| label.contains("temp1"))
&& let Some(temp) = component.temperature()
{
cpu_temp = temp as f64;
if cpu_temp > 0.0 {
break;
}
}
}
@@ -40,127 +78,220 @@ impl HardwareDaemon {
let load_avg = System::load_average();
let uptime = System::uptime();
let mut process_count = 0;
if let Ok(loadavg_str) = std::fs::read_to_string("/proc/loadavg") {
let parts: Vec<&str> = loadavg_str.split_whitespace().collect();
if parts.len() >= 4 {
if let Some(total_procs) = parts[3].split('/').nth(1) {
process_count = total_procs.parse().unwrap_or(0);
}
if parts.len() >= 4
&& let Some(total_procs) = parts[3].split('/').nth(1)
{
process_count = total_procs.parse().unwrap_or(0);
}
}
if let Ok(mut state_lock) = state.write() {
state_lock.cpu.usage = cpu_usage as f64;
state_lock.cpu.temp = cpu_temp as f64;
state_lock.cpu.model = cpu_model;
let mut cpu = cpu_tx.borrow().clone();
cpu.usage = cpu_usage as f64;
cpu.temp = cpu_temp;
cpu.model = cpu_model;
let _ = cpu_tx.send(cpu);
state_lock.memory.total_gb = total_mem;
state_lock.memory.used_gb = used_mem;
let mut mem = mem_tx.borrow().clone();
mem.total_gb = total_mem;
mem.used_gb = used_mem;
let _ = mem_tx.send(mem);
state_lock.sys.load_1 = load_avg.one;
state_lock.sys.load_5 = load_avg.five;
state_lock.sys.load_15 = load_avg.fifteen;
state_lock.sys.uptime = uptime;
state_lock.sys.process_count = process_count;
let mut sys = sys_tx.borrow().clone();
sys.load_1 = load_avg.one;
sys.load_5 = load_avg.five;
sys.load_15 = load_avg.fifteen;
sys.uptime = uptime;
sys.process_count = process_count;
let _ = sys_tx.send(sys);
}
self.poll_gpu(&mut state_lock.gpu);
/// Slow path: GPU every 5 ticks, disks every 10 ticks. Each sub-poll
/// runs off the hot loop before any state is published.
pub async fn poll_slow(
&mut self,
gpu_tx: &watch::Sender<GpuState>,
disks_tx: &watch::Sender<Vec<DiskInfo>>,
) {
let mut gpu_state = crate::state::GpuState::default();
self.gpu_poll_counter = (self.gpu_poll_counter + 1) % 5;
let should_poll_gpu = self.gpu_poll_counter == 0;
if should_poll_gpu {
self.poll_gpu(&mut gpu_state).await;
}
let mut disks_data = None;
self.disk_poll_counter = (self.disk_poll_counter + 1) % 10;
if self.disk_poll_counter == 0 {
disks_data = Some(
tokio::task::spawn_blocking(|| {
Disks::new_with_refreshed_list()
.iter()
.map(|d| DiskInfo {
mount_point: d.mount_point().to_string_lossy().into_owned(),
filesystem: d.file_system().to_string_lossy().to_lowercase(),
total_bytes: d.total_space(),
available_bytes: d.available_space(),
})
.collect::<Vec<DiskInfo>>()
})
.await
.unwrap_or_default(),
);
}
if should_poll_gpu {
let _ = gpu_tx.send(gpu_state);
}
if let Some(d) = disks_data {
let _ = disks_tx.send(d);
}
}
fn poll_gpu(&mut self, gpu: &mut crate::state::GpuState) {
/// Dispatch to the cached vendor's probe, or run detection on first call.
async fn poll_gpu(&mut self, gpu: &mut crate::state::GpuState) {
gpu.active = false;
if self.gpu_vendor.as_deref() == Some("NVIDIA") || self.gpu_vendor.is_none() {
if let Ok(output) = std::process::Command::new("nvidia-smi")
.args(["--query-gpu=utilization.gpu,memory.used,memory.total,temperature.gpu,name", "--format=csv,noheader,nounits"])
.output()
{
if output.status.success() {
let stdout = String::from_utf8_lossy(&output.stdout);
if let Some(line) = stdout.lines().next() {
let parts: Vec<&str> = line.split(',').collect();
if parts.len() >= 5 {
gpu.active = true;
gpu.vendor = "NVIDIA".to_string();
gpu.usage = parts[0].trim().parse().unwrap_or(0.0);
gpu.vram_used = parts[1].trim().parse::<f64>().unwrap_or(0.0) / 1024.0;
gpu.vram_total = parts[2].trim().parse::<f64>().unwrap_or(0.0) / 1024.0;
gpu.temp = parts[3].trim().parse().unwrap_or(0.0);
gpu.model = parts[4].trim().to_string();
self.gpu_vendor = Some("NVIDIA".to_string());
return;
}
}
match self.gpu_vendor.as_deref() {
Some("NVIDIA") => {
Self::poll_nvidia(gpu).await;
}
Some("AMD") => {
Self::poll_amd(gpu);
}
Some("Intel") => {
Self::poll_intel(gpu);
}
_ => {
// First run — probe each vendor and cache the first hit.
Self::poll_nvidia(gpu).await;
if gpu.active {
self.gpu_vendor = Some("NVIDIA".to_string());
return;
}
Self::poll_amd(gpu);
if gpu.active {
self.gpu_vendor = Some("AMD".to_string());
return;
}
Self::poll_intel(gpu);
if gpu.active {
self.gpu_vendor = Some("Intel".to_string());
}
}
}
}
if self.gpu_vendor.as_deref() == Some("AMD") || self.gpu_vendor.as_deref() == Some("Intel") || self.gpu_vendor.is_none() {
for i in 0..=3 {
let base = format!("/sys/class/drm/card{}/device", i);
if self.gpu_vendor.as_deref() == Some("AMD") || self.gpu_vendor.is_none() {
if let Ok(usage_str) = std::fs::read_to_string(format!("{}/gpu_busy_percent", base)) {
gpu.active = true;
gpu.vendor = "AMD".to_string();
gpu.usage = usage_str.trim().parse().unwrap_or(0.0);
if let Ok(mem_used) = std::fs::read_to_string(format!("{}/mem_info_vram_used", base)) {
gpu.vram_used = mem_used.trim().parse::<f64>().unwrap_or(0.0) / 1024.0 / 1024.0 / 1024.0;
}
if let Ok(mem_total) = std::fs::read_to_string(format!("{}/mem_info_vram_total", base)) {
gpu.vram_total = mem_total.trim().parse::<f64>().unwrap_or(0.0) / 1024.0 / 1024.0 / 1024.0;
}
if let Ok(entries) = std::fs::read_dir(format!("{}/hwmon", base)) {
for entry in entries.flatten() {
let temp_path = entry.path().join("temp1_input");
if let Ok(temp_str) = std::fs::read_to_string(temp_path) {
gpu.temp = temp_str.trim().parse::<f64>().unwrap_or(0.0) / 1000.0;
break;
}
}
}
gpu.model = "AMD GPU".to_string();
self.gpu_vendor = Some("AMD".to_string());
return;
/// Shell out to `nvidia-smi --query-gpu=...` for utilization/VRAM/temp.
async fn poll_nvidia(gpu: &mut crate::state::GpuState) {
let Ok(output) = tokio::process::Command::new("nvidia-smi")
.args([
"--query-gpu=utilization.gpu,memory.used,memory.total,temperature.gpu,name",
"--format=csv,noheader,nounits",
])
.output()
.await
else {
return;
};
if !output.status.success() {
return;
}
let stdout = String::from_utf8_lossy(&output.stdout);
if let Some(line) = stdout.lines().next() {
let parts: Vec<&str> = line.split(',').collect();
if parts.len() >= 5 {
gpu.active = true;
gpu.vendor = "NVIDIA".to_string();
gpu.usage = parts[0].trim().parse().unwrap_or(0.0);
gpu.vram_used = parts[1].trim().parse::<f64>().unwrap_or(0.0) / 1024.0;
gpu.vram_total = parts[2].trim().parse::<f64>().unwrap_or(0.0) / 1024.0;
gpu.temp = parts[3].trim().parse().unwrap_or(0.0);
gpu.model = parts[4].trim().to_string();
}
}
}
/// Read amdgpu sysfs entries under `/sys/class/drm/card*/device`.
fn poll_amd(gpu: &mut crate::state::GpuState) {
for i in 0..=3 {
let base = format!("/sys/class/drm/card{}/device", i);
let Ok(usage_str) = std::fs::read_to_string(format!("{}/gpu_busy_percent", base))
else {
continue;
};
gpu.active = true;
gpu.vendor = "AMD".to_string();
gpu.usage = usage_str.trim().parse().unwrap_or(0.0);
if let Ok(mem_used) = std::fs::read_to_string(format!("{}/mem_info_vram_used", base)) {
gpu.vram_used =
mem_used.trim().parse::<f64>().unwrap_or(0.0) / 1024.0 / 1024.0 / 1024.0;
}
if let Ok(mem_total) = std::fs::read_to_string(format!("{}/mem_info_vram_total", base))
{
gpu.vram_total =
mem_total.trim().parse::<f64>().unwrap_or(0.0) / 1024.0 / 1024.0 / 1024.0;
}
if let Ok(entries) = std::fs::read_dir(format!("{}/hwmon", base)) {
for entry in entries.flatten() {
let temp_path = entry.path().join("temp1_input");
if let Ok(temp_str) = std::fs::read_to_string(temp_path) {
gpu.temp = temp_str.trim().parse::<f64>().unwrap_or(0.0) / 1000.0;
break;
}
}
}
gpu.model = "AMD GPU".to_string();
return;
}
}
if self.gpu_vendor.as_deref() == Some("Intel") || self.gpu_vendor.is_none() {
let freq_path = if std::path::Path::new(&format!("{}/gt_cur_freq_mhz", base)).exists() {
Some(format!("{}/gt_cur_freq_mhz", base))
} else if std::path::Path::new(&format!("/sys/class/drm/card{}/gt_cur_freq_mhz", i)).exists() {
Some(format!("/sys/class/drm/card{}/gt_cur_freq_mhz", i))
} else {
None
};
/// Read i915/xe sysfs `gt_cur_freq_mhz`; approximate "usage" as
/// current/max frequency since Intel has no direct utilization counter.
fn poll_intel(gpu: &mut crate::state::GpuState) {
for i in 0..=3 {
let base = format!("/sys/class/drm/card{}/device", i);
let freq_path = if std::path::Path::new(&format!("{}/gt_cur_freq_mhz", base)).exists() {
Some(format!("{}/gt_cur_freq_mhz", base))
} else if std::path::Path::new(&format!("/sys/class/drm/card{}/gt_cur_freq_mhz", i))
.exists()
{
Some(format!("/sys/class/drm/card{}/gt_cur_freq_mhz", i))
} else {
None
};
if let Some(path) = freq_path {
if let Ok(freq_str) = std::fs::read_to_string(&path) {
gpu.active = true;
gpu.vendor = "Intel".to_string();
let cur_freq = freq_str.trim().parse::<f64>().unwrap_or(0.0);
let mut max_freq = 0.0;
let max_path = path.replace("gt_cur_freq_mhz", "gt_max_freq_mhz");
if let Ok(max_str) = std::fs::read_to_string(max_path) {
max_freq = max_str.trim().parse::<f64>().unwrap_or(0.0);
}
if let Some(path) = freq_path
&& let Ok(freq_str) = std::fs::read_to_string(&path)
{
gpu.active = true;
gpu.vendor = "Intel".to_string();
gpu.usage = if max_freq > 0.0 { (cur_freq / max_freq) * 100.0 } else { 0.0 };
gpu.temp = 0.0;
gpu.vram_used = 0.0;
gpu.vram_total = 0.0;
gpu.model = format!("Intel iGPU ({}MHz)", cur_freq);
self.gpu_vendor = Some("Intel".to_string());
return;
}
}
let cur_freq = freq_str.trim().parse::<f64>().unwrap_or(0.0);
let mut max_freq = 0.0;
let max_path = path.replace("gt_cur_freq_mhz", "gt_max_freq_mhz");
if let Ok(max_str) = std::fs::read_to_string(max_path) {
max_freq = max_str.trim().parse::<f64>().unwrap_or(0.0);
}
gpu.usage = if max_freq > 0.0 {
(cur_freq / max_freq) * 100.0
} else {
0.0
};
gpu.temp = 0.0;
gpu.vram_used = 0.0;
gpu.vram_total = 0.0;
gpu.model = format!("Intel iGPU ({}MHz)", cur_freq);
return;
}
}
}
+112
View File
@@ -0,0 +1,112 @@
//! Keyboard layout indicator backed by Hyprland's event socket
//! (`.socket2.sock`). Also seeds the initial layout by shelling out to
//! `hyprctl devices -j` once at startup.
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};
/// Renders the current keyboard layout from [`KeyboardState`].
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,
})
}
}
/// Background watcher that subscribes to `activelayout>>` events emitted by
/// Hyprland's event socket.
pub struct KeyboardDaemon;
impl KeyboardDaemon {
/// Construct a new (stateless) daemon.
pub fn new() -> Self {
Self
}
/// Spawn a supervised listen loop that reconnects with a 5 s backoff.
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);
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();
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()
{
// `keyboards.last()` is the most recently registered device,
// which is typically the main one for single-keyboard setups.
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 {
// Event payload: `keyboard_name,layout_name`.
if let Some(payload) = line.strip_prefix("activelayout>>") {
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"))
}
}
+82 -21
View File
@@ -1,42 +1,42 @@
//! RAM usage renderer. Reads from the `memory` watch channel.
use crate::config::Config;
use crate::error::Result;
use crate::modules::WaybarModule;
use crate::output::WaybarOutput;
use crate::state::SharedState;
use crate::utils::{format_template, TokenValue};
use anyhow::Result;
use crate::state::AppReceivers;
use crate::utils::{TokenValue, classify_usage, format_template};
/// Renders used/total GB with usage classification for Waybar CSS.
pub struct MemoryModule;
impl WaybarModule for MemoryModule {
fn run(&self, config: &Config, state: &SharedState, _args: &[&str]) -> Result<WaybarOutput> {
async fn run(
&self,
config: &Config,
state: &AppReceivers,
_args: &[&str],
) -> Result<WaybarOutput> {
let (used_gb, total_gb) = {
if let Ok(state_lock) = state.read() {
(
state_lock.memory.used_gb,
state_lock.memory.total_gb,
)
} else {
(0.0, 0.0)
}
let s = state.memory.borrow();
(s.used_gb, s.total_gb)
};
let ratio = if total_gb > 0.0 { (used_gb / total_gb) * 100.0 } else { 0.0 };
let ratio = if total_gb > 0.0 {
(used_gb / total_gb) * 100.0
} else {
0.0
};
let text = format_template(
&config.memory.format,
&[
("used", TokenValue::Float(used_gb)),
("total", TokenValue::Float(total_gb)),
]
],
);
let class = if ratio > 95.0 {
"max"
} else if ratio > 75.0 {
"high"
} else {
"normal"
};
let class = classify_usage(ratio, 75.0, 95.0);
Ok(WaybarOutput {
text,
@@ -46,3 +46,64 @@ impl WaybarModule for MemoryModule {
})
}
}
#[cfg(test)]
mod tests {
use super::*;
use crate::state::{AppState, MemoryState, mock_state};
#[tokio::test]
async fn test_memory_normal() {
let state = mock_state(AppState {
memory: MemoryState {
used_gb: 8.0,
total_gb: 32.0,
},
..Default::default()
});
let config = Config::default();
let output = MemoryModule
.run(&config, &state.receivers, &[])
.await
.unwrap();
assert!(output.text.contains("8.00"));
assert!(output.text.contains("32.00"));
assert_eq!(output.class.as_deref(), Some("normal"));
assert_eq!(output.percentage, Some(25)); // 8/32 = 25%
}
#[tokio::test]
async fn test_memory_high() {
let state = mock_state(AppState {
memory: MemoryState {
used_gb: 26.0,
total_gb: 32.0,
},
..Default::default()
});
let config = Config::default();
let output = MemoryModule
.run(&config, &state.receivers, &[])
.await
.unwrap();
assert_eq!(output.class.as_deref(), Some("high")); // 81%
}
#[tokio::test]
async fn test_memory_zero_total() {
let state = mock_state(AppState {
memory: MemoryState {
used_gb: 0.0,
total_gb: 0.0,
},
..Default::default()
});
let config = Config::default();
let output = MemoryModule
.run(&config, &state.receivers, &[])
.await
.unwrap();
assert_eq!(output.class.as_deref(), Some("normal"));
assert_eq!(output.percentage, Some(0));
}
}
+58 -14
View File
@@ -1,23 +1,67 @@
pub mod network;
pub mod cpu;
pub mod memory;
pub mod hardware;
pub mod disk;
pub mod btrfs;
//! Waybar module implementations.
//!
//! Each submodule exposes a [`WaybarModule`] type (CPU, network, audio, …)
//! and is feature-gated by a `mod-<kind>` flag. The [`WaybarModule`] trait is
//! what [`crate::registry::dispatch`] uses to evaluate a module on demand.
//!
//! Modules come in two flavours:
//!
//! * **Watched** — the daemon spawns a background polling/event task that
//! pushes state into a `watch::Receiver`, which the module reads lock-free
//! (network, cpu, audio, bluetooth, …).
//! * **Dispatch-only** — evaluated on demand only, without a watch channel
//! (power, game, btrfs).
#[cfg(feature = "mod-audio")]
pub mod audio;
pub mod gpu;
pub mod sys;
#[cfg(feature = "mod-dbus")]
pub mod backlight;
#[cfg(feature = "mod-bt")]
pub mod bt;
pub mod buds;
pub mod power;
#[cfg(feature = "mod-hardware")]
pub mod btrfs;
#[cfg(feature = "mod-hardware")]
pub mod cpu;
#[cfg(feature = "mod-hardware")]
pub mod disk;
#[cfg(feature = "mod-dbus")]
pub mod dnd;
#[cfg(feature = "mod-hardware")]
pub mod game;
#[cfg(feature = "mod-hardware")]
pub mod gpu;
#[cfg(feature = "mod-hardware")]
pub mod hardware;
#[cfg(feature = "mod-dbus")]
pub mod keyboard;
#[cfg(feature = "mod-hardware")]
pub mod memory;
#[cfg(feature = "mod-dbus")]
pub mod mpris;
#[cfg(feature = "mod-network")]
pub mod network;
#[cfg(feature = "mod-hardware")]
pub mod power;
#[cfg(feature = "mod-hardware")]
pub mod sys;
use crate::config::Config;
use crate::error::Result as FluxoResult;
use crate::output::WaybarOutput;
use crate::state::SharedState;
use anyhow::Result;
use crate::state::AppReceivers;
/// Common interface implemented by every Waybar module.
///
/// Given the current daemon config, the shared state receivers, and any
/// caller-supplied arguments, a module produces a single [`WaybarOutput`].
/// Implementations must be cheap to evaluate — they are invoked on every
/// client request and on every signaler state change.
pub trait WaybarModule {
fn run(&self, config: &Config, state: &SharedState, args: &[&str]) -> Result<WaybarOutput>;
/// Evaluate the module and return its rendered output.
fn run(
&self,
config: &Config,
state: &AppReceivers,
args: &[&str],
) -> impl std::future::Future<Output = FluxoResult<WaybarOutput>> + Send;
}
+362
View File
@@ -0,0 +1,362 @@
//! MPRIS media player indicator.
//!
//! Subscribes to `PlaybackStatus` and `Metadata` property-changed streams on
//! the first `org.mpris.MediaPlayer2.*` name that appears on the session bus,
//! so the indicator is truly signal-driven (no 2 s polling). A 10 s heartbeat
//! verifies the player is still there. Optional marquee scrolling is driven
//! by [`mpris_scroll_ticker`] from [`crate::daemon`].
use crate::config::Config;
use crate::error::Result;
use crate::modules::WaybarModule;
use crate::output::WaybarOutput;
use crate::state::{AppReceivers, MprisScrollState, MprisState};
use crate::utils::{TokenValue, format_template};
use std::sync::Arc;
use tokio::sync::{RwLock, watch};
use tokio::time::Duration;
use tracing::{debug, info};
use zbus::{Connection, proxy};
/// Render the user's format string + derive the Waybar CSS class from state.
fn format_mpris_text(format: &str, mpris: &MprisState) -> (String, &'static str) {
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(
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())),
],
);
(text, class)
}
/// Return a cyclic `max_len`-wide window over `full_text + separator`.
fn apply_scroll_window(full_text: &str, max_len: usize, offset: usize, separator: &str) -> String {
let char_count = full_text.chars().count();
let total_len = char_count + separator.chars().count();
let offset = offset % total_len;
full_text
.chars()
.chain(separator.chars())
.cycle()
.skip(offset)
.take(max_len)
.collect()
}
/// Truncate `text` to `max_len` chars, appending `...` when cut.
fn truncate_with_ellipsis(text: &str, max_len: usize) -> String {
let char_count = text.chars().count();
if char_count <= max_len {
return text.to_string();
}
let truncated: String = text.chars().take(max_len.saturating_sub(3)).collect();
format!("{}...", truncated)
}
/// Renders the current player state, applying scroll/truncate per config.
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 (full_text, class) = format_mpris_text(&config.mpris.format, &mpris);
let text = if config.mpris.scroll {
if let Some(max_len) = config.mpris.max_length {
let scroll = state.mpris_scroll.read().await;
apply_scroll_window(
&full_text,
max_len,
scroll.offset,
&config.mpris.scroll_separator,
)
} else {
full_text.clone()
}
} else if let Some(max_len) = config.mpris.max_length {
truncate_with_ellipsis(&full_text, max_len)
} else {
full_text.clone()
};
Ok(WaybarOutput {
text,
tooltip: Some(format!("{} - {}", mpris.artist, mpris.title)),
class: Some(class.to_string()),
percentage: None,
})
}
}
/// Drive the marquee animation: advance the offset every `scroll_speed` ms
/// while a track is playing, and emit a fresh generation on `tick_tx` so the
/// mpris signaler arm fires. Resets offset when the track changes.
pub async fn mpris_scroll_ticker(
config: Arc<RwLock<Config>>,
mut mpris_rx: watch::Receiver<MprisState>,
scroll_state: Arc<RwLock<MprisScrollState>>,
tick_tx: watch::Sender<u64>,
) {
let mut generation: u64 = 0;
let mut last_track_key = String::new();
loop {
let mpris = mpris_rx.borrow_and_update().clone();
let cfg = config.read().await;
let scroll_enabled = cfg.mpris.scroll;
let has_max_length = cfg.mpris.max_length.is_some();
let scroll_speed = cfg.mpris.scroll_speed;
let format_str = cfg.mpris.format.clone();
drop(cfg);
let (full_text, _) = format_mpris_text(&format_str, &mpris);
let track_key = format!("{}|{}|{}", mpris.artist, mpris.title, mpris.album);
if track_key != last_track_key {
let mut state = scroll_state.write().await;
state.offset = 0;
state.full_text = full_text.clone();
last_track_key = track_key;
generation += 1;
let _ = tick_tx.send(generation);
}
if scroll_enabled && has_max_length && mpris.is_playing {
tokio::time::sleep(Duration::from_millis(scroll_speed)).await;
let mut state = scroll_state.write().await;
state.offset += 1;
state.full_text = full_text;
drop(state);
generation += 1;
let _ = tick_tx.send(generation);
continue;
}
// Not scrolling — sleep until the next player state change.
if mpris_rx.changed().await.is_err() {
break;
}
}
}
/// Background watcher that discovers the active MPRIS player and mirrors
/// its `PlaybackStatus` + `Metadata` properties into [`MprisState`].
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 {
/// Construct a new (stateless) daemon.
pub fn new() -> Self {
Self
}
/// Spawn a supervised listen loop with a 5 s reconnect backoff.
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<()> {
use futures::StreamExt;
let connection = Connection::session().await?;
info!("Connected to D-Bus for MPRIS monitoring");
let dbus_proxy = DBusProxy::new(&connection).await?;
loop {
let names = dbus_proxy.list_names().await?;
let active_player = names
.into_iter()
.find(|n| n.starts_with("org.mpris.MediaPlayer2."));
let Some(player_name) = active_player else {
send_stopped_if_changed(tx);
tokio::time::sleep(Duration::from_secs(5)).await;
continue;
};
let player_proxy = match MprisPlayerProxy::builder(&connection)
.destination(player_name.clone())?
.build()
.await
{
Ok(p) => p,
Err(_) => {
tokio::time::sleep(Duration::from_secs(2)).await;
continue;
}
};
update_from_player(&player_proxy, tx).await;
let mut status_stream = player_proxy.receive_playback_status_changed().await;
let mut metadata_stream = player_proxy.receive_metadata_changed().await;
loop {
tokio::select! {
Some(_) = status_stream.next() => {
update_from_player(&player_proxy, tx).await;
}
Some(_) = metadata_stream.next() => {
update_from_player(&player_proxy, tx).await;
}
_ = tokio::time::sleep(Duration::from_secs(10)) => {
// Heartbeat: re-check that the player name is still owned.
let current = dbus_proxy.list_names().await.unwrap_or_default();
if !current.iter().any(|n| n == &player_name) {
break;
}
}
}
}
}
}
}
/// Fetch `PlaybackStatus` + `Metadata` and publish only when they differ
/// from the previous [`MprisState`] (to avoid spurious watch wake-ups).
async fn update_from_player(player: &MprisPlayerProxy<'_>, tx: &watch::Sender<MprisState>) {
let status = player.playback_status().await.unwrap_or_default();
let metadata = player.metadata().await.unwrap_or_default();
let is_playing = status == "Playing";
let is_paused = status == "Paused";
let is_stopped = status == "Stopped";
let (artist, title, album) = parse_metadata(&metadata);
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);
let _ = tx.send(MprisState {
is_playing,
is_paused,
is_stopped,
artist,
title,
album,
});
}
}
/// Extract `xesam:artist` (string or array), `xesam:title`, `xesam:album`.
fn parse_metadata(
metadata: &std::collections::HashMap<String, zbus::zvariant::Value<'_>>,
) -> (String, String, String) {
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();
}
(artist, title, album)
}
/// Publish a cleared/stopped [`MprisState`] if the current state isn't already
/// that. Called when no player is on the bus.
fn send_stopped_if_changed(tx: &watch::Sender<MprisState>) {
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(),
});
}
}
+223 -92
View File
@@ -1,14 +1,25 @@
//! Primary-interface throughput renderer + polling daemon.
//!
//! The daemon picks the interface with the longest-prefix default route (see
//! [`get_primary_interface`]) and computes rx/tx rates as byte-count deltas
//! between successive polls. Well-known VPN interface prefixes get a lock
//! glyph prepended to the rendered text.
use crate::config::Config;
use crate::error::Result;
use crate::modules::WaybarModule;
use crate::output::WaybarOutput;
use crate::state::SharedState;
use crate::utils::{format_template, TokenValue};
use anyhow::Result;
use crate::state::{AppReceivers, NetworkState};
use crate::utils::{TokenValue, format_template};
use nix::ifaddrs::getifaddrs;
use std::fs;
use std::time::{SystemTime, UNIX_EPOCH};
use tokio::sync::watch;
/// Renders interface / IP / rx / tx for the detected primary route.
pub struct NetworkModule;
/// Background poller that tracks byte counters across ticks to derive rates.
pub struct NetworkDaemon {
last_time: u64,
last_rx_bytes: u64,
@@ -17,7 +28,10 @@ pub struct NetworkDaemon {
cached_ip: Option<String>,
}
type PollResult = crate::error::Result<(String, Option<String>, Option<(u64, u64)>)>;
impl NetworkDaemon {
/// Build a fresh daemon with no prior byte-count samples.
pub fn new() -> Self {
Self {
last_time: 0,
@@ -28,60 +42,116 @@ impl NetworkDaemon {
}
}
pub fn poll(&mut self, state: SharedState) {
// Cache invalidation: if the interface directory doesn't exist, clear cache
if let Some(ref iface) = self.cached_interface {
if !std::path::Path::new(&format!("/sys/class/net/{}", iface)).exists() {
self.cached_interface = None;
self.cached_ip = None;
/// Detect the primary interface, read `/sys/class/net/*/statistics`, and
/// publish a new [`NetworkState`] onto `state_tx`. Interface/byte reads
/// run via [`tokio::task::spawn_blocking`] so the runtime isn't starved.
pub async fn poll(
&mut self,
state_tx: &watch::Sender<NetworkState>,
) -> crate::error::Result<()> {
let (iface, ip_opt, bytes_opt) = tokio::task::spawn_blocking(|| -> PollResult {
let iface = get_primary_interface()?;
if iface.is_empty() {
return Ok((String::new(), None, None));
}
let ip = get_ip_address(&iface);
let bytes = get_bytes(&iface).ok();
Ok((iface, ip, bytes))
})
.await
.map_err(|e| crate::error::FluxoError::System(e.to_string()))??;
if !iface.is_empty() {
if self.cached_interface.as_ref() != Some(&iface) || self.cached_ip.is_none() {
self.cached_ip = ip_opt;
self.cached_interface = Some(iface.clone());
}
} else {
self.cached_interface = None;
self.cached_ip = None;
let mut network = state_tx.borrow().clone();
network.interface.clear();
network.ip.clear();
network.rx_mbps = 0.0;
network.tx_mbps = 0.0;
let _ = state_tx.send(network);
return Err(crate::error::FluxoError::Network(
"No primary interface found".into(),
));
}
// Re-detect interface if needed
if self.cached_interface.is_none() {
if let Ok(iface) = get_primary_interface() {
if !iface.is_empty() {
self.cached_interface = Some(iface.clone());
self.cached_ip = get_ip_address(&iface);
}
}
}
let interface = if let Some(ref interface) = self.cached_interface {
interface.clone()
} else {
let mut network = state_tx.borrow().clone();
network.interface.clear();
network.ip.clear();
network.rx_mbps = 0.0;
network.tx_mbps = 0.0;
let _ = state_tx.send(network);
return Err(crate::error::FluxoError::Network(
"Interface disappeared during poll".into(),
));
};
if let Some(ref interface) = self.cached_interface {
let time_now = SystemTime::now()
.duration_since(UNIX_EPOCH)
.unwrap_or_default()
.as_secs();
let time_now = SystemTime::now()
.duration_since(UNIX_EPOCH)
.unwrap_or_default()
.as_secs();
if let Ok((rx_bytes_now, tx_bytes_now)) = get_bytes(interface) {
if self.last_time > 0 && time_now > self.last_time {
let time_diff = time_now - self.last_time;
let rx_bps = (rx_bytes_now.saturating_sub(self.last_rx_bytes)) / time_diff;
let tx_bps = (tx_bytes_now.saturating_sub(self.last_tx_bytes)) / time_diff;
if let Some((rx_bytes_now, tx_bytes_now)) = bytes_opt {
if self.last_time > 0 && time_now > self.last_time {
let time_diff = (time_now - self.last_time) as f64;
let rx_mbps = (rx_bytes_now.saturating_sub(self.last_rx_bytes)) as f64
/ time_diff
/ 1024.0
/ 1024.0;
let tx_mbps = (tx_bytes_now.saturating_sub(self.last_tx_bytes)) as f64
/ time_diff
/ 1024.0
/ 1024.0;
let rx_mbps = (rx_bps as f64) / 1024.0 / 1024.0;
let tx_mbps = (tx_bps as f64) / 1024.0 / 1024.0;
if let Ok(mut state_lock) = state.write() {
state_lock.network.rx_mbps = rx_mbps;
state_lock.network.tx_mbps = tx_mbps;
}
}
self.last_time = time_now;
self.last_rx_bytes = rx_bytes_now;
self.last_tx_bytes = tx_bytes_now;
let mut network = state_tx.borrow().clone();
network.rx_mbps = rx_mbps;
network.tx_mbps = tx_mbps;
network.interface = interface.clone();
network.ip = self.cached_ip.clone().unwrap_or_default();
let _ = state_tx.send(network);
} else {
// Read failed, might be down
self.cached_interface = None;
// First poll has no prior sample — publish iface/ip only.
let mut network = state_tx.borrow().clone();
network.interface = interface.clone();
network.ip = self.cached_ip.clone().unwrap_or_default();
let _ = state_tx.send(network);
}
self.last_time = time_now;
self.last_rx_bytes = rx_bytes_now;
self.last_tx_bytes = tx_bytes_now;
} else {
self.cached_interface = None;
return Err(crate::error::FluxoError::Network(format!(
"Failed to read bytes for {}",
interface
)));
}
Ok(())
}
}
impl WaybarModule for NetworkModule {
fn run(&self, config: &Config, state: &SharedState, _args: &[&str]) -> Result<WaybarOutput> {
let interface = get_primary_interface()?;
async fn run(
&self,
config: &Config,
state: &AppReceivers,
_args: &[&str],
) -> Result<WaybarOutput> {
let (interface, ip, rx_mbps, tx_mbps) = {
let s = state.network.borrow();
(s.interface.clone(), s.ip.clone(), s.rx_mbps, s.tx_mbps)
};
if interface.is_empty() {
return Ok(WaybarOutput {
text: "No connection".to_string(),
@@ -91,96 +161,80 @@ impl WaybarModule for NetworkModule {
});
}
let ip = get_ip_address(&interface).unwrap_or_else(|| String::from("No IP"));
let (rx_mbps, tx_mbps) = {
if let Ok(state_lock) = state.read() {
(state_lock.network.rx_mbps, state_lock.network.tx_mbps)
} else {
(0.0, 0.0)
}
};
let ip_display = if ip.is_empty() { "No IP" } else { &ip };
let mut output_text = format_template(
&config.network.format,
&[
("interface", TokenValue::String(&interface)),
("ip", TokenValue::String(&ip)),
("interface", TokenValue::String(interface.clone())),
("ip", TokenValue::String(ip_display.to_string())),
("rx", TokenValue::Float(rx_mbps)),
("tx", TokenValue::Float(tx_mbps)),
]
],
);
if interface.starts_with("tun")
|| interface.starts_with("wg")
|| interface.starts_with("ppp")
|| interface.starts_with("pvpn")
|| interface.starts_with("proton")
|| interface.starts_with("ipsec")
{
output_text = format!(" {}", output_text);
output_text = format!(" {}", output_text);
}
Ok(WaybarOutput {
text: output_text,
tooltip: Some(format!("Interface: {}\nIP: {}", interface, ip)),
tooltip: Some(format!("Interface: {}\nIP: {}", interface, ip_display)),
class: Some(interface),
percentage: None,
})
}
}
/// Parse `/proc/net/route` to find the default-route interface. When several
/// defaults exist, prefer the one with the longest netmask, then lowest metric.
fn get_primary_interface() -> Result<String> {
let output = std::process::Command::new("ip")
.args(["route", "list"])
.output()?;
let stdout = String::from_utf8_lossy(&output.stdout);
let content = std::fs::read_to_string("/proc/net/route")?;
let mut defaults = Vec::new();
for line in stdout.lines() {
if line.starts_with("default") {
let parts: Vec<&str> = line.split_whitespace().collect();
let mut dev = "";
let mut metric = 0;
for i in 0..parts.len() {
if parts[i] == "dev" && i + 1 < parts.len() {
dev = parts[i + 1];
}
if parts[i] == "metric" && i + 1 < parts.len() {
metric = parts[i + 1].parse::<i32>().unwrap_or(0);
}
}
if !dev.is_empty() {
defaults.push((metric, dev.to_string()));
for line in content.lines().skip(1) {
let parts: Vec<&str> = line.split_whitespace().collect();
if parts.len() >= 8 {
let iface = parts[0];
let dest = parts[1];
let metric = parts[6].parse::<i32>().unwrap_or(0);
let mask = u32::from_str_radix(parts[7], 16).unwrap_or(0);
if dest == "00000000" {
defaults.push((mask, metric, iface.to_string()));
}
}
}
defaults.sort_by_key(|k| k.0);
if let Some((_, dev)) = defaults.first() {
defaults.sort_by(|a, b| b.0.cmp(&a.0).then(a.1.cmp(&b.1)));
if let Some((_, _, dev)) = defaults.first() {
Ok(dev.clone())
} else {
Ok(String::new())
}
}
/// First IPv4 address for `interface`, via `getifaddrs`. `None` if absent.
fn get_ip_address(interface: &str) -> Option<String> {
let output = std::process::Command::new("ip")
.args(["-4", "addr", "show", interface])
.output()
.ok()?;
let stdout = String::from_utf8_lossy(&output.stdout);
for line in stdout.lines() {
if line.trim().starts_with("inet ") {
let parts: Vec<&str> = line.trim().split_whitespace().collect();
if parts.len() > 1 {
let ip_cidr = parts[1];
let ip = ip_cidr.split('/').next().unwrap_or(ip_cidr);
return Some(ip.to_string());
}
let addrs = getifaddrs().ok()?;
for ifaddr in addrs {
if ifaddr.interface_name == interface
&& let Some(address) = ifaddr.address
&& let Some(sockaddr) = address.as_sockaddr_in()
{
return Some(sockaddr.ip().to_string());
}
}
None
}
/// Read `(rx_bytes, tx_bytes)` counters from sysfs for `interface`.
fn get_bytes(interface: &str) -> Result<(u64, u64)> {
let rx_path = format!("/sys/class/net/{}/statistics/rx_bytes", interface);
let tx_path = format!("/sys/class/net/{}/statistics/tx_bytes", interface);
@@ -198,3 +252,80 @@ fn get_bytes(interface: &str) -> Result<(u64, u64)> {
Ok((rx, tx))
}
#[cfg(test)]
mod tests {
use super::*;
use crate::state::{AppState, NetworkState, mock_state};
#[tokio::test]
async fn test_network_no_connection() {
let state = mock_state(AppState::default());
let config = Config::default();
let output = NetworkModule
.run(&config, &state.receivers, &[])
.await
.unwrap();
assert_eq!(output.text, "No connection");
}
#[tokio::test]
async fn test_network_connected() {
let state = mock_state(AppState {
network: NetworkState {
rx_mbps: 1.5,
tx_mbps: 0.3,
interface: "eth0".to_string(),
ip: "192.168.1.100".to_string(),
},
..Default::default()
});
let config = Config::default();
let output = NetworkModule
.run(&config, &state.receivers, &[])
.await
.unwrap();
assert!(output.text.contains("eth0"));
assert!(output.text.contains("192.168.1.100"));
assert!(output.text.contains("1.50"));
assert_eq!(output.class.as_deref(), Some("eth0"));
}
#[tokio::test]
async fn test_network_vpn_prefix() {
let state = mock_state(AppState {
network: NetworkState {
rx_mbps: 0.0,
tx_mbps: 0.0,
interface: "wg0".to_string(),
ip: "10.0.0.1".to_string(),
},
..Default::default()
});
let config = Config::default();
let output = NetworkModule
.run(&config, &state.receivers, &[])
.await
.unwrap();
assert!(output.text.starts_with(" "));
}
#[tokio::test]
async fn test_network_no_ip() {
let state = mock_state(AppState {
network: NetworkState {
rx_mbps: 0.0,
tx_mbps: 0.0,
interface: "eth0".to_string(),
ip: String::new(),
},
..Default::default()
});
let config = Config::default();
let output = NetworkModule
.run(&config, &state.receivers, &[])
.await
.unwrap();
assert!(output.text.contains("No IP"));
}
}
+24 -15
View File
@@ -1,19 +1,27 @@
//! Battery/AC indicator via `/sys/class/power_supply`. Dispatch-only - reads
//! sysfs on demand rather than polling into a watch channel.
use crate::config::Config;
use crate::error::Result;
use crate::modules::WaybarModule;
use crate::output::WaybarOutput;
use crate::state::SharedState;
use crate::utils::{format_template, TokenValue};
use anyhow::Result;
use crate::state::AppReceivers;
use crate::utils::{TokenValue, format_template};
use std::fs;
/// Renders battery percentage + charge state (critical/warning/bat/charging/ac).
pub struct PowerModule;
impl WaybarModule for PowerModule {
fn run(&self, config: &Config, _state: &SharedState, _args: &[&str]) -> Result<WaybarOutput> {
async fn run(
&self,
config: &Config,
_state: &AppReceivers,
_args: &[&str],
) -> Result<WaybarOutput> {
let critical_threshold = 15;
let warning_threshold = 50;
// Find the first battery
let mut battery_path = None;
if let Ok(entries) = fs::read_dir("/sys/class/power_supply") {
for entry in entries.flatten() {
@@ -25,18 +33,17 @@ impl WaybarModule for PowerModule {
}
}
// Check AC status
let mut ac_online = false;
if let Ok(entries) = fs::read_dir("/sys/class/power_supply") {
for entry in entries.flatten() {
let name = entry.file_name().to_string_lossy().to_string();
if name.starts_with("AC") || name.starts_with("ADP") {
let online_path = entry.path().join("online");
if let Ok(online_str) = fs::read_to_string(online_path) {
if online_str.trim() == "1" {
ac_online = true;
break;
}
if let Ok(online_str) = fs::read_to_string(online_path)
&& online_str.trim() == "1"
{
ac_online = true;
break;
}
}
}
@@ -60,9 +67,11 @@ impl WaybarModule for PowerModule {
}
};
let capacity_str = fs::read_to_string(bat_path.join("capacity")).unwrap_or_else(|_| "0".to_string());
let capacity_str =
fs::read_to_string(bat_path.join("capacity")).unwrap_or_else(|_| "0".to_string());
let percentage: u8 = capacity_str.trim().parse().unwrap_or(0);
let status_str = fs::read_to_string(bat_path.join("status")).unwrap_or_else(|_| "Unknown".to_string());
let status_str =
fs::read_to_string(bat_path.join("status")).unwrap_or_else(|_| "Unknown".to_string());
let state = status_str.trim().to_lowercase();
let (icon, class, tooltip) = if state == "charging" || ac_online {
@@ -94,8 +103,8 @@ impl WaybarModule for PowerModule {
&config.power.format,
&[
("percentage", TokenValue::Int(percentage as i64)),
("icon", TokenValue::String(icon)),
]
("icon", TokenValue::String(icon.to_string())),
],
);
Ok(WaybarOutput {
+22 -17
View File
@@ -1,26 +1,31 @@
//! Uptime + load average renderer. Reads from the `sys` watch channel.
use crate::config::Config;
use crate::error::Result;
use crate::modules::WaybarModule;
use crate::output::WaybarOutput;
use crate::state::SharedState;
use crate::utils::{format_template, TokenValue};
use anyhow::Result;
use crate::state::AppReceivers;
use crate::utils::{TokenValue, format_template};
/// Renders uptime and load averages with a detailed tooltip.
pub struct SysModule;
impl WaybarModule for SysModule {
fn run(&self, config: &Config, state: &SharedState, _args: &[&str]) -> Result<WaybarOutput> {
async fn run(
&self,
config: &Config,
state: &AppReceivers,
_args: &[&str],
) -> Result<WaybarOutput> {
let (load1, load5, load15, uptime_secs, process_count) = {
if let Ok(state_lock) = state.read() {
(
state_lock.sys.load_1,
state_lock.sys.load_5,
state_lock.sys.load_15,
state_lock.sys.uptime,
state_lock.sys.process_count,
)
} else {
(0.0, 0.0, 0.0, 0, 0)
}
let state_lock = state.sys.borrow();
(
state_lock.load_1,
state_lock.load_5,
state_lock.load_15,
state_lock.uptime,
state_lock.process_count,
)
};
let hours = uptime_secs / 3600;
@@ -34,11 +39,11 @@ impl WaybarModule for SysModule {
let text = format_template(
&config.sys.format,
&[
("uptime", TokenValue::String(&uptime_str)),
("uptime", TokenValue::String(uptime_str.clone())),
("load1", TokenValue::Float(load1)),
("load5", TokenValue::Float(load5)),
("load15", TokenValue::Float(load15)),
]
],
);
Ok(WaybarOutput {
+139 -2
View File
@@ -1,12 +1,149 @@
use serde::Serialize;
//! JSON payload returned to Waybar custom modules, plus client-side
//! output formatting utilities.
#[derive(Serialize)]
use serde::{Deserialize, Serialize};
/// Waybar renders in a proportional font — replacing normal spaces with
/// figure-spaces (U+2007) keeps column widths stable across updates.
pub const FIGURE_SPACE: char = '\u{2007}';
/// Zero-width space used as cosmetic padding around module text so Waybar
/// doesn't clip leading/trailing glyphs.
pub const ZERO_WIDTH_SPACE: char = '\u{200B}';
/// A Waybar custom module return value.
///
/// Serialises to the schema Waybar's `return-type: json` expects — the
/// optional fields are omitted from the output when unset.
#[derive(Serialize, Deserialize, Clone, Debug, Default)]
pub struct WaybarOutput {
/// Primary text shown in the bar.
pub text: String,
/// Tooltip text shown on hover.
#[serde(skip_serializing_if = "Option::is_none")]
pub tooltip: Option<String>,
/// CSS class applied to the module (for styling).
#[serde(skip_serializing_if = "Option::is_none")]
pub class: Option<String>,
/// Optional 0-100 value usable by bar progress indicators.
#[serde(skip_serializing_if = "Option::is_none")]
pub percentage: Option<u8>,
}
impl WaybarOutput {
/// A blank output for disabled modules.
pub fn disabled() -> Self {
Self {
text: String::new(),
tooltip: Some("Module disabled".to_string()),
class: Some("disabled".to_string()),
percentage: None,
}
}
/// A user-visible error with tooltip detail.
pub fn error(message: &str) -> Self {
Self {
text: format!("{}Error{}", ZERO_WIDTH_SPACE, ZERO_WIDTH_SPACE),
tooltip: Some(message.to_string()),
class: Some("error".to_string()),
percentage: None,
}
}
}
/// Apply Waybar font-stabilisation to a text string.
///
/// Replaces normal spaces with figure-spaces (unless the string contains
/// markup), and wraps in zero-width spaces for cosmetic padding.
pub fn stabilize_text(text: &str) -> String {
let processed = if text.contains('<') {
text.to_string()
} else {
text.replace(' ', &FIGURE_SPACE.to_string())
};
format!("{}{}{}", ZERO_WIDTH_SPACE, processed, ZERO_WIDTH_SPACE)
}
/// Process an IPC response and print Waybar-compatible JSON to stdout.
///
/// On IPC failure, prints a "Daemon offline" error output and exits
/// non-zero so Waybar surfaces the problem visually.
pub fn print_waybar_response(response: anyhow::Result<String>) {
match response {
Ok(json_str) => match serde_json::from_str::<serde_json::Value>(&json_str) {
Ok(mut val) => {
if let Some(text) = val.get("text").and_then(|t| t.as_str()) {
val["text"] = serde_json::Value::String(stabilize_text(text));
}
println!("{}", serde_json::to_string(&val).unwrap());
}
Err(_) => println!("{}", json_str),
},
Err(e) => {
let err_out = WaybarOutput {
text: format!(
"{}Daemon offline ({}){}",
ZERO_WIDTH_SPACE, e, ZERO_WIDTH_SPACE
),
tooltip: Some(e.to_string()),
class: Some("error".to_string()),
percentage: None,
};
println!("{}", serde_json::to_string(&err_out).unwrap());
std::process::exit(1);
}
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_full_output_serialization() {
let output = WaybarOutput {
text: "CPU: 50%".to_string(),
tooltip: Some("Details".to_string()),
class: Some("normal".to_string()),
percentage: Some(50),
};
let json = serde_json::to_string(&output).unwrap();
let val: serde_json::Value = serde_json::from_str(&json).unwrap();
assert_eq!(val["text"], "CPU: 50%");
assert_eq!(val["tooltip"], "Details");
assert_eq!(val["class"], "normal");
assert_eq!(val["percentage"], 50);
}
#[test]
fn test_optional_fields_omitted() {
let output = WaybarOutput {
text: "test".to_string(),
tooltip: None,
class: None,
percentage: None,
};
let json = serde_json::to_string(&output).unwrap();
let val: serde_json::Value = serde_json::from_str(&json).unwrap();
assert_eq!(val["text"], "test");
assert!(val.get("tooltip").is_none());
assert!(val.get("class").is_none());
assert!(val.get("percentage").is_none());
}
#[test]
fn test_partial_optional_fields() {
let output = WaybarOutput {
text: "test".to_string(),
tooltip: Some("tip".to_string()),
class: None,
percentage: Some(75),
};
let json = serde_json::to_string(&output).unwrap();
let val: serde_json::Value = serde_json::from_str(&json).unwrap();
assert_eq!(val["tooltip"], "tip");
assert!(val.get("class").is_none());
assert_eq!(val["percentage"], 75);
}
}
+85
View File
@@ -0,0 +1,85 @@
//! Central module dispatch: name → [`WaybarModule::run`] lookup.
//!
//! The `dispatch` function is called by the IPC request handler with a module
//! name (e.g. `"cpu"`, `"vol"`) and returns the rendered [`WaybarOutput`]. All
//! per-module match arms are generated by [`for_each_watched_module!`];
//! dispatch-only modules without a watch channel (power/game/btrfs) are
//! listed manually.
use crate::config::Config;
use crate::error::{FluxoError, Result as FluxoResult};
use crate::output::WaybarOutput;
use crate::state::AppReceivers;
#[allow(unused_imports)]
use crate::modules::WaybarModule;
/// Expands the module registry into a single [`dispatch`] match and a
/// [`signaler_default_args`] lookup.
macro_rules! gen_dispatch {
($( { $feature:literal, $field:ident, $state:ty, [$($name:literal),+], [$($sig_name:literal),+], $module:path, $signal:ident, [$($default_arg:literal),*], $config:ident } )*) => {
/// Look up a module by name and render its [`WaybarOutput`].
///
/// Returns [`FluxoError::Disabled`] when the module is disabled in
/// config, and [`FluxoError::Ipc`] when the name is unknown.
pub async fn dispatch(
module_name: &str,
#[allow(unused)] config: &Config,
#[allow(unused)] state: &AppReceivers,
#[allow(unused)] args: &[&str],
) -> FluxoResult<WaybarOutput> {
if !config.is_module_enabled(module_name) {
return Err(FluxoError::Disabled(module_name.to_string()));
}
match module_name {
$(
#[cfg(feature = $feature)]
$($name)|+ => {
$module.run(config, state, args).await
}
)*
// Dispatch-only modules (no watch channel)
#[cfg(feature = "mod-audio")]
"mic" => {
crate::modules::audio::AudioModule
.run(config, state, args)
.await
}
#[cfg(feature = "mod-hardware")]
"power" => {
crate::modules::power::PowerModule
.run(config, state, args)
.await
}
#[cfg(feature = "mod-hardware")]
"game" => {
crate::modules::game::GameModule
.run(config, state, args)
.await
}
#[cfg(feature = "mod-hardware")]
"pool" | "btrfs" => {
crate::modules::btrfs::BtrfsModule
.run(config, state, args)
.await
}
_ => Err(FluxoError::Ipc(format!("Unknown module: {}", module_name))),
}
}
/// Returns the default args used by the signaler when evaluating a module.
pub fn signaler_default_args(module_name: &str) -> &'static [&'static str] {
match module_name {
$(
$($name)|+ => &[$($default_arg),*],
)*
"mic" => &["source", "show"],
_ => &[],
}
}
};
}
for_each_watched_module!(gen_dispatch);
+163
View File
@@ -0,0 +1,163 @@
//! Waybar signalling: watch state channels, send `SIGRTMIN+N` on changes.
//!
//! Waybar's custom modules use `signal = N` to rerun their command when they
//! receive `SIGRTMIN+N`. This task subscribes to every watched module's
//! `watch::Receiver`, evaluates the module when its channel fires, and only
//! signals Waybar when the rendered output actually changed. A 50 ms per-signal
//! debounce prevents storms during rapid state churn.
use crate::config::Config;
use crate::state::AppReceivers;
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};
/// Sends real-time signals to the Waybar process.
///
/// Resolves Waybar's PID lazily and caches it — the PID is invalidated on
/// signal failure (e.g. Waybar was restarted) and rediscovered via `sysinfo`.
pub struct WaybarSignaler {
cached_pid: Option<i32>,
sys: System,
last_signal_sent: HashMap<i32, Instant>,
}
impl WaybarSignaler {
/// Create a new signaler with no cached PID.
pub fn new() -> Self {
Self {
cached_pid: None,
sys: System::new(),
last_signal_sent: HashMap::new(),
}
}
fn find_waybar_pid(&mut self) -> Option<i32> {
self.sys.refresh_processes(ProcessesToUpdate::All, true);
for (pid, process) in self.sys.processes() {
if process.name() == "waybar" {
return Some(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
&& unsafe { libc::kill(pid, 0) } == 0
{
valid_pid = true;
}
if !valid_pid {
self.cached_pid = self.find_waybar_pid();
}
if let Some(pid) = self.cached_pid {
let sig = libc::SIGRTMIN() + signal_num;
if unsafe { libc::kill(pid, sig) } == 0 {
debug!("Sent SIGRTMIN+{} to waybar (PID: {})", signal_num, pid);
self.last_signal_sent.insert(signal_num, Instant::now());
} else {
warn!("Failed to send SIGRTMIN+{} to waybar", signal_num);
self.cached_pid = None;
}
} else {
debug!("Waybar process not found, skipping signal.");
}
}
}
/// Generates [`WaybarSignaler::run`] from the central module registry.
///
/// For each watched module we emit:
/// * a cfg-gated `watch::Receiver::changed()` future (or `pending::<()>` when
/// the feature is disabled, so the `tokio::select!` arm compiles uniformly),
/// * a `select!` arm that re-evaluates the module and signals Waybar once per
/// changed output.
macro_rules! gen_signaler_run {
($( { $feature:literal, $field:ident, $state:ty, [$($name:literal),+], [$($sig_name:literal),+], $module:path, $signal:ident, [$($default_arg:literal),*], $config:ident } )*) => {
impl WaybarSignaler {
/// Run the signaler event loop forever.
///
/// Consumes `self`; intended to be spawned as a long-lived task.
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);
}
}
}
};
}
// Generate cfg-gated futures for each watched module
$(
#[cfg(not(feature = $feature))]
let $field = std::future::pending::<
std::result::Result<(), tokio::sync::watch::error::RecvError>,
>();
#[cfg(feature = $feature)]
let $field = receivers.$field.changed();
)*
// MPRIS scroll tick (special case — not a watched module)
#[cfg(not(feature = "mod-dbus"))]
let mpris_scroll_tick = std::future::pending::<
std::result::Result<(), tokio::sync::watch::error::RecvError>,
>();
#[cfg(feature = "mod-dbus")]
let mpris_scroll_tick = receivers.mpris_scroll_tick.changed();
tokio::select! {
$(
res = $field, if signals.$signal.is_some() => {
if res.is_ok() {
$(check_and_signal!($sig_name, signals.$signal);)+
}
}
)*
// MPRIS scroll tick (separate from mpris data changes)
res = mpris_scroll_tick, if signals.mpris.is_some() => {
if res.is_ok()
&& let Some(sig) = signals.mpris { self.send_signal(sig); }
}
_ = sleep(Duration::from_secs(5)) => {
// heartbeat: re-read signals config on each iteration
}
}
}
}
}
};
}
for_each_watched_module!(gen_signaler_run);
+318 -9
View File
@@ -1,20 +1,126 @@
use std::sync::{Arc, RwLock};
//! Shared state types exchanged between daemon tasks and modules.
//!
//! Every "watched" module has a `watch::Receiver<StateType>` held by
//! [`AppReceivers`]; background tasks write into the paired sender. Readers
//! (the request handler, the signaler) take a cheap snapshot via
//! `Receiver::borrow()` and render their output from it.
#[derive(Default, Clone)]
pub struct AppState {
pub network: NetworkState,
pub cpu: CpuState,
pub memory: MemoryState,
pub sys: SysState,
pub gpu: GpuState,
use crate::output::WaybarOutput;
use std::collections::HashMap;
use std::sync::Arc;
use tokio::sync::{RwLock, mpsc, watch};
use tokio::time::Instant;
/// Generates the [`AppReceivers`] struct, with one `watch::Receiver` field
/// per watched module (cfg-gated by feature flag).
macro_rules! gen_app_receivers {
($( { $feature:literal, $field:ident, $state:ty, [$($name:literal),+], [$($sig_name:literal),+], $module:path, $signal:ident, [$($default_arg:literal),*], $config:ident } )*) => {
/// Bundle of every watch-channel receiver and control-channel handle.
///
/// Cloned into each IPC request handler and each module daemon so that
/// they can snapshot state without locking. Per-module receivers are
/// generated from [`for_each_watched_module!`]; the remaining fields
/// (`health`, `bt_cycle`, `audio_cmd_tx`, `mpris_scroll_*`) are shared
/// control state.
#[derive(Clone)]
pub struct AppReceivers {
$(
#[cfg(feature = $feature)]
pub $field: watch::Receiver<$state>,
)*
#[cfg(feature = "mod-bt")]
pub bt_cycle: Arc<RwLock<usize>>,
#[cfg(feature = "mod-dbus")]
pub mpris_scroll: Arc<RwLock<MprisScrollState>>,
#[cfg(feature = "mod-dbus")]
pub mpris_scroll_tick: watch::Receiver<u64>,
pub health: Arc<RwLock<HashMap<String, ModuleHealth>>>,
#[cfg(feature = "mod-bt")]
pub bt_force_poll: mpsc::Sender<()>,
#[cfg(feature = "mod-audio")]
pub audio_cmd_tx: mpsc::Sender<crate::modules::audio::AudioCommand>,
}
};
}
for_each_watched_module!(gen_app_receivers);
/// Runtime health statistics used for backoff decisions.
#[derive(Clone, Default)]
pub struct ModuleHealth {
pub consecutive_failures: u32,
pub last_failure: Option<Instant>,
pub backoff_until: Option<Instant>,
pub last_successful_output: Option<WaybarOutput>,
}
/// Current default audio sink + source plus all available devices.
#[derive(Default, Clone)]
pub struct AudioState {
pub sink: AudioDeviceInfo,
pub source: AudioDeviceInfo,
pub available_sinks: Vec<String>,
pub available_sources: Vec<String>,
}
/// Metadata for one PulseAudio sink or source.
#[derive(Default, Clone)]
pub struct AudioDeviceInfo {
pub name: String,
pub description: String,
pub volume: u8,
pub muted: bool,
pub channels: u8,
}
/// A single BlueZ device, optionally annotated by a plugin (buds, maestro…).
#[derive(Default, Clone)]
pub struct BtDeviceInfo {
pub device_alias: String,
pub device_address: String,
pub battery_percentage: Option<u8>,
pub plugin_data: Vec<(String, String)>,
}
/// Snapshot of the Bluetooth adapter plus every tracked device.
#[derive(Default, Clone)]
pub struct BtState {
pub adapter_powered: bool,
pub devices: Vec<BtDeviceInfo>,
}
impl BtState {
/// Return the device the client is currently cycled to, if any.
///
/// The `index` is taken modulo `devices.len()` so cycling past the end
/// wraps around naturally.
pub fn active_device(&self, index: usize) -> Option<&BtDeviceInfo> {
if self.devices.is_empty() {
None
} else {
Some(&self.devices[index % self.devices.len()])
}
}
}
/// Per-mountpoint disk usage.
#[derive(Default, Clone)]
pub struct DiskInfo {
pub mount_point: String,
pub filesystem: String,
pub total_bytes: u64,
pub available_bytes: u64,
}
/// Throughput and identity of the active network interface.
#[derive(Default, Clone)]
pub struct NetworkState {
pub rx_mbps: f64,
pub tx_mbps: f64,
pub interface: String,
pub ip: String,
}
/// CPU utilisation and temperature.
#[derive(Clone)]
pub struct CpuState {
pub usage: f64,
@@ -32,12 +138,14 @@ impl Default for CpuState {
}
}
/// RAM usage in gigabytes.
#[derive(Default, Clone)]
pub struct MemoryState {
pub used_gb: f64,
pub total_gb: f64,
}
/// Load averages, uptime, and process count.
#[derive(Default, Clone)]
pub struct SysState {
pub load_1: f64,
@@ -47,6 +155,9 @@ pub struct SysState {
pub process_count: usize,
}
/// GPU snapshot (vendor-agnostic).
///
/// `active` is `false` until detection finds a supported GPU.
#[derive(Clone)]
pub struct GpuState {
pub active: bool,
@@ -72,4 +183,202 @@ impl Default for GpuState {
}
}
pub type SharedState = Arc<RwLock<AppState>>;
/// Do-Not-Disturb toggle state.
#[derive(Default, Clone)]
pub struct DndState {
pub is_dnd: bool,
}
/// Currently active keyboard layout.
#[derive(Default, Clone)]
pub struct KeyboardState {
pub layout: String,
}
/// Display backlight brightness as a 0-100 percentage.
#[derive(Default, Clone)]
pub struct BacklightState {
pub percentage: u8,
}
/// Marquee scroll position for the MPRIS module.
#[derive(Default, Clone)]
pub struct MprisScrollState {
pub offset: usize,
pub full_text: String,
}
/// Currently playing media metadata from an MPRIS player.
#[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,
}
/// Test harness holding a synthetic [`AppReceivers`] plus its senders.
///
/// The senders are kept alive via `_*_tx` fields so test code can drive the
/// watch channels without them being dropped.
#[cfg(test)]
pub struct MockState {
pub receivers: AppReceivers,
#[cfg(feature = "mod-network")]
_net_tx: watch::Sender<NetworkState>,
#[cfg(feature = "mod-hardware")]
_cpu_tx: watch::Sender<CpuState>,
#[cfg(feature = "mod-hardware")]
_mem_tx: watch::Sender<MemoryState>,
#[cfg(feature = "mod-hardware")]
_sys_tx: watch::Sender<SysState>,
#[cfg(feature = "mod-hardware")]
_gpu_tx: watch::Sender<GpuState>,
#[cfg(feature = "mod-hardware")]
_disks_tx: watch::Sender<Vec<DiskInfo>>,
#[cfg(feature = "mod-bt")]
_bt_tx: watch::Sender<BtState>,
#[cfg(feature = "mod-audio")]
_audio_tx: watch::Sender<AudioState>,
#[cfg(feature = "mod-dbus")]
_mpris_tx: watch::Sender<MprisState>,
#[cfg(feature = "mod-dbus")]
_backlight_tx: watch::Sender<BacklightState>,
#[cfg(feature = "mod-dbus")]
_keyboard_tx: watch::Sender<KeyboardState>,
#[cfg(feature = "mod-dbus")]
_dnd_tx: watch::Sender<DndState>,
}
/// Plain-data counterpart of [`AppReceivers`] used to seed a [`MockState`].
#[cfg(test)]
#[derive(Default, Clone)]
pub struct AppState {
#[cfg(feature = "mod-network")]
pub network: NetworkState,
#[cfg(feature = "mod-hardware")]
pub cpu: CpuState,
#[cfg(feature = "mod-hardware")]
pub memory: MemoryState,
#[cfg(feature = "mod-hardware")]
pub sys: SysState,
#[cfg(feature = "mod-hardware")]
pub gpu: GpuState,
#[cfg(feature = "mod-hardware")]
pub disks: Vec<DiskInfo>,
#[cfg(feature = "mod-bt")]
pub bluetooth: BtState,
#[cfg(feature = "mod-audio")]
pub audio: AudioState,
#[cfg(feature = "mod-dbus")]
pub mpris: MprisState,
#[cfg(feature = "mod-dbus")]
pub backlight: BacklightState,
#[cfg(feature = "mod-dbus")]
pub keyboard: KeyboardState,
#[cfg(feature = "mod-dbus")]
pub dnd: DndState,
pub health: HashMap<String, ModuleHealth>,
}
/// Build a [`MockState`] from a plain [`AppState`] snapshot for unit tests.
#[cfg(test)]
pub fn mock_state(state: AppState) -> MockState {
#[cfg(feature = "mod-network")]
let (net_tx, net_rx) = watch::channel(state.network);
#[cfg(feature = "mod-hardware")]
let (cpu_tx, cpu_rx) = watch::channel(state.cpu);
#[cfg(feature = "mod-hardware")]
let (mem_tx, mem_rx) = watch::channel(state.memory);
#[cfg(feature = "mod-hardware")]
let (sys_tx, sys_rx) = watch::channel(state.sys);
#[cfg(feature = "mod-hardware")]
let (gpu_tx, gpu_rx) = watch::channel(state.gpu);
#[cfg(feature = "mod-hardware")]
let (disks_tx, disks_rx) = watch::channel(state.disks);
#[cfg(feature = "mod-bt")]
let (bt_tx, bt_rx) = watch::channel(state.bluetooth);
#[cfg(feature = "mod-audio")]
let (audio_tx, audio_rx) = watch::channel(state.audio);
#[cfg(feature = "mod-dbus")]
let (mpris_tx, mpris_rx) = watch::channel(state.mpris);
#[cfg(feature = "mod-dbus")]
let (backlight_tx, backlight_rx) = watch::channel(state.backlight);
#[cfg(feature = "mod-dbus")]
let (keyboard_tx, keyboard_rx) = watch::channel(state.keyboard);
#[cfg(feature = "mod-dbus")]
let (dnd_tx, dnd_rx) = watch::channel(state.dnd);
#[cfg(feature = "mod-bt")]
let (bt_force_tx, _) = mpsc::channel(1);
#[cfg(feature = "mod-audio")]
let (audio_cmd_tx, _) = mpsc::channel(1);
MockState {
receivers: AppReceivers {
#[cfg(feature = "mod-network")]
network: net_rx,
#[cfg(feature = "mod-hardware")]
cpu: cpu_rx,
#[cfg(feature = "mod-hardware")]
memory: mem_rx,
#[cfg(feature = "mod-hardware")]
sys: sys_rx,
#[cfg(feature = "mod-hardware")]
gpu: gpu_rx,
#[cfg(feature = "mod-hardware")]
disks: disks_rx,
#[cfg(feature = "mod-bt")]
bluetooth: bt_rx,
#[cfg(feature = "mod-bt")]
bt_cycle: Arc::new(RwLock::new(0usize)),
#[cfg(feature = "mod-audio")]
audio: audio_rx,
#[cfg(feature = "mod-dbus")]
mpris: mpris_rx,
#[cfg(feature = "mod-dbus")]
backlight: backlight_rx,
#[cfg(feature = "mod-dbus")]
keyboard: keyboard_rx,
#[cfg(feature = "mod-dbus")]
dnd: dnd_rx,
#[cfg(feature = "mod-dbus")]
mpris_scroll: Arc::new(RwLock::new(MprisScrollState::default())),
#[cfg(feature = "mod-dbus")]
mpris_scroll_tick: {
let (_, rx) = watch::channel(0u64);
rx
},
health: Arc::new(RwLock::new(state.health)),
#[cfg(feature = "mod-bt")]
bt_force_poll: bt_force_tx,
#[cfg(feature = "mod-audio")]
audio_cmd_tx,
},
#[cfg(feature = "mod-network")]
_net_tx: net_tx,
#[cfg(feature = "mod-hardware")]
_cpu_tx: cpu_tx,
#[cfg(feature = "mod-hardware")]
_mem_tx: mem_tx,
#[cfg(feature = "mod-hardware")]
_sys_tx: sys_tx,
#[cfg(feature = "mod-hardware")]
_gpu_tx: gpu_tx,
#[cfg(feature = "mod-hardware")]
_disks_tx: disks_tx,
#[cfg(feature = "mod-bt")]
_bt_tx: bt_tx,
#[cfg(feature = "mod-audio")]
_audio_tx: audio_tx,
#[cfg(feature = "mod-dbus")]
_mpris_tx: mpris_tx,
#[cfg(feature = "mod-dbus")]
_backlight_tx: backlight_tx,
#[cfg(feature = "mod-dbus")]
_keyboard_tx: keyboard_tx,
#[cfg(feature = "mod-dbus")]
_dnd_tx: dnd_tx,
}
}
+226 -38
View File
@@ -1,24 +1,38 @@
//! Shared helpers: menu spawning, Hyprland socket lookup, and template formatting.
use anyhow::{Context, Result};
use std::io::Write;
use std::process::{Command, Stdio};
/// Pipe `items` into an external menu program (rofi/dmenu/wofi) and return
/// the user's selection.
///
/// The prompt is exposed to the command as `$FLUXO_PROMPT` (preferred) and
/// as a legacy `{prompt}` placeholder that is substituted in the shell string.
pub fn show_menu(prompt: &str, items: &[String], menu_cmd: &str) -> Result<String> {
let cmd_str = menu_cmd.replace("{prompt}", prompt);
let mut child = Command::new("sh")
.arg("-c")
.arg(&cmd_str)
.env("FLUXO_PROMPT", prompt)
.stdin(Stdio::piped())
.stdout(Stdio::piped())
// Suppress GTK/Wayland noise from tools like wofi.
.stderr(Stdio::null())
.spawn()
.context("Failed to spawn menu command")?;
if let Some(mut stdin) = child.stdin.take() {
let input = items.join("\n");
stdin.write_all(input.as_bytes()).context("Failed to write to menu stdin")?;
let mut input = items.join("\n");
// Trailing newline is required by wofi/rofi.
input.push('\n');
stdin
.write_all(input.as_bytes())
.context("Failed to write to menu stdin")?;
}
let output = child.wait_with_output().context("Failed to wait on menu")?;
if !output.status.success() {
return Err(anyhow::anyhow!("Menu cancelled or failed"));
}
@@ -31,64 +45,238 @@ pub fn show_menu(prompt: &str, items: &[String], menu_cmd: &str) -> Result<Strin
Ok(selected)
}
/// Resolve the path to a Hyprland IPC socket for the current session.
///
/// Looks under `$XDG_RUNTIME_DIR/hypr/$HYPRLAND_INSTANCE_SIGNATURE/` first,
/// then falls back to `/tmp/hypr/...` for older Hyprland builds.
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")?;
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);
}
}
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;
pub enum TokenValue<'a> {
Float(f64),
Int(i64),
String(&'a str),
/// Bucket `value` into `"normal"`, `"high"`, or `"max"` for CSS class output.
pub fn classify_usage(value: f64, high: f64, max: f64) -> &'static str {
if value > max {
"max"
} else if value > high {
"high"
} else {
"normal"
}
}
pub fn format_template(template: &str, values: &[(&str, TokenValue)]) -> String {
static RE: LazyLock<Regex> = LazyLock::new(|| {
Regex::new(r"\{([a-zA-Z0-9_]+)(?::([<>\^])?(\d+)?(?:\.(\d+))?)?\}").unwrap()
});
/// A typed value supplied to [`format_template`] — chosen at call site so
/// formatting handles precision and alignment correctly per type.
pub enum TokenValue {
Float(f64),
Int(i64),
String(String),
}
RE.replace_all(template, |caps: &regex::Captures| {
let name = &caps[1];
if let Some((_, val)) = values.iter().find(|(k, _)| *k == name) {
let align = caps.get(2).map(|m| m.as_str()).unwrap_or(">");
let width = caps.get(3).map(|m| m.as_str().parse::<usize>().unwrap_or(0)).unwrap_or(0);
let precision = caps.get(4).map(|m| m.as_str().parse::<usize>().unwrap_or(0));
/// Token grammar: `{name}`, `{name:>5}`, `{name:<8.2}`, `{name:^6}`, etc.
pub static TOKEN_RE: LazyLock<Regex> =
LazyLock::new(|| Regex::new(r"\{([a-zA-Z0-9_]+)(?::([<>\^])?(\d+)?(?:\.(\d+))?)?\}").unwrap());
match val {
TokenValue::Float(f) => format_float(*f, align, width, precision),
TokenValue::Int(i) => format_int(*i, align, width),
TokenValue::String(s) => format_str(s, align, width),
/// Substitute `{name[:align[width[.precision]]]}` tokens in a template string.
///
/// Unknown tokens are left verbatim. Width/alignment/precision follow Rust's
/// `std::fmt` semantics (`<` left, `^` center, `>` right).
pub fn format_template<K>(template: &str, values: &[(K, TokenValue)]) -> String
where
K: AsRef<str>,
{
TOKEN_RE
.replace_all(template, |caps: &regex::Captures| {
let name = &caps[1];
if let Some((_, val)) = values.iter().find(|(k, _)| k.as_ref() == name) {
let align = caps.get(2).map(|m| m.as_str()).unwrap_or(">");
let width = caps
.get(3)
.map(|m| m.as_str().parse::<usize>().unwrap_or(0))
.unwrap_or(0);
let precision = caps
.get(4)
.map(|m| m.as_str().parse::<usize>().unwrap_or(0));
match val {
TokenValue::Float(f) => format_float(*f, align, width, precision),
TokenValue::Int(i) => format_int(*i, align, width),
TokenValue::String(s) => format_str(s, align, width),
}
} else {
caps[0].to_string()
}
} else {
caps[0].to_string()
}
}).into_owned()
})
.into_owned()
}
fn format_float(f: f64, align: &str, width: usize, precision: Option<usize>) -> String {
match (align, precision) {
("<", Some(p)) => format!("{:<width$.p$}", f, width=width, p=p),
("^", Some(p)) => format!("{:^width$.p$}", f, width=width, p=p),
(">", Some(p)) => format!("{:>width$.p$}", f, width=width, p=p),
("<", None) => format!("{:<width$}", f, width=width),
("^", None) => format!("{:^width$}", f, width=width),
(">", None) => format!("{:>width$}", f, width=width),
("<", Some(p)) => format!("{:<width$.p$}", f, width = width, p = p),
("^", Some(p)) => format!("{:^width$.p$}", f, width = width, p = p),
(">", Some(p)) => format!("{:>width$.p$}", f, width = width, p = p),
("<", None) => format!("{:<width$}", f, width = width),
("^", None) => format!("{:^width$}", f, width = width),
(">", None) => format!("{:>width$}", f, width = width),
_ => format!("{}", f),
}
}
fn format_int(i: i64, align: &str, width: usize) -> String {
match align {
"<" => format!("{:<width$}", i, width=width),
"^" => format!("{:^width$}", i, width=width),
">" => format!("{:>width$}", i, width=width),
"<" => format!("{:<width$}", i, width = width),
"^" => format!("{:^width$}", i, width = width),
">" => format!("{:>width$}", i, width = width),
_ => format!("{}", i),
}
}
fn format_str(s: &str, align: &str, width: usize) -> String {
match align {
"<" => format!("{:<width$}", s, width=width),
"^" => format!("{:^width$}", s, width=width),
">" => format!("{:>width$}", s, width=width),
_ => format!("{}", s),
"<" => format!("{:<width$}", s, width = width),
"^" => format!("{:^width$}", s, width = width),
">" => format!("{:>width$}", s, width = width),
_ => s.to_string(),
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_simple_string_token() {
let result = format_template(
"{name}",
&[("name", TokenValue::String("hello".to_string()))],
);
assert_eq!(result, "hello");
}
#[test]
fn test_simple_float_token() {
let result = format_template("{val}", &[("val", TokenValue::Float(3.15))]);
assert_eq!(result, "3.15");
}
#[test]
fn test_simple_int_token() {
let result = format_template("{count}", &[("count", TokenValue::Int(42))]);
assert_eq!(result, "42");
}
#[test]
fn test_float_right_align_with_precision() {
let result = format_template("{val:>8.2}", &[("val", TokenValue::Float(3.15))]);
assert_eq!(result, " 3.15");
assert_eq!(result.len(), 8);
}
#[test]
fn test_float_left_align_with_precision() {
let result = format_template("{val:<8.2}", &[("val", TokenValue::Float(3.15))]);
assert_eq!(result, "3.15 ");
assert_eq!(result.len(), 8);
}
#[test]
fn test_float_center_align_with_precision() {
let result = format_template("{val:^8.2}", &[("val", TokenValue::Float(3.15))]);
assert_eq!(result, " 3.15 ");
assert_eq!(result.len(), 8);
}
#[test]
fn test_int_right_align() {
let result = format_template("{val:>5}", &[("val", TokenValue::Int(42))]);
assert_eq!(result, " 42");
}
#[test]
fn test_string_left_align() {
let result = format_template(
"{val:<10}",
&[("val", TokenValue::String("hi".to_string()))],
);
assert_eq!(result, "hi ");
assert_eq!(result.len(), 10);
}
#[test]
fn test_unknown_token_preserved() {
let result = format_template(
"{unknown}",
&[("name", TokenValue::String("test".to_string()))],
);
assert_eq!(result, "{unknown}");
}
#[test]
fn test_multiple_tokens() {
let result = format_template(
"CPU: {usage:>4.1}% {temp:>4.1}C",
&[
("usage", TokenValue::Float(55.3)),
("temp", TokenValue::Float(65.0)),
],
);
assert_eq!(result, "CPU: 55.3% 65.0C");
}
#[test]
fn test_no_tokens() {
let result = format_template::<&'static str>("plain text", &[]);
assert_eq!(result, "plain text");
}
#[test]
fn test_empty_template() {
let result = format_template("", &[("x", TokenValue::Int(1))]);
assert_eq!(result, "");
}
#[test]
fn test_mixed_token_types() {
let result = format_template(
"{name} ({ip}): {rx:>5.2} MB/s",
&[
("name", TokenValue::String("eth0".to_string())),
("ip", TokenValue::String("10.0.0.1".to_string())),
("rx", TokenValue::Float(1.5)),
],
);
assert_eq!(result, "eth0 (10.0.0.1): 1.50 MB/s");
}
#[test]
fn test_float_precision_zero() {
let result = format_template("{val:>3.0}", &[("val", TokenValue::Float(99.7))]);
assert_eq!(result, "100");
}
}