mirror of
https://github.com/Homebrew/brew.git
synced 2025-07-14 16:09:03 +08:00
68 lines
1.8 KiB
Ruby
68 lines
1.8 KiB
Ruby
# typed: false
|
|
# frozen_string_literal: true
|
|
|
|
module Utils
|
|
IO_DEFAULT_BUFFER_SIZE = 4096
|
|
private_constant :IO_DEFAULT_BUFFER_SIZE
|
|
|
|
def self.popen_read(*args, **options, &block)
|
|
popen(args, "rb", options, &block)
|
|
end
|
|
|
|
def self.safe_popen_read(*args, **options, &block)
|
|
output = popen_read(*args, **options, &block)
|
|
return output if $CHILD_STATUS.success?
|
|
|
|
raise ErrorDuringExecution.new(args, status: $CHILD_STATUS, output: [[:stdout, output]])
|
|
end
|
|
|
|
def self.popen_write(*args, **options)
|
|
popen(args, "w+b", options) do |pipe|
|
|
output = ""
|
|
|
|
# Before we yield to the block, capture as much output as we can
|
|
loop do
|
|
output += pipe.read_nonblock(IO_DEFAULT_BUFFER_SIZE)
|
|
rescue IO::WaitReadable, EOFError
|
|
break
|
|
end
|
|
|
|
yield pipe
|
|
pipe.close_write
|
|
IO.select([pipe])
|
|
|
|
# Capture the rest of the output
|
|
output += pipe.read
|
|
output.freeze
|
|
end
|
|
end
|
|
|
|
def self.safe_popen_write(*args, **options, &block)
|
|
output = popen_write(*args, **options, &block)
|
|
return output if $CHILD_STATUS.success?
|
|
|
|
raise ErrorDuringExecution.new(args, status: $CHILD_STATUS, output: [[:stdout, output]])
|
|
end
|
|
|
|
def self.popen(args, mode, options = {})
|
|
IO.popen("-", mode) do |pipe|
|
|
if pipe
|
|
return pipe.read unless block_given?
|
|
|
|
yield pipe
|
|
else
|
|
options[:err] ||= :close unless ENV["HOMEBREW_STDERR"]
|
|
begin
|
|
exec(*args, options)
|
|
rescue Errno::ENOENT
|
|
$stderr.puts "brew: command not found: #{args[0]}" unless options[:err] == :close
|
|
exit! 127
|
|
rescue SystemCallError
|
|
$stderr.puts "brew: exec failed: #{args[0]}" unless options[:err] == :close
|
|
exit! 1
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|