#!/bin/bash

set -o nounset

# Make sure umask is sane
umask 022

# Check for uninitialized variables
NOUNSET=${NOUNSET:-}
if [[ -n "$NOUNSET" ]]; then
        set -o nounset
fi

# TBD: Defaults to $HOME, otherwise if we want to address:
# pid path could be /etc/pid_path
# monit_script could be 
# log_path /var/log/monit/
# exec path(for monitrc) could be /etc/monitrc, /usr/local/etc/monitrc

# Check if directory for monit logs exists
if [ ! -d "$HOME/.bigchaindb-monit" ]; then
  mkdir -p "$HOME/.bigchaindb-monit"
fi

monit_pid_path=${MONIT_PID_PATH:=$HOME/.bigchaindb-monit/monit_processes}
monit_script_path=${MONIT_SCRIPT_PATH:=$HOME/.bigchaindb-monit/monit_script}
monit_log_path=${MONIT_LOG_PATH:=$HOME/.bigchaindb-monit/logs}
monit_exec_path=${MONIT_EXEC_PATH:=$HOME}

function usage() {
        cat <<EOM

    Usage: ${0##*/} [-h]

    Configure Monit for BigchainDB and Tendermint process management.

    ENV[MONIT_PID_PATH] || --monit-pid-path PATH

    Absolute path to directory where the the program's pid-file will reside. 
    The pid-file contains the ID(s) of the process(es).

    ENV[MONIT_SCRIPT_PATH] || --monit-script-path PATH

    Absolute path to the directory where the executable program or
    script is present.

    ENV[MONIT_LOG_PATH] || --monit-log-path PATH

    Absolute path to the directory where all the logs for processes
    monitored by Monit are stored.

    ENV[MONIT_EXEC_PATH] || --monit-exec-path PATH

    Absolute path to the directory to run the script form.

    -h|--help
        Show this help and exit.

EOM
}

while [[ $# -gt 0 ]]; do
    arg="$1"
    case $arg in
        --monit-pid-path)
            monit_pid_path="$2"
            shift
        ;;
        --monit-script-path)
            monit_script_path="$2"
            shift
        ;;
        --monit-log-path)
            monit_log_path="$2"
            shift
        ;;
        --monit-exec-path)
            monit_exec_path="$2"
            shift
        ;;
        -h|--help)
            usage
            exit
        ;;
        *)
            echo "Unknown option: $1"
            usage
            exit 1
        ;;
    esac
    shift
done

# Check if directory for monit logs exists
if [ ! -d "$monit_log_path" ]; then
  mkdir -p "$monit_log_path"
fi

# Check if directory for monit pid files exists
if [ ! -d "$monit_pid_path" ]; then
  mkdir -p "$monit_pid_path"
fi

cat >${monit_script_path} <<EOF
#!/bin/bash
case \$1 in

  start_bigchaindb)

    pushd \$4
      nohup bigchaindb start >> \$3/bigchaindb.out.log 2>> \$3/bigchaindb.err.log &

      echo \$! > \$2
    popd

    ;;

  stop_bigchaindb)

    kill -2 \`cat \$2\`
    rm -f \$2

    ;;

  start_tendermint)

    pushd \$4
      nohup tendermint node --consensus.create_empty_blocks=false >> \$3/tendermint.out.log 2>> \$3/tendermint.err.log &

      echo \$! > \$2
    popd

    ;;

  stop_tendermint)

    kill -2 \`cat \$2\`
    rm -f \$2

    ;;

esac
exit 0
EOF
chmod +x ${monit_script_path}

# configure monitrc
cat >${monit_exec_path}/.monitrc <<EOF
set httpd
  port 2812
  allow localhost

check process bigchaindb
    with pidfile ${monit_pid_path}/bigchaindb.pid
    start program "${monit_script_path} start_bigchaindb $monit_pid_path/bigchaindb.pid ${monit_log_path} ${monit_exec_path}"
    restart program "${monit_script_path} start_bigchaindb $monit_pid_path/bigchaindb.pid ${monit_log_path} ${monit_exec_path}"
    stop program "${monit_script_path} stop_bigchaindb $monit_pid_path/bigchaindb.pid ${monit_log_path} ${monit_exec_path}"

check process tendermint
    with pidfile ${monit_pid_path}/tendermint.pid
    start program "${monit_script_path} start_tendermint ${monit_pid_path}/tendermint.pid ${monit_log_path} ${monit_exec_path}"
    restart program "${monit_script_path} start_bigchaindb ${monit_pid_path}/bigchaindb.pid ${monit_log_path} ${monit_exec_path}"
    stop program "${monit_script_path} stop_tendermint ${monit_pid_path}/tendermint.pid ${monit_log_path} ${monit_exec_path}"
    depends on bigchaindb
EOF

# Setting permissions for control file
chmod 0700 ${monit_exec_path}/.monitrc

# Kill background processes on exit
trap exit_trap EXIT
function exit_trap {
    exit $?
}
# Exit on any errors so that errors don't compound and kill if any services already started

echo -e "BigchainDB process manager configured!"
set -o errexit