#! /usr/bin/perl -w
################################################################################
# Copyright 2005-2009 MERETHIS
# Centreon is developped by : Julien Mathis and Romain Le Merlus under
# GPL Licence 2.0.
# 
# 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 2 of the License.
# 
# 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>.
# 
# Linking this program statically or dynamically with other modules is making a 
# combined work based on this program. Thus, the terms and conditions of the GNU 
# General Public License cover the whole combination.
# 
# As a special exception, the copyright holders of this program give MERETHIS 
# permission to link this program with independent modules to produce an executable, 
# regardless of the license terms of these independent modules, and to copy and 
# distribute the resulting executable under terms of MERETHIS choice, provided that 
# MERETHIS also meet, for each linked independent module, the terms  and conditions 
# of the license of that module. An independent module is a module which is not 
# derived from this program. If you modify this program, you may extend this 
# exception to your version of the program, but you are not obliged to do so. If you
# do not wish to do so, delete this exception statement from your version.
# 
# For more information : contact@centreon.com
# 
# SVN : $URL
# SVN : $Id :
#
####################################################################################
#
# Script init
#

use strict;
use Getopt::Long;
use DBI;

use vars qw($mysql_database_oreon $mysql_database_ods $mysql_host $mysql_user $mysql_passwd);
require "/etc/centreon/conf.pm";

######################################
## Get snmptt configuration files path
#

sub getPath($) {
    my $dbh = shift;
    my $sth = $dbh->prepare("Select `snmp_trapd_path_conf` FROM `general_opt`");
    $sth->execute();
    my $path = $sth->fetchrow_array();
    undef($sth);
    $path .= "/" if (!($path =~ /\/$/));
    $path = "/etc/snmp/" if (!$path);
    return $path;
}

sub main() {
    print "Generating SNMPTT configuration files...\n";
    my ($nbMan, $nbTraps) = (0,0);

    my $dbh = DBI->connect("DBI:mysql:database=".$mysql_database_oreon.";host=".$mysql_host, $mysql_user, $mysql_passwd, {'RaiseError' => 0, 'PrintError' => 0, 'AutoCommit' => 1});
    my $confFiles_path = getPath($dbh);

    my $sth = $dbh->prepare("SELECT `oreon_path` FROM `general_opt` LIMIT 1");
    $sth->execute();
    my @conf = $sth->fetchrow_array();
    $sth->finish();
    my $NAGIOS_TRAPS = $conf[0]."bin/";

    $sth = $dbh->prepare("SELECT `id`, `name` FROM `traps_vendor`");
    $sth->execute();
    my $snmpttIni = "";
    while (my ($man_id, $man_name) = $sth->fetchrow_array()) {
		my $sth2 = $dbh->prepare("SELECT `traps_name`, `traps_oid`, `traps_status`, `traps_args`, `traps_comments` FROM `traps` WHERE `manufacturer_id` = '$man_id'");
		$sth2->execute();
		if (!open(FILE, "> ".$confFiles_path."snmptt-".$man_name.".conf")) {
		    print "Cannot open ".$confFiles_path."snmptt-".$man_name.".conf in write mode - Export aborded\n";
		    exit;
		}
		$nbMan++ if ($sth2->rows);
		while (my @values = $sth2->fetchrow_array()) {
		    $nbTraps++;
		    my $args = "";
		    print FILE "EVENT ".$values[0]." ".$values[1]." \"Status Event\" ".$values[2]."\n";
		    if (defined($values[3])) {
				print FILE "FORMAT ".$values[3]."\n";
		    } else {
				$values[3] = "no output for trap!";
		    }
		    print FILE "EXEC ".$NAGIOS_TRAPS."centTrapHandler-2.x \$aA \$A \$o \"$values[3]\"\n";
		    if (defined($values[4])) {
				print FILE "SDESC\n".$values[4];
				if ($values[4] =~ /\n$/) {
				    print FILE "EDESC\n\n";
				} else {
				    print FILE "\nEDESC\n\n";
				}
		    } else {
				print FILE "\n";
		    }
		}
		close FILE;
		$snmpttIni .= $confFiles_path."snmptt-".$man_name.".conf\n";
		$sth2->finish();
    }
    print "$nbTraps traps for $nbMan manufacturers are defined.\n";
    $sth->finish();
    $dbh->disconnect();
    if (!open(FILE, $confFiles_path."snmptt.ini")) {
		print "Cannot open ".$confFiles_path."snmptt.ini - Export Aborded\n";
		exit;
    }
    if (!open(TEMP, "> /tmp/snmptt.ini.tmp")) {
		print "Cannot open /tmp/snmptt.ini.tmp in write mode - Export Aborded\n";
		exit;
    }
    my $continue = 1;
    while ($continue == 1) {
		my $line = <FILE>;
		if ($line) {
		    if (!($line =~ /^snmptt\_conf\_files/)) {
				print TEMP $line;
		    } else {
				$continue = 0;
		    }
		} else {
		    $continue = -1;
		}
    }
    if (!$continue) {
		print TEMP "snmptt_conf_files = <<END\n";
		print TEMP $snmpttIni."END\n";
		my $command = "mv /tmp/snmptt.ini.tmp ".$confFiles_path."snmptt.ini";
		my $mv = `$command`;
		print "SNMPTT configuration files generated.\n";
    } else {
		print "Couldn't export ".$confFiles_path."snmptt.ini, please put these lines at the end of file snmptt.ini :\n";
		print "snmptt_conf_files = <<END\n".$snmpttIni."END\n";
    }
}

main();
