allow for simple service monitoring via tagged members

[moira.git] / gen / nagios-wsh.gen

#!/moira/bin/perl -Tw

# $Id$
# The following exit codes are defined and MUST BE CONSISTENT with the
# error codes the library uses:
$MR_DBMS_ERR = 47836421;
$MR_OCONFIG = 47836460;

$outdir = '/moira/dcm/nagios-wsh';

use DBI;

$dbh = DBI->connect("dbi:Oracle:moira", "moira", "moira")
    || exit $MR_DBMS_ERR;

$sth0 = $dbh->prepare("SELECT l.list_id, m.name " .
                      "FROM list l, machine m, serverhosts sh " .
                      "WHERE sh.value3 = l.name AND sh.service = " .
		      "'NAGIOS-WSH' AND m.mach_id = sh.mach_id")
    || exit $MR_DBMS_ERR;
$sth0->execute;

while (($root_list_id, $hostname) = $sth0->fetchrow_array) {
    umask 022;
    open(OUT, ">$outdir/$hostname") || exit $MR_OCONFIG;
    print OUT "# This file is automatically generated by Moira.  Do not edit.\n";
    $sth = $dbh->prepare("SELECT m.name, i.tag FROM machine m, imembers i " .
			 "WHERE i.list_id = " . $dbh->quote($root_list_id) .
			 "AND i.member_type = 'MACHINE' AND m.status = 1 " .
			 "AND i.member_id = m.mach_id AND i.direct = 1 ORDER BY m.name") 
	|| exit $MR_DBMS_ERR;
    $sth->execute;

    while (($name, $tag) = $sth->fetchrow_array) {
	next if $name eq "[NONE]";
	$name = lc($name);
	push(@allwshhosts, $name);
	print OUT <<END;
define host{
	host_name		$name
	alias			$name
	address			$name
	use			generic-host
	}

define service{
  	host_name		$name
	contact_groups		wsh
	use			ping-service
	}

define hostescalation{
        host_name               $name
        contact_groups          wsh,wsh-mail
        first_notification      2
        last_notification       0
        notification_interval   0
        }

define serviceescalation{
        host_name               $name
        contact_groups          wsh,wsh-mail
        service_description     PING
        first_notification      2
        last_notification       0
        notification_interval   0
        }

END

	next if $tag == 0; 
	$sth1 = $dbh->prepare("SELECT s.string " .
			      "FROM strings s " .
			      "WHERE s.string_id = " . $tag)
	    || exit $MR_DBMS_ERR;
	$sth1->execute;
	my @tags = split /\s+/,$sth1->fetchrow_array;
	$sth1->finish;
	my %services;
	while (<@tags>) {
	    my $service = $_;
	    chomp $service;
	    if ($service =~ /^(FTP|NFS|HTTP|HTTPS|SMTP|SSHD|TELNET|TNS)$/i) {
		my $ucservice = $service;
		$ucservice =~ tr/a-z/A-Z/;
		my $lcservice = $service;
		$lcservice =~ tr/A-Z/a-z/;
		$services{$ucservice}++;
		print OUT <<END;
define service{
	host_name		$name
	contact_groups		wsh
	use			$lcservice-service
	}

define serviceescalation{
	host_name		$name
	contact_groups		wsh,wsh-mail
	service_description	$ucservice
	first_notification	2
	last_notification	0
	notification_interval	0
	}

END
	    } elsif ($service =~ /^HTTPS-CERT$/i) {
		$services{'HTTPS-CERT'}++;
		print OUT <<END;
define service{
        host_name               $name
        contact_groups          wsh-mail
        use                     https-cert-service
        }

END
	    } else {
		printf STDERR "Machine %s has unknown service %s; ignoring\n", $name, $service;
	    }

	    if ($services{'HTTPS'} && $services{'HTTPS-CERT'}) {
		print OUT <<END;
define servicedependency{
	dependent_host_name		$name
	dependent_service_description	HTTPS-CERT
	host_name			$name
	service_description		HTTPS
	execution_failure_criteria	n
	notification_failure_criteria	w,u,c
	}

END
	    }
	}
    } # while $sth->fetchrow_array

    print OUT <<END;
define hostgroup{
        hostgroup_name          wsh-hosts
        alias                   wsh-hosts
        contact_groups          wsh
END

print OUT "\tmembers\t\t\t";

$maybecomma = "";
foreach $host (@allwshhosts) {
    print OUT "$maybecomma$host";
    $maybecomma = ",";
}

print OUT "\n\t}\n\n";

close(OUT);
}

$dbh->disconnect;

exit 0;