File Coverage

blib/lib/Rex/Apache/Inject/Properties.pm

Criterion	Covered	Total	%
statement	33	164	20.1
branch	0	74	0.0
condition			n/a
subroutine	11	21	52.3
pod	0	4	0.0
total	44	263	16.7

#
# (c) Jan Gehring
#
# vim: set ts=2 sw=2 tw=0:
# vim: set expandtab:

package Rex::Apache::Inject::Properties;

=begin

=head2 SYNOPSIS

This is a (R)?ex module to ease the deployments of PHP, Perl or other languages.

=cut

use strict;
use warnings;

use Rex::Commands::Run;
use Rex::Commands::Fs;
use Rex::Commands::Upload;
use Rex::Commands;
use Rex::Config;
use File::Basename qw(dirname basename);
use Cwd qw(getcwd);

#require Exporter;
#use base qw(Exporter);

use vars qw(@EXPORT $real_name_from_template $template_file $template_pattern);
@EXPORT = qw(inject
  generate_real_name
  template_file template_search_for);

my $work_dir = getcwd;

############ deploy functions ################

sub inject {
  my ( $to, @options ) = @_;

  my $option = {@options};

  my $tmp_to = $to;
  if ( $tmp_to !~ m/^\// ) { $tmp_to = "../$tmp_to"; }

  my $cmd1 = sprintf( _get_extract_command($to), "$tmp_to" );
  my $cmd2 = sprintf( _get_pack_command($to), "$tmp_to", "." );

  my $template_params = _get_template_params($template_file);

  mkdir("tmp");
  chdir("tmp");
  run $cmd1;

  if ( exists $option->{"extract"} ) {
    for my $file_pattern ( @{ $option->{"extract"} } ) {

      my $find = "find ../ -name '$file_pattern'";
      if ( $^O =~ m/^MSWin/i ) {
        $find = "find2 ../ -name \"$file_pattern\"";
      }

      for my $found_file (`$find`) {
        chomp $found_file;

        mkdir "tmp-b";
        chdir "tmp-b";

        my $extract_cmd =
          sprintf( _get_extract_command($found_file), "../$found_file" );
        my $compress_cmd =
          sprintf( _get_pack_command($found_file), "../$found_file", "." );

        run $extract_cmd;

        _find_and_parse_templates();

        if ( exists $option->{"pre_pack_hook"} ) {
          &{ $option->{"pre_pack_hook"} }($found_file);
        }

        run $compress_cmd;

        if ( exists $option->{"post_pack_hook"} ) {
          &{ $option->{"post_pack_hook"} }($found_file);
        }

        chdir "..";
        rmdir "tmp-b";
      }

    }
  }

  _find_and_parse_templates();

  if ( exists $option->{"pre_pack_hook"} ) {
    &{ $option->{"pre_pack_hook"} };
  }

  run $cmd2;
  if ( $? != 0 ) {
    chdir("..");
    system("rm -rf tmp");
    die("Can't re-pack archive. Please check permissions. Command was: $cmd2");
  }

  if ( exists $option->{"post_pack_hook"} ) {
    &{ $option->{"post_pack_hook"} };
  }

  chdir("..");
  system("rm -rf tmp");
}

sub _find_and_parse_templates {

  my $template_params = _get_template_params($template_file);

  my $find = "find . -name '$template_pattern'";
  if ( $^O =~ m/^MSWin/i ) {
    $find = "find2 . -name \"$template_pattern\"";
  }

  for my $file (`$find`) {
    $/ = "\n";
    chomp $file;
    Rex::Logger::debug("Opening file $file");

    open( my $fh, "<$file" ) or die($!);
    my %con;
    while ( my $line = <$fh> ) {
      chomp $line;
      $line =~ s/\r//gs;
      next if ( $line =~ /^#/ );
      next if ( $line =~ /^$/ );

      my ( $key, $val ) = $line =~ m/^(.*?)\s?[=:]\s?(.*)$/;
      Rex::Logger::debug("key: -$key- => val: -$val-");

      unless ($key) {
        Rex::Logger::info("Parser Error in $file:");
        Rex::Logger::info("($.)===$line===");
        next;
      }
      $val = "" unless $val;
      $con{$key} = $val;
    }
    close($fh);

    for my $k ( keys %$template_params ) {
      Rex::Logger::debug( "(t): -$k- => -" . $template_params->{$k} . "-" );
      if ( exists $con{$k} ) {
        Rex::Logger::info(
          "setting $k to " . $template_params->{$k} . " ($file)" );
        $con{$k} = $template_params->{$k};
      }

      #				 else {
      #				 	  Rex::Logger::info("$k doesn't exists in $file");
      #				 }
    }

    my $new_file_name =
      $real_name_from_template ? &$real_name_from_template($file) : $file;
    Rex::Logger::debug("Writing file $new_file_name");
    open( $fh, ">", $new_file_name ) or die($!);
    for my $key ( keys %con ) {
      print $fh "$key=" . $con{$key} . "\n";
    }
    close($fh);
  }

}

############ configuration functions #############

sub generate_real_name {
  $real_name_from_template = shift;
}

sub template_file {
  $template_file = shift;
}

sub template_search_for {
  $template_pattern = shift;
}

############ helper functions #############

sub _get_extract_command {
  my ($file) = @_;

  if ( $file =~ m/\.tar\.gz$/ ) {
    return "tar xzf %s";
  }
  elsif ( $file =~ m/\.zip$/ ) {
    return "unzip %s";
  }
  elsif ( $file =~ m/\.tar\.bz2$/ ) {
    return "tar xjf %s";
  }
  elsif ( $file =~ m/\.war$/ ) {
    return "unzip %s";
  }
  elsif ( $file =~ m/\.jar$/ ) {
    return "unzip %s";
  }

  die("Unknown Archive Format.");
}

sub _get_pack_command {
  my ($file) = @_;

  if ( $file =~ m/\.tar\.gz$/ ) {
    return "tar czf %s %s";
  }
  elsif ( $file =~ m/\.zip$/ ) {
    return "zip -r %s %s";
  }
  elsif ( $file =~ m/\.tar\.bz2$/ ) {
    return "tar cjf %s %s";
  }
  elsif ( $file =~ m/\.war$/ ) {
    return "zip -r %s %s";
  }
  elsif ( $file =~ m/\.jar$/ ) {
    return "zip -r %s %s";
  }

  die("Unknown Archive Format.");
}

# read the template file and return a hashref.
sub _get_template_params {
  my %inject;

  my $t_file;

  if ( $template_file =~ m/^\// ) {
    $t_file = $template_file;
  }
  else {
    $t_file = "$work_dir/$template_file";
  }

  if ( -f "$t_file." . Rex::Config->get_environment ) {
    $t_file = "$t_file." . Rex::Config->get_environment;
  }

  open( my $fh, "<", $t_file ) or die($!);
  while ( my $line = <$fh> ) {
    chomp $line;
    $line =~ s/\r//gs;
    next if ( $line =~ /^#/ );
    next if ( $line =~ /^$/ );
    $line =~ s/\r//gs;

    my ( $key, $val ) = $line =~ m/^(.*?)\s?[=:\t]\s?(.*)$/;
    $inject{$key} = $val;
  }
  close($fh);

  return \%inject;
}

sub _get_ext {
  my ($file) = @_;

  if ( $file =~ m/\.tar\.gz$/ ) {
    return ".tar.gz";
  }
  elsif ( $file =~ m/\.zip$/ ) {
    return ".zip";
  }
  elsif ( $file =~ m/\.tar\.bz2$/ ) {
    return ".tar.bz2";
  }
  elsif ( $file =~ m/\.war$/ ) {
    return ".war";
  }
  elsif ( $file =~ m/\.jar$/ ) {
    return ".jar";
  }

  die("Unknown Archive Format.");

}

####### import function #######

sub import {

  no strict 'refs';
  for my $func (@EXPORT) {
    Rex::Logger::debug("Registering main::$func");
    *{"$_[1]::$func"} = \&$func;
  }

}

1;