Changeset 3701 for lang/perl/Archer

Timestamp:

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

Author:

tokuhirom

Message:

r3654@mnk (orig r304): tokuhiro | 2007-03-05 21:15:30 -0800
apply franck's patch.

Files:

• lang/perl/Archer/lib/Archer/Shell.pm (modified) (2 diffs)

lang/perl/Archer/lib/Archer/Shell.pm

@@ -20 +20 @@
     my $HISTSIZE = 256;
 
-        # this won't work with Term::ReadLine::Perl
-        # If there is Term::ReadLine::Gnu, be sure to do : export "PERL_RL=Gnu o=0"
-        eval { $term->stifle_history($HISTSIZE);};
-        
-        if (@!){
-                $self->{context}->log('debug' => "You will need Term::ReadLine::Gnu");
-        }else{
-        if (-f $HISTFILE) {
-                $term->ReadHistory($HISTFILE) or $self->{context}->log('warn' => "cannot read history file: $!");
-        }
-        }
-
-    while ( defined( my $line = $term->readline('archer> ') )) {
+   # this won't work with Term::ReadLine::Perl
+   # If there is Term::ReadLine::Gnu, be sure to do : export "PERL_RL=Gnu o=0"
+    eval { $term->stifle_history( $HISTSIZE ); };
+
+    if ( @! ) {
+        $self->{ context }->log( 'debug' => "You will need Term::ReadLine::Gnu" );
+    } else {
+        if ( -f $HISTFILE ) {
+            $term->ReadHistory( $HISTFILE )
+                or $self->{ context }->log( 'warn' => "cannot read history file: $!" );
+        }
+    }
+
+    while ( defined( my $line = $term->readline( 'archer> ' ) ) ) {
         next if $line =~ /^\s*$/;
-        $self->catch_run($line);
+        $self->catch_run( $line );
     }
 
     print "\n";
 
-        eval {$term->WriteHistory($HISTFILE);};
-        if (@!){
-                $self->{context}->log('debug' => "perlsh: cannot write history file: $!");
-        }
+    eval { $term->WriteHistory( $HISTFILE ); };
+    if ( @! ) {
+        $self->{context}->log( 'debug' => "perlsh: cannot write history file: $!" );
+    }
 
 }
 
 sub catch_run {
-    my ($self, $cmd) = @_;
-
-    my $parallel = $self->{context}->{config}->{global}->{parallel}
+    my ( $self, $cmd ) = @_;
+
+    $self->{ parallel }
+        = $self->{ context }->{ config }->{ global }->{ parallel }
         || 'Archer::Parallel::ForkManager';
-    $parallel->use or die $@;
-
-    my $manager = $parallel->new;
-    $manager->run({
-        elems => $self->{servers},
-        callback => sub {
-            my $server = shift;
-            $self->callback($server, $cmd);
-        },
-        num => $self->{context}->{parallel_num},
-    });
+    $self->{ parallel }->use or die $@;
+
+    if ( $cmd =~ /^on/ ) {
+        if ( $cmd =~ /^on\s(.*)\sdo\s(.*)$/ ) {
+            $self->process_host( $1, $2 );
+        } else {
+            print "[WARNING] error in your syntax, see help\n";
+        }
+    } elsif ( $cmd =~ /^with/ ) {
+        if ( $cmd =~ /^with\s(.*)\sdo\s(.*)$/ ) {
+            $self->process_role( $1, $2 );
+        } else {
+            print "[WARNING] error in your syntax, see help\n";
+        }
+    } elsif ( $cmd =~ /^help/ ) {
+        $self->help();
+    } elsif ( $cmd =~ /^(quit|exit)/ ) {
+        print "bye bye\n";
+        exit;
+    } elsif ( $cmd =~ /^!/ ) {
+        if ( $cmd =~ /^!(\w+)\s?(on|with)?\s?(.*)?$/ ) {
+            my $task     = $1;
+            my $action   = $2;
+            my $machines = $3;
+            if (    defined $action
+                 && defined $machines
+                 && length( $machines ) < 1 )
+            {
+                return print "[WARNING] error in your syntax, see help\n";
+            }
+            my $executed = 0;
+            for
+                my $plugin ( @{ $self->{ config }->{ tasks }->{ process } } )
+            {
+                next if $plugin->{ name } ne $task;
+                $executed = 1;
+                if ( defined $action ) {
+                    if ( $action eq "on" ) {
+                        my @hosts = split " ", $machines;
+                        for my $host ( @hosts ) {
+                            $self->process_task( $plugin, $host );
+                        }
+                    } else {
+                        my @roles = split " ", $machines;
+                        for my $role ( @roles ) {
+                            for my $host (
+                                          @{ $self->{ servers }->{ $role } } )
+                            {
+                                $self->process_task( $plugin, $host );
+                            }
+                        }
+                    }
+                } else {
+                    for my $host (
+                                   @{  $self->{ servers }
+                                           ->{ $plugin->{ config }->{ role } }
+                                   } )
+                    {
+                        $self->process_task( $plugin, $host );
+                    }
+                }
+            }
+            if ( $executed == 0 ) {
+                print "[WARNING] unable to find the requested task: $task\n";
+            }
+        } else {
+            print "[WARNING] error in your syntax\n";
+        }
+    } else {
+        $self->process_command( $cmd );
+    }
+}
+
+sub process_host {
+    my ( $self, $hosts, $cmd ) = @_;
+
+    my @hosts = split /\s/, $hosts;
+
+    # check if hosts are in our config.
+    for my $host ( @hosts ) {
+        for my $role ( keys %{ $self->{ servers } } ) {
+            @hosts = grep ( /$host/, @{ $self->{ servers }->{ $role } } );
+        }
+    }
+
+    if ( @hosts ) {
+        $self->process_command( $cmd, \@hosts );
+    }
+}
+
+sub process_role {
+    my ( $self, $roles, $cmd ) = @_;
+
+    my @roles      = split /\s/, $roles;
+    my @hosts      = ();
+    my @inexistant = ();
+    for my $role ( @roles ) {
+        if ( !defined $self->{ servers }->{ $role } ) {
+            push( @inexistant, $role );
+            next;
+        }
+        for my $host ( @{ $self->{ servers }->{ $role } } ) {
+            push @hosts, $host;
+        }
+    }
+    if ( @inexistant ) {
+        print "[WARNING] inexisting role(s) for "
+            . join( ' ', @inexistant ) . "\n";
+    }
+    $self->process_command( $cmd, \@hosts );
+}
+
+sub process_command {
+    my ( $self, $cmd, $hosts ) = @_;
+    my $manager = $self->{parallel}->new;
+
+    if ( !$hosts ) {
+        for my $role (
+            keys
+            %{ $self->{config}->{projects}->{ $self->{context}->{project} } }
+            )
+        {
+            for my $host (
+                @{  $self->{config}->{projects}
+                        ->{ $self->{context}->{project} }->{$role}
+                }
+                )
+            {
+                push @{$hosts}, $host;
+            }
+        }
+    }
+
+    $manager->run(
+        {   elems    => $hosts,
+            callback => sub {
+                my $server = shift;
+                $self->callback( $server, $cmd );
+            },
+            num => $self->{context}->{parallel_num},
+        }
+    );
+}
+
+sub process_task {
+    my ( $self, $plugin, $host ) = @_;
+    my $class = "Archer::Plugin::$plugin->{module}";
+    $class->use or die $@;
+    $class->new( { config  => $plugin->{ config },
+                   project => $self->{ context }->{ project },
+                   server  => $host
+                 } )->run( $self->{ context } );
 }
 
@@ -76 +219 @@
 }
 
+sub help {
+    my ( $self ) = @_;
+    my $help = <<HELP;
+ To quit, just type quit, exit, or press ctrl-D. 
+ This shell is still experimental.
+
+ execute a command on all servers, just type it directly, like:
+
+archer> ping
+
+ To execute a command on a specific set of servers, specify an 'on' clause.
+ Note that if you specify more than one host name, they must be 
+ space-delimited.
+
+archer> on app1.foo.com app2.foo.com do ping
+
+ To execute a command on all servers matching a set of roles:
+
+archer> with web db do ping
+
+ To execute an Archer task, prefix the name with a bang, by default it
+ will be executed only on the role applyed to this task.
+
+archer> !restart
+
+ To execute an Archer task on a specific set of servers:
+
+archer> !restart on app1.foo.com app2.foo.com
+
+ To execute an Archer task on all servers matching a set of roles:
+
+archer> !restart with web db
+
+HELP
+    print $help;
+}
+
 1;
 __END__