improve handling of environment groups

1 files changed, 10 insertions, 6 deletions

diff --git a/upgrade.sh b/upgrade.sh
index 48849b82..e44379c5 100755
--- a/upgrade.sh
+++ b/upgrade.sh
@@ -1,13 +1,17 @@
 #!/bin/bash
 
-if [ -z "$1" ] || [ -z "$2" ]; then
-  echo "$0 <environment> <host(s)>"
+if [ -z "$1" ]; then
+  echo "$0 <host(s)>"
   exit 1
 fi
-env="$1"
-shift
 hosts="$1"
 shift
 
-echo "######## upgrading host(s) '$hosts' in environment '$env' ########"
-exec ansible-playbook -e "myhosts=$hosts" -e "myrole=upgrade" $@ "$env/generic.yaml"
+source "${BASH_SOURCE%/*}/common/utils.sh"
+get_ansible_variable env_group "$hosts"
+if [ $? -ne 0 ]; then
+  exit 1
+fi
+
+echo "######## upgrading host(s) '$hosts' in environment '$env_group' ########"
+exec ansible-playbook -e "myhosts=$hosts" -e "myrole=upgrade" $@ "$env_group/generic.yaml"