54 lines
1.3 KiB
Ruby
54 lines
1.3 KiB
Ruby
require_relative "formatter"
|
|
|
|
module ErrorHighlight
|
|
module CoreExt
|
|
# This is a marker to let `DidYouMean::Correctable#original_message` skip
|
|
# the following method definition of `to_s`.
|
|
# See https://github.com/ruby/did_you_mean/pull/152
|
|
SKIP_TO_S_FOR_SUPER_LOOKUP = true
|
|
private_constant :SKIP_TO_S_FOR_SUPER_LOOKUP
|
|
|
|
def to_s
|
|
msg = super.dup
|
|
|
|
locs = backtrace_locations
|
|
return msg unless locs
|
|
|
|
loc = locs.first
|
|
return msg unless loc
|
|
begin
|
|
node = RubyVM::AbstractSyntaxTree.of(loc, keep_script_lines: true)
|
|
opts = {}
|
|
|
|
case self
|
|
when NoMethodError, NameError
|
|
opts[:point_type] = :name
|
|
opts[:name] = name
|
|
when TypeError, ArgumentError
|
|
opts[:point_type] = :args
|
|
end
|
|
|
|
spot = ErrorHighlight.spot(node, **opts)
|
|
|
|
rescue SyntaxError
|
|
rescue SystemCallError # file not found or something
|
|
rescue ArgumentError # eval'ed code
|
|
end
|
|
|
|
if spot
|
|
points = ErrorHighlight.formatter.message_for(spot)
|
|
msg << points if !msg.include?(points)
|
|
end
|
|
|
|
msg
|
|
end
|
|
end
|
|
|
|
NameError.prepend(CoreExt)
|
|
|
|
# The extension for TypeError/ArgumentError is temporarily disabled due to many test failures
|
|
|
|
#TypeError.prepend(CoreExt)
|
|
#ArgumentError.prepend(CoreExt)
|
|
end
|