#
# (c) Jan Gehring <jan.gehring@gmail.com>
#
package Rex::Virtualization::VBox::forward_port;
use v5.12.5;
use warnings;
our $VERSION = '1.15.0.2'; # TRIAL VERSION
use Rex::Logger;
use Rex::Helper::Run;
use Rex::Virtualization::VBox::info;
#
# vm forward_port => "foovm", add => { http => [80, 90] };
# vm forward_port => "foovm", remove => "http";
#
sub execute {
  my ( $class, $arg1, $action, $option ) = @_;
  unless ($arg1) {
    die("You have to define the vm name!");
  }
  my $dom = $arg1;
  unless ($dom) {
    die("VM $dom not found.");
  }
  if ( $action eq "add" ) {
    for my $rule ( keys %{$option} ) {
      my $from_port = $option->{$rule}->[0];
      my $to_port   = $option->{$rule}->[1];
      i_run
        "VBoxManage modifyvm \"$dom\" --natpf1 \"$rule,tcp,,$from_port,,$to_port\"";
    }
  }
  else {
    if ( $option ne "-all" ) {
      i_run "VBoxManage modifyvm \"$dom\" --natpf1 delete \"$option\"";
    }
    else {
      # if no name is given, remove all redirects
      # output: Forwarding(0)="ssh,tcp,,2222,,22"
      my $info = Rex::Virtualization::VBox::info->execute($dom);
      my @keys = grep { m/^Forwarding/ } keys %{$info};
      for my $k (@keys) {
        my @_t = split( /,/, $info->{$k} );
        i_run "VBoxManage modifyvm \"$dom\" --natpf1 delete \"$_t[0]\"";
      }
    }
  }
  if ( $? != 0 ) {
    die("Error setting port forwarding options for vm $dom");
  }
}
1;