Ticket #1214: puppetmaster-mongrel2

#!/bin/bash
# puppetmaster  This shell script enables the puppetmaster server.
#
# Author:       Duane Griffin <d.griffin@psenterprise.com>
#
# chkconfig: - 65 45
#
# description: Server for the puppet system management tool.
# processname: puppetmaster

PATH=/usr/bin:/sbin:/bin:/usr/sbin
export PATH

# Source function library.
. /etc/rc.d/init.d/functions

if [ -f /etc/sysconfig/puppetmaster ]; then
        . /etc/sysconfig/puppetmaster
fi

prog=puppetmasterd
PUPPETMASTER=/usr/sbin/$prog
RUNDIR=/var/run/puppet
LOCKDIR=/var/lock/subsys/
lockfile=${LOCKDIR}/puppetmaster
PUPPETMASTERPORTS="8150 8151 8152 8153 8154"
PUPPETMASTER_OPTS=""

[ -n "$PUPPETMASTER_MANIFEST" ] && PUPPETMASTER_OPTS="--manifest=${PUPPETMASTER_MANIFEST}"
[ -n "$PUPPETMASTER_LOG" ] && PUPPETMASTER_OPTS="${PUPPETMASTER_OPTS} --logdest=${PUPPETMASTER_LOG}"
[ -n "$PUPPETMASTER_PORTS" ] && PUPPETMASTERPORTS=${PUPPETMASTER_PORTS}
case "$PUPPETMASTER_SERVER_TYPE" in
        MONGREL)
            FP="m"
            ;;
        *) # implied WEBBRICK
            FP="wb" 
esac
PUPPETMASTER_OPTS="${PUPPETMASTER_OPTS} ${PUPPETMASTER_EXTRA_OPTS}"

RETVAL=0


# wb_* functions => webbrick specific
# m_* functions  => mongrel specific

wb_start() {
        echo -n $"Starting puppetmaster: "

        # Confirm the manifest exists
        if [ -r $PUPPETMASTER_MANIFEST ]; then
                daemon $PUPPETMASTER $PUPPETMASTER_OPTS
                RETVAL=$?
        else
                failure $"Manifest does not exist: $PUPPETMASTER_MANIFEST"
                echo
                return 1
        fi
        [ $RETVAL -eq 0 ] && touch "$lockfile"
        echo
        return $RETVAL
}

wb_stop() {
        echo -n  $"Stopping puppetmaster: "
        killproc $PUPPETMASTER
        RETVAL=$?
        echo
        [ $RETVAL -eq 0 ] && rm -f "$lockfile"
        return $RETVAL
}

wb_status() {
        status $PUPPETMASTER
        RETVAL=$?
}

m_start() {
        local daemonargs=' --check $base --pidfile $pidfile'

        for pmp in $PUPPETMASTERPORTS ; do
                local pmargs="${PUPPETMASTER_OPTS} --servertype mongrel --masterport $pmp --pidfile \$pidfile"
                local base=puppetmaster-$pmp
                local pidfile=$RUNDIR/$base.pid
                local lockfile=$LOCKDIR/$base
        
                echo -n $"Starting $base: "

                eval daemon $daemonargs puppetmasterd $pmargs
                RETVAL=$?
                [ $RETVAL = 0 ] && touch ${lockfile}

                echo
        done
}

m_stop() {
        for pmp in $PUPPETMASTERPORTS ; do
            local base=puppetmaster-$pmp
            local pidfile=$RUNDIR/$base.pid
            local lockfile=$LOCKDIR/$base

            echo -n  $"Stopping $base: "

            killproc -p $pidfile $base
            RETVAL=$?
            [ $RETVAL -eq 0 ] && rm -f "$lockfile"

            echo
        done
}

m_status() {
        for pmp in $PUPPETMASTERPORTS ; do

            local base=puppetmaster-$pmp
            local pidfile=$RUNDIR/$base.pid

            status -p $pidfile $base
        done
}

start() {
    if [ "x${FP}" = "xm" ] ; then
        m_start
    elif  [ "x${FP}" = "xwb" ] ; then
        wb_start
    fi
}

stop() {
    if [ "x${FP}" = "xm" ] ; then
        m_stop
    elif  [ "x${FP}" = "xwb" ] ; then
        wb_stop
    fi
}

p_status() {
    if [ "x${FP}" = "xm" ] ; then
        m_status
    elif  [ "x${FP}" == "xwb" ] ; then
        wb_status
    fi
}

restart() {
    if [ "x${FP}" = "xm" ] ; then
        m_stop
        m_start
    elif  [ "x${FP}" = "xwb" ] ; then
        wb_stop
        wb_start
    fi
}

case "$1" in
  start)
        start
        ;;
  stop) 
        stop
        ;;
  restart|reload|force-reload)
        restart
        ;;
  condrestart)
        [ -f "$lockfile" ] && restart
        ;;
  status)
        p_status
        ;;
  *)
        echo $"Usage: $0 {start|stop|status|restart|reload|force-reload|condrestart}"
        exit 1
esac

exit $RETVAL