#!/usr/bin/perl
#
# Munin plugin for different stats over a spotweb MySQL database
#
# Copyright (C) 2011 - Rowdy Schwachöfer (http://rowdy.nl)
#
# Spotweb       : http://github.com/spotweb/spotweb
# Original idea : smeerbartje (http://gathering.tweakers.net/forum/myreact/190949)
#                 slommer     (http://gathering.tweakers.net/forum/myreact/353335)
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
#
#
# Configuration example
#
# [spotweb*]
# env.mysql /usr/bin/mysql                               # MySQL binary (optional)
# env.mysqlopts -u <MYSQL_USERNAME> -p<MYSQL_PASSWORD>   # How to connect to the database (optional if no password is set)
# env.database spotweb                                   # Spotweb database (optional)
#
# [spotweb_total]
# env.spotsfull yes                                      # Also show Full spots?
#
#

use strict;

my $MYSQL = $ENV{mysql} || "/usr/bin/mysql";
my $MYSQLOPTS = $ENV{mysqlopts} || "";
my $DATABASE = $ENV{database} || "spotweb";
my $SPOTSFULL = $ENV{spotsfull} || "yes";

# Output for config
if(defined $ARGV[0] && $ARGV[0] eq 'config') {
    print <<EOC
graph_title Spotweb spots and comments (Totals)
graph_vlabel Spots/comments
graph_category tv
spots.label Spots
EOC
;

    if($SPOTSFULL eq 'yes') {
        print <<EOC
spotsfull.label Full spots
EOC
;
    }

    print <<EOC
comments.label Comments
EOC
;
        exit 0;
}

# Spot count
my $spots = `$MYSQL $MYSQLOPTS -e 'SELECT count(*) as spots FROM $DATABASE.spots;'`;
$spots =~ /(\d+)/;
print "spots.value ".$1."\n";

if($SPOTSFULL eq 'yes') {
    my $spotsfull = `$MYSQL $MYSQLOPTS -e 'SELECT count(*) as fullspots FROM $DATABASE.spotsfull;'`;
    $spotsfull =~ /(\d+)/;
    print "spotsfull.value ".$1."\n";
}

# Comment count
my $comments = `$MYSQL $MYSQLOPTS -e 'SELECT count(*) as comments FROM $DATABASE.commentsxover'`;
$comments =~/(\d+)/;
print "comments.value ".$1."\n";
