Compare commits
2 Commits
c8a50d7b67
...
a932e6b422
| Author | SHA1 | Date | |
|---|---|---|---|
| a932e6b422 | |||
| 81c9b78cb3 |
Generated
+1
-1
@@ -353,7 +353,7 @@ checksum = "1d674e81391d1e1ab681a28d99df07927c6d4aa5b027d7da16ba32d1d21ecd99"
|
|||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "fluxo-rs"
|
name = "fluxo-rs"
|
||||||
version = "0.3.0"
|
version = "0.3.1"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"anyhow",
|
"anyhow",
|
||||||
"bluer",
|
"bluer",
|
||||||
|
|||||||
@@ -1,95 +1,66 @@
|
|||||||
# fluxo-rs
|
# fluxo-rs
|
||||||
|
|
||||||
fluxo-rs is a high-performance system metrics daemon and client designed specifically for waybar. It replaces standard shell scripts with a compiled rust binary that collects data via a background polling loop and serves it over a unix domain socket (`/tmp/fluxo.sock`).
|
fluxo-rs is a high-performance system metrics daemon and client designed specifically for Waybar. It replaces standard shell scripts with a compiled Rust binary that collects data via a background polling loop and serves it over a Unix socket.
|
||||||
|
|
||||||
## Description
|
## Key Features
|
||||||
|
|
||||||
The project follows a client-server architecture:
|
- **Asynchronous Architecture**: Built on **Tokio**, the daemon handles concurrent IPC requests and background tasks with zero latency and minimal CPU overhead.
|
||||||
- **Daemon**: Handles heavy lifting (polling cpu, memory, network, gpu) and stores state in memory.
|
- **Native Library Integrations**:
|
||||||
- **Client**: A thin cli wrapper that connects to the daemon's socket to retrieve formatted json for waybar.
|
- **Audio**: Direct `libpulse` integration for event-driven, instant volume and device updates.
|
||||||
|
- **Bluetooth**: Native `bluer` integration for robust device monitoring.
|
||||||
This approach eliminates process spawning overhead and temporary file locking, resulting in near-zero cpu usage for custom modules.
|
- **Pixel Buds Pro**: Custom native RPC implementation for real-time battery and ANC control.
|
||||||
|
- **Network**: Native `nix` and `/proc` inspection for high-speed interface monitoring.
|
||||||
## Features
|
- **Hyprland**: Direct IPC Unix socket communication for gamemode and animation status.
|
||||||
|
- **Circuit Breaker (Failsafe)**: Automatically detects failing modules and enters a "Cool down" state to prevent resource waste and log spam.
|
||||||
- **Ultra-lightweight**: Background polling is highly optimized (e.g., O(1) process counting).
|
- **Multi-threaded Polling**: Decoupled subsystem threads ensure that a hang in one system (e.g., a slow GPU probe) never freezes your entire bar.
|
||||||
- **Jitter-free**: Uses zero-width sentinels and figure spaces to prevent waybar from trimming padding.
|
|
||||||
- **Configurable**: Fully customizable output formats via toml config.
|
|
||||||
- **Interactive Menus**: Integrated support for selecting items (like Bluetooth devices) via external menus (e.g., Rofi, Wofi, Fuzzel).
|
|
||||||
- **Live Reload**: Configuration can be reloaded without restarting the daemon.
|
|
||||||
- **Multi-vendor GPU**: Native support for intel (igpu), amd, and nvidia.
|
|
||||||
|
|
||||||
## Modules
|
## Modules
|
||||||
|
|
||||||
| Command | Description | Tokens |
|
| Command | Description | Tokens |
|
||||||
| :--- | :--- | :--- |
|
| :--- | :--- | :--- |
|
||||||
| `net` | Network speed (rx/tx) | `{interface}`, `{ip}`, `{rx}`, `{tx}` |
|
| `cpu` | CPU usage and temperature | `{usage}`, `{temp}`, `{model}` |
|
||||||
| `cpu` | CPU usage and temp | `{usage}`, `{temp}` |
|
|
||||||
| `mem` | Memory usage | `{used}`, `{total}` |
|
| `mem` | Memory usage | `{used}`, `{total}` |
|
||||||
| `gpu` | GPU utilization | `{usage}`, `{vram_used}`, `{vram_total}`, `{temp}` |
|
| `net` | Network status & speeds | `{interface}`, `{ip}`, `{rx}`, `{tx}` |
|
||||||
| `sys` | System load and uptime | `{uptime}`, `{load1}`, `{load5}`, `{load15}` |
|
| `sys` | System load and uptime | `{uptime}`, `{load1}`, `{load5}`, `{load15}`, `{procs}` |
|
||||||
| `disk` | Disk usage (default: /) | `{mount}`, `{used}`, `{total}` |
|
| `disk` | Disk usage | `{mount}`, `{used}`, `{total}` |
|
||||||
| `pool` | Aggregate storage (btrfs) | `{used}`, `{total}` |
|
| `pool` | Btrfs aggregate storage | `{used}`, `{total}` |
|
||||||
| `vol` | Audio output volume | `{name}`, `{volume}`, `{icon}` |
|
| `vol` | Audio output (sink) | `{name}`, `{volume}`, `{icon}` |
|
||||||
| `mic` | Audio input volume | `{name}`, `{volume}`, `{icon}` |
|
| `mic` | Audio input (source) | `{name}`, `{volume}`, `{icon}` |
|
||||||
| `bt` | Bluetooth status & plugins | `{alias}`, `{mac}`, `{left}`, `{right}`, `{anc}` |
|
| `bt` | Bluetooth status & plugins | `{alias}`, `{mac}`, `{left}`, `{right}`, `{anc}` |
|
||||||
| `power` | Battery and AC status | `{percentage}`, `{icon}` |
|
| `power` | Battery and AC status | `{percentage}`, `{icon}` |
|
||||||
| `game` | Hyprland gamemode status | active/inactive icon strings |
|
| `game` | Hyprland status | active/inactive icons |
|
||||||
|
|
||||||
## Setup
|
## Setup
|
||||||
|
|
||||||
1. Build the project:
|
1. **Build**: `cargo build --release`
|
||||||
```bash
|
2. **Configure**: Create `~/.config/fluxo/config.toml` (see `example.config.toml`).
|
||||||
cd fluxo-rs
|
3. **Daemon**: Start `fluxo-rs daemon`. It's recommended to run this as a systemd user service.
|
||||||
cargo build --release
|
|
||||||
```
|
|
||||||
|
|
||||||
2. Start the daemon:
|
## Waybar Configuration
|
||||||
```bash
|
|
||||||
# Starts the daemon using the default config path (~/.config/fluxo/config.toml)
|
|
||||||
./target/release/fluxo-rs daemon &
|
|
||||||
|
|
||||||
# Or provide a custom path
|
To achieve zero-latency updates, use **Waybar Signals**:
|
||||||
./target/release/fluxo-rs daemon --config /path/to/your/config.toml &
|
|
||||||
```
|
|
||||||
|
|
||||||
3. Configuration:
|
```jsonc
|
||||||
Create `~/.config/fluxo/config.toml` (see `example.config.toml` for all default options).
|
"custom/audio": {
|
||||||
|
"exec": "fluxo vol",
|
||||||
4. Waybar configuration (`config.jsonc`):
|
"return-type": "json",
|
||||||
```json
|
"interval": 5,
|
||||||
"custom/cpu": {
|
"signal": 8,
|
||||||
"exec": "~/path/to/fluxo-rs cpu",
|
"on-click": "fluxo audio cycle sink && pkill -RTMIN+8 waybar"
|
||||||
"return-type": "json"
|
},
|
||||||
}
|
"custom/bluetooth": {
|
||||||
```
|
"exec": "fluxo bt",
|
||||||
|
"return-type": "json",
|
||||||
## Development
|
"interval": 5,
|
||||||
|
"signal": 9,
|
||||||
### Architecture
|
"on-click": "fluxo bt menu && pkill -RTMIN+9 waybar",
|
||||||
- `src/main.rs`: Entry point, CLI parsing, interactive GUI spawns (menus), and client-side formatting logic.
|
"on-click-right": "fluxo bt cycle_mode && pkill -RTMIN+9 waybar"
|
||||||
- `src/daemon.rs`: UDS listener, configuration management, and polling orchestration.
|
}
|
||||||
- `src/ipc.rs`: Unix domain socket communication logic.
|
|
||||||
- `src/utils.rs`: Generic GUI utilities (like the menu spawner).
|
|
||||||
- `src/modules/`: Individual metric implementations.
|
|
||||||
- `src/state.rs`: Shared thread-safe data structures.
|
|
||||||
|
|
||||||
### Adding a Module
|
|
||||||
1. Add the required config block to `src/config.rs`.
|
|
||||||
2. Add the required state fields to `src/state.rs`.
|
|
||||||
3. Implement the `WaybarModule` trait in a new file in `src/modules/`.
|
|
||||||
4. Add polling logic to `src/modules/hardware.rs` or `src/daemon.rs`.
|
|
||||||
5. Register the new subcommand in `src/main.rs` and the router in `src/daemon.rs`.
|
|
||||||
|
|
||||||
### Configuration Reload
|
|
||||||
The daemon can reload its configuration live:
|
|
||||||
```bash
|
|
||||||
fluxo-rs reload
|
|
||||||
```
|
```
|
||||||
|
|
||||||
### Logging & Debugging
|
## Debugging
|
||||||
`fluxo-rs` uses the `tracing` ecosystem. If a module isn't behaving properly or your configuration isn't applying, start the daemon with debug logging enabled in the foreground:
|
|
||||||
|
Start the daemon with `RUST_LOG=debug` to see detailed logs of library interactions and circuit breaker status:
|
||||||
```bash
|
```bash
|
||||||
RUST_LOG=debug fluxo-rs daemon
|
RUST_LOG=debug fluxo-rs daemon
|
||||||
```
|
```
|
||||||
This will print verbose information regarding config parsing errors, subprocess failures, and IPC requests.
|
|
||||||
|
|||||||
+13
-13
@@ -5,7 +5,7 @@ use std::path::PathBuf;
|
|||||||
use std::sync::LazyLock;
|
use std::sync::LazyLock;
|
||||||
use tracing::{debug, info, warn};
|
use tracing::{debug, info, warn};
|
||||||
|
|
||||||
#[derive(Deserialize, Default)]
|
#[derive(Deserialize, Default, Clone)]
|
||||||
pub struct Config {
|
pub struct Config {
|
||||||
#[serde(default)]
|
#[serde(default)]
|
||||||
pub general: GeneralConfig,
|
pub general: GeneralConfig,
|
||||||
@@ -33,7 +33,7 @@ pub struct Config {
|
|||||||
pub game: GameConfig,
|
pub game: GameConfig,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Deserialize)]
|
#[derive(Deserialize, Clone)]
|
||||||
pub struct GeneralConfig {
|
pub struct GeneralConfig {
|
||||||
pub menu_command: String,
|
pub menu_command: String,
|
||||||
}
|
}
|
||||||
@@ -46,7 +46,7 @@ impl Default for GeneralConfig {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Deserialize)]
|
#[derive(Deserialize, Clone)]
|
||||||
pub struct NetworkConfig {
|
pub struct NetworkConfig {
|
||||||
pub format: String,
|
pub format: String,
|
||||||
}
|
}
|
||||||
@@ -59,7 +59,7 @@ impl Default for NetworkConfig {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Deserialize)]
|
#[derive(Deserialize, Clone)]
|
||||||
pub struct CpuConfig {
|
pub struct CpuConfig {
|
||||||
pub format: String,
|
pub format: String,
|
||||||
}
|
}
|
||||||
@@ -72,7 +72,7 @@ impl Default for CpuConfig {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Deserialize)]
|
#[derive(Deserialize, Clone)]
|
||||||
pub struct MemoryConfig {
|
pub struct MemoryConfig {
|
||||||
pub format: String,
|
pub format: String,
|
||||||
}
|
}
|
||||||
@@ -85,7 +85,7 @@ impl Default for MemoryConfig {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Deserialize)]
|
#[derive(Deserialize, Clone)]
|
||||||
pub struct GpuConfig {
|
pub struct GpuConfig {
|
||||||
pub format_amd: String,
|
pub format_amd: String,
|
||||||
pub format_intel: String,
|
pub format_intel: String,
|
||||||
@@ -104,7 +104,7 @@ impl Default for GpuConfig {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Deserialize)]
|
#[derive(Deserialize, Clone)]
|
||||||
pub struct SysConfig {
|
pub struct SysConfig {
|
||||||
pub format: String,
|
pub format: String,
|
||||||
}
|
}
|
||||||
@@ -117,7 +117,7 @@ impl Default for SysConfig {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Deserialize)]
|
#[derive(Deserialize, Clone)]
|
||||||
pub struct DiskConfig {
|
pub struct DiskConfig {
|
||||||
pub format: String,
|
pub format: String,
|
||||||
}
|
}
|
||||||
@@ -130,7 +130,7 @@ impl Default for DiskConfig {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Deserialize)]
|
#[derive(Deserialize, Clone)]
|
||||||
pub struct PoolConfig {
|
pub struct PoolConfig {
|
||||||
pub format: String,
|
pub format: String,
|
||||||
}
|
}
|
||||||
@@ -143,7 +143,7 @@ impl Default for PoolConfig {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Deserialize)]
|
#[derive(Deserialize, Clone)]
|
||||||
pub struct PowerConfig {
|
pub struct PowerConfig {
|
||||||
pub format: String,
|
pub format: String,
|
||||||
}
|
}
|
||||||
@@ -156,7 +156,7 @@ impl Default for PowerConfig {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Deserialize)]
|
#[derive(Deserialize, Clone)]
|
||||||
pub struct AudioConfig {
|
pub struct AudioConfig {
|
||||||
pub format_sink_unmuted: String,
|
pub format_sink_unmuted: String,
|
||||||
pub format_sink_muted: String,
|
pub format_sink_muted: String,
|
||||||
@@ -175,7 +175,7 @@ impl Default for AudioConfig {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Deserialize)]
|
#[derive(Deserialize, Clone)]
|
||||||
pub struct BtConfig {
|
pub struct BtConfig {
|
||||||
pub format_connected: String,
|
pub format_connected: String,
|
||||||
pub format_plugin: String,
|
pub format_plugin: String,
|
||||||
@@ -194,7 +194,7 @@ impl Default for BtConfig {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Deserialize)]
|
#[derive(Deserialize, Clone)]
|
||||||
pub struct GameConfig {
|
pub struct GameConfig {
|
||||||
pub format_active: String,
|
pub format_active: String,
|
||||||
pub format_inactive: String,
|
pub format_inactive: String,
|
||||||
|
|||||||
+3
-1
@@ -89,11 +89,13 @@ pub async fn run_daemon(config_path: Option<PathBuf>) -> Result<()> {
|
|||||||
|
|
||||||
// 4. Bluetooth Task
|
// 4. Bluetooth Task
|
||||||
let poll_state = Arc::clone(&state);
|
let poll_state = Arc::clone(&state);
|
||||||
|
let poll_config = Arc::clone(&config);
|
||||||
tokio::spawn(async move {
|
tokio::spawn(async move {
|
||||||
info!("Starting Bluetooth polling task");
|
info!("Starting Bluetooth polling task");
|
||||||
let mut daemon = BtDaemon::new();
|
let mut daemon = BtDaemon::new();
|
||||||
loop {
|
loop {
|
||||||
daemon.poll(Arc::clone(&poll_state)).await;
|
let config = poll_config.read().await;
|
||||||
|
daemon.poll(Arc::clone(&poll_state), &config).await;
|
||||||
sleep(Duration::from_secs(1)).await;
|
sleep(Duration::from_secs(1)).await;
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
|
|||||||
@@ -1,606 +0,0 @@
|
|||||||
use crate::config::Config;
|
|
||||||
use crate::error::Result as FluxoResult;
|
|
||||||
use crate::modules::WaybarModule;
|
|
||||||
use crate::output::WaybarOutput;
|
|
||||||
use crate::state::{BtState, SharedState};
|
|
||||||
use crate::utils::{TokenValue, format_template};
|
|
||||||
use anyhow::{Context, Result};
|
|
||||||
use futures::StreamExt;
|
|
||||||
use std::collections::HashMap;
|
|
||||||
use std::process::Command;
|
|
||||||
use std::sync::{Arc, LazyLock, Mutex};
|
|
||||||
use tokio::sync::mpsc;
|
|
||||||
use tokio::time::{Duration, Instant};
|
|
||||||
use tracing::{debug, error, info, warn};
|
|
||||||
|
|
||||||
// Maestro imports
|
|
||||||
#[allow(unused_imports)]
|
|
||||||
use maestro::protocol::codec::Codec;
|
|
||||||
#[allow(unused_imports)]
|
|
||||||
use maestro::pwrpc::client::{Client, ClientHandle};
|
|
||||||
#[allow(unused_imports)]
|
|
||||||
use maestro::service::MaestroService;
|
|
||||||
#[allow(unused_imports)]
|
|
||||||
use maestro::service::settings::{self, Setting, SettingValue};
|
|
||||||
|
|
||||||
#[derive(Clone, Default)]
|
|
||||||
struct BudsStatus {
|
|
||||||
left_battery: Option<u8>,
|
|
||||||
right_battery: Option<u8>,
|
|
||||||
case_battery: Option<u8>,
|
|
||||||
anc_state: String,
|
|
||||||
#[allow(dead_code)]
|
|
||||||
last_update: Option<Instant>,
|
|
||||||
error: Option<String>,
|
|
||||||
}
|
|
||||||
|
|
||||||
enum BudsCommand {
|
|
||||||
SetAnc(String),
|
|
||||||
}
|
|
||||||
|
|
||||||
enum ManagerCommand {
|
|
||||||
EnsureTask(String),
|
|
||||||
SendCommand(String, BudsCommand),
|
|
||||||
}
|
|
||||||
|
|
||||||
struct MaestroManager {
|
|
||||||
statuses: Arc<Mutex<HashMap<String, BudsStatus>>>,
|
|
||||||
management_tx: mpsc::UnboundedSender<ManagerCommand>,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl MaestroManager {
|
|
||||||
fn new() -> Self {
|
|
||||||
let (tx, mut rx) = mpsc::unbounded_channel::<ManagerCommand>();
|
|
||||||
let statuses = Arc::new(Mutex::new(HashMap::new()));
|
|
||||||
let statuses_clone = Arc::clone(&statuses);
|
|
||||||
|
|
||||||
// Start dedicated BT management thread
|
|
||||||
std::thread::spawn(move || {
|
|
||||||
let rt = tokio::runtime::Builder::new_current_thread()
|
|
||||||
.enable_all()
|
|
||||||
.build()
|
|
||||||
.unwrap();
|
|
||||||
|
|
||||||
let local = tokio::task::LocalSet::new();
|
|
||||||
|
|
||||||
local.block_on(&rt, async move {
|
|
||||||
let mut command_txs: HashMap<String, mpsc::Sender<BudsCommand>> = HashMap::new();
|
|
||||||
|
|
||||||
loop {
|
|
||||||
tokio::select! {
|
|
||||||
Some(cmd) = rx.recv() => {
|
|
||||||
match cmd {
|
|
||||||
ManagerCommand::EnsureTask(mac) => {
|
|
||||||
if !command_txs.contains_key(&mac) {
|
|
||||||
let (tx, buds_rx) = mpsc::channel::<BudsCommand>(10);
|
|
||||||
command_txs.insert(mac.clone(), tx);
|
|
||||||
|
|
||||||
let mac_clone = mac.clone();
|
|
||||||
let st_clone = Arc::clone(&statuses_clone);
|
|
||||||
|
|
||||||
tokio::task::spawn_local(async move {
|
|
||||||
if let Err(e) = buds_task(&mac_clone, st_clone, buds_rx).await {
|
|
||||||
error!("Buds task for {} failed: {}", mac_clone, e);
|
|
||||||
}
|
|
||||||
});
|
|
||||||
}
|
|
||||||
}
|
|
||||||
ManagerCommand::SendCommand(mac, buds_cmd) => {
|
|
||||||
if let Some(tx) = command_txs.get(&mac) {
|
|
||||||
let _ = tx.try_send(buds_cmd);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
_ = tokio::time::sleep(Duration::from_millis(100)) => {
|
|
||||||
// Cleanup dropped tasks if needed
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
Self {
|
|
||||||
statuses,
|
|
||||||
management_tx: tx,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn get_status(&self, mac: &str) -> BudsStatus {
|
|
||||||
let statuses = self.statuses.lock().unwrap();
|
|
||||||
statuses.get(mac).cloned().unwrap_or_default()
|
|
||||||
}
|
|
||||||
|
|
||||||
fn ensure_task(&self, mac: &str) {
|
|
||||||
let _ = self
|
|
||||||
.management_tx
|
|
||||||
.send(ManagerCommand::EnsureTask(mac.to_string()));
|
|
||||||
}
|
|
||||||
|
|
||||||
fn send_command(&self, mac: &str, cmd: BudsCommand) -> Result<()> {
|
|
||||||
self.ensure_task(mac);
|
|
||||||
let _ = self
|
|
||||||
.management_tx
|
|
||||||
.send(ManagerCommand::SendCommand(mac.to_string(), cmd));
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn buds_task(
|
|
||||||
mac: &str,
|
|
||||||
statuses: Arc<Mutex<HashMap<String, BudsStatus>>>,
|
|
||||||
mut rx: mpsc::Receiver<BudsCommand>,
|
|
||||||
) -> Result<()> {
|
|
||||||
info!("Starting native Maestro connection task for {}", mac);
|
|
||||||
|
|
||||||
loop {
|
|
||||||
let addr: bluer::Address = mac.parse().context("Failed to parse MAC address")?;
|
|
||||||
let session = bluer::Session::new()
|
|
||||||
.await
|
|
||||||
.context("Failed to create bluer session")?;
|
|
||||||
let adapter = session
|
|
||||||
.default_adapter()
|
|
||||||
.await
|
|
||||||
.context("Failed to get default adapter")?;
|
|
||||||
let device = adapter
|
|
||||||
.device(addr)
|
|
||||||
.context("Failed to get device handle")?;
|
|
||||||
|
|
||||||
if !device.is_connected().await.unwrap_or(false) {
|
|
||||||
debug!("Device {} not connected to BT, stopping maestro task", mac);
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Connect to Maestro RFCOMM service
|
|
||||||
// We try channel 1 then 2, which covers most Pixel Buds variants.
|
|
||||||
let mut stream = None;
|
|
||||||
for channel in [1, 2] {
|
|
||||||
let socket = match bluer::rfcomm::Socket::new() {
|
|
||||||
Ok(s) => s,
|
|
||||||
Err(e) => {
|
|
||||||
error!("Failed to create RFCOMM socket: {}", e);
|
|
||||||
return Err(e.into());
|
|
||||||
}
|
|
||||||
};
|
|
||||||
let target = bluer::rfcomm::SocketAddr::new(addr, channel);
|
|
||||||
debug!(
|
|
||||||
"Trying to connect RFCOMM to {} on channel {}...",
|
|
||||||
mac, channel
|
|
||||||
);
|
|
||||||
match socket.connect(target).await {
|
|
||||||
Ok(s) => {
|
|
||||||
stream = Some(s);
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
Err(e) => {
|
|
||||||
debug!("Failed to connect to channel {}: {}", channel, e);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
let stream = match stream {
|
|
||||||
Some(s) => s,
|
|
||||||
None => {
|
|
||||||
warn!(
|
|
||||||
"Failed to connect RFCOMM to {} on any common channel. Retrying in 15s...",
|
|
||||||
mac
|
|
||||||
);
|
|
||||||
tokio::time::sleep(Duration::from_secs(15)).await;
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
info!("Connected Maestro RFCOMM to {} on channel", mac);
|
|
||||||
|
|
||||||
// Initialize Maestro communication stack
|
|
||||||
let codec = Codec::new();
|
|
||||||
let stream = codec.wrap(stream);
|
|
||||||
let mut client = Client::new(stream);
|
|
||||||
let handle = client.handle();
|
|
||||||
|
|
||||||
// Resolve Maestro channel (typically 1 or 2)
|
|
||||||
let channel = match maestro::protocol::utils::resolve_channel(&mut client).await {
|
|
||||||
Ok(c) => c,
|
|
||||||
Err(e) => {
|
|
||||||
error!("Failed to resolve Maestro channel for {}: {}", mac, e);
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
// Run client in background to handle RPC packets
|
|
||||||
tokio::spawn(async move {
|
|
||||||
if let Err(e) = client.run().await {
|
|
||||||
error!("Maestro client loop error: {}", e);
|
|
||||||
}
|
|
||||||
});
|
|
||||||
|
|
||||||
let mut service = MaestroService::new(handle, channel);
|
|
||||||
|
|
||||||
// Query initial ANC state
|
|
||||||
if let Ok(val) = service
|
|
||||||
.read_setting_var(settings::SettingId::CurrentAncrState)
|
|
||||||
.await
|
|
||||||
&& let SettingValue::CurrentAncrState(anc_state) = val
|
|
||||||
{
|
|
||||||
let mut status = MAESTRO.get_status(mac);
|
|
||||||
status.anc_state = anc_state_to_string(&anc_state);
|
|
||||||
statuses.lock().unwrap().insert(mac.to_string(), status);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Subscribe to real-time status updates (battery, ANC, wear)
|
|
||||||
let mut call = match service.subscribe_to_runtime_info() {
|
|
||||||
Ok(c) => c,
|
|
||||||
Err(e) => {
|
|
||||||
error!("Failed to subscribe to runtime info for {}: {}", mac, e);
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
let mut runtime_info = call.stream();
|
|
||||||
|
|
||||||
debug!("Subscribed to runtime info for {}", mac);
|
|
||||||
|
|
||||||
loop {
|
|
||||||
tokio::select! {
|
|
||||||
cmd = rx.recv() => {
|
|
||||||
match cmd {
|
|
||||||
Some(BudsCommand::SetAnc(mode)) => {
|
|
||||||
debug!("Setting ANC mode to {} for {}", mode, mac);
|
|
||||||
let state = mode_to_anc_state(&mode);
|
|
||||||
let val = SettingValue::CurrentAncrState(state);
|
|
||||||
if let Err(e) = service.write_setting(val).await {
|
|
||||||
error!("Failed to write ANC setting for {}: {}", mac, e);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
None => return Ok(()),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Some(Ok(info)) = runtime_info.next() => {
|
|
||||||
let mut status = MAESTRO.get_status(mac);
|
|
||||||
status.last_update = Some(Instant::now());
|
|
||||||
|
|
||||||
if let Some(bat) = info.battery_info {
|
|
||||||
status.left_battery = bat.left.map(|b| b.level as u8);
|
|
||||||
status.right_battery = bat.right.map(|b| b.level as u8);
|
|
||||||
status.case_battery = bat.case.map(|b| b.level as u8);
|
|
||||||
}
|
|
||||||
|
|
||||||
statuses.lock().unwrap().insert(mac.to_string(), status);
|
|
||||||
}
|
|
||||||
_ = tokio::time::sleep(Duration::from_secs(30)) => {
|
|
||||||
// Check if still connected to BT
|
|
||||||
if !device.is_connected().await.unwrap_or(false) {
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if !device.is_connected().await.unwrap_or(false) {
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
|
|
||||||
fn mode_to_anc_state(mode: &str) -> settings::AncState {
|
|
||||||
match mode {
|
|
||||||
"active" => settings::AncState::Active,
|
|
||||||
"aware" => settings::AncState::Aware,
|
|
||||||
"off" => settings::AncState::Off,
|
|
||||||
_ => settings::AncState::Off,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn anc_state_to_string(state: &settings::AncState) -> String {
|
|
||||||
match state {
|
|
||||||
settings::AncState::Active => "active".to_string(),
|
|
||||||
settings::AncState::Aware => "aware".to_string(),
|
|
||||||
settings::AncState::Off => "off".to_string(),
|
|
||||||
_ => "unknown".to_string(),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
static MAESTRO: LazyLock<MaestroManager> = LazyLock::new(MaestroManager::new);
|
|
||||||
|
|
||||||
pub struct BtDaemon {
|
|
||||||
session: Option<bluer::Session>,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl BtDaemon {
|
|
||||||
pub fn new() -> Self {
|
|
||||||
Self { session: None }
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn poll(&mut self, state: SharedState) {
|
|
||||||
if let Err(e) = self.poll_async(state).await {
|
|
||||||
error!("BT daemon error: {}", e);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn poll_async(&mut self, state: SharedState) -> Result<()> {
|
|
||||||
if self.session.is_none() {
|
|
||||||
self.session = Some(bluer::Session::new().await?);
|
|
||||||
}
|
|
||||||
let session = self.session.as_ref().unwrap();
|
|
||||||
let adapter = session.default_adapter().await?;
|
|
||||||
let adapter_powered = adapter.is_powered().await.unwrap_or(false);
|
|
||||||
|
|
||||||
let mut bt_state = BtState {
|
|
||||||
adapter_powered,
|
|
||||||
..Default::default()
|
|
||||||
};
|
|
||||||
|
|
||||||
if adapter_powered {
|
|
||||||
let devices = adapter.device_addresses().await?;
|
|
||||||
for addr in devices {
|
|
||||||
let device = adapter.device(addr)?;
|
|
||||||
if device.is_connected().await.unwrap_or(false) {
|
|
||||||
let uuids = device.uuids().await?.unwrap_or_default();
|
|
||||||
// Audio sink UUID (0x110b)
|
|
||||||
let audio_sink_uuid =
|
|
||||||
bluer::Uuid::from_u128(0x0000110b_0000_1000_8000_00805f9b34fb);
|
|
||||||
if uuids.contains(&audio_sink_uuid) {
|
|
||||||
bt_state.connected = true;
|
|
||||||
bt_state.device_address = addr.to_string();
|
|
||||||
bt_state.device_alias =
|
|
||||||
device.alias().await.unwrap_or_else(|_| addr.to_string());
|
|
||||||
bt_state.battery_percentage =
|
|
||||||
device.battery_percentage().await.unwrap_or(None);
|
|
||||||
|
|
||||||
// Plugin detection
|
|
||||||
for p in PLUGINS.iter() {
|
|
||||||
if p.can_handle(&bt_state.device_alias, &bt_state.device_address) {
|
|
||||||
match p.get_data(&bt_state.device_address) {
|
|
||||||
Ok(data) => {
|
|
||||||
bt_state.plugin_data = data
|
|
||||||
.into_iter()
|
|
||||||
.map(|(k, v)| {
|
|
||||||
let val_str = match v {
|
|
||||||
TokenValue::String(s) => s,
|
|
||||||
TokenValue::Int(i) => i.to_string(),
|
|
||||||
TokenValue::Float(f) => format!("{:.1}", f),
|
|
||||||
};
|
|
||||||
(k, val_str)
|
|
||||||
})
|
|
||||||
.collect();
|
|
||||||
}
|
|
||||||
Err(e) => {
|
|
||||||
warn!("Plugin {} failed for {}: {}", p.name(), addr, e);
|
|
||||||
bt_state
|
|
||||||
.plugin_data
|
|
||||||
.push(("plugin_error".to_string(), e.to_string()));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
let mut lock = state.write().await;
|
|
||||||
lock.bluetooth = bt_state;
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub trait BtPlugin: Send + Sync {
|
|
||||||
fn name(&self) -> &str;
|
|
||||||
fn can_handle(&self, alias: &str, mac: &str) -> bool;
|
|
||||||
fn get_data(&self, mac: &str) -> Result<Vec<(String, TokenValue)>>;
|
|
||||||
fn get_modes(&self, _mac: &str) -> Result<Vec<String>> {
|
|
||||||
Ok(vec![])
|
|
||||||
}
|
|
||||||
fn set_mode(&self, _mode: &str, _mac: &str) -> Result<()> {
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
fn cycle_mode(&self, _mac: &str) -> Result<()> {
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub struct PixelBudsPlugin;
|
|
||||||
|
|
||||||
impl BtPlugin for PixelBudsPlugin {
|
|
||||||
fn name(&self) -> &str {
|
|
||||||
"Pixel Buds Pro"
|
|
||||||
}
|
|
||||||
|
|
||||||
fn can_handle(&self, alias: &str, _mac: &str) -> bool {
|
|
||||||
alias.contains("Pixel Buds Pro")
|
|
||||||
}
|
|
||||||
|
|
||||||
fn get_data(&self, mac: &str) -> Result<Vec<(String, TokenValue)>> {
|
|
||||||
MAESTRO.ensure_task(mac);
|
|
||||||
let status = MAESTRO.get_status(mac);
|
|
||||||
|
|
||||||
if let Some(err) = status.error {
|
|
||||||
return Err(anyhow::anyhow!(err));
|
|
||||||
}
|
|
||||||
|
|
||||||
let left_display = status
|
|
||||||
.left_battery
|
|
||||||
.map(|b| format!("{}%", b))
|
|
||||||
.unwrap_or_else(|| "---".to_string());
|
|
||||||
let right_display = status
|
|
||||||
.right_battery
|
|
||||||
.map(|b| format!("{}%", b))
|
|
||||||
.unwrap_or_else(|| "---".to_string());
|
|
||||||
|
|
||||||
let (anc_icon, class) = match status.anc_state.as_str() {
|
|
||||||
"active" => ("ANC", "anc-active"),
|
|
||||||
"aware" => ("Aware", "anc-aware"),
|
|
||||||
"off" => ("Off", "anc-off"),
|
|
||||||
_ => ("?", "anc-unknown"),
|
|
||||||
};
|
|
||||||
|
|
||||||
Ok(vec![
|
|
||||||
("left".to_string(), TokenValue::String(left_display)),
|
|
||||||
("right".to_string(), TokenValue::String(right_display)),
|
|
||||||
("anc".to_string(), TokenValue::String(anc_icon.to_string())),
|
|
||||||
(
|
|
||||||
"plugin_class".to_string(),
|
|
||||||
TokenValue::String(class.to_string()),
|
|
||||||
),
|
|
||||||
])
|
|
||||||
}
|
|
||||||
|
|
||||||
fn get_modes(&self, _mac: &str) -> Result<Vec<String>> {
|
|
||||||
Ok(vec![
|
|
||||||
"active".to_string(),
|
|
||||||
"aware".to_string(),
|
|
||||||
"off".to_string(),
|
|
||||||
])
|
|
||||||
}
|
|
||||||
|
|
||||||
fn set_mode(&self, mode: &str, mac: &str) -> Result<()> {
|
|
||||||
MAESTRO.send_command(mac, BudsCommand::SetAnc(mode.to_string()))
|
|
||||||
}
|
|
||||||
|
|
||||||
fn cycle_mode(&self, mac: &str) -> Result<()> {
|
|
||||||
let status = MAESTRO.get_status(mac);
|
|
||||||
let next_mode = match status.anc_state.as_str() {
|
|
||||||
"active" => "aware",
|
|
||||||
"aware" => "off",
|
|
||||||
_ => "active",
|
|
||||||
};
|
|
||||||
self.set_mode(next_mode, mac)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
static PLUGINS: LazyLock<Vec<Box<dyn BtPlugin>>> =
|
|
||||||
LazyLock::new(|| vec![Box::new(PixelBudsPlugin)]);
|
|
||||||
|
|
||||||
pub struct BtModule;
|
|
||||||
|
|
||||||
impl WaybarModule for BtModule {
|
|
||||||
async fn run(
|
|
||||||
&self,
|
|
||||||
config: &Config,
|
|
||||||
state: &SharedState,
|
|
||||||
args: &[&str],
|
|
||||||
) -> FluxoResult<WaybarOutput> {
|
|
||||||
let action = args.first().unwrap_or(&"show");
|
|
||||||
let bt_state = {
|
|
||||||
let lock = state.read().await;
|
|
||||||
lock.bluetooth.clone()
|
|
||||||
};
|
|
||||||
|
|
||||||
match *action {
|
|
||||||
"disconnect" if bt_state.connected => {
|
|
||||||
let _ = Command::new("bluetoothctl")
|
|
||||||
.args(["disconnect", &bt_state.device_address])
|
|
||||||
.output();
|
|
||||||
return Ok(WaybarOutput::default());
|
|
||||||
}
|
|
||||||
"cycle_mode" if bt_state.connected => {
|
|
||||||
let plugin = PLUGINS
|
|
||||||
.iter()
|
|
||||||
.find(|p| p.can_handle(&bt_state.device_alias, &bt_state.device_address));
|
|
||||||
if let Some(p) = plugin {
|
|
||||||
p.cycle_mode(&bt_state.device_address)?;
|
|
||||||
}
|
|
||||||
return Ok(WaybarOutput::default());
|
|
||||||
}
|
|
||||||
"get_modes" if bt_state.connected => {
|
|
||||||
let plugin = PLUGINS
|
|
||||||
.iter()
|
|
||||||
.find(|p| p.can_handle(&bt_state.device_alias, &bt_state.device_address));
|
|
||||||
let modes = if let Some(p) = plugin {
|
|
||||||
p.get_modes(&bt_state.device_address)?
|
|
||||||
} else {
|
|
||||||
vec![]
|
|
||||||
};
|
|
||||||
return Ok(WaybarOutput {
|
|
||||||
text: modes.join("\n"),
|
|
||||||
..Default::default()
|
|
||||||
});
|
|
||||||
}
|
|
||||||
"set_mode" if bt_state.connected => {
|
|
||||||
if let Some(mode) = args.get(1) {
|
|
||||||
let plugin = PLUGINS
|
|
||||||
.iter()
|
|
||||||
.find(|p| p.can_handle(&bt_state.device_alias, &bt_state.device_address));
|
|
||||||
if let Some(p) = plugin {
|
|
||||||
p.set_mode(mode, &bt_state.device_address)?;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return Ok(WaybarOutput::default());
|
|
||||||
}
|
|
||||||
"show" => {}
|
|
||||||
_ => {}
|
|
||||||
}
|
|
||||||
|
|
||||||
if !bt_state.adapter_powered {
|
|
||||||
return Ok(WaybarOutput {
|
|
||||||
text: config.bt.format_disabled.clone(),
|
|
||||||
tooltip: Some("Bluetooth Disabled".to_string()),
|
|
||||||
class: Some("disabled".to_string()),
|
|
||||||
percentage: None,
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
if bt_state.connected {
|
|
||||||
let mut tokens: Vec<(String, TokenValue)> = vec![
|
|
||||||
(
|
|
||||||
"alias".to_string(),
|
|
||||||
TokenValue::String(bt_state.device_alias.clone()),
|
|
||||||
),
|
|
||||||
(
|
|
||||||
"mac".to_string(),
|
|
||||||
TokenValue::String(bt_state.device_address.clone()),
|
|
||||||
),
|
|
||||||
];
|
|
||||||
|
|
||||||
let mut class = vec!["connected".to_string()];
|
|
||||||
let mut has_plugin = false;
|
|
||||||
|
|
||||||
for (k, v) in &bt_state.plugin_data {
|
|
||||||
if k == "plugin_class" {
|
|
||||||
class.push(v.clone());
|
|
||||||
has_plugin = true;
|
|
||||||
} else if k == "plugin_error" {
|
|
||||||
class.push("plugin-error".to_string());
|
|
||||||
} else {
|
|
||||||
tokens.push((k.clone(), TokenValue::String(v.clone())));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
let format = if has_plugin {
|
|
||||||
&config.bt.format_plugin
|
|
||||||
} else {
|
|
||||||
&config.bt.format_connected
|
|
||||||
};
|
|
||||||
|
|
||||||
let text = format_template(format, &tokens);
|
|
||||||
let tooltip = format!(
|
|
||||||
"{} | MAC: {}\nBattery: {}",
|
|
||||||
bt_state.device_alias,
|
|
||||||
bt_state.device_address,
|
|
||||||
bt_state
|
|
||||||
.battery_percentage
|
|
||||||
.map(|b| format!("{}%", b))
|
|
||||||
.unwrap_or_else(|| "N/A".to_string())
|
|
||||||
);
|
|
||||||
|
|
||||||
Ok(WaybarOutput {
|
|
||||||
text,
|
|
||||||
tooltip: Some(tooltip),
|
|
||||||
class: Some(class.join(" ")),
|
|
||||||
percentage: bt_state.battery_percentage,
|
|
||||||
})
|
|
||||||
} else {
|
|
||||||
Ok(WaybarOutput {
|
|
||||||
text: config.bt.format_disconnected.clone(),
|
|
||||||
tooltip: Some("Bluetooth On (Disconnected)".to_string()),
|
|
||||||
class: Some("disconnected".to_string()),
|
|
||||||
percentage: None,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -0,0 +1,142 @@
|
|||||||
|
use crate::config::Config;
|
||||||
|
use crate::error::{FluxoError, Result as FluxoResult};
|
||||||
|
use crate::modules::bt::maestro::{BudsCommand, get_maestro};
|
||||||
|
use crate::state::SharedState;
|
||||||
|
use crate::utils::TokenValue;
|
||||||
|
use futures::future::BoxFuture;
|
||||||
|
use std::sync::Arc;
|
||||||
|
|
||||||
|
pub trait BtPlugin: Send + Sync {
|
||||||
|
fn name(&self) -> &str;
|
||||||
|
fn can_handle(&self, alias: &str, mac: &str) -> bool;
|
||||||
|
fn get_data(
|
||||||
|
&self,
|
||||||
|
config: &Config,
|
||||||
|
state: &SharedState,
|
||||||
|
mac: &str,
|
||||||
|
) -> BoxFuture<'static, FluxoResult<Vec<(String, TokenValue)>>>;
|
||||||
|
fn get_modes(
|
||||||
|
&self,
|
||||||
|
mac: &str,
|
||||||
|
state: &SharedState,
|
||||||
|
) -> BoxFuture<'static, FluxoResult<Vec<String>>>;
|
||||||
|
fn set_mode(
|
||||||
|
&self,
|
||||||
|
mode: &str,
|
||||||
|
mac: &str,
|
||||||
|
state: &SharedState,
|
||||||
|
) -> BoxFuture<'static, FluxoResult<()>>;
|
||||||
|
fn cycle_mode(&self, mac: &str, state: &SharedState) -> BoxFuture<'static, FluxoResult<()>>;
|
||||||
|
}
|
||||||
|
|
||||||
|
pub struct PixelBudsPlugin;
|
||||||
|
|
||||||
|
impl BtPlugin for PixelBudsPlugin {
|
||||||
|
fn name(&self) -> &str {
|
||||||
|
"Pixel Buds Pro"
|
||||||
|
}
|
||||||
|
|
||||||
|
fn can_handle(&self, alias: &str, _mac: &str) -> bool {
|
||||||
|
alias.contains("Pixel Buds Pro")
|
||||||
|
}
|
||||||
|
|
||||||
|
fn get_data(
|
||||||
|
&self,
|
||||||
|
_config: &Config,
|
||||||
|
state: &SharedState,
|
||||||
|
mac: &str,
|
||||||
|
) -> BoxFuture<'static, FluxoResult<Vec<(String, TokenValue)>>> {
|
||||||
|
let mac = mac.to_string();
|
||||||
|
let state = Arc::clone(state);
|
||||||
|
Box::pin(async move {
|
||||||
|
let maestro = get_maestro(&state);
|
||||||
|
maestro.ensure_task(&mac);
|
||||||
|
let status = maestro.get_status(&mac);
|
||||||
|
|
||||||
|
if let Some(err) = status.error {
|
||||||
|
return Err(FluxoError::Module {
|
||||||
|
module: "bt.buds",
|
||||||
|
message: err,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
let left_display = status
|
||||||
|
.left_battery
|
||||||
|
.map(|b| format!("{}%", b))
|
||||||
|
.unwrap_or_else(|| "---".to_string());
|
||||||
|
let right_display = status
|
||||||
|
.right_battery
|
||||||
|
.map(|b| format!("{}%", b))
|
||||||
|
.unwrap_or_else(|| "---".to_string());
|
||||||
|
|
||||||
|
let (anc_icon, class) = match status.anc_state.as_str() {
|
||||||
|
"active" => ("ANC", "anc-active"),
|
||||||
|
"aware" => ("Aware", "anc-aware"),
|
||||||
|
"off" => ("Off", "anc-off"),
|
||||||
|
_ => ("?", "anc-unknown"),
|
||||||
|
};
|
||||||
|
|
||||||
|
Ok(vec![
|
||||||
|
("left".to_string(), TokenValue::String(left_display)),
|
||||||
|
("right".to_string(), TokenValue::String(right_display)),
|
||||||
|
("anc".to_string(), TokenValue::String(anc_icon.to_string())),
|
||||||
|
(
|
||||||
|
"plugin_class".to_string(),
|
||||||
|
TokenValue::String(class.to_string()),
|
||||||
|
),
|
||||||
|
])
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
fn get_modes(
|
||||||
|
&self,
|
||||||
|
_mac: &str,
|
||||||
|
_state: &SharedState,
|
||||||
|
) -> BoxFuture<'static, FluxoResult<Vec<String>>> {
|
||||||
|
Box::pin(async move {
|
||||||
|
Ok(vec![
|
||||||
|
"active".to_string(),
|
||||||
|
"aware".to_string(),
|
||||||
|
"off".to_string(),
|
||||||
|
])
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
fn set_mode(
|
||||||
|
&self,
|
||||||
|
mode: &str,
|
||||||
|
mac: &str,
|
||||||
|
state: &SharedState,
|
||||||
|
) -> BoxFuture<'static, FluxoResult<()>> {
|
||||||
|
let mode = mode.to_string();
|
||||||
|
let mac = mac.to_string();
|
||||||
|
let state = Arc::clone(state);
|
||||||
|
Box::pin(async move {
|
||||||
|
get_maestro(&state)
|
||||||
|
.send_command(&mac, BudsCommand::SetAnc(mode))
|
||||||
|
.map_err(|e: anyhow::Error| FluxoError::Module {
|
||||||
|
module: "bt.buds",
|
||||||
|
message: e.to_string(),
|
||||||
|
})
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
fn cycle_mode(&self, mac: &str, state: &SharedState) -> BoxFuture<'static, FluxoResult<()>> {
|
||||||
|
let mac = mac.to_string();
|
||||||
|
let state = Arc::clone(state);
|
||||||
|
Box::pin(async move {
|
||||||
|
let status = get_maestro(&state).get_status(&mac);
|
||||||
|
let next_mode = match status.anc_state.as_str() {
|
||||||
|
"active" => "aware",
|
||||||
|
"aware" => "off",
|
||||||
|
_ => "active",
|
||||||
|
};
|
||||||
|
get_maestro(&state)
|
||||||
|
.send_command(&mac, BudsCommand::SetAnc(next_mode.to_string()))
|
||||||
|
.map_err(|e: anyhow::Error| FluxoError::Module {
|
||||||
|
module: "bt.buds",
|
||||||
|
message: e.to_string(),
|
||||||
|
})
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,362 @@
|
|||||||
|
use crate::state::SharedState;
|
||||||
|
use anyhow::{Context, Result};
|
||||||
|
use futures::StreamExt;
|
||||||
|
use std::collections::HashMap;
|
||||||
|
use std::sync::{Arc, Mutex, OnceLock};
|
||||||
|
use std::time::{Duration, Instant};
|
||||||
|
use tokio::sync::mpsc;
|
||||||
|
use tracing::{debug, error, info, warn};
|
||||||
|
|
||||||
|
// Maestro imports
|
||||||
|
use maestro::protocol::codec::Codec;
|
||||||
|
use maestro::pwrpc::client::Client;
|
||||||
|
use maestro::service::MaestroService;
|
||||||
|
use maestro::service::settings::{self, SettingValue};
|
||||||
|
|
||||||
|
#[derive(Clone, Default)]
|
||||||
|
pub struct BudsStatus {
|
||||||
|
pub left_battery: Option<u8>,
|
||||||
|
pub right_battery: Option<u8>,
|
||||||
|
pub case_battery: Option<u8>,
|
||||||
|
pub anc_state: String,
|
||||||
|
#[allow(dead_code)]
|
||||||
|
pub last_update: Option<Instant>,
|
||||||
|
pub error: Option<String>,
|
||||||
|
}
|
||||||
|
|
||||||
|
pub enum BudsCommand {
|
||||||
|
SetAnc(String),
|
||||||
|
}
|
||||||
|
|
||||||
|
pub enum ManagerCommand {
|
||||||
|
EnsureTask(String),
|
||||||
|
SendCommand(String, BudsCommand),
|
||||||
|
}
|
||||||
|
|
||||||
|
pub struct MaestroManager {
|
||||||
|
statuses: Arc<Mutex<HashMap<String, BudsStatus>>>,
|
||||||
|
management_tx: mpsc::UnboundedSender<ManagerCommand>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl MaestroManager {
|
||||||
|
pub fn new(state: SharedState) -> Self {
|
||||||
|
let (tx, mut rx) = mpsc::unbounded_channel::<ManagerCommand>();
|
||||||
|
let statuses = Arc::new(Mutex::new(HashMap::new()));
|
||||||
|
let statuses_clone = Arc::clone(&statuses);
|
||||||
|
let state_clone = Arc::clone(&state);
|
||||||
|
|
||||||
|
// Start dedicated BT management thread
|
||||||
|
std::thread::spawn(move || {
|
||||||
|
let rt = tokio::runtime::Builder::new_current_thread()
|
||||||
|
.enable_all()
|
||||||
|
.build()
|
||||||
|
.unwrap();
|
||||||
|
|
||||||
|
let local = tokio::task::LocalSet::new();
|
||||||
|
|
||||||
|
local.block_on(&rt, async move {
|
||||||
|
let mut command_txs: HashMap<String, mpsc::Sender<BudsCommand>> = HashMap::new();
|
||||||
|
|
||||||
|
loop {
|
||||||
|
tokio::select! {
|
||||||
|
Some(cmd) = rx.recv() => {
|
||||||
|
match cmd {
|
||||||
|
ManagerCommand::EnsureTask(mac) => {
|
||||||
|
if !command_txs.contains_key(&mac) {
|
||||||
|
let (tx, buds_rx) = mpsc::channel::<BudsCommand>(10);
|
||||||
|
command_txs.insert(mac.clone(), tx);
|
||||||
|
|
||||||
|
let mac_clone = mac.clone();
|
||||||
|
let st_clone = Arc::clone(&statuses_clone);
|
||||||
|
let state_inner = Arc::clone(&state_clone);
|
||||||
|
|
||||||
|
tokio::task::spawn_local(async move {
|
||||||
|
if let Err(e) = buds_task(&mac_clone, st_clone, buds_rx, state_inner).await {
|
||||||
|
error!("Buds task for {} failed: {}", mac_clone, e);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
ManagerCommand::SendCommand(mac, buds_cmd) => {
|
||||||
|
if let Some(tx) = command_txs.get(&mac) {
|
||||||
|
let _ = tx.try_send(buds_cmd);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_ = tokio::time::sleep(Duration::from_millis(100)) => {
|
||||||
|
// Cleanup dropped tasks if needed
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
Self {
|
||||||
|
statuses,
|
||||||
|
management_tx: tx,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn get_status(&self, mac: &str) -> BudsStatus {
|
||||||
|
let statuses = self.statuses.lock().unwrap();
|
||||||
|
statuses.get(mac).cloned().unwrap_or_default()
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn ensure_task(&self, mac: &str) {
|
||||||
|
let _ = self
|
||||||
|
.management_tx
|
||||||
|
.send(ManagerCommand::EnsureTask(mac.to_string()));
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn send_command(&self, mac: &str, cmd: BudsCommand) -> Result<()> {
|
||||||
|
self.ensure_task(mac);
|
||||||
|
let _ = self
|
||||||
|
.management_tx
|
||||||
|
.send(ManagerCommand::SendCommand(mac.to_string(), cmd));
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn buds_task(
|
||||||
|
mac: &str,
|
||||||
|
statuses: Arc<Mutex<HashMap<String, BudsStatus>>>,
|
||||||
|
mut rx: mpsc::Receiver<BudsCommand>,
|
||||||
|
state: SharedState,
|
||||||
|
) -> Result<()> {
|
||||||
|
info!("Starting native Maestro connection task for {}", mac);
|
||||||
|
|
||||||
|
loop {
|
||||||
|
let addr: bluer::Address = match mac.parse() {
|
||||||
|
Ok(a) => a,
|
||||||
|
Err(e) => {
|
||||||
|
error!("Failed to parse MAC address {}: {}", mac, e);
|
||||||
|
return Err(e.into());
|
||||||
|
}
|
||||||
|
};
|
||||||
|
let session = bluer::Session::new()
|
||||||
|
.await
|
||||||
|
.context("Failed to create bluer session")?;
|
||||||
|
let adapter = session
|
||||||
|
.default_adapter()
|
||||||
|
.await
|
||||||
|
.context("Failed to get default adapter")?;
|
||||||
|
let device = adapter
|
||||||
|
.device(addr)
|
||||||
|
.context("Failed to get device handle")?;
|
||||||
|
|
||||||
|
if !device.is_connected().await.unwrap_or(false) {
|
||||||
|
debug!("Device {} not connected to BT, stopping maestro task", mac);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Connect to Maestro RFCOMM service
|
||||||
|
let mut stream = None;
|
||||||
|
for channel in [1, 2] {
|
||||||
|
let socket = match bluer::rfcomm::Socket::new() {
|
||||||
|
Ok(s) => s,
|
||||||
|
Err(e) => {
|
||||||
|
error!("Failed to create RFCOMM socket: {}", e);
|
||||||
|
return Err(e.into());
|
||||||
|
}
|
||||||
|
};
|
||||||
|
let target = bluer::rfcomm::SocketAddr::new(addr, channel);
|
||||||
|
debug!(
|
||||||
|
"Trying to connect RFCOMM to {} on channel {}...",
|
||||||
|
mac, channel
|
||||||
|
);
|
||||||
|
match socket.connect(target).await {
|
||||||
|
Ok(s) => {
|
||||||
|
stream = Some(s);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
Err(e) => {
|
||||||
|
debug!("Failed to connect to channel {}: {}", channel, e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
let stream = match stream {
|
||||||
|
Some(s) => s,
|
||||||
|
None => {
|
||||||
|
warn!(
|
||||||
|
"Failed to connect RFCOMM to {} on any common channel. Retrying in 15s...",
|
||||||
|
mac
|
||||||
|
);
|
||||||
|
tokio::time::sleep(Duration::from_secs(15)).await;
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
info!("Connected Maestro RFCOMM to {} on channel", mac);
|
||||||
|
|
||||||
|
// Initialize Maestro communication stack
|
||||||
|
let codec = Codec::new();
|
||||||
|
let stream = codec.wrap(stream);
|
||||||
|
let mut client = Client::new(stream);
|
||||||
|
let handle = client.handle();
|
||||||
|
|
||||||
|
// Resolve Maestro channel
|
||||||
|
let channel = match maestro::protocol::utils::resolve_channel(&mut client).await {
|
||||||
|
Ok(c) => c,
|
||||||
|
Err(e) => {
|
||||||
|
error!("Failed to resolve Maestro channel for {}: {}", mac, e);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
tokio::spawn(async move {
|
||||||
|
if let Err(e) = client.run().await {
|
||||||
|
error!("Maestro client loop error: {}", e);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
let mut service = MaestroService::new(handle, channel);
|
||||||
|
|
||||||
|
// Update health
|
||||||
|
{
|
||||||
|
let mut lock = state.write().await;
|
||||||
|
let health = lock.health.entry("bt.buds".to_string()).or_default();
|
||||||
|
health.consecutive_failures = 0;
|
||||||
|
health.backoff_until = None;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Query initial ANC state
|
||||||
|
if let Ok(val) = service
|
||||||
|
.read_setting_var(settings::SettingId::CurrentAncrState)
|
||||||
|
.await
|
||||||
|
&& let SettingValue::CurrentAncrState(anc_state) = val
|
||||||
|
{
|
||||||
|
let mut lock = statuses.lock().unwrap();
|
||||||
|
let status = lock.entry(mac.to_string()).or_default();
|
||||||
|
status.anc_state = anc_state_to_string(&anc_state);
|
||||||
|
status.last_update = Some(Instant::now());
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut runtime_info_call = match service.subscribe_to_runtime_info() {
|
||||||
|
Ok(c) => c,
|
||||||
|
Err(e) => {
|
||||||
|
error!("Failed to subscribe to runtime info for {}: {}", mac, e);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let mut runtime_info = runtime_info_call.stream();
|
||||||
|
|
||||||
|
let mut settings_changes_call = match service.subscribe_to_settings_changes() {
|
||||||
|
Ok(s) => s,
|
||||||
|
Err(e) => {
|
||||||
|
error!("Failed to subscribe to settings changes for {}: {}", mac, e);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let mut settings_changes = settings_changes_call.stream();
|
||||||
|
|
||||||
|
debug!("Subscribed to status and settings for {}", mac);
|
||||||
|
|
||||||
|
loop {
|
||||||
|
tokio::select! {
|
||||||
|
cmd = rx.recv() => {
|
||||||
|
match cmd {
|
||||||
|
Some(BudsCommand::SetAnc(mode)) => {
|
||||||
|
debug!("Setting ANC mode to {} for {}", mode, mac);
|
||||||
|
let state = mode_to_anc_state(&mode);
|
||||||
|
let val = SettingValue::CurrentAncrState(state);
|
||||||
|
|
||||||
|
{
|
||||||
|
let mut lock = statuses.lock().unwrap();
|
||||||
|
let status = lock.entry(mac.to_string()).or_default();
|
||||||
|
status.anc_state = mode.clone();
|
||||||
|
status.last_update = Some(Instant::now());
|
||||||
|
}
|
||||||
|
|
||||||
|
if let Err(e) = service.write_setting(val).await {
|
||||||
|
error!("Failed to write ANC setting for {}: {}", mac, e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
None => return Ok(()),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Some(res) = runtime_info.next() => {
|
||||||
|
match res {
|
||||||
|
Ok(info) => {
|
||||||
|
let mut lock = statuses.lock().unwrap();
|
||||||
|
let status = lock.entry(mac.to_string()).or_default();
|
||||||
|
status.last_update = Some(Instant::now());
|
||||||
|
|
||||||
|
if let Some(bat) = info.battery_info {
|
||||||
|
status.left_battery = bat.left.map(|b| b.level as u8);
|
||||||
|
status.right_battery = bat.right.map(|b| b.level as u8);
|
||||||
|
status.case_battery = bat.case.map(|b| b.level as u8);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Err(e) => {
|
||||||
|
warn!("Runtime info stream error for {}: {}", mac, e);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Some(res) = settings_changes.next() => {
|
||||||
|
if let Ok(change) = res {
|
||||||
|
use maestro::protocol::types::settings_rsp::ValueOneof as RspOneof;
|
||||||
|
use maestro::protocol::types::setting_value::ValueOneof as ValOneof;
|
||||||
|
|
||||||
|
if let Some(RspOneof::Value(setting_val)) = change.value_oneof
|
||||||
|
&& let Some(ValOneof::CurrentAncrState(anc_state_raw)) = setting_val.value_oneof {
|
||||||
|
let mut lock = statuses.lock().unwrap();
|
||||||
|
let status = lock.entry(mac.to_string()).or_default();
|
||||||
|
|
||||||
|
let anc_state = match anc_state_raw {
|
||||||
|
1 => settings::AncState::Off,
|
||||||
|
2 => settings::AncState::Active,
|
||||||
|
3 => settings::AncState::Aware,
|
||||||
|
4 => settings::AncState::Adaptive,
|
||||||
|
_ => settings::AncState::Unknown(anc_state_raw),
|
||||||
|
};
|
||||||
|
|
||||||
|
status.anc_state = anc_state_to_string(&anc_state);
|
||||||
|
status.last_update = Some(Instant::now());
|
||||||
|
debug!(mode = %status.anc_state, "Caught physical ANC toggle");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_ = tokio::time::sleep(Duration::from_secs(30)) => {
|
||||||
|
if !device.is_connected().await.unwrap_or(false) {
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if !device.is_connected().await.unwrap_or(false) {
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
fn mode_to_anc_state(mode: &str) -> settings::AncState {
|
||||||
|
match mode {
|
||||||
|
"active" => settings::AncState::Active,
|
||||||
|
"aware" => settings::AncState::Aware,
|
||||||
|
"off" => settings::AncState::Off,
|
||||||
|
_ => settings::AncState::Off,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn anc_state_to_string(state: &settings::AncState) -> String {
|
||||||
|
match state {
|
||||||
|
settings::AncState::Active => "active".to_string(),
|
||||||
|
settings::AncState::Aware => "aware".to_string(),
|
||||||
|
settings::AncState::Off => "off".to_string(),
|
||||||
|
_ => "unknown".to_string(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
static MAESTRO: OnceLock<MaestroManager> = OnceLock::new();
|
||||||
|
|
||||||
|
pub fn get_maestro(state: &SharedState) -> &MaestroManager {
|
||||||
|
MAESTRO.get_or_init(|| MaestroManager::new(Arc::clone(state)))
|
||||||
|
}
|
||||||
@@ -0,0 +1,236 @@
|
|||||||
|
pub mod buds;
|
||||||
|
pub mod maestro;
|
||||||
|
|
||||||
|
use crate::config::Config;
|
||||||
|
use crate::error::Result as FluxoResult;
|
||||||
|
use crate::modules::WaybarModule;
|
||||||
|
use crate::output::WaybarOutput;
|
||||||
|
use crate::state::{BtState, SharedState};
|
||||||
|
use crate::utils::{TokenValue, format_template};
|
||||||
|
use anyhow::Result;
|
||||||
|
use std::process::Command;
|
||||||
|
use std::sync::{Arc, LazyLock};
|
||||||
|
use tracing::{error, warn};
|
||||||
|
|
||||||
|
use self::buds::{BtPlugin, PixelBudsPlugin};
|
||||||
|
|
||||||
|
pub struct BtDaemon {
|
||||||
|
session: Option<bluer::Session>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl BtDaemon {
|
||||||
|
pub fn new() -> Self {
|
||||||
|
Self { session: None }
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn poll(&mut self, state: SharedState, config: &Config) {
|
||||||
|
if let Err(e) = self.poll_async(state, config).await {
|
||||||
|
error!("BT daemon error: {}", e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn poll_async(&mut self, state: SharedState, config: &Config) -> Result<()> {
|
||||||
|
if self.session.is_none() {
|
||||||
|
self.session = Some(bluer::Session::new().await?);
|
||||||
|
}
|
||||||
|
let session = self.session.as_ref().unwrap();
|
||||||
|
let adapter = session.default_adapter().await?;
|
||||||
|
let adapter_powered = adapter.is_powered().await.unwrap_or(false);
|
||||||
|
|
||||||
|
let mut bt_state = BtState {
|
||||||
|
adapter_powered,
|
||||||
|
..Default::default()
|
||||||
|
};
|
||||||
|
|
||||||
|
if adapter_powered {
|
||||||
|
let devices = adapter.device_addresses().await?;
|
||||||
|
for addr in devices {
|
||||||
|
let device = adapter.device(addr)?;
|
||||||
|
if device.is_connected().await.unwrap_or(false) {
|
||||||
|
let uuids = device.uuids().await?.unwrap_or_default();
|
||||||
|
let audio_sink_uuid =
|
||||||
|
bluer::Uuid::from_u128(0x0000110b_0000_1000_8000_00805f9b34fb);
|
||||||
|
if uuids.contains(&audio_sink_uuid) {
|
||||||
|
bt_state.connected = true;
|
||||||
|
bt_state.device_address = addr.to_string();
|
||||||
|
bt_state.device_alias =
|
||||||
|
device.alias().await.unwrap_or_else(|_| addr.to_string());
|
||||||
|
bt_state.battery_percentage =
|
||||||
|
device.battery_percentage().await.unwrap_or(None);
|
||||||
|
|
||||||
|
for p in PLUGINS.iter() {
|
||||||
|
if p.can_handle(&bt_state.device_alias, &bt_state.device_address) {
|
||||||
|
match p.get_data(config, &state, &bt_state.device_address).await {
|
||||||
|
Ok(data) => {
|
||||||
|
bt_state.plugin_data = data
|
||||||
|
.into_iter()
|
||||||
|
.map(|(k, v)| {
|
||||||
|
let val_str = match v {
|
||||||
|
TokenValue::String(s) => s,
|
||||||
|
TokenValue::Int(i) => i.to_string(),
|
||||||
|
TokenValue::Float(f) => format!("{:.1}", f),
|
||||||
|
};
|
||||||
|
(k, val_str)
|
||||||
|
})
|
||||||
|
.collect();
|
||||||
|
}
|
||||||
|
Err(e) => {
|
||||||
|
warn!("Plugin {} failed for {}: {}", p.name(), addr, e);
|
||||||
|
bt_state
|
||||||
|
.plugin_data
|
||||||
|
.push(("plugin_error".to_string(), e.to_string()));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut lock = state.write().await;
|
||||||
|
lock.bluetooth = bt_state;
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
static PLUGINS: LazyLock<Vec<Box<dyn BtPlugin>>> =
|
||||||
|
LazyLock::new(|| vec![Box::new(PixelBudsPlugin)]);
|
||||||
|
|
||||||
|
pub struct BtModule;
|
||||||
|
|
||||||
|
impl WaybarModule for BtModule {
|
||||||
|
fn run(
|
||||||
|
&self,
|
||||||
|
config: &Config,
|
||||||
|
state: &SharedState,
|
||||||
|
args: &[&str],
|
||||||
|
) -> impl std::future::Future<Output = FluxoResult<WaybarOutput>> + Send {
|
||||||
|
let action = args.first().cloned().unwrap_or("show").to_string();
|
||||||
|
let args = args.iter().map(|s| s.to_string()).collect::<Vec<_>>();
|
||||||
|
let state = Arc::clone(state);
|
||||||
|
let config = config.clone();
|
||||||
|
|
||||||
|
async move {
|
||||||
|
let bt_state = {
|
||||||
|
let lock = state.read().await;
|
||||||
|
lock.bluetooth.clone()
|
||||||
|
};
|
||||||
|
|
||||||
|
match action.as_str() {
|
||||||
|
"disconnect" if bt_state.connected => {
|
||||||
|
let _ = Command::new("bluetoothctl")
|
||||||
|
.args(["disconnect", &bt_state.device_address])
|
||||||
|
.output();
|
||||||
|
return Ok(WaybarOutput::default());
|
||||||
|
}
|
||||||
|
"cycle_mode" if bt_state.connected => {
|
||||||
|
let plugin = PLUGINS
|
||||||
|
.iter()
|
||||||
|
.find(|p| p.can_handle(&bt_state.device_alias, &bt_state.device_address));
|
||||||
|
if let Some(p) = plugin {
|
||||||
|
p.cycle_mode(&bt_state.device_address, &state).await?;
|
||||||
|
}
|
||||||
|
return Ok(WaybarOutput::default());
|
||||||
|
}
|
||||||
|
"get_modes" if bt_state.connected => {
|
||||||
|
let plugin = PLUGINS
|
||||||
|
.iter()
|
||||||
|
.find(|p| p.can_handle(&bt_state.device_alias, &bt_state.device_address));
|
||||||
|
let modes = if let Some(p) = plugin {
|
||||||
|
p.get_modes(&bt_state.device_address, &state).await?
|
||||||
|
} else {
|
||||||
|
vec![]
|
||||||
|
};
|
||||||
|
return Ok(WaybarOutput {
|
||||||
|
text: modes.join("\n"),
|
||||||
|
..Default::default()
|
||||||
|
});
|
||||||
|
}
|
||||||
|
"set_mode" if bt_state.connected => {
|
||||||
|
if let Some(mode) = args.get(1) {
|
||||||
|
let plugin = PLUGINS.iter().find(|p| {
|
||||||
|
p.can_handle(&bt_state.device_alias, &bt_state.device_address)
|
||||||
|
});
|
||||||
|
if let Some(p) = plugin {
|
||||||
|
p.set_mode(mode, &bt_state.device_address, &state).await?;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return Ok(WaybarOutput::default());
|
||||||
|
}
|
||||||
|
"show" => {}
|
||||||
|
_ => {}
|
||||||
|
}
|
||||||
|
|
||||||
|
if !bt_state.adapter_powered {
|
||||||
|
return Ok(WaybarOutput {
|
||||||
|
text: config.bt.format_disabled.clone(),
|
||||||
|
tooltip: Some("Bluetooth Disabled".to_string()),
|
||||||
|
class: Some("disabled".to_string()),
|
||||||
|
percentage: None,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
if bt_state.connected {
|
||||||
|
let mut tokens: Vec<(String, TokenValue)> = vec![
|
||||||
|
(
|
||||||
|
"alias".to_string(),
|
||||||
|
TokenValue::String(bt_state.device_alias.clone()),
|
||||||
|
),
|
||||||
|
(
|
||||||
|
"mac".to_string(),
|
||||||
|
TokenValue::String(bt_state.device_address.clone()),
|
||||||
|
),
|
||||||
|
];
|
||||||
|
|
||||||
|
let mut class = vec!["connected".to_string()];
|
||||||
|
let mut has_plugin = false;
|
||||||
|
|
||||||
|
for (k, v) in &bt_state.plugin_data {
|
||||||
|
if k == "plugin_class" {
|
||||||
|
class.push(v.clone());
|
||||||
|
has_plugin = true;
|
||||||
|
} else if k == "plugin_error" {
|
||||||
|
class.push("plugin-error".to_string());
|
||||||
|
} else {
|
||||||
|
tokens.push((k.clone(), TokenValue::String(v.clone())));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
let format = if has_plugin {
|
||||||
|
&config.bt.format_plugin
|
||||||
|
} else {
|
||||||
|
&config.bt.format_connected
|
||||||
|
};
|
||||||
|
|
||||||
|
let text = format_template(format, &tokens);
|
||||||
|
let tooltip = format!(
|
||||||
|
"{} | MAC: {}\nBattery: {}",
|
||||||
|
bt_state.device_alias,
|
||||||
|
bt_state.device_address,
|
||||||
|
bt_state
|
||||||
|
.battery_percentage
|
||||||
|
.map(|b| format!("{}%", b))
|
||||||
|
.unwrap_or_else(|| "N/A".to_string())
|
||||||
|
);
|
||||||
|
|
||||||
|
Ok(WaybarOutput {
|
||||||
|
text,
|
||||||
|
tooltip: Some(tooltip),
|
||||||
|
class: Some(class.join(" ")),
|
||||||
|
percentage: bt_state.battery_percentage,
|
||||||
|
})
|
||||||
|
} else {
|
||||||
|
Ok(WaybarOutput {
|
||||||
|
text: config.bt.format_disconnected.clone(),
|
||||||
|
tooltip: Some("Bluetooth On (Disconnected)".to_string()),
|
||||||
|
class: Some("disconnected".to_string()),
|
||||||
|
percentage: None,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user