tyler.durden/centreon-plugins
1
0
Fork 0
mirror of https://github.com/centreon/centreon-plugins.git synced 2025-07-27 23:54:18 +02:00
Code Issues Packages Projects Releases Wiki Activity

use 'execute' function from centreon

Browse Source
This commit is contained in:
David Sabatié 2015-09-15 18:02:15 +02:00
parent c1eaedb612
commit c6284de6d6
1 changed files with 88 additions and 51 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

139
centreon-plugins/cloud/aws/custom/awscli.pm
View File

@ -24,42 +24,51 @@ use strict;
use warnings; use warnings;
use JSON; use JSON;
use Data::Dumper; use Data::Dumper;
use centreon::plugins::misc;
sub new { sub new {
my ($class, %options) = @_; my ( $class, %options ) = @_;
my $self = {}; my $self = {};
bless $self, $class; bless $self, $class;
# $options{options} = options object # $options{options} = options object
# $options{output} = output object # $options{output} = output object
# $options{exit_value} = integer # $options{exit_value} = integer
# $options{noptions} = integer # $options{noptions} = integer
if (!defined($options{output})) { if ( !defined( $options{output} ) ) {
print "Class Custom: Need to specify 'output' argument.\n"; print "Class Custom: Need to specify 'output' argument.\n";
exit 3; exit 3;
} }
if (!defined($options{options})) { if ( !defined( $options{options} ) ) {
$options{output}->add_option_msg(short_msg => "Class Custom: Need to specify 'options' argument."); $options{output}->add_option_msg( short_msg => "Class Custom: Need to specify 'options' argument." );
$options{output}->option_exit(); $options{output}->option_exit();
} }
if (!defined($options{noptions})) { if ( !defined( $options{noptions} ) ) {
$options{options}->add_options(arguments => $options{options}->add_options( arguments => {
{ "region:s" => { name => 'region' },
"region:s" => { name => 'region' }, "command:s" => { name => 'command', default => 'aws' },
}); "command-path:s" => { name => 'command_path' },
"sudo" => { name => 'sudo' },
} );
} }
$options{options}->add_help(package => __PACKAGE__, sections => 'AWSCLI OPTIONS', once => 1); $options{options}->add_help(
package => __PACKAGE__,
sections => 'AWSCLI OPTIONS',
once => 1
);
$self->{output} = $options{output}; $self->{output} = $options{output};
$self->{mode} = $options{mode}; $self->{mode} = $options{mode};
return $self; return $self;
} }
# Method to manage multiples # Method to manage multiples
sub set_options { sub set_options {
my ($self, %options) = @_; my ( $self, %options ) = @_;
# options{options_result} # options{options_result}
$self->{option_results} = $options{option_results}; $self->{option_results} = $options{option_results};
@ -67,15 +76,17 @@ sub set_options {
# Method to manage multiples # Method to manage multiples
sub set_defaults { sub set_defaults {
my ($self, %options) = @_; my ( $self, %options ) = @_;
# options{default} # options{default}
# Manage default value # Manage default value
foreach (keys %{$options{default}}) { foreach ( keys %{ $options{default} } ) {
if ($_ eq $self->{mode}) { if ( $_ eq $self->{mode} ) {
for (my $i = 0; $i < scalar(@{$options{default}->{$_}}); $i++) { for ( my $i = 0 ; $i < scalar( @{ $options{default}->{$_} } ) ; $i++ )
foreach my $opt (keys %{$options{default}->{$_}[$i]}) { {
if (!defined($self->{option_results}->{$opt}[$i])) { foreach my $opt ( keys %{ $options{default}->{$_}[$i] } ) {
if ( !defined( $self->{option_results}->{$opt}[$i] ) ) {
$self->{option_results}->{$opt}[$i] = $options{default}->{$_}[$i]->{$opt}; $self->{option_results}->{$opt}[$i] = $options{default}->{$_}[$i]->{$opt};
} }
} }
@ -85,46 +96,60 @@ sub set_defaults {
} }
sub check_options { sub check_options {
my ($self, %options) = @_; my ( $self, %options ) = @_;
return 0; return 0;
} }
sub execReq { sub execReq {
my ($self, $options) = @_; my ( $self, $options ) = @_;
my $jsoncontent; my $jsoncontent;
if (!defined($options->{output})) { if ( !defined( $options->{output} ) ) {
$options->{output} = 'json'; $options->{output} = 'json';
} }
my $json = JSON->new;
my $json_encoded = "aws ".$options->{command}." ".$options->{subcommand};
if ( defined($self->{option_results}->{region})) {
$json_encoded = $json_encoded . " --region '" . $self->{option_results}->{region} . "'";
}
if ( defined($options->{json})) {
$json_encoded = $json_encoded . " --cli-input-json '" . $json->encode($options->{json}) . "'";
}
if ($options->{output} eq 'text') { my $json = JSON->new;
my @return = `$json_encoded`; my $json_encoded = $options->{command} . " " . $options->{subcommand};
chomp(@return); if ( defined( $self->{option_results}->{region} ) ) {
$jsoncontent = $json->encode([@return]); $json_encoded = $json_encoded . " --region '". $self->{option_results}->{region} . "'";
}
if ( defined( $options->{json} ) ) {
$json_encoded = $json_encoded . " --cli-input-json '" . $json->encode( $options->{json} ) . "'";
}
$self->{option_results}->{timeout} = 30;
if ( $options->{output} eq 'text' ) {
$self->{stdout} = centreon::plugins::misc::execute(
output => $self->{output},
options => $self->{option_results},
sudo => $self->{option_results}->{sudo},
command => $self->{option_results}->{command},
command_path => $self->{option_results}->{command_path},
command_options => $json_encoded
);
my @return = split /\n/, $self->{stdout};
$jsoncontent = $json->encode( [@return] );
} }
else { else {
$jsoncontent = `$json_encoded`; $jsoncontent = centreon::plugins::misc::execute(
output => $self->{output},
options => $self->{option_results},
sudo => $self->{option_results}->{sudo},
command => $self->{option_results}->{command},
command_path => $self->{option_results}->{command_path},
command_options => $json_encoded
);
}
if ( $? > 0 ) {
$self->{output}->add_option_msg( short_msg => "Cannot run aws" );
$self->{output}->option_exit();
}
eval { $self->{command_return} = $json->decode($jsoncontent); };
if ($@) {
$self->{output}->add_option_msg( short_msg => "Cannot decode json answer" );
$self->{output}->option_exit();
} }
if ($? > 0) {
$self->{output}->add_option_msg(short_msg => "Cannot run aws");
$self->{output}->option_exit();
}
eval {
$self->{command_return} = $json->decode($jsoncontent);
};
if ($@) {
$self->{output}->add_option_msg(short_msg => "Cannot decode json answer");
$self->{output}->option_exit();
}
return $self->{command_return}; return $self->{command_return};
} }
@ -146,6 +171,18 @@ AWS cli API custom mode
(optional) The region to use (should be configured directly in aws). (optional) The region to use (should be configured directly in aws).
=item B<--sudo>
Use 'sudo' to execute the command.
=item B<--command>
(optional) Command to get information (Default: 'aws').
=item B<--command-path>
(optional) Command path (Default: none).
=back =back
=head1 DESCRIPTION =head1 DESCRIPTION