#!/bin/bash
set -o nounset

. /etc/rc.d/init.d/functions

RETVAL=0
PIDFILE="{{ work_path }}/supervisord.pid"
LOCKFILE="{{ work_path }}/supervisord.lock"
OPTIONS="-c {{ config_file_path }}"
WAIT_FOR_SUBPROCESSES="yes"

start() {
    echo "Starting supervisord: "
	if [ -e $PIDFILE ]; then 
		echo "ALREADY STARTED"
		return 1
	fi

    {{ python_bin_dir }}/supervisord $OPTIONS

	{{ python_bin_dir }}/supervisorctl $OPTIONS status
	
    [ -e $PIDFILE ] && touch $LOCKFILE
}

stop() {
    echo -n "Stopping supervisord: "
    {{ python_bin_dir }}/supervisorctl $OPTIONS shutdown
	if [ -n "$WAIT_FOR_SUBPROCESSES" ]; then 
        echo "Waiting roughly 60 seconds for $PIDFILE to be removed after child processes exit"
        for sleep in  2 2 2 2 4 4 4 4 8 8 8 8 last; do
            if [ ! -e $PIDFILE ] ; then
                echo "Supervisord exited as expected in under $total_sleep seconds"
                break
            else
                if [[ $sleep -eq "last" ]] ; then
                    echo "Supervisord still working on shutting down. We've waited roughly 60 seconds, we'll let it do its thing from here"
                    return 1
                else
                    sleep $sleep
                    total_sleep=$(( $total_sleep + $sleep ))
                fi

            fi
        done
    fi

    # always remove the subsys.  we might have waited a while, but just remove it at this point.
    rm -f $LOCKFILE
}

restart() {
        stop
        start
}

case "$1" in
    start)
        start
        RETVAL=$?
        ;;
    stop)
        stop
        RETVAL=$?
        ;;
    restart|force-reload)
        restart
        RETVAL=$?
        ;;
    reload)
        /usr/bin/supervisorctl $OPTIONS reload
        RETVAL=$?
        ;;
    condrestart)
        [ -f $LOCKFILE ] && restart
        RETVAL=$?
        ;;
    status)
        {{ python_bin_dir }}/supervisorctl $OPTIONS status
        status -p $PIDFILE supervisord
        # The 'status' option should return one of the LSB-defined return-codes,
        # in particular, return-code 3 should mean that the service is not
        # currently running. This is particularly important for Ansible's 'service'
        # module, as without this behaviour it won't know if a service is up or down.
        RETVAL=$?
        ;;
    *)
        echo $"Usage: $0 {start|stop|status|restart|reload|force-reload|condrestart}"
        exit 1
esac

exit $RETVAL