Munin contributed plugins

#!/usr/bin/perl

use strict;

use warnings;

my $s3_id = exists $ENV{'s3_id'} ? $ENV{'s3_id'} : "user";

my $s3curl = "perl s3-curl/s3curl.pl --id $s3_id -- -s -S";

sub get_bucket_list()

{

    my $buckets = `$s3curl http://s3.amazonaws.com`;

    my $str = $buckets;

    my @bucket_list;

    while ($buckets =~ s/.<Name>([\w._-]+)<\/Name>//)

	 {

        push @bucket_list, $1;

    }

    return @bucket_list;

}

my @bucket_list = split /\s+/, ($ENV{'buckets'} || '');

if (not @bucket_list)

{

    @bucket_list = get_bucket_list();

}

if ($ARGV[0] and $ARGV[0] eq "autoconf")

{

    if (@bucket_list)

	 {

        print "yes\n";

        exit 0;

    }

	 else

	 {

        print "no\n";

        exit 1;

    }

}

sub get_bucket_stats

{

    my ($name) = @_;

    my $stats = `$s3curl http://$name.s3.amazonaws.com`;

    my %res;

    $res{'size'}  = 0;

    while ($stats =~ s/.<Size>([\w._-]+)<\/Size>//)

	 {

        $res{'size'} += $1;

		  $res{'count'}++;

    }

    return %res;

}

if ($ARGV[0] and $ARGV[0] eq "config")

{

    # The headers

    print "graph_title S3 Storage Usage\n";

    print "graph_category cloud\n";

    print "graph_args --base 1024 -l 0\n";

    print "graph_vlabel bytes\n";

    print "graph_info Plugin available at <a href='https://github.com/aptivate/munin-contrib/blob/master/plugins/s3/s3_storage'>https://github.com/aptivate/munin-contrib/blob/master/plugins/s3/s3_storage</a>\n";

    foreach my $bucket_name (@bucket_list)

	 {

        print "$bucket_name.label Bucket $bucket_name\n";

    }

    exit;

}

foreach my $bucket_name (@bucket_list)

{

    my %stats = get_bucket_stats($bucket_name);

    print "$bucket_name.value " . $stats{'size'} . "\n";

}

exit 0;