forked from rubocop/rubocop
/
inverse_methods.rb
197 lines (168 loc) · 6.6 KB
/
inverse_methods.rb
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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
# frozen_string_literal: true
module RuboCop
module Cop
module Style
# This cop check for usages of not (`not` or `!`) called on a method
# when an inverse of that method can be used instead.
# Methods that can be inverted by a not (`not` or `!`) should be defined
# in `InverseMethods`
# Methods that are inverted by inverting the return
# of the block that is passed to the method should be defined in
# `InverseBlocks`
#
# @example
# # bad
# !foo.none?
# !foo.any? { |f| f.even? }
# !foo.blank?
# !(foo == bar)
# foo.select { |f| !f.even? }
# foo.reject { |f| f != 7 }
#
# # good
# foo.none?
# foo.blank?
# foo.any? { |f| f.even? }
# foo != bar
# foo == bar
# !!('foo' =~ /^\w+$/)
# !(foo.class < Numeric) # Checking class hierarchy is allowed
# # Blocks with guard clauses are ignored:
# foo.select do |f|
# next if f.zero?
# f != 1
# end
class InverseMethods < Cop
include IgnoredNode
include RangeHelp
MSG = 'Use `%<inverse>s` instead of inverting `%<method>s`.'
CLASS_COMPARISON_METHODS = %i[<= >= < >].freeze
EQUALITY_METHODS = %i[== != =~ !~ <= >= < >].freeze
NEGATED_EQUALITY_METHODS = %i[!= !~].freeze
CAMEL_CASE = /[A-Z]+[a-z]+/.freeze
def self.autocorrect_incompatible_with
[Style::Not]
end
def_node_matcher :inverse_candidate?, <<~PATTERN
{
(send $(send $(...) $_ $...) :!)
(send (block $(send $(...) $_) $...) :!)
(send (begin $(send $(...) $_ $...)) :!)
}
PATTERN
def_node_matcher :inverse_block?, <<~PATTERN
(block $(send (...) $_) ... { $(send ... :!)
$(send (...) {:!= :!~} ...)
(begin ... $(send ... :!))
(begin ... $(send (...) {:!= :!~} ...))
})
PATTERN
def on_send(node)
return if part_of_ignored_node?(node)
inverse_candidate?(node) do |_method_call, lhs, method, rhs|
return unless inverse_methods.key?(method)
return if possible_class_hierarchy_check?(lhs, rhs, method)
return if negated?(node)
add_offense(node,
message: format(MSG, method: method,
inverse: inverse_methods[method]))
end
end
def on_block(node)
inverse_block?(node) do |_method_call, method, block|
return unless inverse_blocks.key?(method)
return if negated?(node) && negated?(node.parent)
return if node.each_node(:next).any?
# Inverse method offenses inside of the block of an inverse method
# offense, such as `y.reject { |key, _value| !(key =~ /c\d/) }`,
# can cause auto-correction to apply improper corrections.
ignore_node(block)
add_offense(node,
message: format(MSG, method: method,
inverse: inverse_blocks[method]))
end
end
def autocorrect(node)
if node.block_type?
correct_inverse_block(node)
elsif node.send_type?
correct_inverse_method(node)
end
end
def correct_inverse_method(node)
method_call, _lhs, method, _rhs = inverse_candidate?(node)
return unless method_call && method
lambda do |corrector|
corrector.remove(not_to_receiver(node, method_call))
corrector.replace(method_call.loc.selector,
inverse_methods[method].to_s)
remove_end_parenthesis(corrector, node, method, method_call)
end
end
def correct_inverse_block(node)
method_call, method, block = inverse_block?(node)
lambda do |corrector|
corrector.replace(method_call.loc.selector,
inverse_blocks[method].to_s)
correct_inverse_selector(block, corrector)
end
end
def correct_inverse_selector(block, corrector)
selector_loc = block.loc.selector
selector = selector_loc.source
if NEGATED_EQUALITY_METHODS.include?(selector.to_sym)
selector[0] = '='
corrector.replace(selector_loc, selector)
else
if block.loc.dot
range = dot_range(block.loc)
corrector.remove(range)
end
corrector.remove(selector_loc)
end
end
private
def inverse_methods
@inverse_methods ||= cop_config['InverseMethods']
.merge(cop_config['InverseMethods'].invert)
end
def inverse_blocks
@inverse_blocks ||= cop_config['InverseBlocks']
.merge(cop_config['InverseBlocks'].invert)
end
def negated?(node)
node.parent.respond_to?(:method?) && node.parent.method?(:!)
end
def not_to_receiver(node, method_call)
Parser::Source::Range.new(node.loc.expression.source_buffer,
node.loc.selector.begin_pos,
method_call.loc.expression.begin_pos)
end
def end_parentheses(node, method_call)
Parser::Source::Range.new(node.loc.expression.source_buffer,
method_call.loc.expression.end_pos,
node.loc.expression.end_pos)
end
# When comparing classes, `!(Integer < Numeric)` is not the same as
# `Integer > Numeric`.
def possible_class_hierarchy_check?(lhs, rhs, method)
CLASS_COMPARISON_METHODS.include?(method) &&
(camel_case_constant?(lhs) ||
(rhs.size == 1 &&
camel_case_constant?(rhs.first)))
end
def camel_case_constant?(node)
node.const_type? && node.source =~ CAMEL_CASE
end
def dot_range(loc)
range_between(loc.dot.begin_pos, loc.expression.end_pos)
end
def remove_end_parenthesis(corrector, node, method, method_call)
return unless EQUALITY_METHODS.include?(method) ||
method_call.parent.begin_type?
corrector.remove(end_parentheses(node, method_call))
end
end
end
end
end