le1t0-capistrano 2.5.18.008 → 2.5.18.009

data/CHANGELOG

@@ -1,3 +1,7 @@
+== 2.5.18.009 / Jul 22, 2010
+
+Added TeeIO class which can catch all output to $stderr/$stdout/etc, so the output can be emailed/stored
+
 == 2.5.18.008 / Jul 1, 2010
 
 Actually merged in new code

data/VERSION

@@ -1 +1 @@
-2.5.18.008
+2.5.18.009

data/bin/cap

@@ -1,4 +1,12 @@
 #!/usr/bin/env ruby
 
+require 'capistrano/cli/tee_io'
 require 'capistrano/cli'
+
+Signal.trap("EXIT") do
+  TeeIO.output_all # make sure we request the output, so file pointers get closed also if not done already, just throw away the
+                   # result
+end
+TeeIO.tee_all
+
 Capistrano::CLI.execute

data/lib/capistrano/cli/tee_io.rb

@@ -0,0 +1,88 @@
+require 'stringio'
+class TeeIO
+  attr_accessor :old_io
+  attr_accessor :pipe_io
+  attr_accessor :teed_io_name
+  attr_accessor :auto_close_pipe_io
+
+  class << self
+
+    def tee_all(pipe_io = nil, &block)
+      if block_given?
+        tee("$stdout", pipe_io) do |io|
+          tee("$stderr", io) do
+            yield
+          end
+        end
+      else
+        io = tee("$stdout", pipe_io)
+        tee("$stderr", io)
+      end
+    end
+
+    def tee(teed_io_name, pipe_io = nil, &block)
+      eval("#{teed_io_name} = self.new(teed_io_name, pipe_io || StringIO.new, pipe_io.nil?)")
+      if block_given?
+        yield(eval("#{teed_io_name}.pipe_io"))
+        eval("#{teed_io_name}.output")
+      else
+        eval("#{teed_io_name}.pipe_io")        
+      end
+    end
+
+    def output_all
+      $stderr.output if $stderr.is_a?(TeeIO)
+      $stdout.output if $stdout.is_a?(TeeIO)
+    end
+
+  end
+
+  def initialize(*args)
+    @teed_io_name = args[0]
+    @old_io = eval(args[0])
+    @pipe_io = args[1]
+    @auto_close_pipe_io = args[2]
+  end
+
+  def call_on_io(method_name, *args, &block)
+    call_on_old_io(method_name, *args, &block)
+    call_on_pipe_io(method_name, *args, &block)
+  end
+
+  def call_on_old_io(method_name, *args, &block)
+    if @old_io.is_a?(IO) || @old_io.is_a?(StringIO)
+      @old_io.send(method_name, *args, &block)
+      @old_io.flush
+    end
+  end
+
+  def call_on_pipe_io(method_name, *args, &block)
+    if @pipe_io.is_a?(IO) || @pipe_io.is_a?(StringIO)
+      @pipe_io.send(method_name, *args, &block)
+      @pipe_io.flush
+    end
+  end
+
+  def write(str)
+    call_on_io(:write, str)
+  end
+
+  def method_missing(method_name, *args, &block)
+    call_on_io(method_name, *args, &block)
+  end
+
+  def output
+    reset_io
+    if @auto_close_pipe_io && @pipe_io.is_a?(StringIO)
+      str = @pipe_io.string
+      @pipe_io.close
+      str
+    else
+      @pipe_io
+    end
+  end
+
+  def reset_io
+    eval("#{teed_io_name} = @old_io")
+  end
+end

metadata

@@ -6,8 +6,8 @@ version: !ruby/object:Gem::Version
   - 2
   - 5
   - 18
-  - 8
-  version: 2.5.18.008
+  - 9
+  version: 2.5.18.009
 platform: ruby
 authors: 
 - Le1t0
@@ -15,7 +15,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2010-07-01 00:00:00 +02:00
+date: 2010-07-22 00:00:00 +02:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -123,6 +123,7 @@ files:
 - lib/capistrano/cli/help.rb
 - lib/capistrano/cli/help.txt
 - lib/capistrano/cli/options.rb
+- lib/capistrano/cli/tee_io.rb
 - lib/capistrano/cli/ui.rb
 - lib/capistrano/command.rb
 - lib/capistrano/configuration.rb