Contents

begin
  require "binding_of_caller"
rescue LoadError => e
  puts "could not load binding_of_caller. please make sure it is included in the Gemfile."
  raise e
end

class NameError
  begin
    require "active_support/core_ext/name_error"

    if method_defined?(:missing_name)
      def missing_name_without_did_you_mean
        if /undefined local variable or method/ !~ original_message
          $1 if /((::)?([A-Z]\w*)(::[A-Z]\w*)*)$/ =~ original_message
        end
      end

      alias missing_name_with_did_you_mean missing_name
      alias missing_name missing_name_without_did_you_mean
    end
  rescue LoadError
  end

  original_set_backtrace = instance_method(:set_backtrace)

  define_method :set_backtrace do |*args|
    unless Thread.current[:__did_you_mean_exception_lock]
      Thread.current[:__did_you_mean_exception_lock] = true
      begin
        @__did_you_mean_bindings_stack = binding.callers.drop(1)
      ensure
        Thread.current[:__did_you_mean_exception_lock] = false
      end
    end
    original_set_backtrace.bind(self).call(*args)
  end
  
  def __did_you_mean_bindings_stack
    @__did_you_mean_bindings_stack || []
  end

  def frame_binding
    @frame_binding ||= __did_you_mean_bindings_stack.first
  end

  def to_s_with_did_you_mean
    original_message + did_you_mean?.to_s
  end

  alias original_message to_s
  alias             to_s to_s_with_did_you_mean

  def did_you_mean?
    method_finder.did_you_mean? if not method_finder.empty?
  end

  def method_finder
    @method_finder ||= DidYouMean.strategies[self.class.to_s].build(self)
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
did_you_mean-0.5.0	lib/did_you_mean/core_ext/name_error.rb
did_you_mean-0.4.0	lib/did_you_mean/core_ext/name_error.rb