Модуль: Utils Частный

143
144
145
146
147

# File 'utils.rb', line 143

def self.convert_to_string_or_symbol(string)
  return T.must(string[1..]).to_sym if string.start_with?(":")

  string
end

17
18
19

# File 'utils.rb', line 17

def self.deconstantize(path)
  T.must(path[0, path.rindex("::") || 0]) # implementation based on the one in facets' Module#spacename
end

197
198
199
200
201
202
203
204
205
206
207
208
209
210
211

# File 'utils.rb', line 197

def self.deep_compact_blank(obj, compact_zero: true)
  obj = case obj
  when Hash
    obj.transform_values { |v| deep_compact_blank(v, compact_zero:) }
       .compact
  when Array
    obj.filter_map { |v| deep_compact_blank(v, compact_zero:) }
  else
    obj
  end

  return if obj.blank? || (compact_zero && obj.is_a?(Numeric) && obj.zero?)

  obj
end

150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170

# File 'utils.rb', line 150

def self.deep_stringify_symbols(obj)
  case obj
  when String
    # Escape leading : or \ to avoid confusion with stringified symbols
    # ":foo" -> "\:foo"
    # "\foo" -> "\\foo"
    if obj.start_with?(":", "\\")
      "\\#{obj}"
    else
      obj
    end
  when Symbol
    ":#{obj}"
  when Hash
    obj.to_h { |k, v| [deep_stringify_symbols(k), deep_stringify_symbols(v)] }
  when Array
    obj.map { |v| deep_stringify_symbols(v) }
  else
    obj
  end
end

173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190

# File 'utils.rb', line 173

def self.deep_unstringify_symbols(obj)
  case obj
  when String
    if obj.start_with?("\\")
      obj[1..]
    elsif obj.start_with?(":")
      T.must(obj[1..]).to_sym
    else
      obj
    end
  when Hash
    obj.to_h { |k, v| [deep_unstringify_symbols(k), deep_unstringify_symbols(v)] }
  when Array
    obj.map { |v| deep_unstringify_symbols(v) }
  else
    obj
  end
end

33
34
35
36
37
38
39
40
41

# File 'utils.rb', line 33

def self.demodulize(path)
  raise ArgumentError, "No constant path provided" if path.nil?

  if (i = path.rindex("::"))
    T.must(path[(i + 2)..])
  else
    path
  end
end

65
66
67

# File 'utils.rb', line 65

def self.full_name?(full_name)
  full_name.count("/") == 2
end

38
39
40

# File 'utils/git_repository.rb', line 38

def self.git_branch(repo = Pathname.pwd, safe: true)
  GitRepository.new(Pathname(repo)).branch_name(safe:)
end

13
14
15
16
17

# File 'utils/git_repository.rb', line 13

def self.git_head(repo = Pathname.pwd, length: nil, safe: true)
  return git_short_head(repo, length:) if length

  GitRepository.new(Pathname(repo)).head_ref(safe:)
end

27
28
29

# File 'utils/git_repository.rb', line 27

def self.git_short_head(repo = Pathname.pwd, length: nil, safe: true)
  GitRepository.new(Pathname(repo)).short_head_ref(length:, safe:)
end

44
45
46
47
48

# File 'utils.rb', line 44

def self.name_from_full_name(full_name)
  _, _, name = full_name.split("/", 3)

  name || full_name
end

51
52
53

# File 'utils.rb', line 51

def self.name_or_token(formula_or_cask)
  formula_or_cask.is_a?(Cask::Cask) ? formula_or_cask.token : formula_or_cask.name
end

91
92
93
94
95
96
97
98
99
100

# File 'utils.rb', line 91

def self.parse_author!(author)
  match_data = /^(?<name>[^<]+?)[ \t]*<(?<email>[^>]+?)>$/.match(author)
  if match_data
    name = match_data[:name]
    email = match_data[:email]
  end
  raise UsageError, "Unable to parse name and email." if name.blank? && email.blank?

  { name: T.must(name), email: T.must(email) }
end

75
76
77
78
79
80
81
82
83
84
85
86
87
88

# File 'utils.rb', line 75

def self.pluralize(stem, count, plural: "s", singular: "", include_count: false)
  case stem
  when "formula"
    plural = "e"
  when "dependency", "try"
    stem = stem.delete_suffix("y")
    plural = "ies"
    singular = "y"
  end

  prefix = include_count ? "#{count}" : ""
  suffix = (count == 1) ? singular : plural
  "#{prefix}#{stem}#{suffix}"
end

88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126

# File 'utils/popen.rb', line 88

def self.popen(args, mode, options = {}, &_block)
  # `brew prof --vernier` uses this to avoid inheriting Vernier's active
  # native collector state through `IO.popen("-")` fork paths.
  if ENV["HOMEBREW_SPAWN_SYSTEM"] == "1"
    options[:err] ||= File::NULL unless ENV["HOMEBREW_STDERR"]
    IO.popen(args, mode, options) do |pipe|
      return pipe.read unless block_given?

      return yield pipe
    end
  end

  IO.popen("-", mode) do |pipe|
    if pipe
      return pipe.read unless block_given?

      yield pipe
    else
      options[:err] ||= File::NULL unless ENV["HOMEBREW_STDERR"]
      cmd = if args[0].is_a? Hash
        args[1]
      else
        args[0]
      end
      begin
        exec(*args, options)
      rescue Errno::ENOENT
        $stderr.puts "brew: command not found: #{cmd}" if options[:err] != :close
        exit! 127
      rescue SystemCallError => e
        if options[:err] != :close
          require "utils"
          $stderr.puts "brew: exec failed (#{Utils.demodulize(e.class.name)}): #{cmd}"
        end
        exit! 1
      end
    end
  end
end

17
18
19
20
21
22

# File 'utils/popen.rb', line 17

def self.popen_read(*args, safe: false, **options, &block)
  output = popen(args, "rb", options, &block)
  return output if !safe || $CHILD_STATUS.success?

  raise ErrorDuringExecution.new(args, status: $CHILD_STATUS, output: [[:stdout, T.cast(output, String)]])
end

44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65

# File 'utils/popen.rb', line 44

def self.popen_write(*args, safe: false, **options, &_block)
  output = ""
  popen(args, "w+b", options) do |pipe|
    # 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
    pipe.wait_readable

    # Capture the rest of the output
    output += pipe.read
    output.freeze
  end
  return output if !safe || $CHILD_STATUS.success?

  raise ErrorDuringExecution.new(args, status: $CHILD_STATUS, output: [[:stdout, output]])
end

9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35

# File 'utils/fork.rb', line 9

def self.rewrite_child_error(child_error)
  # The error class name comes from the forked child's serialised JSON.
  # rubocop:disable Sorbet/ConstantsFromStrings
  inner_class = Object.const_get(child_error["json_class"])
  # rubocop:enable Sorbet/ConstantsFromStrings
  error = if child_error["cmd"] && inner_class == ErrorDuringExecution
    ErrorDuringExecution.new(child_error["cmd"],
                             status: child_error["status"],
                             output: child_error["output"])
  elsif child_error["cmd"] && inner_class == BuildError
    # We fill `BuildError#formula` and `BuildError#options` in later,
    # when we rescue this in `FormulaInstaller#build`.
    BuildError.new(nil, child_error["cmd"], child_error["args"], child_error["env"])
  elsif inner_class == Interrupt
    Interrupt.new
  else
    # Everything other error in the child just becomes a RuntimeError.
    RuntimeError.new <<~EOS
      An exception occurred within a child process:
#{inner_class}: #{child_error["m"]}    EOS
  end

  error.set_backtrace child_error["b"]

  error
end

133
134
135

# File 'utils.rb', line 133

def self.safe_filename(basename)
  basename.gsub(SAFE_FILENAME_REGEX, "")
end

128
129
130

# File 'utils.rb', line 128

def self.safe_filename?(basename)
  !SAFE_FILENAME_REGEX.match?(basename)
end

44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128

# File 'utils/fork.rb', line 44

def self.safe_fork(directory: nil, yield_parent: false, &_blk)
  require "json/add/exception"

  block = proc do |tmpdir|
    UNIXServerExt.open("#{tmpdir}/socket") do |server|
      read, write = IO.pipe

      pid = fork do
        # bootsnap doesn't like these forked processes
        ENV["HOMEBREW_NO_BOOTSNAP"] = "1"
        error_pipe = server.path
        ENV["HOMEBREW_ERROR_PIPE"] = error_pipe
        server.close
        read.close
        write.fcntl(Fcntl::F_SETFD, Fcntl::FD_CLOEXEC)

        Process::UID.change_privilege(Process.euid) if Process.euid != Process.uid

        yield(error_pipe)
      # This could be any type of exception, so rescue them all.
      rescue Exception => e # rubocop:disable Lint/RescueException
        error_hash = JSON.parse e.to_json

        # Special case: We need to recreate ErrorDuringExecutions
        # for proper error messages and because other code expects
        # to rescue them further down.
        if e.is_a?(ErrorDuringExecution)
          error_hash["cmd"] = e.cmd
          error_hash["status"] = if e.status.is_a?(Process::Status)
            {
              exitstatus: e.exitstatus,
              termsig:    e.termsig,
            }
          else
            e.status
          end
          error_hash["output"] = e.output
        end

        write.puts error_hash.to_json
        write.close

        exit!
      else
        exit!(true)
      end

      begin
        yield(nil) if yield_parent

        begin
          socket = server.accept_nonblock
        rescue Errno::EAGAIN, Errno::EWOULDBLOCK, Errno::ECONNABORTED, Errno::EPROTO, Errno::EINTR
          retry unless Process.waitpid(pid, Process::WNOHANG)
        else
          socket.send_io(write)
          socket.close
        end
        write.close
        data = read.read
        read.close
        Process.waitpid(pid) unless socket.nil?
      rescue Interrupt
        Process.waitpid(pid)
      end

      # 130 is the exit status for a process interrupted via Ctrl-C.
      raise Interrupt if $CHILD_STATUS.exitstatus == 130
      raise Interrupt if $CHILD_STATUS.termsig == Signal.list["INT"]

      if data.present?
        error_hash = JSON.parse(data.lines.fetch(0))
        raise rewrite_child_error(error_hash)
      end

      raise ChildProcessError, $CHILD_STATUS unless $CHILD_STATUS.success?
    end
  end

  if directory
    block.call(directory)
  else
    Dir.mktmpdir("homebrew-fork", HOMEBREW_TEMP, &block)
  end
end

32
33
34

# File 'utils/popen.rb', line 32

def self.safe_popen_read(*args, **options, &block)
  popen_read(*args, safe: true, **options, &block)
end

75
76
77

# File 'utils/popen.rb', line 75

def self.safe_popen_write(*args, **options, &block)
  popen_write(*args, safe: true, **options, &block)
end

56
57
58
59
60
61

# File 'utils.rb', line 56

def self.tap_from_full_name(full_name)
  user, repository, name = full_name.split("/", 3)
  return unless name

  "#{user}/#{repository}"
end

112
113
114
115
116
117
118
119
120
121
122

# File 'utils.rb', line 112

def self.underscore(camel_cased_word)
  return camel_cased_word.to_s unless /[A-Z-]|::/.match?(camel_cased_word)

  word = camel_cased_word.to_s.gsub("::", "/")
  word.gsub!(/([A-Z])(?=[A-Z][a-z])|([a-z\d])(?=[A-Z])/) do
    T.must(::Regexp.last_match(1) || ::Regexp.last_match(2)) << "_"
  end
  word.tr!("-", "_")
  word.downcase!
  word
end