mshots_ctl.sh

#!/bin/bash

### BEGIN INIT INFO
# Provides:          mShots.JS
# Required-Start:
# Required-Stop:
# Default-Start:     2 3 4 5
# Default-Stop:      0 1 6
# Short-Description: Start/stop mShots.JS
# Description:       Start/stop the mShots service.
### END INIT INFO

# installation directory
INSTALL_DIR=/opt/mshots
# port to run the mShots.JS service on
PORT=7777
# number of workers to put to work
MSHOTS_WORKERS=${MSHOTS_WORKERS:-20}

function startservice {
	if [ ! -d ${INSTALL_DIR} ]; then
		echo "mShots.JS is not installed in the correct directory: ${INSTALL_DIR}"
		exit 1
	fi
	pid="`ps -ef | grep 'mShots.JS - Master' | grep -v 'grep' | awk ' { print $(2) }'`"
	if [ -z $pid ]; then
		pid=0
	fi
	if [ $pid -gt 0 ] ; then
		echo "mShots.JS service is already running"
		exit 1
	fi

	running=`lsmod | grep vfb | awk '{ print $(1) }'`
	if [ -z $running ] && [ "$ARCH" != "arm64" ] ; then
		echo "loading vfb"
		modprobe -r vfb
		modprobe vfb vfb_enable=yes videomemorysize=7733248
		fbset 1600x1200-76 -depth 24 -accel true
	else
		echo "vfb already loaded, skipping"
	fi

	if [ ! -d "${INSTALL_DIR}/logs" ]; then
		mkdir "${INSTALL_DIR}/logs"
	fi
	if [ ! -d "${INSTALL_DIR}/stats" ]; then
		mkdir "${INSTALL_DIR}/stats"
	fi
	if [ -z "$PORT" ]; then
		echo "PORT not specified. Define it in this script"
		exit 1
	fi

	pid="`ps -ef | grep 'mShots.JS - Worker' | grep -v 'grep' | awk 'NR==1 { print $(2) }'`"
	if [ -z $pid ]; then
		pid=0
	else
		echo "Some orphaned Worker threads found from a previous Master:"
	fi
	while [ $pid -gt 0 ]; do
		echo "Killing orphaned Worker thread: $pid"
		kill -s 9 $pid
		pid="`ps -ef | grep 'mShots.JS - Worker' | grep -v 'grep' | awk 'NR==1 { print $(2) }'`"
		if [ -z $pid ]; then
			pid=0
		fi
	done

	echo "Starting mShots.JS"
	ARGS="${INSTALL_DIR}/lib/mshots.js -p $PORT -n $MSHOTS_WORKERS 2"
	start-stop-daemon -S -q -m -b --pidfile /var/run/mshots.pid --exec /usr/local/node/bin/node --chdir $INSTALL_DIR -- $ARGS
	sleep 1
}

function stopservice {
	echo "Stopping mShots"
	start-stop-daemon --stop --retry=TERM/10/KILL/5 --pidfile /var/run/mshots.pid --name "mShots.JS - Mas"
	sleep 2
	start-stop-daemon --stop --oknodo --retry=0/10/KILL/5 --name "mShots.JS - Wor"

	rm -f /var/run/mshots.pid
}

function reload_config {
	pid="`cat /var/run/mshots.pid`"
	if [ -z $pid ]; then
		pid=0
	fi
	if [ $pid -gt 0 ]; then
		echo "Sending the master mShots.JS process the reload signal."
		start-stop-daemon -K -s 12 -q --pidfile /var/run/mshots.pid
	else
		echo "There is no mShots.JS Master process running to receive the reload signal."
	fi
}

case "$1" in
start )
	startservice
	;;
stop )
	# Sends the Master Thread a SIGHUP signal to shutdown the slaves and exit
	stopservice
	;;
restart )
	stopservice
	sleep 2
	startservice
	;;
reload )
	reload_config
	;;
* )
	echo "Usage:$0 start|stop|restart|reload"
	exit 1
	;;
esac
exit 0