summaryrefslogtreecommitdiff
path: root/lib/rubygems/core_ext/kernel_warn.rb
blob: 1291e72870f52bbd069d9a10ca18556250af8684 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
# 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)

    module_function define_method(:warn) {|*messages, uplevel: nil|
      if uplevel
        while (loc, = caller_locations(uplevel, 1); loc && loc.path.start_with?(path))
          uplevel += 1
        end
        original_warn.call(*messages, uplevel: uplevel)
      else
        original_warn.call(*messages)
      end
    }
  end
end