#!/bin/bash
#
# plc	Manages all PLC services on this machine
#
# chkconfig: 2345 5 99
#
# description:	Manages all PLC services on this machine
#
# $Id: guest.init,v 1.20 2006/08/08 23:19:52 mlhuang Exp $
#

# Source function library and configuration
. /etc/plc.d/functions

# Verbosity
verbose=0

# All steps should be idempotent. This means that you should be able
# to run them multiple times without depending on anything previously
# being run. The idea is that when the configuration changes, "service
# plc restart" is called, all dependencies are fixed up, and
# everything just works.
steps=($(
for step in /etc/plc.d/* ; do
    if [ -f $step -a -x $step ] ; then
	priority=$(sed -ne 's/# priority: \(.*\)/\1/p' $step)
	echo $priority $(basename $step)
    fi
done | sort -n | cut -d' ' -f2
))
nsteps=${#steps[@]}

# Regenerate configuration files
reload ()
{
    force=$1

    # Regenerate the main configuration file from default values
    # overlaid with site-specific and current values.
    files=(
	/etc/planetlab/default_config.xml 
	/etc/planetlab/configs/*
	/etc/planetlab/plc_config.xml
    )
    for file in "${files[@]}" ; do
	if [ -n "$force" -o $file -nt /etc/planetlab/plc_config.xml ] ; then
	    tmp=$(mktemp /tmp/plc_config.xml.XXXXXX)
	    plc-config --xml "${files[@]}" >$tmp
	    if [ $? -eq 0 ] ; then
		mv $tmp /etc/planetlab/plc_config.xml
		chmod 644 /etc/planetlab/plc_config.xml
	    else
		echo "PLC: Warning: Invalid configuration file(s) detected"
		rm -f $tmp
	    fi
	    break
	fi
    done

    # Convert configuration to various formats
    if [ -n "$force" -o /etc/planetlab/plc_config.xml -nt /etc/planetlab/plc_config ] ; then
	plc-config --shell >/etc/planetlab/plc_config
    fi
    if [ -n "$force" -o /etc/planetlab/plc_config.xml -nt /etc/planetlab/php/plc_config.php ] ; then
	mkdir -p /etc/planetlab/php
	plc-config --php >/etc/planetlab/php/plc_config.php
    fi
}

usage()
{
    echo "Usage: $0 [OPTION]... [COMMAND] [STEP]..."
    echo "	-v		Be verbose"
    echo "	-h		This message"
    echo
    echo "Commands:"
    echo "	start		Start all PLC subsystems"
    echo "	stop		Stop all PLC subsystems"
    echo "	reload		Regenerate configuration files"
    echo "	restart		Restart all PLC subsystems"
    echo
    echo "Steps:"
    for step in "${steps[@]}" ; do
	if [ -x /etc/plc.d/$step ] ; then
	    echo "	$(basename $step)"
	fi
    done
    exit 1
}

# Get options
while getopts "vh" opt ; do
    case $opt in
	v)
	    verbose=1
	    ;;
	h|*)
	    usage
	    ;;
    esac
done

# Redirect stdout and stderr of each step to /var/log/boot.log
if [ $verbose -eq 0 ] ; then
    touch /var/log/boot.log
    chmod 600 /var/log/boot.log
    exec 1>>/var/log/boot.log
    exec 2>>/var/log/boot.log
fi

# Get command
shift $(($OPTIND - 1))
if [ -z "$1" ] ; then
    usage
fi
command=$1

# Get step(s)
shift 1
if [ -z "$1" ] ; then
    # Start or stop everything. Regenerate configuration first.
    reload force
else
    # Start or stop a particular step
    steps=("$@")
    nsteps=${#steps[@]}
fi

RETVAL=0

start ()
{
    for step in "${steps[@]}" ; do
	if [ -x /etc/plc.d/$step ] ; then
	    /etc/plc.d/$step start
	    # Steps may alter the configuration, may need to regenerate
	    reload
	else
	    echo "PLC: $step: unrecognized step" >&4
	    exit 1
	fi
    done
}

stop ()
{
    for i in $(seq 1 $nsteps) ; do
	step=${steps[$(($nsteps - $i))]}
	if [ -x /etc/plc.d/$step ] ; then
	    /etc/plc.d/$step stop
	    # Steps may alter the configuration, may need to regenerate
	    reload
	else
	    echo "PLC: $step: unrecognized step" >&4
	    exit 1
	fi
    done
}

case "$command" in
    start|stop)
	$command
	;;

    restart)
	stop
	start
	;;

    reload)
	reload force
	;;

    *)
	usage >&3
	;;
esac

exit $RETVAL