45bfd2b94a
Provides feature parity between the block and non-block forms of inreplace by creating a four-argument version of the non-block form, where the fourth argument is an optional Boolean value, defaulting to true, which specifies whether a failed inreplace should cause an InreplaceError error to be raised. The fourth argument is passed along to StringInreplaceExtension#gsub!, which already supports an optional audit_result argument. This resolves the Catch-22 that single replacements aren't permissible in the block form (in that they now cause `brew audit` to complain), but the audit_result argument is not available in the non-block form. Closes #552. Signed-off-by: ilovezfs <ilovezfs@icloud.com>
41 lines
1.3 KiB
Ruby
41 lines
1.3 KiB
Ruby
module Utils
|
|
class InreplaceError < RuntimeError
|
|
def initialize(errors)
|
|
super errors.inject("inreplace failed\n") { |s, (path, errs)|
|
|
s << "#{path}:\n" << errs.map { |e| " #{e}\n" }.join
|
|
}
|
|
end
|
|
end
|
|
|
|
module Inreplace
|
|
# Sometimes we have to change a bit before we install. Mostly we
|
|
# prefer a patch but if you need the `prefix` of this formula in the
|
|
# patch you have to resort to `inreplace`, because in the patch
|
|
# you don't have access to any var defined by the formula. Only
|
|
# HOMEBREW_PREFIX is available in the embedded patch.
|
|
# inreplace supports regular expressions.
|
|
# <pre>inreplace "somefile.cfg", /look[for]what?/, "replace by #{bin}/tool"</pre>
|
|
def inreplace(paths, before = nil, after = nil, audit_result = true)
|
|
errors = {}
|
|
|
|
Array(paths).each do |path|
|
|
s = File.open(path, "rb", &:read).extend(StringInreplaceExtension)
|
|
|
|
if before.nil? && after.nil?
|
|
yield s
|
|
else
|
|
after = after.to_s if Symbol === after
|
|
s.gsub!(before, after, audit_result)
|
|
end
|
|
|
|
errors[path] = s.errors if s.errors.any?
|
|
|
|
Pathname(path).atomic_write(s)
|
|
end
|
|
|
|
raise InreplaceError.new(errors) if errors.any?
|
|
end
|
|
module_function :inreplace
|
|
end
|
|
end
|