#!/bin/bash
# vim: set sw=4 et :
#
#	Resource Agent for BOOTH site daemon.
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of version 2 of the GNU General Public License as
# published by the Free Software Foundation.
#
# This program is distributed in the hope that it would be useful, but
# WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
#
# Further, this software is distributed without any warranty that it is
# free of the rightful claim of any third person regarding infringement
# or the like.  Any license provided herein, whether implied or
# otherwise, applies only to this software file.  Patent licenses, if
# any, provided herein do not apply to combinations of this program with
# other software, or any other product whatsoever.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write the Free Software Foundation,
# Inc., 59 Temple Place - Suite 330, Boston MA 02111-1307, USA.
#

#######################################################################
# Initialization:

BOOTH_DAEMON_STARTED=0
BOOTH_DAEMON_STARTING=1
BOOTH_DAEMON_EXIST=2
BOOTH_DAEMON_NOT_RUNNING=3
BOOTH_ERROR_GENERIC=4

. ${OCF_ROOT}/resource.d/heartbeat/.ocf-shellfuncs

#######################################################################

meta_data() {
	cat <<END
<?xml version="1.0"?>
<!DOCTYPE resource-agent SYSTEM "ra-api-1.dtd">
<resource-agent name="booth-site" version="0.9">
<version>1.0</version>


<longdesc lang="en">
This Resource Agent can control the BOOTH site daemon.
It assumes that the binary boothd is in your default PATH.
In most cases, it should be run as a primitive resource.
</longdesc>
<shortdesc lang="en">BOOTH site daemon</shortdesc>

<parameters>

<parameter name="config" unique="1">
<longdesc lang="en">
The configuration name (or configuration filename) to use.
</longdesc>
<shortdesc lang="en">BOOTH Options</shortdesc>
<content type="string" default="/etc/booth/booth.conf" />
</parameter>

<parameter name="args" unique="1">
<longdesc lang="en">
Any additional options to start the BOOTH daemon with
</longdesc>
<shortdesc lang="en">BOOTH Options</shortdesc>
<content type="string" default="" />
</parameter>

<parameter name="daemon" unique="1">
<longdesc lang="en">
The daemon to start
</longdesc>
<shortdesc lang="en">The daemon to start</shortdesc>
<content type="string" default="boothd" />
</parameter>

</parameters>

<actions>
<action name="start"        timeout="90" />
<action name="stop"         timeout="100" />
<action name="monitor"      timeout="20" interval="10" depth="0" start-delay="0" />
<action name="meta-data"    timeout="5" />
<action name="validate-all"   timeout="30" />
</actions>
</resource-agent>
END
}

#######################################################################

booth_usage() {
	cat <<END
usage: $0 {start|stop|monitor|validate-all|meta-data}

Expects to have a fully populated OCF RA-compliant environment set.
END
}


booth_check_daemon_state(){
	rc=$BOOTH_ERROR_GENERIC
	eval `"$OCF_RESKEY_daemon" status "-c$OCF_RESKEY_config" ; echo rc=$?`
	case $rc in 
	0)
		case  "$booth_state" in 
		started)  return $BOOTH_DAEMON_STARTED;;
		starting) return $BOOTH_DAEMON_STARTING;;
		*) return $BOOTH_ERROR_GENERIC;;	
		esac
	;;
	$OCF_NOT_RUNNING) return $BOOTH_DAEMON_NOT_RUNNING;;
	$OCF_ERR_GENERIC) return $BOOTH_ERROR_GENERIC;;
	*) return $BOOTH_ERROR_GENERIC;;
	esac
}

booth_start() {
	booth_check_daemon_state; rc=$?

	case $rc in
	$BOOTH_DAEMON_STARTED)     return $OCF_SUCCESS;;
	$BOOTH_DAEMON_STARTING) 
		while [ $rc != $BOOTH_DAEMON_STARTED ]; do
			sleep .7
			booth_check_daemon_state ; rc=$?
		done
		return $OCF_SUCCESS;;
	$BOOTH_DAEMON_NOT_RUNNING) ;;
	*) return $OCF_ERR_GENERIC;;
	esac

	"$OCF_RESKEY_daemon" daemon "-c$OCF_RESKEY_config" $OCF_RESKEY_args

	sleep 1.1
	rc=$BOOTH_DAEMON_STARTING
	while [ $rc != $BOOTH_DAEMON_STARTED ]; do
		booth_check_daemon_state; rc=$?
		sleep .8
	done
	
	return $OCF_SUCCESS
}

booth_stop() {
	booth_check_daemon_state; rc=$?

	case $rc in
	$BOOTH_DAEMON_STARTED) ;;
	$BOOTH_DAEMON_STARTING) ;;
	$BOOTH_DAEMON_NOT_RUNNING) return $OCF_SUCCESS;;
	*) return $OCF_ERR_GENERIC;;
	esac


	if [ "$booth_lockpid" == "" ] ; then
		return $OCF_ERR_GENERIC
	fi
	if pkill -TERM -s $booth_lockpid boothd ; then
	    sleep 1
	    pkill -KILL -s $booth_lockpid boothd
	fi

	rc=$BOOTH_DAEMON_STARTED
	while [ $rc != $BOOTH_DAEMON_NOT_RUNNING ]; do
		booth_check_daemon_state; rc=$?
		sleep 0.9
	done

	if [ $rc = $BOOTH_DAEMON_NOT_RUNNING ]; then
		rc=$OCF_SUCCESS
	else
		rc=$OCF_ERR_GENERIC
	fi

	return $rc
}

booth_monitor() {
	booth_check_daemon_state; rc=$?

	case $rc in
	$BOOTH_DAEMON_STARTED) return $OCF_SUCCESS;;
	$BOOTH_DAEMON_STARTING) return $OCF_NOT_RUNNING;;
	$BOOTH_DAEMON_EXIST) return $OCF_NOT_RUNNING;;
	$BOOTH_DAEMON_NOT_RUNNING) return $OCF_NOT_RUNNING;;
	*) return $OCF_ERR_GENERIC;;
	esac
}

booth_validate() {
    check_binary ${OCF_RESKEY_daemon}

    case ${OCF_RESKEY_CRM_meta_globally_unique} in
	yes|Yes|true|True|1) 
	    ocf_log err "$OCF_RESOURCE_INSTANCE must be configured with the globally_unique=false meta attribute"
	    exit $OCF_ERR_CONFIGURED
	    ;;
    esac

    return $OCF_SUCCESS
}

# Set OCF_RESKEY_daemon to boothd as default if not set
if [ ! ${OCF_RESKEY_daemon} ]; then
    OCF_RESKEY_daemon=`which boothd`
fi

if ! test -x "$OCF_RESKEY_daemon" ; then
    exit $OCF_ERR_INSTALLED
fi

if [ "$OCF_RESKEY_config" == "" ] ; then
    OCF_RESKEY_config=/etc/booth/booth.conf
fi


# for testing:
# ocf-tester -n p_booth -v /usr/lib/ocf/resource.d/pacemaker/booth-site

case $__OCF_ACTION in
meta-data)	meta_data
		exit $OCF_SUCCESS
		;;
start)		booth_validate; booth_start;;
stop)		booth_stop;;
monitor)	booth_validate; booth_monitor;;
validate-all)	booth_validate;;
usage|help)	booth_usage
		exit $OCF_SUCCESS
		;;
migrate_to|migrate_from|reload|demote|promote|notify)
		exit $OCF_ERR_UNIMPLEMENTED
		;;
*)		booth_usage
		exit $OCF_ERR_UNIMPLEMENTED
		;;
esac
rc=$?

exit $rc

