From 673503204bc96f17016bd3b2117b797fb9c3b366 Mon Sep 17 00:00:00 2001
From: root <root@gw-spr.sprachenatelier.netz>
Date: Sun, 18 Mar 2018 13:58:58 +0100
Subject: [PATCH] Add script 'check_service.sh'.

---
 check_service.sh | 221 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 221 insertions(+)
 create mode 100755 check_service.sh

diff --git a/check_service.sh b/check_service.sh
new file mode 100755
index 0000000..c9ede3e
--- /dev/null
+++ b/check_service.sh
@@ -0,0 +1,221 @@
+#!/usr/bin/env bash
+
+
+#---------------------------------------
+#-----------------------------
+# Base Function(s)
+#-----------------------------
+#---------------------------------------
+
+fatal(){
+   echo ""
+   if $terminal ; then
+      echo -e "   [ \033[31m\033[1mFatal\033[m ] $*"
+   else
+      echo -e "   [ Fatal ] $*"
+   fi
+   echo ""
+   if $terminal ; then
+      echo -e "             \033[1mScript terminated\033[m.."
+   else
+      echo -e "             Script terminated.."
+   fi
+   echo ""
+   rm -rf $LOCK_DIR
+   exit 1
+}
+
+error (){
+   echo ""
+   if $terminal ; then
+      echo -e "   [ \033[31m\033[1mError\033[m ] $*"
+   else
+      echo "   [ Error ] $*"
+   fi
+   echo ""
+}
+
+warn (){
+   echo ""
+   if $terminal ; then
+      echo -e "   [ \033[33m\033[1mWarn\033[m ]  $*"
+   else
+      echo "   [ Warn ]  $*"
+   fi
+   echo ""
+}
+
+info (){
+   echo ""
+   if $terminal ; then
+      echo -e "   [ \033[32m\033[1mInfo\033[m ]  $*"
+   else
+      echo "   [ Info ]  $*"
+   fi
+   echo ""
+}
+
+ok (){
+   echo ""
+   if $terminal ; then
+      echo -e "   [ \033[32m\033[1mOk\033[m ]    $*"
+   else
+      echo "   [ Ok ]    $*"
+   fi
+   echo ""
+}
+
+trim() {
+    local var="$*"
+    var="${var#"${var%%[![:space:]]*}"}"   # remove leading whitespace characters
+    var="${var%"${var##*[![:space:]]}"}"   # remove trailing whitespace characters
+    echo -n "$var"
+}
+
+
+#---------------------------------------
+#-----------------------------
+# Check some prerequisites
+#-----------------------------
+#---------------------------------------
+
+LOCK_DIR=`mktemp -d`
+
+# - Running in a terminal?
+# -
+if [[ -t 1 ]] ; then
+   terminal=true
+   LOGGING=true
+else
+   terminal=false
+   LOGGING=false
+fi
+
+if [[ -n "$1" ]] ; then
+   service_name=$1
+else
+   fatal "$(basename $0): No Service given"
+fi
+
+# - Is Service installed ?
+# -
+_check_binary="$(which ${service_name,,}d)"
+if [[ -z "$_check_binary" ]] ; then
+   _check_binary="$(which ${service_name,,})"
+fi
+
+
+if [[ -z "$_check_binary" ]]; then
+   fatal "$(basename $0): $service_name Service seems NOT to be installed"
+else
+   check_binary="$_check_binary"
+   check_string_ps="$check_binary"
+fi
+
+
+# - Systemd supported ?
+# -
+systemd=$(which systemd)
+systemctl=$(which systemctl)
+
+systemd_supported=false
+if [[ -n "$systemd" ]] && [[ -n "$systemctl" ]] ; then
+   systemd_supported=true
+fi
+
+SYSTEMD_SERVICE=
+SYSV_INIT_SCRIPT=
+
+if $systemd_supported ; then
+   if systemctl -t service list-unit-files \
+         | grep -e "^${service_name,,}d" \
+         | grep -q -E "(enabled|disabled)" 2> /devnull ; then
+
+      SYSTEMD_SERVICE="$(systemctl -t service list-unit-files | grep -e "^${service_name,,}d" | awk '{print$1}' | head -1)"
+   elif systemctl -t service list-unit-files \
+         | grep -e "^${service_name,,}" \
+         | grep -q -E "(enabled|disabled)" 2> /devnull ; then
+
+      SYSTEMD_SERVICE="$(systemctl -t service list-unit-files | grep -e "^${service_name,,}" | awk '{print$1}' | head -1)"
+   fi
+fi
+
+if [[ -z "$SYSTEMD_SERVICE" ]]; then
+   if [[ -x "/etc/init.d/${service_name,,}" ]]; then
+      SYSV_INIT_SCRIPT="/etc/init.d/${service_name,,}"
+   elif [[ -x "/etc/init.d/${service_name,,}d" ]]; then
+      SYSV_INIT_SCRIPT="/etc/init.d/${service_name,,}d"
+   fi
+fi
+
+if [[ -z "$SYSTEMD_SERVICE" ]] && [[ -z "$SYSV_INIT_SCRIPT" ]] ; then
+   fatal "Neither an init-script nor a service file for $service_name found!"
+fi
+
+
+#---------------------------------------
+#-----------------------------
+# Check if SSH service is running
+#-----------------------------
+#---------------------------------------
+
+if $LOGGING ; then
+   echo -e "\n   Check if $service_name service is running.."
+   echo -e "   ================================="
+fi
+if ! ps -e f | grep -E "[[:digit:]]\ ${check_string_ps}" | grep -v grep > /dev/null ; then
+   error "$service_name service seems to be down! Trying to restart service now.."
+
+   if [[ -n "$SYSTEMD_SERVICE" ]] ; then
+      $systemctl daemon-reload > /dev/null 2> ${LOCK_DIR}/err_msg.log
+      if [[ $? -ne 0 ]]; then
+         error "$(cat ${LOCK_DIR}/err_msg.log)"
+      fi
+      sleep 2
+      $systemctl stop $SYSTEMD_SERVICE > /dev/null 2> ${LOCK_DIR}/err_msg.log
+      if [[ $? -ne 0 ]]; then
+         error "$(cat ${LOCK_DIR}/err_msg.log)"
+      fi
+      sleep 10
+      $systemctl start $SYSTEMD_SERVICE > /dev/null 2> ${LOCK_DIR}/err_msg.log
+      if [[ $? -ne 0 ]]; then
+         error "$(cat ${LOCK_DIR}/err_msg.log)"
+      fi
+   else
+      $SYSV_INIT_SCRIPT stop > /dev/null 2>&1
+      if [[ $? -ne 0 ]]; then
+         error "Stopping $service_name service failed!"
+      fi
+      sleep 10
+      $SYSV_INIT_SCRIPT start > /dev/null 2>&1
+      if [[ $? -ne 0 ]]; then
+         error "Starting $service_name service failed!"
+      fi
+   fi
+
+   declare -i counter=0
+   PID=$(ps -e f | grep -E "[[:digit:]]\ ${check_string_ps}"| grep -v grep | awk '{print$2}')
+   while [[ "X${PID}" = "X" ]]; do
+      sleep 1
+      PID=$(ps -e f | grep -E "[[:digit:]]\ ${check_string_ps}"| grep -v grep | awk '{print$2}')
+      if [[ $counter -gt 10 ]]; then
+         break
+      else 
+         ((counter++))
+      fi
+   done
+
+   if [[ "X${PID}" = "X" ]] ; then
+      error "Restarting $service_name service failed!"
+   else
+      ok "$service_name service is up and running."
+   fi
+         
+else
+   if $LOGGING ; then
+      ok "$service_name service is up and running."
+   fi
+fi
+
+rm -rf $LOCK_DIR
+exit 0