#!/bin/bash
# chkconfig: - 21 79
# description: Sheepdog
# processname: sheep
#
### BEGIN INIT INFO
# Provides:		sheepdog
# Required-Start:	$network
# Should-Start:		$syslog
# Required-Stop:	$network
# Default-Start:
# Default-Stop:
# Short-Description:	Starts and stops Sheepdog.
# Description:		Starts and stops Sheepdog.
### END INIT INFO
desc="Sheepdog QEMU/KVM Block Storage"
prog="sheep"

# set secure PATH
PATH="/sbin:/bin:/usr/sbin:/usr/bin:/usr/sbin"
SHEEPDOGD=/usr/sbin/$prog
SHEEPDOG_PATH=/var/lib/sheepdog 

success()
{
	echo -ne "[  OK  ]\r"
}

failure()
{
	echo -ne "[FAILED]\r"
}

status()
{
	pid=$(pidof $1 2>/dev/null)
	rtrn=$?
	if [ $rtrn -ne 0 ]; then
		echo "$1 is stopped"
	else
		echo "$1 (pid $pid) is running..."
	fi
	return $rtrn
}

# rpm based distros
if [ -d /etc/sysconfig ]; then
	[ -f /etc/rc.d/init.d/functions ] && . /etc/rc.d/init.d/functions
	[ -f /etc/sysconfig/$prog ] && . /etc/sysconfig/$prog
	[ -z "$LOCK_FILE" ] && LOCK_FILE="/var/lock/subsys/$prog"
fi

# deb based distros
if [ -d /etc/default ]; then
	[ -s @LIBDIR@/lsb/init-functions ] && . @LIBDIR@/lsb/init-functions
	[ -f /etc/default/$prog ] && . /etc/default/$prog
	[ -z "$LOCK_FILE" ] && LOCK_FILE="/var/lock/$prog"
fi

# gentoo/exherbo/sabayon based distros
if [ -d /etc/conf.d ]; then
  [ -f /etc/conf.d/$prog.conf ] && . /etc/conf.d/$prog.conf
  [ -z "$LOCK_FILE" ] && LOCK_FILE="/var/lock/$prog"
fi


# The version of __pids_pidof in /etc/init.d/functions calls pidof with -x
# This means it matches scripts, including this one.
# Redefine it here so that status (from the same file) works.
# Otherwise simultaneous calls to stop() will loop forever
__pids_pidof() {
        pidof -c -o $$ -o $PPID -o %PPID "$1" || \
                pidof -c -o $$ -o $PPID -o %PPID "${1##*/}"
}

start()
{
	echo -n "Starting $desc ($prog): "

	# most recent distributions use tmpfs for /var/run
	# to avoid to clean it up on every boot.
	# they also assume that init scripts will create
	# required subdirectories for proper operations
	mkdir -p /var/run

	if status $prog > /dev/null 2>&1; then
		success
	else
		$SHEEPDOGD -p 7000 $SHEEPDOG_PATH > /dev/null 2>&1

		# give it time to fail
		sleep 2
		if status $prog > /dev/null 2>&1; then
			touch $LOCK_FILE
			success
		else
			failure
			rtrn=1
		fi
	fi
	echo
}

stop()
{
	! status $prog > /dev/null 2>&1 && return

	echo -n "Stopping $desc ($prog): "
	killproc $prog
	RETVAL=$?
	rm -f $LOCK_FILE
	success
	echo
}

restart()
{
	stop
	start
}

rtrn=0

case "$1" in
start)
	start
;;
restart|reload|force-reload)
	restart
;;
condrestart|try-restart)
	if status $prog > /dev/null 2>&1; then
		restart
	fi
;;
status)
	status $prog
	rtrn=$?
;;
stop)
	stop
;;
*)
	echo "usage: $0 {start|stop|restart|reload|force-reload|condrestart|try-restart|status}"
	rtrn=2
;;
esac


exit $rtrn
