mirror of
https://github.com/ruby/ruby.git
synced 2022-11-09 12:17:21 -05:00
15 lines
297 B
Ruby
15 lines
297 B
Ruby
|
# frozen_string_literal: true
|
||
|
|
require "stringio"
|
||
|
|
|
||
|
|
def capture(*streams)
|
||
|
|
streams.map!(&:to_s)
|
||
|
|
begin
|
||
|
|
result = StringIO.new
|
||
|
|
streams.each {|stream| eval "$#{stream} = result" }
|
||
|
|
yield
|
||
|
|
ensure
|
||
|
|
streams.each {|stream| eval("$#{stream} = #{stream.upcase}") }
|
||
|
|
end
|
||
|
|
result.string
|
||
|
|
end
|