Improve command line argument processing

lug-helper.sh

@@ -1212,43 +1212,68 @@ if [ "$#" -gt 0 ]; then
         case "$1" in
             --help | -h )
                 printf "Star Citizen Linux Users Group Helper Script
-
 Usage: lug-helper <options>
-
   -p, --preflight-check     Run system optimization checks
   -m, --manage-runners      Install or remove Lutris runners
- -s,  --sanitize            Delete Star Citizen USER folder, preserving keybinds
+  -u, --delete-user-folder  Delete Star Citizen USER folder, preserving keybinds
- -d,  --delete-shaders      Delete Star Citizen shaders directory
+  -s, --delete-shaders      Delete Star Citizen shaders directory
   -c, --delete-dxvk-cache   Delete Star Citizen dxvk cache file
- -g,  --get-referral        Get a random LUG member's Star Citizen referral code
+  -t, --target=[live|ptu]   Target LIVE or PTU (default live)
- -r,  --reset-helper        Delete saved lug-helper configs
+  -g, --use-gui=[yes|no]    Use Zenity GUI (default yes)
+  -r, --get-referral        Get a random LUG member's Star Citizen referral code
+  -x, --reset-helper        Delete saved lug-helper configs
 "
                 exit 0
                 ;;
             --preflight-check | -p )
-                preflight_check
+                cargs+=("preflight_check")
                 ;;
             --manage-runners | -m )
-                runner_manage
+                cargs+=("runner_manage")
                 ;;
-            --sanitize | -s)
+            --delete-user-folder | -u )
-                sanitize
+                cargs+=("sanitize")
                 ;;
-            --delete-shaders | -d)
+            --delete-shaders | -s )
-                rm_shaders
+                cargs+=("rm_shaders")
                 ;;
             --delete-dxvk-cache | -c )
-                rm_dxvkcache
+                cargs+=("rm_dxvkcache")
                 ;;
-            --get-referral | -g)
+            --target=* | -t=* )
-                referral_randomizer
+                live_or_ptu="$(echo "$1" | cut -d'=' -f2)"
+                if [ "$live_or_ptu" = "live" ] || [ "$live_or_ptu" = "LIVE" ]; then
+                    live_or_ptu="LIVE"
+                elif [ "$live_or_ptu" = "ptu" ] || [ "$live_or_ptu" = "PTU" ]; then
+                    live_or_ptu="PTU"
+                else
+                    printf "$0: Invalid option '$1'\n"
+                    exit 0
+                fi
                 ;;
-            --reset-helper | -r)
+            --use-gui=* | -g=* )
-                reset_helper
+                # If zenity is unavailable, it has already been set to 0
+                # and this setting has no effect
+                if [ -x "$(command -v zenity)" ]; then
+                    has_zen="$(echo "$1" | cut -d'=' -f2)"
+                    if [ "$has_zen" = "yes" ] || [ "$has_zen" = "YES" ]; then
+                        has_zen=1
+                    elif [ "$has_zen" = "no" ] || [ "$has_zen" = "NO" ]; then
+                        has_zen=0
+                    else
+                        printf "$0: Invalid option '$1'\n"
+                        exit 0
+                    fi
+                fi
+                ;;
+            --get-referral | -r )
+                cargs+=("referral_randomizer")
+                ;;
+            --reset-helper | -x )
+                cargs+=("reset_helper")
                 ;;
             * )
                 printf "$0: Invalid option '$1'\n"
-                printf  "For more information try \x1B[32m--help\n"
                 exit 0
                 ;;
         esac
@@ -1256,8 +1281,15 @@ Usage: lug-helper <options>
         shift
     done
 
+    # Call the requested functions and exit
+    if [ "${#cargs[@]}" -gt 0 ]; then
+        for (( i=0; i<"${#cargs[@]}"; i++ )); do
+            ${cargs[i]}
+        done
+
         exit 0
     fi
+fi
 
 # Loop the main menu until the user selects quit
 while true; do