2018-09-23 23:14:43 -04:00
|
|
|
# frozen_string_literal: true
|
|
|
|
|
|
|
|
if RUBY_VERSION >= "2.5"
|
|
|
|
module Kernel
|
|
|
|
path = "#{__dir__}/"
|
|
|
|
original_warn = instance_method(:warn)
|
|
|
|
Module.new {define_method(:warn, original_warn)}
|
|
|
|
original_warn = method(:warn)
|
|
|
|
|
2018-09-24 09:04:40 -04:00
|
|
|
module_function define_method(:warn) {|*messages, uplevel: nil|
|
2018-09-23 23:14:43 -04:00
|
|
|
if uplevel
|
|
|
|
while (loc, = caller_locations(uplevel, 1); loc && loc.path.start_with?(path))
|
|
|
|
uplevel += 1
|
|
|
|
end
|
2018-09-24 09:04:40 -04:00
|
|
|
original_warn.call(*messages, uplevel: uplevel)
|
2018-09-23 23:14:43 -04:00
|
|
|
else
|
2018-09-24 09:04:40 -04:00
|
|
|
original_warn.call(*messages)
|
2018-09-23 23:14:43 -04:00
|
|
|
end
|
|
|
|
}
|
|
|
|
end
|
|
|
|
end
|