#!/usr/bin/bash
#
# Plugin to monitor FTP bytes.
# based on previous work by jintxo
#
# Parameters understood:
#
# 	config   (required)
# 	autoconf (optional)
#
# Magic markers (optional - used by munin-config and installation
# scripts):
#
#%# family=auto
#%# capabilities=autoconf

MAXLABEL=20
PATH=/opt/local/bin:${PATH}

mktempfile () {
mktemp -t $1
}

LOGFILE=${logfile:-/var/log/proftpd/xferlog}
LOGTAIL=${logtail:-`which logtail`}
STATEFILE=/opt/local/var/munin/plugin-state/xferlog-bytes.offset

if [ "$1" = "autoconf" ]; then
	if [ -f "${LOGFILE}" -a -n "${LOGTAIL}" -a -x "${LOGTAIL}" ] ; then
		echo yes
		exit 0
	else
		echo no
		exit 1
	fi
fi

if [ "$1" = "config" ]; then
	echo 'graph_title FTP Server Bytes'
	echo 'graph_args --base 1000 -l 0'
	echo 'graph_vlabel FTP Server Bytes'
	echo 'graph_category FTP'
	echo 'ftp_get.label Bytes GET'
	echo 'ftp_put.label Bytes PUT'
	exit 0
fi


ftp_get=U
ftp_put=U

TEMP_FILE=`mktempfile munin-xferlog-bytes.XXXXXX`

if [ -n "$TEMP_FILE" -a -f "$TEMP_FILE" ]
then
	$LOGTAIL ${LOGFILE} $STATEFILE | grep "[[:space:]][oi][[:space:]]" > ${TEMP_FILE}
	ftp_get=`grep "[[:space:]]o[[:space:]]" ${TEMP_FILE} | awk '{s += $8} END { if ( s ) print s ; else print "0" }'`
	ftp_put=`grep "[[:space:]]i[[:space:]]" ${TEMP_FILE} | awk '{s += $8} END { if ( s ) print s ; else print "0" }'`

	rm -f $TEMP_FILE
fi

echo "ftp_get.value ${ftp_get}"
echo "ftp_put.value ${ftp_put}"