#!/usr/bin/perl

############################################################
#
# Config Section


# The @server_list array is just that - duh.
# The "Local" value is a keyword and refers to the local machine.

@server_list = ("Local", "host1.server.com", "host2.server.com.com", "host3.server.com", "host4.server.com");

# This is the warning level (in percent). When a filesystem passes
# this percentage in used space, a warning report is generated

$warning_threshold = 80;

############################################################

if ($ARGV[0] eq "-?" || $ARGV[0] eq "--help")
{
        print "Usage: diskfree [-w] [-? | --help]\n";
        print "\t-w\t\t Display warnings only\n";
        print "\t-? | --help\t Usage screen\n";
        exit 1;
}

$warnings = "";
$detail_report = "";

foreach $server (@server_list)
{
        $detail_report = $detail_report . "$server Disk Usage\n";
        $detail_report = $detail_report . "------------------------------------------\n";
        $detail_report = $detail_report . "Use% |  Mount  | Size | Avail\n";
        $detail_report = $detail_report . "------------------------------------------\n";

        if ($server eq "Local") {
                @diskusage = `df -hP`;
        } else {
                @diskusage = `ssh "$server" "df -hP"`;
        }

        for ($i = 1; $i <= $#diskusage; $i++)
        {
                @usage_detail = split (/\s+/,$diskusage[$i]);

                $usage_detail[4] =~ s/%//g;

                if ($usage_detail[4] > $warning_threshold)
                {
                        $warnings = $warnings . "*WARNING* $server: $usage_detail[5] ($usage_detail[0]) is at $usage_detail[4]%\n";
                        $detail_report = $detail_report . "$usage_detail[4]% \t $usage_detail[5] \t $usage_detail[1] \t $usage_detail[3]" . "\n";
                } else {
                        $detail_report = $detail_report . "$usage_detail[4]% \t $usage_detail[5] \t $usage_detail[1] \t $usage_detail[3]" . "\n";
                }
        }

        $detail_report = $detail_report . "------------------------------------------\n\n";

}

if ($ARGV[0] eq "-w")
{
        if ($warnings eq "")
        {
                print "No warnings.\n";
        } else {
                print "$warnings\n";
        }
} else {
        print "$warnings\n";
        print "$detail_report\n";
}