sample Takfile
Matt S Trout [Fri, 11 Nov 2011 12:55:08 +0000 (12:55 +0000)]
Takfile [new file with mode: 0644]

diff --git a/Takfile b/Takfile
new file mode 100644 (file)
index 0000000..62b8135
--- /dev/null
+++ b/Takfile
@@ -0,0 +1,57 @@
+package Tak::MyScript;
+
+use Tak::Takfile;
+
+sub every_exec (stream|s) {
+  my ($self, $remotes, $options, @command) = @_;
+
+  my @requests;
+
+  $_->ensure(command_service => 'Tak::CommandService') for @$remotes;
+
+  foreach my $remote (@$remotes) {
+    if ($options->{stream}) {
+      my $stdout = $self->stdout;
+      my $host = $remote->host;
+      push @requests, $remote->start(
+        {
+          on_result => sub { $self->print_exec_result($remote, @_) },
+          on_progress => sub {
+            $stdout->print($host.' '.$_[0].': '.$_[1]);
+            $stdout->print("\n") unless $_[1] =~ /\n\Z/;
+          }
+        },
+        command_service => stream_exec => \@command
+      );
+    } else {
+      push @requests, $remote->start(
+        { on_result => sub { $self->print_exec_result($remote, @_) } },
+        command_service => exec => \@command
+      );
+    }
+  }
+  Tak->await_all(@requests);
+}
+
+sub print_exec_result {
+  my ($self, $remote, $result) = @_;
+
+  my $res = eval { $result->get }
+    or do {
+      $self->stderr->print("Host ${\$remote->host}: Error: $@\n");
+      return;
+    };
+
+  my $code = $res->{exit_code};
+  $self->stdout->print(
+    "Host ${\$remote->host}: ".($code ? "NOT OK ${code}" : "OK")."\n"
+  );
+  if ($res->{stderr}) {
+    $self->stdout->print("Stderr:\n${\$res->{stderr}}\n");
+  }
+  if ($res->{stdout}) {
+    $self->stdout->print("Stdout:\n${\$res->{stdout}}\n");
+  }
+}
+
+1;