File Coverage

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

Criterion	Covered	Total	%
statement	15	30	50.0
branch	0	8	0.0
condition	0	3	0.0
subroutine	5	6	83.3
pod	0	1	0.0
total	20	48	41.6

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

package Rex::Virtualization::CBSD::p9shares_add;

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 all the keys are sane
	my @required_keys = ( 'vm', 'device', 'path' );
	foreach my $key (@required_keys) {

		if ( !defined( $opts{$key} ) ) {
			die 'The required variable "' . $key . '" is not set';
		}

		# make sure it does not contain any possible characters we don't want
		if (   ( $key ne 'path' )
			&& ( $opts{$key} =~ /[\t\ \=\\\/\'\"\n\;\&]/ ) )
		{
			die 'The value for "'
				. $key . '", "'
				. $opts{$key}
				. '", matched /[\t\ \=\/\\\'\"\n\;\&]/, meaning it is not a valid value';
		}
		elsif ( $opts{$key} =~ /[\t\'\"\n\;\&]/ ) {
			die 'The value for "'
				. $key . '", "'
				. $opts{$key}
				. '", matched /[\t\'\"\n\;\&]/, meaning it is not a valid value';
		}
	}

	# put together the command
	my $command
		= 'cbsd bhyve-p9shares mode=attach jname='
		. $opts{vm}
		. ' device='
		. $opts{device}
		. " path='"
		. $opts{path} . "'";

	Rex::Logger::debug( "Adding CBSD p9share 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;