#!/usr/bin/perl -w
use strict;
use File::Find;
use File::stat;
use Getopt::Std;
my %opts;
getopts('hd:p:s:t:rm',\%opts);
usage() if ( $opts{h} || ( !$opts{d} ) );
my ($directory) = ("")x1;
if ($opts{d}) { $directory = $opts{'d'}; }
if ($opts{r}) { report(); }
if ($opts{m}) { manage(); }
my @dirList = "";
sub usage
{
print "Usage:\n\n";
print "\t dirmonitor -d <dir>";
}
sub report
{
my ($owner,@ownersDirs,@ownersDirSizes,@fileList,%HoA_Sizes,%HoA_Dirs,@tokens,$user,$file,$dir,$size,$cat,$sum,%dirHash) = ("")x14;
print "Inspecting directory: $directory ...\n\n";
#find( {wanted => \&wanted, dangling_symlinks => 1 }, $directory);
find(\&wanted, $directory);
#print "values: @dirList\n";
# Create a hash of arrays with our newly found directories, each array in the
# hash contains directories that the user owns. Each array is keyed by the
# owners user name.
foreach (@dirList) {
$dir = $_;
$user = getpwuid(stat($dir)->uid);
# split up the info that du returns into tokens, resuse @tokens array
#@tokens = split(' ', qx(du -S "$dir"), 2);
#$size = $tokens[0];
$size = -s $dir;
# push the sizes of the users directories onto users stack
push @{$HoA_Sizes{$user}}, $size;
push @{$HoA_Dirs{$user}}, $dir;
}
# sum up the sizes of directories that the user owns
foreach $owner ( keys %HoA_Sizes ) {
# dereference the array reference and grab the list of values
@ownersDirSizes = @{ $HoA_Sizes{$owner} };
@ownersDirs = @{ $HoA_Dirs{$owner} };
@dirHash{ @ownersDirs } = @ownersDirSizes;
#for my $i (0 .. $#ownersDirs) {
# if ($i ne $directory) {
# $sum += $ownersDirSizes[$i];
# $dirHash{$ownersDirs[$i]} = $ownersDirSizes[$i];
# print "$ownersDirSizes[$i] dir: $ownersDirs[$i]\n";
# }
#}
foreach my $toke ( keys %dirHash ) {
$sum += $dirHash{$toke};
#print "$dirHash{$toke}KB\t $toke\n";
}
push @{$HoA_Sizes{$owner}}, $sum;
#print "sum is: $sum\n";
$sum = 0;
}
# sort the hash of arrays by comparing the first value of a users
# array(sum of all users dirs)
foreach $owner ( sort { $HoA_Sizes{$b}->[-1] <=> $HoA_Sizes{$a}->[-1] } keys %HoA_Sizes ) {
# print "$owner: @{[int($HoA_Sizes{$owner}->[-1]/1000000)]}MB\n";
#print "$owner: @{[$HoA_Sizes{$owner}->[-1]/1000000]}MB\n";
print "$owner: ";
printf "%2.2f", @{[$HoA_Sizes{$owner}->[-1]/1000000]};
print "MB\n";
}
}
sub wanted {
#if (-d $_) {
if ($_ ne ".") {
#print "we have: $_ and $directory\n";
push(@dirList,$File::Find::name);
}
#}
}