use strict;

use lib "<<Make:LIB>>";

use Netdot::Model::Device;

use Netdot::Topology;

use Getopt::Long qw(:config no_ignore_case bundling);

use Log::Log4perl::Level;

my ($host, $blocks, $db, $matching, $file, $commstrs, $version, $sec_name,

$sec_level, $auth_proto, $auth_pass, $priv_proto, $priv_pass);

my $retries = Netdot->config->get('DEFAULT_SNMPRETRIES');

my $timeout = Netdot->config->get('DEFAULT_SNMPTIMEOUT');

my ($ATOMIC, $ADDSUBNETS, $SUBSINHERIT, $BGPPEERS, $RECURSIVE, $INFO, $FWT,

$TOPO, $ARP, $AUTO_DNS, $PRETEND, $HELP, $_DEBUG);

$ENV{REMOTE_USER} = "netdot";

my $USAGE = <<EOF;

my $result = GetOptions( "H|host=s" => \$host,

"B|blocks=s" => \$blocks,

"D|db" => \$db,

"m|matching=s" => \$matching,

"E|file=s" => \$file,

"I|info" => \$INFO,

"F|fwt" => \$FWT,

"A|arp" => \$ARP,

"T|topology" => \$TOPO,

"N|auto-dns" => \$AUTO_DNS,

"c|communities:s" => \$commstrs,

"r|retries:s" => \$retries,

"o|timeout:s" => \$timeout,

"v|version:s" => \$version,

"sec-name:s" => \$sec_name,

"sec-level:s" => \$sec_level,

"auth-proto:s" => \$auth_proto,

"auth-pass:s" => \$auth_pass,

"priv-proto:s" => \$priv_proto,

"priv-pass:s" => \$priv_pass,

"atomic" => \$ATOMIC,

"add-subnets:s" => \$ADDSUBNETS,

"subs-inherit:s" => \$SUBSINHERIT,

"with-bgp-peers:s" => \$BGPPEERS,

"pretend" => \$PRETEND,

"recursive" => \$RECURSIVE,

"h|help" => \$HELP,

"d|debug" => \$_DEBUG,

);

if ( ! $result ) {

print $USAGE;

die "Error: Problem with cmdline args\n";

}

if ( $HELP ) {

print $USAGE;

exit;

}

if ( ($host && $db) || ($host && $blocks) || ($host && $file ) || ($db && $blocks)

|| ($db && $file) || ($blocks && $file) ){

print $USAGE;

die "Error: arguments -H, -B, -D and -E are mutually exclusive\n";

}

unless ( $INFO || $FWT || $ARP || $TOPO || $AUTO_DNS ){

print $USAGE;

die "Error: You need to specify at least one of -I, -F, -A, -T or N\n";

}

foreach my $flagref ( \$ADDSUBNETS, \$SUBSINHERIT, \$BGPPEERS ){

if ( defined $$flagref && !($$flagref =~ /^(:?0|1)$/o) ){

print $USAGE;

die "Error: Invalid value for boolean parameter";

}

}

my %uargs = (version => $version,

timeout => $timeout,

retries => $retries,

sec_name => $sec_name,

sec_level => $sec_level,

auth_proto => $auth_proto,

auth_pass => $auth_pass,

priv_proto => $priv_proto,

priv_pass => $priv_pass,

);

my @communities = split ',', $commstrs if defined $commstrs;

map { $_ =~ s/\s+// } @communities;

$uargs{communities} = \@communities if @communities;

my $logscr = Netdot::Util::Log->new_appender('Screen', stderr=>0);

my $logger = Netdot->log->get_logger('Netdot::Model::Device');

$logger->add_appender($logscr);

my $dns_logger = Netdot->log->get_logger('Netdot::Model::DNS');

$dns_logger->add_appender($logscr);

my $ip_logger = Netdot->log->get_logger('Netdot::Model::Ipblock');

$ip_logger->add_appender($logscr);

if ( $_DEBUG ){

$logger->level($DEBUG);

$dns_logger->level($DEBUG);

$ip_logger->level($DEBUG);

}

$logger->warn("Warning: Pretend (-p) flag enabled. Changes will not be committed to the DB")

if ( $PRETEND );

my $start = time;

$logger->info(sprintf("$0 started at %s", scalar localtime($start)));

if ( $INFO || $FWT || $ARP ){

# Arguments specific to these functions

my %dargs = (

pretend => $PRETEND,

atomic => $ATOMIC,

add_subnets => $ADDSUBNETS,

subs_inherit => $SUBSINHERIT,

bgp_peers => $BGPPEERS,

do_info => $INFO,

do_fwt => $FWT,

do_arp => $ARP,

matching => $matching,

);

# Add the common update arguments

while ( my($key, $val) = each %uargs ){

$dargs{$key} = $val;

}

if ( $host ){

$logger->info("Updating single device: $host");

$dargs{name} = $host;

eval {

Device->discover(%dargs);

};

die "ERROR: $@\n" if $@;

}elsif ( $blocks ){

my @blocks = split ',', $blocks;

map { $_ =~ s/\s+//g } @blocks;

$logger->info("Updating all devices in $blocks");

$dargs{blocks} = \@blocks;

Netdot::Model::Device->snmp_update_block(%dargs);

}elsif ( $db ){

$logger->info("Updating all devices in the DB");

Netdot::Model::Device->snmp_update_all(%dargs);

}elsif ( $file ){

$logger->info("Updating all devices in given file: $file");

$dargs{file} = $file;

Netdot::Model::Device->snmp_update_from_file(%dargs);

}else{

print $USAGE;

die "Error: You need to specify one of: -H, -B, -E, or -D\n";

}

}

if ( $AUTO_DNS ){

if ( $host || $blocks || $file ){

die "Error: AUTO DNS option only works with -D"

}elsif ( $db ){

$logger->info("Generating DNS for interface IPs on all devices");

Netdot::Model::Device->do_auto_dns_all();

}

}

if ( $TOPO ){

$logger = Netdot->log->get_logger('Netdot::Topology');

$logger->level($DEBUG) if ( $_DEBUG ); # Notice that $DEBUG is imported from Log::Log4perl

$logger->add_appender($logscr);

$uargs{recursive} = 1 if $RECURSIVE;

Netdot::Topology->discover(%uargs);

}

$logger->info(sprintf("$0 total runtime: %s", Netdot->sec2dhms(time-$start)));