Add check for /root

Heavybullets8 · Oct 27, 2023 · 2c08ae1 · 2c08ae1
1 parent 058c2cd
commit 2c08ae1
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/bin/heavyscript b/bin/heavyscript
@@ -25,7 +25,9 @@ err_exit() {
     exit "${2:-1}"
 }
 
-if [[ $EUID -eq 0 && -n ${SUDO_USER:-} ]]; then
+if [[ -d "/root/heavy_script" ]]; then
+    script_dir="/root/heavy_script"
+elif [[ $EUID -eq 0 && -n ${SUDO_USER:-} ]]; then
     script_dir="/home/$SUDO_USER/heavy_script"
 elif [[ -d "$HOME/heavy_script" ]]; then
     script_dir="$HOME/heavy_script"