#!/usr/bin/perl
#
# Munin plugin for total spot by categories in a spotweb MySQL database
#
# Copyright (C) 2011 - Rowdy Schwachöfer (http://rowdy.nl)
#
# Spotweb       : http://github.com/spotweb/spotweb
#
# 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_cat_total]
# env.countporn 'yes'                                    # Also count porn?
#
#

use strict;

# Environment variables
my $MYSQL 		= $ENV{mysql}     || "/usr/bin/mysql";
my $MYSQLOPTS 	= $ENV{mysqlopts} || "";
my $DATABASE 	= $ENV{database}  || "spotweb";
my $COUNTPORN 	= $ENV{countporn} || "yes";


# Output for config
if(defined $ARGV[0] && $ARGV[0] eq 'config') {
print <<EOC
graph_title Spotweb spots by Category (Totals)
graph_vlabel Spots
graph_category tv
cat0.label Videos/images
cat0.colour 0f94e1
cat1.label Music
cat1.colour efac26
cat2.label Games
cat2.colour 1ae31a
cat3.label Applications
cat3.colour e33b1a
EOC
;
exit 0;
}


if($COUNTPORN eq 'yes') { &print_value(0); }
else {
# 11-04-2011 - Added the new pr0n filters
#   &print_value(0, " AND subcatd NOT LIKE '%23%' AND subcatd NOT LIKE '%24%' AND subcatd NOT LIKE '%25%' AND subcatd NOT LIKE '%26%'");
    &print_value(0, " AND (subcatz IS NULL OR subcatz NOT LIKE '%z4%') AND subcatd NOT LIKE '%d23%' AND subcatd NOT LIKE '%d24%' AND subcatd NOT LIKE '%d25%' AND subcatd NOT LIKE '%d26%' AND subcatd NOT LIKE '%d72%' AND subcatd NOT LIKE '%d73%' AND subcatd NOT LIKE '%d74%' AND subcatd NOT LIKE '%d75%' AND subcatd NOT LIKE '%d76%' AND subcatd NOT LIKE '%d77%' AND subcatd NOT LIKE '%d78%' AND subcatd NOT LIKE '%d79%' AND subcatd NOT LIKE '%d80%' AND subcatd NOT LIKE '%d81%' AND subcatd NOT LIKE '%d82%' AND subcatd NOT LIKE '%d83%' AND subcatd NOT LIKE '%d84%' AND subcatd NOT LIKE '%d85%' AND subcatd NOT LIKE '%d86%' AND subcatd NOT LIKE '%d87%' AND subcatd NOT LIKE '%d88%' AND subcatd NOT LIKE '%d88%' AND subcatd NOT LIKE '%d89%'");
}
&print_value(1);
&print_value(2);
&print_value(3);

sub print_value {
    my $spots = `$MYSQL $MYSQLOPTS -e "SELECT count(*) as spots FROM $DATABASE.spots WHERE category = $_[0]$_[1];"`;
    $spots =~ /(\d+)/;
    print "cat".$_[0].".value ".$1."\n";
}