isislab / spod-manager

  #!/bin/bash
  
  # The MIT License (MIT)
  #

  # Copyright (c) 2015 ISISLab, Università degli Studi di Salerno

  #
  # Permission is hereby granted, free of charge, to any person obtaining a copy
  # of this software and associated documentation files (the "Software"), to deal
  # in the Software without restriction, including without limitation the rights
  # to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  # copies of the Software, and to permit persons to whom the Software is
  # furnished to do so, subject to the following conditions:
  #
  # The above copyright notice and this permission notice shall be included in
  # all copies or substantial portions of the Software.
  #
  # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  # IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  # FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  # AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  # LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  # THE SOFTWARE.
  
  #

  # FUNCTIONS DEFINITIONS
  #
  
  # host_exec_cmd
  #
  # Description:
  #     Calls a command on a host, abstracting the locality of the host.
  #
  # Parameters:
  #     $1 hostname where to perform the command
  #     $2 the command to execute
  function host_exec_cmd {
  	if [ "$MYHOST" == "$1" ]; then
  		"$2"
  	else
  		ssh -t ${SSH_USER}${1}.routetopa.eu "$2"
  	fi
  }
  
  # host_send_file
  #
  # Description:
  #     Send a local file to remote host. If the copy is issued on the local host
  #     itself, the command is ignored.
  #
  # Parameters:
  #     $1 hostname to copy the file to
  #     $2 local filename
  #     $3 remote localhost
  function host_send_file {
  	if [ "$MYHOST" == "$1" ]; then

  		cp ${2} ${3}

  	else

  		scp ${2} ${SSH_USER}${1}.routetopa.eu:${3}

  	fi
  }
  
  # host_recv_file
  #
  # Description:
  #     Receive a remote file to localhost host. If the copy is issued on the local host
  #     itself, the command is ignored.
  #
  # Parameters:
  #     $1 hostname to copy the file from
  #     $2 remote filename
  #     $3 local filename
  function host_recv_file {
  	if [ "$MYHOST" == "$1" ]; then

  		cp ${2} ${3}

  	else
  		scp ${SSH_USER}${1}.routetopa.eu:${2} ./${3}
  	fi
  }
  
  #

  # DEFAULT OPTIONS
  #
  
  SSH_USER=
  SSH_PASS=
  
  # MYSQL_HOST=localhost # IGNORED
  MYSQL_USER=root
  MYSQL_PASS=is15rdc
  MYSQL_DB=oxwall
  
  HOSTLIST=(

  	'spod'

  	'prato'
  	'dublin'
  	'issy'
  	'denhaag'
  	'groningen'
  )
  
  #
  # OPTION PROCESSING
  #
  

  OPTIONS=$(getopt -o s:h: -- $@)

  eval set -- $OPTIONS
  
  while true; do
  	case "$1" in

  		-h)
  			IFS=',' read -a HOSTLIST <<< "$2"
  			shift 2
  			;;

  		-s)
  			SSH_USER="${2}@"
  			shift 2
  			;;
  		--)
  			shift
  			break
  			;;
  		*)
  			echo "Unknown option: $1"
  			shift
  			break
  			;;
  	esac
  done
  
  #
  # VARIABLES
  #
  
  ALLOWEDHOST="spod.routetopa.eu"
  MYHOST=$(hostname)
  
  #if [ "$MYHOST" != "$ALLOWEDHOST" ]; then
  	#echo "This script can be run only on ${ALLOWEDHOST}. It appears you are running it on ${MYHOST}"
  	#echo "Exiting..."
  	#exit -1;
  #fi
  
  #
  # PERFORM SELECTED ACTION
  #
  ACTION=$1
  case "$ACTION" in

  	cmd)
  		shift

  		ARG_COMMAND=$@

  		if [ -z "$ARG_COMMAND" ]; then

  			echo "ERROR: Please specify a command: $0 $ACTION touch last_access"

  			exit 1
  		fi

  		for HOST in ${HOSTLIST[@]}; do
  			host_exec_cmd ${HOST} "${ARG_COMMAND}"
  		done
  		;;
  

  	db-backup)

  		ARG_BACKUP_NAME=$2
  		if [ -z $ARG_BACKUP_NAME ]; then

  			ARG_BACKUP_NAME="${MYSQL_DB}-$(date +%Y%m%d-%H%M%S).sql"
  			echo "No given backup filename. Using $ARG_BACKUP_NAME"

  		fi
  		for HOST in ${HOSTLIST[@]}; do

  			host_exec_cmd ${HOST} "mysqldump -u${MYSQL_USER} -p${MYSQL_PASS} ${MYSQL_DB} > ${ARG_BACKUP_NAME}"
  			host_recv_file ${HOST} ${ARG_BACKUP_NAME} ${HOST}-${ARG_BACKUP_NAME}

  		done

  		;;

  	db-restore)

  		ARG_BACKUP_NAME=$2

  		if [ -z ${ARG_BACKUP_NAME} ]; then

  			echo "ERROR: Please specify the name of the backup. e.g.: $0 $ACTION backup.sql"

  			exit 1

  		fi
  		for HOST in ${HOSTLIST[@]}; do

  			if [ ! -f "${HOST}-${ARG_BACKUP_NAME}" ]; then
  				echo "ERROR: File ${HOST}-${ARG_BACKUP_NAME} does not exists"
  				exit 1
  			fi
  		done
  		for HOST in ${HOSTLIST[@]}; do
  			host_send_file ${HOST} "${HOST}-${ARG_BACKUP_NAME}" "${ARG_BACKUP_NAME}"
  			host_exec_cmd ${HOST} "mysql -u${MYSQL_USER} -p${MYSQL_PASS} ${MYSQL_DB} < ${ARG_BACKUP_NAME}"

  		done
  		;;

  	db-sql)

  		ARG_SCRIPT_NAME=$2
  		if [ -z $ARG_SCRIPT_NAME ]; then

  			echo "ERROR: Please specify the name of the sql. e.g.: $0 $ACTION do_something.sql"

  			exit 1
  		fi
  		if [ ! -f $ARG_SCRIPT_NAME ]; then
  			echo "ERROR: File $ARG_SCRIPT_NAME does not exists"
  			exit 1
  		fi
  		for HOST in ${HOSTLIST[@]}; do

  			host_send_file ${HOST} ${ARG_SCRIPT_NAME} ${HOST}-${ARG_SCRIPT_NAME}
  			host_exec_cmd ${HOST} "mysql -u${MYSQL_USER} -p${MYSQL_PASS} ${MYSQL_DB} < ${HOST}-${ARG_SCRIPT_NAME}"

  		done
  		;;
  

  	db-transfer)
  		ARG_TABLES=$2
  		ARG_HOST_FROM=$3
  		IFS=',' read -a ARG_HOST_TO <<< "$4"
  
  		if [ -z ARG_TABLES ] || [ -z $ARG_HOST_FROM ]; then
  			echo "ERROR: Please specify at least the table(s) and the source host."
  			echo "Examples"
  			echo "    $0 $ACTION table sourcehost"
  			echo "    $0 $ACTION table1,table2 sourcehost destinationhost"
  			echo "    $0 $ACTION table sourcehost destinationhost1,destinationhost2,destinationhost3"
  			exit 1
  		fi
  

  		if [ -z "$ARG_HOST_TO" ]; then
  			ARG_HOST_TO="${HOSTLIST[@]}"
  		fi
  

  		TABLES=$(echo "$ARG_TABLES" | tr , " ")
  		ARG_BACKUP_NAME="TEMP-$(date +%Y%m%d-%H%M%S).sql"
  		host_exec_cmd ${ARG_HOST_FROM} "mysqldump -u${MYSQL_USER} -p${MYSQL_PASS} --add-drop-table ${MYSQL_DB} ${TABLES} > ${ARG_BACKUP_NAME}"
  		host_recv_file ${ARG_HOST_FROM} ${ARG_BACKUP_NAME} LOCAL-${ARG_BACKUP_NAME}
  		host_exec_cmd ${ARG_HOST_FROM} "rm ${ARG_BACKUP_NAME}"
  		for HOST_TO in ${ARG_HOST_TO[@]}; do

  			if [ "$ARG_HOST_FROM" == "$HOST_TO" ]; then
  				continue
  			fi

  			host_send_file ${HOST_TO} LOCAL-${ARG_BACKUP_NAME} ${HOST_TO}-${ARG_BACKUP_NAME}
  			host_exec_cmd ${HOST_TO} "mysql -u${MYSQL_USER} -p${MYSQL_PASS} ${MYSQL_DB} < ${HOST_TO}-${ARG_BACKUP_NAME}; rm ${HOST_TO}-${ARG_BACKUP_NAME}"
  		done
  		rm LOCAL-${ARG_BACKUP_NAME}
  		;;
  

  	lang-add)
  		ARG_PREFIX=$2
  		ARG_KEY=$3
  		if [ -z $ARG_KEY ]; then
  			echo "ERROR: Please specify language prefix and key. e.g.: $0 $ACTION ode add_button"
  			exit 1
  		fi
  		SCRIPT_NAME="TEMP-$(date +%Y%m%d-%H%M%S).sql"
  		echo "INSERT INTO ow_base_language_prefix(prefix, label) SELECT '$ARG_PREFIX', '$ARG_PREFIX' FROM DUAL WHERE NOT EXISTS (SELECT id FROM ow_base_language_prefix WHERE prefix LIKE '$ARG_PREFIX' LIMIT 1);" > $SCRIPT_NAME
  		echo "SET @prefix_id = (SELECT id FROM ow_base_language_prefix WHERE prefix LIKE '$ARG_PREFIX' LIMIT 1);" >> $SCRIPT_NAME
  		echo "INSERT INTO ow_base_language_key(prefixId, \`key\`) VALUES(@prefix_id, '$ARG_KEY');" >> $SCRIPT_NAME
  		for HOST in ${HOSTLIST[@]}; do

  			host_send_file ${HOST} ${SCRIPT_NAME} ${HOST}-${SCRIPT_NAME}
  			host_exec_cmd ${HOST} "mysql -u${MYSQL_USER} -p${MYSQL_PASS} ${MYSQL_DB} < ${HOST}-${SCRIPT_NAME}; rm ${HOST}-${SCRIPT_NAME}"

  		done
  		;;
  

  	script)
  		ARG_SCRIPT_NAME=$2
  		if [ -z $ARG_SCRIPT_NAME ]; then
  			echo "ERROR: Please specify the name of the script. e.g.: $0 $ACTION do_something.sh"
  			exit 1
  		fi
  		if [ ! -f $ARG_SCRIPT_NAME ]; then
  			echo "ERROR: File $ARG_SCRIPT_NAME does not exists"
  			exit 1
  		fi
  		for HOST in ${HOSTLIST[@]}; do

  			host_send_file ${HOST} ${ARG_SCRIPT_NAME} ${HOST}-${ARG_SCRIPT_NAME}
  			host_exec_cmd ${HOST} "source ${HOST}-$ARG_SCRIPT_NAME; rm ${HOST}-$ARG_SCRIPT_NAME"

  		done
  		;;

  	update)

  		echo "Updating production sites..."
  		for HOST in ${HOSTLIST[@]}; do

  			host_exec_cmd ${HOST} "sudo ./spod-manager.sh -a update -p isisadmin -q isislab.unisa@gmail.com -r is15rdc1 -w http://${HOST}.routetopa.eu/"

  		done
  		;;

  	""|help)
  		echo "ROUTE-TO-PA PROJECT: CLUSTER-MANAGER.SH"
  		echo " "

  		echo "Usage: $0 <action> [options]"

  		echo " "
  		echo "List of actions:"
  		echo "    help                  : Show this screen"

  		echo "    cmd <command>         : Execute <command> on hosts"

  		echo "    db-backup [filename]  : Perform a database backup"

  		echo "    db-restore <filename> : Restore a database backup"

  		echo "    db-transfer table1,table2 srchost dsthost1,dsthost2"
  		echo "                          : Copy table(s) from srchost to desthost(s)"

  		echo "    db-sql <filename>     : Copy <filename> to hosts and run with mysql client"

  		echo "    lang-add <prefix> <key>"
  		echo "                          : Add a language string"

  		echo "    script <filename>     : Copy <filename> to hosts and execute it"

  		echo "    update                : Update code using git pull"
  		echo " "
  		echo "List of options:"

  		echo "    -s <ssh_login>"

  		echo "    -h <hostname,hostname,...>"

  		;;

  	*)
  		echo "Unknown action: \"$ACTION\". Try calling this script without arguments to get help."
  		;;
  esac