Changeset 3707 for lang/perl/Archer

Timestamp:

12/28/07 11:23:49 (11 months ago)

Author:

tokuhirom

Message:

r3854@mnk (orig r504): tokuhiro | 2007-08-03 22:03:01 -0700
apply the franck patch.thanks franck++.
added configmaker & recipe.

Location:

lang/perl/Archer

Files:

• archer.pl (modified) (2 diffs)
• assets/kwalify/schema.yaml (modified) (1 diff)
• assets/recipe (added)
• assets/recipe/Deploy (added)
• lib/Archer.pm (modified) (6 diffs)
• lib/Archer/Plugin.pm (modified) (3 diffs)
• lib/Archer/Plugin/Exec.pm (modified) (1 diff)

lang/perl/Archer/archer.pl

@@ -7 +7 @@
 use FindBin ();
 use Path::Class;
+use File::Util;
+use Cwd;
 
-use lib dir($FindBin::RealBin, 'lib')->stringify;
+use lib dir( $FindBin::RealBin, 'lib' )->stringify;
 
 use Archer;
@@ -14 +16 @@
 my $argv_str = "@ARGV";
 my $fork_num = 1;
-my $config = file($FindBin::RealBin, 'config.yaml')->stringify;
+my $config   = file( $FindBin::RealBin, 'config.yaml' )->stringify;
 Getopt::Long::GetOptions(
-    '--para=i'         => \$fork_num,
-    '--dry-run'        => \my $dry_run_fg,
-    '--skip=s'         => \my $skips,
-    '--shell',         => \my $shell,
-    '--man'            => \my $man,
-    '--config=s'       => \$config,
-  )
-  or pod2usage(2);
-Getopt::Long::Configure("bundling");    # allows -p
-pod2usage(-verbose => 2) if $man;
-pod2usage(2) unless @ARGV;
+    '--para=i'       => \$fork_num,
+    '--dry-run'      => \my $dry_run_fg,
+    '--skip=s'       => \my $skips,
+    '--shell',       => \my $shell,
+    '--man'          => \my $man,
+    '--config=s'     => \$config,
+    '--write-config' => \my $wc,
+) or pod2usage( 2 );
+Getopt::Long::Configure( "bundling" );    # allows -p
+pod2usage( -verbose => 2 ) if $man;
 
-for my $proj (@ARGV) {
+if ( !@ARGV ) {
+
+    # name of the current dir, will be the project name
+    my $dir     = getcwd;
+    my $f       = File::Util->new;
+    my $project = $f->strip_path( $dir );
+    $config = '.archer.yaml';
+    if ( $f->existent( $config ) ) {
+        Archer->new(
+            {   project      => $project,
+                dry_run_fg   => $dry_run_fg,
+                parallel_num => $fork_num,
+                skips => +{ map { $_ => 1 } split /,/, ( $skips || '' ) },
+                config_yaml => $config,
+                argv_str    => $argv_str,
+                shell       => $shell,
+            }
+        )->run;
+        exit;
+    }
+    else {
+        Archer->new(
+            {   project      => $project,
+                dry_run_fg   => $dry_run_fg,
+                config_yaml  => $config,
+                write_config => $wc,
+            }
+        )->run;
+        exit;
+    }
+    pod2usage( 2 ) unless @ARGV;
+}
+
+for my $proj ( @ARGV ) {
     Archer->new(
-        {
-            project        => $proj,
-            dry_run_fg     => $dry_run_fg,
-            parallel_num   => $fork_num,
-            skips          => +{ map { $_ => 1 } split /,/, ($skips || '')},
-            config_yaml    => $config,
-            argv_str       => $argv_str,
-            shell          => $shell,
+        {   project      => $proj,
+            dry_run_fg   => $dry_run_fg,
+            parallel_num => $fork_num,
+            skips        => +{ map { $_ => 1 } split /,/, ( $skips || '' ) },
+            config_yaml  => $config,
+            argv_str     => $argv_str,
+            shell        => $shell,
+            write_config => $wc,
         }
     )->run;

lang/perl/Archer/assets/kwalify/schema.yaml

@@ -15 +15 @@
         name: dest_dir
         desc: dest. directory
+        type: str
+      "recipe":
+        name: recipe
+        desc: additional recipe path
         type: str
       "log":

lang/perl/Archer/lib/Archer.pm

@@ -11 +11 @@
 my $context;
 sub context { $context }
+
 sub set_context {
-    my ($class, $c) = @_;
+    my ( $class, $c ) = @_;
     $context = $c;
 }
 
 sub new {
-    my ($class, $opts) = @_;
-    my $self = bless {%$opts}, $class;
+    my ( $class, $opts ) = @_;
+    my $self = bless { %$opts }, $class;
 
-    my $config_loader = Archer::ConfigLoader->new;
-    $self->{config} = $config_loader->load($opts->{config_yaml});
-    $self->{config}->{global}->{log} ||= { level => 'debug' };
+    if ( !$$opts{ write_config } ) {
+        my $config_loader = Archer::ConfigLoader->new;
+        $self->{ config } = $config_loader->load( $opts->{ config_yaml } );
+    }
+    $self->{ config }->{ global }->{ log } ||= { level => 'debug' };
 
-    Archer->set_context($self);
+    Archer->set_context( $self );
 
     return $self;
@@ -30 +33 @@
 
 sub run {
-    my ($self, ) = @_;
+    my ( $self, ) = @_;
 
-    if ($self->{shell}) {
+    if ( $self->{ shell } ) {
+
         # TODO: role support
         require Archer::Shell;
-        my @servers = map { @{$_} } values %{ $context->{config}->{projects}->{$self->{project}} };
+        my @servers
+            = map { @{ $_ } }
+            values
+            %{ $context->{ config }->{ projects }->{ $self->{ project } } };
         my $shell = Archer::Shell->new(
             {   context => $self,
-                config  => $self->{config},
+                config  => $self->{ config },
                 servers => \@servers,
             }
@@ -44 +51 @@
 
         $shell->run_loop;
-    } else {
-        $self->run_hook('init');
+    }
+    elsif ( $self->{ write_config } ) {
+        require Archer::Util;
+        my $util = Archer::Util->new;
+        $util->templatize( $self );
+    }
+    else {
+        $self->run_hook( 'init' );
 
         $self->run_process;
 
-        $self->run_hook('finalize');
+        $self->run_hook( 'finalize' );
     }
 }
 
 sub run_hook {
-    my ($self, $hook, $args) = @_;
+    my ( $self, $hook, $args ) = @_;
     $args ||= {};
 
-    $self->log('info' => "run hook $hook");
-    for my $plugin (@{ $self->{config}->{tasks}->{$hook} }) {
-        if ($self->{skips}->{$plugin->{name}}) {
-            $self->log(info => "skipped: $plugin->{name}");
+    $self->log( 'info' => "run hook $hook" );
+    for my $plugin ( @{ $self->{ config }->{ tasks }->{ $hook } } ) {
+        if ( $self->{ skips }->{ $plugin->{ name } } ) {
+            $self->log( info => "skipped: $plugin->{name}" );
             next;
         }
 
-        if ( $plugin->{role} && $plugin->{role} ne $args->{role} ) {
-            $self->log(debug => "skip $args->{server}. because $plugin->{role} ne $args->{role}");
+        if ( $plugin->{ role } && $plugin->{ role } ne $args->{ role } ) {
+            $self->log( debug =>
+                    "skip $args->{server}. because $plugin->{role} ne $args->{role}"
+            );
             next;
         }
 
         my $class = "Archer::Plugin::$plugin->{module}";
-        $self->log('debug' => "load $class");
+        $self->log( 'debug' => "load $class" );
         $class->use or die $@;
 
-        $self->log('info' => "run $class");
+        $self->log( 'info' => "run $class" );
         $class->new(
-            { config => $plugin->{config}, project => $self->{project}, %$args } )
-            ->run( $self, $args );
+            {   config  => $plugin->{ config },
+                project => $self->{ project },
+                %$args
+            }
+        )->run( $self, $args );
 
-        print "\n\n"; # for debug.
+        print "\n\n";    # for debug.
     }
 }
 
 sub run_process {
-    my ($self) = @_;
+    my ( $self ) = @_;
 
-    my $parallel = $self->{config}->{global}->{parallel} || 'Archer::Parallel::ForkManager';
+    my $parallel = $self->{ config }->{ global }->{ parallel }
+        || 'Archer::Parallel::ForkManager';
     $parallel->use or die $@;
 
     # construct elements
-    my $server_tree = $self->{config}->{projects}->{$self->{project}};
+    # this one doesn't work for me
+    # my $server_tree = $self->{config}->{projects}->{$self->{project}};
+    # but this one do
+    my $server_tree = $self->{ config }->{ projects };
+
     my @elems;
-    while (my ($role, $servers) = each %$server_tree) {
-        for my $server (@$servers) {
-            push @elems, {server => $server, role => $role};
+    while ( my ( $role, $servers ) = each %$server_tree ) {
+        for my $server ( @$servers ) {
+            push @elems, { server => $server, role => $role };
         }
     }
-
-    $self->log(debug => "run parallel : $self->{parallel_num}");
+    $self->log( debug => "run parallel : $self->{parallel_num}" );
     my $manager = $parallel->new;
     $manager->run(
-        {   elems => \@elems,
+        {   elems    => \@elems,
             callback => sub {
                 my $args = shift;
                 $self->run_hook( 'process', $args );
-                },
-            num => $self->{parallel_num},
+            },
+            num => $self->{ parallel_num },
         }
     );
@@ -111 +133 @@
 
 sub bootstrap {
-    my ($class, $opts) = @_;
+    my ( $class, $opts ) = @_;
 
-    my $self = $class->new($opts);
+    my $self = $class->new( $opts );
     $self->run;
     return $self;
@@ -122 +144 @@
     my ( $self, $level, $msg, %opt ) = @_;
 
-    return unless $self->should_log($level);
+    return unless $self->should_log( $level );
 
     # hack to get the original caller as Plugin or Rule
     # from plagger.
-    my $caller = $opt{caller};
-    unless ($caller) {
+    my $caller = $opt{ caller };
+    unless ( $caller ) {
         my $i = 0;
-        while (my $c = caller($i++)) {
+        while ( my $c = caller( $i++ ) ) {
             last if $c !~ /Plugin|Rule/;
             $caller = $c;
         }
-        $caller ||= caller(0);
+        $caller ||= caller( 0 );
     }
 
@@ -147 +169 @@
 
 sub should_log {
-    my($self, $level) = @_;
+    my ( $self, $level ) = @_;
 
-    $levels{$level} >= $levels{ $self->{config}->{global}->{log}->{level} };
+    $levels{ $level }
+        >= $levels{ $self->{ config }->{ global }->{ log }->{ level } };
 }
 

lang/perl/Archer/lib/Archer/Plugin.pm

@@ -9 +9 @@
 
 sub new {
-    my ($class, $args) = @_;
-    bless {%$args}, $class;
+    my ( $class, $args ) = @_;
+    bless { %$args }, $class;
 }
 
 sub log {
     my $self = shift;
-    Archer->context->log(@_);
+    Archer->context->log( @_ );
 }
 
 sub templatize {
-    my ($self, $cmd) = @_;
+    my ( $self, $cmd ) = @_;
 
     my $vars = {
-        config  => $self->{config},
-        project => $self->{project},
+        config    => $self->{ config },
+        project   => $self->{ project },
         l_project => $self->l_project,
-        work_dir => Archer->context->{config}->{global}->{work_dir},
-        dest_dir => Archer->context->{config}->{global}->{dest_dir},
-        server => $self->{server},
-        user   => $ENV{USER},
+        work_dir  => Archer->context->{ config }->{ global }->{ work_dir },
+        dest_dir  => Archer->context->{ config }->{ global }->{ dest_dir },
+        server    => $self->{ server },
+        user      => $ENV{ USER },
     };
 
     my $tt = Template->new;
-    $tt->process( \$cmd, $vars, \my $out ) or $self->log('error' => 'Tmplate Error: ' . $tt->error);
+    $tt->process( \$cmd, $vars, \my $out )
+        or $self->log( 'error' => 'Template Error: ' . $tt->error );
 
     $out;
@@ -38 +39 @@
 
 sub detach {
-    my ($self, $msg) = @_;
+    my ( $self, $msg ) = @_;
 
     croak "$msg\n";
@@ -45 +46 @@
 # FIXME: so bad...following method...
 sub l_project {
-    my ($self, ) = @_;
+    my ( $self, ) = @_;
 
-    my $work_dir = Archer->context->{config}->{global}->{work_dir};
+    my $work_dir = Archer->context->{ config }->{ global }->{ work_dir };
 
-    my $lc = String::CamelCase::decamelize( $self->{project} );
-    if ( -e file( $work_dir, $self->{project}, $lc )
-        ->stringify )
-    {
+    my $lc = String::CamelCase::decamelize( $self->{ project } );
+    if ( -e file( $work_dir, $self->{ project }, $lc )->stringify ) {
         return $lc;
     }
     else {
-        return lc Archer->context->{project};
+        return lc Archer->context->{ project };
     }
 }
 
+sub check_recipe {
+    my ( $self, $recipe_name, $altern_path ) = @_;
+    my ( $f, $path );
+
+    $f = File::Util->new;
+    $path = File::Spec->catfile( $FindBin::Bin, 'assets', 'recipe',
+        $recipe_name );
+
+    # check first in assets
+    if ( $f->existent( $path ) ) {
+        return $f->load_file( $path );
+    }
+
+    # if there is another path for recipe in the config, check this one
+    $path = File::Spec->catfile( $altern_path, $recipe_name );
+    if ( $f->existent( $path ) ) {
+        return $f->load_file( $path );
+    }
+
+    # fail
+    return undef;
+}
+
 1;

lang/perl/Archer/lib/Archer/Plugin/Exec.pm

@@ -7 +7 @@
 
 sub run {
-    my ($self, $context, $args) = @_;
+    my ( $self, $context, $args ) = @_;
+    my $cmd;
 
-    my $cmd = $self->{config}->{command};
-    $self->log(debug => "template: $cmd");
+    if ( $self->{ config }->{ command } ) {
+        $cmd = $self->{ config }->{ command };
+        $self->log( debug => "template: $cmd" );
 
-    $cmd = $self->templatize($cmd);
-    $self->log(info => "* execute " . colored($cmd, 'red'));
+        $cmd = $self->templatize( $cmd );
+        $self->log( info => "* execute " . colored( $cmd, 'red' ) );
+    }
+    elsif ( $self->{ config }->{ recipe } ) {
+        require Archer::Util;
+        $cmd = $self > check_recipe( $self->{ config }->{ recipe },
+            $context->{ config }->{ global }->{ recipe } );
 
-    if ($context->{dry_run_fg}) {
-        $self->log(debug => "dry-run");
+        if ( !defined $cmd ) {
+            $self->log(
+                'warn' => 'The recipe '
+                    . $self->{ config }->{ recipe } . '
+                can\'t be found'
+            );
+            return;
+        }
+        $cmd = $self->templatize( $cmd );
+        warn $cmd;
+    }
+
+    if ( $context->{ dry_run_fg } ) {
+        $self->log( debug => "dry-run" );
     }
     else {
-        $self->log(debug => "run!");
-
-        $self->_execute($_) for grep !/^\s*$/, split /\n/, $cmd;
+        $self->log( debug => "run!" );
+        if ( $cmd ) {
+            $self->_execute( $_ ) for grep !/^\s*$/, split /\n/, $cmd;
+        }
     }
 }