#!/bin/sh
#
# Plugin to monitor CPU usage, for a selected set of processes. Tested on FreeBSD.
#
# Author: Erik Cederstrand
# Based on http://waste.mandragor.org/munin_tutorial/cpubyuser
# Thanks to Yann Hamon.
#
# Usage: Place in /usr/local/etc/munin/plugins/ (or link it there  using ln -s)
#        Add this to your /ur/local/etc/munin/plugin-conf.d/plugins.conf:
#       [cpubyproc]
#       env.procs httpd java
#
#    httpd and java being a list of the processes to monitor.

#
# Parameters understood:
#
#       config   (required)
#       autoconf (optional - used by munin-config)
#

#%# family=auto
#%# capabilities=autoconf



if [ "$1" = "autoconf" ] ; then 
    if [ -n "$procs" ] ; then
        echo "yes"
    else
        echo "\$procs not defined."
    fi
    exit
fi

if [ "$1" = "config" ] ; then
    echo "graph_args --base 1000 -r --lower-limit 0";
    echo "graph_title CPU usage, by process";
    echo "graph_category processes";
    echo "graph_info This graph shows CPU usage, for monitored processes.";
        echo 'graph_vlabel %'
        echo 'graph_scale no'
        echo 'graph_period second'

    echo "graph_order $procs"
    
    FIRSTPROC=1;
    for proc in $procs; do  
        echo "${proc}.label $proc"
        echo "${proc}.info CPU used by process $proc"
        echo "${proc}.type GAUGE"
        if [ $FIRSTPROC -eq 1 ] ; then
            echo "${proc}.draw AREA"
            export FIRSTPROC=0;
        else    
            echo "${proc}.draw STACK"
        fi
    done ;
    
    exit
fi



for proc in $procs ; do {

    ps axo 'pcpu,comm' | grep "$proc" |
    awk '
    BEGIN {
        FS=" "
        CPU_PROC=0
    }

    {
        CPU_PROC+=$0
    }

    END { 
    print  "'$proc'.value "CPU_PROC
    }'  
}

done;