Ticket #1214: puppetmaster-mongrel

#!/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

#set -x

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

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"
            PUPPETMASTER_OPTS="${PUPPETMASTER_OPTS} --servertype mongrel"
            ;;
        *) # implied WEBBRICK
            lockfile=/var/lock/subsys/puppetmaster
            FP="wb" 
esac
PUPPETMASTER_OPTS="${PUPPETMASTER_OPTS} ${PUPPETMASTER_EXTRA_OPTS}"

RETVAL=0

prog=puppetmasterd
PUPPETMASTER=/usr/sbin/$prog
RUNDIR=/var/run/puppet
LOCKDIR=/var/lock/subsys/

# 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="--servertype mongrel --masterport $pmp --pidfile \$pidfile"
                m_start_proc puppetmaster-$pmp $daemonargs puppetmasterd $pmargs
        done

}

m_start_proc() {
        local base=$1
        local pidfile=$RUNDIR/$base.pid
        local lockfile=$LOCKDIR/$base
        shift
        
        echo -n $"Starting $base: "

        eval daemon $*
        RETVAL=$?
        [ $RETVAL = 0 ] && touch ${lockfile}

        echo
        return $RETVAL
}

m_stop() {
        for pmp in $PUPPETMASTERPORTS ; do
            m_stop_proc puppetmaster-$pmp
        done
}

m_stop_proc() {
        local base=$1
        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
        return $RETVAL
}


m_status() {
        for pmp in $PUPPETMASTERPORTS ; do
            m_status_proc puppetmaster-$pmp
        done
}

m_status_proc() {
        local base=$1
        local pidfile=$RUNDIR/$base.pid
        local lockfile=$LOCKDIR/$base

        status -p $pidfile $base
}

restart() {
        ${FP}_stop
        ${FP}_start
}

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

exit $RETVAL