forked from rubocop/rubocop
/
access_modifier_declarations.rb
212 lines (187 loc) · 5.79 KB
/
access_modifier_declarations.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
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
# frozen_string_literal: true
module RuboCop
module Cop
module Style
# Access modifiers should be declared to apply to a group of methods
# or inline before each method, depending on configuration.
# EnforcedStyle config covers only method definitions.
# Applications of visibility methods to symbols can be controlled
# using AllowModifiersOnSymbols config.
#
# @safety
# Autocorrection is not safe, because the visibility of dynamically
# defined methods can vary depending on the state determined by
# the group access modifier.
#
# @example EnforcedStyle: group (default)
# # bad
# class Foo
#
# private def bar; end
# private def baz; end
#
# end
#
# # good
# class Foo
#
# private
#
# def bar; end
# def baz; end
#
# end
#
# @example EnforcedStyle: inline
# # bad
# class Foo
#
# private
#
# def bar; end
# def baz; end
#
# end
#
# # good
# class Foo
#
# private def bar; end
# private def baz; end
#
# end
#
# @example AllowModifiersOnSymbols: true (default)
# # good
# class Foo
#
# private :bar, :baz
#
# end
#
# @example AllowModifiersOnSymbols: false
# # bad
# class Foo
#
# private :bar, :baz
#
# end
class AccessModifierDeclarations < Base
extend AutoCorrector
include ConfigurableEnforcedStyle
include RangeHelp
GROUP_STYLE_MESSAGE = [
'`%<access_modifier>s` should not be',
'inlined in method definitions.'
].join(' ')
INLINE_STYLE_MESSAGE = [
'`%<access_modifier>s` should be',
'inlined in method definitions.'
].join(' ')
RESTRICT_ON_SEND = %i[private protected public module_function].freeze
# @!method access_modifier_with_symbol?(node)
def_node_matcher :access_modifier_with_symbol?, <<~PATTERN
(send nil? {:private :protected :public :module_function} (sym _))
PATTERN
def on_send(node)
return unless node.access_modifier?
return if node.parent&.pair_type?
return if allow_modifiers_on_symbols?(node)
if offense?(node)
add_offense(node.loc.selector) do |corrector|
autocorrect(corrector, node)
end
opposite_style_detected
else
correct_style_detected
end
end
private
def autocorrect(corrector, node)
case style
when :group
def_node = find_corresponding_def_node(node)
return unless def_node
remove_node(corrector, def_node)
remove_node(corrector, node)
insert_def(corrector, node, def_node.source)
when :inline
remove_node(corrector, node)
select_grouped_def_nodes(node).each do |grouped_def_node|
insert_inline_modifier(corrector, grouped_def_node, node.method_name)
end
end
end
def allow_modifiers_on_symbols?(node)
cop_config['AllowModifiersOnSymbols'] && access_modifier_with_symbol?(node)
end
def offense?(node)
(group_style? && access_modifier_is_inlined?(node)) ||
(inline_style? && access_modifier_is_not_inlined?(node))
end
def group_style?
style == :group
end
def inline_style?
style == :inline
end
def access_modifier_is_inlined?(node)
node.arguments.any?
end
def access_modifier_is_not_inlined?(node)
!access_modifier_is_inlined?(node)
end
def message(range)
access_modifier = range.source
if group_style?
format(GROUP_STYLE_MESSAGE, access_modifier: access_modifier)
elsif inline_style?
format(INLINE_STYLE_MESSAGE, access_modifier: access_modifier)
end
end
def find_corresponding_def_node(node)
if access_modifier_with_symbol?(node)
method_name = node.arguments.first.value
node.parent.each_child_node(:def).find do |child|
child.method?(method_name)
end
else
node.arguments.first
end
end
def find_argument_less_modifier_node(node)
node.parent.each_child_node(:send).find do |child|
child.method?(node.method_name) && child.arguments.empty?
end
end
def select_grouped_def_nodes(node)
node.right_siblings.take_while do |sibling|
!(sibling.send_type? && sibling.bare_access_modifier_declaration?)
end.select(&:def_type?)
end
def insert_def(corrector, node, source)
argument_less_modifier_node = find_argument_less_modifier_node(node)
if argument_less_modifier_node
corrector.insert_after(argument_less_modifier_node, "\n\n#{source}")
else
corrector.insert_before(
node.each_ancestor(:block, :class, :module).first.location.end,
"#{node.method_name}\n\n#{source}\n"
)
end
end
def insert_inline_modifier(corrector, node, modifier_name)
corrector.insert_before(node, "#{modifier_name} ")
end
def remove_node(corrector, node)
corrector.remove(
range_by_whole_lines(
node.location.expression,
include_final_newline: true
)
)
end
end
end
end
end