From 612404bdce4d7066b4f56b9373e91074b19c6348 Mon Sep 17 00:00:00 2001
From: Jaromil <jaromil@dyne.org>
Date: Sun, 20 Feb 2022 21:23:01 +0100
Subject: [PATCH] cleanup and support sup,sud,pkexec

---
 tomb | 34 ++++++++--------------------------
 1 file changed, 8 insertions(+), 26 deletions(-)

diff --git a/tomb b/tomb
index 71e43c4..5be3a9e 100755
--- a/tomb
+++ b/tomb
@@ -114,11 +114,12 @@ export TEXTDOMAIN=tomb
 # Wrap sudo with a more visible message or apply user-supplied alternative to sudo
 _sudo() {
 	if option_is_set --sudo; then
-		case `basename $(option_value --sudo)` in
-			"doas")
-				command -v doas >/dev/null || _failure "Doas executable not found in PATH"
-				_verbose "Super user execution using doas"
-				doas ${@}
+               pescmd=`option_value --sudo`
+		case `basename $pescmd` in
+			"doas"|"sup"|"sud"|"pkexec")
+				command -v $pescmd > /dev/null || _failure "$pescmd executable not found"
+				_verbose "Super user execution using $pescmd"
+				${pescmd} ${@}
 				return $?
 				;;
 			"skip"|"none")
@@ -3290,27 +3291,8 @@ main() {
 	option_is_set --tmp && TMPDIR=$(option_value --tmp)
 
 	option_is_set --sudo && {
-	    local _opt=`option_value --sudo`
-	    case $_opt in
-		"sudo") # the worst default ever
-		    SUDO=sudo
-		    ;;
-		"doas") # openbsd
-		    SUDO=doas
-		    ;;
-		"pkexec") # policykit
-		    SUDO=pkexec
-		    ;;
-		"sup") # https://sup.dyne.org
-		    SUDO=sup
-		    ;;
-		"sud") # https://sud.dyne.org
-		    SUDO=sud
-		    ;;
-		*)
-		    _failure "Invalid privilege escalation tool: ::1 exec::" $_opt
-		    ;;
-	    esac
+	    local _opt=`basename $(option_value --sudo)`
+            _message "Privilege escalation tool configured: ::1 exec::" $_opt
 	}
 
 	# When we run as root, we remember the original uid:gid to set