#!/usr/bin/perl -wT

# Copyright (c) 2005, 2006, 2007, 2008 Peter Palfrader <peter@palfrader.org>
#
# Permission is hereby granted, free of charge, to any person obtaining
# a copy of this software and associated documentation files (the
# "Software"), to deal in the Software without restriction, including
# without limitation the rights to use, copy, modify, merge, publish,
# distribute, sublicense, and/or sell copies of the Software, and to
# permit persons to whom the Software is furnished to do so, subject to
# the following conditions:
#
# The above copyright notice and this permission notice shall be
# included in all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
# LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
# OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
# WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.


use strict;
use English;
use File::Basename;

my $MAX_LINES = 4;
my $ENVELOPE_FROM = 'nobody@commit.noreply.org';
my $HEADER_FROM = 'nobody@commit.noreply.org';
my $BOT_ADDRESS = 'commit@commit.noreply.org';
my $SENDMAIL = '/usr/sbin/sendmail';

$ENV{'PATH'} = '/bin:/usr/bin';
delete @ENV{'IFS', 'CDPATH', 'ENV', 'BASH_ENV'};

die ("Usage: $PROGRAM_NAME <project> <mailfile>\n") unless (scalar @ARGV == 2);
my $project = $ARGV[0];

umask 077;


$project =~ m/^([a-zA-Z0-9-]+(\.[a-zA-Z0-9-]+)*)+$/;
$project = $1;
die ("Project is not a nice name.\n") unless defined $project;


my $type;
my $service;
my $host;
my $state;
my $info;
my $running_number;
my $ack_by;
my $comment;
my $hosttype;

open (MAIL, $ARGV[1]) || die ("Cannot open $ARGV[1]: $!\n");
my @mail = <MAIL>;
close(MAIL);

my $line;
while (defined($line = shift @mail)) {
	if ($line =~ m/^Subject:.*?alert ([0-9]+) - /i) {
		$running_number = $1;
	} elsif ($line =~ m/^Notification Type:\s*(.*)/) {
		$type = $1;
	} elsif ($line =~ m/^Service:\s*(.*)/) {
		$service = $1;
	} elsif ($line =~ m/^Host:\s*(.*)/) {
		$host = $1;
	} elsif ($line =~ m/^State:\s*(.*)/) {
		$state = $1;
	} elsif ($line =~ m/^ACK by:\s*(.*)/) {
		$ack_by = $1;
	} elsif ($line =~ m/^Comment:\s*(.*)/) {
		$comment = $1;
	} elsif ($line =~ m/^(Additional )?Info:/) {
		while (defined($line = shift @mail) && !($line =~ /\S/)) {
			# null
		}
		if (defined $line) {
			chomp $line;
			$info = $line;
		};
	}
};
die ("$PROGRAM_NAME - $project: No type found.\n") unless defined $type;
# die ("$PROGRAM_NAME - $project: No service found.\n") unless defined $service;
die ("$PROGRAM_NAME - $project: No host found.\n") unless defined $host;
die ("$PROGRAM_NAME - $project: No state found.\n") unless defined $state;
die ("$PROGRAM_NAME - $project: No info found.\n") unless defined $info;

$type =~ tr/A-Z/a-z/;
$service = $host unless defined $service;

if ($state =~ m/CRITICAL|DOWN|UNREACHABLE/ ) {
    $hosttype = "5$host";
} elsif ($state =~ m/WARNING/) {
    $hosttype = "7$host";
} elsif ($state =~ m/OK|UP/ ) {
    $hosttype = "3$host";
} elsif ($state =~ m/UNKNOWN/) {
    $hosttype = "10$host";
} else {
    $hosttype = "$type\@$host";
}

my $extra = (defined $running_number) ? " (#$running_number)" : "";

#open(MAIL, "|cat") or
open(MAIL, "|$SENDMAIL -t -oi -f $ENVELOPE_FROM") or
	die ("Cannot exec sendmail: $!\n");
print MAIL "From: $HEADER_FROM\n";
print MAIL "To: $BOT_ADDRESS\n";
print MAIL "Subject: Announce $project\n";
print MAIL "Precedence: junk\n";
print MAIL "\n";
print MAIL "[$hosttype] $service is $state$extra: $info\n";
print MAIL "ACK by: $ack_by\n" if ($ack_by);
print MAIL "Comment: $comment\n" if ($comment);
close(MAIL);