uwsgi.init

#!/usr/bin/env bash

### BEGIN INIT INFO
# Provides:          uwsgi
# Required-Start:    $all
# Required-Stop:     $all
# Default-Start:     2 3 4 5
# Default-Stop:      0 1 6
# Short-Description: starts the uwsgi app server
# Description:       starts uwsgi app server using start-stop-daemon
### END INIT INFO
set -e

VERSION=$(basename $0)

if [[ "uwsgi uwsgi.init" =~ $VERSION ]]; then
    echo "Please use version-specific uwsgi!"
    find /etc/init.d/ -name 'uwsgi-2.?' -exec echo "  {}" \;
    exit 1
fi

PATH=/usr/local/bin:/sbin:/bin:/usr/sbin:/usr/bin
DAEMON=/usr/local/bin/$VERSION
RUN=/var/run/$VERSION/
CONFIGS_DIR=/etc/uwsgi/$VERSION/sites-enabled
OWNER=www-data
NAME=$VERSION
DESC=$VERSION
OP=$1

if [[ -n $2 ]]; then
    shift
    ENABLED_CONFIGS=""
    for c in $@; do
        [[ -f $CONFIGS_DIR/$c ]] && \
            ENABLED_CONFIGS=$(echo -e "$ENABLED_CONFIGS\n$CONFIGS_DIR/$c")
    done
else
    ENABLED_CONFIGS=$(find $CONFIGS_DIR/ -not -type d)
fi

[[ -x $DAEMON ]] || exit 0
[[ -d $RUN ]] || mkdir $RUN && chown www-data $RUN

DAEMON_OPTS=""

# Include uwsgi defaults if available
if [[ -f /etc/default/$VERSION ]]; then
    . /etc/default/$VERSION
fi

do_pid_check()
{
    local PIDFILE=$1
    [[ -f $PIDFILE ]] || return 0
    local PID=$(cat $PIDFILE)
    for p in $(pgrep $VERSION); do
        [[ $p == $PID ]] && return 1
    done
    return 0
}


do_start()
{
    for config in $ENABLED_CONFIGS; do
        local PIDFILE=$RUN/$(basename $config).pid
        if do_pid_check $PIDFILE; then
            sudo -u $OWNER -i $VERSION $config $DAEMON_OPTS --pidfile $PIDFILE
        else
            echo "Already running!"
        fi
    done
}

send_sig()
{
    for config in $ENABLED_CONFIGS; do
        local PIDFILE=$RUN/$(basename $config).pid
        set +e
        [[ -f $PIDFILE ]] && kill $1 $(cat $PIDFILE) > /dev/null 2>&1
        set -e
    done
}

wait_and_clean_pidfiles()
{
    for config in $ENABLED_CONFIGS; do
        local PIDFILE=$RUN/$(basename $config).pid
        until do_pid_check $PIDFILE; do
            echo -n "";
        done
        rm -f $PIDFILE
    done
}

do_stop()
{
    send_sig -3
    wait_and_clean_pidfiles
}

do_reload()
{
    send_sig -1
}

do_force_reload()
{
    send_sig -15
}

get_status()
{
    send_sig -10
}

case "$OP" in
    start)
        echo "Starting $DESC: "
        do_start
        echo "$NAME."
        ;;
    stop)
        echo -n "Stopping $DESC: "
        do_stop
        echo "$NAME."
        ;;
    reload)
        echo -n "Reloading $DESC: "
        do_reload
        echo "$NAME."
        ;;
    force-reload)
        echo -n "Force-reloading $DESC: "
        do_force_reload
        echo "$NAME."
       ;;
    restart)
        echo  "Restarting $DESC: "
        do_stop
        sleep 1
        do_start
        echo "$NAME."
        ;;
    status)
        get_status
        ;;
    *)
        N=/etc/init.d/$NAME
        echo "Usage: $N {start|stop|restart|reload|force-reload|status}" >&2
        exit 1
        ;;
esac
exit 0