#!/usr/bin/perl -W
#
# [ML] SOURCE: http://munin.projects.linpro.no/attachment/ticket/143/quota-usage 
#
# Plugin to monitor quota on a specified device. Needs repquota and root privs
#
# Usage: place in /etc/munin/node.d/quota-usage_<dev> (or link it there using
# ln -s), for example quota-usage_hda3. Use underscores instead of slashes, for
# example to monitor /dev/mapper/vol-foo, name this quota-usage_mapper_vol-foo
#
# This plugin uses the soft and hard quota data for warning and critical levels
# respectively.
#
# Parameters understood:
#
#       config   (required)
#
#%# family=manual
#
use strict;
use warnings;

# We do some magic to allow device strings with underscore to mean a /
# So to monitor /dev/mapper/vol-foo use quota-usage_mapper_vol-foo
my @tmp = split(/_/, $0);
shift @tmp;
my $dev = join("/", @tmp);
my %users;

open (REP, "/usr/sbin/repquota /dev/$dev |") or exit 22;
while (<REP>) {
        chomp;
        if (/^-+$/../^$/) {
                my @data = split(/ +/);
                if ( @data > 2 && $data[2] > 0 ) {
                    $users{$data[0]}[0] = $data[2]; # current usage
                    $users{$data[0]}[1] = $data[3]; # Soft quota
                    $users{$data[0]}[2] = $data[4]; # Hard quota
                }
        }
}
close REP;


my @sorted_users;
@sorted_users = sort keys %users;    # sorted by username



my $user;
my @value;

if ($ARGV[0] and $ARGV[0] eq "config") {
        print "graph_title Filesystem usage by user (quota based) on $dev\n";
        print "graph_args --base 1024 --lower-limit 0\n";
        print "graph_vlabel bytes\n";
        print "graph_category disk\n";

        foreach (@sorted_users) {
            my $user = $_;
            my ($uid, $name) = (getpwnam($user))[2,6];
			my $esc = $user;
            $esc =~ s/-/_/g;
            $name = (split(/,/, $name))[0];
            printf "%s.label %s\n", $esc, $name;
            if ( $users{$user}[1] > 0 ) {
                printf "%s.warning %s\n", $esc, $users{$user}[1];
            }
            if ( $users{$user}[2] > 0 ) {
                printf "%s.critical %s\n", $esc, $users{$user}[2];
            }
            printf "%s.cdef %s,1024,*\n", $esc, $esc;
            if (($uid < 500)) {
                printf "%s.graph no\n", $esc;
            }
        }
} else {
        foreach (@sorted_users) {
            my $user = $_;
            my $esc = $user;
            $esc =~ s/-/_/g;
            printf "%s.value %s\n", $esc, $users{$user}[0];
        }
}

# vim: set ts=4 sw=4: