#!/bin/bash
# File	: cdot_set_stor-class.sh
# By	: Maarten.deBoer@atos.net, 141222
# Subject	: Script to set (a different) storage class on a volume by using ETCFILE
#set -x
PGM="`basename $0|cut -d\. -f1`"
VER="0.1"
TMP="/tmp/${PGM}.$$"
ASC="/tmp/${PGM}.asc"
LOG="${HOME}/log/${PGM}.log"
SSH="/usr/bin/ssh -n"
HOSTNAME="`hostname|cut -d\. -f1`"
CFILTER="[?]*"
SFILTER="[?]*"
ETCFILE="${HOME}/etc/cdot_rpt_qos_settings.excl"

SSHCMD()
# 1: Filername 2:Command-string
# When issue with connection to cluster, try the nodes (-01 & -02)
# "There are no entries matching your query." => EC=255
# "no connection" is also EC=255
{
  TMPERR="/tmp/${PGM}.$$.err"
  /usr/bin/ssh -n ${1} "${2}" 2> ${TMPERR}
  EC=${?}
  # Check if "ssh: connect to host 10.192.109.202 port 22: Connection refused" If so (EC2=0), the 2nd
  grep 'Connection refused' ${TMPERR}
  EC2=${?}
  if [ ${EC} -ne 0 ] && [ ${EC2} -eq 0 ]; then
    sleep 2
    /usr/bin/ssh -n ${1}-01 "${2}" 2> ${TMPERR}
    EC=${?}
    grep 'Connection refused' ${TMPERR}
    EC2=${?}
    if [ ${EC} -ne 0 ] && [ ${EC2} -eq 0 ]; then
      sleep 2
      /usr/bin/ssh -n ${1}-02 "${2}" 2> ${TMPERR}
      EC=${?}
      grep 'Connection refused' ${TMPERR}
      EC2=${?}
      if [ ${EC} -ne 0 ] && [ ${EC2} -eq 0 ]; then
        sleep 2
        /usr/bin/ssh -n ${1}-03 "${2}" 2> ${TMPERR}
        EC=${?}
        grep 'Connection refused' ${TMPERR}
        EC2=${?}
        if [ ${EC} -ne 0 ] && [ ${EC2} -eq 0 ]; then
          echo "  EC=${EC} sleep 2 ..."
          sleep 2
          /usr/bin/ssh -n ${1}-04 "${2}" 2> ${TMPERR}
          EC=${?}
          grep 'Connection refused' ${TMPERR}
          EC2=${?}
          if [ ${EC} -ne 0 ] && [ ${EC2} -eq 0 ]; then
            echo  "`date` ${PGM} ERROR with communication to ${1}. Connection to -01 - -04 failed too."|tee -a ${LOG} 
          fi
        fi
      fi
    fi
  fi
  rm ${TMPERR}
}

USAGE()
{
  echo "Usage: ${PGM} [options]"
  echo "  Version: ${VER}"
  echo "  options       :"
  echo "    -e | --etc  : Etc-file (${ETCFILE})"
  echo "    -c          : Clustername Filter (${CFILTER})"
  echo "    -h | --help : this help"
  echo "    -v          : SVM filter (${SFILTER})"
  echo "    -V          : Version"
  echo "    -x          : set -x"
}

# Check options
while [ ${#} -gt 0 ]
  do
  case ${1} in
    -e | --etc) ETCFILE=${2}; shift ;;
    -h | --help) USAGE; exit 1 ;;
    -c | --cluster) CFILTER=${2}; shift ;;
    -s | --svm) SFILTER=${2}; shift ;;
    -V) echo "${PGM}: v${VER}"; exit 2 ;;
    -x)  set -x ;;
    *)  echo "Option ${1} not known."; USAGE; exit 1 ;;
  esac
    shift
done

# MAIN
echo "`date` ${PGM} v${VER} started."|tee -a ${LOG} ${TMP}
echo "CLUSTERS=${CLUSTERS}"
echo "CFILTER=${CFILTER}"
echo "ETCFILE=${ETCFILE}"
echo "SFILTER=${SFILTER}"
sleep 1

if [ ! -f ${ETCFILE} ]; then
  echo "  ETCFILE (${ETCFILE}) NOT found. Exiting ..."
  exit 3
fi

touch ${TMP}

cat ${ETCFILE}|grep -v ^#|while read LINE
do
  echo "${LINE} ..."
  CLUSTER=`echo ${LINE}|grep ${CFILTER}|awk -F\; '{print $1}'`
  VSERVER=`echo ${LINE}|grep ${SFILTER}|awk -F\; '{print $2}'`
  AGGR=`echo ${LINE}|grep ${SFILTER}|awk -F\; '{print $3}'`
  VOLUME=`echo ${LINE}|grep ${SFILTER}|awk -F\; '{print $4}'`
  STORCLASS=`echo ${LINE}|grep ${SFILTER}|awk -F\; '{print $5}'`
  NEWCLASS="`echo ${VSERVER}|sed 's/\-dr//g'`_${STORCLASS}"
  echo "  C:${CLUSTER} V:${VSERVER} A:${AGGR} V:${VOLUME} N:${NEWCLASS} "|tee -a ${LOG}

  if [ "${CLUSTER}" != "" ] && [ "${VSERVER}" != "" ] && [ ${AGGR} != "" ] && [ "${VOLUME}" != "" ] && [ "${NEWCLASS}" != "" ]; then
    SSHCMD ${CLUSTER} "volume modify -vserver ${VSERVER} -volume ${VOLUME} -qos-policy ${NEWCLASS}"
    SSHCMD ${CLUSTER} "volume show -vserver ${VSERVER} -volume ${VOLUME} -field vserver,volume,aggregate,qos-policy" | tee -a ${LOG}
  fi  

done  # LINE

# Finish
rm ${TMP} 
echo "`date` ${PGM} v${VER} finished."|tee -a ${LOG}
exit 0