#!@@PERL@@ -w
#
# Graphs statistics for e-mails passing through a MailScanner filter
#
# Usage: Link into /etc/munin/plugins/
#
# Requirements:
# 
# 	The logtail command
#
# Parameters supported:
# 
# 	config (required)
#
# Configurable variables:
# 
# 	logfile		- The file where MailScanner logs its action (default: /var/log/mail.log)
# 	logtail		- The location of the logtail command (default: /usr/sbin/logtail)
# 	offsetfile	- The location of the offset file (default: /tmp/munin-mailscanner.offset)
#
# Bugs:
# 
# 	None known
#
# Magic markers:
#%# family=contrib
#%# capabilities=

use strict;

my $logfile = '/var/log/mail.log';
my $logtail = '/usr/sbin/logtail';
my $offsetfile = "@@PLUGSTATE@@/munin-mailscanner.offset";
my ($clean, $viruses, $spams, $others, $total) = (0, 0, 0, 0, 0);
my $cmd = (defined($ARGV[0])) ? $ARGV[0] : '';

(defined($ENV{'logfile'})) and $logfile = $ENV{'logfile'};
(defined($ENV{'logtail'})) and $logtail = $ENV{'logtail'};
(defined($ENV{'offsetfile'})) and $offsetfile = $ENV{'offsetfile'};

if ($cmd eq 'config') {
	print("graph_title MailScanner statistics\n");
	print("graph_args --lower-limit 0\n");
	print("graph_vlabel messages\n");
	print("clean.label clean\n");
	print("clean.type GAUGE\n");
	print("clean.draw LINE2\n");
	print("viruses.label viruses\n");
	print("viruses.type GAUGE\n");
	print("viruses.draw LINE2\n");
	print("spams.label spams\n");
	print("spams.type GAUGE\n");
	print("spams.draw LINE2\n");
	print("others.label others\n");
	print("others.type GAUGE\n");
	print("others.draw LINE2\n");
	print("total.label total\n");
	print("total.type GAUGE\n");
	print("total.draw LINE2\n");

	exit(0);
}

my @lines = split(/\n/, qx($logtail -f $logfile -o $offsetfile));
foreach (@lines) {
	SWITCH: {
		if (/New Batch: S/) {
			s/.*New Batch: Scanning ([0-9]+) messages.*/$1/;
			$total += $_;
			last SWITCH;
		}
		if (/Virus Scanning: F/) {
			s/.*Virus Scanning: Found ([0-9]+) viruses.*/$1/g;
			$viruses += $_;
			last SWITCH;
		}
		if (/Spam Checks: F/) {
			s/.*Spam Checks: Found ([0-9]+) spam.*/$1/g;
			$spams += $_;
			last SWITCH;
		}
		if (/Other Checks: F/) {
			s/.*Other Checks: Found ([0-9]+) problems.*/$1/g;
			$others += $_;
			last SWITCH;
		}
	}
}
$clean = $total - ($viruses + $spams + $others);

print("clean.value $clean\n");
print("viruses.value $viruses\n");
print("spams.value $spams\n");
print("others.value $others\n");
print("total.value $total\n");

exit(0);
