File Coverage

blib/lib/XAS/Lib/WS/Transfer.pm

Criterion	Covered	Total	%
statement	3	113	2.6
branch	0	6	0.0
condition			n/a
subroutine	1	15	6.6
pod	7	7	100.0
total	11	141	7.8

package XAS::Lib::WS::Transfer;

our $VERSION = '0.01';

use XAS::Class
  version  => $VERSION,
  base     => 'XAS::Lib::WS::RemoteShell',
  codecs   => 'base64 unicode',
  utils    => ':validation dotid',
  constant => 'SCALAR',
;

# ----------------------------------------------------------------------
# Public Methods
# ----------------------------------------------------------------------

sub get {
    my $self = shift;
    my $p = validate_params(\@_, {
        -local  => { type => SCALAR },
        -remote => { type => SCALAR },
    });

    my $local  = $p->{'local'};
    my $remote = $p->{'remote'};

    # this assumes that the remote WS-Manage server is Microsoft based
    # otherwise you would use something sensible like scp

    my $fh;
    my $code   = $self->_code_get_powershell($remote);
    my $invoke = "powershell -noprofile -encodedcommand $code";

    $self->command($invoke);
    $self->receive();
    $self->check_exitcode();

    if (open($fh, '>', $local)) {

        print $fh decode_base64($self->stdout);
        close $fh;

    } else {

        $self->throw_msg(
            dotid($self->class) . '.put.badfile',
            'file_create',
            $local, $!
        );

    }

    return $self->exitcode;

}

sub put {
    my $self = shift;
    my $p = validate_params(\@_, {
        -local  => { type => SCALAR },
        -remote => { type => SCALAR },
    });

    my $local  = $p->{'local'};
    my $remote = $p->{'remote'};

    # this assumes that the remote WS-Manage server is Microsoft based
    # otherwise you would use something sensible like scp

    my $fh;
    my $size   = 30 * 57;
    my $invoke = 'powershell -noprofile -encodedcommand %s';

    if (open($fh, '<', $local)) {

        while (read($fh, my $buf, $size)) {

            my $data = encode_base64($buf, '');
            my $code = $self->_code_put_powershell($remote, $data);
            my $cmd  = sprintf($invoke, $code);

            $self->command($cmd);
            $self->receive();
            $self->check_exitcode();

        }

        close $fh;

    } else {

        $self->throw_msg(
            dotid($self->class) . '.put.badfile',
            'file_create',
            $local, $!
        );

    }

    return $self->exitcode;

}

sub exists {
    my $self = shift;
    my ($path) = validate_params(\@_, [1]);

    my $code   = $self->_code_exists_powershell($path);
    my $invoke = "powershell -noprofile -encodedcommand $code";

    $self->command($invoke);
    $self->receive();
    $self->check_exitcode();

    return $self->exitcode ? 0 : 1;

}

sub mkdir {
    my $self = shift;
    my ($path) = validate_params(\@_, [1]);

    my $code   = $self->_code_mkdir_powershell($path);
    my $invoke = "powershell -noprofile -encodedcommand $code";

    $self->command($invoke);
    $self->receive();
    $self->check_exitcode();

    return $self->exitcode;

}

sub rmdir {
    my $self = shift;
    my ($path) = validate_params(\@_, [1]);

    my $code   = $self->_code_rmdir_powershell($path);
    my $invoke = "powershell -noprofile -encodedcommand $code";

    $self->command($invoke);
    $self->receive();
    $self->check_exitcode();

    return $self->exitcode;

}

sub del {
    my $self = shift;
    my ($path) = validate_params(\@_, [1]);

    my $code   = $self->_code_del_powershell($path);
    my $invoke = "powershell -noprofile -encodedcommand $code";

    $self->command($invoke);
    $self->receive();
    $self->check_exitcode();

    return $self->exitcode;

}

sub dir {
    my $self = shift;
    my ($path) = validate_params(\@_, [1]);

    my $code   = $self->_code_dir_powershell($path);
    my $invoke = "powershell -noprofile -encodedcommand $code";

    $self->command($invoke);
    $self->receive();
    $self->check_exitcode();

    return $self->stdout;

}

# ----------------------------------------------------------------------
# Private Methods
# ----------------------------------------------------------------------

# ----------------------------------------------------------------------
# Powershell Boilerplate - yeah heredoc...
#
# some powershell code borrowed from
#    https://github.com/WinRb/winrm-fs/tree/master/lib/winrm-fs/scripts
# ----------------------------------------------------------------------

sub _code_put_powershell {
    my $self     = shift;
    my $filename = shift;
    my $data     = shift;

    my $code = <<'CODE';
$ProgressPreference='SilentlyContinue'
try {
    $data = '__DATA__'
    $bytes = [System.Convert]::FromBase64String($data)
    $file = [System.IO.File]::Open('__FILENAME__', 'Append')
    $file.Write($bytes, 0, $bytes.Length)
    $file.Close()
    exit 0
} catch {
    Write-Error -Message $_.Exception.Message
    exit 1
}
CODE

    $code =~ s/__FILENAME__/$filename/;
    $code =~ s/__DATA__/$data/;

    return encode_base64(encode_unicode('UTF-16LE', $code), '');

}

sub _code_get_powershell {
    my $self = shift;
    my $filename = shift;

my $code = <<'CODE';
$ProgressPreference='SilentlyContinue'
$p = $ExecutionContext.SessionState.Path
$path = $p.GetUnresolvedProviderPathFromPSPath('__FILENAME__')
if (Test-Path $path -PathType Leaf) {
    $bytes = [System.convert]::ToBase64String([System.IO.File]::ReadAllBytes($path))
    Write-Host $bytes
    exit 0
}
Write-Error -Message 'File not found'
exit 1
CODE

    $code =~ s/__FILENAME__/$filename/;

    return encode_base64(encode_unicode('UTF-16LE', $code), '');

}

sub _code_exists_powershell {
    my $self = shift;
    my $path = shift;

my $code = <<'CODE';
$ProgressPreference='SilentlyContinue'
$p = $ExecutionContext.SessionState.Path
$path = $p.GetUnresolvedProviderPathFromPSPath('__PATH__')
if (Test-Path $path) {
   exit 0
} else {
   Write-Error -Message '__PATH__ not found'
   exit 1
}
CODE

    $code =~ s/__PATH__/$path/g;

    return encode_base64(encode_unicode('UTF-16LE', $code), '');

}

sub _code_mkdir_powershell {
    my $self = shift;
    my $path = shift;

my $code = <<'CODE';
$ProgressPreference='SilentlyContinue'
$p = $ExecutionContext.SessionState.Path
$path = $p.GetUnresolvedProviderPathFromPSPath('__PATH__')
if (!(Test-Path $path)) {
    New-Item -ItemType Directory -Force -Path $path | Out-Null
    exit 0
}
Write-Error -Message '__PATH__ not found'
exit 1
CODE

    $code =~ s/__PATH__/$path/g;

    return encode_base64(encode_unicode('UTF-16LE', $code), '');

}

sub _code_rmdir_powershell {
    my $self = shift;
    my $path = shift;

my $code = <<'CODE';
$ProgressPreference='SilentlyContinue'
$p = $ExecutionContext.SessionState.Path
$path = $p.GetUnresolvedProviderPathFromPSPath('__PATH__')
if (Test-Path $path) {
    Remove-Item $path -Force
    exit 0
}
Write-Error -Message '__PATH__ not found'
exit 1
CODE

    $code =~ s/__PATH__/$path/g;

    return encode_base64(encode_unicode('UTF-16LE', $code), '');

}

sub _code_del_powershell {
    my $self = shift;
    my $path = shift;

my $code = <<'CODE';
$ProgressPreference='SilentlyContinue'
$p = $ExecutionContext.SessionState.Path
$path = $p.GetUnresolvedProviderPathFromPSPath('__PATH__')
if (Test-Path $path) {
    Remove-Item $path -Force
    exit 0
}
Write-Error -Message '__PATH__ not found'
exit 1
CODE

    $code =~ s/__PATH__/$path/g;

    return encode_base64(encode_unicode('UTF-16LE', $code), '');

}

sub _code_dir_powershell {
    my $self = shift;
    my $path = shift;

my $code = <<'CODE';
$ProgressPreference='SilentlyContinue'
$p = $ExecutionContext.SessionState.Path
$path = $p.GetUnresolvedProviderPathFromPSPath('__PATH__')
if (Test-Path $path) {
    Get-ChildItem -Path __PATH__
    exit 0
}
Write-Error -Message '__PATH__ not found'
exit 1
CODE

    $code =~ s/__PATH__/$path/g;

    return encode_base64(encode_unicode('UTF-16LE', $code), '');

}

1;

__END__