check if is $XAUTHORITY set

coastalwhite · Jan 7, 2025 · 783ea39 · 783ea39
1 parent 1d4be7d
commit 783ea39
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/post_login/x.rs b/src/post_login/x.rs
@@ -85,8 +85,11 @@ pub fn setup_x(
     let vtnr_value = env::var("XDG_VTNR").map_err(|_| XSetupError::VTNREnvVar)?;
 
     // Setup xauth
-    let xauth_dir = PathBuf::from(env::var("HOME").map_err(|_| XSetupError::HomeEnvVar)?);
-    let xauth_path = xauth_dir.join(".Xauthority");
+    info!("Check if `XAUTHORITY` enviroment variable is set");
+    let xauth_path = match env::var("XAUTHORITY") {
+        Ok(var) => PathBuf::from(env::var(var).map_err(|_| XSetupError::FillingXAuth)?),
+        Err(_) => PathBuf::from(env::var("HOME").map_err(|_| XSetupError::HomeEnvVar)?).join(".Xauthority")
+        };
 
     info!(
         "Filling `.Xauthority` file at `{xauth_path}`",