20 lines
425 B
Ruby
20 lines
425 B
Ruby
|
module Kernel
|
||
|
unless method_defined?(:capture)
|
||
|
def capture(stream)
|
||
|
stream = stream.to_s
|
||
|
captured_stream = Tempfile.new(stream)
|
||
|
stream_io = eval("$#{stream}")
|
||
|
origin_stream = stream_io.dup
|
||
|
stream_io.reopen(captured_stream)
|
||
|
|
||
|
yield
|
||
|
|
||
|
stream_io.rewind
|
||
|
return captured_stream.read
|
||
|
ensure
|
||
|
captured_stream.unlink
|
||
|
stream_io.reopen(origin_stream)
|
||
|
end
|
||
|
end
|
||
|
end
|