File Coverage

blib/lib/Rex/Virtualization/CBSD/bsnapshot_rollback.pm

Criterion	Covered	Total	%
statement	15	29	51.7
branch	0	8	0.0
condition	0	3	0.0
subroutine	5	6	83.3
pod	0	1	0.0
total	20	47	42.5

#
# (c) Zane C. Bowers-Hadley
#

package Rex::Virtualization::CBSD::bsnapshot_rollback;

use strict;
use warnings;

our $VERSION = '0.0.1';    # VERSION

use Rex::Logger;
use Rex::Helper::Run;
use Term::ANSIColor qw(colorstrip);

sub execute {
	my ( $class, %opts ) = @_;

	# make sure the required variables are set
	if ( !defined( $opts{vm} ) ) {
		die 'The required variable "vm" is not set';
	}
	if ( !defined( $opts{name} ) ) {
		die 'The required variable "name" is not set';
	}

	# make sure all the keys are sane
	if (   ( $opts{vm} =~ /[\t\ \=\\\/\'\"\n\;\&]/ )
		|| ( $opts{name} =~ /[\t\ \=\\\/\'\"\n\;\&]/ ) )
	{
		die 'The value either for "vm", "'
			. $opts{vm}
			. '" or "name", "'
			. $opts{name}
			. '", matched /[\t\ \=\/\\\'\"\n\;\&]/, meaning it is not a valid value';
	}

	# put together the command
	my $command
		= 'cbsd bsnapshot mode=rollback jname=' . $opts{vm} . " snapname='" . $opts{name} . "'";

	Rex::Logger::debug( "Rolling back to a CBSD VM snapshot via... " . $command );

	my $returned = i_run( $command, fail_ok => 1 );

	# the output is colorized, if there is an error
	$returned = colorstrip($returned);

	# check for this second as no VM will also exit non-zero
	if ( $? != 0 ) {
		die( "Error running '" . $command . "'" );
	}

	return 1;
}

1;