mirror of
https://github.com/YaLTeR/niri.git
synced 2024-09-11 12:35:58 +03:00
Add dinit support (#246)
* Add dinit support - Add --notify-fd cli flag for ready notifications - Set dinit activation environment when "dinit" feature flag is enabled * Make systemd and dinit environment activation additive * Use NOTIFY_FD env variable instead of --notify-fd cli flag for sending ready notifications * Format with rustfmt
This commit is contained in:
parent
9cbbffc23c
commit
431f070481
@ -107,6 +107,8 @@ systemd = ["dbus"]
|
||||
xdp-gnome-screencast = ["dbus", "pipewire"]
|
||||
# Enables the Tracy profiler instrumentation.
|
||||
profile-with-tracy = ["profiling/profile-with-tracy", "tracy-client/default"]
|
||||
# Enables dinit integration (global environment)
|
||||
dinit = []
|
||||
|
||||
[profile.release]
|
||||
debug = "line-tables-only"
|
||||
|
35
src/main.rs
35
src/main.rs
@ -1,8 +1,10 @@
|
||||
#[macro_use]
|
||||
extern crate tracing;
|
||||
|
||||
use std::fmt::Write as _;
|
||||
use std::fs::{self, File};
|
||||
use std::io::{self, Write};
|
||||
use std::os::fd::FromRawFd;
|
||||
use std::path::PathBuf;
|
||||
use std::process::Command;
|
||||
use std::{env, mem};
|
||||
@ -211,6 +213,11 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
|
||||
warn!("error notifying systemd: {err:?}");
|
||||
};
|
||||
|
||||
// Send ready notification to specified file descriptor
|
||||
if let Err(err) = notify_fd() {
|
||||
warn!("error notifying fd: {err:?}");
|
||||
}
|
||||
|
||||
// Set up config file watcher.
|
||||
let _watcher = if let Some(path) = path.clone() {
|
||||
let (tx, rx) = calloop::channel::sync_channel(1);
|
||||
@ -258,16 +265,23 @@ fn import_environment() {
|
||||
]
|
||||
.join(" ");
|
||||
|
||||
#[cfg(feature = "systemd")]
|
||||
let systemctl = format!("systemctl --user import-environment {variables} && ");
|
||||
#[cfg(not(feature = "systemd"))]
|
||||
let systemctl = String::new();
|
||||
let mut init_system_import = String::new();
|
||||
if cfg!(feature = "systemd") {
|
||||
write!(
|
||||
init_system_import,
|
||||
"systemctl --user import-environment {variables};"
|
||||
)
|
||||
.unwrap();
|
||||
}
|
||||
if cfg!(feature = "dinit") {
|
||||
write!(init_system_import, "dinitctl setenv {variables};").unwrap();
|
||||
}
|
||||
|
||||
let rv = Command::new("/bin/sh")
|
||||
.args([
|
||||
"-c",
|
||||
&format!(
|
||||
"{systemctl}\
|
||||
"{init_system_import}\
|
||||
hash dbus-update-activation-environment 2>/dev/null && \
|
||||
dbus-update-activation-environment {variables}"
|
||||
),
|
||||
@ -302,3 +316,14 @@ fn default_config_path() -> Option<PathBuf> {
|
||||
path.push("config.kdl");
|
||||
Some(path)
|
||||
}
|
||||
|
||||
fn notify_fd() -> anyhow::Result<()> {
|
||||
let fd = match env::var("NOTIFY_FD") {
|
||||
Ok(notify_fd) => notify_fd.parse()?,
|
||||
Err(env::VarError::NotPresent) => return Ok(()),
|
||||
Err(err) => return Err(err.into()),
|
||||
};
|
||||
let mut notif = unsafe { File::from_raw_fd(fd) };
|
||||
notif.write_all(b"READY=1\n")?;
|
||||
Ok(())
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user