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


mktempfile () {
    mktemp -t "$1"
}

LOGFILE=${logfile:-/var/log/proftpd/xferlog}
LOGTAIL=${logtail:-$(which logtail)}
STATEFILE=$MUNIN_PLUGSTATE/xferlog-bytes.offset

if [ "$1" = "autoconf" ]; then
    if [ -f "$LOGFILE" ] && [ -n "$LOGTAIL" ] && [ -x "$LOGTAIL" ] ; then
            echo yes
    else
            echo "no (missing logfile or 'logtail' executable)"
    fi
    exit 0
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 network'
    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" ] && [ -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" }')

        /bin/rm -f "$TEMP_FILE"
fi

echo "ftp_get.value $ftp_get"
echo "ftp_put.value $ftp_put"