#!/bin/sh
#
# Author: Petter Reinholdtsen

set -e

nodes=/usr/sbin/sitesummary-nodes
muninopts=""

# Modify this in collector.cfg to /etc/munin/ to automatically replace
# the default munin configuration.
MUNINDIR=/var/lib/sitesummary

if [ -d /var/cache/munin/www ] ; then
    # Used since munin version 1.4.0-1
    muninhtmldir="/var/cache/munin/www"
else
    muninhtmldir="/var/www/munin"
fi
# Based on of active config from munin version 1.2.5-1
munindbdir=/var/lib/munin
muninlogdir=/var/log/munin
muninrundir=/var/run/munin
munintmpldir=/etc/munin/templates

[ -f /etc/sitesummary/collector.cfg ] && . /etc/sitesummary/collector.cfg

# The storage area is not configurable, because too many scripts have
# it hardcoded
entriesdir=/var/lib/sitesummary/entries

generate_munin_config() {
    # Generate munin config.  Edit /etc/cron.d/munin to enable it.
    # Add -c /var/lib/sitesummary/sitesummary-munin.conf to the calls
    # to the munin scripts, or change MUNINDIR above.
    (
	if [ -f $MUNINDIR/munin.conf.pre ] ; then
	    cat $MUNINDIR/munin.conf.pre
	else
	    cat <<EOF
# Munin server configuration generated from cron using sitesummary
# data by $0
# Do not edit, it will be overwritten. 
# Edit $MUNINDIR/munin.conf.pre and
# $MUNINDIR/munin.conf.post instead.

dbdir	$munindbdir
htmldir	$muninhtmldir
logdir	$muninlogdir
rundir	$muninrundir
tmpldir	$munintmpldir

EOF

            # Munin supports the includedir statement since Squeeze
            muninver=$(dpkg -l munin|grep munin|awk '{print $3}')
            if dpkg --compare-versions "1.4.5" le "$muninver" ; then
	        cat <<EOF
# (Exactly one) directory to include all files from.
#
includedir /etc/munin/munin-conf.d

EOF
            fi
	fi

	$nodes -m $muninopts

	[ -f $MUNINDIR/munin.conf.post ] && cat $MUNINDIR/munin.conf.post

        # Make sure the subshell return true to trigger the mv below.
	true
    ) > $MUNINDIR/munin.conf.new && \
	chown root:root $MUNINDIR/munin.conf.new && \
	chmod a+r $MUNINDIR/munin.conf.new && \
	mv $MUNINDIR/munin.conf.new $MUNINDIR/munin.conf
}

# Only enable if munin and sitesummary is installed.
if [ -f /etc/munin/munin.conf ] && [ -x /usr/sbin/sitesummary-nodes ]; then
    generate_munin_config
fi