Export NIRI_SOCKET to systemd/dbus environment

2 files changed, 8 insertions, 4 deletions

diff --git a/resources/niri-session b/resources/niri-session
index 0ba9af34..fb3121d8 100755
--- a/resources/niri-session
+++ b/resources/niri-session
@@ -44,4 +44,4 @@ systemctl --user --wait start niri.service
 systemctl --user start --job-mode=replace-irreversibly niri-shutdown.target
 
 # Unset environment that we've set.
-systemctl --user unset-environment WAYLAND_DISPLAY XDG_SESSION_TYPE XDG_CURRENT_DESKTOP
+systemctl --user unset-environment WAYLAND_DISPLAY XDG_SESSION_TYPE XDG_CURRENT_DESKTOP NIRI_SOCKET
diff --git a/src/main.rs b/src/main.rs
index ed9ac6d4..bcf0feae 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -235,12 +235,16 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
 }
 
 fn import_env_to_systemd() {
+    let variables = ["WAYLAND_DISPLAY", niri_ipc::SOCKET_PATH_ENV].join(" ");
+
     let rv = Command::new("/bin/sh")
         .args([
             "-c",
-            "systemctl --user import-environment WAYLAND_DISPLAY && \
-             hash dbus-update-activation-environment 2>/dev/null && \
-             dbus-update-activation-environment WAYLAND_DISPLAY",
+            &format!(
+                "systemctl --user import-environment {variables} && \
+                 hash dbus-update-activation-environment 2>/dev/null && \
+                 dbus-update-activation-environment {variables}"
+            ),
         ])
         .spawn();
     // Wait for the import process to complete, otherwise services will start too fast without