feat: support to call pkexec-like on *nix

ArcticLampyrid · Mar 18, 2024 · 6247fac · 6247fac
1 parent aa54092
commit 6247fac
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 1 deletion.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/gui/Cargo.toml b/gui/Cargo.toml
@@ -13,6 +13,8 @@ portable-pty = "0.8.1"
 strip-ansi-escapes = "0.2.0"
 once_cell = "1.18.0"
 native-dialog = "0.7"
+which = "6.0"
+libc = "0.2"
 
 [target.'cfg(windows)'.dependencies]
 windows = { version = "0.51", features = [

diff --git a/gui/src/main.rs b/gui/src/main.rs
@@ -106,7 +106,22 @@ fn do_env_up(ui_handle: &Weak<AppWindow>) {
         std::process::exit(0);
     }
 
-    let mut cmd = CommandBuilder::new(PATH_OF_CLI.as_os_str());
+    let mut cmd: CommandBuilder = 'b: {
+        #[cfg(unix)]
+        if unsafe { libc::getuid() } != 0 {
+            if which::which("lxqt-sudo").is_ok() {
+                let mut cmd = CommandBuilder::new("lxqt-sudo");
+                cmd.arg(PATH_OF_CLI.as_os_str());
+                break 'b cmd;
+            }
+            if which::which("pkexec").is_ok() {
+                let mut cmd = CommandBuilder::new("pkexec");
+                cmd.arg(PATH_OF_CLI.as_os_str());
+                break 'b cmd;
+            }
+        }
+        break 'b CommandBuilder::new(PATH_OF_CLI.as_os_str());
+    };
     cmd.arg("env");
     cmd.arg("up");
     let ui_handle_2 = ui_handle.clone();