io-capture 0.1.0

data/lib/io-capture.rb

@@ -0,0 +1,88 @@
+class Capture
+
+  class << self
+    attr_accessor :currently_capturing
+  end
+
+  Capture.currently_capturing ||= {}
+
+  def self.[] name
+    Capture.currently_capturing[name]
+  end
+
+  def self.[]= name, value
+    Capture.currently_capturing[name] = value
+  end
+
+  def self.do io = nil, &block
+    io = :stdout if io.nil?
+
+    if global = is_a_global_variable_and_IO?(io)
+      start io
+      block.call
+      stop io
+    else
+      raise "Not sure how to capture writes to io: #{ io.inspect }"
+    end
+  end
+
+  def self.is_a_global_variable_and_IO? io
+    io = '$' + io.to_s.sub(/^\$/,'')
+    if global_variables.include?(io) and eval(io).respond_to?(:write)
+      io
+    else
+      nil
+    end
+  end
+
+  def self.start io = nil
+    io = :stdout if io.nil?
+    if global = is_a_global_variable_and_IO?(io)
+
+      # backup original value of global
+      Capture[global] = eval(global).clone
+
+      # reset global to a StringIO for capturing
+      eval("#{global} = StringIO.open '', 'w'")
+      
+      # simple return nil
+      nil
+
+    end
+  end
+
+  def self.stop io = nil
+    io = :stdout if io.nil?
+    if global = is_a_global_variable_and_IO?(io)
+
+      # get captured output
+      output = eval(global).string
+
+      # reset global to backed up original
+      eval("#{global} = Capture[#{ global.inspect }]")
+
+      # we're no longer capturing ... delete from hash
+      currently_capturing.delete global
+
+      # reutrn the output
+      return output
+
+    end
+  end
+
+end
+
+# Captures STDOUT or STDERR and returns the output.
+#
+#  >> Capture { puts 'hello' }
+#  => 'hello'
+#
+#  >> Capture(:stdout){ puts 'hello stdout' }
+#  => 'hello stdout'
+#
+#  >> Capture(:stderr){ warn 'hello stderr' }
+#  => 'hello stderr'
+#
+def Capture io = nil, &block
+  Capture.do io, &block
+end

metadata

@@ -0,0 +1,62 @@
+--- !ruby/object:Gem::Specification 
+name: io-capture
+version: !ruby/object:Gem::Version 
+  prerelease: false
+  segments: 
+  - 0
+  - 1
+  - 0
+  version: 0.1.0
+platform: ruby
+authors: 
+- remi
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-03-11 00:00:00 -07:00
+default_executable: 
+dependencies: []
+
+description: 
+email: remi@remitaylor.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- lib/io-capture.rb
+has_rdoc: true
+homepage: http://github.com/remi/capture
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.6
+signing_key: 
+specification_version: 3
+summary: easily capture writes to STDOUT, Files, or any IO
+test_files: []
+