diff options
Diffstat (limited to 'lib/PINGDOMFETCH/Pingdomfetch.pm')
| -rw-r--r-- | lib/PINGDOMFETCH/Pingdomfetch.pm | 245 |
1 files changed, 245 insertions, 0 deletions
diff --git a/lib/PINGDOMFETCH/Pingdomfetch.pm b/lib/PINGDOMFETCH/Pingdomfetch.pm new file mode 100644 index 0000000..3aa0046 --- /dev/null +++ b/lib/PINGDOMFETCH/Pingdomfetch.pm @@ -0,0 +1,245 @@ +package PINGDOMFETCH::Pingdomfetch; + +use strict; +use warnings; + +use PINGDOMFETCH::Config; +use PINGDOMFETCH::DateHelper; +use PINGDOMFETCH::Display; +use PINGDOMFETCH::Notify; +use PINGDOMFETCH::Pingdom; +use PINGDOMFETCH::Utils; + +our @ISA = ('PINGDOMFETCH::Display'); + +sub new { + my ( $class, $opts ) = @_; + + my $config = PINGDOMFETCH::Config->new($opts); + my $pingdom = PINGDOMFETCH::Pingdom->new($config); + + my $self = bless { + config => $config, + pingdom => $pingdom, + dots_counter => 0, + }, $class; + + $self->init_from_to_interval(); + + return $self; +} + +sub init_from_to_interval { + my ($self) = @_; + + my $config = $self->{config}; + + # Yeah, Hash Slices are hellworks! + my ( $from, $to ) = @{$config}{qw(arg.from arg.to)}; + + my $dh_from = $config->{dh_from} = + PINGDOMFETCH::DateHelper->new( $config, $from ); + my $dh_to = $config->{dh_to} = + PINGDOMFETCH::DateHelper->new( $config, $to ); + + $dh_from->begin_of_day() if $from eq ''; + + # Handle the --flatten switcht + my $flatten = $config->{'arg.flatten'}; + my ( $flatten_from, $flatten_to ) = split ':', $flatten; + + if ( defined $flatten_from ) { + $dh_from->flatten($flatten_from) + if $flatten_from ne ''; + + $dh_to->flatten($flatten_to) + if defined $flatten_to and $flatten_to ne ''; + } + + $self->error( +"Interval '$dh_from' - '$dh_to' is negative or zero. 'from' must be < 'to'." + ) if $dh_from->time() >= $dh_to->time(); + + $config->{interval_is_in_future} = $dh_to->is_in_future(); + + $self->{dh_from} = $dh_from; + $self->{dh_to} = $dh_to; + + return undef; +} + +sub get_checkid_avail { + my ( $self, $checkid ) = @_; + + my $config = $self->{config}; + my $services = $config->{services}; + + while ( my ( $k, $v ) = each %$services ) { + if ( $v->{checkid} eq $checkid ) { + $self->verbose("Checkid $checkid belongs to service $k"); + $self->get_all_services_avail( { $k => $v } ); + + return ($v); + } + } + + $self->error("No such service with checkid '$checkid'"); + + return (); +} + +sub get_service_avail { + my ( $self, $servicename ) = @_; + + my $config = $self->{config}; + my $services = $config->{services}; + + if ( exists $services->{$servicename} ) { + my $service = $services->{$servicename}; + $self->get_all_services_avail( { $servicename => $service } ); + + return $service; + } + + $self->error("No such service '$servicename'"); + + return (); +} + +sub get_tls_avail { + my ( $self, $tlsname ) = @_; + + my $config = $self->{config}; + + my @results; + + if ( ref $config->{tls}{$tlsname} ) { + my $tls = $config->{tls}{$tlsname}; + my $services = $tls->{services}; + + $self->get_all_services_avail($services); + $tls->acc(); + + return ($tls); + + } + else { + $self->error("No such TLS '$tlsname'"); + } + + return (); +} + +sub get_all_services_avail { + my ( $self, $services ) = @_; + + my $pingdom = $self->{pingdom}; + my $config = $self->{config}; + + my @return; + + while ( my ( $k, $v ) = each %$services ) { + unless ( $config->is_verbose() ) { + $self->{dots_counter}++; + + if ( $self->{dots_counter} == 3 ) { + print '...'; + + } + elsif ( $self->{dots_counter} > 3 ) { + print '.'; + } + } + $v->{result} = $pingdom->fetch_avail_result($v); + push @return, $v; + } + + return @return; +} + +sub run { + my ($self) = @_; + my $retval = 0; + + my $config = $self->{config}; + my $pingdom = $self->{pingdom}; + + $config->read_services($pingdom); + $config->read_tls(); + + return $config->print_services() if $config->{'arg.list-services'}; + + return $config->print_tls() if $config->{'arg.list-tls'}; + + $self->info( + "Fetching stats of interval '$self->{dh_from}' - '$self->{dh_to}'"); + + my @data; + + push @data, $self->get_checkid_avail( $config->{'arg.checkid'} ) + if $config->{'arg.checkid'} ne ''; + + push @data, $self->get_service_avail( $config->{'arg.service'} ) + if $config->{'arg.service'} ne ''; + + if ( $config->{'arg.tls'} ne '' ) { + if ( $config->{'arg.tls'} =~ /,/ ) { + push @data, $self->get_tls_avail($_) + for split ',', $config->{'arg.tls'}; + } + else { + push @data, $self->get_tls_avail( $config->{'arg.tls'} ); + } + } + + push @data, $self->get_all_services_avail( $config->{services} ) + if $config->{'arg.all-services'}; + + if ( $config->{'arg.all-tls'} ) { + push @data, $self->get_tls_avail($_) for sort keys %{ $config->{tls} }; + } + + if (@data) { + my @sorted_data = + sort { $b->{result}{avail_perc} <=> $a->{result}{avail_perc} } @data; + @sorted_data = reverse @sorted_data + if $config->bool('arg.sort-reverse'); + + if ( $self->is_verbose() ) { + $self->error("--notify* can not be used together with --verbose") + if $config->bool('arg.notify-info') + or $config->bool('arg.notify'); + + for (@sorted_data) { + $_->print_full(); + $self->nl(); + } + + } + else { + print "\n" if $self->{dots_counter} > 2; + + my $notify = $config->{notify}; + + for (@sorted_data) { + $_->print(); + $self->nl($notify); + } + + $notify->info_notification_send() + if $config->bool('arg.notify-info'); + + $notify->notification_send() + if $config->bool('arg.notify') + and ( $notify->has_warnings(), or $notify->has_criticals() ); + } + } + else { + $self->warning( + "No results found. Use --all-tls for all TLS or --help for help!"); + } + + return 0; +} + +1; |