pbot/modules/compiler_vm/languages/ruby.pm

#!/usr/bin/perl

# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.

use warnings;
use strict;

package ruby;
use parent '_default';

use Text::ParseWords qw(shellwords);

sub initialize {
  my ($self, %conf) = @_;

  $self->{sourcefile}      = 'prog.rb';
  $self->{execfile}        = 'prog.rb';
  $self->{default_options} = '-w';
  $self->{cmdline}         = 'ruby $options $sourcefile';

  if (length $self->{arguments}) {
    $self->{cmdline} .= " $self->{arguments}";
  }
}

sub preprocess_code {
  my $self = shift;
  $self->SUPER::preprocess_code;

  if (defined $self->{arguments}) {
    my @args = shellwords($self->{arguments});
    my $prelude .= "\nmy \$arglen = " . (scalar @args) . ";\n";

    if (@args) {
      $prelude .= "my \@args = (";

      my $comma = "";
      foreach my $arg (@args) {
        $arg = quotemeta $arg;
        $prelude .= "$comma\"$arg\"";
        $comma = ", ";
      }

      $prelude .= ");\n";
    } else {
      $prelude .= "my \@args;\n";
    }

    $self->{code} = "$prelude\n$self->{code}";
  }
}

sub postprocess_output {
  my $self = shift;
  $self->SUPER::postprocess_output;

  $self->{output} =~ s/\s+at $self->{sourcefile} line \d+, near ".*?"//;
  $self->{output} =~ s/\s*Execution of $self->{sourcefile} aborted due to compilation errors.//;

  $self->{cmdline_opening_comment} = "=cut =============== CMDLINE ===============\n";
  $self->{cmdline_closing_comment} = "=cut\n";

  $self->{output_opening_comment} = "=cut =============== OUTPUT ===============\n";
  $self->{output_closing_comment} = "=cut\n";
}

1;