#!/bin/sh
#
# 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:

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

L7SOCKFILE=${OCF_RESKEY_socket_default:-/var/run/l7vs/l7vs}

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

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

<longdesc lang="en">
This is a L7vsd Resource Agent.
</longdesc>
<shortdesc lang="en">L7vsd resource agent</shortdesc>

<parameters>
<parameter name="socket" unique="0" required="0">
<longdesc lang="en">
The socket to be used for l7vsadm.
</longdesc>
<shortdesc lang="en">l7vsadm socket</shortdesc>
<content type="string" default="${OCF_RESKEY_socket_default}"/>
</parameter>
</parameters>

<actions>
<action name="start"        timeout="60" />
<action name="stop"         timeout="60" />
<action name="monitor"      timeout="60" interval="10" depth="0" start-delay="0" />
<action name="meta-data"    timeout="5" />
</actions>
</resource-agent>
END
}

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

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

END
}

###############################
# Resource start Method
###############################
l7vsd_start() {
        ocf_log info "l7vsd is starting ..."
        l7vsd_monitor info
        RET=$?
        if [ $RET -eq $OCF_SUCCESS ]; then
                ocf_log info "l7vsd is already running."
                return $OCF_SUCCESS
        elif [ $RET -eq $OCF_NOT_RUNNING ]; then
                L7SOCKDIR=`dirname L7SOCKFILE`
                if [ ! -d $L7SOCKDIR ] ; then
                        ocf_log info "Creating l7vsadm socket dir: $L7SOCKDIR"
                        mkdir -p $L7SOCKDIR
                fi
                if [ -e $L7SOCKFILE ] ; then
                        ocf_log info "Delete l7vsadm socket filer: $L7SOCKFILE"
                        rm -rf $L7SOCKFILE
                fi
                /usr/sbin/l7vsd > /dev/null 2>&1
                RET=$?
                if [ $RET -ne 0 ]; then
                        MSG="l7vsd start error!"
                        outputLog err ${OCF_ERR_GENERIC} $MSG
                        return $OCF_ERR_GENERIC
                fi
                l7vsd_status
                if [ $? -eq $OCF_SUCCESS ]; then
                        # l7vsd is running
                        ocf_log info "l7vsd starts."
                        return $OCF_SUCCSESS
                fi
        fi
        MSG="l7vsd does not work. (ps=$RET) "
        outputLog err ${OCF_ERR_GENERIC} ${MSG}
        return $OCF_ERR_GENERIC
}

###############################
# Resource stop Method 
###############################
l7vsd_stop() {
        ocf_log info "l7vsd stopping ..."
        l7vsd_status
        RET=$?
        if [ $RET -eq $OCF_NOT_RUNNING ]; then
                ocf_log info "l7vsd stopped."
                return $OCF_SUCCESS
        fi
        pkill -f "/usr/sbin/l7vsd"
        count=0
        while [ $count -le 10 ]
        do
                l7vsd_status
                RET=$?
                if [ $RET -eq $OCF_NOT_RUNNING ]; then
                        ocf_log info "l7vsd stopped."
                        return $OCF_SUCCESS
                fi
                count=`expr $count + 1`
                sleep 1
        done
        l7vsd_pkill
        RET=$?
        rm -rf $L7SOCKFILE
        return $RET
}

###############################
# Logging Method
###############################
outputLog(){
	MODE=$1
	shift
	case $MODE in
		info)	RET=$1; shift;
			ocf_log $MODE "[$0 ${__OCF_ACTION}] OK;return=$RET" "$@";;
		err)	RET=$1; shift;
			ocf_log $MODE "[$0 ${__OCF_ACTION}] NG;return=$RET" "$@";;
	esac
}

###############################
# kill process
###############################
l7vsd_pkill(){
        pkill -9 -f "/usr/sbin/l7vsd"
        ocf_log info "kill l7vsd process!"
        while true
        do
                sleep 1
                l7vsd_status
                RET=$?
                if [ $RET -eq $OCF_NOT_RUNNING ]; then
                     # stop OK
                     ocf_log info "l7vsd process stopped!"
                     return $OCF_SUCCESS
                fi
        done
}

###############################
# Get Resource Status Method
###############################
l7vsd_status(){
	RET=0
	RET=`pgrep -fox "/usr/sbin/l7vsd" | wc -l`
	if [ $RET -eq 1 ]; then
		return $OCF_SUCCESS
	elif [ $RET -eq 0 ]; then
		MSG="l7vsd is not running."
		outputLog $loglevel ${OCF_NOT_RUNNING} ${MSG}
		return $OCF_NOT_RUNNING
	else
		MSG="l7vsd status ERROR!: (ps=$RET)"
		outputLog err ${OCF_ERR_GENERIC} ${MSG}
		return $OCF_ERR_GENERIC
	fi
}

###############################
# Get Resource Monitor Method
###############################
l7vsd_monitor() {
	local loglevel
	loglevel=${1:-err}
	if ocf_is_probe; then
		loglevel="info"
	fi

	l7vsd_status
	RET=$?
	if [ $RET -eq $OCF_SUCCESS ]; then
		# status OK
		return $OCF_SUCCESS
	elif [ $RET -eq $OCF_NOT_RUNNING ]; then
		# status NG
		return $OCF_NOT_RUNNING
	fi
	MSG="l7vsd duplicated. "
	outputLog err ${OCF_ERR_GENERIC} ${MSG}
	return $OCF_ERR_GENERIC
}

case $__OCF_ACTION in
meta-data)	meta_data
		exit $OCF_SUCCESS ;;
start)		l7vsd_start;;
stop)		l7vsd_stop;;
monitor)	l7vsd_monitor;;
usage|help)	l7vsd_usage
		exit $OCF_SUCCESS
		;;
*)		l7vsd_usage
		exit $OCF_ERR_UNIMPLEMENTED
    
		;;
esac
rc=$?
ocf_log debug "${OCF_RESOURCE_INSTANCE} $__OCF_ACTION : $rc"
exit $rc
