blob: 955bff1be63fd04a2e817a38cd898ad5e3f8fa50 (
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
|
require_relative "../spell_checker"
module DidYouMean
class KeyErrorChecker
def initialize(key_error)
@key = key_error.key
@keys = key_error.receiver.keys
end
def corrections
@corrections ||= exact_matches.empty? ? SpellChecker.new(dictionary: @keys).correct(@key).map(&:inspect) : exact_matches
end
private
def exact_matches
@exact_matches ||= @keys.select { |word| @key == word.to_s }.map { |obj| format_object(obj) }
end
def format_object(symbol_or_object)
if symbol_or_object.is_a?(Symbol)
":#{symbol_or_object}"
else
symbol_or_object.to_s
end
end
end
end
|