#!/bin/sh
#
# K2HR3 REST API
#
# Copyright 2017 Yahoo Japan Corporation.
#
# K2HR3 is K2hdkc based Resource and Roles and policy Rules, gathers 
# common management information for the cloud.
# K2HR3 can dynamically manage information as "who", "what", "operate".
# These are stored as roles, resources, policies in K2hdkc, and the
# client system can dynamically read and modify these information.
#
# For the full copyright and license information, please view
# the license file that was distributed with this source code.
#
# AUTHOR:   Takeshi Nakatani
# CREATE:   Wed Jun 8 2017
# REVISION:
#

#==========================================================
# Common Variables
#==========================================================
PRGNAME=$(basename "$0")
SCRIPTDIR=$(dirname "$0")
SCRIPTDIR=$(cd "${SCRIPTDIR}" || exit 1; pwd)
SRCTOP=$(cd "${SCRIPTDIR}/.." || exit 1; pwd)

#
# Variables
#
CMDLINE_COMMAND="$0"
CMDLINE_PARAMETERS="$*"

LOCAL_HOSTNAME="$(hostname | tr -d '\n')"
PID_FILE_BASEDIR="/var/run/antpickax"
PID_FILE_TMPDIR="/tmp"

PRGNAME_PREFIX=$(echo "${PRGNAME}" | sed -e 's#\.sh.*$##g')
MAINPROC_PID_BASENAME="k2hr3pi-${PRGNAME_PREFIX}.pid"
WATCHPROC_PID_BASENAME="k2hr3pi-${PRGNAME_PREFIX}-watch.pid"

TARGET_PROGRAM=""
MAIN_PROGRAM="bin/www"
WATCH_PROGRAM="bin/watcher"

PROC_PID_FILE=""
MAINPROC_PID_FILE=""
WATCHPROC_PID_FILE=""

#==============================================================
# Utility functions
#==============================================================
#
# Usage
#
PrintUsage()
{
	echo "Usage:   $1 [--production(-prod) | --development(-dev)]"
	echo "            [--stop(-s)]"
	echo "            [--background(-bg) | --foreground(-fg)]"
	echo "            [--debug(-d) | --debug-nobrk(-dnobrk)]"
	echo "            [--debuglevel(-dl) DBG/MSG/WARN/ERR/(custom debug level)]"
	echo "            [--watcher(-w) [--oneshot(-o)]]"
	echo ""
	echo "Option:"
	echo "         --production(-prod)    : [default] Set 'production' to NODE_ENV environment (this is default and exclusive with the '--development' option)"
	echo "         --development(-dev)    : Set 'development' to NODE_ENV environment (exclusive with the '--production' option)"
	echo "         --stop(-s)             : Stop www or watcher nodejs process"
	echo "         --background(-bg)      : Run process background"
	echo "         --foreground(-fg)      : Run process foreground (this takes precedence over --background(-bg) option)"
	echo "         --debug(-d)            : Run with nodejs inspector option"
	echo "         --debug-nobrk(-dnobrk) : Run with nodejs inspector option (no break at start)"
	echo "         --debuglevel(-dl)      : Specify the level of debug output (DBG/MSG/WARN/ERR/custom debug level)"
	echo "         --watcher(-w)          : Run IP watcher process as daemon"
	echo "         --oneshot(-os)         : Run watcher process only once"
	echo ""
	echo "Environment:"
	echo "         DEBUG_LEVEL            : Specify DBG/MSG/WARN/ERR. Same as --debuglevel(-dl) option, and --debuglevel(-dl) takes precedence."
	echo ""
}

#
# Using ALL_CHILD_PIDS global variable
#
get_all_pid_list()
{
	if [ -z "${ALL_CHILD_PIDS}" ]; then
		return 0
	fi
	_ADD_PIDS=0
	for _ONE_PID in ${ALL_CHILD_PIDS}; do
		if _CIHLD_PIDS=$(pgrep -P "${_ONE_PID}"); then
			for _ONE_CPID in ${_CIHLD_PIDS}; do
				_FOUND_PID=0
				for _ONE_PPID in ${ALL_CHILD_PIDS}; do
					if [ "${_ONE_CPID}" = "${_ONE_PPID}" ]; then
						_FOUND_PID=1
						break
					fi
				done
				if [ "${_FOUND_PID}" -eq 0 ]; then
					#
					# Add child PID
					#
					ALL_CHILD_PIDS="${ALL_CHILD_PIDS} ${_ONE_CPID}"
					_ADD_PIDS=1
				fi
			done
		fi
	done
	if [ "${_ADD_PIDS}" -eq 1 ]; then
		#
		# Reentrant for check adding PIDs
		#
		get_all_pid_list
	fi
	return 0
}

#
# Stop processes
#
stop_old_process()
{
	if [ -z "${PROC_PID_FILE}" ]; then
		#
		# Not initialize yet
		#
		return 0
	fi

	if [ ! -f "${PROC_PID_FILE}" ]; then
		return 0
	fi

	ALL_CHILD_PIDS="$(tr -d '\n' < "${PROC_PID_FILE}")"

	if pgrep -f "${PRGNAME}" | grep -q "${ALL_CHILD_PIDS}"; then
		#
		# List up all chid processes
		#
		get_all_pid_list

		#
		# Try to stop(HUP) process and child processes
		#
		for _ONE_PID in ${ALL_CHILD_PIDS}; do
			kill -HUP "${_ONE_PID}" >/dev/null 2>&1
		done
		sleep 1

		#
		# If processes are running yet, try to kill it.
		#
		for _ONE_PID in ${ALL_CHILD_PIDS}; do
			# shellcheck disable=SC2009
			if ( ps -o pid,stat ax 2>/dev/null | grep -v 'PID' | awk '$2~/^[^Z]/ { print $1 }' | grep -q "^${_ONE_PID}$" || exit 1 && exit 0 ); then
				kill -KILL "${_ONE_PID}" >/dev/null 2>&1
			fi
		done
		sleep 1

		#
		# Result
		#
		for _ONE_PID in ${ALL_CHILD_PIDS}; do
			# shellcheck disable=SC2009
			if ( ps -o pid,stat ax 2>/dev/null | grep -v 'PID' | awk '$2~/^[^Z]/ { print $1 }' | grep -q "^${_ONE_PID}$" || exit 1 && exit 0 ); then
				echo "[ERROR] Could not stop old processes."
				return 1
			fi
		done

		echo "[INFO] Stop old processes."
	fi
	rm -f "${PROC_PID_FILE}"

	return 0
}

#==========================================================
# Parse options and check environments
#==========================================================
NODE_ENV_VALUE=""
FOREGROUND=0
BACKGROUND=0
STOP_OLD_PROCESS=0
INSPECTOR_OPT=""
DEBUG_ENV_LEVEL=0
DEBUG_ENV_CUSTOM=""
WATCHER_PROC=0
WATCH_ONESHOT=0

while [ $# -ne 0 ]; do
	if [ -z "$1" ]; then
		break

	elif echo "$1" | grep -q -i -e "^-h$" -e "^--help$"; then
		PrintUsage "${PRGNAME}"
		exit 0

	elif echo "$1" | grep -q -i -e "^-prod$" -e "^--production$"; then
		if [ -n "${NODE_ENV_VALUE}" ]; then
			echo "[ERROR] already specified --production(-prod) or --development(-dev) option"
			exit 1
		fi
		NODE_ENV_VALUE="production"

	elif echo "$1" | grep -q -i -e "^-dev$" -e "^--development$"; then
		if [ -n "${NODE_ENV_VALUE}" ]; then
			echo "[ERROR] already specified --production(-prod) or --development(-dev) option"
			exit 1
		fi
		NODE_ENV_VALUE="development"

	elif echo "$1" | grep -q -i -e "^-bg$" -e "^--background$"; then
		#
		# Not check multi same option...
		#
		BACKGROUND=1

	elif echo "$1" | grep -q -i -e "^-fg$" -e "^--foreground$"; then
		#
		# Not check multi same option...
		#
		FOREGROUND=1

	elif echo "$1" | grep -q -i -e "^-s$" -e "^--stop$"; then
		if [ "${STOP_OLD_PROCESS}" -ne 0 ]; then
			echo "[ERROR] already specified --stop(-s) option"
			exit 1
		fi
		STOP_OLD_PROCESS=1

	elif echo "$1" | grep -q -i -e "^-d$" -e "^--debug$"; then
		if [ -n "${INSPECTOR_OPT}" ]; then
			echo "[ERROR] already specified --debug(-d) or --debug-nobrk(-dnobrk) option"
			exit 1
		fi
		INSPECTOR_OPT="--inspect-brk=${LOCAL_HOSTNAME}:9229"

	elif echo "$1" | grep -q -i -e "^-dnobrk$" -e "^--debug-nobrk$"; then
		if [ -n "${INSPECTOR_OPT}" ]; then
			echo "[ERROR] already specified --debug(-d) or --debug-nobrk(-dnobrk) option"
			exit 1
		fi
		INSPECTOR_OPT="--inspect=${LOCAL_HOSTNAME}:9229"

	elif echo "$1" | grep -q -i -e "^-dl$" -e "^--debuglevel$"; then
		shift
		if [ $# -eq 0 ]; then
			echo "[ERROR] --debuglevel(-dl) option needs parameter(dbg/msg/warn/err/custom debug level)"
			exit 1
		fi
		if echo "$1" | grep -q -i -e "^dbg$" -e "^debug$"; then
			if [ "${DEBUG_ENV_LEVEL}" -ne 0 ]; then
				echo "[ERROR] --debuglevel(-dl) option already is set"
				exit 1
			fi
			DEBUG_ENV_LEVEL=4
		elif echo "$1" | grep -q -i -e "^msg$" -e "^message$" -e "^info$"; then
			if [ "${DEBUG_ENV_LEVEL}" -ne 0 ]; then
				echo "[ERROR] --debuglevel(-dl) option already is set"
				exit 1
			fi
			DEBUG_ENV_LEVEL=3
		elif echo "$1" | grep -q -i -e "^wan$" -e "^warn$" -e "^warning$"; then
			if [ "${DEBUG_ENV_LEVEL}" -ne 0 ]; then
				echo "[ERROR] --debuglevel(-dl) option already is set"
				exit 1
			fi
			DEBUG_ENV_LEVEL=2
		elif echo "$1" | grep -q -i -e "^err$" -e "^error$"; then
			if [ "${DEBUG_ENV_LEVEL}" -ne 0 ]; then
				echo "[ERROR] --debuglevel(-dl) option already is set"
				exit 1
			fi
			DEBUG_ENV_LEVEL=1
		else
			#
			# Custom debug level value
			#
			if [ -n "${DEBUG_ENV_CUSTOM}" ]; then
				DEBUG_ENV_CUSTOM="${DEBUG_ENV_CUSTOM},"
			fi
			DEBUG_ENV_CUSTOM="${DEBUG_ENV_CUSTOM}$1"
		fi

	elif echo "$1" | grep -q -i -e "^-w$" -e "^--watcher$" -e "^--watch$"; then
		if [ "${WATCHER_PROC}" -ne 0 ]; then
			echo "[ERROR] already specified --watcher(-w) option"
			exit 1
		fi
		WATCHER_PROC=1

	elif echo "$1" | grep -q -i -e "^-os$" -e "^--oneshot$"; then
		if [ "${WATCH_ONESHOT}" -ne 0 ]; then
			echo "[ERROR] already specified --oneshot(-os) option"
			exit 1
		fi
		WATCH_ONESHOT=1

	else
		echo "[WARNING] Unknown option $1 is specified, it is ignored."
	fi
	shift
done

#
# Check NODE_ENV_VALUE
#
if [ -z "${NODE_ENV_VALUE}" ]; then
	NODE_ENV_VALUE="production"
fi

#
# Check watcher option & set process variables
#
if [ "${WATCHER_PROC}" -ne 1 ] && [ "${WATCH_ONESHOT}" -eq 1 ]; then
    echo "[ERROR] invalid option is specified, --oneshot(-os) must be specified with --watcher(-w)."
    exit 1
fi

#----------------------------------------------------------
# Set pid file varibales
#----------------------------------------------------------
PROG_EXTRA_OPTIONS=""

#
# Check PID directory exists
#
if [ -d "${PID_FILE_BASEDIR}" ]; then
	MAINPROC_PID_FILE="${PID_FILE_BASEDIR}/${MAINPROC_PID_BASENAME}"
	WATCHPROC_PID_FILE="${PID_FILE_BASEDIR}/${WATCHPROC_PID_BASENAME}"
else
	MAINPROC_PID_FILE="${PID_FILE_TMPDIR}/${MAINPROC_PID_BASENAME}"
	WATCHPROC_PID_FILE="${PID_FILE_TMPDIR}/${WATCHPROC_PID_BASENAME}"
fi

#
# PID files
#
if [ ${WATCHER_PROC} -ne 1 ]; then
	PROC_PID_FILE="${MAINPROC_PID_FILE}"
	TARGET_PROGRAM="${MAIN_PROGRAM}"
else
	PROC_PID_FILE="${WATCHPROC_PID_FILE}"
	TARGET_PROGRAM="${WATCH_PROGRAM}"

	if [ "${WATCH_ONESHOT}" -eq 1 ]; then
		PROG_EXTRA_OPTIONS="--oneshot"
	fi
fi

#==========================================================
# Do work
#==========================================================
#
# Check background
#
if [ "${BACKGROUND}" -eq 1 ] && [ "${FOREGROUND}" -eq 0 ]; then
	#
	# Run another process as child
	#
	"${CMDLINE_COMMAND}" "${CMDLINE_PARAMETERS}" -fg > /dev/null 2>&1 &
	exit 0
fi

#
# Stop old process if exists
#
if ! stop_old_process; then
	exit 1
fi
if [ "${STOP_OLD_PROCESS}" -eq 1 ]; then
	#
	# Only sto pole processes
	#
	exit 0
fi

#
# Make NODE_DEBUG environment
#
DEBUG_ENV_PARAM=""
if [ "${DEBUG_ENV_LEVEL}" -ge 4 ]; then
	DEBUG_ENV_PARAM="LOGLEVEL_DBG"
elif [ "${DEBUG_ENV_LEVEL}" -ge 3 ]; then
	DEBUG_ENV_PARAM="LOGLEVEL_MSG"
elif [ "${DEBUG_ENV_LEVEL}" -ge 2 ]; then
	DEBUG_ENV_PARAM="LOGLEVEL_WAN"
elif [ "${DEBUG_ENV_LEVEL}" -ge 1 ]; then
	DEBUG_ENV_PARAM="LOGLEVEL_ERR"
else
	DEBUG_ENV_PARAM="LOGLEVEL_SILENT"
    if [ -n "${DEBUG_LEVEL}" ]; then
		if echo "${DEBUG_LEVEL}" | grep -q -i -e "^dbg$" -e "^debug$"; then
			DEBUG_ENV_PARAM="LOGLEVEL_DBG"
		elif echo "${DEBUG_LEVEL}" | grep -q -i -e "^msg$" -e "^message$" -e "^info$"; then
			DEBUG_ENV_PARAM="LOGLEVEL_MSG"
		elif echo "${DEBUG_LEVEL}" | grep -q -i -e "^wan$" -e "^warn$" -e "^warning$"; then
			DEBUG_ENV_PARAM="LOGLEVEL_WAN"
		elif echo "${DEBUG_LEVEL}" | grep -q -i -e "^err$" -e "^error$"; then
			DEBUG_ENV_PARAM="LOGLEVEL_ERR"
		fi
    fi
fi
if [ -n "${DEBUG_ENV_CUSTOM}" ]; then
	if [ -n "${DEBUG_ENV_PARAM}" ]; then
		DEBUG_ENV_PARAM="${DEBUG_ENV_PARAM},"
	fi
	DEBUG_ENV_PARAM="${DEBUG_ENV_PARAM}${DEBUG_ENV_CUSTOM}"
fi

#
# Push this process id to PID file
#
if ! echo "$$" > "${PROC_PID_FILE}"; then
	echo "[ERROR] Could create PID file(${PROC_PID_FILE}) and push this process PID."
	exit 1
fi

#
# Run
#
cd "${SRCTOP}" || exit 1
if [ "${DEBUG_ENV_LEVEL}" -ge 4 ]; then
	echo "***** RUN *****"
	echo "NODE_PATH=${NODE_PATH} NODE_ENV=${NODE_ENV_VALUE} NODE_DEBUG=${DEBUG_ENV_PARAM} node ${INSPECTOR_OPT} ${TARGET_PROGRAM} ${PROG_EXTRA_OPTIONS}"
	echo ""
fi

EXIT_CODE=0
if ! /bin/sh -c "NODE_PATH=${NODE_PATH} NODE_ENV=${NODE_ENV_VALUE} NODE_DEBUG=${DEBUG_ENV_PARAM} node ${INSPECTOR_OPT} ${TARGET_PROGRAM} ${PROG_EXTRA_OPTIONS}"; then
	EXIT_CODE="$?"
	echo "[INFO] Process exited with exit code: ${EXIT_CODE}"
fi

exit "${EXIT_CODE}"

#
# Local variables:
# tab-width: 4
# c-basic-offset: 4
# End:
# vim600: noexpandtab sw=4 ts=4 fdm=marker
# vim<600: noexpandtab sw=4 ts=4
#