feat: switch to smol

This commit is contained in:
Vladimir Rubin 2025-04-27 19:31:24 +03:00
parent b2f03f048b
commit 640dab8f56
Signed by: vavakado
GPG key ID: CAB744727F36B524
3 changed files with 448 additions and 267 deletions

385
Cargo.lock generated
View file

@ -3,40 +3,155 @@
version = 4 version = 4
[[package]] [[package]]
name = "addr2line" name = "async-channel"
version = "0.24.2" version = "2.3.1"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "dfbe277e56a376000877090da837660b4427aad530e3028d44e0bffe4f89a1c1" checksum = "89b47800b0be77592da0afd425cc03468052844aff33b84e33cc696f64e77b6a"
dependencies = [ dependencies = [
"gimli", "concurrent-queue",
"event-listener-strategy",
"futures-core",
"pin-project-lite",
] ]
[[package]] [[package]]
name = "adler2" name = "async-executor"
version = "1.13.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "30ca9a001c1e8ba5149f91a74362376cc6bc5b919d92d988668657bd570bdcec"
dependencies = [
"async-task",
"concurrent-queue",
"fastrand",
"futures-lite",
"slab",
]
[[package]]
name = "async-fs"
version = "2.1.2"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "ebcd09b382f40fcd159c2d695175b2ae620ffa5f3bd6f664131efff4e8b9e04a"
dependencies = [
"async-lock",
"blocking",
"futures-lite",
]
[[package]]
name = "async-io"
version = "2.4.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "43a2b323ccce0a1d90b449fd71f2a06ca7faa7c54c2751f06c9bd851fc061059"
dependencies = [
"async-lock",
"cfg-if",
"concurrent-queue",
"futures-io",
"futures-lite",
"parking",
"polling",
"rustix",
"slab",
"tracing",
"windows-sys",
]
[[package]]
name = "async-lock"
version = "3.4.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "ff6e472cdea888a4bd64f342f09b3f50e1886d32afe8df3d663c01140b811b18"
dependencies = [
"event-listener",
"event-listener-strategy",
"pin-project-lite",
]
[[package]]
name = "async-net"
version = "2.0.0" version = "2.0.0"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "512761e0bb2578dd7380c6baaa0f4ce03e84f95e960231d1dec8bf4d7d6e2627" checksum = "b948000fad4873c1c9339d60f2623323a0cfd3816e5181033c6a5cb68b2accf7"
[[package]]
name = "backtrace"
version = "0.3.74"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "8d82cb332cdfaed17ae235a638438ac4d4839913cc2af585c3c6746e8f8bee1a"
dependencies = [ dependencies = [
"addr2line", "async-io",
"cfg-if", "blocking",
"libc", "futures-lite",
"miniz_oxide",
"object",
"rustc-demangle",
"windows-targets",
] ]
[[package]] [[package]]
name = "bytes" name = "async-process"
version = "1.10.1" version = "2.3.0"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "d71b6127be86fdcfddb610f7182ac57211d4b18a3e9c82eb2d17662f2227ad6a" checksum = "63255f1dc2381611000436537bbedfe83183faa303a5a0edaf191edef06526bb"
dependencies = [
"async-channel",
"async-io",
"async-lock",
"async-signal",
"async-task",
"blocking",
"cfg-if",
"event-listener",
"futures-lite",
"rustix",
"tracing",
]
[[package]]
name = "async-signal"
version = "0.2.10"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "637e00349800c0bdf8bfc21ebbc0b6524abea702b0da4168ac00d070d0c0b9f3"
dependencies = [
"async-io",
"async-lock",
"atomic-waker",
"cfg-if",
"futures-core",
"futures-io",
"rustix",
"signal-hook-registry",
"slab",
"windows-sys",
]
[[package]]
name = "async-task"
version = "4.7.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "8b75356056920673b02621b35afd0f7dda9306d03c79a30f5c56c44cf256e3de"
[[package]]
name = "atomic-waker"
version = "1.1.2"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "1505bd5d3d116872e7271a6d4e16d81d0c8570876c8de68093a09ac269d8aac0"
[[package]]
name = "autocfg"
version = "1.4.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "ace50bade8e6234aa140d9a2f552bbee1db4d353f69b8217bc503490fc1a9f26"
[[package]]
name = "bitflags"
version = "2.9.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "5c8214115b7bf84099f1309324e63141d4c5d7cc26862f97a0a857dbefe165bd"
[[package]]
name = "blocking"
version = "1.6.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "703f41c54fc768e63e091340b424302bb1c29ef4aa0c7f10fe849dfb114d29ea"
dependencies = [
"async-channel",
"async-task",
"futures-io",
"futures-lite",
"piper",
]
[[package]] [[package]]
name = "cfg-if" name = "cfg-if"
@ -45,17 +160,94 @@ source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "baf1de4339761588bc0619e3cbc0120ee582ebb74b53b4efbf79117bd2da40fd" checksum = "baf1de4339761588bc0619e3cbc0120ee582ebb74b53b4efbf79117bd2da40fd"
[[package]] [[package]]
name = "gimli" name = "concurrent-queue"
version = "0.31.1" version = "2.5.0"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "07e28edb80900c19c28f1072f2e8aeca7fa06b23cd4169cefe1af5aa3260783f" checksum = "4ca0197aee26d1ae37445ee532fefce43251d24cc7c166799f4d46817f1d3973"
dependencies = [
"crossbeam-utils",
]
[[package]]
name = "crossbeam-utils"
version = "0.8.21"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "d0a5c400df2834b80a4c3327b3aad3a4c4cd4de0629063962b03235697506a28"
[[package]]
name = "errno"
version = "0.3.11"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "976dd42dc7e85965fe702eb8164f21f450704bdde31faefd6471dba214cb594e"
dependencies = [
"libc",
"windows-sys",
]
[[package]]
name = "event-listener"
version = "5.4.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "3492acde4c3fc54c845eaab3eed8bd00c7a7d881f78bfc801e43a93dec1331ae"
dependencies = [
"concurrent-queue",
"parking",
"pin-project-lite",
]
[[package]]
name = "event-listener-strategy"
version = "0.5.4"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "8be9f3dfaaffdae2972880079a491a1a8bb7cbed0b8dd7a347f668b4150a3b93"
dependencies = [
"event-listener",
"pin-project-lite",
]
[[package]]
name = "fastrand"
version = "2.3.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "37909eebbb50d72f9059c3b6d82c0463f2ff062c9e95845c43a6c9c0355411be"
[[package]]
name = "futures-core"
version = "0.3.31"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "05f29059c0c2090612e8d742178b0580d2dc940c837851ad723096f87af6663e"
[[package]]
name = "futures-io"
version = "0.3.31"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "9e5c1b78ca4aae1ac06c48a526a655760685149f0d465d21f37abfe57ce075c6"
[[package]]
name = "futures-lite"
version = "2.6.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "f5edaec856126859abb19ed65f39e90fea3a9574b9707f13539acf4abf7eb532"
dependencies = [
"fastrand",
"futures-core",
"futures-io",
"parking",
"pin-project-lite",
]
[[package]]
name = "hermit-abi"
version = "0.4.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "fbf6a919d6cf397374f7dfeeea91d974c7c0a7221d0d0f4f20d859d329e53fcc"
[[package]] [[package]]
name = "hinoirisetr" name = "hinoirisetr"
version = "0.2.0" version = "0.2.0"
dependencies = [ dependencies = [
"libloading", "libloading",
"tokio", "smol",
] ]
[[package]] [[package]]
@ -75,39 +267,16 @@ dependencies = [
] ]
[[package]] [[package]]
name = "memchr" name = "linux-raw-sys"
version = "2.7.4" version = "0.4.15"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "78ca9ab1a0babb1e7d5695e3530886289c18cf2f87ec19a575a0abdce112e3a3" checksum = "d26c52dbd32dccf2d10cac7725f8eae5296885fb5703b261f7d0a0739ec807ab"
[[package]] [[package]]
name = "miniz_oxide" name = "parking"
version = "0.8.8" version = "2.2.1"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "3be647b768db090acb35d5ec5db2b0e1f1de11133ca123b9eacf5137868f892a" checksum = "f38d5652c16fde515bb1ecef450ab0f6a219d619a7274976324d5e377f7dceba"
dependencies = [
"adler2",
]
[[package]]
name = "mio"
version = "1.0.3"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "2886843bf800fba2e3377cff24abf6379b4c4d5c6681eaf9ea5b0d15090450bd"
dependencies = [
"libc",
"wasi",
"windows-sys",
]
[[package]]
name = "object"
version = "0.36.7"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "62948e14d923ea95ea2c7c86c71013138b66525b86bdc08d2dcc262bdb497b87"
dependencies = [
"memchr",
]
[[package]] [[package]]
name = "pin-project-lite" name = "pin-project-lite"
@ -116,28 +285,43 @@ source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "3b3cff922bd51709b605d9ead9aa71031d81447142d828eb4a6eba76fe619f9b" checksum = "3b3cff922bd51709b605d9ead9aa71031d81447142d828eb4a6eba76fe619f9b"
[[package]] [[package]]
name = "proc-macro2" name = "piper"
version = "1.0.95" version = "0.2.4"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "02b3e5e68a3a1a02aad3ec490a98007cbc13c37cbe84a3cd7b8e406d76e7f778" checksum = "96c8c490f422ef9a4efd2cb5b42b76c8613d7e7dfc1caf667b8a3350a5acc066"
dependencies = [ dependencies = [
"unicode-ident", "atomic-waker",
"fastrand",
"futures-io",
] ]
[[package]] [[package]]
name = "quote" name = "polling"
version = "1.0.40" version = "3.7.4"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "1885c039570dc00dcb4ff087a89e185fd56bae234ddc7f056a945bf36467248d" checksum = "a604568c3202727d1507653cb121dbd627a58684eb09a820fd746bee38b4442f"
dependencies = [ dependencies = [
"proc-macro2", "cfg-if",
"concurrent-queue",
"hermit-abi",
"pin-project-lite",
"rustix",
"tracing",
"windows-sys",
] ]
[[package]] [[package]]
name = "rustc-demangle" name = "rustix"
version = "0.1.24" version = "0.38.44"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "719b953e2095829ee67db738b3bfa9fa368c94900df327b3f07fe6e794d2fe1f" checksum = "fdb5bc1ae2baa591800df16c9ca78619bf65c0488b41b96ccec5d11220d8c154"
dependencies = [
"bitflags",
"errno",
"libc",
"linux-raw-sys",
"windows-sys",
]
[[package]] [[package]]
name = "signal-hook-registry" name = "signal-hook-registry"
@ -149,71 +333,52 @@ dependencies = [
] ]
[[package]] [[package]]
name = "socket2" name = "slab"
version = "0.5.9" version = "0.4.9"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "4f5fd57c80058a56cf5c777ab8a126398ece8e442983605d280a44ce79d0edef" checksum = "8f92a496fb766b417c996b9c5e57daf2f7ad3b0bebe1ccfca4856390e3d3bb67"
dependencies = [ dependencies = [
"libc", "autocfg",
"windows-sys",
] ]
[[package]] [[package]]
name = "syn" name = "smol"
version = "2.0.100" version = "2.0.2"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "b09a44accad81e1ba1cd74a32461ba89dee89095ba17b32f5d03683b1b1fc2a0" checksum = "a33bd3e260892199c3ccfc487c88b2da2265080acb316cd920da72fdfd7c599f"
dependencies = [ dependencies = [
"proc-macro2", "async-channel",
"quote", "async-executor",
"unicode-ident", "async-fs",
"async-io",
"async-lock",
"async-net",
"async-process",
"blocking",
"futures-lite",
] ]
[[package]] [[package]]
name = "tokio" name = "tracing"
version = "1.44.2" version = "0.1.41"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "e6b88822cbe49de4185e3a4cbf8321dd487cf5fe0c5c65695fef6346371e9c48" checksum = "784e0ac535deb450455cbfa28a6f0df145ea1bb7ae51b821cf5e7927fdcfbdd0"
dependencies = [ dependencies = [
"backtrace",
"bytes",
"libc",
"mio",
"pin-project-lite", "pin-project-lite",
"signal-hook-registry", "tracing-core",
"socket2",
"tokio-macros",
"windows-sys",
] ]
[[package]] [[package]]
name = "tokio-macros" name = "tracing-core"
version = "2.5.0" version = "0.1.33"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "6e06d43f1345a3bcd39f6a56dbb7dcab2ba47e68e8ac134855e7e2bdbaf8cab8" checksum = "e672c95779cf947c5311f83787af4fa8fffd12fb27e4993211a84bdfd9610f9c"
dependencies = [
"proc-macro2",
"quote",
"syn",
]
[[package]]
name = "unicode-ident"
version = "1.0.18"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "5a5f39404a5da50712a4c1eecf25e90dd62b613502b7e925fd4e4d19b5c96512"
[[package]]
name = "wasi"
version = "0.11.0+wasi-snapshot-preview1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "9c8d87e72b64a3b4db28d11ce29237c246188f4f51057d65a7eab63b7987e423"
[[package]] [[package]]
name = "windows-sys" name = "windows-sys"
version = "0.52.0" version = "0.59.0"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "282be5f36a8ce781fad8c8ae18fa3f9beff57ec1b52cb3de0789201425d9a33d" checksum = "1e38bc4d79ed67fd075bcc251a1c39b32a1776bbe92e5bef1f0bf1f8c531853b"
dependencies = [ dependencies = [
"windows-targets", "windows-targets",
] ]

View file

@ -12,7 +12,7 @@ edition = "2024"
[dependencies] [dependencies]
libloading = "0.8.6" libloading = "0.8.6"
tokio = { version = "1", features = ["rt-multi-thread", "macros", "sync", "time","io-util","net","signal" ] } smol = "2.0.2"
[profile.release] [profile.release]
strip = true strip = true

View file

@ -1,29 +1,31 @@
use std::env; use std::env;
use std::path::PathBuf; use std::path::PathBuf;
use std::sync::atomic::{AtomicBool, Ordering}; use std::sync::atomic::Ordering::SeqCst;
use std::sync::atomic::{AtomicBool, AtomicU64, Ordering};
use std::sync::{Arc, OnceLock}; use std::sync::{Arc, OnceLock};
use std::time::{Duration, UNIX_EPOCH}; use std::time::{Duration, UNIX_EPOCH};
use hinoirisetr::notify::InitializedNotificationSystem; use hinoirisetr::notify::InitializedNotificationSystem;
use hinoirisetr::time::Time; use hinoirisetr::time::Time;
use hinoirisetr::{Config, apply_settings, compute_settings, debug, error, info, trace, warn}; use hinoirisetr::{Config, apply_settings, compute_settings, debug, error, info, trace, warn};
use tokio::io::{AsyncBufReadExt, BufReader}; use smol::Timer;
use tokio::net::UnixListener; use smol::channel::{Sender, unbounded};
use tokio::signal::unix::{SignalKind, signal}; use smol::io::{AsyncBufReadExt, BufReader};
use tokio::sync::{Notify, RwLock}; use smol::lock::{RwLock, RwLockReadGuard};
use tokio::time::sleep; use smol::net::unix::UnixListener;
use smol::stream::StreamExt;
const SOCKET_PATH: &str = "/tmp/hinoirisetr.sock"; const SOCKET_PATH: &str = "/tmp/hinoirisetr.sock";
static CONFIG: OnceLock<Arc<RwLock<Config>>> = OnceLock::new(); static CONFIG: OnceLock<Arc<RwLock<Config>>> = OnceLock::new();
static LAST_MODIFIED: OnceLock<Arc<RwLock<Option<u64>>>> = OnceLock::new(); static LAST_MODIFIED: OnceLock<Arc<AtomicU64>> = OnceLock::new();
enum NotifyState { enum NotifyState {
Enabled(InitializedNotificationSystem), Enabled(InitializedNotificationSystem),
Disabled, Disabled,
} }
async fn config_realoader(notify: Arc<Notify>) { async fn config_realoader(notify: Arc<Sender<()>>) {
debug!("config reloader started"); debug!("config reloader started");
// Config polling every 5 seconds // Config polling every 5 seconds
loop { loop {
@ -31,12 +33,12 @@ async fn config_realoader(notify: Arc<Notify>) {
let config_path = get_config_path(); let config_path = get_config_path();
if config_path.exists() { if config_path.exists() {
let last_modified = LAST_MODIFIED.get().expect("last_modified not init"); let last_modified = LAST_MODIFIED.get().expect("last_modified not init");
let last_modified_guard = last_modified.read().await;
if let Ok(current_modified) = std::fs::metadata(&config_path) if let Ok(current_modified) = std::fs::metadata(&config_path)
.and_then(|m| m.modified()) .and_then(|m| m.modified())
.map(|t| t.duration_since(UNIX_EPOCH).unwrap().as_secs()) .map(|t| t.duration_since(UNIX_EPOCH).unwrap().as_secs())
{ {
if let Some(last) = *last_modified_guard { let last: u64 = last_modified.load(Ordering::SeqCst);
if 0 != last {
if current_modified > last { if current_modified > last {
trace!("{current_modified}"); trace!("{current_modified}");
trace!("{last}"); trace!("{last}");
@ -50,11 +52,11 @@ async fn config_realoader(notify: Arc<Notify>) {
} }
} }
} }
sleep(Duration::from_secs(5)).await; Timer::after(Duration::from_secs(5)).await;
} }
} }
async fn socket_server(disabled: Arc<AtomicBool>, notify: Arc<Notify>) { async fn socket_server(disabled: Arc<AtomicBool>, notify: Arc<Sender<()>>) {
let listener = UnixListener::bind(SOCKET_PATH).expect("Failed to bind socket"); let listener = UnixListener::bind(SOCKET_PATH).expect("Failed to bind socket");
trace!("socket server bound"); trace!("socket server bound");
let notification: NotifyState = let notification: NotifyState =
@ -68,29 +70,30 @@ async fn socket_server(disabled: Arc<AtomicBool>, notify: Arc<Notify>) {
loop { loop {
let (stream, _) = listener.accept().await.unwrap(); let (stream, _) = listener.accept().await.unwrap();
let mut lines = BufReader::new(stream).lines(); let reader = BufReader::new(stream);
let mut lines = reader.lines();
trace!("socket server accepted connection"); trace!("socket server accepted connection");
if let Ok(Some(line)) = lines.next_line().await { if let Some(Ok(line)) = lines.next().await {
match line.trim() { match line.trim() {
"disable" => { "disable" => {
trace!("disable dispatched"); trace!("disable dispatched");
disabled.store(true, Ordering::SeqCst); disabled.store(true, Ordering::SeqCst);
notify.notify_one(); let _ = notify.send(()).await;
debug!("dimming is disabled"); debug!("dimming is disabled");
} }
"enable" => { "enable" => {
trace!("enable dispatched"); trace!("enable dispatched");
disabled.store(false, Ordering::SeqCst); disabled.store(false, Ordering::SeqCst);
notify.notify_one(); let _ = notify.send(()).await;
debug!("dimming is enabled"); debug!("dimming is enabled");
} }
"toggle" => { "toggle" => {
trace!("toggle dispatched"); trace!("toggle dispatched");
let now = !disabled.load(Ordering::SeqCst); let now = !disabled.load(Ordering::SeqCst);
disabled.store(now, Ordering::SeqCst); disabled.store(now, Ordering::SeqCst);
notify.notify_one(); let _ = notify.send(()).await;
debug!("dimming is {}", if now { "enabled" } else { "disabled" }); debug!("dimming is {}", if now { "enabled" } else { "disabled" });
} }
"status" => { "status" => {
@ -128,7 +131,7 @@ async fn socket_server(disabled: Arc<AtomicBool>, notify: Arc<Notify>) {
match notification { match notification {
NotifyState::Enabled(ref not) => { NotifyState::Enabled(ref not) => {
trace!("notify notification enabled"); trace!("notify notification enabled");
let timeout = config_handle().read().await.notification_timeout; let timeout = config_guard().await.notification_timeout;
match not.show_notification( match not.show_notification(
"Sunsetting", "Sunsetting",
&body, &body,
@ -150,139 +153,152 @@ async fn socket_server(disabled: Arc<AtomicBool>, notify: Arc<Notify>) {
} }
} }
#[tokio::main] fn main() {
async fn main() { smol::block_on(async {
match env::var("RUST_LOG") { match env::var("RUST_LOG") {
Ok(val) => match val.parse::<hinoirisetr::log::LogLevel>() { Ok(val) => match val.parse::<hinoirisetr::log::LogLevel>() {
Ok(level) => hinoirisetr::log::set_log_level(level), Ok(level) => hinoirisetr::log::set_log_level(level),
Err(err) => error!("Failed to parse RUST_LOG: {err}"), Err(err) => error!("Failed to parse RUST_LOG: {err}"),
}, },
Err(_) => {
if cfg!(debug_assertions) {
hinoirisetr::log::set_log_level(hinoirisetr::log::LogLevel::Debug);
} else {
hinoirisetr::log::set_log_level(hinoirisetr::log::LogLevel::Info);
}
}
}
info!("starting the daemon");
if !is_binary_available("hyprctl") {
error!("hyprctl is not available, exiting.");
std::process::exit(1);
}
let disabled = Arc::new(AtomicBool::new(false));
let notify = Arc::new(Notify::new());
// load config
let config_path = get_config_path();
let cfg: Config = if config_path.exists() {
debug!("Config file found, loading...");
LAST_MODIFIED
.set(Arc::new(RwLock::new(
std::fs::metadata(&config_path)
.and_then(|m| m.modified())
.map(|t| t.duration_since(UNIX_EPOCH).unwrap().as_secs())
.ok(),
)))
.unwrap();
match Config::load(&config_path) {
Ok(cfg) => cfg,
Err(err) => {
error!("Failed to load config: {err:?}");
warn!("Using default config.");
Config::default()
}
}
} else {
warn!("Config file not found, using default config.");
warn!("Config path {}", get_config_path().display());
Config::default()
};
CONFIG.set(Arc::new(RwLock::new(cfg))).unwrap();
if std::path::Path::new(SOCKET_PATH).exists() {
match std::os::unix::net::UnixStream::connect(SOCKET_PATH) {
Ok(_) => {
error!("Another instance is running.");
std::process::exit(1);
}
Err(_) => { Err(_) => {
warn!("Stale socket found, removing."); if cfg!(debug_assertions) {
let _ = std::fs::remove_file(SOCKET_PATH); hinoirisetr::log::set_log_level(hinoirisetr::log::LogLevel::Debug);
}
}
}
// Spawn control socket server
{
let disabled = Arc::clone(&disabled);
let notify = Arc::clone(&notify);
tokio::spawn(async move {
socket_server(disabled, notify).await;
});
}
// Spawn config reloader
{
let notify = Arc::clone(&notify);
tokio::spawn(async move {
config_realoader(notify).await;
});
}
// Signal handling
let mut sigint = signal(SignalKind::interrupt()).unwrap();
let mut sigterm = signal(SignalKind::terminate()).unwrap();
// set initial settings
{
let now = get_time();
let (temp, gamma) = compute_settings(now, &*config_guard().await);
apply_settings(temp, gamma);
trace!("initial settings applied: {temp}K, {gamma}%");
}
// Main loop with shutdown support
tokio::select! {
_ = async {
loop {
if disabled.load(Ordering::SeqCst) {
apply_settings(config_guard().await.temp_day, config_guard().await.gamma_day);
} else { } else {
let now = get_time(); hinoirisetr::log::set_log_level(hinoirisetr::log::LogLevel::Info);
let (temp, gamma) = compute_settings(now,&*config_guard().await);
apply_settings(temp, gamma);
}
tokio::select! {
_ = sleep(Duration::from_secs(300)) => {trace!("main loop tick");},
_ = notify.notified() => {trace!("main loop woke up via notify");},
} }
} }
} => {},
_ = sigint.recv() => {
info!("Received SIGINT, shutting down...");
},
_ = sigterm.recv() => {
info!("Received SIGTERM, shutting down...");
},
}
// Cleanup the socket file on shutdown
if std::path::Path::new(SOCKET_PATH).exists() {
match std::fs::remove_file(SOCKET_PATH) {
Ok(_) => info!("Socket file {SOCKET_PATH} removed."),
Err(e) => warn!("Failed to remove socket file {SOCKET_PATH}: {e}"),
} }
}
info!("starting the daemon");
if !is_binary_available("hyprctl") {
error!("hyprctl is not available, exiting.");
std::process::exit(1);
}
let disabled = Arc::new(AtomicBool::new(false));
let (notify_s, notify_r) = unbounded::<()>();
let notify = Arc::new(notify_s);
// load config
let config_path = get_config_path();
let cfg: Config = if config_path.exists() {
debug!("Config file found, loading...");
LAST_MODIFIED
.set(Arc::new(AtomicU64::new(
std::fs::metadata(&config_path)
.and_then(|m| m.modified())
.map(|t| t.duration_since(UNIX_EPOCH).unwrap().as_secs())
.unwrap_or(0),
)))
.unwrap();
match Config::load(&config_path) {
Ok(cfg) => cfg,
Err(err) => {
error!("Failed to load config: {err:?}");
warn!("Using default config.");
Config::default()
}
}
} else {
warn!("Config file not found, using default config.");
warn!("Config path {}", get_config_path().display());
Config::default()
};
CONFIG.set(Arc::new(RwLock::new(cfg))).unwrap();
if std::path::Path::new(SOCKET_PATH).exists() {
match std::os::unix::net::UnixStream::connect(SOCKET_PATH) {
Ok(_) => {
error!("Another instance is running.");
std::process::exit(1);
}
Err(_) => {
warn!("Stale socket found, removing.");
let _ = std::fs::remove_file(SOCKET_PATH);
}
}
}
// Spawn control socket server
{
let disabled = Arc::clone(&disabled);
let notify = Arc::clone(&notify);
smol::spawn(async move {
socket_server(disabled, notify).await;
})
.detach();
}
// Spawn config reloader
{
let notify = Arc::clone(&notify);
smol::spawn(async move {
config_realoader(notify).await;
})
.detach();
}
// Spawn timer
{
let notify = Arc::clone(&notify);
smol::spawn(async move {
loop {
Timer::after(Duration::from_secs(300)).await;
let _ = notify.send(()).await;
}
})
.detach();
}
// Signal handling
// let mut sigint = signal(SignalKind::interrupt()).unwrap();
// let mut sigterm = signal(SignalKind::terminate()).unwrap();
// set initial settings
{
let now = get_time();
let (temp, gamma) = compute_settings(now, &*config_guard().await);
apply_settings(temp, gamma);
trace!("initial settings applied: {temp}K, {gamma}%");
}
// Main loop with shutdown support
loop {
if disabled.load(Ordering::SeqCst) {
apply_settings(
config_guard().await.temp_day,
config_guard().await.gamma_day,
);
} else {
let now = get_time();
let (temp, gamma) = compute_settings(now, &*config_guard().await);
apply_settings(temp, gamma);
}
let _ = notify_r.recv().await;
}
// _ = sigint.recv() => {
// info!("Received SIGINT, shutting down...");
// },
// _ = sigterm.recv() => {
// info!("Received SIGTERM, shutting down...");
// },
// }
// Cleanup the socket file on shutdown
// if std::path::Path::new(SOCKET_PATH).exists() {
// match std::fs::remove_file(SOCKET_PATH) {
// Ok(_) => info!("Socket file {SOCKET_PATH} removed."),
// Err(e) => warn!("Failed to remove socket file {SOCKET_PATH}: {e}"),
// }
// }
})
} }
// Function to handle config reloading // Function to handle config reloading
async fn reload_config(notify: Arc<Notify>) { async fn reload_config(notify: Arc<Sender<()>>) {
trace!("reload_config called"); trace!("reload_config called");
let config_handle = config_handle(); let config_handle = config_handle();
let mut config = config_handle.write().await; let mut config = config_handle.write().await;
@ -293,18 +309,18 @@ async fn reload_config(notify: Arc<Notify>) {
*config = cfg; *config = cfg;
trace!("new config: {:#?}", config); trace!("new config: {:#?}", config);
let last_modified = LAST_MODIFIED.get().expect("last_modified not init"); // let mut last_modified = LAST_MODIFIED.get().expect("last_modified not init").clone();
let mut last_modified = last_modified.write().await; let new_modified = std::fs::metadata(&config_path)
tokio::task::spawn_blocking(move || { .and_then(|m| m.modified())
let new_modified = std::fs::metadata(&config_path) .map(|t| t.duration_since(UNIX_EPOCH).unwrap().as_secs())
.and_then(|m| m.modified()) .unwrap_or(0);
.map(|t| t.duration_since(UNIX_EPOCH).unwrap().as_secs()) trace!("new_modified: {new_modified:?}");
.ok(); LAST_MODIFIED
trace!("last_modified: {last_modified:?}"); .get()
trace!("new_modified: {new_modified:?}"); .expect("last_modified not init")
*last_modified = new_modified; .store(new_modified, SeqCst);
}); // last_modified = new_modified.into();
notify.notify_one(); let _ = notify.send(()).await;
} }
Err(err) => { Err(err) => {
error!("Failed to reload config: {err:?}"); error!("Failed to reload config: {err:?}");
@ -348,7 +364,7 @@ fn get_config_path() -> PathBuf {
} }
} }
async fn config_guard() -> tokio::sync::RwLockReadGuard<'static, Config> { async fn config_guard() -> RwLockReadGuard<'static, Config> {
CONFIG.get().expect("config not init").read().await CONFIG.get().expect("config not init").read().await
} }