#!/usr/bin/perl -Tw

#
#

use strict;
use DBI;
use Config::IniFiles;
use Getopt::Long;
use vars qw($logfile $configfile $quiet $help $version @nodes);
use POSIX qw(strftime);

my $_version = '1.0';

my %options = (
        "--config-file|C=s"     =>      \$configfile,
        "--quiet|q"             =>      \$quiet,
        "--help|h"              =>      \$help,
        "--version|v"           =>      \$version,
	"--node=s" 		=> 	\@nodes,
);

Getopt::Long::config("no_ignore_case");
GetOptions(%options);

@nodes=split(/,/,join(',',@nodes));

if($help)
{
        print STDERR <<EOF;
lms-mrtg, version $_version
(C) 2005 LMS Developers

-C, --config-file=/etc/lms/lms.ini      alternate config file (default: /etc/lms/lms.ini);
-h, --help                      print this help and exit;
-v, --version                   print version info and exit;
-q, --quiet                     suppress any output, except errors;

EOF
        exit 0;
}

if($version)
{
        print STDERR <<EOF;
lms-mrtg, version $_version
(C) 2001-2004 LMS Developers

EOF
        exit 0;
}

if(!$configfile)
{
        $configfile = "/etc/lms/lms.ini";
}

if(! -r $configfile)
{
        print STDERR "Fatal error: Unable to read configuration file $configfile, exiting.\n";
        exit 1;
}


my $ini = new Config::IniFiles -file => $configfile;
my $dbtype = $ini->val('database', 'type') || 'mysql';
my $dbhost = $ini->val('database', 'host') || 'localhost';
my $dbuser = $ini->val('database', 'user') || 'root';
my $dbpasswd = $ini->val('database', 'password') || '';
my $dbname = $ini->val('database', 'database') || 'lms';

my $dbase;

if($dbtype eq "mysql")
{
        $dbase = DBI->connect("DBI:mysql:database=$dbname;host=$dbhost","$dbuser","$dbpasswd", { RaiseError => 1
});
}
elsif($dbtype eq "postgres")
{
        $dbase = DBI->connect("DBI:Pg:dbname=$dbname;host=$dbhost","$dbuser","$dbpasswd", { RaiseError => 1 });
}
else
{
        print STDERR "Fatal error: unsupported database type: $dbtype, exiting.\n";
        exit 1;
}

if (@nodes) {
	my $nodelist="nodeid=";
	$nodelist.=pop(@nodes);
	foreach my $nodeid (@nodes) {
		$nodelist="$nodelist OR nodeid=$nodeid";
	};
	
	my $dbq=$dbase->prepare("SELECT SUM(download) AS download,SUM(upload) AS upload FROM stats WHERE ($nodelist) AND dt>UNIX_TIMESTAMP()-300");
	$dbq->execute();
	my $row = $dbq->fetchrow_hashref();
	if ($row->{'upload'}) {
		print "$row->{'upload'}\n"
	} else {
		print "0\n";
	}
	if ($row->{'download'}) {
		print "$row->{'download'}\n";
	} else {
		print "0\n";
	}
	print "gateway.perfect.net.pl\n";
	print "-\n";
	$dbq->finish;
	$dbase->disconnect();
}