update SUDO_ASKPASS variable with NH_SUDO_ASKPASS

Author: dvishal485

src/commands.rs

@@ -84,8 +84,9 @@ impl Command {
             };
 
             // use NH_SUDO_ASKPASS program for sudo if present
-            let cmd = if std::env::var("NH_SUDO_ASKPASS").is_ok() {
-                cmd.arg("-A")
+            let askpass = std::env::var("NH_SUDO_ASKPASS");
+            let cmd = if let Ok(askpass) = askpass {
+                cmd.env("SUDO_ASKPASS", askpass).arg("-A")
             } else {
                 cmd
             };

src/main.rs

@@ -55,8 +55,11 @@ fn self_elevate() -> ! {
     use std::os::unix::process::CommandExt;
 
     let mut cmd = std::process::Command::new("sudo");
+
     // use NH_SUDO_ASKPASS program for sudo if present
-    if std::env::var("NH_SUDO_ASKPASS").is_ok() {
+    let askpass = std::env::var("NH_SUDO_ASKPASS");
+    if let Ok(askpass) = askpass {
+        cmd.env("SUDO_ASKPASS", askpass);
         cmd.arg("-A");
     }