#!/bin/sh

### BEGIN INIT INFO
# Provides:          vmon
# Required-Start:    $local_fs $network mosquitto
# Required-Stop:     $local_fs $network $syslog
# Default-Start:     2 3 4 5
# Default-Stop:      0 1 6
# Short-Description: vmon
# Description:       startup and shutdown a vmon service
### END INIT INFO

NAME="vmon"
PATH="/usr/local/sbin:/usr/local/bin:/sbin:/bin:/usr/sbin:/usr/bin"
APPDIR="/opt/inspark"
APPBIN="vmon"
APPARGS=""
USER="inspark"
GROUP="inspark"
MQTT_INIT="/etc/init.d/mosquitto"
MQTT_CONF="/etc/mosquitto/mosquitto.conf"
umask 002


# Include functions 
set -e
. /lib/lsb/init-functions


check_mosquitto_permissions() {
    if [ -d /var/run/mosquitto ]; then
        chmod -R 775 /var/run/mosquitto&& chown -R root:inspark /var/run/mosquitto
    else
        install -d -m 775 -o mosquitto -g mosquitto /var/run/mosquitto
    fi

    if [ -d /var/log/mosquitto ]; then
        chmod -R 770 /var/log/mosquitto&& chown -R root:inspark /var/log/mosquitto
    else
        install -d -m 660 -o root -g inspark /var/log/mosquitto
    fi
}


modify_mosquitto_init() {
cp $1 $1.backup
sed -i '/PIDFILE=/i#### INSPARK modifications/'  "$1"
sed -i '/PIDFILE/s/=.*/=\/var\/run\/mosquitto\/mosquitto.pid/' "$1"
sed -i '/CONFIG=/aUSER="inspark"\nGROUP="inspark"' "$1"
sed -i 's/--quiet/--chuid "$USER:$GROUP"/' "$1"
}

modify_mosquitto_conf() {
cp $1 $1.backup
sed -i '/pid_file/i#### INSPARK modifications/'  "$1"
sed -i '/pid_file/s/ .*/ \/var\/run\/mosquitto\/mosquitto.pid/' "$1"
}




start() {
  if init_is_upstart; then
    exit 1
  fi	
  log_daemon_msg "Starting '$NAME'... "
  start-stop-daemon --start --chuid "$USER:$GROUP" --quiet --oknodo --background --make-pidfile --pidfile /var/run/$NAME.pid --chdir "$APPDIR" --exec "$APPBIN" -- $APPARGS  
  status=$?
  log_end_msg $status
}

#We need this function to ensure the whole process tree will be killed
killtree() {
    local _pid=$1
    local _sig=${2-TERM}
    for _child in $(ps -o pid --no-headers --ppid ${_pid}); do
        killtree ${_child} ${_sig}
    done
    kill -${_sig} ${_pid}
}

stop() {
  if init_is_upstart; then
    exit 0
  fi
  log_daemon_msg "Stopping '$NAME'... "
  [ -z `cat /var/run/$NAME.pid 2>/dev/null` ] || \
  while test -d /proc/$(cat /var/run/$NAME.pid); do
    killtree $(cat /var/run/$NAME.pid) 15
    sleep 0.5
  done 
  [ -z `cat /var/run/$NAME.pid 2>/dev/null` ] || rm /var/run/$NAME.pid
  status=$?
  log_end_msg $status
}

status() {
  if init_is_upstart; then
    exit 1
  fi
  status_of_proc -p /var/run/$NAME.pid "" $NAME && exit 0 || exit $?
}

case "$1" in
  start)
    start
    ;;
  stop)
    stop
    ;;
  restart)
    stop
    start
    ;;
  status)
    status
    ;;
  *)
    echo "Usage: $NAME {start|stop|restart|status}" >&2
    exit 1
    ;;
esac

exit 0
