summaryrefslogtreecommitdiff
path: root/lib/prism/polyfill/warn.rb
blob: 76a4264623ba21896fdce5e699a8955c26ddd925 (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
# frozen_string_literal: true

# Polyfill for Kernel#warn with the category parameter. Not all Ruby engines
# have Method#parameters implemented, so we check the arity instead if
# necessary.
if (method = Kernel.instance_method(:warn)).respond_to?(:parameters) ? method.parameters.none? { |_, name| name == :category } : (method.arity == -1)
  Kernel.prepend(
    Module.new {
      def warn(*msgs, uplevel: nil, category: nil) # :nodoc:
        case uplevel
        when nil
          super(*msgs)
        when Integer
          super(*msgs, uplevel: uplevel + 1)
        else
          super(*msgs, uplevel: uplevel.to_int + 1)
        end
      end
    }
  )

  Object.prepend(
    Module.new {
      def warn(*msgs, uplevel: nil, category: nil) # :nodoc:
        case uplevel
        when nil
          super(*msgs)
        when Integer
          super(*msgs, uplevel: uplevel + 1)
        else
          super(*msgs, uplevel: uplevel.to_int + 1)
        end
      end
    }
  )
end