File Coverage

blib/lib/Armadito/Agent/Task/Enrollment.pm

Criterion	Covered	Total	%
statement	18	79	22.7
branch	0	20	0.0
condition	0	3	0.0
subroutine	6	16	37.5
pod	2	2	100.0
total	26	120	21.6

package Armadito::Agent::Task::Enrollment;

use strict;
use warnings;
use base 'Armadito::Agent::Task';
use Armadito::Agent::Tools::File qw( readFile );
use Data::Dumper;
use JSON;

sub new {
	my ( $class, %params ) = @_;

	my $self = $class->SUPER::new(%params);

	if ( $params{debug} ) {
		$self->{debug} = 1;
	}

	my $task = {
		name      => "Enrollment",
		antivirus => $self->{agent}->{antivirus}->getJobj()
	};

	$self->{jobj}->{task} = $task;

	return $self;
}

sub run {
	my ( $self, %params ) = @_;

	$self = $self->SUPER::run(%params);

	$self->_setEnrollmentKey();

	my $json_text = to_json( $self->{jobj} );
	print $json_text. "\n";

	my $response = $self->{glpi_client}->sendRequest(
		"url"   => $self->{agent}->{config}->{server}[0] . "/api/agents",
		message => $json_text,
		method  => "POST"
	);

	if ( $response->is_success() && $response->content() =~ /^\s*\{/ms ) {
		$self->_handleResponse($response);
		$self->{logger}->info("Enrollment successful...");
	}
	else {
		$self->_handleError($response);
		$self->{logger}->info("Enrollment failed...");
	}

	return $self;
}

sub _handleError {
	my ( $self, $response ) = @_;

	$self->{logger}->error( "Error Response : " . $response->content() . "\n" );
	if ( $response->content() =~ /^\s*\{/ ) {
		my $obj = from_json( $response->content(), { utf8 => 1 } );
		$self->{logger}->error( $obj->{message} );
	}
	return $self;
}

sub _handleResponse {
	my ( $self, $response ) = @_;

	$self->{logger}->info( $response->content() );

	my $jobj = from_json( $response->content(), { utf8 => 1 } );
	$self->_updateStorage($jobj);
	$self->_rmEnrollmentKey();

	return $self;
}

sub _updateStorage {
	my ( $self, $jobj ) = @_;

	$self->{agent}->{agent_id}     = defined( $jobj->{agent_id} )     ? $jobj->{agent_id}     : 0;
	$self->{agent}->{scheduler_id} = defined( $jobj->{scheduler_id} ) ? $jobj->{scheduler_id} : 0;

	if ( $jobj->{agent_id} > 0 ) {
		$self->{agent}->_storeArmaditoIds();
		$self->{logger}->info( "Agent successfully enrolled with id " . $jobj->{agent_id} );
	}
}

sub _rmEnrollmentKey {
	my ($self) = @_;

	my $keyfile = $self->_getEnrollmentKeyPath();

	if ( -f $keyfile ) {
		unlink $keyfile;
	}
}

sub _setEnrollmentKey {
	my ($self) = @_;

	my $key  = '';
	my $jobj = {};

	$key = $self->_getEnrollmentKey();

	if ( $self->_isValidKeyFormat($key) ) {
		$jobj->{key} = $key;
	}
	else {
		die "Invalid Key. Expected key format is : \n"
			. " [A-Z0-9]{5}-[A-Z0-9]{5}-[A-Z0-9]{5}-[A-Z0-9]{5}-[A-Z0-9]{5}\n"
			. " For example : AAAAF-111AF-DZ78F-EE78F-DDD1F";
	}

	$self->{jobj}->{task}->{obj} = $jobj;
}

sub _getEnrollmentKey {
	my ($self) = @_;

	my $key     = '';
	my $keyfile = $self->_getEnrollmentKeyPath();

	if ( $self->{agent}->{key} ne "" ) {
		$key = $self->{agent}->{key};
	}
	elsif ( -f $keyfile ) {
		$key = readFile( filepath => $keyfile );
	}
	else {
		die "No enrollment key found.";
	}

	return $key;
}

sub _getEnrollmentKeyPath {
	my ($self) = @_;

	return $self->{agent}->{vardir} . "enrollment.key";
}

sub _isValidKeyFormat {
	my ( $self, $key ) = @_;

	return $key =~ m/^[A-Z0-9]{5}-[A-Z0-9]{5}-[A-Z0-9]{5}-[A-Z0-9]{5}-[A-Z0-9]{5}$/msi;
}

1;

__END__

=head1 NAME

Armadito::Agent::Task::Enrollment - Enrollment task of Armadito Agent.

=head1 DESCRIPTION

This task inherits from L<Armadito::Agent::Task>. Enroll the device into Armadito plugin for GLPI.

=head1 FUNCTIONS

=head2 run ( $self, %params )

Run the task.

=head2 new ( $self, %params )

Instanciate Task.