From 2e493e086f73337b6f3ba59d5af3b613ec882e28 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Gardais=20J=C3=A9r=C3=A9my?=
 <jeremy.gardais@univ-rennes1.fr>
Date: Mon, 22 Jun 2020 13:48:26 +0200
Subject: [PATCH] Add options management

---
 cluster/sge.enable.host.queue.sh | 103 +++++++++++++++++++++++--------
 1 file changed, 76 insertions(+), 27 deletions(-)

diff --git a/cluster/sge.enable.host.queue.sh b/cluster/sge.enable.host.queue.sh
index d27bfca..30c140c 100755
--- a/cluster/sge.enable.host.queue.sh
+++ b/cluster/sge.enable.host.queue.sh
@@ -6,7 +6,7 @@ readonly PROGDIR=$(readlink -m $(dirname "${0}"))
 readonly ARGS="${*}"
 readonly NBARGS="${#}"
 ## Test if DEBUG is already defined (by parent script,…)
-[ -z "${DEBUG}" ] && readonly DEBUG=0
+[ -z "${DEBUG}" ] && DEBUG=1
 
 # Maco temp file
 readonly MACO_TMP_FILE="/tmp/.maco.upgrade"
@@ -15,6 +15,9 @@ readonly MACO_TMP_URGENT_FILE="/tmp/.maco.urgent.upgrade"
 # APT temp file to monitor
 readonly APT_TMP_FILE="/tmp/.apt.upgrade"
 
+# If the scrip need to check pending upgrade before enable a queue
+CHECK_UPGRADE="0"
+
 ## Colors
 readonly PURPLE='\033[1;35m'
 readonly RED='\033[0;31m'
@@ -22,33 +25,10 @@ readonly RESET='\033[0m'
 readonly COLOR_DEBUG="${PURPLE}"
 # }}}
 
-manage_args() {                                                 # {{{
-
-	case "${NBARGS}" in
-		0 )
-			sge_hostname="$(hostname -f)"
-			## Ensure there is no pending upgrades on the local host
-			CHECK_UPGRADE="0"
-			;;
-		1 )
-			sge_hostname="${1}"
-			## No way to monitor pending upgrades on a remote host
-			CHECK_UPGRADE="1"
-			;;
-		* )
-			printf '%b\n' "${RED}Expect 1 or 0 argument.${RESET}"
-			printf '%b\n' "---"
-			usage
-			exit 1
-			;;
-	esac
-
-}
-# }}}
 usage() {                                                       # {{{
 
 	cat <<- EOF
-usage: $PROGNAME [hostname]
+usage: $PROGNAME [--help|-d,--debug] [hostname]
 
 Try to enable all SGE queues of the current host (default)
 or the one passed as first argument.
@@ -238,8 +218,6 @@ Try to enable SGE queue: ${RED}${local_sge_queue_name:=/dev/null}@${local_sge_ho
 # }}}
 main() {                                                        # {{{
 
-	manage_args "${ARGS}"
-
 	sge_queues_name="$(qhost -h "${sge_hostname:=/dev/null}" -q -xml \
 		| grep "queue name" \
 		| cut -d"'" -f2 )"
@@ -289,6 +267,77 @@ for host: ${RED}${sge_hostname:=/dev/null}${COLOR_DEBUG}."
 }
 # }}}
 
+# Manage arguments                                                # {{{
+# This code can't be in a function due to arguments
+
+if [ ! "${NBARGS}" -eq "0" ]; then
+
+	manage_arg="0"
+
+	## If the first argument is not an option
+	if ! printf -- '%s' "${1}" | grep -q -E -- "^-*";
+	then
+		## Use this argument for sge_hostname
+		sge_hostname="${1}"
+
+		## No way to monitor pending upgrades on a remote host
+		CHECK_UPGRADE="1"
+
+		## Switch to next arg
+		shift
+		manage_arg=$((manage_arg+1))
+	else
+		## Use local host for sge_hostname
+		sge_hostname="$(hostname -f)"
+	fi
+
+	# Parse all options (start with a "-") one by one
+	while printf -- '%s' "${1}" | grep -q -E -- "^-*"; do
+
+	case "${1}" in
+		--help )        ## help
+			usage
+			## Exit after help informations
+			exit 0
+			;;
+		-d|--debug )    ## debug
+			DEBUG=0
+			## Re-export new DEBUG value
+			export DEBUG
+			;;
+		-- )            ## End of options list
+			## End the while loop
+			break
+			;;
+		* )             ## unknow option
+			printf '%b\n' "${RED}Invalid option: ${1}${RESET}"
+			printf '%b\n' "---"
+			usage
+			exit 1
+			;;
+	esac
+
+	debug_message "Arguments management − \
+${RED}${1}${COLOR_DEBUG} option managed."
+
+	## Next arg
+	shift
+	manage_arg=$((manage_arg+1))
+
+	done
+
+	debug_message "Arguments management − \
+${RED}${manage_arg}${COLOR_DEBUG} argument(s) successfully managed."
+else
+	debug_message "Arguments management − \
+No arguments/options to manage."
+
+	## Use local host for sge_hostname
+	sge_hostname="$(hostname -f)"
+fi
+
+# }}}
+
 main
 
 exit 0