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
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
|
# File 'lib/node_mutation/adapter/parser.rb', line 146
def child_node_range(node, child_name)
direct_child_name, nested_child_name = child_name.to_s.split('.', 2)
if node.is_a?(Array)
if direct_child_name =~ INDEX_REGEXP
child_node = node[direct_child_name.to_i]
raise NodeMutation::MethodNotSupported,
"#{direct_child_name} is not supported for #{get_source(node)}" unless child_node
return child_node_range(child_node, nested_child_name) if nested_child_name
return NodeMutation::Struct::Range.new(child_node.loc.expression.begin_pos, child_node.loc.expression.end_pos)
end
raise NodeMutation::MethodNotSupported,
"#{direct_child_name} is not supported for #{get_source(node)}" unless node.respond_to?(direct_child_name)
child_node = node.send(direct_child_name)
return child_node_range(child_node, nested_child_name) if nested_child_name
return NodeMutation::Struct::Range.new(child_node.loc.expression.begin_pos, child_node.loc.expression.end_pos)
end
case [node.type, child_name.to_sym]
when %i[block pipes], %i[def parentheses], %i[defs parentheses]
if node.arguments.empty?
nil
else
NodeMutation::Struct::Range.new(
node.arguments.first.loc.expression.begin_pos - 1,
node.arguments.last.loc.expression.end_pos + 1
)
end
when %i[arg name], %i[class name], %i[const name], %i[cvar name], %i[casgn name], %i[def name], %i[defs name],
%i[gvar name], %i[ivar name], %i[lvar name]
NodeMutation::Struct::Range.new(node.loc.name.begin_pos, node.loc.name.end_pos)
when %i[const double_colon]
NodeMutation::Struct::Range.new(node.loc.double_colon.begin_pos, node.loc.double_colon.end_pos)
when %i[defs dot]
NodeMutation::Struct::Range.new(node.loc.operator.begin_pos, node.loc.operator.end_pos) if node.loc.operator
when %i[defs self]
NodeMutation::Struct::Range.new(node.loc.operator.begin_pos - 'self'.length, node.loc.operator.begin_pos)
when %i[float value], %i[int value], %i[sym value]
NodeMutation::Struct::Range.new(node.loc.expression.begin_pos, node.loc.expression.end_pos)
when %i[lvasgn variable], %i[ivasgn variable], %i[cvasgn variable], %i[gvasgn variable]
NodeMutation::Struct::Range.new(node.loc.name.begin_pos, node.loc.name.end_pos)
when %i[send dot], %i[csend dot]
NodeMutation::Struct::Range.new(node.loc.dot.begin_pos, node.loc.dot.end_pos) if node.loc.dot
when %i[send message], %i[csend message]
if node.loc.operator
NodeMutation::Struct::Range.new(node.loc.selector.begin_pos, node.loc.operator.end_pos)
else
NodeMutation::Struct::Range.new(node.loc.selector.begin_pos, node.loc.selector.end_pos)
end
when %i[send parentheses], %i[csend parentheses]
if node.loc.begin && node.loc.end
NodeMutation::Struct::Range.new(node.loc.begin.begin_pos, node.loc.end.end_pos)
end
when %i[str value]
NodeMutation::Struct::Range.new(node.loc.expression.begin_pos + 1, node.loc.expression.end_pos - 1)
else
raise NodeMutation::MethodNotSupported,
"#{direct_child_name} is not supported for #{get_source(node)}" unless node.respond_to?(direct_child_name)
child_node = node.send(direct_child_name)
return child_node_range(child_node, nested_child_name) if nested_child_name
return nil if child_node.nil?
if child_node.is_a?(Parser::AST::Node)
return(
NodeMutation::Struct::Range.new(child_node.loc.expression.begin_pos, child_node.loc.expression.end_pos)
)
end
return nil if child_node.empty?
return(
NodeMutation::Struct::Range.new(
child_node.first.loc.expression.begin_pos,
child_node.last.loc.expression.end_pos
)
)
end
end
|