blob: 6c5b387e88c0882816c46a217bde29e297540303 (
plain)
1
2
3
4
5
6
7
8
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
36
37
38
39
40
41
42
43
44
45
46
|
# frozen_string_literal: true
# `uplevel` keyword argument of Kernel#warn is available since ruby 2.5.
if RUBY_VERSION >= "2.5"
module Kernel
path = "#{__dir__}/" # Frames to be skipped start with this path.
# Suppress "method redefined" warning
original_warn = instance_method(:warn)
Module.new {define_method(:warn, original_warn)}
original_warn = method(:warn)
module_function define_method(:warn) {|*messages, **kw|
unless uplevel = kw[:uplevel]
return original_warn.call(*messages, **kw)
end
# Ensure `uplevel` fits a `long`
uplevel, = [uplevel].pack("l!").unpack("l!")
if uplevel >= 0
start = 0
while uplevel >= 0
loc, = caller_locations(start, 1)
unless loc
# No more backtrace
start += uplevel
break
end
start += 1
unless loc.path.start_with?(path)
# Non-rubygems frames
uplevel -= 1
end
end
uplevel = start
end
kw[:uplevel] = uplevel
original_warn.call(*messages, **kw)
}
end
end
|