#!/bin/sh
#
# Copyright (c) 2012, Parallels
#
### BEGIN INIT INFO
# Provides:			wdcollect
# Required-Start:		$local_fs $remote_fs $network
# Should-Start:			$named $time mysql ypclient
# Should-Stop:			$named $time mysql ypclient
# Required-Stop:		$local_fs $remote_fs $network
# Default-Start:		3 5
# Default-Stop:			0 1 2 6
# X-Interactive:		true
# Short-Description:		Watchdog wdcollect daemon
# Description:			Start the watchdog monitoring
### END INIT INFO

test -s /etc/rc.status && . /etc/rc.status && rc_reset

pname=wdcollect

wd_service="/usr/local/psa/admin/bin/modules/watchdog/wdservice"
wd_config="/usr/local/psa/etc/modules/watchdog/wdcollect.inc.php"
wd_pid="/var/run/${pname}.pid"

wdcollect_get_pid()
{
	[ -f $wd_pid ] && pid="`cat $wd_pid`"
	proc_pid="`ps ax|grep $pname | grep sw-engine | grep -v grep | awk '{print $1}'`"

	[ "$wd_pid" = "$proc_pid" ] && return $wd_pid

	if [ -z "$proc_pid" ]; then
		[ -f "$wd_pid" ] && rm -f "$wd_pid" 2>/dev/null
		return 0
	fi

	# The process is exists but pid doesn't match in pid file.
	echo "$proc_pid" > $wd_pid

	return $proc_pid
}

wdcollect_status()
{
	wdcollect_get_pid
	[ $? -eq 0 ] && return 1 || return 0
}

wdcollect_stop()
{
	pid="`wdcollect_get_pid`"
	[ $? -eq 0 ] && return 0

	while wdcollect_status; do
		$wd_service $pname stop $wd_config 2>/dev/null
		sleep 1
	done

	return 0
}

wdcollect_start()
{
	wdcollect_status
	[ $? -eq 0 ] || $wd_service $pname start $wd_config

	for i in 1 2 4 6; do
		wdcollect_status
		[ $? -eq 0 ] && return 0
		sleep 1
	done

	return 1
}

action="$1"

case "$action" in
    start)
	wdcollect_start
	;;
    stop)
	wdcollect_stop
	;;
    restart)
	wdcollect_stop
	wdcollect_start
	;;
    status)
	wdcollect_status
	;;
    *)
	echo "Usage: `basename $0` <start|stop|restart|status>"
	exit 1
	;;
esac
