Add methods in FormulaCop to find block nodes

This commit is contained in:
Gautham Goli 2017-06-01 00:52:49 +05:30
parent b6e1dde7c9
commit 134da5b8c2

View File

@ -167,12 +167,18 @@ module RuboCop
nil
end
# Returns an array of block nodes named block_name inside node
# Returns an array of block nodes of depth first order named block_name below node
def find_blocks(node, block_name)
return if node.nil?
node.each_child_node(:block).select { |block_node| block_name == block_node.method_name }
end
# Returns an array of block nodes of any depth below node in AST
def find_all_blocks(node, block_name)
return if node.nil?
node.each_descendant(:block).select { |block_node| block_name == block_node.method_name}
end
# Returns a method definition node with method_name
def find_method_def(node, method_name)
return if node.nil?
@ -250,7 +256,7 @@ module RuboCop
# Returns the array of arguments of the method_node
def parameters(method_node)
return unless method_node.send_type?
return unless method_node.send_type? || method_node.block_type?
method_node.method_args
end