From 4c1dedc2752802fa9da26865f94a4c3a3945c2ec Mon Sep 17 00:00:00 2001
From: Admin9705 <9705@duck.com>
Date: Tue, 17 Sep 2024 05:49:24 -0400
Subject: [PATCH] update

---
 mods/scripts/menu.sh | 32 ++++++++++++++++++++++++++++----
 1 file changed, 28 insertions(+), 4 deletions(-)

diff --git a/mods/scripts/menu.sh b/mods/scripts/menu.sh
index f290839ec..53d76e52b 100644
--- a/mods/scripts/menu.sh
+++ b/mods/scripts/menu.sh
@@ -7,15 +7,39 @@ NC="\033[0m"  # No color
 # Get the username of the user with UID 1000
 REQUIRED_USER=$(getent passwd 1000 | cut -d: -f1)
 
-# Enhanced security check: Block root user and only allow user with UID 1000 and GID 1000
-if [[ $EUID -eq 0 ]] || [[ $(id -u) -ne 1000 ]] || [[ $(id -g) -ne 1000 ]]; then
-    echo -e "${RED}WARNING: This script can only be run by the user '$REQUIRED_USER' (UID 1000 and GID 1000).${NC}"
-    echo -e "${RED}It cannot be run as root or any other user.${NC}"
+# Function to check if the script is being run with sudo
+is_sudo() {
+    if [ -n "$SUDO_USER" ]; then
+        return 0 # True, it's being run with sudo
+    else
+        return 1 # False, it's not being run with sudo
+    fi
+}
+
+# Enhanced security check
+if [[ -z "$SUDO_USER" ]]; then
+    echo -e "${RED}WARNING: This script must be run with sudo.${NC}"
+    echo -e "${RED}Please run it as 'sudo -u $REQUIRED_USER $0 $@'${NC}"
+    read -p "Press [ENTER] to acknowledge"
+    bash /pg/installer/menu_exit.sh
+    exit 1
+elif [[ $SUDO_UID -ne 1000 ]] || [[ $SUDO_GID -ne 1000 ]]; then
+    echo -e "${RED}WARNING: This script can only be run by the user '$REQUIRED_USER' (UID 1000 and GID 1000) using sudo.${NC}"
+    echo -e "${RED}Please run it as 'sudo -u $REQUIRED_USER $0 $@'${NC}"
+    read -p "Press [ENTER] to acknowledge"
+    bash /pg/installer/menu_exit.sh
+    exit 1
+elif [[ $EUID -ne 0 ]]; then
+    echo -e "${RED}WARNING: This script must be run with sudo privileges.${NC}"
+    echo -e "${RED}Please run it as 'sudo -u $REQUIRED_USER $0 $@'${NC}"
     read -p "Press [ENTER] to acknowledge"
     bash /pg/installer/menu_exit.sh
     exit 1
 fi
 
+# If we've made it here, the user is either UID 1000 or is UID 1000 using sudo
+echo "Security check passed. Proceeding with the script..."
+
 # Configuration file path
 CONFIG_FILE="/pg/config/config.cfg"