Sha256: 8b53b941773e37b244d490601293443bf67ee5138ae89def59d479b0b566872e

Contents?: true

Size: 942 Bytes

Versions: 7

Compression:

Stored size: 942 Bytes

Contents

require 'darkext/hash'

module DarkIO
  # Runs a block and captures the output it generates
  def self.capture_output(opts = { }) # yield e
    opts.with_defaults!(:stdout => true, :stderr => false)
    cout,cerr = opts[:stdout],opts[:stderr]

    yield and return if !cout && !cerr

    old_stdout = STDOUT.dup if cout
    old_stderr = STDERR.dup if cerr

    r_stdout,w_stdout = IO.pipe if cout
    r_stderr,w_stderr = IO.pipe if cerr

    STDOUT.reopen(w_stdout) if cout
    STDERR.reopen(w_stderr) if cerr

    begin
      yield
    ensure
      STDOUT.reopen(old_stdout) if cout
      STDERR.reopen(old_stderr) if cerr
      w_stdout.close if cout
      w_stderr.close if cerr
    end

    ret_stdout = r_stdout.read if cout
    ret_stderr = r_stderr.read if cerr
    r_stdout.close if cout
    r_stderr.close if cerr
    return ret_stdout,ret_stderr if cout && cerr
    return ret_stdout if cout
    return ret_stderr if cerr
  end
end

Version data entries

7 entries across 7 versions & 1 rubygems

Version Path
darkhelmet-darkext-0.10.0 lib/darkext/io.rb
darkhelmet-darkext-0.11.0 lib/darkext/io.rb
darkhelmet-darkext-0.11.1 lib/darkext/io.rb
darkhelmet-darkext-0.11.2 lib/darkext/io.rb
darkhelmet-darkext-0.9.5 lib/darkext/io.rb
darkhelmet-darkext-0.9.6 lib/darkext/io.rb
darkhelmet-darkext-0.9.7 lib/darkext/io.rb