Skip to content

Commit

Permalink
Merge pull request #1 from mickenordin/master
Browse files Browse the repository at this point in the history 
Add export method
  • Loading branch information
@theseal
theseal authored Mar 30, 2017
2 parents 84a6967 + c16ef00 commit ea9af76
Show file tree
Hide file tree
Showing 2 changed files with 214 additions and 61 deletions.
253 changes: 194 additions & 59 deletions SU/API/Icinga2.pm
View file
Original file line number Diff line number Diff line change
Expand Up @@ -11,147 +11,282 @@ use Encode qw( encode_utf8 );

sub new {
my $class = shift;
my $self = {
my $self = {
hostname => shift,
port => shift,
path => shift,
version => shift,
port => shift,
path => shift,
version => shift,
insecure => shift,
};

if ($self->{path} !~ /\/$/) {
if ( $self->{path} !~ /\/$/ ) {
$self->{path} .= "/";
}
$self->{url} = "https://$self->{hostname}:$self->{port}$self->{path}v$self->{version}";
$self->{url} =
"https://$self->{hostname}:$self->{port}$self->{path}v$self->{version}";

if ($self->{insecure}) {
$self->{ua} = LWP::UserAgent->new( ssl_opts => {verify_hostname => 0 } );
} else {
if ( $self->{insecure} ) {
$self->{ua} =
LWP::UserAgent->new( ssl_opts => { verify_hostname => 0 } );
}
else {
$self->{ua} = LWP::UserAgent->new;
};
}

$self->{ua}->default_header('Accept' => 'application/json');
$self->{ua}->default_header( 'Accept' => 'application/json' );
$self->{login_status} = "not logged in";

bless $self, $class;
return $self;
};
}

sub do_request {
my ($self,$method,$uri,$params,$data, $plaintext) = @_;
my ( $self, $method, $uri, $params, $data, $plaintext ) = @_;

my $request_url;
$request_url = "$self->{url}/${uri}";

if ($params) {
$params = encode_params($params);
$params = encode_params($params);
$request_url = "$self->{url}/${uri}?$params";
};
my $req = HTTP::Request->new($method => $request_url);
}
my $req = HTTP::Request->new( $method => $request_url );

if ($data) {
$data = encode_json($data);
$req->content($data);
};
}

$self->{res} = $self->{ua}->request($req);

if (!$self->{res}->is_success) {
if ( !$self->{res}->is_success ) {
return undef;
};
}

# Try with first plaintext if plaintext is set
if($plaintext) {
my $str = encode_utf8($self->{res}->content);
if($str) {
if ($plaintext) {
my $str = encode_utf8( $self->{res}->content );
if ($str) {
return $str;
}
}

# Handle non utf8 chars
my $json_result = decode_json(encode_utf8($self->{res}->content));
my $json_result = decode_json( encode_utf8( $self->{res}->content ) );
if ($json_result) {
return $json_result;
};
}
return undef;
};
}

sub encode_params {
my $filter = $_[0];
my @filter_array;
my @encoded_uri_array;

if($filter =~ /&/) {
@filter_array = split('&',$filter);
} else {
if ( $filter =~ /&/ ) {
@filter_array = split( '&', $filter );
}
else {
@filter_array = $filter;
};
for(@filter_array) {
if($_ =~ /=/) {
my ($argument,$value) = split("=",$_);
push(@encoded_uri_array,join("=",uri_escape($argument),uri_escape($value)));
} else {
push(@encoded_uri_array,uri_escape($_));
};
};
return join("&",@encoded_uri_array);
};
}
for (@filter_array) {
if ( $_ =~ /=/ ) {
my ( $argument, $value ) = split( "=", $_ );
push( @encoded_uri_array,
join( "=", uri_escape($argument), uri_escape($value) ) );
}
else {
push( @encoded_uri_array, uri_escape($_) );
}
}
return join( "&", @encoded_uri_array );
}

sub export {

my ( $self, $full, $api_only ) = @_;
my $result = decode_json( encode_utf8( $self->{res}->content ) );
my $type = $result->{results}[0]{type};
my @allowed_types = (
"CheckCommand", "Host", "HostGroup", "Service",
"ServiceGroup", "TimePeriod", "User", "UserGroup"
);

# We only support certain object types
unless ( grep( /$type/, @allowed_types ) ) {
return undef;
}

my @keys;

if ( $type eq "CheckCommand" ) {
@keys = ( "arguments", "command", "env", "vars", "timeout" );
if ($full) {
push @keys, ( "templates", "zone" );
}
}
elsif ( $type eq "Host" ) {
@keys = (
"address6", "address",
"check_command", "display_name",
"event_command", "action_url",
"notes", "notes_url",
"vars", "icon_image",
"icon_image_alt", "check_interval"
);
if ($full) {
push @keys,
(
"check_period", "check_timeout",
"enable_active_checks", "enable_event_handler",
"enable_flapping", "enable_notifications",
"enable_passive_checks", "enable_perfdata",
"groups", "notes",
"retry_interval", "templates",
"zone"
);
}
}
elsif ( $type eq "HostGroup" ) {
@keys = ( "action_url", "display_name", "notes", "notes_url", "vars" );
if ($full) {
push @keys, ( "groups", "templates", "zone" );
}
}
elsif ( $type eq "Service" ) {
@keys = (
"vars", "action_url",
"check_command", "check_interval",
"display_name", "notes",
"notes_url", "event_command",
"max_check_attempts", "retry_interval"
);
if ($full) {
push @keys,
(
"check_period", "check_timeout",
"enable_active_checks", "enable_event_handler",
"enable_flapping", "enable_notifications",
"enable_passive_checks", "enable_perfdata",
"groups", "icon_image",
"icon_image_alt", "notes",
"templates", "zone"
);
}
}
elsif ( $type eq "ServiceGroup" ) {
@keys = ( "action_url", "display_name", "notes", "notes_url", "vars" );
if ($full) {
push @keys, ( "groups", "templates", "zone" );
}
}
elsif ( $type eq "TimePeriod" ) {
@keys = (
"display_name", "excludes",
"includes", "prefer_includes",
"ranges", "vars"
);
if ($full) {
push @keys, ( "templates", "zone" );
}
}
elsif ( $type eq "User" ) {
@keys = (
"display_name", "email",
"enable_notifications", "pager",
"states", "period",
"vars"
);
if ($full) {
push @keys, ( "groups", "templates", "zone" );
}
}
elsif ( $type eq "UserGroup" ) {
@keys = ( "display_name", "vars" );
if ($full) {
push @keys, ( "groups", "templates", "zone" );
}
}
my @results;
foreach my $object ( @{ $result->{results} } ) {
my %hash;
foreach my $key (sort @keys) {
$hash{attrs}{$key} = $object->{attrs}{$key};
}
if ($api_only) {
if ($object->{attrs}{package} eq "_api") {
push @results, \%hash;
}
}
else {
push @results, \%hash;
}

}
return \@results;

}

sub login {
my ($self,$username,$password) = @_;
my ( $self, $username, $password ) = @_;

$self->{username} = $username;
$self->{password} = $password;

$self->{ua}->credentials("$self->{hostname}:$self->{port}", "Icinga 2", $self->{username}, $self->{password});
$self->{ua}->credentials( "$self->{hostname}:$self->{port}",
"Icinga 2", $self->{username}, $self->{password} );

$self->do_request("GET", "/status", "", "");
$self->do_request( "GET", "/status", "", "" );


if ($self->request_code == 200 ) {
if ( $self->request_code == 200 ) {
$self->{login_status} = "login successful";
$self->{logged_in} = 1;
} elsif ($self->request_code == 401) {
$self->{logged_in} = 1;
}
elsif ( $self->request_code == 401 ) {
$self->{login_status} = "wrong username/password";
} else {
$self->{login_status} = "unknown status line: " . $self->{res}->status_line;
}
else {
$self->{login_status} =
"unknown status line: " . $self->{res}->status_line;
}

return $self->{logged_in};
};
}

sub logout {
my ($self) = @_;
$self->{logged_in} = undef;
};
}

sub request_code {
my ($self) = @_;
return $self->{res}->code;
};
}

sub request_status_line {
my ($self) = @_;
return $self->{res}->status_line;
};
}

sub logged_in {
my ($self) = @_;
return $self->{logged_in};
};
}

sub login_status {
my ($self) = @_;
return $self->{login_status};
};
}

sub DESTROY {
my ($self) = @_;
if ($self->{ua} && $self->{logged_in}) {
if ( $self->{ua} && $self->{logged_in} ) {
$self->logout();
} elsif ($self->{logged_in}) {
}
elsif ( $self->{logged_in} ) {
warn "Automatic logout failed";
};
};
}
}

1;
22 changes: 20 additions & 2 deletions SU/API/Icinga2.pod
View file
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,13 @@ SU::API::Icinga2
};


my $put_result = $icinga->do_request("PUT", "/objects/hosts/localhost.example.com"", "", $put_data);
my $put_result = $icinga->do_request("PUT", "/objects/hosts/localhost.example.com", "", $host_object);
if (!$put_result) {
warn $icinga->request_code;
warn $icinga->request_status_line;
};

my $post_result = $icinga->do_request("POST", "/objects/hosts/localhost.example.com", "",$host_object);
if (!$post_result) {
warn $icinga->request_code;
warn $icinga->request_status_line;
Expand Down Expand Up @@ -91,6 +97,18 @@ Perform the actual request.

Returns the result in a perl reference.

=item export (FULL, API_ONLY)

Returns a Perl Data structure without unneccessary meta data. $icinga->export can
only be called after do_request.

"FULL" is optional and if it is true, write once attributes will be included in
the export. Objects exported with FULL set to true can be converted to json and be
used for PUT requests, but not neccessarily for POST requests.

"API_ONLY" is optional and if it is true only objects created via the API
will be exported.

=item request_code

Returns the last HTTP status code.
Expand Down Expand Up @@ -138,4 +156,4 @@ OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
=head1 AUTHOR

Johan Carlquist <[email protected]>,
Mikael Nordin <kano0049@su.se>
Mikael Nordin <mikael.nordin@su.se>

0 comments on commit ea9af76

Please sign in to comment.