Compare commits
15 Commits
492f0787cb
...
v0.5.1
| Author | SHA1 | Date | |
|---|---|---|---|
| f89833a62e | |||
| 198986efac | |||
| d1178ff8ba | |||
| 5c5d0148f8 | |||
| fdfd54b518 | |||
| 708317a10b | |||
| f1e601f9ed | |||
| 57906de920 | |||
| 2050c345f1 | |||
| 75601305e2 | |||
| 52cc041c42 | |||
| cb8b641447 | |||
| 071d973fea | |||
| 31072bc645 | |||
| 941942d550 |
Generated
+44
-209
@@ -111,17 +111,6 @@ dependencies = [
|
|||||||
"slab",
|
"slab",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
|
||||||
name = "async-fs"
|
|
||||||
version = "2.2.0"
|
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
|
||||||
checksum = "8034a681df4aed8b8edbd7fbe472401ecf009251c8b40556b304567052e294c5"
|
|
||||||
dependencies = [
|
|
||||||
"async-lock",
|
|
||||||
"blocking",
|
|
||||||
"futures-lite",
|
|
||||||
]
|
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "async-io"
|
name = "async-io"
|
||||||
version = "2.6.0"
|
version = "2.6.0"
|
||||||
@@ -239,15 +228,6 @@ version = "2.11.0"
|
|||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "843867be96c8daad0d758b57df9392b6d8d271134fce549de6ce169ff98a92af"
|
checksum = "843867be96c8daad0d758b57df9392b6d8d271134fce549de6ce169ff98a92af"
|
||||||
|
|
||||||
[[package]]
|
|
||||||
name = "block-buffer"
|
|
||||||
version = "0.10.4"
|
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
|
||||||
checksum = "3078c7629b62d3f0439517fa394996acacc5cbc91c5a20d8c658e77abd503a71"
|
|
||||||
dependencies = [
|
|
||||||
"generic-array",
|
|
||||||
]
|
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "block2"
|
name = "block2"
|
||||||
version = "0.6.2"
|
version = "0.6.2"
|
||||||
@@ -378,31 +358,12 @@ dependencies = [
|
|||||||
"crossbeam-utils",
|
"crossbeam-utils",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
|
||||||
name = "cpufeatures"
|
|
||||||
version = "0.2.17"
|
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
|
||||||
checksum = "59ed5838eebb26a2bb2e58f6d5b5316989ae9d08bab10e0e6d103e656d1b0280"
|
|
||||||
dependencies = [
|
|
||||||
"libc",
|
|
||||||
]
|
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "crossbeam-utils"
|
name = "crossbeam-utils"
|
||||||
version = "0.8.21"
|
version = "0.8.21"
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "d0a5c400df2834b80a4c3327b3aad3a4c4cd4de0629063962b03235697506a28"
|
checksum = "d0a5c400df2834b80a4c3327b3aad3a4c4cd4de0629063962b03235697506a28"
|
||||||
|
|
||||||
[[package]]
|
|
||||||
name = "crypto-common"
|
|
||||||
version = "0.1.7"
|
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
|
||||||
checksum = "78c8292055d1c1df0cce5d180393dc8cce0abec0a7102adb6c7b1eef6016d60a"
|
|
||||||
dependencies = [
|
|
||||||
"generic-array",
|
|
||||||
"typenum",
|
|
||||||
]
|
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "ctrlc"
|
name = "ctrlc"
|
||||||
version = "3.5.2"
|
version = "3.5.2"
|
||||||
@@ -504,16 +465,6 @@ dependencies = [
|
|||||||
"tokio",
|
"tokio",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
|
||||||
name = "digest"
|
|
||||||
version = "0.10.7"
|
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
|
||||||
checksum = "9ed9a281f7bc9b7576e61468ba615a66a5c8cfdff42420a70aa82701a3b1e292"
|
|
||||||
dependencies = [
|
|
||||||
"block-buffer",
|
|
||||||
"crypto-common",
|
|
||||||
]
|
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "dispatch2"
|
name = "dispatch2"
|
||||||
version = "0.3.1"
|
version = "0.3.1"
|
||||||
@@ -621,13 +572,14 @@ checksum = "1d674e81391d1e1ab681a28d99df07927c6d4aa5b027d7da16ba32d1d21ecd99"
|
|||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "fluxo-rs"
|
name = "fluxo-rs"
|
||||||
version = "0.3.3"
|
version = "0.5.1"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"anyhow",
|
"anyhow",
|
||||||
"bluer",
|
"bluer",
|
||||||
"clap",
|
"clap",
|
||||||
"ctrlc",
|
"ctrlc",
|
||||||
"futures",
|
"futures",
|
||||||
|
"libc",
|
||||||
"libpulse-binding",
|
"libpulse-binding",
|
||||||
"maestro",
|
"maestro",
|
||||||
"nix 0.31.2",
|
"nix 0.31.2",
|
||||||
@@ -768,27 +720,6 @@ dependencies = [
|
|||||||
"slab",
|
"slab",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
|
||||||
name = "generic-array"
|
|
||||||
version = "0.14.7"
|
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
|
||||||
checksum = "85649ca51fd72272d7821adaf274ad91c288277713d9c18820d8499a7ff69e9a"
|
|
||||||
dependencies = [
|
|
||||||
"typenum",
|
|
||||||
"version_check",
|
|
||||||
]
|
|
||||||
|
|
||||||
[[package]]
|
|
||||||
name = "getrandom"
|
|
||||||
version = "0.2.17"
|
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
|
||||||
checksum = "ff2abc00be7fca6ebc474524697ae276ad847ad0a6b3faa4bcb027e9a4614ad0"
|
|
||||||
dependencies = [
|
|
||||||
"cfg-if",
|
|
||||||
"libc",
|
|
||||||
"wasi",
|
|
||||||
]
|
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "getrandom"
|
name = "getrandom"
|
||||||
version = "0.4.2"
|
version = "0.4.2"
|
||||||
@@ -1071,7 +1002,6 @@ dependencies = [
|
|||||||
"cfg-if",
|
"cfg-if",
|
||||||
"cfg_aliases",
|
"cfg_aliases",
|
||||||
"libc",
|
"libc",
|
||||||
"memoffset",
|
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
@@ -1304,15 +1234,6 @@ dependencies = [
|
|||||||
"windows-sys 0.61.2",
|
"windows-sys 0.61.2",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
|
||||||
name = "ppv-lite86"
|
|
||||||
version = "0.2.21"
|
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
|
||||||
checksum = "85eae3c4ed2f50dcfe72643da4befc30deadb458a9b590d720cde2f2b1e97da9"
|
|
||||||
dependencies = [
|
|
||||||
"zerocopy",
|
|
||||||
]
|
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "prettyplease"
|
name = "prettyplease"
|
||||||
version = "0.2.37"
|
version = "0.2.37"
|
||||||
@@ -1407,36 +1328,6 @@ version = "6.0.0"
|
|||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "f8dcc9c7d52a811697d2151c701e0d08956f92b0e24136cf4cf27b57a6a0d9bf"
|
checksum = "f8dcc9c7d52a811697d2151c701e0d08956f92b0e24136cf4cf27b57a6a0d9bf"
|
||||||
|
|
||||||
[[package]]
|
|
||||||
name = "rand"
|
|
||||||
version = "0.8.5"
|
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
|
||||||
checksum = "34af8d1a0e25924bc5b7c43c079c942339d8f0a8b57c39049bef581b46327404"
|
|
||||||
dependencies = [
|
|
||||||
"libc",
|
|
||||||
"rand_chacha",
|
|
||||||
"rand_core",
|
|
||||||
]
|
|
||||||
|
|
||||||
[[package]]
|
|
||||||
name = "rand_chacha"
|
|
||||||
version = "0.3.1"
|
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
|
||||||
checksum = "e6c10a63a0fa32252be49d21e7709d4d4baf8d231c2dbce1eaa8141b9b127d88"
|
|
||||||
dependencies = [
|
|
||||||
"ppv-lite86",
|
|
||||||
"rand_core",
|
|
||||||
]
|
|
||||||
|
|
||||||
[[package]]
|
|
||||||
name = "rand_core"
|
|
||||||
version = "0.6.4"
|
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
|
||||||
checksum = "ec0be4795e2f6a28069bec0b5ff3e2ac9bafc99e6a9a7dc3547996c5c816922c"
|
|
||||||
dependencies = [
|
|
||||||
"getrandom 0.2.17",
|
|
||||||
]
|
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "regex"
|
name = "regex"
|
||||||
version = "1.12.3"
|
version = "1.12.3"
|
||||||
@@ -1563,17 +1454,6 @@ dependencies = [
|
|||||||
"serde_core",
|
"serde_core",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
|
||||||
name = "sha1"
|
|
||||||
version = "0.10.6"
|
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
|
||||||
checksum = "e3bf829a2d51ab4a5ddf1352d8470c140cadc8301b2ae1789db023f01cedd6ba"
|
|
||||||
dependencies = [
|
|
||||||
"cfg-if",
|
|
||||||
"cpufeatures",
|
|
||||||
"digest",
|
|
||||||
]
|
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "sharded-slab"
|
name = "sharded-slab"
|
||||||
version = "0.1.7"
|
version = "0.1.7"
|
||||||
@@ -1615,12 +1495,6 @@ dependencies = [
|
|||||||
"windows-sys 0.61.2",
|
"windows-sys 0.61.2",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
|
||||||
name = "static_assertions"
|
|
||||||
version = "1.1.0"
|
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
|
||||||
checksum = "a2eb9349b6444b326872e140eb1cf5e7c522154d69e7a0ffb0fb81c06b37543f"
|
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "strsim"
|
name = "strsim"
|
||||||
version = "0.11.1"
|
version = "0.11.1"
|
||||||
@@ -1692,7 +1566,7 @@ source = "registry+https://github.com/rust-lang/crates.io-index"
|
|||||||
checksum = "32497e9a4c7b38532efcdebeef879707aa9f794296a4f0244f6f69e9bc8574bd"
|
checksum = "32497e9a4c7b38532efcdebeef879707aa9f794296a4f0244f6f69e9bc8574bd"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"fastrand",
|
"fastrand",
|
||||||
"getrandom 0.4.2",
|
"getrandom",
|
||||||
"once_cell",
|
"once_cell",
|
||||||
"rustix",
|
"rustix",
|
||||||
"windows-sys 0.61.2",
|
"windows-sys 0.61.2",
|
||||||
@@ -1791,7 +1665,7 @@ dependencies = [
|
|||||||
"toml_datetime",
|
"toml_datetime",
|
||||||
"toml_parser",
|
"toml_parser",
|
||||||
"toml_writer",
|
"toml_writer",
|
||||||
"winnow",
|
"winnow 1.0.1",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
@@ -1805,14 +1679,14 @@ dependencies = [
|
|||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "toml_edit"
|
name = "toml_edit"
|
||||||
version = "0.25.8+spec-1.1.0"
|
version = "0.25.10+spec-1.1.0"
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "16bff38f1d86c47f9ff0647e6838d7bb362522bdf44006c7068c2b1e606f1f3c"
|
checksum = "a82418ca169e235e6c399a84e395ab6debeb3bc90edc959bf0f48647c6a32d1b"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"indexmap",
|
"indexmap",
|
||||||
"toml_datetime",
|
"toml_datetime",
|
||||||
"toml_parser",
|
"toml_parser",
|
||||||
"winnow",
|
"winnow 1.0.1",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
@@ -1821,7 +1695,7 @@ version = "1.1.2+spec-1.1.0"
|
|||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "a2abe9b86193656635d2411dc43050282ca48aa31c2451210f4202550afb7526"
|
checksum = "a2abe9b86193656635d2411dc43050282ca48aa31c2451210f4202550afb7526"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"winnow",
|
"winnow 1.0.1",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
@@ -1891,12 +1765,6 @@ dependencies = [
|
|||||||
"tracing-log",
|
"tracing-log",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
|
||||||
name = "typenum"
|
|
||||||
version = "1.19.0"
|
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
|
||||||
checksum = "562d481066bde0658276a35467c4af00bdc6ee726305698a55b86e61d7ad82bb"
|
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "uds_windows"
|
name = "uds_windows"
|
||||||
version = "1.2.1"
|
version = "1.2.1"
|
||||||
@@ -1932,8 +1800,9 @@ version = "1.23.0"
|
|||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "5ac8b6f42ead25368cf5b098aeb3dc8a1a2c05a3eee8a9a1a68c640edbfc79d9"
|
checksum = "5ac8b6f42ead25368cf5b098aeb3dc8a1a2c05a3eee8a9a1a68c640edbfc79d9"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"getrandom 0.4.2",
|
"getrandom",
|
||||||
"js-sys",
|
"js-sys",
|
||||||
|
"serde_core",
|
||||||
"wasm-bindgen",
|
"wasm-bindgen",
|
||||||
]
|
]
|
||||||
|
|
||||||
@@ -1943,12 +1812,6 @@ version = "0.1.1"
|
|||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "ba73ea9cf16a25df0c8caa16c51acb937d5712a8429db78a3ee29d5dcacd3a65"
|
checksum = "ba73ea9cf16a25df0c8caa16c51acb937d5712a8429db78a3ee29d5dcacd3a65"
|
||||||
|
|
||||||
[[package]]
|
|
||||||
name = "version_check"
|
|
||||||
version = "0.9.5"
|
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
|
||||||
checksum = "0b928f33d975fc6ad9f86c8f283853ad26bdd5b10b7f1542aa2fa15e2289105a"
|
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "walkdir"
|
name = "walkdir"
|
||||||
version = "2.5.0"
|
version = "2.5.0"
|
||||||
@@ -2194,15 +2057,6 @@ dependencies = [
|
|||||||
"windows-link",
|
"windows-link",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
|
||||||
name = "windows-sys"
|
|
||||||
version = "0.52.0"
|
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
|
||||||
checksum = "282be5f36a8ce781fad8c8ae18fa3f9beff57ec1b52cb3de0789201425d9a33d"
|
|
||||||
dependencies = [
|
|
||||||
"windows-targets 0.52.6",
|
|
||||||
]
|
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "windows-sys"
|
name = "windows-sys"
|
||||||
version = "0.59.0"
|
version = "0.59.0"
|
||||||
@@ -2368,6 +2222,15 @@ version = "0.53.1"
|
|||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "d6bbff5f0aada427a1e5a6da5f1f98158182f26556f345ac9e04d36d0ebed650"
|
checksum = "d6bbff5f0aada427a1e5a6da5f1f98158182f26556f345ac9e04d36d0ebed650"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "winnow"
|
||||||
|
version = "0.7.15"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "df79d97927682d2fd8adb29682d1140b343be4ac0f08fd68b7765d9c059d3945"
|
||||||
|
dependencies = [
|
||||||
|
"memchr",
|
||||||
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "winnow"
|
name = "winnow"
|
||||||
version = "1.0.1"
|
version = "1.0.1"
|
||||||
@@ -2465,25 +2328,14 @@ dependencies = [
|
|||||||
"wasmparser",
|
"wasmparser",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
|
||||||
name = "xdg-home"
|
|
||||||
version = "1.3.0"
|
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
|
||||||
checksum = "ec1cdab258fb55c0da61328dc52c8764709b249011b2cad0454c72f0bf10a1f6"
|
|
||||||
dependencies = [
|
|
||||||
"libc",
|
|
||||||
"windows-sys 0.59.0",
|
|
||||||
]
|
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "zbus"
|
name = "zbus"
|
||||||
version = "4.4.0"
|
version = "5.14.0"
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "bb97012beadd29e654708a0fdb4c84bc046f537aecfde2c3ee0a9e4b4d48c725"
|
checksum = "ca82f95dbd3943a40a53cfded6c2d0a2ca26192011846a1810c4256ef92c60bc"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"async-broadcast",
|
"async-broadcast",
|
||||||
"async-executor",
|
"async-executor",
|
||||||
"async-fs",
|
|
||||||
"async-io",
|
"async-io",
|
||||||
"async-lock",
|
"async-lock",
|
||||||
"async-process",
|
"async-process",
|
||||||
@@ -2494,20 +2346,18 @@ dependencies = [
|
|||||||
"enumflags2",
|
"enumflags2",
|
||||||
"event-listener",
|
"event-listener",
|
||||||
"futures-core",
|
"futures-core",
|
||||||
"futures-sink",
|
"futures-lite",
|
||||||
"futures-util",
|
|
||||||
"hex",
|
"hex",
|
||||||
"nix 0.29.0",
|
"libc",
|
||||||
"ordered-stream",
|
"ordered-stream",
|
||||||
"rand",
|
"rustix",
|
||||||
"serde",
|
"serde",
|
||||||
"serde_repr",
|
"serde_repr",
|
||||||
"sha1",
|
|
||||||
"static_assertions",
|
|
||||||
"tracing",
|
"tracing",
|
||||||
"uds_windows",
|
"uds_windows",
|
||||||
"windows-sys 0.52.0",
|
"uuid",
|
||||||
"xdg-home",
|
"windows-sys 0.61.2",
|
||||||
|
"winnow 0.7.15",
|
||||||
"zbus_macros",
|
"zbus_macros",
|
||||||
"zbus_names",
|
"zbus_names",
|
||||||
"zvariant",
|
"zvariant",
|
||||||
@@ -2515,48 +2365,30 @@ dependencies = [
|
|||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "zbus_macros"
|
name = "zbus_macros"
|
||||||
version = "4.4.0"
|
version = "5.14.0"
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "267db9407081e90bbfa46d841d3cbc60f59c0351838c4bc65199ecd79ab1983e"
|
checksum = "897e79616e84aac4b2c46e9132a4f63b93105d54fe8c0e8f6bffc21fa8d49222"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"proc-macro-crate",
|
"proc-macro-crate",
|
||||||
"proc-macro2",
|
"proc-macro2",
|
||||||
"quote",
|
"quote",
|
||||||
"syn",
|
"syn",
|
||||||
|
"zbus_names",
|
||||||
|
"zvariant",
|
||||||
"zvariant_utils",
|
"zvariant_utils",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "zbus_names"
|
name = "zbus_names"
|
||||||
version = "3.0.0"
|
version = "4.3.1"
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "4b9b1fef7d021261cc16cba64c351d291b715febe0fa10dc3a443ac5a5022e6c"
|
checksum = "ffd8af6d5b78619bab301ff3c560a5bd22426150253db278f164d6cf3b72c50f"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"serde",
|
"serde",
|
||||||
"static_assertions",
|
"winnow 0.7.15",
|
||||||
"zvariant",
|
"zvariant",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
|
||||||
name = "zerocopy"
|
|
||||||
version = "0.8.48"
|
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
|
||||||
checksum = "eed437bf9d6692032087e337407a86f04cd8d6a16a37199ed57949d415bd68e9"
|
|
||||||
dependencies = [
|
|
||||||
"zerocopy-derive",
|
|
||||||
]
|
|
||||||
|
|
||||||
[[package]]
|
|
||||||
name = "zerocopy-derive"
|
|
||||||
version = "0.8.48"
|
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
|
||||||
checksum = "70e3cd084b1788766f53af483dd21f93881ff30d7320490ec3ef7526d203bad4"
|
|
||||||
dependencies = [
|
|
||||||
"proc-macro2",
|
|
||||||
"quote",
|
|
||||||
"syn",
|
|
||||||
]
|
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "zmij"
|
name = "zmij"
|
||||||
version = "1.0.21"
|
version = "1.0.21"
|
||||||
@@ -2565,22 +2397,23 @@ checksum = "b8848ee67ecc8aedbaf3e4122217aff892639231befc6a1b58d29fff4c2cabaa"
|
|||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "zvariant"
|
name = "zvariant"
|
||||||
version = "4.2.0"
|
version = "5.10.0"
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "2084290ab9a1c471c38fc524945837734fbf124487e105daec2bb57fd48c81fe"
|
checksum = "5708299b21903bbe348e94729f22c49c55d04720a004aa350f1f9c122fd2540b"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"endi",
|
"endi",
|
||||||
"enumflags2",
|
"enumflags2",
|
||||||
"serde",
|
"serde",
|
||||||
"static_assertions",
|
"winnow 0.7.15",
|
||||||
"zvariant_derive",
|
"zvariant_derive",
|
||||||
|
"zvariant_utils",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "zvariant_derive"
|
name = "zvariant_derive"
|
||||||
version = "4.2.0"
|
version = "5.10.0"
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "73e2ba546bda683a90652bac4a279bc146adad1386f25379cf73200d2002c449"
|
checksum = "5b59b012ebe9c46656f9cc08d8da8b4c726510aef12559da3e5f1bf72780752c"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"proc-macro-crate",
|
"proc-macro-crate",
|
||||||
"proc-macro2",
|
"proc-macro2",
|
||||||
@@ -2591,11 +2424,13 @@ dependencies = [
|
|||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "zvariant_utils"
|
name = "zvariant_utils"
|
||||||
version = "2.1.0"
|
version = "3.3.0"
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "c51bcff7cc3dbb5055396bcf774748c3dab426b4b8659046963523cee4808340"
|
checksum = "f75c23a64ef8f40f13a6989991e643554d9bef1d682a281160cf0c1bc389c5e9"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"proc-macro2",
|
"proc-macro2",
|
||||||
"quote",
|
"quote",
|
||||||
|
"serde",
|
||||||
"syn",
|
"syn",
|
||||||
|
"winnow 0.7.15",
|
||||||
]
|
]
|
||||||
|
|||||||
+19
-9
@@ -1,12 +1,21 @@
|
|||||||
[package]
|
[package]
|
||||||
name = "fluxo-rs"
|
name = "fluxo-rs"
|
||||||
version = "0.4.0"
|
version = "0.5.1"
|
||||||
edition = "2024"
|
edition = "2024"
|
||||||
|
|
||||||
|
[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]
|
[dependencies]
|
||||||
anyhow = "1.0.102"
|
anyhow = "1.0.102"
|
||||||
clap = { version = "4.6.0", features = ["derive"] }
|
clap = { version = "4.6.0", features = ["derive"] }
|
||||||
ctrlc = "3"
|
ctrlc = "3"
|
||||||
|
libc = "0.2"
|
||||||
regex = "1.12"
|
regex = "1.12"
|
||||||
serde = { version = "1.0.228", features = ["derive"] }
|
serde = { version = "1.0.228", features = ["derive"] }
|
||||||
serde_json = "1.0.149"
|
serde_json = "1.0.149"
|
||||||
@@ -15,16 +24,17 @@ thiserror = "2.0"
|
|||||||
toml = "1.1.2"
|
toml = "1.1.2"
|
||||||
tracing = "0.1.44"
|
tracing = "0.1.44"
|
||||||
tracing-subscriber = { version = "0.3.23", features = ["env-filter"] }
|
tracing-subscriber = { version = "0.3.23", features = ["env-filter"] }
|
||||||
maestro = { git = "https://github.com/qzed/pbpctrl", package = "maestro" }
|
tokio = { version = "1", features = ["rt", "rt-multi-thread", "sync", "time", "macros", "signal", "process", "io-util", "net"] }
|
||||||
bluer = { version = "0.17", features = ["bluetoothd", "rfcomm", "id"] }
|
|
||||||
tokio = { version = "1", features = ["rt", "rt-multi-thread", "sync", "time", "macros", "signal", "process"] }
|
|
||||||
tokio-util = { version = "0.7", features = ["codec", "time"] }
|
tokio-util = { version = "0.7", features = ["codec", "time"] }
|
||||||
futures = "0.3"
|
|
||||||
libpulse-binding = "2.30"
|
|
||||||
nix = { version = "0.31", features = ["net"] }
|
|
||||||
notify = "8.2.0"
|
notify = "8.2.0"
|
||||||
zbus = "4"
|
|
||||||
|
# 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]
|
[dev-dependencies]
|
||||||
|
|
||||||
tempfile = "3"
|
tempfile = "3"
|
||||||
|
|||||||
@@ -29,40 +29,49 @@ backlight = 13
|
|||||||
dnd = 14
|
dnd = 14
|
||||||
|
|
||||||
[network]
|
[network]
|
||||||
|
# enabled = false # set to false to disable this module at runtime
|
||||||
# tokens: {interface}, {ip}, {rx}, {tx}
|
# tokens: {interface}, {ip}, {rx}, {tx}
|
||||||
format = "{interface} ({ip}): {rx:^4.1} MB/s {tx:^4.1} MB/s"
|
format = "{interface} ({ip}): {rx:^4.1} MB/s {tx:^4.1} MB/s"
|
||||||
|
|
||||||
[cpu]
|
[cpu]
|
||||||
|
# enabled = false # set to false to disable this module at runtime
|
||||||
# tokens: {usage}, {temp}, {model}
|
# tokens: {usage}, {temp}, {model}
|
||||||
format = "CPU: {usage:^4.1}% {temp:^4.1}C"
|
format = "CPU: {usage:^4.1}% {temp:^4.1}C"
|
||||||
|
|
||||||
[memory]
|
[memory]
|
||||||
|
# enabled = false # set to false to disable this module at runtime
|
||||||
# tokens: {used}, {total}
|
# tokens: {used}, {total}
|
||||||
format = "MEM: {used:^4.1}/{total:^4.1}GB"
|
format = "MEM: {used:^4.1}/{total:^4.1}GB"
|
||||||
|
|
||||||
[gpu]
|
[gpu]
|
||||||
|
# enabled = false # set to false to disable this module at runtime
|
||||||
# tokens: {usage}, {vram_used}, {vram_total}, {temp}
|
# 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_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_intel = "iGPU: {usage:>3.0}%"
|
||||||
format_nvidia = "NV: {usage:>3.0}% {vram_used:>4.1}/{vram_total:>4.1}GB {temp:>4.1}C"
|
format_nvidia = "NV: {usage:>3.0}% {vram_used:>4.1}/{vram_total:>4.1}GB {temp:>4.1}C"
|
||||||
|
|
||||||
[sys]
|
[sys]
|
||||||
|
# enabled = false # set to false to disable this module at runtime
|
||||||
# tokens: {uptime}, {load1}, {load5}, {load15}, {procs}
|
# tokens: {uptime}, {load1}, {load5}, {load15}, {procs}
|
||||||
format = "UP: {uptime} LOAD: {load1:^3.1} "
|
format = "UP: {uptime} LOAD: {load1:^3.1} "
|
||||||
|
|
||||||
[disk]
|
[disk]
|
||||||
|
# enabled = false # set to false to disable this module at runtime
|
||||||
# tokens: {mount}, {used}, {total}
|
# tokens: {mount}, {used}, {total}
|
||||||
format = "{mount} {used:^3.0}/{total:^3.0}G"
|
format = "{mount} {used:^3.0}/{total:^3.0}G"
|
||||||
|
|
||||||
[pool]
|
[pool]
|
||||||
|
# enabled = false # set to false to disable this module at runtime
|
||||||
# tokens: {used}, {total}
|
# tokens: {used}, {total}
|
||||||
format = "{used:>4.0}G / {total:>4.0}G"
|
format = "{used:>4.0}G / {total:>4.0}G"
|
||||||
|
|
||||||
[power]
|
[power]
|
||||||
|
# enabled = false # set to false to disable this module at runtime
|
||||||
# tokens: {percentage}, {icon}
|
# tokens: {percentage}, {icon}
|
||||||
format = "{percentage:>3}% {icon}"
|
format = "{percentage:>3}% {icon}"
|
||||||
|
|
||||||
[audio]
|
[audio]
|
||||||
|
# enabled = false # set to false to disable this module at runtime
|
||||||
# tokens: {name}, {volume}, {icon}
|
# tokens: {name}, {volume}, {icon}
|
||||||
format_sink_unmuted = "{name} {volume:>3}% {icon}"
|
format_sink_unmuted = "{name} {volume:>3}% {icon}"
|
||||||
format_sink_muted = "{name} {icon}"
|
format_sink_muted = "{name} {icon}"
|
||||||
@@ -70,6 +79,7 @@ format_source_unmuted = "{name} {volume:>3}% {icon}"
|
|||||||
format_source_muted = "{name} {icon}"
|
format_source_muted = "{name} {icon}"
|
||||||
|
|
||||||
[bt]
|
[bt]
|
||||||
|
# enabled = false # set to false to disable this module at runtime
|
||||||
# tokens: {alias}, {mac}, {left}, {right}, {anc}
|
# tokens: {alias}, {mac}, {left}, {right}, {anc}
|
||||||
format_plugin = "{alias} [{left}|{right}] {anc} "
|
format_plugin = "{alias} [{left}|{right}] {anc} "
|
||||||
format_connected = " {alias}"
|
format_connected = " {alias}"
|
||||||
@@ -77,21 +87,30 @@ format_disconnected = " Disconnected"
|
|||||||
format_disabled = " Off"
|
format_disabled = " Off"
|
||||||
|
|
||||||
[game]
|
[game]
|
||||||
|
# enabled = false # set to false to disable this module at runtime
|
||||||
format_active = "<span size='large'></span>"
|
format_active = "<span size='large'></span>"
|
||||||
format_inactive = "<span size='large'></span>"
|
format_inactive = "<span size='large'></span>"
|
||||||
|
|
||||||
[mpris]
|
[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}
|
# tokens: {artist}, {title}, {album}, {status_icon}
|
||||||
format = "{status_icon} {artist} - {title}"
|
format = "{status_icon} {artist} - {title}"
|
||||||
|
|
||||||
[backlight]
|
[backlight]
|
||||||
|
# enabled = false # set to false to disable this module at runtime
|
||||||
# tokens: {percentage}, {icon}
|
# tokens: {percentage}, {icon}
|
||||||
format = "{percentage:>3}% {icon}"
|
format = "{percentage:>3}% {icon}"
|
||||||
|
|
||||||
[keyboard]
|
[keyboard]
|
||||||
|
# enabled = false # set to false to disable this module at runtime
|
||||||
# tokens: {layout}
|
# tokens: {layout}
|
||||||
format = "{layout}"
|
format = "{layout}"
|
||||||
|
|
||||||
[dnd]
|
[dnd]
|
||||||
|
# enabled = false # set to false to disable this module at runtime
|
||||||
format_dnd = "<span size='large'></span>"
|
format_dnd = "<span size='large'></span>"
|
||||||
format_normal = "<span size='large'></span>"
|
format_normal = "<span size='large'></span>"
|
||||||
|
|||||||
+225
-69
@@ -1,50 +1,73 @@
|
|||||||
use regex::Regex;
|
//! 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 serde::Deserialize;
|
||||||
use std::fs;
|
use std::fs;
|
||||||
use std::path::PathBuf;
|
use std::path::PathBuf;
|
||||||
use std::sync::LazyLock;
|
|
||||||
use tracing::{debug, info, warn};
|
use tracing::{debug, info, warn};
|
||||||
|
|
||||||
|
/// Top-level parsed `config.toml`. Each field corresponds to a `[section]`.
|
||||||
#[derive(Deserialize, Default, Clone)]
|
#[derive(Deserialize, Default, Clone)]
|
||||||
pub struct Config {
|
pub struct Config {
|
||||||
#[serde(default)]
|
#[serde(default)]
|
||||||
pub general: GeneralConfig,
|
pub general: GeneralConfig,
|
||||||
#[serde(default)]
|
#[serde(default)]
|
||||||
pub signals: SignalsConfig,
|
pub signals: SignalsConfig,
|
||||||
|
#[cfg(feature = "mod-network")]
|
||||||
#[serde(default)]
|
#[serde(default)]
|
||||||
pub network: NetworkConfig,
|
pub network: NetworkConfig,
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
#[serde(default)]
|
#[serde(default)]
|
||||||
pub cpu: CpuConfig,
|
pub cpu: CpuConfig,
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
#[serde(default)]
|
#[serde(default)]
|
||||||
pub memory: MemoryConfig,
|
pub memory: MemoryConfig,
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
#[serde(default)]
|
#[serde(default)]
|
||||||
pub gpu: GpuConfig,
|
pub gpu: GpuConfig,
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
#[serde(default)]
|
#[serde(default)]
|
||||||
pub sys: SysConfig,
|
pub sys: SysConfig,
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
#[serde(default)]
|
#[serde(default)]
|
||||||
pub disk: DiskConfig,
|
pub disk: DiskConfig,
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
#[serde(default)]
|
#[serde(default)]
|
||||||
pub pool: PoolConfig,
|
pub pool: PoolConfig,
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
#[serde(default)]
|
#[serde(default)]
|
||||||
pub power: PowerConfig,
|
pub power: PowerConfig,
|
||||||
|
#[cfg(feature = "mod-audio")]
|
||||||
#[serde(default)]
|
#[serde(default)]
|
||||||
pub audio: AudioConfig,
|
pub audio: AudioConfig,
|
||||||
|
#[cfg(feature = "mod-bt")]
|
||||||
#[serde(default)]
|
#[serde(default)]
|
||||||
pub bt: BtConfig,
|
pub bt: BtConfig,
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
#[serde(default)]
|
#[serde(default)]
|
||||||
pub game: GameConfig,
|
pub game: GameConfig,
|
||||||
|
#[cfg(feature = "mod-dbus")]
|
||||||
#[serde(default)]
|
#[serde(default)]
|
||||||
pub mpris: MprisConfig,
|
pub mpris: MprisConfig,
|
||||||
|
#[cfg(feature = "mod-dbus")]
|
||||||
#[serde(default)]
|
#[serde(default)]
|
||||||
pub backlight: BacklightConfig,
|
pub backlight: BacklightConfig,
|
||||||
|
#[cfg(feature = "mod-dbus")]
|
||||||
#[serde(default)]
|
#[serde(default)]
|
||||||
pub keyboard: KeyboardConfig,
|
pub keyboard: KeyboardConfig,
|
||||||
|
#[cfg(feature = "mod-dbus")]
|
||||||
#[serde(default)]
|
#[serde(default)]
|
||||||
pub dnd: DndConfig,
|
pub dnd: DndConfig,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Process-wide settings that aren't tied to a single module.
|
||||||
#[derive(Deserialize, Clone)]
|
#[derive(Deserialize, Clone)]
|
||||||
pub struct GeneralConfig {
|
pub struct GeneralConfig {
|
||||||
|
/// Shell command spawned by [`crate::utils::show_menu`] for interactive pickers.
|
||||||
pub menu_command: String,
|
pub menu_command: String,
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -56,6 +79,9 @@ impl Default for GeneralConfig {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Which Waybar `SIGRTMIN+N` signal each module should trigger, if any.
|
||||||
|
///
|
||||||
|
/// `None` disables signalling for that module entirely.
|
||||||
#[allow(dead_code)]
|
#[allow(dead_code)]
|
||||||
#[derive(Deserialize, Default, Clone)]
|
#[derive(Deserialize, Default, Clone)]
|
||||||
pub struct SignalsConfig {
|
pub struct SignalsConfig {
|
||||||
@@ -76,47 +102,62 @@ pub struct SignalsConfig {
|
|||||||
pub dnd: Option<i32>,
|
pub dnd: Option<i32>,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Network module config. `format` tokens: `interface`, `ip`, `rx`, `tx`.
|
||||||
#[derive(Deserialize, Clone)]
|
#[derive(Deserialize, Clone)]
|
||||||
pub struct NetworkConfig {
|
pub struct NetworkConfig {
|
||||||
|
#[serde(default = "default_true")]
|
||||||
|
pub enabled: bool,
|
||||||
pub format: String,
|
pub format: String,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Default for NetworkConfig {
|
impl Default for NetworkConfig {
|
||||||
fn default() -> Self {
|
fn default() -> Self {
|
||||||
Self {
|
Self {
|
||||||
|
enabled: true,
|
||||||
format: "{interface} ({ip}): {rx:>5.2} MB/s {tx:>5.2} MB/s".to_string(),
|
format: "{interface} ({ip}): {rx:>5.2} MB/s {tx:>5.2} MB/s".to_string(),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// CPU module config. `format` tokens: `usage`, `temp`.
|
||||||
#[derive(Deserialize, Clone)]
|
#[derive(Deserialize, Clone)]
|
||||||
pub struct CpuConfig {
|
pub struct CpuConfig {
|
||||||
|
#[serde(default = "default_true")]
|
||||||
|
pub enabled: bool,
|
||||||
pub format: String,
|
pub format: String,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Default for CpuConfig {
|
impl Default for CpuConfig {
|
||||||
fn default() -> Self {
|
fn default() -> Self {
|
||||||
Self {
|
Self {
|
||||||
|
enabled: true,
|
||||||
format: "CPU: {usage:>4.1}% {temp:>4.1}C".to_string(),
|
format: "CPU: {usage:>4.1}% {temp:>4.1}C".to_string(),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Memory module config. `format` tokens: `used`, `total` (gigabytes).
|
||||||
#[derive(Deserialize, Clone)]
|
#[derive(Deserialize, Clone)]
|
||||||
pub struct MemoryConfig {
|
pub struct MemoryConfig {
|
||||||
|
#[serde(default = "default_true")]
|
||||||
|
pub enabled: bool,
|
||||||
pub format: String,
|
pub format: String,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Default for MemoryConfig {
|
impl Default for MemoryConfig {
|
||||||
fn default() -> Self {
|
fn default() -> Self {
|
||||||
Self {
|
Self {
|
||||||
|
enabled: true,
|
||||||
format: "{used:>5.2}/{total:>5.2}GB".to_string(),
|
format: "{used:>5.2}/{total:>5.2}GB".to_string(),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// GPU module config with per-vendor format strings.
|
||||||
#[derive(Deserialize, Clone)]
|
#[derive(Deserialize, Clone)]
|
||||||
pub struct GpuConfig {
|
pub struct GpuConfig {
|
||||||
|
#[serde(default = "default_true")]
|
||||||
|
pub enabled: bool,
|
||||||
pub format_amd: String,
|
pub format_amd: String,
|
||||||
pub format_intel: String,
|
pub format_intel: String,
|
||||||
pub format_nvidia: String,
|
pub format_nvidia: String,
|
||||||
@@ -125,6 +166,7 @@ pub struct GpuConfig {
|
|||||||
impl Default for GpuConfig {
|
impl Default for GpuConfig {
|
||||||
fn default() -> Self {
|
fn default() -> Self {
|
||||||
Self {
|
Self {
|
||||||
|
enabled: true,
|
||||||
format_amd: "AMD: {usage:>3.0}% {vram_used:>4.1}/{vram_total:>4.1}GB {temp:>4.1}C"
|
format_amd: "AMD: {usage:>3.0}% {vram_used:>4.1}/{vram_total:>4.1}GB {temp:>4.1}C"
|
||||||
.to_string(),
|
.to_string(),
|
||||||
format_intel: "iGPU: {usage:>3.0}%".to_string(),
|
format_intel: "iGPU: {usage:>3.0}%".to_string(),
|
||||||
@@ -134,60 +176,79 @@ impl Default for GpuConfig {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Sys module config. `format` tokens: `uptime`, `load1`, `load5`, `load15`, `procs`.
|
||||||
#[derive(Deserialize, Clone)]
|
#[derive(Deserialize, Clone)]
|
||||||
pub struct SysConfig {
|
pub struct SysConfig {
|
||||||
|
#[serde(default = "default_true")]
|
||||||
|
pub enabled: bool,
|
||||||
pub format: String,
|
pub format: String,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Default for SysConfig {
|
impl Default for SysConfig {
|
||||||
fn default() -> Self {
|
fn default() -> Self {
|
||||||
Self {
|
Self {
|
||||||
|
enabled: true,
|
||||||
format: "UP: {uptime} | LOAD: {load1:>4.2} {load5:>4.2} {load15:>4.2}".to_string(),
|
format: "UP: {uptime} | LOAD: {load1:>4.2} {load5:>4.2} {load15:>4.2}".to_string(),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Disk module config. `format` tokens: `mount`, `used`, `total`.
|
||||||
#[derive(Deserialize, Clone)]
|
#[derive(Deserialize, Clone)]
|
||||||
pub struct DiskConfig {
|
pub struct DiskConfig {
|
||||||
|
#[serde(default = "default_true")]
|
||||||
|
pub enabled: bool,
|
||||||
pub format: String,
|
pub format: String,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Default for DiskConfig {
|
impl Default for DiskConfig {
|
||||||
fn default() -> Self {
|
fn default() -> Self {
|
||||||
Self {
|
Self {
|
||||||
|
enabled: true,
|
||||||
format: "{mount} {used:>5.1}/{total:>5.1}G".to_string(),
|
format: "{mount} {used:>5.1}/{total:>5.1}G".to_string(),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Btrfs pool module config. `format` tokens: `used`, `total`.
|
||||||
#[derive(Deserialize, Clone)]
|
#[derive(Deserialize, Clone)]
|
||||||
pub struct PoolConfig {
|
pub struct PoolConfig {
|
||||||
|
#[serde(default = "default_true")]
|
||||||
|
pub enabled: bool,
|
||||||
pub format: String,
|
pub format: String,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Default for PoolConfig {
|
impl Default for PoolConfig {
|
||||||
fn default() -> Self {
|
fn default() -> Self {
|
||||||
Self {
|
Self {
|
||||||
|
enabled: true,
|
||||||
format: "{used:>4.0}G / {total:>4.0}G".to_string(),
|
format: "{used:>4.0}G / {total:>4.0}G".to_string(),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Battery/power module config. `format` tokens: `percentage`, `icon`.
|
||||||
#[derive(Deserialize, Clone)]
|
#[derive(Deserialize, Clone)]
|
||||||
pub struct PowerConfig {
|
pub struct PowerConfig {
|
||||||
|
#[serde(default = "default_true")]
|
||||||
|
pub enabled: bool,
|
||||||
pub format: String,
|
pub format: String,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Default for PowerConfig {
|
impl Default for PowerConfig {
|
||||||
fn default() -> Self {
|
fn default() -> Self {
|
||||||
Self {
|
Self {
|
||||||
|
enabled: true,
|
||||||
format: "{percentage:>3}% {icon}".to_string(),
|
format: "{percentage:>3}% {icon}".to_string(),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Audio module config, one format per (sink|source) × (muted|unmuted) state.
|
||||||
#[derive(Deserialize, Clone)]
|
#[derive(Deserialize, Clone)]
|
||||||
pub struct AudioConfig {
|
pub struct AudioConfig {
|
||||||
|
#[serde(default = "default_true")]
|
||||||
|
pub enabled: bool,
|
||||||
pub format_sink_unmuted: String,
|
pub format_sink_unmuted: String,
|
||||||
pub format_sink_muted: String,
|
pub format_sink_muted: String,
|
||||||
pub format_source_unmuted: String,
|
pub format_source_unmuted: String,
|
||||||
@@ -197,6 +258,7 @@ pub struct AudioConfig {
|
|||||||
impl Default for AudioConfig {
|
impl Default for AudioConfig {
|
||||||
fn default() -> Self {
|
fn default() -> Self {
|
||||||
Self {
|
Self {
|
||||||
|
enabled: true,
|
||||||
format_sink_unmuted: "{name} {volume:>3}% {icon}".to_string(),
|
format_sink_unmuted: "{name} {volume:>3}% {icon}".to_string(),
|
||||||
format_sink_muted: "{name} {icon}".to_string(),
|
format_sink_muted: "{name} {icon}".to_string(),
|
||||||
format_source_unmuted: "{name} {volume:>3}% {icon}".to_string(),
|
format_source_unmuted: "{name} {volume:>3}% {icon}".to_string(),
|
||||||
@@ -205,8 +267,12 @@ impl Default for AudioConfig {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Bluetooth module config. Plugin line tokens: `alias`, `left`, `right`,
|
||||||
|
/// `anc`, `mac`. Plain connect line tokens: `alias`.
|
||||||
#[derive(Deserialize, Clone)]
|
#[derive(Deserialize, Clone)]
|
||||||
pub struct BtConfig {
|
pub struct BtConfig {
|
||||||
|
#[serde(default = "default_true")]
|
||||||
|
pub enabled: bool,
|
||||||
pub format_connected: String,
|
pub format_connected: String,
|
||||||
pub format_plugin: String,
|
pub format_plugin: String,
|
||||||
pub format_disconnected: String,
|
pub format_disconnected: String,
|
||||||
@@ -216,6 +282,7 @@ pub struct BtConfig {
|
|||||||
impl Default for BtConfig {
|
impl Default for BtConfig {
|
||||||
fn default() -> Self {
|
fn default() -> Self {
|
||||||
Self {
|
Self {
|
||||||
|
enabled: true,
|
||||||
format_connected: "{alias} ".to_string(),
|
format_connected: "{alias} ".to_string(),
|
||||||
format_plugin: "{alias} [{left}|{right}] {anc} ".to_string(),
|
format_plugin: "{alias} [{left}|{right}] {anc} ".to_string(),
|
||||||
format_disconnected: "".to_string(),
|
format_disconnected: "".to_string(),
|
||||||
@@ -224,8 +291,11 @@ impl Default for BtConfig {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Gamemode indicator config (active/inactive glyphs).
|
||||||
#[derive(Deserialize, Clone)]
|
#[derive(Deserialize, Clone)]
|
||||||
pub struct GameConfig {
|
pub struct GameConfig {
|
||||||
|
#[serde(default = "default_true")]
|
||||||
|
pub enabled: bool,
|
||||||
pub format_active: String,
|
pub format_active: String,
|
||||||
pub format_inactive: String,
|
pub format_inactive: String,
|
||||||
}
|
}
|
||||||
@@ -233,71 +303,110 @@ pub struct GameConfig {
|
|||||||
impl Default for GameConfig {
|
impl Default for GameConfig {
|
||||||
fn default() -> Self {
|
fn default() -> Self {
|
||||||
Self {
|
Self {
|
||||||
|
enabled: true,
|
||||||
format_active: "<span size='large'></span>".to_string(),
|
format_active: "<span size='large'></span>".to_string(),
|
||||||
format_inactive: "<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)]
|
#[derive(Deserialize, Clone)]
|
||||||
pub struct MprisConfig {
|
pub struct MprisConfig {
|
||||||
|
#[serde(default = "default_true")]
|
||||||
|
pub enabled: bool,
|
||||||
pub format: String,
|
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 {
|
impl Default for MprisConfig {
|
||||||
fn default() -> Self {
|
fn default() -> Self {
|
||||||
Self {
|
Self {
|
||||||
|
enabled: true,
|
||||||
format: "{status_icon} {artist} - {title}".to_string(),
|
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)]
|
#[derive(Deserialize, Clone)]
|
||||||
pub struct BacklightConfig {
|
pub struct BacklightConfig {
|
||||||
|
#[serde(default = "default_true")]
|
||||||
|
pub enabled: bool,
|
||||||
pub format: String,
|
pub format: String,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Default for BacklightConfig {
|
impl Default for BacklightConfig {
|
||||||
fn default() -> Self {
|
fn default() -> Self {
|
||||||
Self {
|
Self {
|
||||||
|
enabled: true,
|
||||||
format: "{percentage:>3}% {icon}".to_string(),
|
format: "{percentage:>3}% {icon}".to_string(),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Keyboard layout module config. `format` tokens: `layout`.
|
||||||
#[derive(Deserialize, Clone)]
|
#[derive(Deserialize, Clone)]
|
||||||
pub struct KeyboardConfig {
|
pub struct KeyboardConfig {
|
||||||
|
#[serde(default = "default_true")]
|
||||||
|
pub enabled: bool,
|
||||||
pub format: String,
|
pub format: String,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Default for KeyboardConfig {
|
impl Default for KeyboardConfig {
|
||||||
fn default() -> Self {
|
fn default() -> Self {
|
||||||
Self {
|
Self {
|
||||||
|
enabled: true,
|
||||||
format: "{layout}".to_string(),
|
format: "{layout}".to_string(),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Do-Not-Disturb indicator config (dnd/normal glyphs).
|
||||||
#[derive(Deserialize, Clone)]
|
#[derive(Deserialize, Clone)]
|
||||||
pub struct DndConfig {
|
pub struct DndConfig {
|
||||||
|
#[serde(default = "default_true")]
|
||||||
|
pub enabled: bool,
|
||||||
pub format_dnd: String,
|
pub format_dnd: String,
|
||||||
pub format_normal: String,
|
pub format_normal: String,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
fn default_true() -> bool {
|
||||||
|
true
|
||||||
|
}
|
||||||
|
|
||||||
impl Default for DndConfig {
|
impl Default for DndConfig {
|
||||||
fn default() -> Self {
|
fn default() -> Self {
|
||||||
Self {
|
Self {
|
||||||
|
enabled: true,
|
||||||
format_dnd: "<span size='large'></span>".to_string(),
|
format_dnd: "<span size='large'></span>".to_string(),
|
||||||
format_normal: "<span size='large'></span>".to_string(),
|
format_normal: "<span size='large'></span>".to_string(),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
static TOKEN_RE: LazyLock<Regex> =
|
|
||||||
LazyLock::new(|| Regex::new(r"\{([a-zA-Z0-9_]+)(?::([<>\^])?(\d+)?(?:\.(\d+))?)?\}").unwrap());
|
|
||||||
|
|
||||||
fn extract_tokens(format_str: &str) -> Vec<String> {
|
fn extract_tokens(format_str: &str) -> Vec<String> {
|
||||||
TOKEN_RE
|
crate::utils::TOKEN_RE
|
||||||
.captures_iter(format_str)
|
.captures_iter(format_str)
|
||||||
.map(|cap| cap[1].to_string())
|
.map(|cap| cap[1].to_string())
|
||||||
.collect()
|
.collect()
|
||||||
@@ -315,79 +424,126 @@ fn validate_format(label: &str, format_str: &str, known_tokens: &[&str]) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
impl Config {
|
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) {
|
pub fn validate(&self) {
|
||||||
|
#[cfg(feature = "mod-network")]
|
||||||
validate_format(
|
validate_format(
|
||||||
"network",
|
"network",
|
||||||
&self.network.format,
|
&self.network.format,
|
||||||
&["interface", "ip", "rx", "tx"],
|
&["interface", "ip", "rx", "tx"],
|
||||||
);
|
);
|
||||||
validate_format("cpu", &self.cpu.format, &["usage", "temp"]);
|
#[cfg(feature = "mod-hardware")]
|
||||||
validate_format("memory", &self.memory.format, &["used", "total"]);
|
{
|
||||||
validate_format(
|
validate_format("cpu", &self.cpu.format, &["usage", "temp"]);
|
||||||
"gpu.amd",
|
validate_format("memory", &self.memory.format, &["used", "total"]);
|
||||||
&self.gpu.format_amd,
|
validate_format(
|
||||||
&["usage", "vram_used", "vram_total", "temp"],
|
"gpu.amd",
|
||||||
);
|
&self.gpu.format_amd,
|
||||||
validate_format("gpu.intel", &self.gpu.format_intel, &["usage", "freq"]);
|
&["usage", "vram_used", "vram_total", "temp"],
|
||||||
validate_format(
|
);
|
||||||
"gpu.nvidia",
|
validate_format("gpu.intel", &self.gpu.format_intel, &["usage", "freq"]);
|
||||||
&self.gpu.format_nvidia,
|
validate_format(
|
||||||
&["usage", "vram_used", "vram_total", "temp"],
|
"gpu.nvidia",
|
||||||
);
|
&self.gpu.format_nvidia,
|
||||||
validate_format(
|
&["usage", "vram_used", "vram_total", "temp"],
|
||||||
"sys",
|
);
|
||||||
&self.sys.format,
|
validate_format(
|
||||||
&["uptime", "load1", "load5", "load15", "procs"],
|
"sys",
|
||||||
);
|
&self.sys.format,
|
||||||
validate_format("disk", &self.disk.format, &["mount", "used", "total"]);
|
&["uptime", "load1", "load5", "load15", "procs"],
|
||||||
validate_format("pool", &self.pool.format, &["used", "total"]);
|
);
|
||||||
validate_format("power", &self.power.format, &["percentage", "icon"]);
|
validate_format("disk", &self.disk.format, &["mount", "used", "total"]);
|
||||||
validate_format(
|
validate_format("pool", &self.pool.format, &["used", "total"]);
|
||||||
"audio.sink_unmuted",
|
validate_format("power", &self.power.format, &["percentage", "icon"]);
|
||||||
&self.audio.format_sink_unmuted,
|
}
|
||||||
&["name", "icon", "volume"],
|
#[cfg(feature = "mod-audio")]
|
||||||
);
|
{
|
||||||
validate_format(
|
validate_format(
|
||||||
"audio.sink_muted",
|
"audio.sink_unmuted",
|
||||||
&self.audio.format_sink_muted,
|
&self.audio.format_sink_unmuted,
|
||||||
&["name", "icon"],
|
&["name", "icon", "volume"],
|
||||||
);
|
);
|
||||||
validate_format(
|
validate_format(
|
||||||
"audio.source_unmuted",
|
"audio.sink_muted",
|
||||||
&self.audio.format_source_unmuted,
|
&self.audio.format_sink_muted,
|
||||||
&["name", "icon", "volume"],
|
&["name", "icon"],
|
||||||
);
|
);
|
||||||
validate_format(
|
validate_format(
|
||||||
"audio.source_muted",
|
"audio.source_unmuted",
|
||||||
&self.audio.format_source_muted,
|
&self.audio.format_source_unmuted,
|
||||||
&["name", "icon"],
|
&["name", "icon", "volume"],
|
||||||
);
|
);
|
||||||
validate_format("bt.connected", &self.bt.format_connected, &["alias"]);
|
validate_format(
|
||||||
validate_format(
|
"audio.source_muted",
|
||||||
"bt.plugin",
|
&self.audio.format_source_muted,
|
||||||
&self.bt.format_plugin,
|
&["name", "icon"],
|
||||||
&["alias", "left", "right", "anc", "mac"],
|
);
|
||||||
);
|
}
|
||||||
validate_format(
|
#[cfg(feature = "mod-bt")]
|
||||||
"mpris",
|
{
|
||||||
&self.mpris.format,
|
validate_format("bt.connected", &self.bt.format_connected, &["alias"]);
|
||||||
&["artist", "title", "album", "status_icon"],
|
validate_format(
|
||||||
);
|
"bt.plugin",
|
||||||
validate_format("backlight", &self.backlight.format, &["percentage", "icon"]);
|
&self.bt.format_plugin,
|
||||||
validate_format("keyboard", &self.keyboard.format, &["layout"]);
|
&["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 {
|
pub fn load_config(custom_path: Option<PathBuf>) -> Config {
|
||||||
let config_path = custom_path.unwrap_or_else(|| {
|
let config_path = custom_path.unwrap_or_else(default_config_path);
|
||||||
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")
|
|
||||||
});
|
|
||||||
|
|
||||||
if let Ok(content) = fs::read_to_string(&config_path) {
|
if let Ok(content) = fs::read_to_string(&config_path) {
|
||||||
match toml::from_str::<Config>(&content) {
|
match toml::from_str::<Config>(&content) {
|
||||||
|
|||||||
+309
-398
@@ -1,14 +1,35 @@
|
|||||||
|
//! 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.
|
||||||
|
|
||||||
use crate::config::Config;
|
use crate::config::Config;
|
||||||
use crate::error::FluxoError;
|
|
||||||
use crate::ipc::socket_path;
|
use crate::ipc::socket_path;
|
||||||
use crate::modules::WaybarModule;
|
#[cfg(feature = "mod-audio")]
|
||||||
use crate::modules::audio::AudioDaemon;
|
use crate::modules::audio::AudioDaemon;
|
||||||
|
#[cfg(feature = "mod-dbus")]
|
||||||
use crate::modules::backlight::BacklightDaemon;
|
use crate::modules::backlight::BacklightDaemon;
|
||||||
|
#[cfg(feature = "mod-bt")]
|
||||||
use crate::modules::bt::BtDaemon;
|
use crate::modules::bt::BtDaemon;
|
||||||
|
#[cfg(feature = "mod-dbus")]
|
||||||
use crate::modules::dnd::DndDaemon;
|
use crate::modules::dnd::DndDaemon;
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
use crate::modules::hardware::HardwareDaemon;
|
use crate::modules::hardware::HardwareDaemon;
|
||||||
|
#[cfg(feature = "mod-dbus")]
|
||||||
use crate::modules::keyboard::KeyboardDaemon;
|
use crate::modules::keyboard::KeyboardDaemon;
|
||||||
|
#[cfg(feature = "mod-dbus")]
|
||||||
use crate::modules::mpris::MprisDaemon;
|
use crate::modules::mpris::MprisDaemon;
|
||||||
|
#[cfg(feature = "mod-network")]
|
||||||
use crate::modules::network::NetworkDaemon;
|
use crate::modules::network::NetworkDaemon;
|
||||||
use crate::signaler::WaybarSignaler;
|
use crate::signaler::WaybarSignaler;
|
||||||
use crate::state::AppReceivers;
|
use crate::state::AppReceivers;
|
||||||
@@ -21,9 +42,77 @@ use std::sync::Arc;
|
|||||||
use tokio::io::{AsyncBufReadExt, AsyncWriteExt, BufReader};
|
use tokio::io::{AsyncBufReadExt, AsyncWriteExt, BufReader};
|
||||||
use tokio::net::UnixListener;
|
use tokio::net::UnixListener;
|
||||||
use tokio::sync::{RwLock, mpsc, watch};
|
use tokio::sync::{RwLock, mpsc, watch};
|
||||||
use tokio::time::{Duration, Instant, sleep};
|
use tokio::time::{Duration, sleep};
|
||||||
use tokio_util::sync::CancellationToken;
|
use tokio_util::sync::CancellationToken;
|
||||||
use tracing::{debug, error, info, warn};
|
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 {
|
struct SocketGuard {
|
||||||
path: String,
|
path: String,
|
||||||
@@ -37,17 +126,14 @@ impl Drop for SocketGuard {
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn get_config_path(custom_path: Option<PathBuf>) -> PathBuf {
|
fn get_config_path(custom_path: Option<PathBuf>) -> PathBuf {
|
||||||
custom_path.unwrap_or_else(|| {
|
custom_path.unwrap_or_else(crate::config::default_config_path)
|
||||||
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")
|
|
||||||
})
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// 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<()> {
|
pub async fn run_daemon(config_path: Option<PathBuf>) -> Result<()> {
|
||||||
let sock_path = socket_path();
|
let sock_path = socket_path();
|
||||||
|
|
||||||
@@ -56,37 +142,77 @@ pub async fn run_daemon(config_path: Option<PathBuf>) -> Result<()> {
|
|||||||
fs::remove_file(&sock_path)?;
|
fs::remove_file(&sock_path)?;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[cfg(feature = "mod-network")]
|
||||||
let (net_tx, net_rx) = watch::channel(Default::default());
|
let (net_tx, net_rx) = watch::channel(Default::default());
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
let (cpu_tx, cpu_rx) = watch::channel(Default::default());
|
let (cpu_tx, cpu_rx) = watch::channel(Default::default());
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
let (mem_tx, mem_rx) = watch::channel(Default::default());
|
let (mem_tx, mem_rx) = watch::channel(Default::default());
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
let (sys_tx, sys_rx) = watch::channel(Default::default());
|
let (sys_tx, sys_rx) = watch::channel(Default::default());
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
let (gpu_tx, gpu_rx) = watch::channel(Default::default());
|
let (gpu_tx, gpu_rx) = watch::channel(Default::default());
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
let (disks_tx, disks_rx) = watch::channel(Default::default());
|
let (disks_tx, disks_rx) = watch::channel(Default::default());
|
||||||
|
#[cfg(feature = "mod-bt")]
|
||||||
let (bt_tx, bt_rx) = watch::channel(Default::default());
|
let (bt_tx, bt_rx) = watch::channel(Default::default());
|
||||||
|
#[cfg(feature = "mod-audio")]
|
||||||
let (audio_tx, audio_rx) = watch::channel(Default::default());
|
let (audio_tx, audio_rx) = watch::channel(Default::default());
|
||||||
|
#[cfg(feature = "mod-dbus")]
|
||||||
let (mpris_tx, mpris_rx) = watch::channel(Default::default());
|
let (mpris_tx, mpris_rx) = watch::channel(Default::default());
|
||||||
|
#[cfg(feature = "mod-dbus")]
|
||||||
let (backlight_tx, backlight_rx) = watch::channel(Default::default());
|
let (backlight_tx, backlight_rx) = watch::channel(Default::default());
|
||||||
|
#[cfg(feature = "mod-dbus")]
|
||||||
let (keyboard_tx, keyboard_rx) = watch::channel(Default::default());
|
let (keyboard_tx, keyboard_rx) = watch::channel(Default::default());
|
||||||
|
#[cfg(feature = "mod-dbus")]
|
||||||
let (dnd_tx, dnd_rx) = watch::channel(Default::default());
|
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()));
|
let health = Arc::new(RwLock::new(HashMap::new()));
|
||||||
|
#[cfg(feature = "mod-bt")]
|
||||||
let (bt_force_tx, mut bt_force_rx) = mpsc::channel(1);
|
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 (audio_cmd_tx, audio_cmd_rx) = mpsc::channel(8);
|
||||||
|
|
||||||
let receivers = AppReceivers {
|
let receivers = AppReceivers {
|
||||||
|
#[cfg(feature = "mod-network")]
|
||||||
network: net_rx,
|
network: net_rx,
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
cpu: cpu_rx,
|
cpu: cpu_rx,
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
memory: mem_rx,
|
memory: mem_rx,
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
sys: sys_rx,
|
sys: sys_rx,
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
gpu: gpu_rx,
|
gpu: gpu_rx,
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
disks: disks_rx,
|
disks: disks_rx,
|
||||||
|
#[cfg(feature = "mod-bt")]
|
||||||
bluetooth: bt_rx,
|
bluetooth: bt_rx,
|
||||||
|
#[cfg(feature = "mod-bt")]
|
||||||
|
bt_cycle,
|
||||||
|
#[cfg(feature = "mod-audio")]
|
||||||
audio: audio_rx,
|
audio: audio_rx,
|
||||||
|
#[cfg(feature = "mod-dbus")]
|
||||||
mpris: mpris_rx,
|
mpris: mpris_rx,
|
||||||
|
#[cfg(feature = "mod-dbus")]
|
||||||
backlight: backlight_rx,
|
backlight: backlight_rx,
|
||||||
|
#[cfg(feature = "mod-dbus")]
|
||||||
keyboard: keyboard_rx,
|
keyboard: keyboard_rx,
|
||||||
|
#[cfg(feature = "mod-dbus")]
|
||||||
dnd: dnd_rx,
|
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),
|
health: Arc::clone(&health),
|
||||||
|
#[cfg(feature = "mod-bt")]
|
||||||
bt_force_poll: bt_force_tx,
|
bt_force_poll: bt_force_tx,
|
||||||
|
#[cfg(feature = "mod-audio")]
|
||||||
audio_cmd_tx,
|
audio_cmd_tx,
|
||||||
};
|
};
|
||||||
|
|
||||||
@@ -95,7 +221,8 @@ pub async fn run_daemon(config_path: Option<PathBuf>) -> Result<()> {
|
|||||||
path: sock_path.clone(),
|
path: sock_path.clone(),
|
||||||
};
|
};
|
||||||
|
|
||||||
// Signal handling for graceful shutdown
|
// 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 cancel_token = CancellationToken::new();
|
||||||
let token_clone = cancel_token.clone();
|
let token_clone = cancel_token.clone();
|
||||||
tokio::spawn(async move {
|
tokio::spawn(async move {
|
||||||
@@ -106,10 +233,140 @@ pub async fn run_daemon(config_path: Option<PathBuf>) -> Result<()> {
|
|||||||
|
|
||||||
let resolved_config_path = get_config_path(config_path.clone());
|
let resolved_config_path = get_config_path(config_path.clone());
|
||||||
let config = Arc::new(RwLock::new(crate::config::load_config(config_path.clone())));
|
let config = Arc::new(RwLock::new(crate::config::load_config(config_path.clone())));
|
||||||
|
spawn_config_watchers(&config, &resolved_config_path);
|
||||||
|
|
||||||
// 0. Config Watcher (Hot Reload)
|
#[cfg(feature = "mod-network")]
|
||||||
let watcher_config = Arc::clone(&config);
|
if config.read().await.network.enabled {
|
||||||
let watcher_path = resolved_config_path.clone();
|
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 {
|
tokio::spawn(async move {
|
||||||
let (ev_tx, mut ev_rx) = mpsc::channel(1);
|
let (ev_tx, mut ev_rx) = mpsc::channel(1);
|
||||||
let mut watcher = RecommendedWatcher::new(
|
let mut watcher = RecommendedWatcher::new(
|
||||||
@@ -133,7 +390,7 @@ pub async fn run_daemon(config_path: Option<PathBuf>) -> Result<()> {
|
|||||||
loop {
|
loop {
|
||||||
tokio::select! {
|
tokio::select! {
|
||||||
_ = ev_rx.recv() => {
|
_ = ev_rx.recv() => {
|
||||||
// Debounce reloads
|
// Coalesce rapid editor writes into one reload.
|
||||||
sleep(Duration::from_millis(100)).await;
|
sleep(Duration::from_millis(100)).await;
|
||||||
while ev_rx.try_recv().is_ok() {}
|
while ev_rx.try_recv().is_ok() {}
|
||||||
reload_config(&watcher_config, Some(watcher_path.clone())).await;
|
reload_config(&watcher_config, Some(watcher_path.clone())).await;
|
||||||
@@ -142,9 +399,8 @@ pub async fn run_daemon(config_path: Option<PathBuf>) -> Result<()> {
|
|||||||
}
|
}
|
||||||
});
|
});
|
||||||
|
|
||||||
// 0.1 SIGHUP Handler
|
let hup_config = Arc::clone(config);
|
||||||
let hup_config = Arc::clone(&config);
|
let hup_path = resolved_path.to_path_buf();
|
||||||
let hup_path = resolved_config_path.clone();
|
|
||||||
tokio::spawn(async move {
|
tokio::spawn(async move {
|
||||||
use tokio::signal::unix::{SignalKind, signal};
|
use tokio::signal::unix::{SignalKind, signal};
|
||||||
let mut stream = signal(SignalKind::hangup()).unwrap();
|
let mut stream = signal(SignalKind::hangup()).unwrap();
|
||||||
@@ -154,120 +410,24 @@ pub async fn run_daemon(config_path: Option<PathBuf>) -> Result<()> {
|
|||||||
reload_config(&hup_config, Some(hup_path.clone())).await;
|
reload_config(&hup_config, Some(hup_path.clone())).await;
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
|
}
|
||||||
|
|
||||||
// 1. Network Task
|
/// Accept loop for the client Unix socket.
|
||||||
let token = cancel_token.clone();
|
///
|
||||||
let net_health = Arc::clone(&health);
|
/// Each client request spawns a short-lived task that reads one line, looks
|
||||||
tokio::spawn(async move {
|
/// up the module via [`crate::registry::dispatch`], and writes the JSON
|
||||||
info!("Starting Network polling task");
|
/// response back. Broken-pipe errors are logged at `debug` — they just mean
|
||||||
let mut daemon = NetworkDaemon::new();
|
/// the client timed out before we responded.
|
||||||
loop {
|
async fn run_ipc_loop(
|
||||||
if !is_in_backoff("net", &net_health).await {
|
listener: UnixListener,
|
||||||
let res = daemon.poll(&net_tx).await;
|
receivers: AppReceivers,
|
||||||
handle_poll_result("net", res, &net_health).await;
|
config: Arc<RwLock<Config>>,
|
||||||
}
|
config_path: Option<PathBuf>,
|
||||||
tokio::select! {
|
cancel_token: CancellationToken,
|
||||||
_ = token.cancelled() => break,
|
) -> Result<()> {
|
||||||
_ = sleep(Duration::from_secs(1)) => {}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
info!("Network task shut down.");
|
|
||||||
});
|
|
||||||
|
|
||||||
// 2. Fast Hardware Task (CPU, Mem, Load)
|
|
||||||
let token = cancel_token.clone();
|
|
||||||
let hw_health = Arc::clone(&health);
|
|
||||||
tokio::spawn(async move {
|
|
||||||
info!("Starting Fast Hardware polling task");
|
|
||||||
let mut daemon = HardwareDaemon::new();
|
|
||||||
loop {
|
|
||||||
if !is_in_backoff("cpu", &hw_health).await {
|
|
||||||
daemon.poll_fast(&cpu_tx, &mem_tx, &sys_tx).await;
|
|
||||||
}
|
|
||||||
tokio::select! {
|
|
||||||
_ = token.cancelled() => break,
|
|
||||||
_ = sleep(Duration::from_secs(1)) => {}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
info!("Fast Hardware task shut down.");
|
|
||||||
});
|
|
||||||
|
|
||||||
// 3. Slow Hardware Task (GPU, Disks)
|
|
||||||
let token = cancel_token.clone();
|
|
||||||
let slow_health = Arc::clone(&health);
|
|
||||||
tokio::spawn(async move {
|
|
||||||
info!("Starting Slow Hardware polling task");
|
|
||||||
let mut daemon = HardwareDaemon::new();
|
|
||||||
loop {
|
|
||||||
if !is_in_backoff("gpu", &slow_health).await {
|
|
||||||
daemon.poll_slow(&gpu_tx, &disks_tx).await;
|
|
||||||
}
|
|
||||||
tokio::select! {
|
|
||||||
_ = token.cancelled() => break,
|
|
||||||
_ = sleep(Duration::from_secs(5)) => {}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
info!("Slow Hardware task shut down.");
|
|
||||||
});
|
|
||||||
|
|
||||||
// 4. Bluetooth Task
|
|
||||||
let token = cancel_token.clone();
|
|
||||||
let bt_health = Arc::clone(&health);
|
|
||||||
let poll_config = Arc::clone(&config);
|
|
||||||
let poll_receivers = receivers.clone();
|
|
||||||
tokio::spawn(async move {
|
|
||||||
info!("Starting Bluetooth polling task");
|
|
||||||
let mut daemon = BtDaemon::new();
|
|
||||||
loop {
|
|
||||||
if !is_in_backoff("bt", &bt_health).await {
|
|
||||||
let config = poll_config.read().await;
|
|
||||||
daemon.poll(&bt_tx, &poll_receivers, &config).await;
|
|
||||||
}
|
|
||||||
tokio::select! {
|
|
||||||
_ = token.cancelled() => break,
|
|
||||||
_ = bt_force_rx.recv() => {},
|
|
||||||
_ = sleep(Duration::from_secs(2)) => {}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
info!("Bluetooth task shut down.");
|
|
||||||
});
|
|
||||||
|
|
||||||
// 5. Audio Thread (Event driven)
|
|
||||||
let audio_daemon = AudioDaemon::new();
|
|
||||||
audio_daemon.start(&audio_tx, audio_cmd_rx);
|
|
||||||
|
|
||||||
// 5.1 Backlight Thread (Event driven)
|
|
||||||
let backlight_daemon = BacklightDaemon::new();
|
|
||||||
backlight_daemon.start(backlight_tx);
|
|
||||||
|
|
||||||
// 5.2 Keyboard Thread (Event driven)
|
|
||||||
let keyboard_daemon = KeyboardDaemon::new();
|
|
||||||
keyboard_daemon.start(keyboard_tx);
|
|
||||||
|
|
||||||
// 5.3 DND Thread (Event driven)
|
|
||||||
let dnd_daemon = DndDaemon::new();
|
|
||||||
dnd_daemon.start(dnd_tx);
|
|
||||||
|
|
||||||
// 5.4 MPRIS Thread
|
|
||||||
let mpris_daemon = MprisDaemon::new();
|
|
||||||
mpris_daemon.start(mpris_tx);
|
|
||||||
|
|
||||||
// 6. Waybar Signaler Task
|
|
||||||
let signaler = WaybarSignaler::new();
|
|
||||||
let sig_config = Arc::clone(&config);
|
|
||||||
let sig_receivers = receivers.clone();
|
|
||||||
tokio::spawn(async move {
|
|
||||||
info!("Starting Waybar Signaler task");
|
|
||||||
signaler.run(sig_config, sig_receivers).await;
|
|
||||||
});
|
|
||||||
|
|
||||||
info!("Fluxo daemon successfully bound to socket: {}", sock_path);
|
|
||||||
|
|
||||||
loop {
|
loop {
|
||||||
tokio::select! {
|
tokio::select! {
|
||||||
_ = cancel_token.cancelled() => {
|
_ = cancel_token.cancelled() => break,
|
||||||
break;
|
|
||||||
}
|
|
||||||
res = listener.accept() => {
|
res = listener.accept() => {
|
||||||
match res {
|
match res {
|
||||||
Ok((mut stream, _)) => {
|
Ok((mut stream, _)) => {
|
||||||
@@ -324,59 +484,7 @@ pub async fn run_daemon(config_path: Option<PathBuf>) -> Result<()> {
|
|||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn handle_poll_result(
|
/// Re-read the configuration file and swap it into the shared lock.
|
||||||
module_name: &str,
|
|
||||||
result: crate::error::Result<()>,
|
|
||||||
health_lock: &Arc<RwLock<HashMap<String, crate::state::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() {
|
|
||||||
// Fatal errors trigger immediate long backoff
|
|
||||||
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 {
|
|
||||||
// Exponential backoff for transient errors: 30s, 60s, 120s...
|
|
||||||
let backoff_secs = 30 * (2u64.pow(health.consecutive_failures.saturating_sub(3)));
|
|
||||||
let backoff_secs = backoff_secs.min(3600); // Cap at 1 hour
|
|
||||||
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);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn is_in_backoff(
|
|
||||||
module_name: &str,
|
|
||||||
health_lock: &Arc<RwLock<HashMap<String, crate::state::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
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn reload_config(config_lock: &Arc<RwLock<Config>>, path: Option<PathBuf>) {
|
pub async fn reload_config(config_lock: &Arc<RwLock<Config>>, path: Option<PathBuf>) {
|
||||||
info!("Reloading configuration...");
|
info!("Reloading configuration...");
|
||||||
let new_config = crate::config::load_config(path);
|
let new_config = crate::config::load_config(path);
|
||||||
@@ -385,79 +493,20 @@ pub async fn reload_config(config_lock: &Arc<RwLock<Config>>, path: Option<PathB
|
|||||||
info!("Configuration reloaded successfully.");
|
info!("Configuration reloaded successfully.");
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// 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(
|
pub async fn evaluate_module_for_signaler(
|
||||||
module_name: &str,
|
module_name: &str,
|
||||||
state: &AppReceivers,
|
state: &AppReceivers,
|
||||||
config: &Config,
|
config: &Config,
|
||||||
) -> Option<String> {
|
) -> Option<String> {
|
||||||
let result = match module_name {
|
let args = crate::registry::signaler_default_args(module_name);
|
||||||
"net" | "network" => {
|
crate::registry::dispatch(module_name, config, state, args)
|
||||||
crate::modules::network::NetworkModule
|
.await
|
||||||
.run(config, state, &[])
|
.ok()
|
||||||
.await
|
.and_then(|out| serde_json::to_string(&out).ok())
|
||||||
}
|
|
||||||
"cpu" => crate::modules::cpu::CpuModule.run(config, state, &[]).await,
|
|
||||||
"mem" | "memory" => {
|
|
||||||
crate::modules::memory::MemoryModule
|
|
||||||
.run(config, state, &[])
|
|
||||||
.await
|
|
||||||
}
|
|
||||||
"disk" => {
|
|
||||||
crate::modules::disk::DiskModule
|
|
||||||
.run(config, state, &["/"])
|
|
||||||
.await
|
|
||||||
}
|
|
||||||
"pool" | "btrfs" => {
|
|
||||||
crate::modules::btrfs::BtrfsModule
|
|
||||||
.run(config, state, &["btrfs"])
|
|
||||||
.await
|
|
||||||
}
|
|
||||||
"vol" | "audio" => {
|
|
||||||
crate::modules::audio::AudioModule
|
|
||||||
.run(config, state, &["sink", "show"])
|
|
||||||
.await
|
|
||||||
}
|
|
||||||
"mic" => {
|
|
||||||
crate::modules::audio::AudioModule
|
|
||||||
.run(config, state, &["source", "show"])
|
|
||||||
.await
|
|
||||||
}
|
|
||||||
"gpu" => crate::modules::gpu::GpuModule.run(config, state, &[]).await,
|
|
||||||
"sys" => crate::modules::sys::SysModule.run(config, state, &[]).await,
|
|
||||||
"bt" | "bluetooth" => {
|
|
||||||
crate::modules::bt::BtModule
|
|
||||||
.run(config, state, &["show"])
|
|
||||||
.await
|
|
||||||
}
|
|
||||||
"power" => {
|
|
||||||
crate::modules::power::PowerModule
|
|
||||||
.run(config, state, &[])
|
|
||||||
.await
|
|
||||||
}
|
|
||||||
"game" => {
|
|
||||||
crate::modules::game::GameModule
|
|
||||||
.run(config, state, &[])
|
|
||||||
.await
|
|
||||||
}
|
|
||||||
"backlight" => {
|
|
||||||
crate::modules::backlight::BacklightModule
|
|
||||||
.run(config, state, &[])
|
|
||||||
.await
|
|
||||||
}
|
|
||||||
"kbd" | "keyboard" => {
|
|
||||||
crate::modules::keyboard::KeyboardModule
|
|
||||||
.run(config, state, &[])
|
|
||||||
.await
|
|
||||||
}
|
|
||||||
"dnd" => crate::modules::dnd::DndModule.run(config, state, &[]).await,
|
|
||||||
"mpris" => {
|
|
||||||
crate::modules::mpris::MprisModule
|
|
||||||
.run(config, state, &[])
|
|
||||||
.await
|
|
||||||
}
|
|
||||||
_ => return None,
|
|
||||||
};
|
|
||||||
result.ok().and_then(|out| serde_json::to_string(&out).ok())
|
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn handle_request(
|
async fn handle_request(
|
||||||
@@ -466,160 +515,22 @@ async fn handle_request(
|
|||||||
state: &AppReceivers,
|
state: &AppReceivers,
|
||||||
config_lock: &Arc<RwLock<Config>>,
|
config_lock: &Arc<RwLock<Config>>,
|
||||||
) -> String {
|
) -> String {
|
||||||
// 1. Check Circuit Breaker status
|
let (is_in_backoff, cached_output) = crate::health::check_backoff(module_name, state).await;
|
||||||
let (is_in_backoff, cached_output) = {
|
|
||||||
let lock = state.health.read().await;
|
|
||||||
if let Some(health) = lock.get(module_name) {
|
|
||||||
let in_backoff = if let Some(until) = health.backoff_until {
|
|
||||||
Instant::now() < until
|
|
||||||
} else {
|
|
||||||
false
|
|
||||||
};
|
|
||||||
(in_backoff, health.last_successful_output.clone())
|
|
||||||
} else {
|
|
||||||
(false, None)
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
if is_in_backoff {
|
if is_in_backoff {
|
||||||
if let Some(mut cached) = cached_output {
|
return crate::health::backoff_response(module_name, cached_output);
|
||||||
// Add a "warning" class to indicate stale data
|
|
||||||
let class = cached.class.unwrap_or_default();
|
|
||||||
cached.class = Some(format!("{} warning", class).trim().to_string());
|
|
||||||
return serde_json::to_string(&cached).unwrap_or_else(|_| "{}".to_string());
|
|
||||||
}
|
|
||||||
return format!(
|
|
||||||
"{{\"text\":\"\u{200B}Cooling down ({})\u{200B}\",\"class\":\"error\"}}",
|
|
||||||
module_name
|
|
||||||
);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
let config = config_lock.read().await;
|
let config = config_lock.read().await;
|
||||||
|
let result = crate::registry::dispatch(module_name, &config, state, args).await;
|
||||||
|
|
||||||
let result = match module_name {
|
crate::health::update_health(module_name, &result, state).await;
|
||||||
"net" | "network" => {
|
|
||||||
crate::modules::network::NetworkModule
|
|
||||||
.run(&config, state, args)
|
|
||||||
.await
|
|
||||||
}
|
|
||||||
"cpu" => {
|
|
||||||
crate::modules::cpu::CpuModule
|
|
||||||
.run(&config, state, args)
|
|
||||||
.await
|
|
||||||
}
|
|
||||||
"mem" | "memory" => {
|
|
||||||
crate::modules::memory::MemoryModule
|
|
||||||
.run(&config, state, args)
|
|
||||||
.await
|
|
||||||
}
|
|
||||||
"disk" => {
|
|
||||||
crate::modules::disk::DiskModule
|
|
||||||
.run(&config, state, args)
|
|
||||||
.await
|
|
||||||
}
|
|
||||||
"pool" | "btrfs" => {
|
|
||||||
crate::modules::btrfs::BtrfsModule
|
|
||||||
.run(&config, state, args)
|
|
||||||
.await
|
|
||||||
}
|
|
||||||
"vol" | "audio" => {
|
|
||||||
crate::modules::audio::AudioModule
|
|
||||||
.run(&config, state, args)
|
|
||||||
.await
|
|
||||||
}
|
|
||||||
"mic" => {
|
|
||||||
crate::modules::audio::AudioModule
|
|
||||||
.run(&config, state, args)
|
|
||||||
.await
|
|
||||||
}
|
|
||||||
"gpu" => {
|
|
||||||
crate::modules::gpu::GpuModule
|
|
||||||
.run(&config, state, args)
|
|
||||||
.await
|
|
||||||
}
|
|
||||||
"sys" => {
|
|
||||||
crate::modules::sys::SysModule
|
|
||||||
.run(&config, state, args)
|
|
||||||
.await
|
|
||||||
}
|
|
||||||
"bt" | "bluetooth" => crate::modules::bt::BtModule.run(&config, state, args).await,
|
|
||||||
"power" => {
|
|
||||||
crate::modules::power::PowerModule
|
|
||||||
.run(&config, state, args)
|
|
||||||
.await
|
|
||||||
}
|
|
||||||
"game" => {
|
|
||||||
crate::modules::game::GameModule
|
|
||||||
.run(&config, state, args)
|
|
||||||
.await
|
|
||||||
}
|
|
||||||
"backlight" => {
|
|
||||||
crate::modules::backlight::BacklightModule
|
|
||||||
.run(&config, state, args)
|
|
||||||
.await
|
|
||||||
}
|
|
||||||
"kbd" | "keyboard" => {
|
|
||||||
crate::modules::keyboard::KeyboardModule
|
|
||||||
.run(&config, state, args)
|
|
||||||
.await
|
|
||||||
}
|
|
||||||
"dnd" => {
|
|
||||||
crate::modules::dnd::DndModule
|
|
||||||
.run(&config, state, args)
|
|
||||||
.await
|
|
||||||
}
|
|
||||||
"mpris" => {
|
|
||||||
crate::modules::mpris::MprisModule
|
|
||||||
.run(&config, state, args)
|
|
||||||
.await
|
|
||||||
}
|
|
||||||
_ => {
|
|
||||||
warn!("Received request for unknown module: '{}'", module_name);
|
|
||||||
Err(FluxoError::Ipc(format!("Unknown module: {}", module_name)))
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
// 2. Update Health and Cache based on result
|
|
||||||
{
|
|
||||||
let mut lock = state.health.write().await;
|
|
||||||
let health = lock.entry(module_name.to_string()).or_default();
|
|
||||||
match &result {
|
|
||||||
Ok(output) => {
|
|
||||||
health.consecutive_failures = 0;
|
|
||||||
health.backoff_until = None;
|
|
||||||
health.last_successful_output = Some(output.clone());
|
|
||||||
}
|
|
||||||
Err(e) => {
|
|
||||||
health.consecutive_failures += 1;
|
|
||||||
health.last_failure = Some(Instant::now());
|
|
||||||
if health.consecutive_failures >= 3 {
|
|
||||||
// Backoff for 30 seconds after 3 failures
|
|
||||||
health.backoff_until = Some(Instant::now() + Duration::from_secs(30));
|
|
||||||
warn!(module = module_name, error = %e, "Module entered backoff state due to repeated failures");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
match result {
|
match result {
|
||||||
Ok(output) => serde_json::to_string(&output).unwrap_or_else(|_| "{}".to_string()),
|
Ok(output) => serde_json::to_string(&output).unwrap_or_else(|_| "{}".to_string()),
|
||||||
Err(e) => {
|
Err(crate::error::FluxoError::Disabled(_)) => {
|
||||||
// If we have a cached output, return it as fallback even on immediate error
|
"{\"text\":\"\",\"tooltip\":\"Module disabled\",\"class\":\"disabled\"}".to_string()
|
||||||
if let Some(mut cached) = cached_output {
|
|
||||||
let class = cached.class.unwrap_or_default();
|
|
||||||
cached.class = Some(format!("{} warning", class).trim().to_string());
|
|
||||||
return serde_json::to_string(&cached).unwrap_or_else(|_| "{}".to_string());
|
|
||||||
}
|
|
||||||
|
|
||||||
let error_msg = e.to_string();
|
|
||||||
error!(module = module_name, error = %error_msg, "Module execution failed");
|
|
||||||
let err_out = crate::output::WaybarOutput {
|
|
||||||
text: "\u{200B}Error\u{200B}".to_string(),
|
|
||||||
tooltip: Some(error_msg),
|
|
||||||
class: Some("error".to_string()),
|
|
||||||
percentage: None,
|
|
||||||
};
|
|
||||||
serde_json::to_string(&err_out).unwrap_or_else(|_| "{}".to_string())
|
|
||||||
}
|
}
|
||||||
|
Err(e) => crate::health::error_response(module_name, &e, cached_output),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,5 +1,11 @@
|
|||||||
|
//! Error types used across the crate.
|
||||||
|
|
||||||
use thiserror::Error;
|
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)]
|
#[derive(Error, Debug)]
|
||||||
#[allow(dead_code)]
|
#[allow(dead_code)]
|
||||||
pub enum FluxoError {
|
pub enum FluxoError {
|
||||||
@@ -33,11 +39,17 @@ pub enum FluxoError {
|
|||||||
#[error("Serialization error: {0}")]
|
#[error("Serialization error: {0}")]
|
||||||
Serialization(#[from] serde_json::Error),
|
Serialization(#[from] serde_json::Error),
|
||||||
|
|
||||||
|
#[error("Module disabled: {0}")]
|
||||||
|
Disabled(String),
|
||||||
|
|
||||||
#[error("Other error: {0}")]
|
#[error("Other error: {0}")]
|
||||||
Other(#[from] anyhow::Error),
|
Other(#[from] anyhow::Error),
|
||||||
}
|
}
|
||||||
|
|
||||||
impl FluxoError {
|
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 {
|
pub fn is_transient(&self) -> bool {
|
||||||
matches!(
|
matches!(
|
||||||
self,
|
self,
|
||||||
@@ -51,4 +63,5 @@ impl FluxoError {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Crate-wide `Result` alias using [`FluxoError`].
|
||||||
pub type Result<T> = std::result::Result<T, FluxoError>;
|
pub type Result<T> = std::result::Result<T, FluxoError>;
|
||||||
|
|||||||
+151
@@ -0,0 +1,151 @@
|
|||||||
|
//! 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());
|
||||||
|
}
|
||||||
|
format!(
|
||||||
|
"{{\"text\":\"\u{200B}Cooling down ({})\u{200B}\",\"class\":\"error\"}}",
|
||||||
|
module_name
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// 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 {
|
||||||
|
text: "\u{200B}Error\u{200B}".to_string(),
|
||||||
|
tooltip: Some(error_msg),
|
||||||
|
class: Some("error".to_string()),
|
||||||
|
percentage: None,
|
||||||
|
};
|
||||||
|
serde_json::to_string(&err_out).unwrap_or_else(|_| "{}".to_string())
|
||||||
|
}
|
||||||
+11
-1
@@ -1,8 +1,16 @@
|
|||||||
|
//! 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::io::{Read, Write};
|
||||||
use std::os::unix::net::UnixStream;
|
use std::os::unix::net::UnixStream;
|
||||||
use std::time::Duration;
|
use std::time::Duration;
|
||||||
use tracing::debug;
|
use tracing::debug;
|
||||||
|
|
||||||
|
/// 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 {
|
pub fn socket_path() -> String {
|
||||||
if let Ok(dir) = std::env::var("XDG_RUNTIME_DIR") {
|
if let Ok(dir) = std::env::var("XDG_RUNTIME_DIR") {
|
||||||
format!("{}/fluxo.sock", dir)
|
format!("{}/fluxo.sock", dir)
|
||||||
@@ -11,13 +19,15 @@ pub fn socket_path() -> 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> {
|
pub fn request_data(module: &str, args: &[&str]) -> anyhow::Result<String> {
|
||||||
let sock = socket_path();
|
let sock = socket_path();
|
||||||
debug!(module, ?args, "Connecting to daemon socket: {}", sock);
|
debug!(module, ?args, "Connecting to daemon socket: {}", sock);
|
||||||
let mut stream = UnixStream::connect(&sock)?;
|
let mut stream = UnixStream::connect(&sock)?;
|
||||||
stream.set_read_timeout(Some(Duration::from_secs(5)))?;
|
stream.set_read_timeout(Some(Duration::from_secs(5)))?;
|
||||||
|
|
||||||
// Send module and args
|
|
||||||
let mut request = module.to_string();
|
let mut request = module.to_string();
|
||||||
for arg in args {
|
for arg in args {
|
||||||
request.push(' ');
|
request.push(' ');
|
||||||
|
|||||||
@@ -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 }
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
+87
-27
@@ -1,9 +1,27 @@
|
|||||||
|
//! `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;
|
||||||
mod config;
|
mod config;
|
||||||
mod daemon;
|
mod daemon;
|
||||||
mod error;
|
mod error;
|
||||||
|
mod health;
|
||||||
mod ipc;
|
mod ipc;
|
||||||
mod modules;
|
mod modules;
|
||||||
mod output;
|
mod output;
|
||||||
|
mod registry;
|
||||||
mod signaler;
|
mod signaler;
|
||||||
mod state;
|
mod state;
|
||||||
mod utils;
|
mod utils;
|
||||||
@@ -75,48 +93,83 @@ fn main() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
if let Some(module) = &cli.module {
|
if let Some(module) = &cli.module {
|
||||||
// Special case for client-side Bluetooth menu which requires UI
|
// Bluetooth menu is handled client-side: it needs access to the user's
|
||||||
|
// menu command (rofi/dmenu/wofi) which the daemon has no business spawning.
|
||||||
|
#[cfg(feature = "mod-bt")]
|
||||||
if module == "bt" && cli.args.first().map(|s| s.as_str()) == Some("menu") {
|
if module == "bt" && cli.args.first().map(|s| s.as_str()) == Some("menu") {
|
||||||
let config = config::load_config(None);
|
let config = config::load_config(None);
|
||||||
let mut items = Vec::new();
|
let mut items = Vec::new();
|
||||||
|
|
||||||
if let Ok(json_str) = ipc::request_data("bt", &["get_modes"])
|
// Ask the daemon for the device list; tuples are (alias, mac).
|
||||||
&& let Ok(val) = serde_json::from_str::<serde_json::Value>(&json_str)
|
let mut connected: Vec<(String, String)> = Vec::new();
|
||||||
&& let Some(modes_str) = val.get("text").and_then(|t| t.as_str())
|
let mut paired: Vec<(String, String)> = Vec::new();
|
||||||
&& !modes_str.is_empty()
|
|
||||||
{
|
|
||||||
for mode in modes_str.lines() {
|
|
||||||
items.push(format!("Mode: {}", mode));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if !items.is_empty() {
|
|
||||||
items.push("Disconnect".to_string());
|
|
||||||
}
|
|
||||||
|
|
||||||
items.push("--- Connect Device ---".to_string());
|
|
||||||
|
|
||||||
if let Ok(json_str) = ipc::request_data("bt", &["menu_data"])
|
if let Ok(json_str) = ipc::request_data("bt", &["menu_data"])
|
||||||
&& let Ok(val) = serde_json::from_str::<serde_json::Value>(&json_str)
|
&& let Ok(val) = serde_json::from_str::<serde_json::Value>(&json_str)
|
||||||
&& let Some(devices_str) = val.get("text").and_then(|t| t.as_str())
|
&& let Some(text) = val.get("text").and_then(|t| t.as_str())
|
||||||
{
|
{
|
||||||
for line in devices_str.lines() {
|
for line in text.lines() {
|
||||||
if !line.is_empty() {
|
if let Some(rest) = line.strip_prefix("CONNECTED:")
|
||||||
items.push(line.to_string());
|
&& 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()));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
for (alias, mac) in &connected {
|
||||||
|
if let Ok(json_str) = 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!("{}: Mode: {} [{}]", alias, mode, mac));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
items.push(format!("Disconnect {} [{}]", alias, mac));
|
||||||
|
}
|
||||||
|
|
||||||
|
if !paired.is_empty() {
|
||||||
|
items.push("--- Connect Device ---".to_string());
|
||||||
|
for (alias, mac) in &paired {
|
||||||
|
items.push(format!("{} ({})", alias, mac));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
if !items.is_empty() {
|
if !items.is_empty() {
|
||||||
if let Ok(selected) =
|
if let Ok(selected) =
|
||||||
utils::show_menu("BT Menu: ", &items, &config.general.menu_command)
|
utils::show_menu("BT Menu: ", &items, &config.general.menu_command)
|
||||||
{
|
{
|
||||||
if let Some(mode) = selected.strip_prefix("Mode: ") {
|
if selected.contains(": Mode: ") {
|
||||||
handle_ipc_response(ipc::request_data("bt", &["set_mode", mode]));
|
// Parse "<alias>: Mode: <mode> [<MAC>]".
|
||||||
} else if selected == "Disconnect" {
|
if let Some(bracket_start) = selected.rfind('[')
|
||||||
handle_ipc_response(ipc::request_data("bt", &["disconnect"]));
|
&& let Some(bracket_end) = selected.rfind(']')
|
||||||
|
{
|
||||||
|
let mac = &selected[bracket_start + 1..bracket_end];
|
||||||
|
if let Some(mode_start) = selected.find(": Mode: ") {
|
||||||
|
let mode =
|
||||||
|
&selected[mode_start + ": Mode: ".len()..bracket_start - 1];
|
||||||
|
handle_ipc_response(ipc::request_data(
|
||||||
|
"bt",
|
||||||
|
&["set_mode", mode, mac],
|
||||||
|
));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else if selected.starts_with("Disconnect ") {
|
||||||
|
// Parse "Disconnect <alias> [<MAC>]".
|
||||||
|
if let Some(bracket_start) = selected.rfind('[')
|
||||||
|
&& let Some(bracket_end) = selected.rfind(']')
|
||||||
|
{
|
||||||
|
let mac = &selected[bracket_start + 1..bracket_end];
|
||||||
|
handle_ipc_response(ipc::request_data("bt", &["disconnect", mac]));
|
||||||
|
}
|
||||||
} else if selected == "--- Connect Device ---" {
|
} else if selected == "--- Connect Device ---" {
|
||||||
// Do nothing
|
// section header
|
||||||
} else if let Some(mac_start) = selected.rfind('(')
|
} else if let Some(mac_start) = selected.rfind('(')
|
||||||
&& let Some(mac_end) = selected.rfind(')')
|
&& let Some(mac_end) = selected.rfind(')')
|
||||||
{
|
{
|
||||||
@@ -130,8 +183,8 @@ fn main() {
|
|||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Generic module dispatch
|
// `vol` and `mic` both dispatch to the audio module; we just prepend
|
||||||
// Translate module-specific shorthand targets
|
// the "sink" / "source" argument so the server picks the right device.
|
||||||
let (actual_module, actual_args) = if module == "vol" {
|
let (actual_module, actual_args) = if module == "vol" {
|
||||||
let mut new_args = vec!["sink".to_string()];
|
let mut new_args = vec!["sink".to_string()];
|
||||||
new_args.extend(cli.args.clone());
|
new_args.extend(cli.args.clone());
|
||||||
@@ -152,6 +205,13 @@ fn main() {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Post-process the daemon's response for direct output to Waybar.
|
||||||
|
///
|
||||||
|
/// Normal spaces are replaced with figure-spaces (U+2007) so Waybar's
|
||||||
|
/// proportional font does not jitter between updates, and the text is wrapped
|
||||||
|
/// in zero-width spaces (U+200B) as a cosmetic padding trick. Markup strings
|
||||||
|
/// (containing `<`) pass through untouched. On IPC failure an `error` output
|
||||||
|
/// is emitted and the client exits non-zero.
|
||||||
fn handle_ipc_response(response: anyhow::Result<String>) {
|
fn handle_ipc_response(response: anyhow::Result<String>) {
|
||||||
match response {
|
match response {
|
||||||
Ok(json_str) => match serde_json::from_str::<serde_json::Value>(&json_str) {
|
Ok(json_str) => match serde_json::from_str::<serde_json::Value>(&json_str) {
|
||||||
|
|||||||
+120
-87
@@ -2,13 +2,14 @@ use crate::config::Config;
|
|||||||
use crate::error::{FluxoError, Result};
|
use crate::error::{FluxoError, Result};
|
||||||
use crate::modules::WaybarModule;
|
use crate::modules::WaybarModule;
|
||||||
use crate::output::WaybarOutput;
|
use crate::output::WaybarOutput;
|
||||||
use crate::state::{AppReceivers, AudioState};
|
use crate::state::{AppReceivers, AudioDeviceInfo, AudioState};
|
||||||
use crate::utils::{TokenValue, format_template};
|
use crate::utils::{TokenValue, format_template};
|
||||||
use libpulse_binding::callbacks::ListResult;
|
use libpulse_binding::callbacks::ListResult;
|
||||||
use libpulse_binding::context::subscribe::{Facility, InterestMaskSet};
|
use libpulse_binding::context::subscribe::{Facility, InterestMaskSet};
|
||||||
use libpulse_binding::context::{Context, FlagSet as ContextFlag};
|
use libpulse_binding::context::{Context, FlagSet as ContextFlag};
|
||||||
use libpulse_binding::mainloop::threaded::Mainloop as ThreadedMainloop;
|
use libpulse_binding::mainloop::threaded::Mainloop as ThreadedMainloop;
|
||||||
use libpulse_binding::volume::Volume;
|
use libpulse_binding::volume::Volume;
|
||||||
|
use std::sync::Arc;
|
||||||
use tokio::sync::{mpsc, watch};
|
use tokio::sync::{mpsc, watch};
|
||||||
use tracing::error;
|
use tracing::error;
|
||||||
|
|
||||||
@@ -229,92 +230,124 @@ fn fetch_audio_data_sync(
|
|||||||
let _ = tx_server.send(current);
|
let _ = tx_server.send(current);
|
||||||
});
|
});
|
||||||
|
|
||||||
let tx_sink = state_tx.clone();
|
fetch_sinks(context, state_tx);
|
||||||
context.introspect().get_sink_info_list(move |res| {
|
fetch_sources(context, state_tx);
|
||||||
let mut current = tx_sink.borrow().clone();
|
|
||||||
match res {
|
|
||||||
ListResult::Item(item) => {
|
|
||||||
if let Some(name) = item.name.as_ref() {
|
|
||||||
let name_str = name.to_string();
|
|
||||||
if !current.available_sinks.contains(&name_str) {
|
|
||||||
current.available_sinks.push(name_str);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
let is_default = item
|
|
||||||
.name
|
|
||||||
.as_ref()
|
|
||||||
.map(|s| s.as_ref() == current.sink.name)
|
|
||||||
.unwrap_or(false);
|
|
||||||
if is_default {
|
|
||||||
current.sink.description = item
|
|
||||||
.description
|
|
||||||
.as_ref()
|
|
||||||
.map(|s| s.to_string())
|
|
||||||
.unwrap_or_default();
|
|
||||||
current.sink.volume = ((item.volume.avg().0 as f64 / Volume::NORMAL.0 as f64)
|
|
||||||
* 100.0)
|
|
||||||
.round() as u8;
|
|
||||||
current.sink.muted = item.mute;
|
|
||||||
current.sink.channels = item.volume.len();
|
|
||||||
}
|
|
||||||
let _ = tx_sink.send(current);
|
|
||||||
}
|
|
||||||
ListResult::End => {
|
|
||||||
// Clear the list on End so it rebuilds fresh next time
|
|
||||||
current.available_sinks.clear();
|
|
||||||
let _ = tx_sink.send(current);
|
|
||||||
}
|
|
||||||
ListResult::Error => {}
|
|
||||||
}
|
|
||||||
});
|
|
||||||
|
|
||||||
let tx_source = state_tx.clone();
|
|
||||||
context.introspect().get_source_info_list(move |res| {
|
|
||||||
let mut current = tx_source.borrow().clone();
|
|
||||||
match res {
|
|
||||||
ListResult::Item(item) => {
|
|
||||||
if let Some(name) = item.name.as_ref() {
|
|
||||||
let name_str = name.to_string();
|
|
||||||
// PulseAudio includes monitor sources, ignore them if we want to
|
|
||||||
if !name_str.contains(".monitor")
|
|
||||||
&& !current.available_sources.contains(&name_str)
|
|
||||||
{
|
|
||||||
current.available_sources.push(name_str);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
let is_default = item
|
|
||||||
.name
|
|
||||||
.as_ref()
|
|
||||||
.map(|s| s.as_ref() == current.source.name)
|
|
||||||
.unwrap_or(false);
|
|
||||||
if is_default {
|
|
||||||
current.source.description = item
|
|
||||||
.description
|
|
||||||
.as_ref()
|
|
||||||
.map(|s| s.to_string())
|
|
||||||
.unwrap_or_default();
|
|
||||||
current.source.volume = ((item.volume.avg().0 as f64 / Volume::NORMAL.0 as f64)
|
|
||||||
* 100.0)
|
|
||||||
.round() as u8;
|
|
||||||
current.source.muted = item.mute;
|
|
||||||
current.source.channels = item.volume.len();
|
|
||||||
}
|
|
||||||
let _ = tx_source.send(current);
|
|
||||||
}
|
|
||||||
ListResult::End => {
|
|
||||||
// Clear the list on End so it rebuilds fresh next time
|
|
||||||
current.available_sources.clear();
|
|
||||||
let _ = tx_source.send(current);
|
|
||||||
}
|
|
||||||
ListResult::Error => {}
|
|
||||||
}
|
|
||||||
});
|
|
||||||
|
|
||||||
Ok(())
|
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)
|
||||||
|
}
|
||||||
|
|
||||||
|
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;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
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 => {}
|
||||||
|
}
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
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 => {}
|
||||||
|
}
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
pub struct AudioModule;
|
pub struct AudioModule;
|
||||||
|
|
||||||
impl WaybarModule for AudioModule {
|
impl WaybarModule for AudioModule {
|
||||||
@@ -331,19 +364,19 @@ impl WaybarModule for AudioModule {
|
|||||||
match *action {
|
match *action {
|
||||||
"up" => {
|
"up" => {
|
||||||
self.change_volume(state, target_type, step, true).await?;
|
self.change_volume(state, target_type, step, true).await?;
|
||||||
Ok(WaybarOutput::default())
|
self.get_status(config, state, target_type).await
|
||||||
}
|
}
|
||||||
"down" => {
|
"down" => {
|
||||||
self.change_volume(state, target_type, step, false).await?;
|
self.change_volume(state, target_type, step, false).await?;
|
||||||
Ok(WaybarOutput::default())
|
self.get_status(config, state, target_type).await
|
||||||
}
|
}
|
||||||
"mute" => {
|
"mute" => {
|
||||||
self.toggle_mute(state, target_type).await?;
|
self.toggle_mute(state, target_type).await?;
|
||||||
Ok(WaybarOutput::default())
|
self.get_status(config, state, target_type).await
|
||||||
}
|
}
|
||||||
"cycle" => {
|
"cycle" => {
|
||||||
self.cycle_device(state, target_type).await?;
|
self.cycle_device(state, target_type).await?;
|
||||||
Ok(WaybarOutput::default())
|
self.get_status(config, state, target_type).await
|
||||||
}
|
}
|
||||||
"show" => self.get_status(config, state, target_type).await,
|
"show" => self.get_status(config, state, target_type).await,
|
||||||
other => Err(FluxoError::Module {
|
other => Err(FluxoError::Module {
|
||||||
|
|||||||
+178
-110
@@ -5,7 +5,7 @@ use crate::config::Config;
|
|||||||
use crate::error::Result as FluxoResult;
|
use crate::error::Result as FluxoResult;
|
||||||
use crate::modules::WaybarModule;
|
use crate::modules::WaybarModule;
|
||||||
use crate::output::WaybarOutput;
|
use crate::output::WaybarOutput;
|
||||||
use crate::state::{AppReceivers, BtState};
|
use crate::state::{AppReceivers, BtDeviceInfo, BtState};
|
||||||
use crate::utils::{TokenValue, format_template};
|
use crate::utils::{TokenValue, format_template};
|
||||||
use anyhow::Result;
|
use anyhow::Result;
|
||||||
use std::sync::LazyLock;
|
use std::sync::LazyLock;
|
||||||
@@ -47,60 +47,64 @@ impl BtDaemon {
|
|||||||
let adapter = session.default_adapter().await?;
|
let adapter = session.default_adapter().await?;
|
||||||
let adapter_powered = adapter.is_powered().await.unwrap_or(false);
|
let adapter_powered = adapter.is_powered().await.unwrap_or(false);
|
||||||
|
|
||||||
let mut bt_state = BtState {
|
let mut connected_devices = Vec::new();
|
||||||
adapter_powered,
|
|
||||||
..Default::default()
|
|
||||||
};
|
|
||||||
|
|
||||||
if adapter_powered {
|
if adapter_powered {
|
||||||
let devices = adapter.device_addresses().await?;
|
let mut addresses = adapter.device_addresses().await?;
|
||||||
for addr in devices {
|
addresses.sort();
|
||||||
let device = adapter.device(addr)?;
|
let audio_sink_uuid = bluer::Uuid::from_u128(0x0000110b_0000_1000_8000_00805f9b34fb);
|
||||||
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() {
|
for addr in addresses {
|
||||||
if p.can_handle(&bt_state.device_alias, &bt_state.device_address) {
|
let device = adapter.device(addr)?;
|
||||||
match p.get_data(config, state, &bt_state.device_address).await {
|
if !device.is_connected().await.unwrap_or(false) {
|
||||||
Ok(data) => {
|
continue;
|
||||||
bt_state.plugin_data = data
|
}
|
||||||
.into_iter()
|
let uuids = device.uuids().await?.unwrap_or_default();
|
||||||
.map(|(k, v)| {
|
if !uuids.contains(&audio_sink_uuid) {
|
||||||
let val_str = match v {
|
continue;
|
||||||
TokenValue::String(s) => s,
|
}
|
||||||
TokenValue::Int(i) => i.to_string(),
|
|
||||||
TokenValue::Float(f) => format!("{:.1}", f),
|
let mut dev_info = BtDeviceInfo {
|
||||||
};
|
device_address: addr.to_string(),
|
||||||
(k, val_str)
|
device_alias: device.alias().await.unwrap_or_else(|_| addr.to_string()),
|
||||||
})
|
battery_percentage: device.battery_percentage().await.unwrap_or(None),
|
||||||
.collect();
|
plugin_data: vec![],
|
||||||
}
|
};
|
||||||
Err(e) => {
|
|
||||||
warn!("Plugin {} failed for {}: {}", p.name(), addr, e);
|
for p in PLUGINS.iter() {
|
||||||
bt_state
|
if p.can_handle(&dev_info.device_alias, &dev_info.device_address) {
|
||||||
.plugin_data
|
match p.get_data(config, state, &dev_info.device_address).await {
|
||||||
.push(("plugin_error".to_string(), e.to_string()));
|
Ok(data) => {
|
||||||
}
|
dev_info.plugin_data = data
|
||||||
}
|
.into_iter()
|
||||||
break;
|
.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;
|
break;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
connected_devices.push(dev_info);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
let _ = tx.send(bt_state);
|
let _ = tx.send(BtState {
|
||||||
|
adapter_powered,
|
||||||
|
devices: connected_devices,
|
||||||
|
});
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
@@ -111,8 +115,6 @@ static PLUGINS: LazyLock<Vec<Box<dyn BtPlugin>>> =
|
|||||||
|
|
||||||
fn trigger_robust_poll(state: AppReceivers) {
|
fn trigger_robust_poll(state: AppReceivers) {
|
||||||
tokio::spawn(async move {
|
tokio::spawn(async move {
|
||||||
// Poll immediately and then a few times over the next few seconds
|
|
||||||
// to catch slow state changes in bluez or plugins.
|
|
||||||
for delay in [200, 500, 1000, 2000, 3000] {
|
for delay in [200, 500, 1000, 2000, 3000] {
|
||||||
tokio::time::sleep(std::time::Duration::from_millis(delay)).await;
|
tokio::time::sleep(std::time::Duration::from_millis(delay)).await;
|
||||||
let _ = state.bt_force_poll.try_send(());
|
let _ = state.bt_force_poll.try_send(());
|
||||||
@@ -120,6 +122,26 @@ fn trigger_robust_poll(state: AppReceivers) {
|
|||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// 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)
|
||||||
|
}
|
||||||
|
|
||||||
pub struct BtModule;
|
pub struct BtModule;
|
||||||
|
|
||||||
impl WaybarModule for BtModule {
|
impl WaybarModule for BtModule {
|
||||||
@@ -129,12 +151,10 @@ impl WaybarModule for BtModule {
|
|||||||
state: &AppReceivers,
|
state: &AppReceivers,
|
||||||
args: &[&str],
|
args: &[&str],
|
||||||
) -> FluxoResult<WaybarOutput> {
|
) -> FluxoResult<WaybarOutput> {
|
||||||
let action = args.first().cloned().unwrap_or("show").to_string();
|
let action = args.first().cloned().unwrap_or("show");
|
||||||
let args = args.iter().map(|s| s.to_string()).collect::<Vec<_>>();
|
|
||||||
|
|
||||||
let bt_state = state.bluetooth.borrow().clone();
|
let bt_state = state.bluetooth.borrow().clone();
|
||||||
|
|
||||||
match action.as_str() {
|
match action {
|
||||||
"connect" => {
|
"connect" => {
|
||||||
if let Some(mac) = args.get(1) {
|
if let Some(mac) = args.get(1) {
|
||||||
if let Ok(session) = bluer::Session::new().await
|
if let Ok(session) = bluer::Session::new().await
|
||||||
@@ -148,77 +168,125 @@ impl WaybarModule for BtModule {
|
|||||||
}
|
}
|
||||||
return Ok(WaybarOutput::default());
|
return Ok(WaybarOutput::default());
|
||||||
}
|
}
|
||||||
"disconnect" if bt_state.connected => {
|
"disconnect" => {
|
||||||
if let Ok(session) = bluer::Session::new().await
|
let target_mac = resolve_target_mac(&bt_state, state, args.get(1).copied()).await;
|
||||||
&& let Ok(adapter) = session.default_adapter().await
|
if let Some(mac) = target_mac {
|
||||||
&& let Ok(addr) = bt_state.device_address.parse::<bluer::Address>()
|
if let Ok(session) = bluer::Session::new().await
|
||||||
&& let Ok(device) = adapter.device(addr)
|
&& let Ok(adapter) = session.default_adapter().await
|
||||||
{
|
&& let Ok(addr) = mac.parse::<bluer::Address>()
|
||||||
let _ = device.disconnect().await;
|
&& let Ok(device) = adapter.device(addr)
|
||||||
}
|
{
|
||||||
trigger_robust_poll(state.clone());
|
let _ = device.disconnect().await;
|
||||||
return Ok(WaybarOutput::default());
|
|
||||||
}
|
|
||||||
"menu_data" => {
|
|
||||||
let mut devs = Vec::new();
|
|
||||||
if let Ok(session) = bluer::Session::new().await
|
|
||||||
&& let Ok(adapter) = session.default_adapter().await
|
|
||||||
&& let Ok(addresses) = adapter.device_addresses().await
|
|
||||||
{
|
|
||||||
for addr in addresses {
|
|
||||||
if let Ok(device) = adapter.device(addr)
|
|
||||||
&& device.is_paired().await.unwrap_or(false)
|
|
||||||
{
|
|
||||||
let alias = device.alias().await.unwrap_or_else(|_| addr.to_string());
|
|
||||||
devs.push(format!("{} ({})", alias, addr));
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
|
||||||
return Ok(WaybarOutput {
|
|
||||||
text: devs.join("\n"),
|
|
||||||
..Default::default()
|
|
||||||
});
|
|
||||||
}
|
|
||||||
"cycle_mode" if bt_state.connected => {
|
|
||||||
let plugin = PLUGINS
|
|
||||||
.iter()
|
|
||||||
.find(|p| p.can_handle(&bt_state.device_alias, &bt_state.device_address));
|
|
||||||
if let Some(p) = plugin {
|
|
||||||
p.cycle_mode(&bt_state.device_address, state).await?;
|
|
||||||
trigger_robust_poll(state.clone());
|
trigger_robust_poll(state.clone());
|
||||||
}
|
}
|
||||||
return Ok(WaybarOutput::default());
|
return Ok(WaybarOutput::default());
|
||||||
}
|
}
|
||||||
"get_modes" if bt_state.connected => {
|
"cycle" => {
|
||||||
let plugin = PLUGINS
|
let count = bt_state.devices.len();
|
||||||
.iter()
|
if count > 1 {
|
||||||
.find(|p| p.can_handle(&bt_state.device_alias, &bt_state.device_address));
|
let mut idx = state.bt_cycle.write().await;
|
||||||
let modes = if let Some(p) = plugin {
|
*idx = (*idx + 1) % count;
|
||||||
p.get_modes(&bt_state.device_address, state).await?
|
}
|
||||||
} else {
|
let _ = state.bt_force_poll.try_send(());
|
||||||
vec![]
|
return Ok(WaybarOutput::default());
|
||||||
};
|
}
|
||||||
|
"menu_data" => {
|
||||||
|
let mut lines = Vec::new();
|
||||||
|
|
||||||
|
// Connected devices
|
||||||
|
for dev in &bt_state.devices {
|
||||||
|
lines.push(format!(
|
||||||
|
"CONNECTED:{}|{}",
|
||||||
|
dev.device_alias, dev.device_address
|
||||||
|
));
|
||||||
|
}
|
||||||
|
|
||||||
|
// Paired-but-not-connected devices
|
||||||
|
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 {
|
return Ok(WaybarOutput {
|
||||||
text: modes.join("\n"),
|
text: lines.join("\n"),
|
||||||
..Default::default()
|
..Default::default()
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
"set_mode" if bt_state.connected => {
|
"get_modes" => {
|
||||||
if let Some(mode) = args.get(1) {
|
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
|
let plugin = PLUGINS
|
||||||
.iter()
|
.iter()
|
||||||
.find(|p| p.can_handle(&bt_state.device_alias, &bt_state.device_address));
|
.find(|p| p.can_handle(&dev.device_alias, &dev.device_address));
|
||||||
if let Some(p) = plugin {
|
if let Some(p) = plugin {
|
||||||
p.set_mode(mode, &bt_state.device_address, state).await?;
|
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());
|
trigger_robust_poll(state.clone());
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return Ok(WaybarOutput::default());
|
return Ok(WaybarOutput::default());
|
||||||
}
|
}
|
||||||
"show" => {}
|
|
||||||
_ => {}
|
_ => {}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// "show" and fallthrough
|
||||||
if !bt_state.adapter_powered {
|
if !bt_state.adapter_powered {
|
||||||
return Ok(WaybarOutput {
|
return Ok(WaybarOutput {
|
||||||
text: config.bt.format_disabled.clone(),
|
text: config.bt.format_disabled.clone(),
|
||||||
@@ -228,22 +296,23 @@ impl WaybarModule for BtModule {
|
|||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
if bt_state.connected {
|
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![
|
let mut tokens: Vec<(String, TokenValue)> = vec![
|
||||||
(
|
(
|
||||||
"alias".to_string(),
|
"alias".to_string(),
|
||||||
TokenValue::String(bt_state.device_alias.clone()),
|
TokenValue::String(dev.device_alias.clone()),
|
||||||
),
|
),
|
||||||
(
|
(
|
||||||
"mac".to_string(),
|
"mac".to_string(),
|
||||||
TokenValue::String(bt_state.device_address.clone()),
|
TokenValue::String(dev.device_address.clone()),
|
||||||
),
|
),
|
||||||
];
|
];
|
||||||
|
|
||||||
let mut class = vec!["connected".to_string()];
|
let mut class = vec!["connected".to_string()];
|
||||||
let mut has_plugin = false;
|
let mut has_plugin = false;
|
||||||
|
|
||||||
for (k, v) in &bt_state.plugin_data {
|
for (k, v) in &dev.plugin_data {
|
||||||
if k == "plugin_class" {
|
if k == "plugin_class" {
|
||||||
class.push(v.clone());
|
class.push(v.clone());
|
||||||
has_plugin = true;
|
has_plugin = true;
|
||||||
@@ -263,10 +332,9 @@ impl WaybarModule for BtModule {
|
|||||||
let text = format_template(format, &tokens);
|
let text = format_template(format, &tokens);
|
||||||
let tooltip = format!(
|
let tooltip = format!(
|
||||||
"{} | MAC: {}\nBattery: {}",
|
"{} | MAC: {}\nBattery: {}",
|
||||||
bt_state.device_alias,
|
dev.device_alias,
|
||||||
bt_state.device_address,
|
dev.device_address,
|
||||||
bt_state
|
dev.battery_percentage
|
||||||
.battery_percentage
|
|
||||||
.map(|b| format!("{}%", b))
|
.map(|b| format!("{}%", b))
|
||||||
.unwrap_or_else(|| "N/A".to_string())
|
.unwrap_or_else(|| "N/A".to_string())
|
||||||
);
|
);
|
||||||
@@ -275,7 +343,7 @@ impl WaybarModule for BtModule {
|
|||||||
text,
|
text,
|
||||||
tooltip: Some(tooltip),
|
tooltip: Some(tooltip),
|
||||||
class: Some(class.join(" ")),
|
class: Some(class.join(" ")),
|
||||||
percentage: bt_state.battery_percentage,
|
percentage: dev.battery_percentage,
|
||||||
})
|
})
|
||||||
} else {
|
} else {
|
||||||
Ok(WaybarOutput {
|
Ok(WaybarOutput {
|
||||||
|
|||||||
@@ -3,7 +3,7 @@ use crate::error::Result;
|
|||||||
use crate::modules::WaybarModule;
|
use crate::modules::WaybarModule;
|
||||||
use crate::output::WaybarOutput;
|
use crate::output::WaybarOutput;
|
||||||
use crate::state::AppReceivers;
|
use crate::state::AppReceivers;
|
||||||
use crate::utils::{TokenValue, format_template};
|
use crate::utils::{TokenValue, classify_usage, format_template};
|
||||||
|
|
||||||
pub struct BtrfsModule;
|
pub struct BtrfsModule;
|
||||||
|
|
||||||
@@ -44,13 +44,7 @@ impl WaybarModule for BtrfsModule {
|
|||||||
let size_gb = total_size / 1024.0 / 1024.0 / 1024.0;
|
let size_gb = total_size / 1024.0 / 1024.0 / 1024.0;
|
||||||
let percentage = (total_used / total_size) * 100.0;
|
let percentage = (total_used / total_size) * 100.0;
|
||||||
|
|
||||||
let class = if percentage > 95.0 {
|
let class = classify_usage(percentage, 80.0, 95.0);
|
||||||
"max"
|
|
||||||
} else if percentage > 80.0 {
|
|
||||||
"high"
|
|
||||||
} else {
|
|
||||||
"normal"
|
|
||||||
};
|
|
||||||
|
|
||||||
let text = format_template(
|
let text = format_template(
|
||||||
&config.pool.format,
|
&config.pool.format,
|
||||||
|
|||||||
+2
-8
@@ -3,7 +3,7 @@ use crate::error::Result;
|
|||||||
use crate::modules::WaybarModule;
|
use crate::modules::WaybarModule;
|
||||||
use crate::output::WaybarOutput;
|
use crate::output::WaybarOutput;
|
||||||
use crate::state::AppReceivers;
|
use crate::state::AppReceivers;
|
||||||
use crate::utils::{TokenValue, format_template};
|
use crate::utils::{TokenValue, classify_usage, format_template};
|
||||||
|
|
||||||
pub struct CpuModule;
|
pub struct CpuModule;
|
||||||
|
|
||||||
@@ -27,13 +27,7 @@ impl WaybarModule for CpuModule {
|
|||||||
],
|
],
|
||||||
);
|
);
|
||||||
|
|
||||||
let class = if usage > 95.0 {
|
let class = classify_usage(usage, 75.0, 95.0);
|
||||||
"max"
|
|
||||||
} else if usage > 75.0 {
|
|
||||||
"high"
|
|
||||||
} else {
|
|
||||||
"normal"
|
|
||||||
};
|
|
||||||
|
|
||||||
Ok(WaybarOutput {
|
Ok(WaybarOutput {
|
||||||
text,
|
text,
|
||||||
|
|||||||
+3
-9
@@ -3,7 +3,7 @@ use crate::error::{FluxoError, Result};
|
|||||||
use crate::modules::WaybarModule;
|
use crate::modules::WaybarModule;
|
||||||
use crate::output::WaybarOutput;
|
use crate::output::WaybarOutput;
|
||||||
use crate::state::AppReceivers;
|
use crate::state::AppReceivers;
|
||||||
use crate::utils::{TokenValue, format_template};
|
use crate::utils::{TokenValue, classify_usage, format_template};
|
||||||
|
|
||||||
pub struct DiskModule;
|
pub struct DiskModule;
|
||||||
|
|
||||||
@@ -41,13 +41,7 @@ impl WaybarModule for DiskModule {
|
|||||||
0.0
|
0.0
|
||||||
};
|
};
|
||||||
|
|
||||||
let class = if percentage > 95.0 {
|
let class = classify_usage(percentage, 80.0, 95.0);
|
||||||
"max"
|
|
||||||
} else if percentage > 80.0 {
|
|
||||||
"high"
|
|
||||||
} else {
|
|
||||||
"normal"
|
|
||||||
};
|
|
||||||
|
|
||||||
let text = format_template(
|
let text = format_template(
|
||||||
&config.disk.format,
|
&config.disk.format,
|
||||||
@@ -121,7 +115,7 @@ mod tests {
|
|||||||
|
|
||||||
#[tokio::test]
|
#[tokio::test]
|
||||||
async fn test_disk_not_found() {
|
async fn test_disk_not_found() {
|
||||||
let state = mock_state(AppState::default());
|
let state = state_with_disk("/", 100, 50);
|
||||||
let config = Config::default();
|
let config = Config::default();
|
||||||
let result = DiskModule
|
let result = DiskModule
|
||||||
.run(&config, &state.receivers, &["/nonexistent"])
|
.run(&config, &state.receivers, &["/nonexistent"])
|
||||||
|
|||||||
+108
-12
@@ -5,18 +5,82 @@ use crate::output::WaybarOutput;
|
|||||||
use crate::state::{AppReceivers, DndState};
|
use crate::state::{AppReceivers, DndState};
|
||||||
use futures::StreamExt;
|
use futures::StreamExt;
|
||||||
use tokio::sync::watch;
|
use tokio::sync::watch;
|
||||||
|
use tokio::time::Duration;
|
||||||
use tracing::{debug, error, info};
|
use tracing::{debug, error, info};
|
||||||
use zbus::{Connection, fdo::PropertiesProxy, proxy};
|
use zbus::proxy;
|
||||||
|
use zbus::zvariant::OwnedValue;
|
||||||
|
use zbus::{Connection, fdo::PropertiesProxy};
|
||||||
|
|
||||||
pub struct DndModule;
|
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 {
|
impl WaybarModule for DndModule {
|
||||||
async fn run(
|
async fn run(
|
||||||
&self,
|
&self,
|
||||||
config: &Config,
|
config: &Config,
|
||||||
state: &AppReceivers,
|
state: &AppReceivers,
|
||||||
_args: &[&str],
|
args: &[&str],
|
||||||
) -> Result<WaybarOutput> {
|
) -> 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),
|
||||||
|
})?;
|
||||||
|
|
||||||
|
// Try SwayNC
|
||||||
|
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());
|
||||||
|
}
|
||||||
|
|
||||||
|
// Try Dunst via raw D-Bus
|
||||||
|
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;
|
let is_dnd = state.dnd.borrow().is_dnd;
|
||||||
|
|
||||||
if is_dnd {
|
if is_dnd {
|
||||||
@@ -47,6 +111,8 @@ pub struct DndDaemon;
|
|||||||
trait SwayncControl {
|
trait SwayncControl {
|
||||||
#[zbus(property)]
|
#[zbus(property)]
|
||||||
fn dnd(&self) -> zbus::Result<bool>;
|
fn dnd(&self) -> zbus::Result<bool>;
|
||||||
|
#[zbus(property)]
|
||||||
|
fn set_dnd(&self, value: bool) -> zbus::Result<()>;
|
||||||
}
|
}
|
||||||
|
|
||||||
impl DndDaemon {
|
impl DndDaemon {
|
||||||
@@ -70,16 +136,13 @@ impl DndDaemon {
|
|||||||
|
|
||||||
info!("Connected to D-Bus for DND monitoring");
|
info!("Connected to D-Bus for DND monitoring");
|
||||||
|
|
||||||
// Try SwayNC first
|
// Try SwayNC first (signal-based)
|
||||||
if let Ok(proxy) = SwayncControlProxy::new(&connection).await {
|
if let Ok(proxy) = SwayncControlProxy::new(&connection).await
|
||||||
debug!("Found SwayNC, using it for DND state.");
|
&& let Ok(is_dnd) = proxy.dnd().await
|
||||||
|
{
|
||||||
|
debug!("Found SwayNC, using signal-based DND monitoring");
|
||||||
|
let _ = tx.send(DndState { is_dnd });
|
||||||
|
|
||||||
// Get initial state
|
|
||||||
if let Ok(is_dnd) = proxy.dnd().await {
|
|
||||||
let _ = tx.send(DndState { is_dnd });
|
|
||||||
}
|
|
||||||
|
|
||||||
// Monitor properties changed
|
|
||||||
if let Ok(props_proxy) = PropertiesProxy::builder(&connection)
|
if let Ok(props_proxy) = PropertiesProxy::builder(&connection)
|
||||||
.destination("org.erikreider.swaync.control")?
|
.destination("org.erikreider.swaync.control")?
|
||||||
.path("/org/erikreider/swaync/control")?
|
.path("/org/erikreider/swaync/control")?
|
||||||
@@ -97,8 +160,41 @@ impl DndDaemon {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
return Err(anyhow::anyhow!("SwayNC DND stream ended"));
|
||||||
}
|
}
|
||||||
|
|
||||||
Err(anyhow::anyhow!("DND stream ended or daemon not found"))
|
// Try Dunst via raw D-Bus calls (bypasses zbus proxy issues)
|
||||||
|
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)"
|
||||||
|
))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
+2
-8
@@ -3,7 +3,7 @@ use crate::error::Result;
|
|||||||
use crate::modules::WaybarModule;
|
use crate::modules::WaybarModule;
|
||||||
use crate::output::WaybarOutput;
|
use crate::output::WaybarOutput;
|
||||||
use crate::state::AppReceivers;
|
use crate::state::AppReceivers;
|
||||||
use crate::utils::{TokenValue, format_template};
|
use crate::utils::{TokenValue, classify_usage, format_template};
|
||||||
|
|
||||||
pub struct GpuModule;
|
pub struct GpuModule;
|
||||||
|
|
||||||
@@ -36,13 +36,7 @@ impl WaybarModule for GpuModule {
|
|||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
let class = if usage > 95.0 {
|
let class = classify_usage(usage, 75.0, 95.0);
|
||||||
"max"
|
|
||||||
} else if usage > 75.0 {
|
|
||||||
"high"
|
|
||||||
} else {
|
|
||||||
"normal"
|
|
||||||
};
|
|
||||||
|
|
||||||
let format_str = match vendor.as_str() {
|
let format_str = match vendor.as_str() {
|
||||||
"Intel" => &config.gpu.format_intel,
|
"Intel" => &config.gpu.format_intel,
|
||||||
|
|||||||
+125
-108
@@ -143,122 +143,139 @@ impl HardwareDaemon {
|
|||||||
async fn poll_gpu(&mut self, gpu: &mut crate::state::GpuState) {
|
async fn poll_gpu(&mut self, gpu: &mut crate::state::GpuState) {
|
||||||
gpu.active = false;
|
gpu.active = false;
|
||||||
|
|
||||||
if (self.gpu_vendor.as_deref() == Some("NVIDIA") || self.gpu_vendor.is_none())
|
match self.gpu_vendor.as_deref() {
|
||||||
&& let Ok(output) = tokio::process::Command::new("nvidia-smi")
|
Some("NVIDIA") => {
|
||||||
.args([
|
Self::poll_nvidia(gpu).await;
|
||||||
"--query-gpu=utilization.gpu,memory.used,memory.total,temperature.gpu,name",
|
}
|
||||||
"--format=csv,noheader,nounits",
|
Some("AMD") => {
|
||||||
])
|
Self::poll_amd(gpu);
|
||||||
.output()
|
}
|
||||||
.await
|
Some("Intel") => {
|
||||||
&& output.status.success()
|
Self::poll_intel(gpu);
|
||||||
{
|
}
|
||||||
let stdout = String::from_utf8_lossy(&output.stdout);
|
_ => {
|
||||||
if let Some(line) = stdout.lines().next() {
|
// Detection pass: try each vendor, cache the first that responds.
|
||||||
let parts: Vec<&str> = line.split(',').collect();
|
Self::poll_nvidia(gpu).await;
|
||||||
if parts.len() >= 5 {
|
if gpu.active {
|
||||||
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());
|
self.gpu_vendor = Some("NVIDIA".to_string());
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
}
|
Self::poll_amd(gpu);
|
||||||
}
|
if gpu.active {
|
||||||
|
|
||||||
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())
|
|
||||||
&& 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());
|
self.gpu_vendor = Some("AMD".to_string());
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
Self::poll_intel(gpu);
|
||||||
if self.gpu_vendor.as_deref() == Some("Intel") || self.gpu_vendor.is_none() {
|
if gpu.active {
|
||||||
let freq_path =
|
self.gpu_vendor = Some("Intel".to_string());
|
||||||
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
|
|
||||||
&& 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);
|
|
||||||
}
|
|
||||||
|
|
||||||
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;
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
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();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
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;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
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
|
||||||
|
&& 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);
|
||||||
|
}
|
||||||
|
|
||||||
|
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;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -3,7 +3,7 @@ use crate::error::Result;
|
|||||||
use crate::modules::WaybarModule;
|
use crate::modules::WaybarModule;
|
||||||
use crate::output::WaybarOutput;
|
use crate::output::WaybarOutput;
|
||||||
use crate::state::AppReceivers;
|
use crate::state::AppReceivers;
|
||||||
use crate::utils::{TokenValue, format_template};
|
use crate::utils::{TokenValue, classify_usage, format_template};
|
||||||
|
|
||||||
pub struct MemoryModule;
|
pub struct MemoryModule;
|
||||||
|
|
||||||
@@ -33,13 +33,7 @@ impl WaybarModule for MemoryModule {
|
|||||||
],
|
],
|
||||||
);
|
);
|
||||||
|
|
||||||
let class = if ratio > 95.0 {
|
let class = classify_usage(ratio, 75.0, 95.0);
|
||||||
"max"
|
|
||||||
} else if ratio > 75.0 {
|
|
||||||
"high"
|
|
||||||
} else {
|
|
||||||
"normal"
|
|
||||||
};
|
|
||||||
|
|
||||||
Ok(WaybarOutput {
|
Ok(WaybarOutput {
|
||||||
text,
|
text,
|
||||||
|
|||||||
@@ -1,18 +1,48 @@
|
|||||||
|
//! 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 audio;
|
||||||
|
#[cfg(feature = "mod-dbus")]
|
||||||
pub mod backlight;
|
pub mod backlight;
|
||||||
|
#[cfg(feature = "mod-bt")]
|
||||||
pub mod bt;
|
pub mod bt;
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
pub mod btrfs;
|
pub mod btrfs;
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
pub mod cpu;
|
pub mod cpu;
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
pub mod disk;
|
pub mod disk;
|
||||||
|
#[cfg(feature = "mod-dbus")]
|
||||||
pub mod dnd;
|
pub mod dnd;
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
pub mod game;
|
pub mod game;
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
pub mod gpu;
|
pub mod gpu;
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
pub mod hardware;
|
pub mod hardware;
|
||||||
|
#[cfg(feature = "mod-dbus")]
|
||||||
pub mod keyboard;
|
pub mod keyboard;
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
pub mod memory;
|
pub mod memory;
|
||||||
|
#[cfg(feature = "mod-dbus")]
|
||||||
pub mod mpris;
|
pub mod mpris;
|
||||||
|
#[cfg(feature = "mod-network")]
|
||||||
pub mod network;
|
pub mod network;
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
pub mod power;
|
pub mod power;
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
pub mod sys;
|
pub mod sys;
|
||||||
|
|
||||||
use crate::config::Config;
|
use crate::config::Config;
|
||||||
@@ -20,7 +50,14 @@ use crate::error::Result as FluxoResult;
|
|||||||
use crate::output::WaybarOutput;
|
use crate::output::WaybarOutput;
|
||||||
use crate::state::AppReceivers;
|
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 {
|
pub trait WaybarModule {
|
||||||
|
/// Evaluate the module and return its rendered output.
|
||||||
fn run(
|
fn run(
|
||||||
&self,
|
&self,
|
||||||
config: &Config,
|
config: &Config,
|
||||||
|
|||||||
+239
-105
@@ -2,12 +2,66 @@ use crate::config::Config;
|
|||||||
use crate::error::Result;
|
use crate::error::Result;
|
||||||
use crate::modules::WaybarModule;
|
use crate::modules::WaybarModule;
|
||||||
use crate::output::WaybarOutput;
|
use crate::output::WaybarOutput;
|
||||||
use crate::state::{AppReceivers, MprisState};
|
use crate::state::{AppReceivers, MprisScrollState, MprisState};
|
||||||
use crate::utils::{TokenValue, format_template};
|
use crate::utils::{TokenValue, format_template};
|
||||||
use tokio::sync::watch;
|
use std::sync::Arc;
|
||||||
|
use tokio::sync::{RwLock, watch};
|
||||||
|
use tokio::time::Duration;
|
||||||
use tracing::{debug, info};
|
use tracing::{debug, info};
|
||||||
use zbus::{Connection, proxy};
|
use zbus::{Connection, proxy};
|
||||||
|
|
||||||
|
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)
|
||||||
|
}
|
||||||
|
|
||||||
|
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()
|
||||||
|
}
|
||||||
|
|
||||||
|
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)
|
||||||
|
}
|
||||||
|
|
||||||
pub struct MprisModule;
|
pub struct MprisModule;
|
||||||
|
|
||||||
impl WaybarModule for MprisModule {
|
impl WaybarModule for MprisModule {
|
||||||
@@ -28,32 +82,26 @@ impl WaybarModule for MprisModule {
|
|||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
let status_icon = if mpris.is_playing {
|
let (full_text, class) = format_mpris_text(&config.mpris.format, &mpris);
|
||||||
""
|
|
||||||
} else if mpris.is_paused {
|
|
||||||
""
|
|
||||||
} else {
|
|
||||||
""
|
|
||||||
};
|
|
||||||
|
|
||||||
let class = if mpris.is_playing {
|
let text = if config.mpris.scroll {
|
||||||
"playing"
|
if let Some(max_len) = config.mpris.max_length {
|
||||||
} else if mpris.is_paused {
|
let scroll = state.mpris_scroll.read().await;
|
||||||
"paused"
|
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 {
|
} else {
|
||||||
"stopped"
|
full_text.clone()
|
||||||
};
|
};
|
||||||
|
|
||||||
let text = format_template(
|
|
||||||
&config.mpris.format,
|
|
||||||
&[
|
|
||||||
("artist", TokenValue::String(mpris.artist.clone())),
|
|
||||||
("title", TokenValue::String(mpris.title.clone())),
|
|
||||||
("album", TokenValue::String(mpris.album.clone())),
|
|
||||||
("status_icon", TokenValue::String(status_icon.to_string())),
|
|
||||||
],
|
|
||||||
);
|
|
||||||
|
|
||||||
Ok(WaybarOutput {
|
Ok(WaybarOutput {
|
||||||
text,
|
text,
|
||||||
tooltip: Some(format!("{} - {}", mpris.artist, mpris.title)),
|
tooltip: Some(format!("{} - {}", mpris.artist, mpris.title)),
|
||||||
@@ -63,6 +111,54 @@ impl WaybarModule for MprisModule {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
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 — wait for next state change
|
||||||
|
if mpris_rx.changed().await.is_err() {
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
pub struct MprisDaemon;
|
pub struct MprisDaemon;
|
||||||
|
|
||||||
#[proxy(
|
#[proxy(
|
||||||
@@ -105,103 +201,141 @@ impl MprisDaemon {
|
|||||||
}
|
}
|
||||||
|
|
||||||
async fn listen_loop(tx: &watch::Sender<MprisState>) -> anyhow::Result<()> {
|
async fn listen_loop(tx: &watch::Sender<MprisState>) -> anyhow::Result<()> {
|
||||||
|
use futures::StreamExt;
|
||||||
|
|
||||||
let connection = Connection::session().await?;
|
let connection = Connection::session().await?;
|
||||||
|
|
||||||
info!("Connected to D-Bus for MPRIS monitoring");
|
info!("Connected to D-Bus for MPRIS monitoring");
|
||||||
|
|
||||||
// Periodically poll for the active player and update the MPRIS state.
|
|
||||||
// This avoids complex dynamic signal tracking across ephemeral player instances.
|
|
||||||
|
|
||||||
let dbus_proxy = DBusProxy::new(&connection).await?;
|
let dbus_proxy = DBusProxy::new(&connection).await?;
|
||||||
|
|
||||||
loop {
|
loop {
|
||||||
|
// Discovery pass: find an active MPRIS player.
|
||||||
let names = dbus_proxy.list_names().await?;
|
let names = dbus_proxy.list_names().await?;
|
||||||
let mut active_player = None;
|
let active_player = names
|
||||||
|
.into_iter()
|
||||||
|
.find(|n| n.starts_with("org.mpris.MediaPlayer2."));
|
||||||
|
|
||||||
for name in names {
|
let Some(player_name) = active_player else {
|
||||||
if name.starts_with("org.mpris.MediaPlayer2.") {
|
send_stopped_if_changed(tx);
|
||||||
active_player = Some(name);
|
// No player — wait and re-discover.
|
||||||
break; // Just grab the first active player for now
|
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;
|
||||||
}
|
}
|
||||||
}
|
};
|
||||||
|
|
||||||
if let Some(player_name) = active_player {
|
// Initial fetch and then signal-driven updates via PropertiesChanged.
|
||||||
if let Ok(player_proxy) = MprisPlayerProxy::builder(&connection)
|
update_from_player(&player_proxy, tx).await;
|
||||||
.destination(player_name.clone())?
|
|
||||||
.build()
|
|
||||||
.await
|
|
||||||
{
|
|
||||||
let status = player_proxy.playback_status().await.unwrap_or_default();
|
|
||||||
let metadata = player_proxy.metadata().await.unwrap_or_default();
|
|
||||||
|
|
||||||
let is_playing = status == "Playing";
|
let mut status_stream = player_proxy.receive_playback_status_changed().await;
|
||||||
let is_paused = status == "Paused";
|
let mut metadata_stream = player_proxy.receive_metadata_changed().await;
|
||||||
let is_stopped = status == "Stopped";
|
|
||||||
|
|
||||||
let mut artist = String::new();
|
loop {
|
||||||
let mut title = String::new();
|
tokio::select! {
|
||||||
let mut album = String::new();
|
Some(_) = status_stream.next() => {
|
||||||
|
update_from_player(&player_proxy, tx).await;
|
||||||
if let Some(v) = metadata.get("xesam:artist") {
|
}
|
||||||
if let Ok(arr) = zbus::zvariant::Array::try_from(v) {
|
Some(_) = metadata_stream.next() => {
|
||||||
let mut artists = Vec::new();
|
update_from_player(&player_proxy, tx).await;
|
||||||
for i in 0..arr.len() {
|
}
|
||||||
if let Ok(Some(s)) = arr.get::<&str>(i) {
|
_ = tokio::time::sleep(Duration::from_secs(10)) => {
|
||||||
artists.push(s.to_string());
|
// Heartbeat: verify the player is still on the bus.
|
||||||
}
|
let current = dbus_proxy.list_names().await.unwrap_or_default();
|
||||||
}
|
if !current.iter().any(|n| n == &player_name) {
|
||||||
artist = artists.join(", ");
|
break;
|
||||||
} else if let Ok(a) = <&str>::try_from(v) {
|
|
||||||
artist = a.to_string();
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
if let Some(v) = metadata.get("xesam:title")
|
|
||||||
&& let Ok(t) = <&str>::try_from(v)
|
|
||||||
{
|
|
||||||
title = t.to_string();
|
|
||||||
}
|
|
||||||
if let Some(v) = metadata.get("xesam:album")
|
|
||||||
&& let Ok(a) = <&str>::try_from(v)
|
|
||||||
{
|
|
||||||
album = a.to_string();
|
|
||||||
}
|
|
||||||
|
|
||||||
// Only send if changed
|
|
||||||
let current = tx.borrow();
|
|
||||||
if current.is_playing != is_playing
|
|
||||||
|| current.is_paused != is_paused
|
|
||||||
|| current.is_stopped != is_stopped
|
|
||||||
|| current.title != title
|
|
||||||
|| current.artist != artist
|
|
||||||
|| current.album != album
|
|
||||||
{
|
|
||||||
drop(current); // Drop borrow before send
|
|
||||||
let _ = tx.send(MprisState {
|
|
||||||
is_playing,
|
|
||||||
is_paused,
|
|
||||||
is_stopped,
|
|
||||||
artist,
|
|
||||||
title,
|
|
||||||
album,
|
|
||||||
});
|
|
||||||
}
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
let current = tx.borrow();
|
|
||||||
if !current.is_stopped || !current.title.is_empty() {
|
|
||||||
drop(current);
|
|
||||||
let _ = tx.send(MprisState {
|
|
||||||
is_playing: false,
|
|
||||||
is_paused: false,
|
|
||||||
is_stopped: true,
|
|
||||||
artist: String::new(),
|
|
||||||
title: String::new(),
|
|
||||||
album: String::new(),
|
|
||||||
});
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
tokio::time::sleep(tokio::time::Duration::from_secs(2)).await;
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
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,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
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)
|
||||||
|
}
|
||||||
|
|
||||||
|
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(),
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
@@ -1,12 +1,22 @@
|
|||||||
|
//! JSON payload returned to Waybar custom modules.
|
||||||
|
|
||||||
use serde::{Deserialize, Serialize};
|
use serde::{Deserialize, Serialize};
|
||||||
|
|
||||||
|
/// 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)]
|
#[derive(Serialize, Deserialize, Clone, Debug, Default)]
|
||||||
pub struct WaybarOutput {
|
pub struct WaybarOutput {
|
||||||
|
/// Primary text shown in the bar.
|
||||||
pub text: String,
|
pub text: String,
|
||||||
|
/// Tooltip text shown on hover.
|
||||||
#[serde(skip_serializing_if = "Option::is_none")]
|
#[serde(skip_serializing_if = "Option::is_none")]
|
||||||
pub tooltip: Option<String>,
|
pub tooltip: Option<String>,
|
||||||
|
/// CSS class applied to the module (for styling).
|
||||||
#[serde(skip_serializing_if = "Option::is_none")]
|
#[serde(skip_serializing_if = "Option::is_none")]
|
||||||
pub class: Option<String>,
|
pub class: Option<String>,
|
||||||
|
/// Optional 0-100 value usable by bar progress indicators.
|
||||||
#[serde(skip_serializing_if = "Option::is_none")]
|
#[serde(skip_serializing_if = "Option::is_none")]
|
||||||
pub percentage: Option<u8>,
|
pub percentage: Option<u8>,
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -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);
|
||||||
+97
-90
@@ -1,8 +1,13 @@
|
|||||||
|
//! 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::config::Config;
|
||||||
use crate::state::AppReceivers;
|
use crate::state::AppReceivers;
|
||||||
use nix::libc;
|
|
||||||
use nix::sys::signal::{Signal, kill};
|
|
||||||
use nix::unistd::Pid;
|
|
||||||
use std::collections::HashMap;
|
use std::collections::HashMap;
|
||||||
use std::sync::Arc;
|
use std::sync::Arc;
|
||||||
use sysinfo::{ProcessesToUpdate, System};
|
use sysinfo::{ProcessesToUpdate, System};
|
||||||
@@ -10,13 +15,18 @@ use tokio::sync::RwLock;
|
|||||||
use tokio::time::{Duration, Instant, sleep};
|
use tokio::time::{Duration, Instant, sleep};
|
||||||
use tracing::{debug, warn};
|
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 {
|
pub struct WaybarSignaler {
|
||||||
cached_pid: Option<Pid>,
|
cached_pid: Option<i32>,
|
||||||
sys: System,
|
sys: System,
|
||||||
last_signal_sent: HashMap<i32, Instant>,
|
last_signal_sent: HashMap<i32, Instant>,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl WaybarSignaler {
|
impl WaybarSignaler {
|
||||||
|
/// Create a new signaler with no cached PID.
|
||||||
pub fn new() -> Self {
|
pub fn new() -> Self {
|
||||||
Self {
|
Self {
|
||||||
cached_pid: None,
|
cached_pid: None,
|
||||||
@@ -25,11 +35,11 @@ impl WaybarSignaler {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn find_waybar_pid(&mut self) -> Option<Pid> {
|
fn find_waybar_pid(&mut self) -> Option<i32> {
|
||||||
self.sys.refresh_processes(ProcessesToUpdate::All, true);
|
self.sys.refresh_processes(ProcessesToUpdate::All, true);
|
||||||
for (pid, process) in self.sys.processes() {
|
for (pid, process) in self.sys.processes() {
|
||||||
if process.name() == "waybar" {
|
if process.name() == "waybar" {
|
||||||
return Some(Pid::from_raw(pid.as_u32() as i32));
|
return Some(pid.as_u32() as i32);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
None
|
None
|
||||||
@@ -44,7 +54,7 @@ impl WaybarSignaler {
|
|||||||
|
|
||||||
let mut valid_pid = false;
|
let mut valid_pid = false;
|
||||||
if let Some(pid) = self.cached_pid
|
if let Some(pid) = self.cached_pid
|
||||||
&& kill(pid, None).is_ok()
|
&& unsafe { libc::kill(pid, 0) } == 0
|
||||||
{
|
{
|
||||||
valid_pid = true;
|
valid_pid = true;
|
||||||
}
|
}
|
||||||
@@ -54,103 +64,100 @@ impl WaybarSignaler {
|
|||||||
}
|
}
|
||||||
|
|
||||||
if let Some(pid) = self.cached_pid {
|
if let Some(pid) = self.cached_pid {
|
||||||
let rt_sig = match Signal::try_from(libc::SIGRTMIN() + signal_num) {
|
let sig = libc::SIGRTMIN() + signal_num;
|
||||||
Ok(sig) => sig,
|
if unsafe { libc::kill(pid, sig) } == 0 {
|
||||||
Err(_) => {
|
|
||||||
unsafe {
|
|
||||||
if libc::kill(pid.as_raw(), libc::SIGRTMIN() + signal_num) == 0 {
|
|
||||||
debug!("Sent raw SIGRTMIN+{} to waybar (PID: {})", signal_num, pid);
|
|
||||||
self.last_signal_sent.insert(signal_num, Instant::now());
|
|
||||||
} else {
|
|
||||||
warn!("Failed to send raw SIGRTMIN+{} to waybar", signal_num);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
if let Err(e) = kill(pid, rt_sig) {
|
|
||||||
warn!("Failed to signal waybar: {}", e);
|
|
||||||
self.cached_pid = None;
|
|
||||||
} else {
|
|
||||||
debug!("Sent SIGRTMIN+{} to waybar (PID: {})", signal_num, pid);
|
debug!("Sent SIGRTMIN+{} to waybar (PID: {})", signal_num, pid);
|
||||||
self.last_signal_sent.insert(signal_num, Instant::now());
|
self.last_signal_sent.insert(signal_num, Instant::now());
|
||||||
|
} else {
|
||||||
|
warn!("Failed to send SIGRTMIN+{} to waybar", signal_num);
|
||||||
|
self.cached_pid = None;
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
debug!("Waybar process not found, skipping signal.");
|
debug!("Waybar process not found, skipping signal.");
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
pub async fn run(mut self, config_lock: Arc<RwLock<Config>>, mut receivers: AppReceivers) {
|
/// Generates [`WaybarSignaler::run`] from the central module registry.
|
||||||
let mut last_outputs: HashMap<&'static str, String> = HashMap::new();
|
///
|
||||||
|
/// 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 {
|
loop {
|
||||||
let signals = config_lock.read().await.signals.clone();
|
let signals = config_lock.read().await.signals.clone();
|
||||||
|
|
||||||
macro_rules! check_and_signal {
|
macro_rules! check_and_signal {
|
||||||
($module_name:expr, $signal_opt:expr) => {
|
($module_name:expr, $signal_opt:expr) => {
|
||||||
if let Some(sig) = $signal_opt {
|
if let Some(sig) = $signal_opt {
|
||||||
let config = config_lock.read().await;
|
let config = config_lock.read().await;
|
||||||
if let Some(out) = crate::daemon::evaluate_module_for_signaler(
|
if let Some(out) = crate::daemon::evaluate_module_for_signaler(
|
||||||
$module_name,
|
$module_name,
|
||||||
&receivers,
|
&receivers,
|
||||||
&config,
|
&config,
|
||||||
)
|
)
|
||||||
.await
|
.await
|
||||||
{
|
{
|
||||||
if last_outputs.get($module_name) != Some(&out) {
|
if last_outputs.get($module_name) != Some(&out) {
|
||||||
last_outputs.insert($module_name, out);
|
last_outputs.insert($module_name, out);
|
||||||
self.send_signal(sig);
|
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
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
tokio::select! {
|
|
||||||
res = receivers.network.changed(), if signals.network.is_some() => {
|
|
||||||
if res.is_ok() { check_and_signal!("net", signals.network); }
|
|
||||||
}
|
|
||||||
res = receivers.cpu.changed(), if signals.cpu.is_some() => {
|
|
||||||
if res.is_ok() { check_and_signal!("cpu", signals.cpu); }
|
|
||||||
}
|
|
||||||
res = receivers.memory.changed(), if signals.memory.is_some() => {
|
|
||||||
if res.is_ok() { check_and_signal!("mem", signals.memory); }
|
|
||||||
}
|
|
||||||
res = receivers.sys.changed(), if signals.sys.is_some() => {
|
|
||||||
if res.is_ok() { check_and_signal!("sys", signals.sys); }
|
|
||||||
}
|
|
||||||
res = receivers.gpu.changed(), if signals.gpu.is_some() => {
|
|
||||||
if res.is_ok() { check_and_signal!("gpu", signals.gpu); }
|
|
||||||
}
|
|
||||||
res = receivers.disks.changed(), if signals.disk.is_some() => {
|
|
||||||
if res.is_ok() { check_and_signal!("disk", signals.disk); }
|
|
||||||
}
|
|
||||||
res = receivers.bluetooth.changed(), if signals.bt.is_some() => {
|
|
||||||
if res.is_ok() { check_and_signal!("bt", signals.bt); }
|
|
||||||
}
|
|
||||||
res = receivers.audio.changed(), if signals.audio.is_some() => {
|
|
||||||
if res.is_ok() {
|
|
||||||
check_and_signal!("vol", signals.audio);
|
|
||||||
check_and_signal!("mic", signals.audio);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
res = receivers.backlight.changed(), if signals.backlight.is_some() => {
|
|
||||||
if res.is_ok() { check_and_signal!("backlight", signals.backlight); }
|
|
||||||
}
|
|
||||||
res = receivers.keyboard.changed(), if signals.keyboard.is_some() => {
|
|
||||||
if res.is_ok() { check_and_signal!("keyboard", signals.keyboard); }
|
|
||||||
}
|
|
||||||
res = receivers.dnd.changed(), if signals.dnd.is_some() => {
|
|
||||||
if res.is_ok() { check_and_signal!("dnd", signals.dnd); }
|
|
||||||
}
|
|
||||||
res = receivers.mpris.changed(), if signals.mpris.is_some() => {
|
|
||||||
if res.is_ok() { check_and_signal!("mpris", signals.mpris); }
|
|
||||||
}
|
|
||||||
_ = sleep(Duration::from_secs(5)) => {
|
|
||||||
// loop and refresh config
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
|
for_each_watched_module!(gen_signaler_run);
|
||||||
|
|||||||
+163
-31
@@ -1,28 +1,50 @@
|
|||||||
|
//! 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.
|
||||||
|
|
||||||
use crate::output::WaybarOutput;
|
use crate::output::WaybarOutput;
|
||||||
use std::collections::HashMap;
|
use std::collections::HashMap;
|
||||||
use std::sync::Arc;
|
use std::sync::Arc;
|
||||||
use tokio::sync::{RwLock, mpsc, watch};
|
use tokio::sync::{RwLock, mpsc, watch};
|
||||||
use tokio::time::Instant;
|
use tokio::time::Instant;
|
||||||
|
|
||||||
#[derive(Clone)]
|
/// Generates the [`AppReceivers`] struct, with one `watch::Receiver` field
|
||||||
pub struct AppReceivers {
|
/// per watched module (cfg-gated by feature flag).
|
||||||
pub network: watch::Receiver<NetworkState>,
|
macro_rules! gen_app_receivers {
|
||||||
pub cpu: watch::Receiver<CpuState>,
|
($( { $feature:literal, $field:ident, $state:ty, [$($name:literal),+], [$($sig_name:literal),+], $module:path, $signal:ident, [$($default_arg:literal),*], $config:ident } )*) => {
|
||||||
pub memory: watch::Receiver<MemoryState>,
|
/// Bundle of every watch-channel receiver and control-channel handle.
|
||||||
pub sys: watch::Receiver<SysState>,
|
///
|
||||||
pub gpu: watch::Receiver<GpuState>,
|
/// Cloned into each IPC request handler and each module daemon so that
|
||||||
pub disks: watch::Receiver<Vec<DiskInfo>>,
|
/// they can snapshot state without locking. Per-module receivers are
|
||||||
pub bluetooth: watch::Receiver<BtState>,
|
/// generated from [`for_each_watched_module!`]; the remaining fields
|
||||||
pub audio: watch::Receiver<AudioState>,
|
/// (`health`, `bt_cycle`, `audio_cmd_tx`, `mpris_scroll_*`) are shared
|
||||||
pub mpris: watch::Receiver<MprisState>,
|
/// control state.
|
||||||
pub backlight: watch::Receiver<BacklightState>,
|
#[derive(Clone)]
|
||||||
pub keyboard: watch::Receiver<KeyboardState>,
|
pub struct AppReceivers {
|
||||||
pub dnd: watch::Receiver<DndState>,
|
$(
|
||||||
pub health: Arc<RwLock<HashMap<String, ModuleHealth>>>,
|
#[cfg(feature = $feature)]
|
||||||
pub bt_force_poll: mpsc::Sender<()>,
|
pub $field: watch::Receiver<$state>,
|
||||||
pub audio_cmd_tx: mpsc::Sender<crate::modules::audio::AudioCommand>,
|
)*
|
||||||
|
#[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)]
|
#[derive(Clone, Default)]
|
||||||
pub struct ModuleHealth {
|
pub struct ModuleHealth {
|
||||||
pub consecutive_failures: u32,
|
pub consecutive_failures: u32,
|
||||||
@@ -31,14 +53,16 @@ pub struct ModuleHealth {
|
|||||||
pub last_successful_output: Option<WaybarOutput>,
|
pub last_successful_output: Option<WaybarOutput>,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Current default audio sink + source plus all available devices.
|
||||||
#[derive(Default, Clone)]
|
#[derive(Default, Clone)]
|
||||||
pub struct AudioState {
|
pub struct AudioState {
|
||||||
pub sink: AudioDeviceInfo,
|
pub sink: AudioDeviceInfo,
|
||||||
pub source: AudioSourceInfo,
|
pub source: AudioDeviceInfo,
|
||||||
pub available_sinks: Vec<String>,
|
pub available_sinks: Vec<String>,
|
||||||
pub available_sources: Vec<String>,
|
pub available_sources: Vec<String>,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Metadata for one PulseAudio sink or source.
|
||||||
#[derive(Default, Clone)]
|
#[derive(Default, Clone)]
|
||||||
pub struct AudioDeviceInfo {
|
pub struct AudioDeviceInfo {
|
||||||
pub name: String,
|
pub name: String,
|
||||||
@@ -48,25 +72,37 @@ pub struct AudioDeviceInfo {
|
|||||||
pub channels: u8,
|
pub channels: u8,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// A single BlueZ device, optionally annotated by a plugin (buds, maestro…).
|
||||||
#[derive(Default, Clone)]
|
#[derive(Default, Clone)]
|
||||||
pub struct AudioSourceInfo {
|
pub struct BtDeviceInfo {
|
||||||
pub name: String,
|
|
||||||
pub description: String,
|
|
||||||
pub volume: u8,
|
|
||||||
pub muted: bool,
|
|
||||||
pub channels: u8,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Default, Clone)]
|
|
||||||
pub struct BtState {
|
|
||||||
pub connected: bool,
|
|
||||||
pub adapter_powered: bool,
|
|
||||||
pub device_alias: String,
|
pub device_alias: String,
|
||||||
pub device_address: String,
|
pub device_address: String,
|
||||||
pub battery_percentage: Option<u8>,
|
pub battery_percentage: Option<u8>,
|
||||||
pub plugin_data: Vec<(String, String)>,
|
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)]
|
#[derive(Default, Clone)]
|
||||||
pub struct DiskInfo {
|
pub struct DiskInfo {
|
||||||
pub mount_point: String,
|
pub mount_point: String,
|
||||||
@@ -75,6 +111,7 @@ pub struct DiskInfo {
|
|||||||
pub available_bytes: u64,
|
pub available_bytes: u64,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Throughput and identity of the active network interface.
|
||||||
#[derive(Default, Clone)]
|
#[derive(Default, Clone)]
|
||||||
pub struct NetworkState {
|
pub struct NetworkState {
|
||||||
pub rx_mbps: f64,
|
pub rx_mbps: f64,
|
||||||
@@ -83,6 +120,7 @@ pub struct NetworkState {
|
|||||||
pub ip: String,
|
pub ip: String,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// CPU utilisation and temperature.
|
||||||
#[derive(Clone)]
|
#[derive(Clone)]
|
||||||
pub struct CpuState {
|
pub struct CpuState {
|
||||||
pub usage: f64,
|
pub usage: f64,
|
||||||
@@ -100,12 +138,14 @@ impl Default for CpuState {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// RAM usage in gigabytes.
|
||||||
#[derive(Default, Clone)]
|
#[derive(Default, Clone)]
|
||||||
pub struct MemoryState {
|
pub struct MemoryState {
|
||||||
pub used_gb: f64,
|
pub used_gb: f64,
|
||||||
pub total_gb: f64,
|
pub total_gb: f64,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Load averages, uptime, and process count.
|
||||||
#[derive(Default, Clone)]
|
#[derive(Default, Clone)]
|
||||||
pub struct SysState {
|
pub struct SysState {
|
||||||
pub load_1: f64,
|
pub load_1: f64,
|
||||||
@@ -115,6 +155,9 @@ pub struct SysState {
|
|||||||
pub process_count: usize,
|
pub process_count: usize,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// GPU snapshot (vendor-agnostic).
|
||||||
|
///
|
||||||
|
/// `active` is `false` until detection finds a supported GPU.
|
||||||
#[derive(Clone)]
|
#[derive(Clone)]
|
||||||
pub struct GpuState {
|
pub struct GpuState {
|
||||||
pub active: bool,
|
pub active: bool,
|
||||||
@@ -140,21 +183,32 @@ impl Default for GpuState {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Do-Not-Disturb toggle state.
|
||||||
#[derive(Default, Clone)]
|
#[derive(Default, Clone)]
|
||||||
pub struct DndState {
|
pub struct DndState {
|
||||||
pub is_dnd: bool,
|
pub is_dnd: bool,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Currently active keyboard layout.
|
||||||
#[derive(Default, Clone)]
|
#[derive(Default, Clone)]
|
||||||
pub struct KeyboardState {
|
pub struct KeyboardState {
|
||||||
pub layout: String,
|
pub layout: String,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Display backlight brightness as a 0-100 percentage.
|
||||||
#[derive(Default, Clone)]
|
#[derive(Default, Clone)]
|
||||||
pub struct BacklightState {
|
pub struct BacklightState {
|
||||||
pub percentage: u8,
|
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)]
|
#[derive(Default, Clone)]
|
||||||
pub struct MprisState {
|
pub struct MprisState {
|
||||||
pub is_playing: bool,
|
pub is_playing: bool,
|
||||||
@@ -165,88 +219,166 @@ pub struct MprisState {
|
|||||||
pub album: 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)]
|
#[cfg(test)]
|
||||||
pub struct MockState {
|
pub struct MockState {
|
||||||
pub receivers: AppReceivers,
|
pub receivers: AppReceivers,
|
||||||
// Keep senders alive so receivers don't return Closed errors
|
#[cfg(feature = "mod-network")]
|
||||||
_net_tx: watch::Sender<NetworkState>,
|
_net_tx: watch::Sender<NetworkState>,
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
_cpu_tx: watch::Sender<CpuState>,
|
_cpu_tx: watch::Sender<CpuState>,
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
_mem_tx: watch::Sender<MemoryState>,
|
_mem_tx: watch::Sender<MemoryState>,
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
_sys_tx: watch::Sender<SysState>,
|
_sys_tx: watch::Sender<SysState>,
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
_gpu_tx: watch::Sender<GpuState>,
|
_gpu_tx: watch::Sender<GpuState>,
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
_disks_tx: watch::Sender<Vec<DiskInfo>>,
|
_disks_tx: watch::Sender<Vec<DiskInfo>>,
|
||||||
|
#[cfg(feature = "mod-bt")]
|
||||||
_bt_tx: watch::Sender<BtState>,
|
_bt_tx: watch::Sender<BtState>,
|
||||||
|
#[cfg(feature = "mod-audio")]
|
||||||
_audio_tx: watch::Sender<AudioState>,
|
_audio_tx: watch::Sender<AudioState>,
|
||||||
|
#[cfg(feature = "mod-dbus")]
|
||||||
_mpris_tx: watch::Sender<MprisState>,
|
_mpris_tx: watch::Sender<MprisState>,
|
||||||
|
#[cfg(feature = "mod-dbus")]
|
||||||
_backlight_tx: watch::Sender<BacklightState>,
|
_backlight_tx: watch::Sender<BacklightState>,
|
||||||
|
#[cfg(feature = "mod-dbus")]
|
||||||
_keyboard_tx: watch::Sender<KeyboardState>,
|
_keyboard_tx: watch::Sender<KeyboardState>,
|
||||||
|
#[cfg(feature = "mod-dbus")]
|
||||||
_dnd_tx: watch::Sender<DndState>,
|
_dnd_tx: watch::Sender<DndState>,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Plain-data counterpart of [`AppReceivers`] used to seed a [`MockState`].
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
#[derive(Default, Clone)]
|
#[derive(Default, Clone)]
|
||||||
pub struct AppState {
|
pub struct AppState {
|
||||||
|
#[cfg(feature = "mod-network")]
|
||||||
pub network: NetworkState,
|
pub network: NetworkState,
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
pub cpu: CpuState,
|
pub cpu: CpuState,
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
pub memory: MemoryState,
|
pub memory: MemoryState,
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
pub sys: SysState,
|
pub sys: SysState,
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
pub gpu: GpuState,
|
pub gpu: GpuState,
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
pub disks: Vec<DiskInfo>,
|
pub disks: Vec<DiskInfo>,
|
||||||
|
#[cfg(feature = "mod-bt")]
|
||||||
pub bluetooth: BtState,
|
pub bluetooth: BtState,
|
||||||
|
#[cfg(feature = "mod-audio")]
|
||||||
pub audio: AudioState,
|
pub audio: AudioState,
|
||||||
|
#[cfg(feature = "mod-dbus")]
|
||||||
pub mpris: MprisState,
|
pub mpris: MprisState,
|
||||||
|
#[cfg(feature = "mod-dbus")]
|
||||||
pub backlight: BacklightState,
|
pub backlight: BacklightState,
|
||||||
|
#[cfg(feature = "mod-dbus")]
|
||||||
pub keyboard: KeyboardState,
|
pub keyboard: KeyboardState,
|
||||||
|
#[cfg(feature = "mod-dbus")]
|
||||||
pub dnd: DndState,
|
pub dnd: DndState,
|
||||||
pub health: HashMap<String, ModuleHealth>,
|
pub health: HashMap<String, ModuleHealth>,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Build a [`MockState`] from a plain [`AppState`] snapshot for unit tests.
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
pub fn mock_state(state: AppState) -> MockState {
|
pub fn mock_state(state: AppState) -> MockState {
|
||||||
|
#[cfg(feature = "mod-network")]
|
||||||
let (net_tx, net_rx) = watch::channel(state.network);
|
let (net_tx, net_rx) = watch::channel(state.network);
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
let (cpu_tx, cpu_rx) = watch::channel(state.cpu);
|
let (cpu_tx, cpu_rx) = watch::channel(state.cpu);
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
let (mem_tx, mem_rx) = watch::channel(state.memory);
|
let (mem_tx, mem_rx) = watch::channel(state.memory);
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
let (sys_tx, sys_rx) = watch::channel(state.sys);
|
let (sys_tx, sys_rx) = watch::channel(state.sys);
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
let (gpu_tx, gpu_rx) = watch::channel(state.gpu);
|
let (gpu_tx, gpu_rx) = watch::channel(state.gpu);
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
let (disks_tx, disks_rx) = watch::channel(state.disks);
|
let (disks_tx, disks_rx) = watch::channel(state.disks);
|
||||||
|
#[cfg(feature = "mod-bt")]
|
||||||
let (bt_tx, bt_rx) = watch::channel(state.bluetooth);
|
let (bt_tx, bt_rx) = watch::channel(state.bluetooth);
|
||||||
|
#[cfg(feature = "mod-audio")]
|
||||||
let (audio_tx, audio_rx) = watch::channel(state.audio);
|
let (audio_tx, audio_rx) = watch::channel(state.audio);
|
||||||
|
#[cfg(feature = "mod-dbus")]
|
||||||
let (mpris_tx, mpris_rx) = watch::channel(state.mpris);
|
let (mpris_tx, mpris_rx) = watch::channel(state.mpris);
|
||||||
|
#[cfg(feature = "mod-dbus")]
|
||||||
let (backlight_tx, backlight_rx) = watch::channel(state.backlight);
|
let (backlight_tx, backlight_rx) = watch::channel(state.backlight);
|
||||||
|
#[cfg(feature = "mod-dbus")]
|
||||||
let (keyboard_tx, keyboard_rx) = watch::channel(state.keyboard);
|
let (keyboard_tx, keyboard_rx) = watch::channel(state.keyboard);
|
||||||
|
#[cfg(feature = "mod-dbus")]
|
||||||
let (dnd_tx, dnd_rx) = watch::channel(state.dnd);
|
let (dnd_tx, dnd_rx) = watch::channel(state.dnd);
|
||||||
|
#[cfg(feature = "mod-bt")]
|
||||||
let (bt_force_tx, _) = mpsc::channel(1);
|
let (bt_force_tx, _) = mpsc::channel(1);
|
||||||
|
#[cfg(feature = "mod-audio")]
|
||||||
let (audio_cmd_tx, _) = mpsc::channel(1);
|
let (audio_cmd_tx, _) = mpsc::channel(1);
|
||||||
|
|
||||||
MockState {
|
MockState {
|
||||||
receivers: AppReceivers {
|
receivers: AppReceivers {
|
||||||
|
#[cfg(feature = "mod-network")]
|
||||||
network: net_rx,
|
network: net_rx,
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
cpu: cpu_rx,
|
cpu: cpu_rx,
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
memory: mem_rx,
|
memory: mem_rx,
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
sys: sys_rx,
|
sys: sys_rx,
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
gpu: gpu_rx,
|
gpu: gpu_rx,
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
disks: disks_rx,
|
disks: disks_rx,
|
||||||
|
#[cfg(feature = "mod-bt")]
|
||||||
bluetooth: bt_rx,
|
bluetooth: bt_rx,
|
||||||
|
#[cfg(feature = "mod-bt")]
|
||||||
|
bt_cycle: Arc::new(RwLock::new(0usize)),
|
||||||
|
#[cfg(feature = "mod-audio")]
|
||||||
audio: audio_rx,
|
audio: audio_rx,
|
||||||
|
#[cfg(feature = "mod-dbus")]
|
||||||
mpris: mpris_rx,
|
mpris: mpris_rx,
|
||||||
|
#[cfg(feature = "mod-dbus")]
|
||||||
backlight: backlight_rx,
|
backlight: backlight_rx,
|
||||||
|
#[cfg(feature = "mod-dbus")]
|
||||||
keyboard: keyboard_rx,
|
keyboard: keyboard_rx,
|
||||||
|
#[cfg(feature = "mod-dbus")]
|
||||||
dnd: dnd_rx,
|
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)),
|
health: Arc::new(RwLock::new(state.health)),
|
||||||
|
#[cfg(feature = "mod-bt")]
|
||||||
bt_force_poll: bt_force_tx,
|
bt_force_poll: bt_force_tx,
|
||||||
|
#[cfg(feature = "mod-audio")]
|
||||||
audio_cmd_tx,
|
audio_cmd_tx,
|
||||||
},
|
},
|
||||||
|
#[cfg(feature = "mod-network")]
|
||||||
_net_tx: net_tx,
|
_net_tx: net_tx,
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
_cpu_tx: cpu_tx,
|
_cpu_tx: cpu_tx,
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
_mem_tx: mem_tx,
|
_mem_tx: mem_tx,
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
_sys_tx: sys_tx,
|
_sys_tx: sys_tx,
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
_gpu_tx: gpu_tx,
|
_gpu_tx: gpu_tx,
|
||||||
|
#[cfg(feature = "mod-hardware")]
|
||||||
_disks_tx: disks_tx,
|
_disks_tx: disks_tx,
|
||||||
|
#[cfg(feature = "mod-bt")]
|
||||||
_bt_tx: bt_tx,
|
_bt_tx: bt_tx,
|
||||||
|
#[cfg(feature = "mod-audio")]
|
||||||
_audio_tx: audio_tx,
|
_audio_tx: audio_tx,
|
||||||
|
#[cfg(feature = "mod-dbus")]
|
||||||
_mpris_tx: mpris_tx,
|
_mpris_tx: mpris_tx,
|
||||||
|
#[cfg(feature = "mod-dbus")]
|
||||||
_backlight_tx: backlight_tx,
|
_backlight_tx: backlight_tx,
|
||||||
|
#[cfg(feature = "mod-dbus")]
|
||||||
_keyboard_tx: keyboard_tx,
|
_keyboard_tx: keyboard_tx,
|
||||||
|
#[cfg(feature = "mod-dbus")]
|
||||||
_dnd_tx: dnd_tx,
|
_dnd_tx: dnd_tx,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
+58
-30
@@ -1,23 +1,31 @@
|
|||||||
|
//! Shared helpers: menu spawning, Hyprland socket lookup, and template formatting.
|
||||||
|
|
||||||
use anyhow::{Context, Result};
|
use anyhow::{Context, Result};
|
||||||
use std::io::Write;
|
use std::io::Write;
|
||||||
use std::process::{Command, Stdio};
|
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> {
|
pub fn show_menu(prompt: &str, items: &[String], menu_cmd: &str) -> Result<String> {
|
||||||
// Backward compatibility for {prompt}, but environment variable is safer
|
|
||||||
let cmd_str = menu_cmd.replace("{prompt}", prompt);
|
let cmd_str = menu_cmd.replace("{prompt}", prompt);
|
||||||
let mut child = Command::new("sh")
|
let mut child = Command::new("sh")
|
||||||
.arg("-c")
|
.arg("-c")
|
||||||
.arg(&cmd_str)
|
.arg(&cmd_str)
|
||||||
.env("FLUXO_PROMPT", prompt) // Safer shell injection
|
.env("FLUXO_PROMPT", prompt)
|
||||||
.stdin(Stdio::piped())
|
.stdin(Stdio::piped())
|
||||||
.stdout(Stdio::piped())
|
.stdout(Stdio::piped())
|
||||||
.stderr(Stdio::null()) // Suppress GTK/Wayland warnings from tools like wofi
|
// Suppress GTK/Wayland noise from tools like wofi.
|
||||||
|
.stderr(Stdio::null())
|
||||||
.spawn()
|
.spawn()
|
||||||
.context("Failed to spawn menu command")?;
|
.context("Failed to spawn menu command")?;
|
||||||
|
|
||||||
if let Some(mut stdin) = child.stdin.take() {
|
if let Some(mut stdin) = child.stdin.take() {
|
||||||
let mut input = items.join("\n");
|
let mut input = items.join("\n");
|
||||||
input.push('\n'); // Ensure trailing newline for wofi/rofi
|
// Trailing newline is required by wofi/rofi.
|
||||||
|
input.push('\n');
|
||||||
stdin
|
stdin
|
||||||
.write_all(input.as_bytes())
|
.write_all(input.as_bytes())
|
||||||
.context("Failed to write to menu stdin")?;
|
.context("Failed to write to menu stdin")?;
|
||||||
@@ -37,11 +45,14 @@ pub fn show_menu(prompt: &str, items: &[String], menu_cmd: &str) -> Result<Strin
|
|||||||
Ok(selected)
|
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> {
|
pub fn get_hyprland_socket(socket_name: &str) -> Result<std::path::PathBuf> {
|
||||||
let signature = std::env::var("HYPRLAND_INSTANCE_SIGNATURE")
|
let signature = std::env::var("HYPRLAND_INSTANCE_SIGNATURE")
|
||||||
.context("HYPRLAND_INSTANCE_SIGNATURE not set")?;
|
.context("HYPRLAND_INSTANCE_SIGNATURE not set")?;
|
||||||
|
|
||||||
// Try XDG_RUNTIME_DIR first (usually /run/user/1000)
|
|
||||||
if let Ok(runtime_dir) = std::env::var("XDG_RUNTIME_DIR") {
|
if let Ok(runtime_dir) = std::env::var("XDG_RUNTIME_DIR") {
|
||||||
let path = std::path::PathBuf::from(runtime_dir)
|
let path = std::path::PathBuf::from(runtime_dir)
|
||||||
.join("hypr")
|
.join("hypr")
|
||||||
@@ -52,7 +63,6 @@ pub fn get_hyprland_socket(socket_name: &str) -> Result<std::path::PathBuf> {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Fallback to /tmp
|
|
||||||
let path = std::path::PathBuf::from("/tmp/hypr")
|
let path = std::path::PathBuf::from("/tmp/hypr")
|
||||||
.join(&signature)
|
.join(&signature)
|
||||||
.join(socket_name);
|
.join(socket_name);
|
||||||
@@ -70,42 +80,60 @@ pub fn get_hyprland_socket(socket_name: &str) -> Result<std::path::PathBuf> {
|
|||||||
use regex::Regex;
|
use regex::Regex;
|
||||||
use std::sync::LazyLock;
|
use std::sync::LazyLock;
|
||||||
|
|
||||||
|
/// 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"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// A typed value supplied to [`format_template`] — chosen at call site so
|
||||||
|
/// formatting handles precision and alignment correctly per type.
|
||||||
pub enum TokenValue {
|
pub enum TokenValue {
|
||||||
Float(f64),
|
Float(f64),
|
||||||
Int(i64),
|
Int(i64),
|
||||||
String(String),
|
String(String),
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// 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());
|
||||||
|
|
||||||
|
/// 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
|
pub fn format_template<K>(template: &str, values: &[(K, TokenValue)]) -> String
|
||||||
where
|
where
|
||||||
K: AsRef<str>,
|
K: AsRef<str>,
|
||||||
{
|
{
|
||||||
static RE: LazyLock<Regex> = LazyLock::new(|| {
|
TOKEN_RE
|
||||||
Regex::new(r"\{([a-zA-Z0-9_]+)(?::([<>\^])?(\d+)?(?:\.(\d+))?)?\}").unwrap()
|
.replace_all(template, |caps: ®ex::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));
|
||||||
|
|
||||||
RE.replace_all(template, |caps: ®ex::Captures| {
|
match val {
|
||||||
let name = &caps[1];
|
TokenValue::Float(f) => format_float(*f, align, width, precision),
|
||||||
if let Some((_, val)) = values.iter().find(|(k, _)| k.as_ref() == name) {
|
TokenValue::Int(i) => format_int(*i, align, width),
|
||||||
let align = caps.get(2).map(|m| m.as_str()).unwrap_or(">");
|
TokenValue::String(s) => format_str(s, align, width),
|
||||||
let width = caps
|
}
|
||||||
.get(3)
|
} else {
|
||||||
.map(|m| m.as_str().parse::<usize>().unwrap_or(0))
|
caps[0].to_string()
|
||||||
.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()
|
.into_owned()
|
||||||
}
|
|
||||||
})
|
|
||||||
.into_owned()
|
|
||||||
}
|
}
|
||||||
|
|
||||||
fn format_float(f: f64, align: &str, width: usize, precision: Option<usize>) -> String {
|
fn format_float(f: f64, align: &str, width: usize, precision: Option<usize>) -> String {
|
||||||
|
|||||||
Reference in New Issue
Block a user