#!/sbin/runscript
# Copyright 1999-2013 Gentoo Foundation
# Distributed under the terms of the GNU General Public License v2
# $Header: $

extra_started_commands="reload"

depend() {
	need net
}

start() {
	ebegin "Start ${SVCNAME}"

	# warn on low ulimit
	local ulimit=$(ulimit -n)
	if [ $ulimit -lt 4096 ]; then
		ewarn "Current ulimit -n is $ulimit. 4096 is the recommended minimum."
	fi

	start-stop-daemon --background --start --user "${RIAK_USER}" \
			--pidfile "${RIAK_PIDFILE}" \
			--exec "${RIAK_EXEC}" -- start

	# wait for riak to start up before creating pidfile
	while [ ${RIAK_STARTUP_TIMEOUT} -ge 0 ]; do
		# ping riak
		local res=$(${RIAK_EXEC} ping)

		# on pong it has fully started
		if [[ $res =~ "pong" ]]; then
			while [ ${RIAK_STARTUP_TIMEOUT} -ge 0 ]; do
				# get the beam's pid
				local beam=$(pidof "${RIAK_ERTS_PATH}/bin/beam.smp")
				# retry if multiple pids are returned
				if [[ $beam =~ ' ' ]]; then
					# let pidfile creation time out
					RIAK_STARTUP_TIMEOUT=$(expr "${RIAK_STARTUP_TIMEOUT}" - 1)
					sleep 1
				else
					# create pidfile and be done
					[ $beam ] && echo -n $beam > "${RIAK_PIDFILE}"
					break
				fi
			done
			break
		else
			# let pidfile creation time out
			RIAK_STARTUP_TIMEOUT=$(expr "${RIAK_STARTUP_TIMEOUT}" - 1)
			sleep 1
		fi
	done
	
	[ -f ${RIAK_PIDFILE} ] || ewarn "Slow startup. No pidfile created."

	eend $?
}

stop() {
	ebegin "Stopping ${SVCNAME}"
	local beam="${RIAK_ERTS_PATH}/bin/beam.smp"

	# if riak stop fails use more force
	if [ $(${RIAK_EXEC} stop 2>&1 > /dev/null) ]; then
		start-stop-daemon \
			--stop --quiet --retry=TERM/30/KILL/5 \
			--user ${RIAK_USER} --exec $beam
		[ -f "${RIAK_PIDFILE}" ] && unlink "${RIAK_PIDFILE}"
	else
		[ -f "${RIAK_PIDFILE}" ] && unlink "${RIAK_PIDFILE}"
	fi

	einfo "Stopping epmd"
	local epmd=$(pidof ${RIAK_ERTS_PATH}/bin/epmd)
	[[ $epmd ]] && kill $epmd

	eend $?
}

reload() {
	ebegin "Reloading ${SVCNAME}"
	${RIAK_EXEC} restart 2>&1 > /dev/null
	eend $?
}

status() {
	local res=$(${RIAK_EXEC} ping)
	local epmd=$(pidof ${RIAK_ERTS_PATH}/bin/epmd)
	if [[ $res =~ "pong" ]]; then
		einfo "status: started"
	elif [ -f "${RIAK_PIDFILE}" ]; then
		eerror "status: crashed"
	else
		# cleanup
		[ $epmd ] && kill $epmd
		einfo "status: stopped"
	fi
}